AIs.ais_rebel.Rebel._colonize_planets() - Code Metrics - Inspection of "Ai obsolete designs" - ospaceteam/outerspace - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — master (#230)

by Marek

created 2018-07-15 22:13 UTC

AIs.ais_rebel.Rebel._colonize_planets() A

↳ Parent: AIs.ais_rebel

Complexity

Conditions

Size

Total Lines	12
Code Lines	11

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	3
eloc	11
nop	3
dl	0
loc	12
rs	9.85
c	0
b	0
f	0

#
#  Copyright 2001 - 2016 Ludek Smid [http://www.ospace.net/]
#
#  This file is part of Outer Space.
#
#  Outer Space is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  Outer Space is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with Outer Space; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#
from ige import log
from ige.ospace import Const
from ige.ospace import Rules
from ige.ospace import Utils
from ige.ospace import TechHandlers

import ai_tools as tool
from ai import AI

import copy, random, math

class Rebel(AI):
    def __init__(self, client):
        super(Rebel, self).__init__(client)
        tool.doRelevance(self.data, self.client, self.db, 10)
        self.designs = {}

    def offense_manager(self):
        # rebel is not suited for attacks, yet
        pass

    def _get_idle_planets(self, system):
        idle_planets = set()
        for planet_id in self.data.myProdPlanets & set(system.planets):
            planet = self.db[planet_id]
            if not getattr(planet, 'prodQueue', None):
                idle_planets.add(planet_id)
                continue
        return idle_planets

    def _fill_planets(self, system, idle_planets):
        system_stats = tool.getSystemStructStats(self.data, self.client, self.db, system.oid)
        for planet_id in copy.copy(idle_planets):
            planet = self.db[planet_id]
            prod_tech_id = Rules.Tech.RESCENTRE1 if planet.plMin < 100 else Rules.Tech.FACTORY1
            prod_tech = self.client.getFullTechInfo(prod_tech_id)
            # now we ignore all already build structures, and try to satisfy
            # outpost/fact or outpost/labs ration [on free slots]
            if planet.plSlots > len(planet.slots):
                if system_stats.en > prod_tech.operEn and system_stats.bio > prod_tech.operWorkers / 100:
                    planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet.oid,
                        prod_tech.id, 1, planet_id, prod_tech.id < 1000, 0, Const.OID_NONE)
                    idle_planets.remove(planet_id)
                else:
                    planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                        Rules.Tech.OUTPOST1, 1, planet_id, Rules.Tech.OUTPOST1 < 1000, 0, Const.OID_NONE)
                    idle_planets.remove(planet_id)

    def _colonize_planets(self, system, idle_planets):
        toColonize = self.data.freePlanets & set(system.planets)
        # colonize remaining planets
        for planet_id in copy.copy(idle_planets):
            planet = self.db[planet_id]
            if toColonize:
                targetID = toColonize.pop()
                target = self.db[targetID]
                planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                    Rules.Tech.OUTPOST1, 1,
                    targetID, Rules.Tech.OUTPOST1 < 1000, 0, Const.OID_NONE)
                idle_planets.remove(planet_id)

    def _build_ships(self, system, idle_planets):
        for planet_id in copy.copy(idle_planets):
            planet = self.db[planet_id]
            systemFleets = getattr(system, 'fleets', [])
            has_colony = False
            has_scouts = False
            shared_system = len(set(system.planets) & self.data.otherPlanets) > 0
            for fleetID in systemFleets:
                fleet = self.db[fleetID]
                if getattr(fleet, 'owner', Const.OID_NONE) == self.player.oid:
                    has_colony |= tool.fleetContains(fleet, {self.designs["colony"]:1})
                    has_scouts |= tool.fleetContains(fleet, {self.designs["scout"]:1})
            if not has_colony:
                planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                    self.designs["colony"], 1, planet_id, self.designs["colony"] < 1000, 0, Const.OID_NONE)
            elif not has_scouts:
                planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                    self.designs["scout"], 1, planet_id, self.designs["scout"] < 1000, 0, Const.OID_NONE)
            elif Rules.Tech.UPGRADESHIPS in self.player.techs and self.player.fleetUpgradePool < 2000:
                planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                    Rules.Tech.UPGRADESHIPS, 1, planet_id, Rules.Tech.UPGRADESHIPS < 1000, 0, Const.OID_NONE)

            elif not shared_system:
                # build fighters
                planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                    self.designs["fighter"], 1, planet_id, self.designs["fighter"] < 1000, 0, Const.OID_NONE)

    def _planet_manager(self):
        for planet_id in self.data.myPlanets:
            tool.sortStructures(self.client, self.db, planet_id)
        for system_id in self.data.mySystems:
            system = self.db[system_id]
            idle_planets = self._get_idle_planets(system)
            # build production buildings if nothing is needed, or outposts
            self._fill_planets(system, idle_planets)
            self._colonize_planets(system, idle_planets)
            self._build_ships(system, idle_planets)

    def _place_gov_center(self, candidate_planets):
        PLACEHOLDERS = (Rules.Tech.RESCENTRE1, Rules.Tech.FACTORY1)
        for planet_id in candidate_planets:
            planet = self.db[planet_id]
            if planet.prodProd == 0:
                continue
            gov_placeholder = None
            for struct in [x for x in planet.slots if x[0] in PLACEHOLDERS]:
                if gov_placeholder is not None:
                    planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                        Rules.Tech.GOVCENTER1, 1, planet_id, Rules.Tech.GOVCENTER1 < 1000, 0, gov_placeholder)
                    planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
                        Rules.Tech.OUTPOST1, 1, planet_id, Rules.Tech.OUTPOST1 < 1000, 0, struct[0])
                    return
                gov_placeholder = struct[0]

    def _cancel_gov_tasks(self, planet_ids):
        # cancel all tasks
        for planet_id in planet_ids:
            planet = self.db[planet_id]
            indexes = []
            i = 0
            for task in planet.prodQueue:
                if task.techID in set([1000, 3010, 3011]):
                    indexes.append(i)
                i += 1
            indexes.reverse()
            for index in indexes:
                self.client.cmdProxy.abortConstruction(planet_id, index)

    def _get_current_gov(self):
        gov_position = Const.OID_NONE
        gov_productions = []
        for planet_id in self.data.myPlanets:
            # find build gov center
            planet = self.db[planet_id]
            for struct in planet.slots:
                if struct[0] in set([1000, 3010, 3011]):
                    gov_position = planet_id
                    break
            for task in getattr(planet, 'prodQueue', []):
                if task.techID in set([1000, 3010, 3011]):
                    gov_productions.append(planet_id)
                    break
        return gov_position, gov_productions

    def _empire_manager(self):
        if not Rules.Tech.GOVCENTER1 in self.player.techs.keys():
            return
        candidates = tool.findPopCenterPlanets(self.db, self.data.myPlanets)
        candidate_planets = candidates[:10]
        gov_position, gov_productions = self._get_current_gov()
        if not set(candidate_planets) & (set([gov_position]) | set(gov_productions)):
            self._cancel_gov_tasks(gov_productions)
            self._place_gov_center(candidate_planets)
        return

    def _expansion_manager(self):
        should_repeat = True
        pirate_influenced_systems = tool.findInfluence(self.data, self.client, self.db, Rules.pirateInfluenceRange, self.data.pirateSystems)
        while should_repeat:
            should_repeat = False
            should_repeat |= self._explore(self.designs["scout"])
            safe_systems = (self.data.freeSystems & self.data.relevantSystems) - pirate_influenced_systems
            should_repeat |= self._colonize_free_systems(safe_systems, self.designs["colony"])

    def _identify_basic_designs(self):
        for desID in self.player.shipDesigns:
            design = self.player.shipDesigns[desID]
            if design.name == 'Scout':
                self.designs["scout"] = desID
            elif design.name == 'Fighter':
                self.designs["fighter"] = desID
            elif design.name == 'Bomber':
                self.designs["bomber"] = desID
            elif design.name == 'Colony Ship':
                self.designs["colony"] = desID

    def _identify_advanced_designs(self):
        for desID in self.player.shipDesigns:
            design = self.player.shipDesigns[desID]
            if design.name == 'Fighter 2':
                self.designs["fighter0"] = self.designs["fighter"]
                self.designs["fighter"] = desID
            elif design.name == 'Bomber 2':
                self.designs["bomber0"] = self.designs["bomber"]
                self.designs["bomber"] = desID

    def _create_basic_designs(self):
        if "scout" not in self.designs:
            self.designs["scout"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Scout',
                    Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
                    Rules.Tech.SCANNERMOD0:1, Rules.Tech.FTLENG0:3})
        if "fighter" not in self.designs:
            self.designs["fighter"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Fighter',
                    Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
                    Rules.Tech.CANNON0:2, Rules.Tech.FTLENG0:3})
        if "bomber" not in self.designs:
            self.designs["bomber"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Bomber',
                    Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
                    Rules.Tech.CONBOMB0:1, Rules.Tech.FTLENG0:3})
        if "colony" not in self.designs:
            self.designs["colony"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Colony Ship',
                    Rules.Tech.MEDIUMHULL0, {Rules.Tech.SCOCKPIT0:1,
                    Rules.Tech.COLONYMOD0:1, Rules.Tech.FTLENG0:4})

    def _create_advanced_designs(self):
        needed_tech = set([Rules.Tech.SMALLHULL1, Rules.Tech.SCOCKPIT1, Rules.Tech.CANNON1, Rules.Tech.FTLENG1])
        if needed_tech.issubset(self.player.techs) and "fighter0" not in self.designs:
            self.designs["fighter0"] = self.designs["fighter"]
            self.designs["fighter"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Fighter 2',
                    Rules.Tech.SMALLHULL1, {Rules.Tech.SCOCKPIT1:1,
                    Rules.Tech.CANNON1:2, Rules.Tech.FTLENG1:3})
        needed_tech = set([Rules.Tech.SMALLHULL1, Rules.Tech.SCOCKPIT1, Rules.Tech.CONBOMB1, Rules.Tech.FTLENG1])
        if needed_tech.issubset(self.player.techs) and "bomber0" not in self.designs:
            self.designs["bomber0"] = self.designs["bomber"]
            self.designs["bomber"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Bomber 2',
                    Rules.Tech.SMALLHULL1, {Rules.Tech.SCOCKPIT1:1,
                    Rules.Tech.CONBOMB1:1, Rules.Tech.FTLENG1:3})

    def _upgrade_obsolete_design(self, old_code_name, new_code_name):
        if old_code_name in self.designs:
            old_design_id = self.designs[old_code_name]
            new_design_id = self.designs[new_code_name]
            old_design = self.player.shipDesigns[old_design_id]
            if not old_design.upgradeTo:
                self.player.shipDesigns, self.player.stratRes, tasksUpgraded, self.player.prodQueues = \
                        self.client.cmdProxy.upgradeShipDesign(self.player.oid, old_design_id, new_design_id)

    def _get_service_centers(self):
        service_centers = []
        for planet in [self.db[planetID] for planetID in self.data.myPlanets]:
            # assumption: if it repairs, it also upgrades
            if not planet.repairShip: continue
            service_centers.append(planet.compOf)
        return service_centers

    def _get_outdated_fleets(self, obsolete_designs):
        outdated_fleets = []
        for fleet in [self.db[fleetID] for fleetID in self.data.idleFleets]:
            if set(obsolete_designs).intersection([ship[0] for ship in fleet.ships]):
                outdated_fleets.append(fleet)
        return outdated_fleets

    def _recall_to_upgrade(self, service_centers, outdated_fleets, obsolete_designs):
        for fleet in outdated_fleets:
            if fleet.orbiting in service_centers: continue
            subfleet_sheet = {}.fromkeys(obsolete_designs, 0)
            max_range = tool.subfleetMaxRange(self.client, self.db, subfleet_sheet, fleet.oid)
            nearest = tool.findNearest(self.db, fleet, service_centers, max_range)
            if not nearest: continue
            nearest_service = nearest[0]
            fleet, new_fleet, my_fleets = tool.orderPartFleet(self.client, self.db,
                subfleet_sheet, True, fleet.oid, Const.FLACTION_MOVE, nearest_service, None)

    def _upgrade_manager(self):
        UPGRADABLE = ["fighter", "bomber"]
        OBSOL_FUNC = lambda x: x + "0"
        obsoleted = set(map(OBSOL_FUNC, UPGRADABLE)).intersection(self.designs)
        obsolete_designs = [self.designs[obsol] for obsol in obsoleted]
        if not obsolete_designs: return

        service_centers = self._get_service_centers()
        outdated_fleets = self._get_outdated_fleets(obsolete_designs)
        self._recall_to_upgrade(service_centers, outdated_fleets, obsolete_designs)
        for code_name in UPGRADABLE:
            self._upgrade_obsolete_design(OBSOL_FUNC(code_name), code_name)

    def _ship_design_manager(self):
        self._identify_basic_designs()
        self._identify_advanced_designs()
        self._create_basic_designs()
        self._create_advanced_designs()

    def research_manager(self):
        researchable = set()
        if len(self.player.rsrchQueue) < 2:
            for tech_id in self.player.techs.keys():
                tech = self.client.getTechInfo(tech_id)
                improvement = self.player.techs[tech_id]
                if improvement < Rules.techMaxImprovement and\
                        improvement < tech.maxImprovement:
                    researchable.add(tech_id)
            for tech_id in self.client.getAllTechIDs():
                tech = self.client.getTechInfo(tech_id)
                if not hasattr(tech, "partialData") or not hasattr(tech, 'researchMod'):
                    continue
                else:
                    researchable.add(tech_id)
            for task in self.player.rsrchQueue:
                researchable -= set([task.techID])
            # some less useful technologies for AI - deprioritize
            lessTechs = set([1102, 1104, 1107, 1110, 1112, 1404, 1510, 1800, 1802])
            if len(researchable - (lessTechs | set([1990, 1991, 1992]))) > 0:
                researchable -= lessTechs
            # do not advance, for now
            researchable -= set([1990, 1991, 1992])
            if len(researchable) > 0:
                if Rules.Tech.OUTPOST1 in researchable:
                    self.player.rsrchQueue = self.client.cmdProxy.startResearch(self.player.oid, Rules.Tech.OUTPOST1)
                    return
                possibilities = list(researchable)
                random.shuffle(possibilities)
                tech_id = possibilities[0]
                self.player.rsrchQueue = self.client.cmdProxy.startResearch(self.player.oid, tech_id)

    def _help_system(self):
        tool.doDanger(self.data, self.client, self.db)
        pirate_influenced_systems = tool.findInfluence(self.data, self.client, self.db, Rules.pirateInfluenceRange, self.data.pirateSystems)
        one_fighter_mp = self.player.shipDesigns[self.designs["fighter"]].combatPwr
        for system_id in set(self.data.endangeredSystems) - set(pirate_influenced_systems):
            mil_pwr, shipQuantity = self.data.endangeredSystems[system_id]
            mil_pwr = -mil_pwr
            if system_id in self.data.myMPPerSystem:
                mil_pwr += self.data.myMPPerSystem[system_id]
            if mil_pwr < 0:
                system = self.db[system_id]
                nearest = tool.findNearest(self.db, system, self.data.mySystems, 99999, 20)[1:]
                for temp_id in nearest:
                    if temp_id in self.data.myMPPerSystem:
                        temp_mp = self.data.myMPPerSystem[temp_id]
                    else:
                        temp_mp = 0
                        self.data.myMPPerSystem[temp_id] = 0
                    if temp_id in self.data.endangeredSystems:
                        a, b = self.data.endangeredSystems[temp_id]
                        temp_mp -= a * 1.5
                        if temp_mp <= 0: continue
                    orig = temp_mp = min(-mil_pwr, temp_mp) * 1.25
                    # this is just prototype, working only with Fighters
                    quantity = int(math.ceil(temp_mp / float(one_fighter_mp)))
                    if quantity == 0:
                        continue
                    ships_left, mil_pwrSend = tool.orderFromSystem(self.client, self.db,
                        {self.designs["fighter"]:quantity}, temp_id, Const.FLACTION_MOVE, system_id, None)
                    mil_pwr += mil_pwrSend
                    self.data.myMPPerSystem[temp_id] -= mil_pwrSend
                    if mil_pwr > 0: break

    def _return_strays(self):
        for fleetID in self.data.idleFleets:
            fleet = self.db[fleetID]
            # fleets orbiting in systems not belonging to the self.player
            if fleet.orbiting and fleet.orbiting not in self.data.mySystems:
                nearest = tool.findNearest(self.db, fleet, self.data.mySystems, 99999, 1)
                if len(nearest):
                    targetID = nearest[0]
                    tool.orderFleet(self.client, self.db, fleetID, Const.FLACTION_MOVE, targetID, None)

    def defense_manager(self):
        self._help_system()
        self._return_strays()

    def economy_manager(self):
        self._empire_manager()
        self._planet_manager()
        self._expansion_manager()
        self._upgrade_manager()

    def run(self):
        self._ship_design_manager() # this fills self.designs, needs to go first
        self.research_manager()
        self.diplomacy_manager(friendly_types=[Const.T_PLAYER, Const.T_AIPLAYER, Const.T_AIRENPLAYER],
                               pacts=[Const.PACT_ALLOW_CIVILIAN_SHIPS, Const.PACT_ALLOW_TANKING,
                                      Const.PACT_MINOR_SCI_COOP, Const.PACT_MAJOR_SCI_COOP,
                                      Const.PACT_MINOR_CP_COOP, Const.PACT_MAJOR_CP_COOP])
        self.economy_manager()
        self.defense_manager()
        self.offense_manager()


