home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / pyUnit / CvUnitController.py
Encoding:
Python Source  |  2005-11-09  |  1.8 KB  |  47 lines
  1. from CvPythonExtensions import *
  2. import random
  3.  
  4. MoveFormation_BoxRanges = [ [-0.8, -0.8, 0.8, 0.8],    # Two
  5.                             [-0.8, -0.8, -0.1, 0.8 ,  0.1, -0.8, 0.8, 0.8],    # Three
  6.                             [-0.8, -0.8, -0.1, -0.1,  0.1, -0.8, 0.8, -0.1,  -0.5, 0.1, 0.5, 0.8],    # Four
  7.                             [-0.8, -0.8, -0.2, -0.2,  -0.8, 0.2, -0.2, 0.8,  0.2, -0.8, 0.8, -0.2,   0.2, 0.2, 0.8, 0.8], # Five
  8.                             [-0.8, -0.8, -0.2, -0.2,  -0.8, 0.2, -0.2, 0.8,  0.2, -0.8, 0.8, -0.2,   0.2, 0.2, 0.8, 0.8, -0.3, -0.3, 0.3, 0.3]]
  9. random.seed( 37 )   # should this be configurable?
  10.  
  11. class CvUnitController:
  12.  
  13.     def __init__(self):
  14.         pass                        # One
  15.  
  16.                         
  17.     def ProcessEvent ( self, pCyUnitEntity, iEvent):
  18.         """Controls the logic to react to the event."""
  19.  
  20.         if iEvent == EntityEventTypes.ENTEVENT_MOVE:
  21.  
  22.             iNumAlive = pCyUnitEntity.GetUnitsCurrentlyAlive()
  23.             if iNumAlive > 5:   # We can't handle more than 5 sub-units...
  24.                 return 0;
  25.             iCurrentUnit = 0
  26.             for i in range(0, pCyUnitEntity.GetSubEntityCount()):
  27.  
  28.                 # 
  29.                 pCyUnitSubEntity = pCyUnitEntity.GetSubEntity(i);
  30.                 if pCyUnitSubEntity.isDead():
  31.                     continue
  32.                 x = random.uniform( MoveFormation_BoxRange[iNumAlive][iCurrentUnit * 4 + 0], MoveFormation_BoxRange[iNumAlive][iCurrentUnit * 4 + 2] )
  33.                 y = random.uniform( MoveFormation_BoxRange[iNumAlive][iCurrentUnit * 4 + 1], MoveFormation_BoxRange[iNumAlive][iCurrentUnit * 4 + 3] )
  34.                 kTarget = pCyUnitEntity.TranslateFormation( x, y )
  35.                 pCyUnitSubEntity.SetPositionTarget( kTarget.x, kTarget.y, 0.0 )
  36.                 pCyUnitSubEntity.SetAnimationPath( AnimationPathTypes.ANIMATIONPATH_RUN )
  37.                 pCyUnitSubEntity.SetStoppingCriteria( BehaviourStoppingCriterion.STOPCRIT_DESTINATION )
  38.                 pCyUnitSubEntity.SetFacingTargetTowardsPositionTarget()
  39.                 pCyUnitSubEntity.PushBehaviourState( )
  40.                 iCurrentUnit += 1;
  41.  
  42.             return 1;
  43.  
  44.         else:
  45.             return 0;
  46.  
  47.