home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / pyWB / CvWBPopups.py < prev   
Encoding:
Python Source  |  2005-11-09  |  4.6 KB  |  137 lines
  1. ## Sid Meier's Civilization 4
  2. ## Copyright Firaxis Games 2005
  3. from CvPythonExtensions import *
  4. import CvUtil
  5. import Popup as PyPopup
  6. import PyHelpers
  7.  
  8. PyPlayer = PyHelpers.PyPlayer
  9. PyCity = PyHelpers.PyCity
  10.  
  11. gc = CyGlobalContext()
  12. local = CyTranslator()
  13.  
  14. def isWBPopup(context):
  15.     "helper for determining if context is a WBPopup"
  16.     return (context >= CvUtil.PopupTypeWBContextStart and context <= CvUtil.PopupTypeWBContextEnd)
  17.  
  18. class CvWBPopups:
  19.     "World Builder App Popups"
  20.     
  21.     # These ar popup types, not popup context IDs
  22.     WBPopupEditCity         = WorldBuilderPopupTypes.WBPOPUP_CITY
  23.     WBPopupEditUnit         = WorldBuilderPopupTypes.WBPOPUP_UNIT
  24.     WBPopupEditPlot         = WorldBuilderPopupTypes.WBPOPUP_PLOT
  25.     WBPopupEditPlayer        = WorldBuilderPopupTypes.WBPOPUP_PLAYER                            
  26.         
  27.     def __init__(self):
  28.         "set global instance items"
  29.     
  30.         self.WBPopupDictionary =     {
  31.             self.WBPopupEditCity :     { 'Init' : self.initEditCity, 'Apply' : self.applyEditCity },
  32.             }
  33.  
  34.     def initWB(self, argsList):
  35.         "called from Interface - use the PopupDictionary for quick access"
  36.         context = argsList[0]
  37.         entry = self.WBPopupDictionary.get(context)
  38.         
  39.         if (entry):
  40.             init = entry.get('Init')
  41.             return init(argsList)
  42.         return 0
  43.         
  44.     def applyWB(self, argsList):
  45.         "Apply the appropriate Popup using the PopupDictionary"
  46.         context = argsList[0]
  47.         entry = self.WBPopupDictionary.get(context-CvUtil.PopupTypeWBContextStart+WorldBuilderPopupTypes.WBPOPUP_START)
  48.         
  49.         if (entry):
  50.             apply = entry.get('Apply')
  51.             return apply(argsList)
  52.         return 0
  53.     
  54.     
  55.     ########################################################################################
  56.     def initEditCity(self, argsList):
  57.         ' Initializes World Builder City Edit Popup '
  58.         px,py = argsList
  59.         pCity = CyMap().plot(px,py).getPlotCity()
  60.         iOwner = pCity.getOwner()
  61.         iID = pCity.getID()
  62.         userData = (iOwner, iID)
  63.         pCity = PyCity(iOwner, iID)
  64.         
  65.         # create popup
  66.         popup = PyPopup.PyPopup( CvUtil.EventEditCity, EventContextTypes.EVENTCONTEXT_ALL )
  67.         popup.setSize(400,600)
  68.         popup.setUserData( userData )
  69.         popup.setHeaderString(local.getText("TXT_KEY_WB_HEADER_CITY_EDIT", ()))
  70.         # City Name - Editbox 0
  71.         popup.createEditBox( pCity.getName(), 0 )
  72.         # Population - Editbox 1
  73.         popup.setBodyString(local.getText("TXT_KEY_WB_CITY_EDIT_POP", ()))
  74.         popup.createEditBox( '0', 1 )
  75.         # Culture - Editbox 2
  76.         popup.setBodyString(local.getText("TXT_KEY_WB_CITY_EDIT_CULTURE", (pCity.getCulture(),)))
  77.         popup.createEditBox( '0', 2)
  78.         # Buildings - Listboxes Group
  79.         popup.createListBox(0)
  80.         iNumBuildings = gc.getNumBuildingInfos()
  81.         lBuildings = []
  82.         for i in range( iNumBuildings ):
  83.             # ('Library', iIndex)
  84.             lBuildings.append( (str(gc.getBuildingInfo(i).getDescription()), i) )
  85.         lBuildings.sort()
  86.         
  87.         popup.addListBoxString( local.getText("TXT_KEY_WB_CITY_NOTHING", ()), -1)
  88.         popup.addListBoxString( local.getText("TXT_KEY_WB_CITY_ALL", ()), iNumBuildings)        #for adding/removing every building
  89.         for i in range( len(lBuildings) ):
  90.             entry = lBuildings[i]
  91.             popup.addListBoxString(entry[0], entry[1])
  92.  
  93.         # Add/Remove - Pulldown Group 0
  94.         popup.setBodyString(local.getText("TXT_KEY_WB_CITY_ADD_REMOVE", ()))
  95.         popup.createPullDown(0)
  96.         popup.addPullDownString( local.getText("TXT_KEY_WB_CITY_ADD", ()), 1)
  97.         popup.addPullDownString( local.getText("TXT_KEY_WB_CITY_NOCHANGE", ()), 0)  #for clean exit
  98.         popup.addPullDownString( local.getText("TXT_KEY_WB_CITY_REMOVE", ()), 2)
  99.  
  100.         # Launch Popup
  101.         popup.launch()
  102.         return 0
  103.     
  104.     def applyEditCity( self, argsList):
  105.         ' Apply World Builder City Edit'
  106.         popupReturn, userData = argsList
  107.         iOwner, iID = userData
  108.         pCity = PyCity( iOwner, iID )
  109.         
  110.         if pCity.isNone():
  111.             return 0
  112.         
  113.         # EDITABLE: Name, Population, Culture, Buildings
  114.         # Name
  115.         newName = str(popupReturn.getEditBoxString( 0 ))
  116.         currName = pCity.getName()
  117.         if ( newName != currName ):
  118.             pCity.setName( newName, False )
  119.         
  120.         # Population
  121.         PopModifier = int(popupReturn.getEditBoxString( 1 ))
  122.         if PopModifier:
  123.             pCity.changePopulation( PopModifier )
  124.         
  125.         iCultureModifier = int(popupReturn.getEditBoxString( 2 ))
  126.         if iCultureModifier:
  127.             pCity.setCulture( iCultureModifier )
  128.         
  129.         # Buildings
  130.         iNumBuildings = gc.getNumBuildingInfos()    #get total # of units from Game
  131.         if ( popupReturn.getSelectedListBoxValue(0)>=0 and popupReturn.getSelectedPullDownValue(0)>0 ):
  132.             BuildingIdx = popupReturn.getSelectedListBoxValue(0)
  133.             if (BuildingIdx == iNumBuildings+1):
  134.                 return 0
  135.             AddRemoveIdx = popupReturn.getSelectedPullDownValue(0)
  136.             CvUtil.AdjustBuilding(AddRemoveIdx==1, BuildingIdx==iNumBuildings, BuildingIdx, pCity)
  137.         return 0