-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Introduced enforcement of AI cost limits via a new automatic method - This new algorithm does not require manual maintenance - It supports all units and races by default, even the mod ones - It is possible it is CPU heavy / makes AI turns longer - Requires testing - Fixed a bug that caused the Tooltip text to be hidden if Dynamic Costs were disabled - Added a new Tooltip text for Dynamic Costs if the capacity has reached its maximum amount - Technical changes: - New method `is_faction_punishable(faction)` added to the CBAC lib API - New method `is_hero(unit_key)` added to the CBAC lib API
- Loading branch information
1 parent
63e6fd2
commit 3cf38a1
Showing
7 changed files
with
193 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| local cbac = core:get_static_object("cbac"); | ||
|
|
||
| -- UTILS -- | ||
|
|
||
| function shuffle(tbl) | ||
| for i = #tbl, 2, -1 do | ||
| local j = cm:random_number(i); | ||
| tbl[i], tbl[j] = tbl[j], tbl[i]; | ||
| end | ||
| return tbl; | ||
| end | ||
|
|
||
| function get_table_keys(tbl) | ||
| cbac:log("get_table_keys " .. #tbl); | ||
|
|
||
| local keys = {}; | ||
| for key, _ in pairs(tbl) do | ||
| table.insert(keys, key); | ||
| end | ||
|
|
||
| return keys; | ||
| end | ||
|
|
||
| -- AI -- | ||
|
|
||
| local function generate_recruitment_pool(faction) | ||
| cbac:log("Generating recruitment pool"); | ||
| local recruitment_pool = {}; | ||
|
|
||
| local characters = faction:character_list(); | ||
| for i = 0, characters:num_items() - 1 do | ||
| if cm:char_is_mobile_general_with_army(characters:item_at(i)) then | ||
| local unit_list = characters:item_at(i):military_force():unit_list(); | ||
| for i = 1, unit_list:num_items() - 1 do | ||
| local unit = unit_list:item_at(i); | ||
| local unit_cost = cbac:get_unit_cost(unit); | ||
|
|
||
| if not cbac:is_hero(unit:unit_key()) and unit_cost > 0 then | ||
| recruitment_pool[unit:unit_key()] = unit_cost; | ||
| end | ||
| end | ||
| end | ||
| end | ||
|
|
||
| cbac:log("Recruitment pool generated. Size: " .. #get_table_keys(recruitment_pool)); | ||
| return recruitment_pool; | ||
| end | ||
|
|
||
| local function replace_unit(old_unit_key, new_unit_key, character_lookup) | ||
| cbac:log("Replacing " .. old_unit_key .. " with " .. new_unit_key); | ||
| cm:remove_unit_from_character(character_lookup, old_unit_key); | ||
| cm:grant_unit_to_character(character_lookup, new_unit_key); | ||
| end | ||
|
|
||
| local function downgrade_unit_and_get_savings(unit_list, unit_index, recruitment_pool, character_lookup) | ||
| local unit = unit_list:item_at(unit_index); | ||
| local unit_cost = cbac:get_unit_cost(unit); | ||
|
|
||
| if unit_cost > 0 then | ||
| cbac:log("Downgrading unit? " .. unit:unit_key()); | ||
| local recruitment_pool_keys = get_table_keys(recruitment_pool); | ||
| local offset = math.random(0, #recruitment_pool_keys - 1); | ||
|
|
||
| for i = 0, #recruitment_pool_keys - 1 do | ||
| local rec_unit_key = recruitment_pool_keys[(i + offset) % #recruitment_pool_keys + 1]; | ||
| if recruitment_pool[rec_unit_key] < unit_cost then | ||
| replace_unit(unit:unit_key(), rec_unit_key, character_lookup); | ||
| cbac:log("Yay! Points saved: " .. unit_cost - recruitment_pool[rec_unit_key]); | ||
| return unit_cost - recruitment_pool[rec_unit_key]; | ||
| end | ||
| end | ||
| end | ||
|
|
||
| cbac:log("Nay! No points saved"); | ||
| return 0; | ||
| end | ||
|
|
||
| local function enforce_limit_on_ai_army(character, recruitment_pool) | ||
| local required_savings = cbac:get_army_cost(character) - cbac:get_army_limit(character); | ||
| local army_is_over_limit = true; | ||
|
|
||
| local unit_list = character:military_force():unit_list(); | ||
| local unit_indices = {} | ||
| for i = 1, unit_list:num_items() - 1 do | ||
| table.insert(unit_indices, i); | ||
| end | ||
|
|
||
| unit_indices = shuffle(unit_indices); | ||
| for i = 1, #unit_indices do | ||
| required_savings = required_savings - downgrade_unit_and_get_savings(unit_list, unit_indices[i], recruitment_pool, | ||
| cm:char_lookup_str(character)); | ||
| if required_savings <= 0 then | ||
| cbac:log("This army is now under the cost limit, moving on.") | ||
| army_is_over_limit = false; | ||
| break | ||
| end | ||
| end | ||
|
|
||
| if army_is_over_limit then | ||
| cbac:log("Looped through all units in army, but it is still over limit. Will try again in the next turn!"); | ||
| end | ||
| end | ||
|
|
||
| local function check_ai_force_army_limit(faction, character) | ||
| if cm:char_is_mobile_general_with_army(character) then | ||
| local army_cost = cbac:get_army_cost(character); | ||
| local army_limit = cbac:get_army_limit(character); | ||
|
|
||
| local recruitment_pool = generate_recruitment_pool(faction); | ||
|
|
||
| if army_cost > army_limit then | ||
| cbac:log("AI Army of faction " .. faction:name() .. " is over cost limit (" .. army_cost .. "/" .. army_limit .. | ||
| "). Limits will be enforced!"); | ||
| enforce_limit_on_ai_army(character, recruitment_pool); | ||
| end | ||
| end | ||
| end | ||
|
|
||
| local function check_ai_faction_army_limit(faction) | ||
| cbac:log("Checking faction limit: " .. faction:name()); | ||
| if cbac:is_faction_punisheable(faction) then | ||
| cbac:log("Faction " .. faction:name() .. " is punisheable"); | ||
| local characters = faction:character_list(); | ||
| for i = 0, characters:num_items() - 1 do | ||
| check_ai_force_army_limit(faction, characters:item_at(i)); | ||
| end | ||
| end | ||
| end | ||
|
|
||
| -- LISTENERS -- | ||
|
|
||
| local function add_listeners() | ||
| core:add_listener( | ||
| "ArmyCostLimitsAI", | ||
| "FactionTurnStart", | ||
| function(context) | ||
| return not context:faction():is_human(); | ||
| end, | ||
| function(context) | ||
| check_ai_faction_army_limit(context:faction()); | ||
| end, | ||
| true | ||
| ); | ||
| end | ||
|
|
||
| -- MAIN -- | ||
|
|
||
| local function main() | ||
| add_listeners(); | ||
| end | ||
|
|
||
| main(); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters