home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / CvDebugTools.py < prev    next >
Encoding:
Python Source  |  2005-11-09  |  10.0 KB  |  321 lines
  1. ## Sid Meier's Civilization 4
  2. ## Copyright Firaxis Games 2005
  3. #
  4. # CIV Python Tools Code
  5. #
  6. import CvUtil
  7. import PyHelpers
  8. import Popup as PyPopup
  9. import CvScreenEnums
  10. from CvPythonExtensions import *
  11.  
  12. PyPlayer = PyHelpers.PyPlayer
  13. PyGame = PyHelpers.PyGame
  14.  
  15. # globals
  16. gc = CyGlobalContext()
  17. iLastSelectedObject = -1
  18. gSetUnit = 0
  19.  
  20. def toggleDebugMode():
  21.     return
  22.  
  23. def adjustUnitMovement(iPlayer, iUnitType, iNumMoves):
  24.     player = PyPlayer(iPlayer)
  25.     for unit in player.getUnitList():
  26.         if ( int(unit.getUnitType()) == iUnitType ):
  27.             unit.setMoves(iNumMoves)
  28.  
  29. def toggleFogOfWar():
  30.     global g_bToggleFog
  31.     g_bToggleFog = not g_bToggleFog
  32.     CyEngine().setFogOfWar(g_bToggleFog)
  33.     CyInterfaceScreen("MainInterface", CvScreenEnums.MAIN_INTERFACE).updateMinimapSection(True)
  34.  
  35. def giveUnitsLotsOfMoves():
  36.     playerTeam = gc.getActivePlayer().getTeam(0)
  37.     playerTeam.changeExtraMoves(DomainTypes.DOMAIN_LAND, 1000)
  38.     playerTeam.changeExtraMoves(DomainTypes.DOMAIN_SEA, 1000)
  39.     playerTeam.changeExtraMoves(DomainTypes.DOMAIN_AIR, 1000)
  40.     
  41. ############################### DEBUG TOOLS ####################################
  42. class CvDebugTools:
  43.     def __init__(self):
  44.         self._bDebugMode = False
  45.         self.iActiveEffect = -1
  46.         self.pEffectPlot = None
  47.     
  48.     def getDebugMode( self ):
  49.         return self._bDebugMode
  50.         
  51.     def setDebugMode( self, bVal ):
  52.         self._bDebugMode = bVal        
  53.         CyInterface().addImmediateMessage( "CvDebugTools.setDebugMode set to %s" % self.bDebugMode, "" )
  54.  
  55.     def notifyInput( self, argsList ):
  56.         #print "Python Debug Mode Notify"
  57.         return 0
  58.     
  59.     def initEffectViewer( self, argsList ):
  60.         px,py = argsList
  61.         pPlot = CyMap().plot(px,py)
  62.         popup = PyPopup.PyPopup( CvUtil.PopupTypeEffectViewer, EventContextTypes.EVENTCONTEXT_SELF )
  63.         popup.setSize(550,300)
  64.         popup.setUserData( (px,py) )
  65.         popup.setHeaderString( "Python Debug Tools: Object Placer" )
  66.         # Pulldown0 - Player Selection
  67.         numEffects = gc.getNumEffectInfos()    # get total # of units from Game
  68.         
  69.         popup.createPythonPullDown("Choose an Effect")
  70.         for i in range(gc.getNumEffectInfos()):
  71.             popup.addPullDownString(gc.getEffectInfo(i).getType(), i)
  72.         
  73.         popup.createPythonEditBox( "Default", "Modify the scale of the effect" )
  74.         popup.createPythonEditBox( "Default", "Modify the update rate" )
  75.         
  76.         # Launch Popup
  77.         popup.launch()
  78.         return 0
  79.     
  80.     def applyEffectViewer(self, playerID, userData, popupReturn):
  81.         px,py = userData
  82.         self.pEffectPlot = CyMap().plot(px,py)
  83.         
  84.         if self.pEffectPlot.isNone():
  85.             return 0
  86.         
  87.         self.iActiveEffect = popupReturn.getSelectedPullDownValue( 0 )
  88.         
  89.         CyEngine().triggerEffect(self.iActiveEffect, self.pEffectPlot.getPoint())
  90.         #scale = popupReturn.getSelectedListBoxValue( 0 )
  91.         #updateRate = int( popupReturn.getEditBoxString( 0 ) )
  92.  
  93.     ############################
  94.     ## UNIT / CITY PLAYER 
  95.     ############################
  96.     def initUnitPicker( self, argsList ):
  97.         'initUnitPicker - for placing units & cities'
  98.         px,py = argsList
  99.         pPlot = CyMap().plot(px,py)
  100.         popup = PyPopup.PyPopup( CvUtil.EventPlaceObject, EventContextTypes.EVENTCONTEXT_ALL )
  101.         #popup.setSize(1000,800)
  102.         popup.setUserData( (px,py) )
  103.         popup.setHeaderString( "Python Debug Tools: Object Placer" )
  104.         #popup.setBodyString( "Choose Player:" )
  105.  
  106.         # Pulldown0 - Player Selection
  107.         iNumUnits = gc.getNumUnitInfos()    # get total # of units from Game
  108.         iOwner = pPlot.getOwner()
  109.         if ( iOwner == PlayerTypes.NO_PLAYER ):
  110.             iOwner = gc.getGame().getActivePlayer()
  111.         popup.createPythonPullDown("Choose a Player")
  112.         popup.addPullDownString(gc.getPlayer(iOwner).getName(), iOwner)
  113.         for i in range(gc.getMAX_PLAYERS()):
  114.             if (gc.getPlayer(i).isEverAlive()):
  115.                 if (i != iOwner):
  116.                     popup.addPullDownString(gc.getPlayer(i).getName(), i)
  117.  
  118.         popup.addSeparator()
  119.         
  120.         # ListBox0 - Unit List w/ City also selectable
  121.         #popup.setBodyString( "Select Game Object to Add:" )
  122.  
  123.         popup.createPythonListBox( "" )
  124.         popup.addListBoxString( 'Nothing', iNumUnits + 1 )   # for clean exit
  125.         popup.addSeparator()
  126.         lastSelection = u""
  127.         if ( iLastSelectedObject == -1 ):
  128.             pass
  129.         elif ( iLastSelectedObject == iNumUnits+1 ):
  130.             lastSelection = u'Nothing'
  131.         elif ( iLastSelectedObject == iNumUnits ):
  132.             lastSelection = u'City'
  133.         else:
  134.             lastSelection = gc.getUnitInfo(iLastSelectedObject).getDescription()
  135.         if ( not iLastSelectedObject == -1 ):
  136.             popup.addListBoxString( lastSelection, iLastSelectedObject )
  137.         popup.addListBoxString( u'City', iNumUnits )        # list City first
  138.  
  139.         # sort units alphabetically
  140.         unitsList=[(0,0)]*iNumUnits
  141.         for j in range( iNumUnits ):
  142.             unitsList[j] = (gc.getUnitInfo(j).getDescription(), j)
  143.         unitsList.sort()    
  144.  
  145.         for j in range( iNumUnits ):            
  146.             popup.addListBoxString( unitsList[j][0], unitsList[j][1])
  147.         
  148.         # EditBox0 - Customize how many units to build
  149.         #popup.setBodyString( "How many objects?" )
  150.         popup.createPythonEditBox( "1", "This allows you to create multiple units." )
  151.         
  152.         # Launch Popup
  153.         #popup.setSize( 400, 600 )
  154.         popup.launch()
  155.         return 0
  156.     
  157.     def applyUnitPicker( self, argsList ):
  158.         'Apply Unit Picker'
  159.         popupReturn, userData = argsList
  160.         px,py = userData
  161.         pPlot = CyMap().plot(px,py)
  162.         
  163.         if pPlot.isNone():
  164.             return 0
  165.         
  166.         # UNIT DEBUG MENU
  167.         playerID = popupReturn.getSelectedPullDownValue( 0 )
  168.         selectedObject = popupReturn.getSelectedListBoxValue( 0 )
  169.         iSpawnNum = int( popupReturn.getEditBoxString( 0 ) )
  170.         
  171.         player = PyPlayer( playerID )
  172.         if ( player.isNone() ):
  173.             return -1   # Error
  174.             
  175.         iNumUnits = gc.getNumUnitInfos()
  176.         global iLastSelectedObject
  177.         iLastSelectedObject = selectedObject
  178.         if ( selectedObject != iNumUnits + 1 ):# Nothing
  179.             
  180.             if ( selectedObject == iNumUnits ):# City"
  181.                 player.initCity( px,py )
  182.             
  183.             else:
  184.                 player.initUnit( selectedObject, px, py, iSpawnNum )
  185.         else:
  186.             iLastSelectedObject = -1
  187.                 
  188.         return 0    
  189.  
  190.     ############################
  191.     ## TECH / GOLD CHEAT POPUP
  192.     ############################
  193.     def cheatTechs( self ):
  194.         'Cheat techs and gold to the players'
  195.         popup = PyPopup.PyPopup( CvUtil.EventAwardTechsAndGold, EventContextTypes.EVENTCONTEXT_ALL )
  196.         popup.setHeaderString( "Tech & Gold Cheat!" )
  197.         popup.createPullDown()
  198.         popup.addPullDownString( "All", gc.getMAX_CIV_PLAYERS() )
  199.         for i in range( gc.getMAX_CIV_PLAYERS() ):
  200.             if ( gc.getPlayer(i).isAlive() ):
  201.                 popup.addPullDownString( gc.getPlayer(i).getName(), i )
  202.         popup.setBodyString( "Modify Player %s:" %( CvUtil.getIcon('gold'),) )
  203.         popup.createPythonEditBox( "0", "Integer value (positive or negative)" )
  204.         
  205.         # Loop through Era Infos and add names
  206.         for i in range(gc.getNumEraInfos()):
  207.             popup.addButton(gc.getEraInfo(i).getDescription())
  208.         
  209.         popup.launch(true, PopupStates.POPUPSTATE_IMMEDIATE)
  210.     
  211.     def applyTechCheat( self, argsList ):
  212.         'Apply Tech Cheat'
  213.         popupReturn = argsList
  214.         playerID = popupReturn.getSelectedPullDownValue( 0 )
  215.         bAll = 0
  216.         if playerID == gc.getMAX_CIV_PLAYERS():
  217.             bAll = 1
  218.             player = PyGame().getCivPlayerList()
  219.         else:
  220.             player = PyPlayer( playerID )
  221.         era = popupReturn.getButtonClicked()
  222.         
  223.         try:
  224.             goldChange = int( popupReturn.getEditBoxString( 0 ) )
  225.         except:
  226.             return 0
  227.             
  228.         if goldChange:
  229.             if not bAll:
  230.                 player.changeGold(goldChange)
  231.             else:
  232.                 for i in range(len(player)):
  233.                     player[i].changeGold(goldChange)
  234.  
  235.         for tech in PyGame().getEraTechList(era):
  236.             id = tech.getID()
  237.             if not bAll:
  238.                 player.setHasTech( id )
  239.             else:
  240.                 for j in range(len(player)):
  241.                     player[j].setHasTech( id )
  242.     def RotateUnit(self, Direction, px, py ):
  243.         if ( px != -1 and py != -1 ):
  244.             unit = CyMap().plot(px, py).getUnit(0)
  245.             if ( not unit.isNone() ):
  246.                 unitEntity = CyUnitEntity(unit)
  247.                 dir = unitEntity.GetUnitFacingDirection( )
  248.                 dir += Direction * 0.05;
  249.                 unitEntity.SetUnitFacingDirection( dir )
  250.  
  251.     def resetUnitMovement( self ):
  252.         global g_bDebugMode
  253.         if g_bDebugMode == 0:
  254.             return            
  255.         for i in range(gc.getMAX_PLAYERS()):
  256.             (unit, iter) = gc.getPlayer(i).firstUnit(false)
  257.             while (unit):
  258.                 unit.setMoves(0)
  259.                 (unit, iter) = gc.getPlayer(i).nextUnit(iter, false)
  260.  
  261.     def allUnits( self ):
  262.         self.putOneOfEveryUnit();
  263.         
  264.     def allBonuses( self ):
  265.         iNBonuses = gc.getNumBonusInfos()
  266.         map = CyMap()
  267.         if ( iNBonuses < map.getGridWidth() * map.getGridHeight() ):
  268.             for x in range(map.getGridWidth()):
  269.                 for y in range((iNBonuses/map.getGridWidth())+1):
  270.                     map.plot(x,y).setBonusType( (x + y * map.getGridWidth())%iNBonuses );
  271.         
  272.     def allImprovements( self ):
  273.         iNImprovements = gc.getNumImprovementInfos()
  274.         map = CyMap()
  275.         if ( iNImprovements < map.getGridWidth() * map.getGridHeight() ):
  276.             for x in range(map.getGridWidth()):
  277.                 for y in range((iNImprovements/map.getGridWidth())+1):
  278.                     map.plot(x,y).setImprovementType( (x + y * map.getGridWidth())%iNImprovements );
  279.         
  280.     def putOneOfEveryUnit( self ):
  281.         pass
  282.         iNUnits = gc.getNumUnitInfos()
  283.         map = CyMap()
  284.         player = gc.getPlayer(0)
  285.         if ( iNUnits < map.getGridWidth() * map.getGridHeight() ):
  286.             for x in range(map.getGridWidth()):
  287.                 for y in range((iNUnits/map.getGridWidth())+1):
  288.                     player.initUnit( (x + y * map.getGridWidth())%iNUnits, x, y, UnitAITypes.NO_UNITAI )
  289.  
  290. g_CvDebugTools = CvDebugTools()
  291. #class DemoScripts:
  292. #    def __init__(self):
  293. #        self.iCurrentStep = 0
  294. #    
  295. #    def demoStart(self):
  296. #        CyGame().toggleDebugMode()
  297. #        #preLoadUnits()
  298. #        #preloadImprovements()
  299. #        #preloadBonuses()
  300. #        
  301. #        CyInterfaceScreen("MainInterface", CvScreenEnums.MAIN_INTERFACE).updateMinimapSection(True)
  302. #        CyInterface().setShowInterface(InterfaceVisibility.INTERFACE_HIDE_ALL)
  303. #        Globe.zoom(1.0)
  304. #        Globe.rotate(True, -180)
  305. #        toggleScreenCleanup()
  306. #        
  307. #    def incrementStep(self):
  308. #        self.iCurrentStep += 1
  309. #        
  310. #        if self.iCurrentStep == 1:
  311. #            self.demoStart()
  312. #        elif self.iCurrentStep == 2:
  313. #            Globe.stopGlobe()
  314. #            CyEngine().JustLookAt(gc.getPlayer(0).getCity(1).plot().getPoint())
  315. #            CyEngine().SetCameraMovementSpeed(int(CameraMovementSpeeds.CAMERAMOVEMENTSPEED_SLOW))
  316. #        elif self.iCurrentStep == 3:
  317. #            Globe.zoom(0.01, 0.05, 0.0)
  318. #            Globe.rotate()
  319. #        
  320. #g_DemoScript = DemoScripts()
  321. #