def run(aclient):
    ai = Rebel(aclient)
    ai.run()
    aclient.saveDB()



1			#
2			# Copyright 2001 - 2016 Ludek Smid [http://www.ospace.net/]
3			#
4			# This file is part of Outer Space.
5			#
6			# Outer Space is free software; you can redistribute it and/or modify
7			# it under the terms of the GNU General Public License as published by
8			# the Free Software Foundation; either version 2 of the License, or
9			# (at your option) any later version.
10			#
11			# Outer Space is distributed in the hope that it will be useful,
12			# but WITHOUT ANY WARRANTY; without even the implied warranty of
13			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			# GNU General Public License for more details.
15			#
16			# You should have received a copy of the GNU General Public License
17			# along with Outer Space; if not, write to the Free Software
18			# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19			#
20			from ige import log
21			from ige.ospace import Const
22			from ige.ospace import Rules
23			from ige.ospace import Utils
24			from ige.ospace import TechHandlers
25
26			import ai_tools as tool
27			from ai import AI
28
29			import copy, random, math
30
31			class Rebel(AI):
32			def __init__(self, client):
33			super(Rebel, self).__init__(client)
34			tool.doRelevance(self.data, self.client, self.db, 10)
35			self.designs = {}
36
37			def offense_manager(self):
38			# rebel is not suited for attacks, yet
39			pass
40
41			def _get_idle_planets(self, system):
42			idle_planets = set()
43			for planet_id in self.data.myProdPlanets & set(system.planets):
44			planet = self.db[planet_id]
45			if not getattr(planet, 'prodQueue', None):
46			idle_planets.add(planet_id)
47			continue
48			return idle_planets
49
50			def _fill_planets(self, system, idle_planets):
51			system_stats = tool.getSystemStructStats(self.data, self.client, self.db, system.oid)
52			for planet_id in copy.copy(idle_planets):
53			planet = self.db[planet_id]
54			prod_tech_id = Rules.Tech.RESCENTRE1 if planet.plMin < 100 else Rules.Tech.FACTORY1
55			prod_tech = self.client.getFullTechInfo(prod_tech_id)
56			# now we ignore all already build structures, and try to satisfy
57			# outpost/fact or outpost/labs ration [on free slots]
58			if planet.plSlots > len(planet.slots):
59			if system_stats.en > prod_tech.operEn and system_stats.bio > prod_tech.operWorkers / 100:
60			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet.oid,
61			prod_tech.id, 1, planet_id, prod_tech.id < 1000, 0, Const.OID_NONE)
62			idle_planets.remove(planet_id)
63			else:
64			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
65			Rules.Tech.OUTPOST1, 1, planet_id, Rules.Tech.OUTPOST1 < 1000, 0, Const.OID_NONE)
66			idle_planets.remove(planet_id)
67
68			def _colonize_planets(self, system, idle_planets):
69			toColonize = self.data.freePlanets & set(system.planets)
70			# colonize remaining planets
71			for planet_id in copy.copy(idle_planets):
72			planet = self.db[planet_id]
73			if toColonize:
74			targetID = toColonize.pop()
75			target = self.db[targetID]
76			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
77			Rules.Tech.OUTPOST1, 1,
78			targetID, Rules.Tech.OUTPOST1 < 1000, 0, Const.OID_NONE)
79			idle_planets.remove(planet_id)
80
81			def _build_ships(self, system, idle_planets):
82			for planet_id in copy.copy(idle_planets):
83			planet = self.db[planet_id]
84			systemFleets = getattr(system, 'fleets', [])
85			has_colony = False
86			has_scouts = False
87			shared_system = len(set(system.planets) & self.data.otherPlanets) > 0
88			for fleetID in systemFleets:
89			fleet = self.db[fleetID]
90			if getattr(fleet, 'owner', Const.OID_NONE) == self.player.oid:
91			has_colony \|= tool.fleetContains(fleet, {self.designs["colony"]:1})
92			has_scouts \|= tool.fleetContains(fleet, {self.designs["scout"]:1})
93			if not has_colony:
94			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
95			self.designs["colony"], 1, planet_id, self.designs["colony"] < 1000, 0, Const.OID_NONE)
96			elif not has_scouts:
97			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
98			self.designs["scout"], 1, planet_id, self.designs["scout"] < 1000, 0, Const.OID_NONE)
99			elif Rules.Tech.UPGRADESHIPS in self.player.techs and self.player.fleetUpgradePool < 2000:
100			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
101			Rules.Tech.UPGRADESHIPS, 1, planet_id, Rules.Tech.UPGRADESHIPS < 1000, 0, Const.OID_NONE)
102
103			elif not shared_system:
104			# build fighters
105			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
106			self.designs["fighter"], 1, planet_id, self.designs["fighter"] < 1000, 0, Const.OID_NONE)
107
108			def _planet_manager(self):
109			for planet_id in self.data.myPlanets:
110			tool.sortStructures(self.client, self.db, planet_id)
111			for system_id in self.data.mySystems:
112			system = self.db[system_id]
113			idle_planets = self._get_idle_planets(system)
114			# build production buildings if nothing is needed, or outposts
115			self._fill_planets(system, idle_planets)
116			self._colonize_planets(system, idle_planets)
117			self._build_ships(system, idle_planets)
118
119			def _place_gov_center(self, candidate_planets):
120			PLACEHOLDERS = (Rules.Tech.RESCENTRE1, Rules.Tech.FACTORY1)
121			for planet_id in candidate_planets:
122			planet = self.db[planet_id]
123			if planet.prodProd == 0:
124			continue
125			gov_placeholder = None
126			for struct in [x for x in planet.slots if x[0] in PLACEHOLDERS]:
127			if gov_placeholder is not None:
128			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
129			Rules.Tech.GOVCENTER1, 1, planet_id, Rules.Tech.GOVCENTER1 < 1000, 0, gov_placeholder)
130			planet.prodQueue, self.player.stratRes = self.client.cmdProxy.startConstruction(planet_id,
131			Rules.Tech.OUTPOST1, 1, planet_id, Rules.Tech.OUTPOST1 < 1000, 0, struct[0])
132			return
133			gov_placeholder = struct[0]
134
135			def _cancel_gov_tasks(self, planet_ids):
136			# cancel all tasks
137			for planet_id in planet_ids:
138			planet = self.db[planet_id]
139			indexes = []
140			i = 0
141			for task in planet.prodQueue:
142			if task.techID in set([1000, 3010, 3011]):
143			indexes.append(i)
144			i += 1
145			indexes.reverse()
146			for index in indexes:
147			self.client.cmdProxy.abortConstruction(planet_id, index)
148
149			def _get_current_gov(self):
150			gov_position = Const.OID_NONE
151			gov_productions = []
152			for planet_id in self.data.myPlanets:
153			# find build gov center
154			planet = self.db[planet_id]
155			for struct in planet.slots:
156			if struct[0] in set([1000, 3010, 3011]):
157			gov_position = planet_id
158			break
159			for task in getattr(planet, 'prodQueue', []):
160			if task.techID in set([1000, 3010, 3011]):
161			gov_productions.append(planet_id)
162			break
163			return gov_position, gov_productions
164
165			def _empire_manager(self):
166			if not Rules.Tech.GOVCENTER1 in self.player.techs.keys():
167			return
168			candidates = tool.findPopCenterPlanets(self.db, self.data.myPlanets)
169			candidate_planets = candidates[:10]
170			gov_position, gov_productions = self._get_current_gov()
171			if not set(candidate_planets) & (set([gov_position]) \| set(gov_productions)):
172			self._cancel_gov_tasks(gov_productions)
173			self._place_gov_center(candidate_planets)
174			return
175
176			def _expansion_manager(self):
177			should_repeat = True
178			pirate_influenced_systems = tool.findInfluence(self.data, self.client, self.db, Rules.pirateInfluenceRange, self.data.pirateSystems)
179			while should_repeat:
180			should_repeat = False
181			should_repeat \|= self._explore(self.designs["scout"])
182			safe_systems = (self.data.freeSystems & self.data.relevantSystems) - pirate_influenced_systems
183			should_repeat \|= self._colonize_free_systems(safe_systems, self.designs["colony"])
184
185			def _identify_basic_designs(self):
186			for desID in self.player.shipDesigns:
187			design = self.player.shipDesigns[desID]
188			if design.name == 'Scout':
189			self.designs["scout"] = desID
190			elif design.name == 'Fighter':
191			self.designs["fighter"] = desID
192			elif design.name == 'Bomber':
193			self.designs["bomber"] = desID
194			elif design.name == 'Colony Ship':
195			self.designs["colony"] = desID
196
197			def _identify_advanced_designs(self):
198			for desID in self.player.shipDesigns:
199			design = self.player.shipDesigns[desID]
200			if design.name == 'Fighter 2':
201			self.designs["fighter0"] = self.designs["fighter"]
202			self.designs["fighter"] = desID
203			elif design.name == 'Bomber 2':
204			self.designs["bomber0"] = self.designs["bomber"]
205			self.designs["bomber"] = desID
206
207			def _create_basic_designs(self):
208			if "scout" not in self.designs:
209			self.designs["scout"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Scout',
210			Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
211			Rules.Tech.SCANNERMOD0:1, Rules.Tech.FTLENG0:3})
212			if "fighter" not in self.designs:
213			self.designs["fighter"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Fighter',
214			Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
215			Rules.Tech.CANNON0:2, Rules.Tech.FTLENG0:3})
216			if "bomber" not in self.designs:
217			self.designs["bomber"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Bomber',
218			Rules.Tech.SMALLHULL0, {Rules.Tech.SCOCKPIT0:1,
219			Rules.Tech.CONBOMB0:1, Rules.Tech.FTLENG0:3})
220			if "colony" not in self.designs:
221			self.designs["colony"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Colony Ship',
222			Rules.Tech.MEDIUMHULL0, {Rules.Tech.SCOCKPIT0:1,
223			Rules.Tech.COLONYMOD0:1, Rules.Tech.FTLENG0:4})
224
225			def _create_advanced_designs(self):
226			needed_tech = set([Rules.Tech.SMALLHULL1, Rules.Tech.SCOCKPIT1, Rules.Tech.CANNON1, Rules.Tech.FTLENG1])
227			if needed_tech.issubset(self.player.techs) and "fighter0" not in self.designs:
228			self.designs["fighter0"] = self.designs["fighter"]
229			self.designs["fighter"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Fighter 2',
230			Rules.Tech.SMALLHULL1, {Rules.Tech.SCOCKPIT1:1,
231			Rules.Tech.CANNON1:2, Rules.Tech.FTLENG1:3})
232			needed_tech = set([Rules.Tech.SMALLHULL1, Rules.Tech.SCOCKPIT1, Rules.Tech.CONBOMB1, Rules.Tech.FTLENG1])
233			if needed_tech.issubset(self.player.techs) and "bomber0" not in self.designs:
234			self.designs["bomber0"] = self.designs["bomber"]
235			self.designs["bomber"] = self.client.cmdProxy.addShipDesign(self.player.oid, 'Bomber 2',
236			Rules.Tech.SMALLHULL1, {Rules.Tech.SCOCKPIT1:1,
237			Rules.Tech.CONBOMB1:1, Rules.Tech.FTLENG1:3})
238
239			def _upgrade_obsolete_design(self, old_code_name, new_code_name):
240			if old_code_name in self.designs:
241			old_design_id = self.designs[old_code_name]
242			new_design_id = self.designs[new_code_name]
243			old_design = self.player.shipDesigns[old_design_id]
244			if not old_design.upgradeTo:
245			self.player.shipDesigns, self.player.stratRes, tasksUpgraded, self.player.prodQueues = \
246			self.client.cmdProxy.upgradeShipDesign(self.player.oid, old_design_id, new_design_id)
247
248			def _get_service_centers(self):
249			service_centers = []
250			for planet in [self.db[planetID] for planetID in self.data.myPlanets]:
251			# assumption: if it repairs, it also upgrades
252			if not planet.repairShip: continue
253			service_centers.append(planet.compOf)
254			return service_centers
255
256			def _get_outdated_fleets(self, obsolete_designs):
257			outdated_fleets = []
258			for fleet in [self.db[fleetID] for fleetID in self.data.idleFleets]:
259			if set(obsolete_designs).intersection([ship[0] for ship in fleet.ships]):
260			outdated_fleets.append(fleet)
261			return outdated_fleets
262
263			def _recall_to_upgrade(self, service_centers, outdated_fleets, obsolete_designs):
264			for fleet in outdated_fleets:
265			if fleet.orbiting in service_centers: continue
266			subfleet_sheet = {}.fromkeys(obsolete_designs, 0)
267			max_range = tool.subfleetMaxRange(self.client, self.db, subfleet_sheet, fleet.oid)
268			nearest = tool.findNearest(self.db, fleet, service_centers, max_range)
269			if not nearest: continue
270			nearest_service = nearest[0]
271			fleet, new_fleet, my_fleets = tool.orderPartFleet(self.client, self.db,
272			subfleet_sheet, True, fleet.oid, Const.FLACTION_MOVE, nearest_service, None)
273
274			def _upgrade_manager(self):
275			UPGRADABLE = ["fighter", "bomber"]
276			OBSOL_FUNC = lambda x: x + "0"
277			obsoleted = set(map(OBSOL_FUNC, UPGRADABLE)).intersection(self.designs)
278			obsolete_designs = [self.designs[obsol] for obsol in obsoleted]
279			if not obsolete_designs: return
280
281			service_centers = self._get_service_centers()
282			outdated_fleets = self._get_outdated_fleets(obsolete_designs)
283			self._recall_to_upgrade(service_centers, outdated_fleets, obsolete_designs)
284			for code_name in UPGRADABLE:
285			self._upgrade_obsolete_design(OBSOL_FUNC(code_name), code_name)
286
287			def _ship_design_manager(self):
288			self._identify_basic_designs()
289			self._identify_advanced_designs()
290			self._create_basic_designs()
291			self._create_advanced_designs()
292
293			def research_manager(self):
294			researchable = set()
295			if len(self.player.rsrchQueue) < 2:
296			for tech_id in self.player.techs.keys():
297			tech = self.client.getTechInfo(tech_id)
298			improvement = self.player.techs[tech_id]
299			if improvement < Rules.techMaxImprovement and\
300			improvement < tech.maxImprovement:
301			researchable.add(tech_id)
302			for tech_id in self.client.getAllTechIDs():
303			tech = self.client.getTechInfo(tech_id)
304			if not hasattr(tech, "partialData") or not hasattr(tech, 'researchMod'):
305			continue
306			else:
307			researchable.add(tech_id)
308			for task in self.player.rsrchQueue:
309			researchable -= set([task.techID])
310			# some less useful technologies for AI - deprioritize
311			lessTechs = set([1102, 1104, 1107, 1110, 1112, 1404, 1510, 1800, 1802])
312			if len(researchable - (lessTechs \| set([1990, 1991, 1992]))) > 0:
313			researchable -= lessTechs
314			# do not advance, for now
315			researchable -= set([1990, 1991, 1992])
316			if len(researchable) > 0:
317			if Rules.Tech.OUTPOST1 in researchable:
318			self.player.rsrchQueue = self.client.cmdProxy.startResearch(self.player.oid, Rules.Tech.OUTPOST1)
319			return
320			possibilities = list(researchable)
321			random.shuffle(possibilities)
322			tech_id = possibilities[0]
323			self.player.rsrchQueue = self.client.cmdProxy.startResearch(self.player.oid, tech_id)
324
325			def _help_system(self):
326			tool.doDanger(self.data, self.client, self.db)
327			pirate_influenced_systems = tool.findInfluence(self.data, self.client, self.db, Rules.pirateInfluenceRange, self.data.pirateSystems)
328			one_fighter_mp = self.player.shipDesigns[self.designs["fighter"]].combatPwr
329			for system_id in set(self.data.endangeredSystems) - set(pirate_influenced_systems):
330			mil_pwr, shipQuantity = self.data.endangeredSystems[system_id]
331			mil_pwr = -mil_pwr
332			if system_id in self.data.myMPPerSystem:
333			mil_pwr += self.data.myMPPerSystem[system_id]
334			if mil_pwr < 0:
335			system = self.db[system_id]
336			nearest = tool.findNearest(self.db, system, self.data.mySystems, 99999, 20)[1:]
337			for temp_id in nearest:
338			if temp_id in self.data.myMPPerSystem:
339			temp_mp = self.data.myMPPerSystem[temp_id]
340			else:
341			temp_mp = 0
342			self.data.myMPPerSystem[temp_id] = 0
343			if temp_id in self.data.endangeredSystems:
344			a, b = self.data.endangeredSystems[temp_id]
345			temp_mp -= a * 1.5
346			if temp_mp <= 0: continue
347			orig = temp_mp = min(-mil_pwr, temp_mp) * 1.25
348			# this is just prototype, working only with Fighters
349			quantity = int(math.ceil(temp_mp / float(one_fighter_mp)))
350			if quantity == 0:
351			continue
352			ships_left, mil_pwrSend = tool.orderFromSystem(self.client, self.db,
353			{self.designs["fighter"]:quantity}, temp_id, Const.FLACTION_MOVE, system_id, None)
354			mil_pwr += mil_pwrSend
355			self.data.myMPPerSystem[temp_id] -= mil_pwrSend
356			if mil_pwr > 0: break
357
358			def _return_strays(self):
359			for fleetID in self.data.idleFleets:
360			fleet = self.db[fleetID]
361			# fleets orbiting in systems not belonging to the self.player
362			if fleet.orbiting and fleet.orbiting not in self.data.mySystems:
363			nearest = tool.findNearest(self.db, fleet, self.data.mySystems, 99999, 1)
364			if len(nearest):
365			targetID = nearest[0]
366			tool.orderFleet(self.client, self.db, fleetID, Const.FLACTION_MOVE, targetID, None)
367
368			def defense_manager(self):
369			self._help_system()
370			self._return_strays()
371
372			def economy_manager(self):
373			self._empire_manager()
374			self._planet_manager()
375			self._expansion_manager()
376			self._upgrade_manager()
377
378			def run(self):
379			self._ship_design_manager() # this fills self.designs, needs to go first
380			self.research_manager()
381			self.diplomacy_manager(friendly_types=[Const.T_PLAYER, Const.T_AIPLAYER, Const.T_AIRENPLAYER],
382			pacts=[Const.PACT_ALLOW_CIVILIAN_SHIPS, Const.PACT_ALLOW_TANKING,
383			Const.PACT_MINOR_SCI_COOP, Const.PACT_MAJOR_SCI_COOP,
384			Const.PACT_MINOR_CP_COOP, Const.PACT_MAJOR_CP_COOP])
385			self.economy_manager()
386			self.defense_manager()
387			self.offense_manager()
388
389
390			def run(aclient):
391			ai = Rebel(aclient)
392			ai.run()
393			aclient.saveDB()
394
395

ospaceteam / outerspace

Pull Request — master (#230)

AIs.ais_rebel.Rebel._colonize_planets() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like