runGame.py

from soldiers import *
from towers import *
from economy import Economy
from output import ConsoleOutput
import time
import math
import random
import gamemap
from pprint import pformat

#import pygame;
#from pygame.locals import *;

from termcolor import colored
#from utils import Utils;


class Game(object):
  # Static method definitions
  @staticmethod
  def getInstanceOf(sClassName,  economy,  gameMap):
      return eval("soldiers." + sClassName +"(economy, gameMap)");
      
  @staticmethod
  def selectArmy(eEconomy,  gameMap,  armyColor,  output,  
    aUnitPool = [
      # "AssassinClass", 
      # "BarbarianClass",
      # "CartographerClass",
      # "DruidClass",  
      # "EnchanterClass", 
      # "KnightClass",  
      "MageClass",  
      # "RangerClass", 
      "SoldierClass", 
      "TechnicianClass",
      # "BridgeBuilderClass",
      # "WizardClass",  
      ]):
      iTotalValue  = 0;
      aArmy = [];
      curMoney = eEconomy.maxMoney;
      output.log("\nStarting money:" + str(curMoney) + "$");
      output.dump();
      
      
      while (curMoney > 0):
          curChoice = random.choice(aUnitPool);
          
          # If we exceed our money
          if (eEconomy.cost(Game.getInstanceOf(curChoice,  eEconomy,  gameMap)) > curMoney):
              curChoice = None;
              # Try linearly
              for cChoice in aUnitPool:
                # If we have enough money
                if eEconomy.cost(Game.getInstanceOf(cChoice,  eEconomy,  gameMap)) <= curMoney:
                    # buy the army
                    curChoice = cChoice;
              # If we found nothing we can afford
              if (curChoice == None):
                output.log("\nTotal army value: %d$\n"%(iTotalValue));
                # quit
                return aArmy;
          # else buy the army
          armyToAdd = Game.getInstanceOf(curChoice,  eEconomy,  gameMap);
          # assign color
          armyToAdd.color = armyColor;
          
          aArmy.append(armyToAdd);
          output.log("Selected %s for %d$. " %(str(curChoice),  eEconomy.cost(armyToAdd)));
          # update our running total of money
          curMoney -= eEconomy.cost(armyToAdd);
          iTotalValue += eEconomy.cost(armyToAdd);
      output.log("\nTotal army value: %d$\n"%(iTotalValue));
        
  # Class method definitions
  def dealDamage(self, oFrom, oTo):
      output = self.output;
      output.log("%s attacks %s with a %s attack.\t"%(str(oFrom) + str(id(oFrom))[:-3],  str(oTo) + str(id(oTo))[:-3],  oFrom.damageType) );
      
      actualDefence = oTo.defence;
      
      if oFrom.damageType in oTo.immunities:
        output.log("Target is immune!");
        return oTo;

      if oFrom.damageType in oTo.resistances:
        actualDefence = oTo.defence * 2.0;
      if oFrom.damageType in oTo.vulnerabilities:
        actualDefence = oTo.defence * 0.5;

      dDmg = round(max(oFrom.attack * math.pow(1.05, (10 - actualDefence)), 1.0));
      oTo.currentHp -= dDmg;
      output.log("Target is hit for %d damage. %d hit points remaining."%(dDmg,  oTo.currentHp));

      return oTo;

  def printGameState(self):
      self.output.dump();

  def repaintTerrain(self):
    self.output.drawMap(self.gameMap,  self.aAttackers,  []);

  def interactWithTrap(self,  actor,  trap,  friends,  foes,  traps=[]):
      self.act(actor,  trap,  friends,  foes,  traps);
      if (trap.hp <= 0):
        self.gameMap.traps.remove(trap);
        # Reward actor
        actor.score += self.economy.trapValue(trap);
      else:
        self.act(trap,  actor,  [],  [actor]);
      

      return actor.currentHp > 0;

  def interactWithHome(self,  actor):
    actor.fullness = 100;
    self.output.log(str(actor) + " got provisions!")
    return True
    
  def interactWithTreasure(self,  actor,  treasure,  friends,  foes):
      treasure.applyEffectTo(actor);

      self.gameMap.treasures.remove(treasure);
      # Reward actor
      actor.score += self.economy.treasureValue(treasure);
      
      return True;
      
  def interactWithFoe(self,  actor, foe,  friends,  foes):
      # DEBUG LINES
      #output.log(str(actor) + " in " + str(actor.x) + ","  + str(actor.y) + " attacks \n\t" +  str(foe))
      
      self.act(actor,  foe,  friends,  foes);


      return actor.currentHp > 0;
      
  def act(self, actor, target,  friends, foes,  traps = []):
      output = self.output;
      
      if (isinstance(actor,  SoldierClass)):
        try:
          output.color(actor.color);
        except:
          pass
      
      # Use abilities, if they exist
      if len(actor.abilities) > 0:
        # DEBUG LINES
        #output.log("Using ability!");
        for aCurAbility in actor.abilities:          
          # DEBUG LINES
          #output.log("Ability group:" + aCurAbility.group);
          
          foe = None;
          if aCurAbility.group == "traps":
            if target in traps:
              foe = target;
            # DEBUG LINES
