ige.ospace.ISystem.ISystem.getSystemCombatBonuses() - Code Metrics - Inspection of "Polishing" - ospaceteam/outerspace - Measure and Improve Code Quality continuously with Scrutinizer

Completed
Pull Request — master (#188)

by Marek
created 2018-06-17 18:02 UTC
ISystem.getSystemCombatBonuses() B

↳ Parent: ige.ospace.ISystem
Complexity

Conditions
Size

Total Lines	13
Code Lines	13
Duplication

Lines	0
Ratio	0 %
Importance

Changes
Metric	Value
cc	6
eloc	13
nop	4
dl	0
loc	13
rs	8
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
#
import random
import copy

from xml.dom.minidom import Node

import ige
import Rules
import Utils

from Const import *
from ige import log
from ige.IObject import IObject
from ige.IDataHolder import IDataHolder

class ISystem(IObject):

    typeID = T_SYSTEM


    def init(self, obj):
        IObject.init(self, obj)
        #
        obj.x = 0.0
        obj.y = 0.0
        obj.planets = []
        obj.fleets = []
        obj.closeFleets = []
        obj.starClass = u'---' # Star clasification
        obj.signature = 100
        # rotation
        #~ obj.dist = 0.0
        #~ obj.dAngle = 0.0
        #~ obj.sAngle = 0.0
        # renaming
        obj.lastNameChng = 0
        # combat
        obj.combatCounter = 0
        # system wide data
        obj.scannerPwrs = {}
        # mine field
        obj.minefield = {}  # for every owner (key) list of minefield (triplet) exists (mine_id, amount, last_deployed)

    def update(self, tran, obj):
        # check existence of all planets
        if 0:
            for planetID in obj.planets:
                if not tran.db.has_key(planetID):
                    log.debug("CONSISTENCY - planet %d from system %d does not exists" % (planetID, obj.oid))
                elif tran.db[planetID].type != T_PLANET:
                    log.debug("CONSISTENCY - planet %d from system %d is not a T_PLANET" % (planetID, obj.oid))
        if not hasattr(obj,'minefield'):
            obj.minefield = {}
        # TODO: only needed for 0.5.72 update
        old_minefields = False
        for ownerID in obj.minefield:
            if obj.minefield[ownerID] and type(obj.minefield[ownerID][0]) is not tuple:
                old_minefields = True
        if old_minefields:
            for ownerID in obj.minefield:
                temp_mine_dict = {}
                for mine_id in obj.minefield[ownerID]:
                    try:
                        temp_mine_dict[mine_id] += 1
                    except KeyError:
                        temp_mine_dict[mine_id] = 1
                # now translate dictionary into list of tuples
                owners_minefields = []
                turn = tran.db[OID_UNIVERSE].turn

                for mine_id, amount in temp_mine_dict.iteritems():
                    owners_minefields.append((mine_id, amount, turn))
                obj.minefield[ownerID] = owners_minefields

        # check that all .fleet are in .closeFleets
        for fleetID in obj.fleets:
            if fleetID not in obj.closeFleets:
                log.debug("CONSISTENCY - fleet %d is in .fleet but not in .closeFleets - adding" % fleetID)
                obj.closeFleets.append(fleetID)
        # check existence of all fleets
        for fleetID in obj.closeFleets:
            if not tran.db.has_key(fleetID):
                log.debug("CONSISTENCY - fleet %d from system %d does not exists" % (fleetID, obj.oid))
            elif tran.db[fleetID].type not in (T_FLEET, T_ASTEROID):

                log.debug("CONSISTENCY - fleet %d from system %d is not a T_FLEET" % (fleetID, obj.oid))
        # delete nonexistent fleets
        index = 0
        while index < len(obj.closeFleets) and obj.closeFleets:
            fleet = tran.db.get(obj.closeFleets[index], None)
            if fleet == None:
                log.debug("CONSISTENCY - fleet %d does not exists" % obj.closeFleets[index])
                fleetID = obj.closeFleets[index]
                obj.closeFleets.remove(fleetID)
                obj.fleets.remove(fleetID)
            else:
                index += 1
        # check compOf
        if not tran.db.has_key(obj.compOf) or tran.db[obj.compOf].type != T_GALAXY:

            log.debug("CONSISTENCY invalid compOf for system", obj.oid)
        # rebuild closeFleets attribute
        old = obj.closeFleets
        obj.closeFleets = []
        for fleetID in old:
            fleet = tran.db.get(fleetID, None)
            if fleet and fleet.closeSystem == obj.oid and fleetID not in obj.closeFleets:
                obj.closeFleets.append(fleetID)
        if old != obj.closeFleets:
            log.debug("System close fleets fixed", obj.oid, old, obj.closeFleets)
        # TODO: remove, no need to start players on random systems
        # try to find starting planets
        #starting = 0
        #free = 1
        #for planetID in obj.planets:
        #    planet = tran.db[planetID]
        #    if planet.plStarting:
        #        starting = planetID
        #    if planet.owner != OID_NONE:
        #        free = 0
        #if starting and free:
        #    # good starting position
        #    #@log.debug("Found starting position", obj.oid, starting)
        #    # get galaxy
        #    galaxy = tran.db[obj.compOf]
        #    if starting not in galaxy.startingPos:
        #        log.debug("Adding to starting positions of galaxy", galaxy.oid)
        #        galaxy.startingPos.append(starting)
        # check if system has planets
        hasHabitable = 0
        for planetID in obj.planets:
            if tran.db[planetID].plSlots > 0:
                hasHabitable = 1
                break
        if (not obj.planets or not hasHabitable) and obj.starClass[0] != "b" and obj.starClass != "wW0":
            log.debug("No planet for system", obj.oid, obj.name, obj.starClass)
            # delete old planets
            for planetID in obj.planets:
                del tran.db[planetID]
            obj.planets = []
            # find matching systems
            avail = []
            for systemID in tran.db[obj.compOf].systems:
                system = tran.db[systemID]
                if system.starClass[1] == obj.starClass[1] \
                    or (obj.starClass[1] == "G" and system.starClass[1] == "F"):
                    ok = 0
                    for planetID in system.planets:
                        planet = tran.db[planetID]
                        if planet.plStarting:
                            ok = 0
                            break
                        if planet.plSlots > 0:
                            ok = 1
                    if ok and system.planets:
                        avail.append(systemID)
            # select random system
            log.debug("Can copy", avail)
            try:
                systemID = random.choice(avail)
                # copy it
                log.debug("Will copy system", systemID)
                nType = Utils.getPlanetNamesType()
                orbit = 1
                for planetID in tran.db[systemID].planets:
                    orig = tran.db[planetID]
                    planet = tran.db[self.createPlanet(tran, obj)]
                    planet.name = Utils.getPlanetName(obj.name, nType, orbit - 1)
                    planet.x = obj.x
                    planet.y = obj.y
                    planet.plDiameter = orig.plDiameter
                    planet.plType = orig.plType
                    planet.plMin = orig.plMin
                    planet.plBio = orig.plBio
                    planet.plEn = orig.plEn
                    planet.plEnv = orig.plEnv
                    planet.plSlots = orig.plSlots
                    planet.plMaxSlots = orig.plMaxSlots
                    planet.plStratRes = 0
                    planet.plDisease = 0
                    planet.plStarting = 0
                    planet.orbit = orbit
                    planet.storPop = 0
                    planet.slots = []
                    orbit += 1
            except:
                log.debug("Copy failed")
        # TODO: remove in version 0.5.73
        if True:
            owners = []
            for planetID in obj.planets:
                planet = tran.db[planetID]
                if planet.owner not in owners + [OID_NONE]:

                    owners.append(planet.owner)
            for owner_id in owners:
                for tech, struct in self.getSystemMineLauncher(tran, obj, owner_id):
                    if not struct[STRUCT_IDX_STATUS] & STRUCT_STATUS_ON:

                        # structure is offline, reset timer
                        self.addMine(tran, obj, owner_id, tech.mineclass, 0)
                        continue
                    efficiency = struct[STRUCT_IDX_HP] / float(tech.maxHP)

                    minenum = int(tech.minenum * efficiency)
                    # by setting minerate to None, we are forcing a creation of one - so it grabs
                    # galaxyTurn instead of universe.turn
                    if self.addMine(tran, obj, owner_id, tech.mineclass, None, None):
                        log.debug('ISystem', 'Mine timer reset for owner %d in system %d' % (owner_id, obj.oid))

    update.public = 0

    def getReferences(self, tran, obj):
        return obj.planets

    getReferences.public = 0

    def getScanInfos(self, tran, obj, scanPwr, player):
        result = IDataHolder()
        results = [result]
        if scanPwr >= Rules.level1InfoScanPwr:
            result._type = T_SCAN

            result.scanPwr = scanPwr
            result.oid = obj.oid
            result.x = obj.x
            result.y = obj.y
            if hasattr(obj, 'destinationOid'):
                result.destinationOid = obj.destinationOid
            # multiply by 1000 to increase accuracy
            #~ result.dist = obj.dist * 1000
            #~ result.dAngle = obj.dAngle * 1000
            #~ result.sAngle = obj.sAngle * 1000
            result.signature = obj.signature
            result.type = obj.type
            result.compOf = obj.compOf
            result.starClass = obj.starClass
        if scanPwr >= Rules.level2InfoScanPwr:
            result.name = obj.name
            result.combatCounter = obj.combatCounter
        if scanPwr >= Rules.level3InfoScanPwr:
            result.planets = obj.planets
            result.owner = obj.owner
            for planetID in obj.planets:
                planet = tran.db[planetID]
                if planet.owner == player: ####### This was player.owner, which made no sense. Hope this change doesn't break something
                    continue
                newPwr = scanPwr * planet.signature / obj.signature
                results.extend(self.cmd(planet).getScanInfos(tran, planet, newPwr, player))
        if scanPwr >= Rules.level4InfoScanPwr:
            result.fleets = obj.fleets
            for fleetID in obj.fleets:
                fleet = tran.db[fleetID]
                if fleet.owner == player:
                    continue
                newPwr = scanPwr * fleet.signature / obj.signature
                results.extend(self.cmd(fleet).getScanInfos(tran, fleet, newPwr, player))
            result.hasmines = 0 # no
            if len(obj.minefield) > 0:
                result.hasmines = 1 # yes
            result.minefield = self.getMines(obj, player.oid) # only shows mines you own
            if len(obj.minefield) > 1 or (len(obj.minefield) == 1 and len(result.minefield) == 0):
                result.hasmines = 2 # yes, and some aren't my mines
        return results

    def processINITPhase(self, tran, obj, data):
        obj.scannerPwrs = {}

        return obj.planets

    processINITPhase.public = 1
    processINITPhase.accLevel = AL_ADMIN


    def processPRODPhase(self, tran, obj, data):
        # mine deployment
        owners = []
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner not in owners + [OID_NONE]:

                owners.append(planet.owner)
        for owner_id in owners:
            self.deployMines(tran, obj, owner_id)
        return obj.planets

    processPRODPhase.public = 1
    processPRODPhase.accLevel = AL_ADMIN

    def processACTIONPhase(self, tran, obj, data):
        # distribute resources
        planets = {}
        # group planets by owner
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner != OID_NONE:

                tmp = planets.get(planet.owner, [])
                tmp.append(planet)
                planets[planet.owner] = tmp
        # group planets if owners are allied
        # TODO
        # process each group
        for owner in planets.keys():
            # skip alone planets
            if len(planets[owner]) < 2:
                continue
            # process each resource
            for resName in ('Bio', 'En'):
                donors = []
                donees = []
                minRes = 'min%s' % resName
                maxRes = 'max%s' % resName
                storRes = 'stor%s' % resName
                donorsSum = 0
                doneesSum = 0
                # put planets into donors/donees
                for planet in planets[owner]:
                    if getattr(planet, storRes) > getattr(planet, minRes):
                        donors.append(planet)
                        donorsSum += getattr(planet, storRes) - getattr(planet, minRes)
                    elif getattr(planet, storRes) < getattr(planet, minRes):
                        donees.append(planet)
                        doneesSum += getattr(planet, minRes) - getattr(planet, storRes)
                #@log.debug('ISystem', obj.oid, 'Donors / donees for %s' % resName, donorsSum, doneesSum)
                # there are requests for donation and there is somebody able to donate
                if doneesSum > 0 and donorsSum > 0:
                    #@log.debug('ISystem', 'Redistributin %s for' % resName, owner)
                    # give
                    balance = 0
                    tmpRatio = min(float(doneesSum) / donorsSum, 1.0)
                    for planet in donees:
                        diff = getattr(planet, minRes) - getattr(planet, storRes)
                        amount = int(float(diff) / doneesSum * donorsSum * tmpRatio)
                        #@log.debug('ISystem', 'Give res', planet.oid, amount)
                        balance -= amount
                        setattr(planet, storRes, getattr(planet, storRes) + amount)
                    # take
                    assert donorsSum + balance >= 0
                    lastPlanet = None
                    tmpRatio = min(float(donorsSum) / doneesSum, 1.0)
                    for planet in donors:
                        diff = getattr(planet, storRes) - getattr(planet, minRes)
                        amount = int(float(diff) / donorsSum * doneesSum * tmpRatio)
                        balance += amount
                        #@log.debug('ISystem', 'Take res', planet.oid, amount)
                        setattr(planet, storRes, getattr(planet, storRes) - amount)
                        lastPlanet = planet
                    # fix rounding error
                    setattr(lastPlanet, storRes, getattr(lastPlanet, storRes) + balance)
                    #@log.debug('ISystem', 'Rounding error', balance)
                # try to move additional resources to the other planets
                for planet in planets[owner]:
                    if getattr(planet, storRes) > getattr(planet, maxRes):
                        excess = getattr(planet, storRes) - getattr(planet, maxRes)
                        #@log.debug('ISystem', 'Trying to move excess rsrcs from', planet.oid, excess)
                        for planet2 in planets[owner]:
                            if planet == planet2:
                                continue
                            if getattr(planet2, storRes) < getattr(planet2, maxRes):
                                space = getattr(planet2, maxRes) - getattr(planet2, storRes)
                                amount = min(space, excess)
                                #@log.debug('ISystem', 'Moved to', planet2.oid, amount)
                                setattr(planet2, storRes, getattr(planet2, storRes) + amount)
                                excess -= amount
                                if excess == 0:
                                    break
                        #@log.debug('ISystem', 'Cannot move excess rsrcs on', planet.oid, excess)
                        setattr(planet, storRes, getattr(planet, maxRes) + excess)
        #~ # rotate system around the galaxy core
        #~ #log.debug("Rotate, old coords", obj.x, obj.y)
        #~ turn = tran.db[OID_UNIVERSE].turn
        #~ galaxy = tran.db[obj.compOf]
        #~ angle = obj.sAngle + (turn / Rules.rotationMod) * obj.dAngle
        #~ obj.x = galaxy.x + obj.dist * math.cos(angle)
        #~ obj.y = galaxy.y + obj.dist * math.sin(angle)
        #~ #log.debug("Rotate, new coords", obj.x, obj.y)
        #~ # change positions of planets and orbitting fleets
        #~ for planetID in obj.planets:
            #~ planet = tran.db[planetID]
            #~ planet.x = obj.x
            #~ planet.y = obj.y
        #~ for fleetID in obj.fleets:
            #~ fleet = tran.db[fleetID]
            #~ fleet.x = obj.x
            #~ fleet.y = obj.y
        # process planets and fleets
        #@log.debug("System close fleets", obj.oid, obj.closeFleets)
        return obj.planets[:] + obj.closeFleets[:]

    processACTIONPhase.public = 1
    processACTIONPhase.accLevel = AL_ADMIN

    def getObjectsInSpace(self, tran, obj):
        inSpace = obj.closeFleets[:]
        for fleetID in obj.fleets:
            try:
                inSpace.remove(fleetID)
            except ValueError:
                log.warning(obj.oid, "Cannot remove fleet from closeFleets", fleetID, obj.fleets, obj.closeFleets)
        return inSpace

    getObjectsInSpace.public = 1
    getObjectsInSpace.accLevel = AL_ADMIN

    def processBATTLEPhase(self, tran, obj, data):
        system = obj
        #@log.debug('ISystem', 'BATTLE - system', obj.oid)
        # we are processing fleets, planets, ...
        objects = obj.planets[:] + obj.fleets[:]
        # shuffle them to prevent predetermined one-sided battles (temporary hack)
        random.shuffle(objects)
        # store owners of objects
        # find enemies and allies
        attack = {}
        allies = {}
        owners = {}
        ownerIDs = {}
        systemAtt = {}
        systemDef = {}
        hasMine = {}
        isOwnedObject = 0
        for objID in objects:
            attack[objID] = []
            allies[objID] = []
            owner = tran.db[objID].owner
            owners[objID] = owner
            ownerIDs[owner] = owner
            if owner != OID_NONE:

                isOwnedObject = 1
        for owner in ownerIDs:
            tempAtt, tempDef = self.getSystemCombatBonuses(tran, system, owner)
            systemAtt[owner] = tempAtt
            systemDef[owner] = tempDef
            hasMine[owner] = self.getSystemMineSource(tran, system, owner)
        if not isOwnedObject:
            #@log.debug('ISystem', 'No combat')
            # reset combat counters
            system.combatCounter = 0
            return
        # first - direct ones
        index = 1
        for obj1ID in objects:
            obj1 = tran.db[obj1ID]
            if obj1.owner == OID_NONE:
                index += 1
                continue
            commander = tran.db[obj1.owner]
            # relationships
            #for obj2ID in objects[index:]:
            for obj2ID in objects:
                obj2 = tran.db[obj2ID]
                if obj2.owner == OID_NONE or obj1 is obj2:
                    continue
                if obj1.owner == obj2.owner:
                    allies[obj1ID].append(obj2ID)
                    allies[obj2ID].append(obj1ID)
                    continue
                # planet and military object
                elif obj1.type == T_PLANET and obj2.isMilitary and \

                    not self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_ALLOW_MILITARY_SHIPS):

                    #@log.debug("ISystem pl - mil", obj1ID, obj2ID)
                    if obj2ID not in attack[obj1ID]:
                        attack[obj1ID].append(obj2ID)
                    if obj1ID not in attack[obj2ID]:
                        attack[obj2ID].append(obj1ID)
                # planet and civilian object
                elif obj1.type == T_PLANET and not obj2.isMilitary and \
                    not self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_ALLOW_CIVILIAN_SHIPS):

                    #@log.debug("ISystem pl - civ", obj1ID, obj2ID)
                    if obj2ID not in attack[obj1ID]:
                        attack[obj1ID].append(obj2ID)
                    if obj1ID not in attack[obj2ID]:
                        attack[obj2ID].append(obj1ID)
                # military and military object
                elif obj1.isMilitary and obj2.isMilitary and \
                    not self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_ALLOW_MILITARY_SHIPS):
                    #@log.debug("ISystem mil - mil", obj1ID, obj2ID)
                    if obj2ID not in attack[obj1ID]:
                        attack[obj1ID].append(obj2ID)
                    if obj1ID not in attack[obj2ID]:
                        attack[obj2ID].append(obj1ID)
                # military and civilian object
                elif obj1.isMilitary and not obj2.isMilitary and \
                    not self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_ALLOW_CIVILIAN_SHIPS):
                    #@log.debug("ISystem mil - civ", obj1ID, obj2ID)
                    if obj2ID not in attack[obj1ID]:
                        attack[obj1ID].append(obj2ID)
                    if obj1ID not in attack[obj2ID]:
                        attack[obj2ID].append(obj1ID)
                # planet and fleet
                #elif obj1.type == T_PLANET and obj2.type == T_FLEET and \
                #    self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_MUTUAL_DEFENCE):
                #    allies[obj1ID].append(obj2ID)
                #    allies[obj2ID].append(obj1ID)
                # fleet and fleet
                #elif obj1.type == T_FLEET and obj2.type == T_FLEET and \
                #    self.cmd(commander).isPactActive(tran, commander, obj2.owner, PACT_MUTUAL_OFFENCE):
                #    allies[obj1ID].append(obj2ID)
                #    allies[obj2ID].append(obj1ID)
                # asteroid
                if obj2.type == T_ASTEROID:

                    attack[obj1ID].append(obj2ID)
                    attack[obj2ID].append(obj1ID)
            index += 1
        #@log.debug('ISystem', 'Targets:', targets)
        #@log.debug('ISystem', 'Allies:', allies)
        # find indirect a/e
        #for objID in objects:
        #    iTargets = []
        #    iAllies = []
        #    # find indirect a/e
        #    todo = allies[objID][:]
        #    while todo:
        #        id = todo.pop(0)
        #        iTargets.extend(targets[id])
        #        for tmpID in allies[id]:
        #            if tmpID not in iAllies:
        #                todo.append(tmpID)
        #                iAllies.append(tmpID)
        #    # remove allies from targets
        #    for id in iAllies:
        #        if id in iTargets:
        #            iTargets.remove(id)
        #    # IMPORTATNT preffer NOT to fire at possible allies
        #    # add my targets
        #    #for id in targets[objID]:
        #    #    if id not in iTargets:
        #    #        iTargets.append(id)
        #    # that's all folks
        #    for id in iTargets:
        #        if objID not in attack[id]:
        #            attack[id].append(objID)
        #        if id not in attack[objID]:
        #            attack[objID].append(id)
        # NOT VALID: objects with action ACTION_ATTACK will attack only their targets
        # check, if there are any targets
        isCombat = 0
        for objID in objects:
            if attack[objID]:
                isCombat = 1
                break #end loop
        if not isCombat:
            #@log.debug('ISystem', 'No combat')
            # reset combat counters
            system.combatCounter = 0
            for fleetID in system.fleets:
                tran.db[fleetID].combatCounter = 0
            return
        # increase combat counters
        system.combatCounter += 1
        for fleetID in system.fleets:
            tran.db[fleetID].combatCounter += 1
        # debug
        log.debug('ISystem', 'Final attacks in system %d:' % system.oid, attack)
        # mines detonate before battle
        shots = {}
        targets = {}
        firing = {}
        damageCaused = {}
        killsCaused = {}
        damageTaken = {}
        shipsLost = {}
        minesTriggered = {}
        fleetOwners = {}
        isCombat = False
        isMineCombat = False
        for owner in ownerIDs:
            if owner not in hasMine: # no planets
                continue
            if not hasMine[owner]: # no planet with control structure
                continue
            controlPlanetID = hasMine[owner][0]  # there is list returned, all planets have same effect
            if len(self.getMines(system, owner)) == 0:
                continue # no mines, something broke
            if len(attack[controlPlanetID]) == 0:
                continue # no targets
            isMineFired = True
            mineTargets = copy.copy(attack[controlPlanetID])
            while isMineFired:
                while len(mineTargets) > 0:
                    targetID = random.choice(mineTargets) # select random target
                    targetobj = tran.db.get(targetID, None)
                    try:
                        if targetobj.type == T_FLEET:

                            fleetOwners[targetID] = targetobj.owner
                            break # target found
                        mineTargets.remove(targetID)  # remove an object type that a mine can't hit from the temporary targets list
                    except:
                        mineTargets.remove(targetID)  # remove a dead fleet from the temporary targets list

                if len(mineTargets) == 0:
                    break # no fleet targets for mines
                temp, temp, firing[targetID] = self.cmd(targetobj).getPreCombatData(tran, targetobj)  # fix firing for "surrender to" section

                damage, att, ignoreshield, mineID = self.cmd(obj).fireMine(system, owner)
                if not damage: # no more mines
                    isMineFired = False
                    break
                log.debug('ISystem', 'Mine Shooting (damage, att, ignore shield):', damage, att, ignoreshield)
                isMineCombat = True
                minesTriggered[mineID] = minesTriggered.get(mineID, 0) + 1
                # Process Combat
                # for now we assume only one ship can be destroyed by one mine
                dmg, destroyed = self.cmd(targetobj).applyMine(tran, targetobj, att, damage, ignoreshield)
                #log.debug('ISystem-Mines', 'Actual Damage Done:',dmg)
                if dmg > 0:
                    damageTaken[targetID] = damageTaken.get(targetID, 0) + dmg

                    shipsLost[targetID] = shipsLost.get(targetID, 0) + destroyed
                    killsCaused[mineID] = killsCaused.get(mineID, 0) + destroyed
                if dmg > 0:
                    damageCaused[mineID] = damageCaused.get(mineID, 0) + dmg
            # send messages about mine effects to the owner of the minefield
            # collect hit players
            players = {}
            for triggerID in firing.keys():
                players[owners[triggerID]] = None
            controllerPlanet = tran.db.get(controlPlanetID, None)
            damageCausedSum = 0
            killsCausedSum = 0
            for mineID in damageCaused.keys():
                damageCausedSum = damageCausedSum + damageCaused.get(mineID, 0)
                killsCausedSum = killsCausedSum + killsCaused.get(mineID, 0)
            Utils.sendMessage(tran, controllerPlanet, MSG_MINES_OWNER_RESULTS, system.oid, (players.keys(),(damageCaused, killsCaused, minesTriggered),damageCausedSum,killsCausedSum))

        # send messages to the players whose fleets got hit by minefields
        for targetID in damageTaken.keys():
            targetFleet = tran.db.get(targetID, None)
            if targetFleet:
                Utils.sendMessage(tran, targetFleet, MSG_MINES_FLEET_RESULTS, system.oid, (damageTaken[targetID], shipsLost[targetID]))

            else:
                targetFleet = IDataHolder()
                targetFleet.oid = fleetOwners[targetID]
                Utils.sendMessage(tran, targetFleet, MSG_MINES_FLEET_RESULTS, system.oid, (damageTaken[targetID], shipsLost[targetID]))
                Utils.sendMessage(tran, targetFleet, MSG_DESTROYED_FLEET, system.oid, ())

        damageCaused = {}
        killsCaused = {}
        damageTaken = {}
        shipsLost = {}
        # now to battle
        for objID in objects:
            obj = tran.db.get(objID, None)
            # get shots from object, should be sorted by weaponClass
            # shots = [ shot, ...], shot = (combatAtt, weaponID)
            # get target classes and numbers
            # (class1, class2, class3, class4)
            # cls0 == fighters, cls1 == midships, cls2 == capital ships, cls3 == planet installations
            #@log.debug(objID, obj.name, "getting pre combat data")
            if obj: # source already destroyed; ignore
                shots[objID], targets[objID], firing[objID] = self.cmd(obj).getPreCombatData(tran, obj)
                if firing[objID]:
                    isCombat = True
        if not isCombat and not isMineCombat:
            # no shots has been fired
            #@log.debug('ISystem', 'No combat')
            # reset combat counters
            system.combatCounter = 0
            for fleetID in system.fleets:
                tran.db[fleetID].combatCounter = 0
            return
        #@log.debug("Shots:", shots)
        #@log.debug("Targets", targets)
        if isCombat:
            for shotIdx in (3, 2, 1, 0):
                for objID in objects:
                    # obj CAN be deleted at this point
                    obj = tran.db.get(objID, None)
                    if obj == None:
                        continue # source already destroyed; move to next source
                    # if object is fleet, then it's signature is max
                    if obj and obj.type == T_FLEET:
                        obj.signature = Rules.maxSignature
                    # target preselection
                    totalClass = [0, 0, 0, 0]
                    total = 0
                    for targetID in attack[objID]:
                        totalClass[0] += targets[targetID][0]
                        totalClass[1] += targets[targetID][1]
                        totalClass[2] += targets[targetID][2]
                        totalClass[3] += targets[targetID][3]
                    total = totalClass[0] + totalClass[1] + totalClass[2] + totalClass[3]
                    # process shots
                    for combatAtt, weaponID in shots[objID][shotIdx]:
                        weapon = Rules.techs[weaponID]
                        weaponClass = weapon.weaponClass
                        if total == 0:
                            # there are no targets
                            break
                        #@log.debug('ISystem', 'Processing shot', objID, weapon.name, weaponClass)
                        # process from weaponClass up
                        # never shoot on smaller ships than weaponClass
                        applied = 0
                        for tmpWpnClass in xrange(weaponClass, 4):

                            #@log.debug('ISystem', 'Trying target class', tmpWpnClass, totalClass[tmpWpnClass])
                            # select target
                            if totalClass[tmpWpnClass]:
                                target = Utils.rand(0, totalClass[tmpWpnClass])
                                #@log.debug('ISystem', 'Target rnd num', target, totalClass[tmpWpnClass])
                                for targetID in attack[objID]:
                                    if target < targets[targetID][tmpWpnClass]:
                                        #@log.debug(objID, 'attacks', targetID, tmpWpnClass)
                                        # targetID can be deleted at this point
                                        anObj = tran.db.get(targetID, None)
                                        if anObj:
                                            dmg, destroyed, destroyedClass = self.cmd(anObj).applyShot(tran, anObj, systemDef[owners[targetID]], combatAtt + systemAtt[owners[objID]], weaponID, tmpWpnClass, target)
                                            #@log.debug("ISystem result", dmg, destroyed, destroyedClass, tmpWpnClass)
                                            #@print objID, 'dmg, destroyed', dmg, destroyed
                                            damageTaken[targetID] = damageTaken.get(targetID, 0) + dmg
                                            if destroyed > 0:
                                                shipsLost[targetID] = shipsLost.get(targetID, 0) + destroyed
                                                total -= destroyed
                                                totalClass[destroyedClass] -= destroyed
                                            if dmg > 0 and obj:
                                                obj.combatExp += dmg
                                                damageCaused[objID] = damageCaused.get(objID, 0) + dmg
                                            applied = 1
                                        else:
                                            continue # target already destroyed, move to next target
                                        break
                                    else:
                                        #@log.debug('ISystem', 'Lovering target by', targets[targetID][tmpWpnClass])
                                        target -= targets[targetID][tmpWpnClass]
                            if applied:
                                break
        # send messages and modify diplomacy relations
        # distribute experience pts
        for objID in objects:
            obj = tran.db.get(objID, None)
            if obj:
                self.cmd(obj).distributeExp(tran, obj)
            if attack[objID]:
                source = obj or tran.db[owners[objID]]
                # collect players
                players = {}
                for attackerID in attack[objID]:
                    players[owners[attackerID]] = None
                d1 = damageTaken.get(objID,0)
                d2 = damageCaused.get(objID,0)
                l = shipsLost.get(objID, 0)
                if d1 or d2 or l:
                    # send only if damage is taken/caused
                    Utils.sendMessage(tran, source, MSG_COMBAT_RESULTS, system.oid, (d1, d2, l, players.keys()))

                if not obj:
                    # report DESTROYED status
                    Utils.sendMessage(tran, source, MSG_DESTROYED_FLEET, system.oid, ())
                # modify diplomacy relations
                objOwner = tran.db[owners[objID]]
                for attackerID in attack[objID]:
                    attOwner = tran.db.get(owners[attackerID], None)
                    # owner of the fleet
                    rel = self.cmd(objOwner).getDiplomacyWith(tran, objOwner, attOwner.oid)
                    rel.relChng = Rules.relLostWhenAttacked
                    # attacker
                    rel = self.cmd(attOwner).getDiplomacyWith(tran, attOwner, objOwner.oid)
                    rel.rechChng = Rules.relLostWhenAttacked
        # check if object surrenders
        for objID in objects:
            # object surrender IFF it and its allies had target and was not able
            # to fire at it, planet is not counted as ally in this case
            obj = tran.db.get(objID, None)
            if firing[objID] and obj:
                continue
            surrenderTo = []
            for attID in attack[objID]:
                if firing[attID] and tran.db.has_key(attID):
                    surrenderTo.append(tran.db[attID].owner)
            for allyID in allies[objID]:
                if not tran.db.has_key(allyID):
                    continue
                ally = tran.db[allyID]
                if firing[allyID] and ally.type != T_PLANET:
                    surrenderTo = []
                    break
            if surrenderTo:
                index = Utils.rand(0, len(surrenderTo))
                if obj:
                    if self.cmd(obj).surrenderTo(tran, obj, surrenderTo[index]):
                        winner = tran.db[surrenderTo[index]]
                        source = tran.db.get(owners[objID], None)
                        log.debug('ISystem', 'BATTLE - surrender', objID, surrenderTo[index], surrenderTo)
                        if source:
                            Utils.sendMessage(tran, source, MSG_COMBAT_LOST, system.oid, winner.oid)

                            Utils.sendMessage(tran, winner, MSG_COMBAT_WON, system.oid, source.oid)

                        else:
                            Utils.sendMessage(tran, winner, MSG_COMBAT_WON, system.oid, obj.oid)
                else:
                    winner = tran.db[surrenderTo[index]]
                    source = tran.db[owners[objID]]
                    log.debug('ISystem', 'BATTLE - surrender', objID, surrenderTo[index], surrenderTo)
                    Utils.sendMessage(tran, source, MSG_COMBAT_LOST, system.oid, winner.oid)
                    Utils.sendMessage(tran, winner, MSG_COMBAT_WON, system.oid, source.oid)
        return

    processBATTLEPhase.public = 1
    processBATTLEPhase.accLevel = AL_ADMIN

    def processFINALPhase(self, tran, obj, data):
        # TODO find new starting points
        # clean up mines if system ownership was lost
        owners = []
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner not in owners:
                owners.append(planet.owner)
        for ownerid in obj.minefield:
            if ownerid not in owners:
                self.clearMines(obj, ownerid)
        return obj.planets[:] + obj.closeFleets[:]

    processFINALPhase.public = 1
    processFINALPhase.accLevel = AL_ADMIN

    def sortPlanets(self, tran, obj, data):
        obj.planets.sort(key=lambda planetID: tran.db[planetID].plEn, reverse = True)
        orbit = 1
        for planetID in obj.planets:
            planet = tran.db[planetID]
            planet.orbit = orbit
            orbit += 1

    sortPlanets.public = 0

    def rename(self, tran, obj, newName, nType):
        newName = newName.strip()
        # you have to own all planets
        # TODO: Throw another cmdr exc AFTER you have no planet
        haveOne = 0
        anotherComm = 0
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner != tran.session.cid and planet.owner != OID_NONE:

                anotherComm = 1
            if planet.owner == tran.session.cid:
                haveOne = 1
        if not haveOne:
            raise ige.GameException('You cannot change name of this system - you have no planet in this system.')
        if anotherComm:
            raise ige.GameException('You cannot change name of this system - another commander in system.')
        # check validity of name
        if not Utils.isCorrectName(newName):
            raise ige.GameException('Invalid name. Only characters, digits, space, dot and dash permitted, max. length is 30 characters.')
        # check if there is other system with this name
        galaxy = tran.db[obj.compOf]
        for systemID in galaxy.systems:
            if tran.db[systemID].name == newName and systemID != obj.oid:
                raise ige.GameException('This name is already used.')
        # TODO you have to own this system longer than previous owner
        # one change per 1 day allowed
        turn = tran.db[OID_UNIVERSE].turn

        if obj.lastNameChng + Rules.turnsPerDay <= turn:
            # rename system
            obj.name = newName
            # rename planets
            newNames = [obj.name]
            for planetID in obj.planets:
                planet = tran.db[planetID]
                planet.name = Utils.getPlanetName(obj.name, nType, planet.orbit - 1)
                newNames.append(planet.name)
            obj.lastNameChng = turn
        else:
            raise ige.GameException('You cannot change name of this system - name has been changed recently (try it one day later).')
        return newNames

    rename.public = 1
    rename.accLevel = AL_NONE


    def createPlanet(self, tran, obj):
        planet = self.new(T_PLANET)

        planet.compOf = obj.oid
        oid = tran.db.create(planet)
        obj.planets.append(oid)
        return oid

    def deployMines(self, tran, obj, owner_id):
        """ Go through all mine control structures and attempt to add mines.

        """
        for tech, struct in self.getSystemMineLauncher(tran, obj, owner_id):
            if not struct[STRUCT_IDX_STATUS] & STRUCT_STATUS_ON:

                # structure is offline, reset timer
                self.addMine(tran, obj, owner_id, tech.mineclass, 0)
                continue
            efficiency = struct[STRUCT_IDX_HP] / float(tech.maxHP)

            minerate = int(tech.minerate / efficiency)
            minenum = int(tech.minenum * efficiency)
            if self.addMine(tran, obj, owner_id, tech.mineclass, minenum, minerate):
                log.debug('ISystem', 'Mine deployed for owner %d in system %d' % (owner_id, obj.oid))

    deployMines.public = 1
    deployMines.accLevel = AL_ADMIN


    def addMine(self, tran, obj, owner_id, mine_tech_id, max_amount=None, mine_rate=None):
        """ Increment amount within particular minefield of particular player.
        Set current turn as a date of latest deployment.

        Returns True if mine was added.

        """
        current_turn = tran.db[obj.compOf].galaxyTurn
        if owner_id in obj.minefield:
            index = -1
            for mine_id, amount, deploy_turn in obj.minefield[owner_id]:
                index += 1
                if mine_id != mine_tech_id:
                    continue
                if max_amount is not None and amount >= max_amount:
                    # cannot add more, thus update deploy turn to current one
                    # (so replenish starts after that amount of turns if needed)
                    obj.minefield[owner_id][index] = (mine_id, amount, current_turn)
                    return False
                if mine_rate and (current_turn - deploy_turn) < mine_rate:
                    # need more time to deploy new mine
                    return False
                obj.minefield[owner_id][index] = (mine_id, amount + 1, current_turn)
                return True
            # owner has some minefield, but not this type
            obj.minefield[owner_id].append((mine_id, 1, current_turn))

            return True
        else:
            # this will be owner's first minefield in this system
            obj.minefield[owner_id]= [(mine_tech_id, 1, current_turn)]
            return True

    addMine.public = 1
    addMine.accLevel = AL_ADMIN

    def removeMine(self, obj, owner_id, mine_tech_id):
        """ Decrement amount within particular minefield of particular player.
        If amount drops to zero, remove minefield.
        If all minefields are removed, remove player minefield record.

        """
        if owner_id in obj.minefield:
            index = -1
            for mine_id, amount, deploy_turn in obj.minefield[owner_id]:
                index += 1
                if mine_id != mine_tech_id:
                    continue
                amount -= 1
                if amount > 0:
                    obj.minefield[owner_id][index] = (mine_id, amount, deploy_turn)
                else:
                    # we have depleted the minefield, remove any notion of it
                    del obj.minefield[owner_id][index]
                break
            if not len(obj.minefield[owner_id]):
                # all minefields are depleted, remove player record
                del obj.minefield[owner_id]

    removeMine.public = 1
    removeMine.accLevel = AL_ADMIN


    def getMines(self, obj, owner_id):
        """ Return list of tuples representing each minefield.

        """
        if owner_id in obj.minefield:
            return obj.minefield[owner_id]
        else:
            return []

    getMines.public = 1
    getMines.accLevel = AL_ADMIN

    def clearMines(self, obj, owner_id):
        """ Remove all minefields of given owner from the system.

        """
        if owner_id in obj.minefield:
            del obj.minefield[owner_id]

    clearMines.public = 0

    def fireMine(self, obj, owner_id): # shoot the mine
        if owner_id not in obj.minefield:
            return False, False, False, False

        mine_ids = []
        amounts = []
        for mine_id, amount, deploy_turn in obj.minefield[owner_id]:
            mine_ids.append(mine_id)
            amounts.append(amount)
        mine = Utils.weightedRandom(mine_ids, amounts) # select random mine to detonate
        self.removeMine(obj, owner_id, mine)
        tech = Rules.techs[mine]
        damage = random.randrange(tech.weaponDmgMin, tech.weaponDmgMax)
        attack = tech.weaponAtt
        ignore_shield = tech.weaponIgnoreShield
        return damage, attack, ignore_shield, mine

    fireMine.public = 1
    fireMine.accLevel = AL_ADMIN

    def getSystemMineLauncher(self, tran, obj, playerID):
        launchers = []
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner == playerID:
                for struct in planet.slots:
                    tech = Rules.techs[struct[STRUCT_IDX_TECHID]]

                    if tech.mineclass:
                        launchtech = tech
                        launchers.append((launchtech, struct))
        return launchers

    getSystemMineLauncher.public = 0

    def getSystemMineSource(self, tran, obj, playerID):
        sources = []
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner == playerID:
                for struct in planet.slots:
                    tech = Rules.techs[struct[STRUCT_IDX_TECHID]]

                    if tech.mineclass:
                        sources.append(planetID)
        return sources

    getSystemMineSource.public = 0

    def getSystemCombatBonuses(self, tran, obj, playerID):
        systemAtt = 0;
        systemDef = 0;
        for planetID in obj.planets:
            planet = tran.db[planetID]
            if planet.owner == playerID:
                for struct in planet.slots:
                    tech = Rules.techs[struct[STRUCT_IDX_TECHID]]

                    techEff = Utils.getTechEff(tran, struct[STRUCT_IDX_TECHID], planet.owner)
                    if tech.systemAtt > 0 or tech.systemDef > 0:
                        systemAtt = max(systemAtt, tech.systemAtt * techEff)
                        systemDef = max(systemDef, tech.systemDef * techEff)
        return (systemAtt, systemDef)

    getSystemCombatBonuses.public = 0

    def loadDOMNode(self, tran, obj, xoff, yoff, node):
        obj.x = float(node.getAttribute('x')) + xoff
        obj.y = float(node.getAttribute('y')) + yoff
        orbit = 1
        nType = Utils.getPlanetNamesType()
        for elem in node.childNodes:
            if elem.nodeType == Node.ELEMENT_NODE:
                name = elem.tagName
                if name == 'properties':
                    self.loadDOMAttrs(obj, elem)
                elif name == 'planet':
                    # create planet
                    planet = tran.db[self.createPlanet(tran, obj)]
                    self.cmd(planet).loadDOMNode(tran, planet, obj.x, obj.y, orbit, elem)
                    # planet.name = u'%s %s' % (obj.name, '-ABCDEFGHIJKLMNOPQRSTUVWXYZ'[orbit])
                    planet.name = Utils.getPlanetName(obj.name, nType, orbit - 1)
                    orbit += 1
                else:
                    raise ige.GameException('Unknown element %s' % name)
        #~ # compute rotational constants
        #~ galaxy = tran.db[obj.compOf]
        #~ dx = obj.x - galaxy.x
        #~ dy = obj.y - galaxy.y
        #~ obj.dist = math.sqrt(dx * dx + dy * dy)
        #~ if obj.dist > 0:
            #~ obj.dAngle = math.sqrt(galaxy.centerWeight / obj.dist) / obj.dist
        #~ else:
            #~ obj.dAngle = 0.0
        #~ if dx != 0:
            #~ obj.sAngle = math.atan(dy / dx)
            #~ if dx < 0: obj.sAngle += math.pi
        #~ elif dy > 0:
            #~ obj.sAngle = math.pi / 2
        #~ elif dx < 0:
            #~ obj.sAngle = math.pi * 3 / 2
        #~ # this is a check only
        #~ angle = obj.sAngle + (0 / 384.0) * obj.dAngle
        #~ x = galaxy.x + obj.dist * math.cos(angle)
        #~ y = galaxy.y + obj.dist * math.sin(angle)
        #~ if x != obj.x or y != obj.y:
            #~ log.warning(obj.name, obj.x, obj.y, dx, dy, obj.dist, obj.dAngle, obj.sAngle, x, y)
        return SUCC


ospaceteam / outerspace

Pull Request — master (#188)

ISystem.getSystemCombatBonuses() B

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like
