def assess_ship_design_role(design):
parts = [get_part_type(partname) for partname in design.parts if partname and get_part_type(partname)]
if any(p.partClass == fo.shipPartClass.colony and p.capacity == 0 for p in parts):
if design.speed > 0:
return ShipRoleType.CIVILIAN_OUTPOST
else:
return ShipRoleType.BASE_OUTPOST
if any(p.partClass == fo.shipPartClass.colony and p.capacity > 0 for p in parts):
if design.speed > 0:
return ShipRoleType.CIVILIAN_COLONISATION
else:
return ShipRoleType.BASE_COLONISATION
if any(p.partClass == fo.shipPartClass.troops for p in parts):
if design.speed > 0:
return ShipRoleType.MILITARY_INVASION
else:
return ShipRoleType.BASE_INVASION
if design.speed == 0:
if not parts or parts[0].partClass == fo.shipPartClass.shields: # ToDo: Update logic for new ship designs
return ShipRoleType.BASE_DEFENSE
else:
return ShipRoleType.INVALID
if design.isArmed or design.hasFighters:
return ShipRoleType.MILITARY
if any(p.partClass == fo.shipPartClass.detection for p in parts):
return ShipRoleType.CIVILIAN_EXPLORATION
else: # if no suitable role found, use as (bad) scout as it still has inherent detection
print "AI Warning: defaulting ship role to 'exploration' for ship with parts: ", design.parts
return ShipRoleType.CIVILIAN_EXPLORATION
def assess_ship_design_role(design):
parts = [get_part_type(partname) for partname in design.parts if partname and get_part_type(partname)]
if any(p.partClass == fo.shipPartClass.colony and p.capacity == 0 for p in parts):
if design.speed > 0:
return ShipRoleType.CIVILIAN_OUTPOST
else:
return ShipRoleType.BASE_OUTPOST
if any(p.partClass == fo.shipPartClass.colony and p.capacity > 0 for p in parts):
if design.speed > 0:
return ShipRoleType.CIVILIAN_COLONISATION
else:
return ShipRoleType.BASE_COLONISATION
if any(p.partClass == fo.shipPartClass.troops for p in parts):
if design.speed > 0:
return ShipRoleType.MILITARY_INVASION
else:
return ShipRoleType.BASE_INVASION
if design.speed == 0:
if not parts or parts[0].partClass == fo.shipPartClass.shields: # ToDo: Update logic for new ship designs
return ShipRoleType.BASE_DEFENSE
else:
return ShipRoleType.INVALID
if design.isArmed or design.hasFighters:
return ShipRoleType.MILITARY
if any(p.partClass == fo.shipPartClass.detection for p in parts):
return ShipRoleType.CIVILIAN_EXPLORATION
else: # if no suitable role found, use as (bad) scout as it still has inherent detection
print "AI Warning: defaulting ship role to 'exploration' for ship with parts: ", design.parts
return ShipRoleType.CIVILIAN_EXPLORATION
def __get_stats_from_ship(self):
"""Read and store combat related stats from ship"""
universe = fo.getUniverse()
ship = universe.getShip(self.__ship_id)
if not ship:
return # TODO: Add some estimate for stealthed ships
if self._consider_refuel:
structure = ship.initialMeterValue(fo.meterType.maxStructure)
shields = ship.initialMeterValue(fo.meterType.maxShield)
else:
structure = ship.initialMeterValue(fo.meterType.structure)
shields = ship.initialMeterValue(fo.meterType.shield)
attacks = {}
fighter_launch_rate = 0
fighter_capacity = 0
fighter_damage = 0
design = ship.design
if design and (ship.isArmed or ship.hasFighters):
meter_choice = fo.meterType.maxCapacity if self._consider_refuel else fo.meterType.capacity
for partname in design.parts:
if not partname:
continue
pc = get_part_type(partname).partClass
if pc == fo.shipPartClass.shortRange:
damage = ship.currentPartMeterValue(meter_choice, partname)
attacks[damage] = attacks.get(damage, 0) + 1
elif pc == fo.shipPartClass.fighterBay:
fighter_launch_rate += ship.currentPartMeterValue(
fo.meterType.capacity, partname)
elif pc == fo.shipPartClass.fighterHangar:
# for hangars, capacity meter is already counting contributions from ALL hangars.
fighter_capacity = ship.currentPartMeterValue(
meter_choice, partname)
part_damage = ship.currentPartMeterValue(
fo.meterType.secondaryStat, partname)
if part_damage != fighter_damage and fighter_damage > 0:
# the C++ code fails also in this regard, so FOCS content *should* not allow this.
# TODO: Depending on future implementation, might actually need to handle this case.
warn(
"Multiple hangar types present on one ship, estimates expected to be wrong."
)
fighter_damage = max(fighter_damage, part_damage)
self._basic_stats = self.BasicStats(attacks, structure, shields)
self._fighter_stats = self.FighterStats(fighter_capacity,
fighter_launch_rate,
fighter_damage)
def get_fighter_capacity_of_fleet(fleet_id):
"""Return current and max fighter capacity
:param fleet_id:
:type fleet_id: int
:return: current and max fighter capacity
"""
universe = fo.getUniverse()
fleet = universe.getFleet(fleet_id)
cur_capacity = 0
max_capacity = 0
ships = (universe.getShip(ship_id) for ship_id in (fleet.shipIDs if fleet else []))
for ship in ships:
design = ship and ship.design
design_parts = design.parts if design and design.hasFighters else []
for partname in design_parts:
part = get_part_type(partname)
if part and part.partClass == fo.shipPartClass.fighterHangar:
cur_capacity += ship.currentPartMeterValue(fo.meterType.capacity, partname)
max_capacity += ship.currentPartMeterValue(fo.meterType.maxCapacity, partname)
return cur_capacity, max_capacity
def get_fighter_capacity_of_fleet(fleet_id):
"""Return current and max fighter capacity
:param fleet_id:
:type fleet_id: int
:return: current and max fighter capacity
"""
universe = fo.getUniverse()
fleet = universe.getFleet(fleet_id)
cur_capacity = 0
max_capacity = 0
ships = (universe.getShip(ship_id) for ship_id in (fleet.shipIDs if fleet else []))
for ship in ships:
design = ship and ship.design
design_parts = design.parts if design and design.hasFighters else []
for partname in design_parts:
part = get_part_type(partname)
if part and part.partClass == fo.shipPartClass.fighterHangar:
cur_capacity += ship.currentPartMeterValue(fo.meterType.capacity, partname)
max_capacity += ship.currentPartMeterValue(fo.meterType.maxCapacity, partname)
return cur_capacity, max_capacity
def __get_stats_from_ship(self):
"""Read and store combat related stats from ship"""
universe = fo.getUniverse()
ship = universe.getShip(self.__ship_id)
if not ship:
return # TODO: Add some estimate for stealthed ships
if self._consider_refuel:
structure = ship.currentMeterValue(fo.meterType.maxStructure)
shields = ship.currentMeterValue(fo.meterType.maxShield)
else:
structure = ship.currentMeterValue(fo.meterType.structure)
shields = ship.currentMeterValue(fo.meterType.shield)
attacks = {}
fighter_launch_rate = 0
fighter_capacity = 0
fighter_damage = 0
design = ship.design
if design and (ship.isArmed or ship.hasFighters):
meter_choice = fo.meterType.maxCapacity if self._consider_refuel else fo.meterType.capacity
for partname in design.parts:
if not partname:
continue
pc = get_part_type(partname).partClass
if pc == fo.shipPartClass.shortRange:
damage = ship.currentPartMeterValue(meter_choice, partname)
attacks[damage] = attacks.get(damage, 0) + 1
elif pc == fo.shipPartClass.fighterBay:
fighter_launch_rate += ship.currentPartMeterValue(fo.meterType.capacity, partname)
elif pc == fo.shipPartClass.fighterHangar:
fighter_capacity += ship.currentPartMeterValue(meter_choice, partname)
part_damage = ship.currentPartMeterValue(fo.meterType.secondaryStat, partname)
if part_damage != fighter_damage and fighter_damage > 0:
# the C++ code fails also in this regard, so FOCS content *should* not allow this.
# TODO: Depending on future implementation, might actually need to handle this case.
print "WARNING: Multiple hangar types present on one ship, estimates expected to be wrong."
fighter_damage = max(fighter_damage, part_damage)
self._basic_stats = self.BasicStats(attacks, structure, shields)
self._fighter_stats = self.FighterStats(fighter_capacity, fighter_launch_rate, fighter_damage)