#            print "Found TRAP!";
#            raw_input();
          else:
            if aCurAbility.group == "foes":
              if target in foes:
                  foe = target;
                  
          # If not applicable as a foe or trap    
          if (foe == None):
            continue;
            
          # Check probability for probability-based abilities
          try:
             if (float(aCurAbility.frequency) > 0) and (float(aCurAbility.frequency) < 1):                  
                fCurProbability = float(aCurAbility.frequency);
                fRoll = random.random();
                if  fRoll <= fCurProbability:
                    # DEBUG LINES
                    #output.log("Rolled %4.2f with a chance of %4.2f"%(fRoll,  fCurProbability));
                    
                    if (aCurAbility.applyTo(foe, friends, foes) != None):
                        output.log(str(aCurAbility));
                continue;
          except:
            # DEBUG LINES
            #output.log("\n" + str(aCurAbility) + "\n");
            pass; # Ignore

          # Try to apply ability for other types of frequency
          if (aCurAbility.applyTo(foe, friends, foes,  traps) != None):
            output.log(str(aCurAbility));
#          else:
#              # DEBUG LINES
#              print "Could not use " + str(type(aCurAbility));
          
          # Remove battle abilities
          if aCurAbility.frequency == "battle": 
              actor.abilities.remove(aCurAbility);
          # Decrease (and possibly remove) abilities with uses
          try:
              if (int(aCurAbility.frequency) >= 1):
                iCurUses = int(aCurAbility.frequency);
                iCurUses -= 1;
                if iCurUses == 0:
                    actor.abilities.remove(aCurAbility);
                else:
                    aCurAbility.frequency = str(iCurUses);
          except:
            pass; # Ignore

      # Deal damage, if target is still a foe
      if target in foes:
        target.currentHp = self.dealDamage(actor, target).currentHp;
      # Reset color 
      output.color();
    
  def __init__(self,  economy,  gameMap,  army,  output,  msgBaseDelaySecs = 0.10):
    # Init map
    self.economy = economy;
    self.gameMap = gameMap;
    self.aAttackers = army;
    # Init output
    self.output = output;
    self.msgBaseDelaySecs = msgBaseDelaySecs;
    
  def run(self):
    output = self.output;
    msgBaseDelaySecs = self.msgBaseDelaySecs;
    
    output.log("Game begins!");
        
    self.dead = [];
    output.dump();
    time.sleep(2 * msgBaseDelaySecs);
    
    # Position armies
    iX = 0; iY = 2;
    for oItemA in self.aAttackers:
        oItemA.x = iX;
        oItemA.y = iY;
        iX += 1;
        if (iX == self.gameMap.xSize):
            iX = 0;
            iY += 1;
            
    # Main game loop
    iGameTime = 0;
    self.repaintTerrain();
    iRowCnt = 0;
    
    while (True):        
        bShouldRepaint = False;
        
        iGameTime += 1; # Next moment
        if (iGameTime % 100 == 0):
          output.log("The time is now %d..."%(iGameTime));
        self.gameTime = iGameTime;        
        bEndConditions = False;
        
        # Local vars
        aAttackers = self.aAttackers;
        aDefenders = self.gameMap.foes
        
        bActed = False;
        
        # Check attackers
        aToRemove = []
        for cCurAttacker in aAttackers:
            if (cCurAttacker.currentHp <= 0 or cCurAttacker.fullness <= 0):
                output.log(colored("\nAttacker", cCurAttacker.color) + str(id(cCurAttacker)) + " has died!");
                if cCurAttacker.fullness <= 0:
                  output.log("...in fact it was STARVATION...");
                # Put to dead
                self.dead += [(cCurAttacker, iGameTime)];
                aToRemove += [cCurAttacker];
                bActed = True;
            else:
              if self.timeToAct(cCurAttacker):
                # Reduce fullness
                cCurAttacker.fullness -= 1;
                bActed = cCurAttacker.act(aAttackers,  self.gameMap.foes,  self);
        # Update attackers list
        aAttackers = list(set(aAttackers) - set(aToRemove))
        self.aAttackers = aAttackers
        if (len(aToRemove) > 0):
          output.log("Remaining attackers: " + ",".join(map(str,  aAttackers)))

        # DEBUG LINES
