def get_starting_resources(
civ_group: GenieCivilizationGroup) -> list[ForwardRef]:
"""
Returns the starting resources of a civ.
"""
resource_amounts = []
civ_id = civ_group.get_id()
dataset = civ_group.data
civ_lookup_dict = internal_name_lookups.get_civ_lookups(
dataset.game_version)
civ_name = civ_lookup_dict[civ_id][0]
# Find starting resource amounts
food_amount = civ_group.civ["resources"][91].get_value()
wood_amount = civ_group.civ["resources"][92].get_value()
gold_amount = civ_group.civ["resources"][93].get_value()
stone_amount = civ_group.civ["resources"][94].get_value()
# Find civ unique starting resources
tech_tree = civ_group.get_tech_tree_effects()
for effect in tech_tree:
type_id = effect.get_type()
if type_id != 1:
continue
resource_id = effect["attr_a"].get_value()
amount = effect["attr_d"].get_value()
if resource_id == 91:
food_amount += amount
elif resource_id == 92:
wood_amount += amount
elif resource_id == 93:
gold_amount += amount
elif resource_id == 94:
stone_amount += amount
food_ref = f"{civ_name}.FoodStartingAmount"
food_raw_api_object = RawAPIObject(food_ref, "FoodStartingAmount",
dataset.nyan_api_objects)
food_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
food_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Food"].get_nyan_object()
food_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
food_raw_api_object.add_raw_member(
"amount", food_amount, "engine.util.resource.ResourceAmount")
food_forward_ref = ForwardRef(civ_group, food_ref)
resource_amounts.append(food_forward_ref)
wood_ref = f"{civ_name}.WoodStartingAmount"
wood_raw_api_object = RawAPIObject(wood_ref, "WoodStartingAmount",
dataset.nyan_api_objects)
wood_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
wood_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Wood"].get_nyan_object()
wood_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
wood_raw_api_object.add_raw_member(
"amount", wood_amount, "engine.util.resource.ResourceAmount")
wood_forward_ref = ForwardRef(civ_group, wood_ref)
resource_amounts.append(wood_forward_ref)
gold_ref = f"{civ_name}.GoldStartingAmount"
gold_raw_api_object = RawAPIObject(gold_ref, "GoldStartingAmount",
dataset.nyan_api_objects)
gold_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
gold_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Gold"].get_nyan_object()
gold_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
gold_raw_api_object.add_raw_member(
"amount", gold_amount, "engine.util.resource.ResourceAmount")
gold_forward_ref = ForwardRef(civ_group, gold_ref)
resource_amounts.append(gold_forward_ref)
stone_ref = f"{civ_name}.StoneStartingAmount"
stone_raw_api_object = RawAPIObject(stone_ref, "StoneStartingAmount",
dataset.nyan_api_objects)
stone_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
stone_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Stone"].get_nyan_object()
stone_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
stone_raw_api_object.add_raw_member(
"amount", stone_amount, "engine.util.resource.ResourceAmount")
stone_forward_ref = ForwardRef(civ_group, stone_ref)
resource_amounts.append(stone_forward_ref)
civ_group.add_raw_api_object(food_raw_api_object)
civ_group.add_raw_api_object(wood_raw_api_object)
civ_group.add_raw_api_object(gold_raw_api_object)
civ_group.add_raw_api_object(stone_raw_api_object)
return resource_amounts
def get_starting_resources(
civ_group: GenieCivilizationGroup) -> list[ForwardRef]:
"""
Returns the starting resources of a civ.
"""
resource_amounts = []
civ_id = civ_group.get_id()
dataset = civ_group.data
civ_lookup_dict = internal_name_lookups.get_civ_lookups(
dataset.game_version)
civ_name = civ_lookup_dict[civ_id][0]
# Find starting resource amounts
food_amount = civ_group.civ["resources"][91].value
carbon_amount = civ_group.civ["resources"][92].value
nova_amount = civ_group.civ["resources"][93].value
ore_amount = civ_group.civ["resources"][94].value
# Find civ unique starting resources
tech_tree = civ_group.get_tech_tree_effects()
for effect in tech_tree:
type_id = effect.get_type()
if type_id != 1:
continue
resource_id = effect["attr_a"].value
amount = effect["attr_d"].value
if resource_id == 91:
food_amount += amount
elif resource_id == 92:
carbon_amount += amount
elif resource_id == 93:
nova_amount += amount
elif resource_id == 94:
ore_amount += amount
food_ref = f"{civ_name}.FoodStartingAmount"
food_raw_api_object = RawAPIObject(food_ref, "FoodStartingAmount",
dataset.nyan_api_objects)
food_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
food_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Food"].get_nyan_object()
food_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
food_raw_api_object.add_raw_member(
"amount", food_amount, "engine.util.resource.ResourceAmount")
food_forward_ref = ForwardRef(civ_group, food_ref)
resource_amounts.append(food_forward_ref)
carbon_ref = f"{civ_name}.CarbonStartingAmount"
carbon_raw_api_object = RawAPIObject(carbon_ref,
"CarbonStartingAmount",
dataset.nyan_api_objects)
carbon_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
carbon_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Carbon"].get_nyan_object()
carbon_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
carbon_raw_api_object.add_raw_member(
"amount", carbon_amount, "engine.util.resource.ResourceAmount")
carbon_forward_ref = ForwardRef(civ_group, carbon_ref)
resource_amounts.append(carbon_forward_ref)
nova_ref = f"{civ_name}.NovaStartingAmount"
nova_raw_api_object = RawAPIObject(nova_ref, "NovaStartingAmount",
dataset.nyan_api_objects)
nova_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
nova_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Nova"].get_nyan_object()
nova_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
nova_raw_api_object.add_raw_member(
"amount", nova_amount, "engine.util.resource.ResourceAmount")
nova_forward_ref = ForwardRef(civ_group, nova_ref)
resource_amounts.append(nova_forward_ref)
ore_ref = f"{civ_name}.OreStartingAmount"
ore_raw_api_object = RawAPIObject(ore_ref, "OreStartingAmount",
dataset.nyan_api_objects)
ore_raw_api_object.add_raw_parent(
"engine.util.resource.ResourceAmount")
civ_location = ForwardRef(civ_group,
civ_lookup_dict[civ_group.get_id()][0])
ore_raw_api_object.set_location(civ_location)
resource = dataset.pregen_nyan_objects[
"util.resource.types.Ore"].get_nyan_object()
ore_raw_api_object.add_raw_member(
"type", resource, "engine.util.resource.ResourceAmount")
ore_raw_api_object.add_raw_member(
"amount", ore_amount, "engine.util.resource.ResourceAmount")
ore_forward_ref = ForwardRef(civ_group, ore_ref)
resource_amounts.append(ore_forward_ref)
civ_group.add_raw_api_object(food_raw_api_object)
civ_group.add_raw_api_object(carbon_raw_api_object)
civ_group.add_raw_api_object(nova_raw_api_object)
civ_group.add_raw_api_object(ore_raw_api_object)
return resource_amounts
def setup_tech_tree(civ_group: GenieCivilizationGroup) -> list[ForwardRef]:
"""
Patches standard techs and units out of Research and Create.
"""
patches = []
civ_id = civ_group.get_id()
dataset = civ_group.data
name_lookup_dict = internal_name_lookups.get_entity_lookups(
dataset.game_version)
tech_lookup_dict = internal_name_lookups.get_tech_lookups(
dataset.game_version)
civ_lookup_dict = internal_name_lookups.get_civ_lookups(
dataset.game_version)
civ_name = civ_lookup_dict[civ_id][0]
disabled_techs = {}
disabled_entities = {}
tech_tree = civ_group.get_tech_tree_effects()
for effect in tech_tree:
type_id = effect.get_type()
if type_id == 101:
patches.extend(
AoCTechSubprocessor.tech_cost_modify_effect(
civ_group, effect))
continue
if type_id == 103:
patches.extend(
AoCTechSubprocessor.tech_time_modify_effect(
civ_group, effect))
continue
if type_id != 102:
continue
# Get tech id
tech_id = int(effect["attr_d"].get_value())
# Check what the purpose of the tech is
if tech_id in dataset.unit_unlocks.keys():
unlock_tech = dataset.unit_unlocks[tech_id]
unlocked_line = unlock_tech.get_unlocked_line()
train_location_id = unlocked_line.get_train_location_id()
if isinstance(unlocked_line, GenieBuildingLineGroup):
train_location = dataset.unit_lines[train_location_id]
else:
train_location = dataset.building_lines[train_location_id]
if train_location in disabled_entities.keys():
disabled_entities[train_location].append(unlocked_line)
else:
disabled_entities[train_location] = [unlocked_line]
elif tech_id in dataset.civ_boni.keys():
# Disables civ boni of other civs
continue
elif tech_id in dataset.tech_groups.keys():
tech_group = dataset.tech_groups[tech_id]
if tech_group.is_researchable():
research_location_id = tech_group.get_research_location_id(
)
research_location = dataset.building_lines[
research_location_id]
if research_location in disabled_techs.keys():
disabled_techs[research_location].append(tech_group)
else:
disabled_techs[research_location] = [tech_group]
else:
continue
for train_location, entities in disabled_entities.items():
train_location_id = train_location.get_head_unit_id()
train_location_name = name_lookup_dict[train_location_id][0]
patch_target_ref = f"{train_location_name}.Create"
patch_target_forward_ref = ForwardRef(train_location,
patch_target_ref)
# Wrapper
wrapper_name = f"Disable{train_location_name}CreatablesWrapper"
wrapper_ref = f"{civ_name}.{wrapper_name}"
wrapper_location = ForwardRef(civ_group, civ_name)
wrapper_raw_api_object = RawAPIObject(wrapper_ref, wrapper_name,
dataset.nyan_api_objects,
wrapper_location)
wrapper_raw_api_object.add_raw_parent("engine.util.patch.Patch")
# Nyan patch
nyan_patch_name = f"Disable{train_location_name}Creatables"
nyan_patch_ref = f"{civ_name}.{wrapper_name}.{nyan_patch_name}"
nyan_patch_location = ForwardRef(civ_group, wrapper_ref)
nyan_patch_raw_api_object = RawAPIObject(nyan_patch_ref,
nyan_patch_name,
dataset.nyan_api_objects,
nyan_patch_location)
nyan_patch_raw_api_object.add_raw_parent(
"engine.util.patch.NyanPatch")
nyan_patch_raw_api_object.set_patch_target(
patch_target_forward_ref)
entities_forward_refs = []
for entity in entities:
entity_id = entity.get_head_unit_id()
game_entity_name = name_lookup_dict[entity_id][0]
disabled_ref = f"{game_entity_name}.CreatableGameEntity"
disabled_forward_ref = ForwardRef(entity, disabled_ref)
entities_forward_refs.append(disabled_forward_ref)
nyan_patch_raw_api_object.add_raw_patch_member(
"creatables", entities_forward_refs,
"engine.ability.type.Create", MemberOperator.SUBTRACT)
patch_forward_ref = ForwardRef(civ_group, nyan_patch_ref)
wrapper_raw_api_object.add_raw_member("patch", patch_forward_ref,
"engine.util.patch.Patch")
civ_group.add_raw_api_object(wrapper_raw_api_object)
civ_group.add_raw_api_object(nyan_patch_raw_api_object)
wrapper_forward_ref = ForwardRef(civ_group, wrapper_ref)
patches.append(wrapper_forward_ref)
for research_location, techs in disabled_techs.items():
research_location_id = research_location.get_head_unit_id()
research_location_name = name_lookup_dict[research_location_id][0]
patch_target_ref = f"{research_location_name}.Research"
patch_target_forward_ref = ForwardRef(research_location,
patch_target_ref)
# Wrapper
wrapper_name = f"Disable{research_location_name}ResearchablesWrapper"
wrapper_ref = f"{civ_name}.{wrapper_name}"
wrapper_location = ForwardRef(civ_group, civ_name)
wrapper_raw_api_object = RawAPIObject(wrapper_ref, wrapper_name,
dataset.nyan_api_objects,
wrapper_location)
wrapper_raw_api_object.add_raw_parent("engine.util.patch.Patch")
# Nyan patch
nyan_patch_name = f"Disable{research_location_name}Researchables"
nyan_patch_ref = f"{civ_name}.{wrapper_name}.{nyan_patch_name}"
nyan_patch_location = ForwardRef(civ_group, wrapper_ref)
nyan_patch_raw_api_object = RawAPIObject(nyan_patch_ref,
nyan_patch_name,
dataset.nyan_api_objects,
nyan_patch_location)
nyan_patch_raw_api_object.add_raw_parent(
"engine.util.patch.NyanPatch")
nyan_patch_raw_api_object.set_patch_target(
patch_target_forward_ref)
entities_forward_refs = []
for tech_group in techs:
tech_id = tech_group.get_id()
tech_name = tech_lookup_dict[tech_id][0]
disabled_ref = f"{tech_name}.ResearchableTech"
disabled_forward_ref = ForwardRef(tech_group, disabled_ref)
entities_forward_refs.append(disabled_forward_ref)
nyan_patch_raw_api_object.add_raw_patch_member(
"researchables", entities_forward_refs,
"engine.ability.type.Research", MemberOperator.SUBTRACT)
patch_forward_ref = ForwardRef(civ_group, nyan_patch_ref)
wrapper_raw_api_object.add_raw_member("patch", patch_forward_ref,
"engine.util.patch.Patch")
civ_group.add_raw_api_object(wrapper_raw_api_object)
civ_group.add_raw_api_object(nyan_patch_raw_api_object)
wrapper_forward_ref = ForwardRef(civ_group, wrapper_ref)
patches.append(wrapper_forward_ref)
return patches
