home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / pyWB / CvWBDesc.py next >
Encoding:
Python Source  |  2005-11-09  |  45.9 KB  |  1,500 lines
  1. ## Sid Meier's Civilization 4
  2. ## Copyright Firaxis Games 2005
  3. #
  4. # Code for saving and loading a high-level description of the world.
  5. # Used by WorldBuilder
  6. #
  7. # Author -    Mustafa Thamer
  8. #        Jon Shafer
  9. #
  10.  
  11. from CvPythonExtensions import *
  12. import os
  13. import sys
  14. import CvUtil
  15. from array import *
  16.  
  17. # globals
  18. gc = CyGlobalContext()
  19. version = 11
  20. fileencoding = "latin_1"    # aka "iso-8859-1"
  21.  
  22. #############
  23. def getPlayer(idx):
  24.     "helper function which wraps get player in case of bad index"
  25.     if (gc.getPlayer(idx).isAlive()):
  26.         return gc.getPlayer(idx)
  27.     return None
  28.  
  29. #############
  30. class CvWBParser:
  31.     "parser functions for WB desc"
  32.     def getTokens(self, line):
  33.         "return a list of (comma separated) tokens from the line.  Strip whitespace on each token"
  34.         if line==None:
  35.             return list()
  36.         toks=line.split(",")
  37.         toksOut=list()
  38.         for tok in toks:
  39.             toksOut.append(tok.strip())
  40.         return toksOut
  41.         
  42.     def findToken(self, toks, item):
  43.         "return true if item exists in list of tokens"
  44.         for tok in toks:
  45.             if (tok==item):
  46.                 return true
  47.         return false
  48.         
  49.     def findTokenValue(self, toks, item):
  50.         "Search for a token of the form item=value in the list of toks, and return value, or -1 if not found"
  51.         for tok in toks:
  52.             l=tok.split("=")
  53.             if (item==l[0]):
  54.                 if (len(l)==1):
  55.                     return item        
  56.                 return l[1]
  57.         return -1        # failed
  58.             
  59.     def getNextLine(self, f):
  60.         "return the next line from the list of lines"
  61.         return f.readline()
  62.     
  63.     def findNextToken(self, f, item):
  64.         "Find the next line that contains the token item, return false if not found"
  65.         while True:
  66.             line = self.getNextLine(f)
  67.             if (not line):
  68.                 return false    # EOF
  69.             toks=self.getTokens(line)
  70.             if (self.findToken(toks, item)):
  71.                 return true
  72.         return false
  73.         
  74.     def findNextTokenValue(self, f, item):
  75.         "Find the next line that contains item=value, return value or -1 if not found"
  76.         while True:
  77.             line = self.getNextLine(f)
  78.             if (not line):
  79.                 return -1        # EOF
  80.             toks=self.getTokens(line)
  81.             val=self.findTokenValue(toks, item)
  82.             if (val != -1):
  83.                 return val
  84.         return -1        
  85.  
  86. #############
  87. class CvGameDesc:
  88.     "class for serializing game data"
  89.     def __init__(self):
  90.         self.eraType = "NONE"
  91.         self.speedType = "NONE"
  92.         self.calendarType = "CALENDAR_DEFAULT"
  93.         self.options = ()
  94.         self.mpOptions = ()
  95.         self.forceControls = ()
  96.         self.victories = ()
  97.         self.gameTurn = 0
  98.         self.maxTurns = 0
  99.         self.maxCityElimination = 0
  100.         self.targetScore = 0
  101.         self.iStartYear = -4000
  102.         self.szDescription = ""
  103.         self.szModPath = ""
  104.         
  105.     def apply(self):
  106.         "after reading, apply the game data"
  107.         gc.getGame().setStartYear(self.iStartYear)
  108.         
  109.     def write(self, f):
  110.         "write out game data"
  111.         f.write("BeginGame\n")
  112.         f.write("\tEra=%s\n" %(gc.getEraInfo(gc.getGame().getStartEra()).getType(),))
  113.         f.write("\tSpeed=%s\n" %(gc.getGameSpeedInfo(gc.getGame().getGameSpeedType()).getType(),))
  114.         f.write("\tCalendar=%s\n" %(gc.getCalendarInfo(gc.getGame().getCalendar()).getType(),))
  115.         
  116.         # write options
  117.         for i in range(gc.getNumGameOptionInfos()):
  118.             if (gc.getGame().isOption(i)):
  119.                 f.write("\tOption=%s\n" %(gc.getGameOptionInfo(i).getType()))
  120.                 
  121.         # write mp options
  122.         for i in range(gc.getNumMPOptionInfos()):
  123.             if (gc.getGame().isMPOption(i)):
  124.                 f.write("\tMPOption=%s\n" %(gc.getMPOptionInfo(i).getType()))
  125.                 
  126.         # write force controls
  127.         for i in range(gc.getNumForceControlInfos()):
  128.             if (gc.getGame().isForcedControl(i)):
  129.                 f.write("\tForceControl=%s\n" %(gc.getForceControlInfo(i).getType()))
  130.                 
  131.         # write victories
  132.         for i in range(gc.getNumVictoryInfos()):
  133.             if (gc.getGame().isVictoryValid(i)):
  134.                 if (not gc.getVictoryInfo(i).isPermanent()):
  135.                     f.write("\tVictory=%s\n" %(gc.getVictoryInfo(i).getType()))
  136.                 
  137.         f.write("\tGameTurn=%d\n" %(gc.getGame().getGameTurn(),))
  138.         f.write("\tMaxTurns=%d\n" %(gc.getGame().getMaxTurns(),))
  139.         f.write("\tMaxCityElimination=%d\n" %(gc.getGame().getMaxCityElimination(),))
  140.         f.write("\tTargetScore=%d\n" %(gc.getGame().getTargetScore(),))
  141.         
  142.         f.write("\tStartYear=%d\n" %(gc.getGame().getStartYear(),))
  143.         f.write("\tDescription=\n")
  144.         f.write("\tModPath=\n")
  145.         f.write("EndGame\n")
  146.         
  147.     def read(self, f):
  148.         "read in game data"
  149.         self.__init__()
  150.         
  151.         parser = CvWBParser()
  152.         if (parser.findNextTokenValue(f, "BeginGame")!=-1):
  153.             while (true):
  154.                 nextLine = parser.getNextLine(f)
  155.                 toks = parser.getTokens(nextLine)
  156.                 if (len(toks)==0):
  157.                     break
  158.                     
  159.                 v = parser.findTokenValue(toks, "Era")
  160.                 if v!=-1:
  161.                     self.eraType = v
  162.                     continue
  163.                     
  164.                 v = parser.findTokenValue(toks, "Speed")
  165.                 if v!=-1:
  166.                     self.speedType = v
  167.                     continue
  168.  
  169.                 v = parser.findTokenValue(toks, "Calendar")
  170.                 if v!=-1:
  171.                     self.calendarType = v
  172.                     continue
  173.  
  174.                 v = parser.findTokenValue(toks, "Option")
  175.                 if v!=-1:
  176.                     self.options = self.options + (v,)
  177.                     continue
  178.                     
  179.                 v = parser.findTokenValue(toks, "MPOption")
  180.                 if v!=-1:
  181.                     self.mpOptions = self.mpOptions + (v,)
  182.                     continue
  183.                     
  184.                 v = parser.findTokenValue(toks, "ForceControl")
  185.                 if v!=-1:
  186.                     self.forceControls = self.forceControls + (v,)
  187.                     continue
  188.                     
  189.                 v = parser.findTokenValue(toks, "Victory")
  190.                 if v!=-1:
  191.                     self.victories = self.victories + (v,)
  192.                     continue
  193.                     
  194.                 v = parser.findTokenValue(toks, "GameTurn")
  195.                 if v!=-1:
  196.                     self.gameTurn = int(v)
  197.                     continue
  198.  
  199.                 v = parser.findTokenValue(toks, "MaxTurns")
  200.                 if v!=-1:
  201.                     self.maxTurns = int(v)
  202.                     continue
  203.                     
  204.                 v = parser.findTokenValue(toks, "MaxCityElimination")
  205.                 if v!=-1:
  206.                     self.maxCityElimination = int(v)
  207.                     continue
  208.  
  209.                 v = parser.findTokenValue(toks, "TargetScore")
  210.                 if v!=-1:
  211.                     self.targetScore = int(v)
  212.                     continue
  213.  
  214.                 v = parser.findTokenValue(toks, "StartYear")
  215.                 if v!=-1:
  216.                     self.iStartYear = int(v)
  217.                     continue
  218.                     
  219.                 v = parser.findTokenValue(toks, "Description")
  220.                 if v!=-1:
  221.                     self.szDescription = v
  222.                     continue
  223.                     
  224.                 v = parser.findTokenValue(toks, "ModPath")
  225.                 if v!=-1:
  226.                     self.szModPath = v
  227.                     continue
  228.  
  229.                 if parser.findTokenValue(toks, "EndGame") != -1:
  230.                     break
  231.         
  232. #############
  233. class CvTeamDesc:
  234.     def __init__(self):
  235.         self.techTypes = ()
  236.         self.bContactWithTeamList = ()
  237.         self.bWarWithTeamList = ()
  238.         self.bPermanentWarPeaceList = ()
  239.         self.bOpenBordersWithTeamList = ()
  240.         self.bDefensivePactWithTeamList = ()
  241.         self.projectType = []
  242.         self.bRevealMap = 0
  243.         
  244.     def write(self, f, idx):
  245.         "write out team data"
  246.         f.write("BeginTeam\n")
  247.         
  248.         # write techs
  249.         for i in range(gc.getNumTechInfos()):
  250.             if (gc.getTeam(idx).isHasTech(i)):
  251.                 f.write("\tTech=%s\n" %(gc.getTechInfo(i).getType()))
  252.  
  253.         # write met other teams
  254.         for i in range(gc.getMAX_CIV_TEAMS()):
  255.             if (gc.getTeam(idx).isHasMet(i)):
  256.                 f.write("\tContactWithTeam=%d\n" %(i))
  257.  
  258.         # write warring teams
  259.         for i in range(gc.getMAX_CIV_TEAMS()):
  260.             if (gc.getTeam(idx).isAtWar(i)):
  261.                 f.write("\tAtWar=%d\n" %(i))
  262.             
  263.         # write permanent war/peace teams
  264.         for i in range(gc.getMAX_CIV_TEAMS()):
  265.             if (gc.getTeam(idx).isPermanentWarPeace(i)):
  266.                 f.write("\tPermanentWarPeace=%d\n" %(i))
  267.             
  268.         # write open borders other teams
  269.         for i in range(gc.getMAX_CIV_TEAMS()):
  270.             if (gc.getTeam(idx).isOpenBorders(i)):
  271.                 f.write("\tOpenBordersWithTeam=%d\n" %(i))
  272.  
  273.         # write defensive pact other teams
  274.         for i in range(gc.getMAX_CIV_TEAMS()):
  275.             if (gc.getTeam(idx).isDefensivePact(i)):
  276.                 f.write("\tDefensivePactWithTeam=%d\n" %(i))
  277.         
  278.         for i in range(gc.getNumProjectInfos()):
  279.             for j in range(gc.getTeam(idx).getProjectCount(i)):
  280.                 f.write("\tProjectType=%s\n" %(gc.getProjectInfo(i).getType()))
  281.  
  282.         f.write("\tRevealMap=%d\n" %(0))
  283.         
  284.         f.write("EndTeam\n")
  285.         
  286.     def read(self, f):
  287.         "read in team data"
  288.         self.__init__()
  289.  
  290.         parser = CvWBParser()
  291.         if (parser.findNextTokenValue(f, "BeginTeam")!=-1):
  292.             while (true):
  293.                 nextLine = parser.getNextLine(f)
  294.                 toks = parser.getTokens(nextLine)
  295.                 if (len(toks)==0):
  296.                     break
  297.                     
  298.                 v = parser.findTokenValue(toks, "Tech")
  299.                 if v!=-1:
  300.                     self.techTypes = self.techTypes + (v,)
  301.                     continue
  302.  
  303.                 v = parser.findTokenValue(toks, "ContactWithTeam")
  304.                 if v!=-1:
  305.                     self.bContactWithTeamList = self.bContactWithTeamList + (int(v),)
  306.                     continue
  307.  
  308.                 v = parser.findTokenValue(toks, "AtWar")
  309.                 if v!=-1:
  310.                     self.bWarWithTeamList = self.bWarWithTeamList + (int(v),)
  311.                     continue
  312.  
  313.                 v = parser.findTokenValue(toks, "PermanentWarPeace")
  314.                 if v!=-1:
  315.                     self.bPermanentWarPeaceList = self.bPermanentWarPeaceList + (int(v),)
  316.                     continue
  317.  
  318.                 v = parser.findTokenValue(toks, "OpenBordersWithTeam")
  319.                 if v!=-1:
  320.                     self.bOpenBordersWithTeamList = self.bOpenBordersWithTeamList + (int(v),)
  321.                     continue
  322.  
  323.                 v = parser.findTokenValue(toks, "DefensivePactWithTeam")
  324.                 if v!=-1:
  325.                     self.bDefensivePactWithTeamList = self.bDefensivePactWithTeamList + (int(v),)
  326.                     continue
  327.                 
  328.                 v = parser.findTokenValue(toks, "ProjectType")
  329.                 if v!=-1:
  330.                     self.projectType.append(v)
  331.                     continue
  332.  
  333.                 v = parser.findTokenValue(toks, "RevealMap")
  334.                 if v!=-1:
  335.                     self.bRevealMap = int(v)
  336.                     continue
  337.                 
  338.                 if parser.findTokenValue(toks, "EndTeam") != -1:
  339.                     return true        # completed successfully
  340.  
  341.         return false    # failed
  342.  
  343. #############
  344. class CvPlayerDesc:
  345.     def __init__(self):
  346.         self.szCivDesc = ""
  347.         self.szCivShortDesc = ""
  348.         self.szLeaderName = ""
  349.         self.szCivAdjective = ""
  350.         self.szFlagDecal = ""
  351.         self.isWhiteFlag = 0
  352.         
  353.         self.leaderType = "NONE"
  354.         self.civType = "NONE"
  355.         self.handicap = gc.getHandicapInfo(gc.getDefineINT("STANDARD_HANDICAP")).getType()
  356.         self.team = -1        # team index
  357.         self.color = "NONE"
  358.         self.artStyle = "NONE"
  359.         self.isPlayableCiv = 1
  360.         self.isMinorNationCiv = 0
  361.         self.iStartingGold = 0
  362.         self.iStartingX = -1
  363.         self.iStartingY = -1
  364.         self.stateReligion = ""
  365.         self.szStartingEra = ""
  366.         
  367.         self.aaiCivics = []
  368.         self.aaiAttitudeExtras = []
  369.         self.aszCityList = []
  370.  
  371.     def write(self, f, idx):
  372.         "write out player data"
  373.         f.write("BeginPlayer\n")
  374.         
  375.         # write team
  376.         f.write("\tTeam=%d\n" %(int(gc.getPlayer(idx).getTeam())))
  377.         
  378.         # write leader and Civ Description info
  379.         if (gc.getPlayer(idx).getLeaderType() == LeaderHeadTypes.NO_LEADER):
  380.             f.write("\tLeaderType=NONE\n")
  381.         
  382.         else:
  383.             f.write("\tLeaderType=%s\n" %(gc.getLeaderHeadInfo(gc.getPlayer(idx).getLeaderType()).getType()))
  384.             
  385.         # write civ, color, artStyle, isPlayableCiv, isMinorNation, StartingGold and StateReligion
  386.         if (gc.getPlayer(idx).getCivilizationType() == CivilizationTypes.NO_CIVILIZATION):
  387.             f.write("\tCivType=NONE\n")
  388.             f.write("\tColor=NONE\n")
  389.             f.write("\tArtStyle=NONE\n")
  390.         else:
  391.             f.write("\tLeaderName=%s\n" %(gc.getPlayer(idx).getName().encode(fileencoding)))
  392.             f.write("\tCivDesc=%s\n" %(gc.getPlayer(idx).getCivilizationDescription(0).encode(fileencoding)))
  393.             f.write("\tCivShortDesc=%s\n" %(gc.getPlayer(idx).getCivilizationShortDescription(0).encode(fileencoding)))
  394.             f.write("\tCivAdjective=%s\n" %(gc.getPlayer(idx).getCivilizationAdjective(0).encode(fileencoding)))
  395.             f.write("\tFlagDecal=%s\n" %(gc.getPlayer(idx).getFlagDecal().encode(fileencoding)))
  396.             f.write("\tWhiteFlag=%d\n" %(gc.getPlayer(idx).isWhiteFlag(),))
  397.             f.write("\tCivType=%s\n" %(gc.getCivilizationInfo(gc.getPlayer(idx).getCivilizationType()).getType()))
  398.             f.write("\tColor=%s\n" %(gc.getPlayerColorInfo(gc.getPlayer(idx).getPlayerColor()).getType()))
  399.             f.write("\tArtStyle=%s\n" %(gc.getArtStyleTypes(gc.getPlayer(idx).getArtStyleType())))
  400.             f.write("\tPlayableCiv=%d\n" %(int(gc.getPlayer(idx).isPlayable())))
  401.             f.write("\tMinorNationStatus=%d\n" %(gc.getPlayer(idx).isMinorCiv()))
  402.             f.write("\tStartingGold=%d\n" %(gc.getPlayer(idx).getGold()))
  403.             
  404.             pPlot = gc.getPlayer(idx).getStartingPlot()
  405.             if (not pPlot.isNone()):
  406.                 f.write("\tStartingX=%d, StartingY=%d\n" %(pPlot.getX(), pPlot.getY()))
  407.             
  408.             pPlayerReligionInfo = gc.getReligionInfo(gc.getPlayer(idx).getStateReligion())
  409.             if pPlayerReligionInfo:
  410.                 f.write("\tStateReligion=%s\n" %(pPlayerReligionInfo.getType()))
  411.             else:
  412.                 f.write("\tStateReligion=\n")
  413.                 
  414.             f.write("\tStartingEra=%s\n" %(gc.getEraInfo(gc.getPlayer(idx).getCurrentEra()).getType()))
  415.             
  416.             # write Civics
  417.             for iCivicOptionLoop in range(gc.getNumCivicOptionInfos()):
  418.                 for iCivicLoop in range(gc.getNumCivicInfos()):
  419.                     iCivic = gc.getPlayer(idx).getCivics(iCivicOptionLoop)
  420.                     if (iCivicLoop == iCivic):
  421.                         f.write("\tCivicOption=%s, Civic=%s\n" %(gc.getCivicOptionInfo(iCivicOptionLoop).getType(), gc.getCivicInfo(iCivicLoop).getType()))
  422.         
  423.             # write Attitude Extra
  424.             for i in range(gc.getMAX_CIV_PLAYERS()):
  425.                 if (gc.getPlayer(idx).AI_getAttitudeExtra(i) != 0):
  426.                     f.write("\tAttitudePlayer=%d, AttitudeExtra=%d\n" %(i, gc.getPlayer(idx).AI_getAttitudeExtra(i)))
  427.             
  428.             # write City List
  429.             for i in range(gc.getPlayer(idx).getNumCityNames()):
  430.                 f.write("\tCityList=%s\n" %(gc.getPlayer(idx).getCityName(i)))
  431.                         
  432.         if (gc.getPlayer(idx).getHandicapType() == HandicapTypes.NO_HANDICAP):
  433.             f.write("\tHandicap=%s\n" %(gc.getHandicapInfo(gc.getDefineINT("STANDARD_HANDICAP")).getType()))
  434.         else:
  435.             f.write("\tHandicap=%s\n" %(gc.getHandicapInfo(gc.getPlayer(idx).getHandicapType()).getType()))
  436.         
  437.         f.write("EndPlayer\n")
  438.         
  439.     def read(self, f):
  440.         "read in player data"
  441.         self.__init__()
  442.         parser = CvWBParser()
  443.         if (parser.findNextTokenValue(f, "BeginPlayer")!=-1):
  444.             while (true):
  445.                 nextLine = parser.getNextLine(f)
  446.                 toks = parser.getTokens(nextLine)
  447.                 if (len(toks)==0):
  448.                     break
  449.                 
  450.                 v = parser.findTokenValue(toks, "CivDesc")
  451.                 if v!=-1:
  452.                     self.szCivDesc = v.decode(fileencoding)
  453.                     continue
  454.                 
  455.                 v = parser.findTokenValue(toks, "CivShortDesc")
  456.                 if v!=-1:
  457.                     self.szCivShortDesc = v.decode(fileencoding)
  458.                     continue
  459.                 
  460.                 v = parser.findTokenValue(toks, "LeaderName")
  461.                 if v!=-1:
  462.                     self.szLeaderName = v.decode(fileencoding)
  463.                     continue
  464.                 
  465.                 v = parser.findTokenValue(toks, "CivAdjective")
  466.                 if v!=-1:
  467.                     self.szCivAdjective = v.decode(fileencoding)
  468.                     continue
  469.                     
  470.                 v = parser.findTokenValue(toks, "FlagDecal")
  471.                 if v!=-1:
  472.                     self.szFlagDecal = v.decode(fileencoding)
  473.                     continue
  474.                 
  475.                 v = parser.findTokenValue(toks, "WhiteFlag")
  476.                 if v!=-1:
  477.                     self.isWhiteFlag = int(v)
  478.                     continue
  479.                 
  480.                 v = parser.findTokenValue(toks, "LeaderType")
  481.                 if v!=-1:
  482.                     self.leaderType = v
  483.                     continue
  484.                     
  485.                 v = parser.findTokenValue(toks, "CivType")
  486.                 if v!=-1:
  487.                     self.civType = v
  488.                     continue
  489.                     
  490.                 v = parser.findTokenValue(toks, "Team")
  491.                 if v!=-1:
  492.                     self.team = int(v)
  493.                     continue
  494.                     
  495.                 v = parser.findTokenValue(toks, "Handicap")
  496.                 if v!=-1:
  497.                     self.handicap = v
  498.                     continue
  499.                     
  500.                 v = parser.findTokenValue(toks, "Color")
  501.                 if v!=-1:
  502.                     self.color = v
  503.                     continue
  504.                     
  505.                 v = parser.findTokenValue(toks, "ArtStyle")
  506.                 if v!=-1:
  507.                     self.artStyle = v
  508.                     continue
  509.  
  510.                 v = parser.findTokenValue(toks, "PlayableCiv")
  511.                 if v!=-1:
  512.                     self.isPlayableCiv = int(v)
  513.                     continue
  514.  
  515.                 v = parser.findTokenValue(toks, "MinorNationStatus")
  516.                 if v!=-1:
  517.                     self.isMinorNationCiv = int(v)
  518.                     continue
  519.  
  520.                 v = parser.findTokenValue(toks, "StartingGold")
  521.                 if v!=-1:
  522.                     self.iStartingGold = int(v)
  523.                     continue
  524.  
  525.                 vX = parser.findTokenValue(toks, "StartingX")
  526.                 vY = parser.findTokenValue(toks, "StartingY")
  527.                 if vX!=-1 and vY!=-1:
  528.                     self.iStartingX = int(vX)
  529.                     self.iStartingY = int(vY)
  530.                     continue
  531.  
  532.                 v = parser.findTokenValue(toks, "StateReligion")
  533.                 if v!=-1:
  534.                     self.stateReligion = v
  535.                     continue
  536.  
  537.                 v = parser.findTokenValue(toks, "StartingEra")
  538.                 if v!=-1:
  539.                     self.szStartingEra = v
  540.                     continue
  541.                     
  542.                 v = parser.findTokenValue(toks, "CivicOption")
  543.                 if v!=-1:
  544.                     iCivicOptionType = gc.getInfoTypeForString(v)
  545.                     
  546.                     v = parser.findTokenValue(toks, "Civic")
  547.                     if v!=-1:
  548.                         iCivicType = gc.getInfoTypeForString(v)
  549.                         self.aaiCivics.append([iCivicOptionType,iCivicType])
  550.                         continue
  551.                                         
  552.                 v = parser.findTokenValue(toks, "AttitudePlayer")
  553.                 if v!=-1:
  554.                     iPlayer = int(v)
  555.                     
  556.                     iExtra = int(parser.findTokenValue(toks, "AttitudeExtra"))
  557.                     self.aaiAttitudeExtras.append([iPlayer,iExtra])
  558.                     continue
  559.  
  560.                 v = parser.findTokenValue(toks, "CityList")
  561.                 if v!=-1:
  562.                     self.aszCityList.append(v)
  563.                     continue
  564.  
  565.                 if parser.findTokenValue(toks, "EndPlayer") != -1:
  566.                     #print("Civics:")
  567.                     #print self.aaiCivics
  568.                     #print("Attitudes:")
  569.                     #print self.aaiAttitudeExtras
  570.                     break
  571.  
  572. #############
  573. class CvUnitDesc:
  574.     "unit WB serialization"
  575.     def __init__(self):
  576.         self.plotX = -1
  577.         self.plotY = -1
  578.         self.unitType = None
  579.         self.owner =-1
  580.         self.level = -1
  581.         self.experience = -1
  582.         self.promotionType = []
  583.         self.szUnitAIType = "NO_UNITAI"
  584.         self.szScriptData = "NONE"
  585.         
  586.     def apply(self):
  587.         "after reading, this will actually apply the data"
  588.         player = getPlayer(self.owner)
  589.         if (player):
  590.             # print ("unit apply %d %d" %(self.plotX, self.plotY))
  591.             CvUtil.pyAssert(self.plotX>=0 and self.plotY>=0, "invalid plot coords")        
  592.             unitTypeNum = CvUtil.findInfoTypeNum(gc.getUnitInfo, gc.getNumUnitInfos(), self.unitType)
  593.             if (unitTypeNum < 0):
  594.                 unit = None
  595.             else:
  596. #                eUnitAI = gc.getTypesEnum(self.szUnitAIType)
  597.                 if (self.szUnitAIType != "NO_UNITAI"):
  598.                     eUnitAI = CvUtil.findInfoTypeNum(gc.getUnitAIInfo, UnitAITypes.NUM_UNITAI_TYPES, self.szUnitAIType) #pUnitAI.getType()
  599.                 else:
  600.                     eUnitAI = UnitAITypes.NO_UNITAI
  601.                     
  602.                 unit = player.initUnit(unitTypeNum, self.plotX, self.plotY, UnitAITypes(eUnitAI))#UnitAITypes.NO_UNITAI)
  603.             if (unit):
  604.                 if self.level != -1:
  605.                     unit.setLevel(self.level)
  606.                 if self.experience != -1:
  607.                     unit.setExperience(self.experience, -1)
  608.                 for promo in self.promotionType:
  609.                     promotionTypeNum = CvUtil.findInfoTypeNum(gc.getPromotionInfo, gc.getNumPromotionInfos(), promo)
  610.                     unit.setHasPromotion(promotionTypeNum, True)
  611.                 if self.szScriptData != "NONE":
  612.                     unit.setScriptData(self.szScriptData)
  613.                     
  614.     def read(self, f, pX, pY):
  615.         "read in unit data - at this point the first line 'BeginUnit' has already been read"
  616.         self.__init__()
  617.         self.plotX = pX
  618.         self.plotY = pY
  619.         CvUtil.pyAssert(self.plotX>=0 and self.plotY>=0, "invalid plot coords")
  620.         
  621.         parser = CvWBParser()
  622.         while (true):
  623.             nextLine = parser.getNextLine(f)
  624.             toks = parser.getTokens(nextLine)
  625.             if (len(toks)==0):
  626.                 break
  627.  
  628.             v = parser.findTokenValue(toks, "UnitType")
  629.             vOwner = parser.findTokenValue(toks, "UnitOwner")
  630.             if (v!=-1 and vOwner != -1):
  631.                 self.unitType = v
  632.                 self.owner = int(vOwner)
  633.                 continue
  634.                 
  635.             v = parser.findTokenValue(toks, "Level")
  636.             if (v != -1):
  637.                 self.level = (int(v))            
  638.                 self.experience = int(parser.findTokenValue(toks, "Experience"))
  639.                 continue
  640.  
  641.             # Units - Promotions
  642.             v = parser.findTokenValue(toks, "PromotionType")
  643.             if v!=-1:
  644.                 self.promotionType.append(v)
  645.                 continue
  646.                 
  647.             v = parser.findTokenValue(toks, "UnitAIType")
  648.             if (v != -1):
  649.                 self.szUnitAIType = v
  650.                 continue
  651.         
  652.             v = parser.findTokenValue(toks, "ScriptData")
  653.             if v!=-1:
  654.                 print("found script data: %s" %(v))
  655.                 self.szScriptData = v
  656.                 continue
  657.                 
  658.             if parser.findTokenValue(toks, "EndUnit") != -1:
  659.                 break
  660.  
  661.     def write(self, f, unit, plot):
  662.         "save unit desc to a file"
  663.         unitType = gc.getUnitInfo(unit.getUnitType()).getType()
  664.         unitOwner= unit.getOwner()
  665.         f.write("\tBeginUnit\n")
  666.         f.write("\t\tUnitType=%s, UnitOwner=%d\n" %(unitType,unitOwner))
  667.         f.write("\t\tLevel=%d, Experience=%d\n" %(unit.getLevel(), unit.getExperience()))
  668.         for i in range(gc.getNumPromotionInfos()):
  669.             if unit.isHasPromotion(i):
  670.                 f.write("\t\tPromotionType=%s\n" %(gc.getPromotionInfo(i).getType()))
  671.         f.write("\t\tUnitAIType=%s\n" %(gc.getUnitAIInfo(unit.getUnitAIType()).getType()))
  672.         if unit.getScriptData():
  673.             f.write("\t\tScriptData=%s\n" %unit.getScriptData() )
  674.         f.write("\tEndUnit\n")
  675.  
  676. ############
  677. class CvCityDesc:
  678.     "serializes city data" 
  679.     def __init__(self):
  680.         self.city = None
  681.         self.owner = None
  682.         self.name = None
  683.         self.population = 0
  684.         self.productionUnit = "NONE"
  685.         self.productionBuilding = "NONE"
  686.         self.productionProject = "NONE"
  687.         self.productionProcess = "NONE"
  688.         self.culture = 0
  689.         self.bldgType = []
  690.         self.religions = []
  691.         self.holyCityReligions = []
  692.         self.plotX=-1
  693.         self.plotY=-1
  694.         self.szScriptData = "NONE"
  695.         self.aiPlayerCulture = []
  696.         for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  697.             self.aiPlayerCulture.append(0)
  698.         
  699.     def apply(self):
  700.         "after reading, this will actually apply the data"
  701.         player = getPlayer(self.owner)
  702.         if (player):
  703.             self.city = player.initCity(self.plotX, self.plotY)
  704.             
  705.         if (self.name != None):
  706.             self.city.setName(self.name, False)
  707.         
  708.         if (self.population):
  709.             self.city.setPopulation(self.population)
  710.             
  711.         if (self.culture):
  712.             self.city.setCulture(self.city.getOwner(), self.culture, True)
  713.             
  714.         for bldg in (self.bldgType):
  715.             bldgTypeNum = CvUtil.findInfoTypeNum(gc.getBuildingInfo, gc.getNumBuildingInfos(), bldg)
  716.             self.city.setHasRealBuilding(bldgTypeNum, True)
  717.         
  718.         for religion in (self.religions):
  719.             religionTypeNum = CvUtil.findInfoTypeNum(gc.getReligionInfo, gc.getNumReligionInfos(), religion)
  720.             self.city.setHasReligion(religionTypeNum, true, false, true)
  721.             
  722.         for holyCityRel in (self.holyCityReligions):
  723.             religionTypeNum = CvUtil.findInfoTypeNum(gc.getReligionInfo, gc.getNumReligionInfos(), holyCityRel)
  724.             gc.getGame().setHolyCity(religionTypeNum, self.city, false)
  725.         
  726.         for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  727.             iPlayerCulture = self.aiPlayerCulture[iPlayerLoop]
  728.             if (iPlayerCulture > 0):
  729.                 self.city.setCulture(iPlayerLoop, iPlayerCulture, true)
  730.  
  731.         unitTypeNum = CvUtil.findInfoTypeNum(gc.getUnitInfo, gc.getNumUnitInfos(), self.productionUnit)
  732.         buildingTypeNum = CvUtil.findInfoTypeNum(gc.getBuildingInfo, gc.getNumBuildingInfos(), self.productionBuilding)
  733.         projectTypeNum = CvUtil.findInfoTypeNum(gc.getProjectInfo, gc.getNumProjectInfos(), self.productionProject)
  734.         processTypeNum = CvUtil.findInfoTypeNum(gc.getProcessInfo, gc.getNumProcessInfos(), self.productionProcess)
  735.         
  736.         if (unitTypeNum != UnitTypes.NO_UNIT):
  737.             self.city.pushOrder(OrderTypes.ORDER_TRAIN, unitTypeNum, -1, False, False, False)
  738.         elif (buildingTypeNum != BuildingTypes.NO_BUILDING):
  739.             self.city.pushOrder(OrderTypes.ORDER_CONSTRUCT, buildingTypeNum, -1, False, False, False)
  740.         elif (projectTypeNum != ProjectTypes.NO_PROJECT):
  741.             self.city.pushOrder(OrderTypes.ORDER_CREATE, projectTypeNum, -1, False, False, False)
  742.         elif (processTypeNum != ProcessTypes.NO_PROCESS):
  743.             self.city.pushOrder(OrderTypes.ORDER_MAINTAIN, processTypeNum, -1, False, False, False)
  744.  
  745.         if (self.szScriptData != "NONE"):
  746.             self.city.setScriptData(self.szScriptData)    
  747.         
  748.     def write(self, f, plot):
  749.         "write out city data"
  750.         city = plot.getPlotCity()
  751.         CvUtil.pyAssert(city.isNone()==0, "null city?")
  752.         f.write("\tBeginCity\n")
  753.         f.write("\t\tCityOwner=%d\n" %(city.getOwner(),))
  754.         f.write("\t\tCityName=%s\n" %(city.getName().encode(fileencoding),))
  755.         f.write("\t\tCityPopulation=%d\n" %(city.getPopulation(),))
  756.         if (city.isProductionUnit()):
  757.             f.write("\t\tProductionUnit=%s\n" %(gc.getUnitInfo(city.getProductionUnit()).getType(),))
  758.         elif (city.isProductionBuilding()):
  759.             f.write("\t\tProductionBuilding=%s\n" %(gc.getBuildingInfo(city.getProductionBuilding()).getType(),))
  760.         elif (city.isProductionProject()):
  761.             f.write("\t\tProductionProject=%s\n" %(gc.getProjectInfo(city.getProductionProject()).getType(),))
  762.         elif (city.isProductionProcess()):
  763.             f.write("\t\tProductionProcess=%s\n" %(gc.getProcessInfo(city.getProductionProcess()).getType(),))
  764. #        f.write("\t\tCityCulture=%d\n" %(city.getCulture(city.getOwner()),))
  765.         
  766.         for iI in range(gc.getNumBuildingInfos()):
  767.             if city.isHasRealBuilding(iI):
  768.                 f.write("\t\tBuildingType=%s\n" %(gc.getBuildingInfo(iI).getType()))    
  769.         
  770.         for iI in range(gc.getNumReligionInfos()):
  771.             if city.isHasReligion(iI):
  772.                 f.write("\t\tReligionType=%s\n" %(gc.getReligionInfo(iI).getType()))    
  773.             if (city.isHolyCityByType(iI)):
  774.                 f.write("\t\tHolyCityReligionType=%s\n" %(gc.getReligionInfo(iI).getType()))
  775.         
  776.         if city.getScriptData():
  777.             f.write("\t\tScriptData=%s\n" %city.getScriptData())
  778.         
  779.         # Player culture
  780.         for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  781.             iPlayerCulture = city.getCulture(iPlayerLoop)
  782.             if (iPlayerCulture > 0):
  783.                 f.write("\t\tPlayer%dCulture=%d\n" %(iPlayerLoop, iPlayerCulture))
  784.         
  785.         f.write("\tEndCity\n")
  786.                     
  787.     def read(self, f, iX, iY):
  788.         "read in city data - at this point the first line 'BeginCity' has already been read"
  789.         self.__init__()
  790.         self.plotX=iX
  791.         self.plotY=iY
  792.         parser = CvWBParser()
  793.         while (true):
  794.             nextLine = parser.getNextLine(f)
  795.             toks = parser.getTokens(nextLine)
  796.             if (len(toks)==0):
  797.                 break
  798.  
  799.             # City - Owner
  800.             vOwner=parser.findTokenValue(toks, "CityOwner")
  801.             if (vOwner != -1):
  802.                 self.owner = int(vOwner)
  803.                 continue
  804.  
  805.             # City - Name
  806.             vName=parser.findTokenValue(toks, "CityName")
  807.             if (vName != -1):
  808.                 self.name = (vName).decode(fileencoding)
  809.                 continue
  810.                 
  811.             # City - Population
  812.             v=parser.findTokenValue(toks, "CityPopulation")
  813.             if v!=-1:
  814.                 self.population = (int(v))
  815.                 continue
  816.                 
  817.             # City - Production Unit
  818.             v=parser.findTokenValue(toks, "ProductionUnit")
  819.             if v!=-1:
  820.                 self.productionUnit = v
  821.                 continue
  822.                 
  823.             # City - Production Building
  824.             v=parser.findTokenValue(toks, "ProductionBuilding")
  825.             if v!=-1:
  826.                 self.productionBuilding = v
  827.                 continue
  828.                 
  829.             # City - Production Project
  830.             v=parser.findTokenValue(toks, "ProductionProject")
  831.             if v!=-1:
  832.                 self.productionProject = v
  833.                 continue
  834.                 
  835.             # City - Production Process
  836.             v=parser.findTokenValue(toks, "ProductionProcess")
  837.             if v!=-1:
  838.                 self.productionProcess = v
  839.                 continue
  840.                 
  841.             # City - Culture XXX obsolete...
  842.             v=parser.findTokenValue(toks, "CityCulture")
  843.             if v!=-1:
  844.                 self.culture = int(v)
  845.                 continue
  846.                 
  847.             # City - Buildings
  848.             v=parser.findTokenValue(toks, "BuildingType")
  849.             if v!=-1:
  850.                 self.bldgType.append(v)
  851.                 continue
  852.     
  853.             # City - Religions
  854.             v=parser.findTokenValue(toks, "ReligionType")
  855.             if v!=-1:
  856.                 self.religions.append(v)
  857.                 continue
  858.  
  859.             # City - HolyCity
  860.             v=parser.findTokenValue(toks, "HolyCityReligionType")
  861.             if v!=-1:
  862.                 self.holyCityReligions.append(v)
  863.                 continue
  864.  
  865.             # City - ScriptData
  866.             v=parser.findTokenValue(toks, "ScriptData")
  867.             if v!=-1:
  868.                 self.szScriptData = v
  869.                 continue
  870.             
  871.             # Player Culture
  872.             for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  873.                 szCityTag = ("Player%dCulture" %(iPlayerLoop))
  874.                 v = parser.findTokenValue(toks, szCityTag)
  875.                 if v!=-1:
  876.                     self.aiPlayerCulture[iPlayerLoop] = int(v)
  877.                     continue
  878.  
  879.             if parser.findTokenValue(toks, "EndCity")!=-1:
  880.                 break        
  881.                 
  882. ###########
  883. class CvPlotDesc:
  884.     "serializes plot data"
  885.     def __init__(self):
  886.         self.iX = -1
  887.         self.iY = -1
  888.         self.riverNSDirection = CardinalDirectionTypes.NO_CARDINALDIRECTION
  889.         self.isNOfRiver = 0
  890.         self.riverWEDirection = CardinalDirectionTypes.NO_CARDINALDIRECTION
  891.         self.isWOfRiver = 0
  892.         self.isStartingPlot = 0
  893.         self.bonusType = None
  894.         self.improvementType = None
  895.         self.featureType = None
  896.         self.featureVariety = 0
  897.         self.routeType = None
  898.         self.terrainType = None
  899.         self.plotType = PlotTypes.NO_PLOT
  900.         self.unitDescs = list()
  901.         self.cityDesc = None
  902.         self.szLandmark = ""
  903.         self.szScriptData = "NONE"
  904.         self.abTeamPlotRevealed = [0]*gc.getMAX_CIV_TEAMS()
  905.         
  906.     def needToWritePlot(self, plot):
  907.         "returns true if this plot needs to be written out."
  908.         return True
  909.         
  910.     def preApply(self):
  911.         "apply plot and terrain type"
  912.         plot = CyMap().plot(self.iX, self.iY)
  913.         if (self.plotType != PlotTypes.NO_PLOT):
  914.             plot.setPlotType(self.plotType, False, False)
  915.         if (self.terrainType):
  916.             terrainTypeNum = CvUtil.findInfoTypeNum(gc.getTerrainInfo, gc.getNumTerrainInfos(), self.terrainType)
  917.             plot.setTerrainType(terrainTypeNum, False, False)
  918.  
  919.     def apply(self):
  920.         "after reading, this will actually apply the data"
  921.         #print("apply plot %d %d" %(self.iX, self.iY))                    
  922.         plot = CyMap().plot(self.iX, self.iY)
  923.         plot.setNOfRiver(self.isNOfRiver, self.riverWEDirection)
  924.         plot.setWOfRiver(self.isWOfRiver, self.riverNSDirection)
  925.         plot.setStartingPlot(self.isStartingPlot)
  926.         if (self.bonusType):
  927.             bonusTypeNum = CvUtil.findInfoTypeNum(gc.getBonusInfo, gc.getNumBonusInfos(), self.bonusType)
  928.             plot.setBonusType(bonusTypeNum)
  929.         if (self.improvementType):
  930.             improvementTypeNum = CvUtil.findInfoTypeNum(gc.getImprovementInfo, gc.getNumImprovementInfos(), self.improvementType)
  931.             plot.setImprovementType(improvementTypeNum)
  932.         if (self.featureType):
  933.             featureTypeNum = CvUtil.findInfoTypeNum(gc.getFeatureInfo, gc.getNumFeatureInfos(), self.featureType)
  934.             plot.setFeatureType(featureTypeNum, self.featureVariety)
  935.         if (self.routeType):
  936.             routeTypeNum = CvUtil.findInfoTypeNum(gc.getRouteInfo, gc.getNumRouteInfos(), self.routeType)
  937.             plot.setRouteType(routeTypeNum)
  938.         
  939.         if (self.szLandmark != ""):
  940.             plot.setLandmark(u"%s" %(self.szLandmark))
  941.         
  942.         if (self.szScriptData != "NONE"):
  943.             plot.setScriptData(self.szScriptData)
  944.  
  945.     def applyUnits(self):
  946.         #print "--apply units"        
  947.         for u in self.unitDescs:
  948.             u.apply()
  949.             
  950.     def applyCity(self):
  951.         if self.cityDesc:
  952.             #print "--apply city"
  953.             self.cityDesc.apply()
  954.  
  955.     def write(self, f, plot):
  956.         "save plot desc to a file"
  957.         f.write("BeginPlot\n")
  958.         f.write("\tx=%d,y=%d\n" %(plot.getX(), plot.getY()))    
  959.         
  960.         # landmarks
  961.         if (plot.getLandmark() != ""):
  962.             f.write("\tLandmark=%s\n" %plot.getLandmark())
  963.         # scriptData
  964.         if (plot.getScriptData() != ""):
  965.             f.write("\tScriptData=%s\n" %plot.getScriptData())
  966.         # rivers
  967.         if (plot.getRiverNSDirection() != CardinalDirectionTypes.NO_CARDINALDIRECTION):
  968.             f.write("\tRiverNSDirection=%d\n" %(int(plot.getRiverNSDirection()),) )
  969.         if (plot.isNOfRiver()):
  970.             f.write("\tisNOfRiver\n")
  971.         if (plot.getRiverWEDirection() != CardinalDirectionTypes.NO_CARDINALDIRECTION):
  972.             f.write("\tRiverWEDirection=%d\n" %(int(plot.getRiverWEDirection()),) )
  973.         if (plot.isWOfRiver()):
  974.             f.write("\tisWOfRiver\n")
  975.         # extras
  976.         if (plot.isStartingPlot()):
  977.             f.write("\tStartingPlot\n")
  978.         if (plot.getBonusType(-1)!=-1):
  979.             f.write("\tBonusType=%s\n" %(gc.getBonusInfo(plot.getBonusType(-1)).getType()) )
  980.         if (plot.getImprovementType()!=-1):
  981.             f.write("\tImprovementType=%s\n" %(gc.getImprovementInfo(plot.getImprovementType()).getType()) )
  982.         if (plot.getFeatureType()!=-1):
  983.             f.write("\tFeatureType=%s, FeatureVariety=%d\n" 
  984.             %(gc.getFeatureInfo(plot.getFeatureType()).getType(), plot.getFeatureVariety(), ) )    
  985.         if (plot.getRouteType()!=-1):
  986.             f.write("\tRouteType=%s\n" %(gc.getRouteInfo(plot.getRouteType()).getType()) )
  987.         if (plot.getTerrainType()!=-1):
  988.             f.write("\tTerrainType=%s\n" %(gc.getTerrainInfo(plot.getTerrainType()).getType()) )
  989.         if (plot.getPlotType()!=PlotTypes.NO_PLOT):
  990.             f.write("\tPlotType=%d\n" %(int(plot.getPlotType()),) )
  991.             
  992.         # units
  993.         for i in range(plot.getNumUnits()):
  994.             unit=plot.getUnit(i)
  995.             if unit.getUnitType() == -1:
  996.                 continue
  997.             CvUnitDesc().write(f, unit, plot)
  998.         # city
  999.         if (plot.isCity()):
  1000.             CvCityDesc().write(f, plot)
  1001.                 
  1002.         # Fog of War
  1003.         bFirstReveal=true
  1004.         for iTeamLoop in range(gc.getMAX_CIV_TEAMS()):
  1005.             if (gc.getTeam(iTeamLoop).isAlive()):
  1006.                 if (plot.isRevealed(iTeamLoop,0)):
  1007.                     # Plot is revealed for this Team so write out the fact that it is; if not revealed don't write anything
  1008.                     if (bFirstReveal):
  1009.                         f.write("\tTeamReveal=")
  1010.                         bFirstReveal=false
  1011.                     f.write("%d," %(iTeamLoop))                    
  1012.         if (bFirstReveal==false):
  1013.             f.write("\n")    # terminate reveal line
  1014.                 
  1015.         f.write("EndPlot\n")        
  1016.         
  1017.     def read(self, f):
  1018.         "read in a plot desc"
  1019.         self.__init__()
  1020.         parser = CvWBParser()
  1021.         if parser.findNextToken(f, "BeginPlot")==false:
  1022.             return false    # no more plots
  1023.         while (true):
  1024.             nextLine = parser.getNextLine(f)
  1025.             toks = parser.getTokens(nextLine)
  1026.             if (len(toks)==0):
  1027.                 break
  1028.             
  1029.             x = parser.findTokenValue(toks, "x")
  1030.             y = parser.findTokenValue(toks, "y")
  1031.             if (x!=-1 and y!=-1):
  1032.                 self.iX = int(x)
  1033.                 self.iY = int(y)
  1034.                 # print("plot read %d %d" %(self.iX, self.iY))
  1035.                 continue
  1036.             
  1037.             v = parser.findTokenValue(toks, "Landmark")
  1038.             if v!=-1:
  1039.                 self.szLandmark=v
  1040.                 continue
  1041.             
  1042.             v = parser.findTokenValue(toks, "ScriptData")
  1043.             if v!=-1:
  1044.                 self.szScriptData=v
  1045.                 continue
  1046.             
  1047.             v = parser.findTokenValue(toks, "RiverNSDirection")
  1048.             if v!=-1:
  1049.                 self.riverNSDirection = (CardinalDirectionTypes(v))
  1050.                 continue
  1051.                                 
  1052.             if (parser.findTokenValue(toks, "isNOfRiver"))!=-1:
  1053.                 self.isNOfRiver = (true)
  1054.                 continue
  1055.  
  1056.             v = parser.findTokenValue(toks, "RiverWEDirection")
  1057.             if v!=-1:
  1058.                 self.riverWEDirection = (CardinalDirectionTypes(v))
  1059.                 continue
  1060.                 
  1061.             if (parser.findTokenValue(toks, "isWOfRiver"))!=-1:
  1062.                 self.isWOfRiver = (true)
  1063.                 continue
  1064.                 
  1065.             if (parser.findTokenValue(toks, "StartingPlot"))!=-1:
  1066.                 self.isStartingPlot = (true)
  1067.                 continue
  1068.  
  1069.             v = parser.findTokenValue(toks, "BonusType")
  1070.             if v!=-1:
  1071.                 self.bonusType = v
  1072.                 continue
  1073.                 
  1074.             v = parser.findTokenValue(toks, "ImprovementType")
  1075.             if v!=-1:
  1076.                 self.improvementType = v
  1077.                 continue
  1078.             
  1079.             v = parser.findTokenValue(toks, "FeatureType")
  1080.             if v!=-1:
  1081.                 self.featureType = v
  1082.                 v = parser.findTokenValue(toks, "FeatureVariety")
  1083.                 if v!=-1:
  1084.                     self.featureVariety = int(v)
  1085.                 continue
  1086.  
  1087.             v = parser.findTokenValue(toks, "RouteType")
  1088.             if v!=-1:
  1089.                 self.routeType = v
  1090.                 continue
  1091.  
  1092.             v = parser.findTokenValue(toks, "TerrainType")
  1093.             if v!=-1:
  1094.                 self.terrainType = v
  1095.                 continue
  1096.  
  1097.             v = parser.findTokenValue(toks, "PlotType")
  1098.             if v!=-1:
  1099.                 self.plotType = PlotTypes(v)
  1100.                 continue
  1101.  
  1102.             # Units
  1103.             v = parser.findTokenValue(toks, "BeginUnit")
  1104.             if v!=-1:
  1105.                 unitDesc = CvUnitDesc()
  1106.                 unitDesc.read(f, self.iX, self.iY)
  1107.                 self.unitDescs.append(unitDesc)
  1108.                 continue
  1109.  
  1110.             # City
  1111.             v = parser.findTokenValue(toks, "BeginCity")
  1112.             if v!=-1:
  1113.                 self.cityDesc = CvCityDesc()
  1114.                 self.cityDesc.read(f, self.iX, self.iY)
  1115.                 continue
  1116.  
  1117.             # Fog of War
  1118.  
  1119.             v = parser.findTokenValue(toks, "TeamReveal")
  1120.             if v!=-1:
  1121.                 for iTeamLoop in toks:
  1122.                     iTeamLoop = iTeamLoop.lstrip('TeamReveal=')
  1123.                     if len(iTeamLoop):
  1124.                         self.abTeamPlotRevealed[int(iTeamLoop)] = true
  1125.                 continue
  1126.             
  1127.             if parser.findTokenValue(toks, "EndPlot")!=-1:
  1128.                 break
  1129.         return True
  1130.         
  1131. ################
  1132. class CvMapDesc:
  1133.     "serialize map data"
  1134.     def __init__(self):
  1135.         self.iGridW = 0
  1136.         self.iGridH = 0
  1137.         self.iTopLatitude = 0
  1138.         self.iBottomLatitude = 0
  1139.         self.bWrapX = 0
  1140.         self.bWrapY = 0
  1141.         self.worldSize = None
  1142.         self.climate = None
  1143.         self.seaLevel = None
  1144.         self.numPlotsWritten = 0
  1145.         
  1146.     def write(self, f):
  1147.         "write map data"
  1148.         map = CyMap()
  1149.         iGridW = map.getGridWidth()
  1150.         iGridH = map.getGridHeight()
  1151.         iNumPlots = iGridW * iGridH
  1152.         
  1153.         f.write("BeginMap\n")
  1154.         f.write("\tgrid width=%d\n" %(map.getGridWidth(),))
  1155.         f.write("\tgrid height=%d\n" %(map.getGridHeight(),))
  1156.         f.write("\ttop latitude=%d\n" %(map.getTopLatitude(),))
  1157.         f.write("\tbottom latitude=%d\n" %(map.getBottomLatitude(),))
  1158.         f.write("\twrap X=%d\n" %(map.isWrapX(),))
  1159.         f.write("\twrap Y=%d\n" %(map.isWrapY(),))
  1160.         f.write("\tworld size=%s\n" %(gc.getWorldInfo(map.getWorldSize()).getType(),))
  1161.         f.write("\tclimate=%s\n" %(gc.getClimateInfo(map.getClimate()).getType(),))
  1162.         f.write("\tsealevel=%s\n" %(gc.getSeaLevelInfo(map.getSeaLevel()).getType(),))
  1163.         f.write("\tnum plots written=%d\n" %(iNumPlots,))
  1164.         f.write("EndMap\n")
  1165.         
  1166.     def read(self, f):
  1167.         "read map data"        
  1168.         self.__init__()
  1169.         parser = CvWBParser()
  1170.         if parser.findNextToken(f, "BeginMap")==false:
  1171.             print "can't find map"
  1172.             return
  1173.         while (true):
  1174.             nextLine = parser.getNextLine(f)
  1175.             toks = parser.getTokens(nextLine)
  1176.             if (len(toks)==0):
  1177.                 break
  1178.                             
  1179.             v = parser.findTokenValue(toks, "grid width")
  1180.             if v!=-1:
  1181.                 self.iGridW = int(v)
  1182.                 continue
  1183.  
  1184.             v = parser.findTokenValue(toks, "grid height")
  1185.             if v!=-1:
  1186.                 self.iGridH = int(v)
  1187.                 continue
  1188.  
  1189.             v = parser.findTokenValue(toks, "top latitude")
  1190.             if v!=-1:
  1191.                 self.iTopLatitude = int(v)
  1192.                 continue
  1193.  
  1194.             v = parser.findTokenValue(toks, "bottom latitude")
  1195.             if v!=-1:
  1196.                 self.iBottomLatitude = int(v)
  1197.                 continue
  1198.  
  1199.             v = parser.findTokenValue(toks, "wrap X")
  1200.             if v!=-1:
  1201.                 self.bWrapX = int(v)
  1202.                 continue
  1203.  
  1204.             v = parser.findTokenValue(toks, "wrap Y")
  1205.             if v!=-1:
  1206.                 self.bWrapY = int(v)
  1207.                 continue
  1208.  
  1209.             v = parser.findTokenValue(toks, "world size")
  1210.             if v!=-1:
  1211.                 self.worldSize = v
  1212.                 continue
  1213.                 
  1214.             v = parser.findTokenValue(toks, "climate")
  1215.             if v!=-1:
  1216.                 self.climate = v
  1217.                 continue
  1218.                 
  1219.             v = parser.findTokenValue(toks, "sealevel")
  1220.             if v!=-1:
  1221.                 self.seaLevel = v
  1222.                 continue
  1223.                 
  1224.             v = parser.findTokenValue(toks, "num plots written")
  1225.             if v!=-1:
  1226.                 self.numPlotsWritten = int(v)
  1227.                 continue
  1228.             
  1229.             if parser.findTokenValue(toks, "EndMap")!=-1:
  1230.                 break
  1231.                                     
  1232. class CvWBDesc:
  1233.     "handles saving/loading a worldbuilder description file"
  1234.     def __init__(self):
  1235.         # game data
  1236.         self.gameDesc = CvGameDesc()
  1237.         self.playersDesc = None
  1238.         self.plotDesc = None    # list
  1239.         self.mapDesc = CvMapDesc()
  1240.         
  1241.     def getVersion(self):
  1242.         return version
  1243.         
  1244.     def getDescFileName(self, fileName):
  1245.         return fileName+getWBSaveExtension()
  1246.                                             
  1247.     def write(self, fileName):
  1248.         "Save out a high-level desc of the world, and height/terrainmaps"        
  1249.         fileName = os.path.normpath(fileName)
  1250.         fileName,ext = os.path.splitext(fileName)
  1251.         CvUtil.pyPrint( 'saveDesc:%s, curDir:%s' %(fileName,os.getcwd()) )
  1252.  
  1253.         f = file(self.getDescFileName(fileName), "w")
  1254.         f.write("Version=%d\n" %(self.getVersion(),))        
  1255.         self.gameDesc.write(f)    # write game info
  1256.  
  1257.         for i in range(gc.getMAX_CIV_TEAMS()):    
  1258.             CvTeamDesc().write(f, i)        # write team info
  1259.  
  1260.         for i in range(gc.getMAX_CIV_PLAYERS()):    
  1261.             CvPlayerDesc().write(f, i)        # write player info
  1262.             
  1263.         self.mapDesc.write(f)    # write map info
  1264.         
  1265.         f.write("\n### Plot Info ###\n")        
  1266.         iGridW = CyMap().getGridWidth()
  1267.         iGridH = CyMap().getGridHeight()
  1268.         for iX in range(iGridW):
  1269.             for iY in range(iGridH):
  1270.                 plot = CyMap().plot(iX, iY)
  1271.                 pDesc = CvPlotDesc()
  1272.                 if pDesc.needToWritePlot(plot): 
  1273.                     pDesc.write(f, plot)
  1274.  
  1275.         f.close()
  1276.         
  1277.         print("WBSave done\n")
  1278.         return 0    # success
  1279.  
  1280.     def applyMap(self):
  1281.         "after reading setup the map"
  1282.         
  1283.         self.gameDesc.apply()
  1284.         
  1285.         # recreate map        
  1286.         print("map rebuild. gridw=%d, gridh=%d" %(self.mapDesc.iGridW, self.mapDesc.iGridH))
  1287.         worldSizeType = CvUtil.findInfoTypeNum(gc.getWorldInfo, gc.getNumWorldInfos(), self.mapDesc.worldSize)
  1288.         climateType = CvUtil.findInfoTypeNum(gc.getClimateInfo, gc.getNumClimateInfos(), self.mapDesc.climate)
  1289.         seaLevelType = CvUtil.findInfoTypeNum(gc.getSeaLevelInfo, gc.getNumSeaLevelInfos(), self.mapDesc.seaLevel)
  1290.         CyMap().rebuild(self.mapDesc.iGridW, self.mapDesc.iGridH, self.mapDesc.iTopLatitude, self.mapDesc.iBottomLatitude, self.mapDesc.bWrapX, self.mapDesc.bWrapY, WorldSizeTypes(worldSizeType), ClimateTypes(climateType), SeaLevelTypes(seaLevelType), 0, None)
  1291.         
  1292.         print "preapply plots"
  1293.         for pDesc in self.plotDesc:
  1294.             pDesc.preApply()    # set plot type / terrain type
  1295.  
  1296.         print("map apply - recalc areas/regions")
  1297.         CyMap().recalculateAreas()
  1298.  
  1299.         print "apply plots"
  1300.         for pDesc in self.plotDesc:
  1301.             pDesc.apply()
  1302.         
  1303.         print ("WB apply done\n")
  1304.         return 0    # ok
  1305.         
  1306.     def getAssignedStartingPlots(self):
  1307.         "add player starting plots if using random civs"
  1308.         
  1309.         # Player stuff
  1310.         for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  1311.             
  1312.             pPlayer = gc.getPlayer(iPlayerLoop)
  1313.             pWBPlayer = self.playersDesc[iPlayerLoop]
  1314.             
  1315.             # Player's starting plot
  1316.             if ((pWBPlayer.iStartingX != -1) and (pWBPlayer.iStartingY != -1)):
  1317.                 pPlayer.setStartingPlot(CyMap().plot(pWBPlayer.iStartingX, pWBPlayer.iStartingY), True)
  1318.         
  1319.         return 0    # ok
  1320.             
  1321.     def applyInitialItems(self):
  1322.         "add player objects in a last pass"
  1323.         
  1324.         # Team stuff
  1325.         if (len(self.teamsDesc)) :
  1326.             for iTeamLoop in range(gc.getMAX_CIV_TEAMS()):
  1327.                 
  1328.                 if (self.teamsDesc[iTeamLoop]):
  1329.                     
  1330.                     pTeam = gc.getTeam(iTeamLoop)
  1331.                     pWBTeam = self.teamsDesc[iTeamLoop]
  1332.                     
  1333.                     # Techs
  1334.                     for techTypeTag in self.teamsDesc[iTeamLoop].techTypes:
  1335.                         techType = CvUtil.findInfoTypeNum(gc.getTechInfo, gc.getNumTechInfos(), techTypeTag)
  1336.                         gc.getTeam(iTeamLoop).setHasTech(techType, true, PlayerTypes.NO_PLAYER, false, false)
  1337.                     
  1338.                     # Contact with Other Teams
  1339.                     for meetTeam in self.teamsDesc[iTeamLoop].bContactWithTeamList:
  1340.                         gc.getTeam(iTeamLoop).meet(meetTeam, false)
  1341.                     
  1342.                     # Wars
  1343.                     for warTeam in self.teamsDesc[iTeamLoop].bWarWithTeamList:
  1344.                         gc.getTeam(iTeamLoop).declareWar(warTeam, false)
  1345.                     
  1346.                     # Permanent War/Peace
  1347.                     for permanentWarPeaceTeam in self.teamsDesc[iTeamLoop].bPermanentWarPeaceList:
  1348.                         gc.getTeam(iTeamLoop).setPermanentWarPeace(permanentWarPeaceTeam, true)
  1349.                     
  1350.                     # Open Borders
  1351.                     for openBordersTeam in self.teamsDesc[iTeamLoop].bOpenBordersWithTeamList:
  1352.                         gc.getTeam(iTeamLoop).signOpenBorders(openBordersTeam)
  1353.                     
  1354.                     # Defensive Pacts
  1355.                     for defensivePactTeam in self.teamsDesc[iTeamLoop].bDefensivePactWithTeamList:
  1356.                         gc.getTeam(iTeamLoop).signDefensivePact(defensivePactTeam)
  1357.                     
  1358.                     # Projects
  1359.                     for project in (self.teamsDesc[iTeamLoop].projectType):
  1360.                         projectTypeNum = CvUtil.findInfoTypeNum(gc.getProjectInfo, gc.getNumProjectInfos(), project)
  1361.                         gc.getTeam(iTeamLoop).changeProjectCount(projectTypeNum, 1)
  1362.                     
  1363.         # Player stuff
  1364.         if (len(self.playersDesc)) :
  1365.             for iPlayerLoop in range(gc.getMAX_CIV_PLAYERS()):
  1366.             
  1367.                 if (self.playersDesc[iPlayerLoop]):
  1368.                     
  1369.                     pPlayer = gc.getPlayer(iPlayerLoop)
  1370.                     pWBPlayer = self.playersDesc[iPlayerLoop]
  1371.                     
  1372.                     # Player's starting gold
  1373.                     pPlayer.setGold(pWBPlayer.iStartingGold)
  1374.                     
  1375.                     # Player's starting plot
  1376.                     if ((pWBPlayer.iStartingX != -1) and (pWBPlayer.iStartingY != -1)):
  1377.                         pPlayer.setStartingPlot(CyMap().plot(pWBPlayer.iStartingX, pWBPlayer.iStartingY), True)
  1378.                     
  1379.                     # State Religion
  1380.                     if (pWBPlayer.stateReligion != ""):
  1381.                         iStateReligionID = gc.getInfoTypeForString(pWBPlayer.stateReligion)
  1382.                         pPlayer.setLastStateReligion(iStateReligionID)
  1383.                         
  1384.                     # Starting Era
  1385.                     if (pWBPlayer.szStartingEra != ""):
  1386.                         iStartingEra = gc.getInfoTypeForString(pWBPlayer.szStartingEra)
  1387.                         pPlayer.setCurrentEra(iStartingEra)
  1388.                         
  1389.                     # Civics
  1390.                     for iCivicLoop in range(len(pWBPlayer.aaiCivics)):
  1391.                         iCivicOption = pWBPlayer.aaiCivics[iCivicLoop][0]
  1392.                         iCivic = pWBPlayer.aaiCivics[iCivicLoop][1]
  1393.                         pPlayer.setCivics(iCivicOption,iCivic)
  1394.                         
  1395.                     # Attitude Extras
  1396.                     for iAttitudeLoop in range(len(pWBPlayer.aaiAttitudeExtras)):
  1397.                         iPlayer = pWBPlayer.aaiAttitudeExtras[iAttitudeLoop][0]
  1398.                         iExtra = pWBPlayer.aaiAttitudeExtras[iAttitudeLoop][1]
  1399.                         pPlayer.AI_setAttitudeExtra(iPlayer,iExtra)
  1400.  
  1401.                     # City List
  1402.                     for iCityListLoop in range(len(pWBPlayer.aszCityList)):
  1403.                         pPlayer.addCityName(pWBPlayer.aszCityList[iCityListLoop])
  1404.         
  1405.         # cities
  1406.         for pDesc in self.plotDesc:
  1407.             pDesc.applyCity()
  1408.         
  1409.         # Team stuff
  1410.         if (len(self.teamsDesc)) :
  1411.             for iTeamLoop in range(gc.getMAX_CIV_TEAMS()):
  1412.                 
  1413.                 if (self.teamsDesc[iTeamLoop]):
  1414.                     
  1415.                     pTeam = gc.getTeam(iTeamLoop)
  1416.                     pWBTeam = self.teamsDesc[iTeamLoop]
  1417.                     
  1418.                     # Reveal whole map
  1419.                     if (pWBTeam.bRevealMap == 1):
  1420.                         
  1421.                         for iPlotX in range(CyMap().getGridWidth()):
  1422.                             for iPlotY in range(CyMap().getGridHeight()):
  1423.                                 pPlot = CyMap().plot(iPlotX,iPlotY)
  1424.                                 pPlot.setRevealed(pTeam.getID(), True, False, TeamTypes.NO_TEAM)
  1425.  
  1426.         # Per plot stuff
  1427.         for iPlotLoop in range(self.mapDesc.numPlotsWritten):
  1428.             
  1429.             pWBPlot = self.plotDesc[iPlotLoop]
  1430.             
  1431.             # Reveal Fog of War for teams
  1432.             for iTeamLoop in range(gc.getMAX_CIV_TEAMS()):
  1433.                 pTeam = gc.getTeam(iTeamLoop)
  1434.                 if (pWBPlot.abTeamPlotRevealed[iTeamLoop] == 1):
  1435.                     
  1436.                     pPlot = CyMap().plot(pWBPlot.iX, pWBPlot.iY)                    
  1437.                     pPlot.setRevealed(pTeam.getID(), True, False, TeamTypes.NO_TEAM)
  1438.         
  1439.         # units
  1440.         for pDesc in self.plotDesc:
  1441.             pDesc.applyUnits()
  1442.  
  1443.         return 0    # ok
  1444.         
  1445.     def read(self, fileName):
  1446.         "Load in a high-level desc of the world, and height/terrainmaps"        
  1447.         fileName = os.path.normpath(fileName)
  1448.         fileName,ext=os.path.splitext(fileName)            
  1449.         CvUtil.pyPrint( 'loadDesc:%s, curDir:%s' %(fileName,os.getcwd()) )
  1450.     
  1451.         if (not os.path.isfile(self.getDescFileName(fileName))):
  1452.             CvUtil.pyPrint("Error: file %s does not exist" %(self.getDescFileName(fileName),))
  1453.             return -1    # failed
  1454.                 
  1455.         f=file(self.getDescFileName(fileName), "r")        # open text file        
  1456.  
  1457.         parser = CvWBParser()
  1458.         version = int(parser.findNextTokenValue(f, "Version"))
  1459.         if (version != self.getVersion()):
  1460.             CvUtil.pyPrint("Error: wrong WorldBuilder save version.  Expected %d, got %d" %(self.getVersion(), version))
  1461.             return -1    # failed
  1462.             
  1463.         print "Reading game desc"
  1464.         self.gameDesc.read(f)    # read game info
  1465.         
  1466.         print "Reading teams desc"
  1467.         filePos = f.tell()
  1468.         self.teamsDesc = []
  1469.         for i in range(gc.getMAX_CIV_TEAMS()):    
  1470.             print ("reading team %d" %(i))
  1471.             teamsDesc = CvTeamDesc()
  1472.             if (teamsDesc.read(f)==false):                    # read team info        
  1473.                 f.seek(filePos)                                # abort and backup
  1474.                 break
  1475.             self.teamsDesc.append(teamsDesc)
  1476.  
  1477.         print "Reading players desc"
  1478.         self.playersDesc = []
  1479.         for i in range(gc.getMAX_CIV_PLAYERS()):    
  1480.             playerDesc = CvPlayerDesc()
  1481.             playerDesc.read(f)                # read player info        
  1482.             self.playersDesc.append(playerDesc)
  1483.             
  1484.         print "Reading map desc"
  1485.         self.mapDesc.read(f)    # read map info
  1486.             
  1487.         print("Reading/creating %d plot descs" %(self.mapDesc.numPlotsWritten,))
  1488.         self.plotDesc = []
  1489.         for i in range(self.mapDesc.numPlotsWritten):
  1490.             pDesc = CvPlotDesc()
  1491.             if pDesc.read(f)==True:
  1492.                 self.plotDesc.append(pDesc)
  1493.             else:
  1494.                 break
  1495.                                 
  1496.         f.close()
  1497.         print ("WB read done\n")
  1498.         return 0
  1499.         
  1500.