#              output.log("\n" + str(cCurAttacker) + "\n");
#              output.log(pformat(vars(cCurAttacker)) + "\n");
        
        
        # Also check defenders
        for cCurDefender in filter(lambda x: isinstance(x, SoldierClass), aDefenders):
            if (cCurDefender.currentHp <= 0):
              output.log("\nDefender" + str(id(cCurDefender)) + " has died!");
              self.gameMap.foes.remove(cCurDefender)
              bActed = True
            else:
              if self.timeToAct(cCurDefender):
                bActed = bActed or cCurDefender.act(aDefenders,  aAttackers,  self,  canMove = False,  canInteractWTraps = False,  canInteractWFoes = True,  
                  canInteractWTreasure = False,  canInteractWHome = False); # Cannot only interact with foes
        
        if (bActed):
            self.repaintTerrain();
            output.log("\n");
            self.printGameState();
            # time.sleep(3 * msgBaseDelaySecs);
            iRowCnt += 1;
            if (iRowCnt >= 5 or bShouldRepaint):
                if (bShouldRepaint):
                    time.sleep(10 * msgBaseDelaySecs);
                    iRowCnt = 0;
                
        # TODO: Remove
        bEndConditions = (len(self.aAttackers) == 0) or (iGameTime >= 1000);
        # End of game
        if (bEndConditions):
            break;

    dScore = self.getScore(iGameTime,  self.aAttackers, self.dead);
    output.log("Score: %d after %d time"%(dScore,  iGameTime));
    
    if (len(self.aAttackers) == 0):
        output.log("\n\nNo Attackers left! Defenders win!");
    else:
      output.log("Foraging complete! %d died and %d remain."%(len(self.dead),  len(self.aAttackers)));
    
    

    # Final output
    output.dump();
    return dScore;

  def timeToAct(self,  actor):
    return (self.gameTime % math.floor(1000.0 / actor.attackSpeed)) == 0
    
    
  def getScore(self,  iGameTime,  aSurvivors, aDead):
    dScore = iGameTime;
    for curSurvivor in aSurvivors:
      dScore += self.economy.cost(curSurvivor) + curSurvivor.score;
    # Also take into account dead
    for soldier,dieTime in self.dead:
      dScore += (soldier.score / 2) * float(dieTime) / float(iGameTime)
    return dScore;

if __name__ == "__main__":
  # Init economy and map
  economy = Economy(500);

  gameMap = gamemap.GameMap(economy, 10, 10, 0.1, 0.1, 0.05, 0.05);
  # Init messaging
  output = ConsoleOutput();
  # Init  army
  # Set colors
  sAttackerColor = "white";
  army = Game.selectArmy(economy,  gameMap,  sAttackerColor,  output);

  # Output game data
  sGameStats = ("Traps: %d, Treasures: %d, Foes: %d, Army: %d"%(len(gameMap.traps), len(gameMap.treasures), len(gameMap.foes), len(army)))
  # Init game
  g = Game(economy,  gameMap,  army,  output, 0.05);


  evaluation = g.run();
  print("Final evaluation of the map",evaluation);
  print("Map info: " + sGameStats)
  g.output.saveToFile("log.txt")