home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / CvPerfTest.py < prev    next >
Encoding:
Python Source  |  2005-11-09  |  2.4 KB  |  102 lines
  1. ## Sid Meier's Civilization 4
  2. ## Copyright Firaxis Games 2005
  3. from CvPythonExtensions import *
  4.  
  5. gc = CyGlobalContext()
  6.  
  7. def fillMap():
  8.     "Entry point to fill a map"
  9.     CvFillMap().fillMap()
  10.     
  11. def emptyMap():
  12.     "Entry point to empty a map"
  13.     CvFillMap().emptyMap()
  14.  
  15. #
  16. #
  17. #
  18. class CvFillMap:
  19.     "class which handles filling the map"
  20.     def __init__(self):
  21.         self.waterUnits=[]
  22.         self.landUnits=[]
  23.  
  24.     def emptyMap(self):
  25.         "clear players and cities from the map"
  26.         for i in range(gc.getMAX_PLAYERS()):
  27.             gc.getPlayer(i).killUnits()
  28.             gc.getPlayer(i).killCities()
  29.     
  30.     def fillMap(self):
  31.         "populate the map for performance testing"
  32.     
  33.         self.waterUnits=[]
  34.         self.landUnits=[]
  35.         for i in range(gc.getNumUnitInfos()):
  36.             if (gc.getUnitInfo(i).getDomainType()==DomainTypes.DOMAIN_SEA):
  37.                 self.waterUnits.append(i)
  38.             else:
  39.                 self.landUnits.append(i)
  40.     
  41.         seed = 1001
  42.         
  43.         rand = CyRandom()
  44.         rand.init(seed)
  45.     
  46.         map = CyMap()
  47.         for iX in range(map.getGridWidth()):
  48.             for iY in range(map.getGridHeight()):
  49.                 self.fillPlot(iX, iY, rand)
  50.             print "fillPlot", iX
  51.             
  52.     def fillPlot(self, iX, iY, rand):
  53.         "roll the dice and put something on this plot"
  54.         ignoreChance = 4    # 40% chance of doing nothing
  55.         cityChance = 2        # 20% chance of a city
  56.         maxUnitsPerPlot = 10    # stack 1-10 units on a plot
  57.         
  58.         waterPlot = CyMap().plot(iX, iY).isWater()
  59.         if waterPlot:
  60.             ignoreChance = 9    # more likely to ignore a waterplot
  61.             
  62.         r=rand.get(10, "PerfTest")
  63.         
  64.         # check if we should do nothing
  65.         if r<ignoreChance:
  66.             return    
  67.     
  68.         owner = CyMap().plot(iX, iY).getOwner()
  69.         if (owner != -1):
  70.             player = gc.getPlayer(owner)
  71.         else:
  72.             player = gc.getPlayer(GC.getBARBARIAN_PLAYER())
  73.     
  74.         # check if we should place a city 
  75.         r=rand.get(10, "PerfTest")
  76.         if r<cityChance:
  77.             if player.canFound(iX, iY):
  78.                 player.initCity(iX, iY)
  79.                 # add city buildings TODO
  80.                 return
  81.     
  82.         # otherwise place units
  83.         numUnits = rand.get(maxUnitsPerPlot+1, "PerfTest") + 1
  84.     
  85.         if waterPlot:
  86.             r = rand.get(len(self.waterUnits), "PerfTest")
  87.             unitType = self.waterUnits[r]
  88.         else:
  89.             r = rand.get(len(self.landUnits), "PerfTest")
  90.             unitType = self.landUnits[r]
  91.                 
  92.         #    print "units", iX, iY
  93.         unitAI = UnitAITypes.UNITAI_UNKNOWN
  94.         for i in range(numUnits):
  95.             player.initUnit(unitType, iX, iY, unitAI)
  96.         
  97.         # check if we should put improvements down TODO
  98.         
  99.         # check if we should put roads down TODO
  100.     
  101.         
  102.