class AttractScreenBase(GameScreen):
'''
This is the base class for all attract screens.
All attract screens must have the credit flashing at the bottom, so the easiest way
is to have all attract screens sublcass this class, where the common elements live.
'''
_creditAnim = None
def __init__(self, screen_manager, name):
'''
Constructor
'''
super(AttractScreenBase, self).__init__(screen_manager, name)
self._creditText=OnscreenText("FREE PLAY PRESS START",
1,
font=base.fontLoader.load('digital.ttf'),
fg=(1,1,1,1),
pos=(0,-.95),
align=TextNode.ACenter,
scale=.1,
mayChange=True,
parent=self.node2d)
self._creditAnim = None
def startAnimations(self):
self._creditAnim = base.taskMgr.doMethodLater(0.8, self.toggleCreditLine, 'credit_flash')
def stopAnimations(self):
if self._creditAnim == None: return
base.taskMgr.remove(self._creditAnim)
self._creditAnim = None
def toggleCreditLine(self, task):
if self._creditText.isHidden():
self._creditText.show()
else:
self._creditText.hide()
return task.again
def show(self):
self.startAnimations()
super(AttractScreenBase, self).show()
def hide(self):
self.stopAnimations()
super(AttractScreenBase, self).hide()
class PartyCogActivityGui(DirectObject):
notify = directNotify.newCategory('PartyCogActivityGui')
def __init__(self):
DirectObject.__init__(self)
self._piePowerMeter = None
self._victoryBalanceBar = None
self._scoreLabel = None
self._cogTracker = None
self._piePowerTitle = None
self._victoryBalanceTitle = None
self._scoreTitle = None
self._spamWarning = None
self._spamWarningIvalName = 'PartyCogActivityGui-SpamWarning'
def load(self):
self._initPiePowerMeter()
self._initScore()
self._initCogTracker()
self._initSpamWarning()
self._initControlGui()
self._initVictoryBalanceBar()
def unload(self):
if self._cogTracker is not None:
self._cogTracker.destory()
self._cogTracker = None
if self._piePowerMeter is not None:
self._piePowerMeter.destroy()
self._piePowerMeter = None
if self._piePowerTitle is not None:
self._piePowerTitle.destroy()
self._piePowerTitle = None
if self._scoreLabel is not None:
self._scoreLabel.destroy()
self._scoreLabel = None
if self._scoreTitle is not None:
self._scoreTitle.destroy()
self._scoreTitle = None
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.destroy()
self._spamWarning = None
if hasattr(self, '_attackKeys'):
self._attackKeys.detachNode()
del self._attackKeys
if hasattr(self, '_moveKeys'):
self._moveKeys.detachNode()
del self._moveKeys
if self._victoryBalanceBar:
self._victoryBalanceBar.detachNode()
self._victoryBalanceBar = None
if self._victoryBalanceBarOrange:
self._victoryBalanceBarOrange.detachNode()
self._victoryBalanceBarOrange = None
if self._victoryBalanceBarPie:
self._victoryBalanceBarPie.detachNode()
self._victoryBalanceBarPie = None
if self._victoryBalanceBarArrow:
self._victoryBalanceBarArrow.detachNode()
self._victoryBalanceBarArrow = None
def _initVictoryBalanceBar(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
victoryBalanceBar = loader.loadModel('phase_13/models/parties/tt_m_gui_pty_pieToss_balanceBar')
self._victoryBalanceBar = victoryBalanceBar.find('**/*tt_t_gui_pty_pieToss_balanceBarBG')
self._victoryBalanceBar.reparentTo(aspect2d)
self._victoryBalanceBar.setBin('fixed', 0)
self._victoryBalanceBar.setPos(PartyGlobals.CogActivityVictoryBarPos)
self._victoryBalanceBar.setScale(1)
self._victoryBalanceBarOrange = victoryBalanceBar.find('**/*tt_t_gui_pty_pieToss_balanceBarOrange')
self._victoryBalanceBarOrange.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarOrange.setBin('fixed', 1)
self._victoryBalanceBarOrange.setPos(PartyGlobals.CogActivityVictoryBarOrangePos)
self._victoryBalanceBarOrange.setScale(PartyGlobals.CogActivityBarStartScale, 1.0, 1.0)
self._victoryBalanceBarPie = victoryBalanceBar.find('**/*tt_t_gui_pty_pieToss_balanceBarPie')
self._victoryBalanceBarPie.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarPie.setBin('fixed', 2)
self._victoryBalanceBarPie.setX(PartyGlobals.CogActivityVictoryBarPiePos[0])
self._victoryBalanceBarPie.setY(PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(PartyGlobals.CogActivityVictoryBarPiePos[2])
self._victoryBalanceBarPie.setScale(PartyGlobals.CogActivityBarPieScale)
self._victoryBalanceBarArrow = victoryBalanceBar.find('**/*tt_t_gui_pty_pieToss_balanceArrow')
self._victoryBalanceBarArrow.reparentTo(self._victoryBalanceBarPie)
self._victoryBalanceBarArrow.setBin('fixed', 2)
self._victoryBalanceBarArrow.setPos(PartyGlobals.CogActivityVictoryBarArrow)
self._victoryBalanceBarArrow.setScale(1 / PartyGlobals.CogActivityBarPieScale)
def _initControlGui(self):
self._attackIvalName = 'PartyCogActivityGui-attackKeys'
self._moveIvalName = 'PartyCogActivityGui-moveKeys'
pieTossControls = loader.loadModel('phase_13/models/parties/tt_m_gui_pty_pieToss_controls')
self._attackKeys = pieTossControls.find('**/*control*')
self._moveKeys = pieTossControls.find('**/*arrow*')
self._moveKeys.reparentTo(aspect2d)
self._moveKeys.setPos(1.0, 0.0, -0.435)
self._moveKeys.setScale(0.15)
self._attackKeys.reparentTo(aspect2d)
self._attackKeys.setPos(0.85, 0.0, -0.45)
self._attackKeys.setScale(0.15)
self._moveKeys.hide()
self._attackKeys.hide()
def _initPiePowerMeter(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
self._piePowerMeter = DirectWaitBar(frameSize=(-h,
h,
-w,
w), relief=DGG.GROOVE, frameColor=(0.9, 0.9, 0.9, 1.0), borderWidth=(0.01, 0.01), barColor=PartyGlobals.CogActivityColors[0], pos=PartyGlobals.CogActivityPowerMeterPos, hpr=(0.0, 0.0, -90.0))
self._piePowerMeter.setBin('fixed', 0)
self._piePowerTitle = OnscreenText(text=TTLocalizer.PartyCogGuiPowerLabel, pos=PartyGlobals.CogActivityPowerMeterTextPos, scale=0.05, fg=(1.0, 1.0, 1.0, 1.0), align=TextNode.ACenter)
self._piePowerTitle.setBin('fixed', 0)
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def _initScore(self):
self._scoreLabel = OnscreenText(text='0', pos=PartyGlobals.CogActivityScorePos, scale=PartyGlobals.TugOfWarTextWordScale, fg=(1.0, 1.0, 0.0, 1.0), align=TextNode.ARight, font=ToontownGlobals.getSignFont(), mayChange=True)
self._scoreTitle = OnscreenText(text=TTLocalizer.PartyCogGuiScoreLabel, pos=PartyGlobals.CogActivityScoreTitle, scale=0.05, fg=(1.0, 1.0, 1.0, 1.0), align=TextNode.ARight)
self._scoreLabel.hide()
self._scoreTitle.hide()
def _initCogTracker(self):
self._cogTracker = PartyCogTrackerGui()
def _initSpamWarning(self):
self._spamWarning = OnscreenText(text=TTLocalizer.PartyCogGuiSpamWarning, scale=0.15, fg=(1.0, 1.0, 0, 1.0), shadow=(0, 0, 0, 0.62), mayChange=False, pos=(0, 0.33))
self._spamWarning.hide()
def showScore(self):
self._scoreLabel.show()
self._scoreTitle.show()
def hideScore(self):
self._scoreLabel.hide()
self._scoreTitle.hide()
def setScore(self, score = 0):
self._scoreLabel['text'] = str(score)
def resetPiePowerMeter(self):
self._piePowerMeter['value'] = 0
def showPiePowerMeter(self):
self._piePowerMeter.show()
self._piePowerTitle.show()
def hidePiePowerMeter(self):
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def updatePiePowerMeter(self, value):
self._piePowerMeter['value'] = value
def getPiePowerMeterValue(self):
return self._piePowerMeter['value']
def hideSpamWarning(self):
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.hide()
def showSpamWarning(self):
if self._spamWarning.isHidden():
self._spamWarning.show()
taskMgr.remove(self._spamWarningIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(self._spamWarning, ''), Wait(PartyGlobals.CogActivitySpamWarningShowTime), Func(self.hideSpamWarning), name=self._spamWarningIvalName, autoFinish=1).start()
def hide(self):
self.hidePiePowerMeter()
self.hideScore()
self.hideSpamWarning()
self.hideControls()
def disableToontownHUD(self):
base.localAvatar.hideName()
base.localAvatar.laffMeter.hide()
base.setCellsAvailable(base.bottomCells + [base.rightCells[1]], False)
def enableToontownHUD(self):
base.localAvatar.showName()
base.localAvatar.laffMeter.show()
base.setCellsAvailable(base.bottomCells + [base.rightCells[1]], True)
def setTeam(self, team):
self.team = team
if team == 0:
self._cogTracker.frame.setR(180)
self._piePowerMeter['barColor'] = PartyGlobals.CogActivityColors[team]
def startTrackingCogs(self, cogs):
self.cogs = cogs
taskMgr.add(self.trackCogs, 'trackCogs')
def trackCogs(self, task):
if self.cogs is None:
return
self._updateVictoryBar()
for i, cog in enumerate(self.cogs):
self._cogTracker.updateCog(i, cog, self.team)
return task.cont
def _updateVictoryBar(self):
if not (hasattr(self, '_victoryBalanceBar') and self._victoryBalanceBar):
return
netDistance = 0
for cog in self.cogs:
netDistance = netDistance + cog.targetDistance
teamDistance = netDistance / 6.0
self._victoryBalanceBarOrange.setScale(PartyGlobals.CogActivityBarStartScale + teamDistance * 10 * PartyGlobals.CogActivityBarUnitScale, 1.0, 1.0)
self._victoryBalanceBarPie.setX(PartyGlobals.CogActivityVictoryBarPiePos[0] + teamDistance * 10 * PartyGlobals.CogActivityBarPieUnitMove)
self._victoryBalanceBarPie.setY(PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(PartyGlobals.CogActivityVictoryBarPiePos[2])
if teamDistance > 0.0:
self._victoryBalanceBarArrow.setColor(PartyGlobals.CogActivityColors[1])
elif teamDistance < 0.0:
self._victoryBalanceBarArrow.setColor(PartyGlobals.CogActivityColors[0])
else:
self._victoryBalanceBarArrow.setColor(VBase4(1.0, 1.0, 1.0, 1.0))
def stopTrackingCogs(self):
taskMgr.remove('trackCogs')
def showAttackControls(self):
if self._attackKeys.isHidden():
self._attackKeys.show()
taskMgr.remove(self._attackIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(self._attackKeys, '', scale=0.15), Wait(PartyGlobals.CogActivityControlsShowTime), Func(self.hideAttackControls), name=self._attackIvalName, autoFinish=1).start()
def showMoveControls(self):
if self._moveKeys.isHidden() and not self._attackKeys.isHidden():
self._moveKeys.show()
taskMgr.remove(self._moveIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(self._moveKeys, '', scale=0.15), Wait(PartyGlobals.CogActivityControlsShowTime), Func(self.hideMoveControls), name=self._moveIvalName, autoFinish=1).start()
def hideAttackControls(self):
taskMgr.remove(self._attackIvalName)
if hasattr(self, '_attackKeys') and self._attackKeys:
self._attackKeys.hide()
def hideMoveControls(self):
taskMgr.remove(self._moveIvalName)
if hasattr(self, '_moveKeys') and self._moveKeys:
self._moveKeys.hide()
def hideControls(self):
self.hideMoveControls()
self.hideAttackControls()
class SimWorldBase(ShowBase):
def __init__(self, width, height, isFullscreen, title):
ShowBase.__init__(self)
#=======================================================================
# Set window properties.
#=======================================================================
win_props = WindowProperties(self.win.getProperties())
win_props.setTitle(title)
win_props.setFullscreen(isFullscreen)
win_props.setSize(width, height)
self.win.requestProperties(win_props)
# Apply the property changes.
self.graphicsEngine.openWindows()
#=======================================================================
# Scene related fields.
#=======================================================================
self.sceneNP = None
'''
Main scene node path. There can be only one scene at any given time.
'''
#=======================================================================
# Actor related fields.
#=======================================================================
self.actorNP = {}
'''
Dictionary of actor node paths keyed with the actor name.Internal use
only.
'''
self.actorOINP = None
'''
Actor of interest node path. There can be only one actor of interest
at any given time.
'''
self.actorOIName = None
'''
Actor of interest name. There can be only one actor of interest at any
given time.
'''
self.actorOIMoveDir = 0
'''
Actor of interest movement direction indicator. Forward is 1, reverse is
-1, and stop is 0.
'''
self.actorOITurnDir = 0
'''
Actor of interest turn direction indicator. Left is 1, right is -1, and
no turn is 0.
'''
self.actorOIMoveSpeed = 3
'''
Actor of interest translational speed. Distance units per second.
'''
self.actorOITurnSpeed = 30
'''
Actor of interest rotational speed. Degrees per second.
'''
self.actorOILocation = Vec3D.zero()
'''
Actor of interest's location coordinates after the last frame.
'''
self.actorOIVelocity = Vec3D.zero()
'''
Actor of interest's velocity vector after the last frame.
'''
#=======================================================================
# Camera related fields.
#=======================================================================
self.cameraNP = {}
'''
Dictionary of camera node paths keyed with the camera name.
'''
# Add the showbase camera as 'default'.
self.cameraNP['default'] = self.cam
#=======================================================================
# Illumination related fields
#=======================================================================
self.lightNP = {}
'''
Dictionary of light node paths keyed with the light name.
'''
self.goalNP = {}
'''
Dictionary of goal node paths keyed with the goal name.
'''
#=======================================================================
# Display region related fields.
#=======================================================================
self.displayRegion = {}
'''
Dictionary of display regions keyed with the display region name.
'''
self.displayRegionOI = None
'''
Display region of interest. Any action related to display regions
will be applied to the current display region of interest.
'''
self.displayRegionOIName = None
'''
Display region of interest name. Any action related to display
regions will be applied to the current display region of interest.
'''
# Add the display region associated with the default showbase camera.
self.displayRegion['default'] = self.camNode.getDisplayRegion(0)
self.displayRegion['default'].setSort(1)
self.accept('control-`', self.setDisplayRegionOfInterest, ['default'])
# Current display region of interest is the 'default' one.
self.displayRegionOI = self.displayRegion['default']
self.displayRegionOIName = 'default'
#=======================================================================
# AI related fields.
#=======================================================================
self.AIWorld = AIWorld(self.render)
self.AICharacter = {}
#=======================================================================
# Trace file related fields.
#=======================================================================
self.traceFile = None
'''Handle to the trace file.'''
self.traceMessage = ''
'''Message to be written to the trace file.'''
self.onScreenHelpNP = None
'''Text node path holding the on screen help information.'''
#=======================================================================
# Other fields.
#=======================================================================
self.clock = globalClock
'''Exposes Panda3D's GlobalClock.'''
self.userDialog = OnscreenText(text='', bg=(0,0,0,.6), fg=(.8,0,0,1),
mayChange=True)
self.userDialogTask = None
self.isNotifyUser = True
# We do not need the default mouse controls.
self.disableMouse()
# We do not need the default camera.
self.cam.node().setActive(False)
# Make sure a clean exit from the application when requested.
atexit.register(self.exitfunc)
# Default collision traverser.
self.cTrav = CollisionTraverser()
#=======================================================================
# On screen help setup.
#=======================================================================
OSHelp_str = '\nHELP\n'
text_font = self.loader.loadFont('cmtt12', spaceAdvance=.52)
self.onScreenHelpNP = OnscreenText(text=OSHelp_str,
font=text_font,
style=1,
fg=(1,1,1,1),
bg=(0,0,0,.6),
align=TextNode.ALeft,
scale=.05,
mayChange=True)
self.onScreenHelpNP.reparentTo(self.a2dTopLeft)
self.onScreenHelpNP.hide()
#=======================================================================
# Keyboard control related fields and default settings.
#=======================================================================
self.keyMap = {}
'''Dictionary of keyboard controls keyed with the action name.'''
self.setActorOIActionKey('forward', 'arrow_up')
self.setActorOIActionKey('reverse', 'arrow_down')
self.setActorOIActionKey('turn left', 'arrow_left')
self.setActorOIActionKey('turn right', 'arrow_right')
self.addHotKey('t', 'toggle pip', self.toggleDisplayRegion, [])
self.addHotKey('f1', 'help', self.toggleOnScreenHelp, [])
self.addHotKey('escape', 'quit', self.shutDown, [])
def setTracing(self, fileName, flag=True):
if flag and self.traceFile is None:
self.traceFile = open(fileName, 'w')
self.taskMgr.add(self.traceUpdateTask, 'traceUpdate', sort=100)
else:
self.taskMgr.remove('traceUpdate')
self.traceFile.close()
self.traceFile = None
def getTracing(self):
return self.taskMgr.hasTaskNamed('traceUpdate')
def setupScene(self, modelName, position=(0,0,0), orientation=(0,0,0),
scale=(1,1,1)):
base_path = os.path.dirname(__file__)
model_path = os.path.join(base_path, 'models', modelName)
model_path = Filename.fromOsSpecific(model_path)
self.sceneNP = self.loader.loadModel(model_path)
self.sceneNP.setPosHprScale(position, orientation, scale)
self.sceneNP.reparentTo(self.render)
base.setBackgroundColor((.1, .1, .1, 0))
def getObject(self, name):
return self.render.findAllMatches('**/'+name)
def setObjectSolid(self, name, flag):
if flag:
self.getObject(name).setCollideMask(BitMask32(0x001),
BitMask32(0x001))
else:
self.getObject(name).setCollideMask(BitMask32.allOff(),
BitMask32(0x001))
def setObjectGoal(self, name, flag):
if flag:
self.getObject(name).setCollideMask(BitMask32(0x002),
BitMask32(0x002))
else:
self.getObject(name).setCollideMask(BitMask32.allOff(),
BitMask32(0x002))
def addHotKey(self, hotkey, action, callback, args):
self.keyMap[action] = hotkey
self.accept(hotkey, callback, args)
def addActor(self, name, position=(0,0,0), orientation=(0,0,0),
scale=(1,1,1), collisionSphere=None):
# Create the scene node path.
actor_np = NodePath(ActorNode(name))
actor_np.setPosHpr(position, orientation)
actor_np.reparentTo(self.render)
# Attach model to the node path.
base_path = os.path.dirname(__file__)
model_path = os.path.join(base_path, 'models', name)
model_path = Filename.fromOsSpecific(model_path)
model_np = self.loader.loadModel(model_path)
model_np.setScale(scale)
model_np.reparentTo(actor_np)
#=======================================================================
# Make actor physics collidible.
#=======================================================================
# Attach collision solid to the node path.
cn = CollisionNode(name+'PhysicsCollisionNode')
cn.setIntoCollideMask(BitMask32.allOff())
cn.setFromCollideMask(BitMask32.bit(0))
cs = CollisionSphere(collisionSphere[0:3], collisionSphere[3])
cn.addSolid(cs)
actor_cnp = actor_np.attachNewNode(cn)
# Add actor to the physics collision detection system.
handler = PhysicsCollisionHandler()
handler.addCollider(actor_cnp, actor_np)
# self.cTrav.addCollider(actor_cnp, handler)
#=======================================================================
# Make actor AI compatible.
#=======================================================================
ai_character = AICharacter(name, actor_np, mass=50, movt_force=10,
max_force=30)
self.AICharacter[name] = ai_character
self.AIWorld.addAiChar(ai_character)
#=======================================================================
# Make actor events collidible.
#=======================================================================
# Attach collision solid to the node path.
cn = CollisionNode(name+'EventCollisionNode')
cn.setIntoCollideMask(BitMask32.allOff())
cn.setFromCollideMask(BitMask32.bit(1))
cs = CollisionSphere(collisionSphere[0:3], collisionSphere[3])
cn.addSolid(cs)
actor_cnp = actor_np.attachNewNode(cn)
actor_cnp.show()
# Add actor to the collision event detection system.
handler = CollisionHandlerEvent()
handler.addInPattern('%fn-enters')
handler.addOutPattern('%fn-exits')
self.cTrav.addCollider(actor_cnp, handler)
self.accept(name+'EventCollisionNode'+'-enters', self.onActorEnterEventFunc)
self.accept(name+'EventCollisionNode'+'-exits', self.onActorExitEventFunc)
self.actorNP[name] = actor_np
# The most recently added actor becomes actor of interest.
self.setActorOfInterest(name)
self.taskMgr.add(self.actorOIStateUpdateTask,'actorStateUpdate',sort=50)
def activateActorOIControl(self, flag):
if flag:
self.taskMgr.add(self.actorOIControlTask,'actorControl',sort=10)
else:
self.taskMgr.remove('actorControl')
def setActorOfInterest(self, name):
self.actorOINP = self.actorNP[name]
self.actorOIName = name
self.actorOILocation = self.actorOINP.getPos()
def setActorOIActionKey(self, action, key):
self.keyMap[action] = key
self.accept(key, self.setActorOIAction, [action])
self.accept(key+'-up', self.setActorOIAction, [action+' stop'])
self.keyMap[action] = key
def setActorOIAction(self, action):
if action == 'forward':
self.actorOIMoveDir = 1;
elif action == 'reverse':
self.actorOIMoveDir = -1;
elif action == 'forward stop' or action == 'reverse stop':
self.actorOIMoveDir = 0;
elif action == 'turn right':
self.actorOITurnDir = -1;
elif action == 'turn left':
self.actorOITurnDir = 1;
elif action == 'turn right stop' or action == 'turn left stop':
self.actorOITurnDir = 0;
def addLight(self, name, hotkey, type, color=(1,1,1,1), position=(0,0,0),
orientation=(0,0,0)):
if type == 'ambient':
light_np = AmbientLight(name)
light_np.setColor(color)
elif type == 'point':
light_np = PointLight(name)
light_np.setColor(color)
light_np.setPoint(position)
elif type == 'directional':
light_np = DirectionalLight(name)
light_np.setColor(color)
light_np.setPoint(position)
light_np.setDirection(orientation)
else:
return
self.keyMap[name] = hotkey
self.accept(hotkey, self.toggleLight, [name])
self.lightNP[name] = self.render.attachNewNode(light_np)
def toggleLight(self, name):
light_np = self.lightNP[name]
if self.render.hasLight(light_np):
self.render.setLightOff(light_np)
msg = '''Light [{0}] turned off.'''.format(name)
else:
self.render.setLight(light_np)
msg = '''Light [{0}] turned on.'''.format(name)
self.notifyUser(msg, 2)
def activateLight(self, name, flag):
if flag:
self.render.setLight(self.lightNP[name])
else:
self.render.setLightOff(self.lightNP[name])
def addCamera(self, name, hotkey, parent, lens, position=(0,0,0),
orientation=(0,0,0)):
camera_n = Camera(name, lens)
camera_np = NodePath(camera_n)
camera_np.setPosHpr(position, orientation)
camera_np.reparentTo(parent)
self.keyMap[name] = hotkey
self.accept(hotkey, self.activateCamera, [name])
self.cameraNP[name] = camera_np
def activateCamera(self, name):
dr = self.displayRegionOI
ar = float(dr.getPixelWidth()) / dr.getPixelHeight()
self.cameraNP[name].node().getLens().setAspectRatio(ar)
self.displayRegionOI.setCamera(self.cameraNP[name])
self.cameraNP[name].node().setActive(True)
msg = '''Camera [{0}] active on display region [{1}].'''
self.notifyUser(msg.format(name, self.displayRegionOIName), 2)
def addDisplayRegion(self, name, hotkey, lrbt):
dr = self.win.makeDisplayRegion(lrbt[0],lrbt[1],lrbt[2],lrbt[3])
dr.setSort(99)
dr.setActive(False)
self.keyMap[name] = hotkey
self.accept(hotkey, self.setDisplayRegionOfInterest, [name])
self.displayRegion[name] = dr
def setDisplayRegionOfInterest(self, name):
self.displayRegionOI = self.displayRegion[name]
self.displayRegionOIName = name
self.displayRegionOI.setActive(True)
msg = '''Display region [{0}] selected.'''
self.notifyUser(msg.format(self.displayRegionOIName), 2)
def activateDisplayRegion(self, name, flag):
self.displayRegion[name].setActive(flag)
def toggleDisplayRegion(self, name=None):
if name is not None:
dr = self.displayRegion[name]
else:
dr = self.displayRegionOI
dr.setActive(not dr.isActive())
def startAITask(self):
taskMgr.add(self.AIUpdateTask, "AIUpdate")
def stopAITask(self):
taskMgr.remove("AIUpdate")
def shutDown(self):
if self.traceFile: self.traceFile.close()
sys.exit()
def toggleOnScreenHelp(self):
str = '\n'.join([n.title().ljust(10)+' : '+k.title()
for n,k in self.keyMap.iteritems()])
self.onScreenHelpNP.setText('\n'+str)
if self.onScreenHelpNP.isHidden():
self.onScreenHelpNP.show()
else:
self.onScreenHelpNP.hide()
def notifyUser(self, msg, delay):
if self.isNotifyUser:
self.userDialog.setText(msg)
self.userDialog.show()
if self.userDialogTask is not None: self.userDialogTask.remove()
self.userDialogTask = self.taskMgr.doMethodLater(delay,
self.userDialog.hide,
'destroyMessage',
extraArgs=[])
def AIUpdateTask(self, task):
self.AIWorld.update()
return task.cont
def traceUpdateFunc(self):
pass
def traceUpdateTask(self, task):
self.traceUpdateFunc()
message = 'timestamp:{0:.4f},{1}\n'.format(self.clock.getFrameTime(),
self.traceMessage)
self.traceFile.write(message)
return task.cont
def onActorEnterEventFunc(self, collisionEntry):
pass
def onActorExitEventFunc(self, collisionEntry):
pass
def actorOIControlTask(self, task):
dt = self.clock.getDt()
rotation = self.actorOITurnSpeed * self.actorOITurnDir * dt
self.actorOINP.setH(self.actorOINP, rotation)
translation = self.actorOIMoveSpeed * self.actorOIMoveDir * dt
self.actorOINP.setY(self.actorOINP, translation)
return task.cont
def actorOIStateUpdateFunc(self):
pass
def actorOIStateUpdateTask(self, task):
self.actorOIStateUpdateFunc()
return task.cont
class PartyCogActivityGui(DirectObject):
notify = directNotify.newCategory('PartyCogActivityGui')
def __init__(self):
DirectObject.__init__(self)
self._piePowerMeter = None
self._victoryBalanceBar = None
self._scoreLabel = None
self._cogTracker = None
self._piePowerTitle = None
self._victoryBalanceTitle = None
self._scoreTitle = None
self._spamWarning = None
self._spamWarningIvalName = 'PartyCogActivityGui-SpamWarning'
return
def load(self):
self._initPiePowerMeter()
self._initScore()
self._initCogTracker()
self._initSpamWarning()
self._initControlGui()
self._initVictoryBalanceBar()
def unload(self):
if self._cogTracker is not None:
self._cogTracker.destory()
self._cogTracker = None
if self._piePowerMeter is not None:
self._piePowerMeter.destroy()
self._piePowerMeter = None
if self._piePowerTitle is not None:
self._piePowerTitle.destroy()
self._piePowerTitle = None
if self._scoreLabel is not None:
self._scoreLabel.destroy()
self._scoreLabel = None
if self._scoreTitle is not None:
self._scoreTitle.destroy()
self._scoreTitle = None
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.destroy()
self._spamWarning = None
if hasattr(self, '_attackKeys'):
self._attackKeys.detachNode()
del self._attackKeys
if hasattr(self, '_moveKeys'):
self._moveKeys.detachNode()
del self._moveKeys
if self._victoryBalanceBar:
self._victoryBalanceBar.detachNode()
self._victoryBalanceBar = None
if self._victoryBalanceBarOrange:
self._victoryBalanceBarOrange.detachNode()
self._victoryBalanceBarOrange = None
if self._victoryBalanceBarPie:
self._victoryBalanceBarPie.detachNode()
self._victoryBalanceBarPie = None
if self._victoryBalanceBarArrow:
self._victoryBalanceBarArrow.detachNode()
self._victoryBalanceBarArrow = None
return
def _initVictoryBalanceBar(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
victoryBalanceBar = loader.loadModel(
'phase_13/models/parties/tt_m_gui_pty_pieToss_balanceBar')
self._victoryBalanceBar = victoryBalanceBar.find(
'**/*tt_t_gui_pty_pieToss_balanceBarBG')
self._victoryBalanceBar.reparentTo(aspect2d)
self._victoryBalanceBar.setBin('fixed', 0)
self._victoryBalanceBar.setPos(PartyGlobals.CogActivityVictoryBarPos)
self._victoryBalanceBar.setScale(1)
self._victoryBalanceBarOrange = victoryBalanceBar.find(
'**/*tt_t_gui_pty_pieToss_balanceBarOrange')
self._victoryBalanceBarOrange.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarOrange.setBin('fixed', 1)
self._victoryBalanceBarOrange.setPos(
PartyGlobals.CogActivityVictoryBarOrangePos)
self._victoryBalanceBarOrange.setScale(
PartyGlobals.CogActivityBarStartScale, 1.0, 1.0)
self._victoryBalanceBarPie = victoryBalanceBar.find(
'**/*tt_t_gui_pty_pieToss_balanceBarPie')
self._victoryBalanceBarPie.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarPie.setBin('fixed', 2)
self._victoryBalanceBarPie.setX(
PartyGlobals.CogActivityVictoryBarPiePos[0])
self._victoryBalanceBarPie.setY(
PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(
PartyGlobals.CogActivityVictoryBarPiePos[2])
self._victoryBalanceBarPie.setScale(
PartyGlobals.CogActivityBarPieScale)
self._victoryBalanceBarArrow = victoryBalanceBar.find(
'**/*tt_t_gui_pty_pieToss_balanceArrow')
self._victoryBalanceBarArrow.reparentTo(self._victoryBalanceBarPie)
self._victoryBalanceBarArrow.setBin('fixed', 2)
self._victoryBalanceBarArrow.setPos(
PartyGlobals.CogActivityVictoryBarArrow)
self._victoryBalanceBarArrow.setScale(
1 / PartyGlobals.CogActivityBarPieScale)
def _initControlGui(self):
self._attackIvalName = 'PartyCogActivityGui-attackKeys'
self._moveIvalName = 'PartyCogActivityGui-moveKeys'
pieTossControls = loader.loadModel(
'phase_13/models/parties/tt_m_gui_pty_pieToss_controls')
self._attackKeys = pieTossControls.find('**/*control*')
self._moveKeys = pieTossControls.find('**/*arrow*')
self._moveKeys.reparentTo(aspect2d)
self._moveKeys.setPos(1.0, 0.0, -0.435)
self._moveKeys.setScale(0.15)
self._attackKeys.reparentTo(aspect2d)
self._attackKeys.setPos(0.85, 0.0, -0.45)
self._attackKeys.setScale(0.15)
self._moveKeys.hide()
self._attackKeys.hide()
def _initPiePowerMeter(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
self._piePowerMeter = DirectWaitBar(
frameSize=(-h, h, -w, w),
relief=DGG.GROOVE,
frameColor=(0.9, 0.9, 0.9, 1.0),
borderWidth=(0.01, 0.01),
barColor=PartyGlobals.CogActivityColors[0],
pos=PartyGlobals.CogActivityPowerMeterPos,
hpr=(0.0, 0.0, -90.0))
self._piePowerMeter.setBin('fixed', 0)
self._piePowerTitle = OnscreenText(
text=TTLocalizer.PartyCogGuiPowerLabel,
pos=PartyGlobals.CogActivityPowerMeterTextPos,
scale=0.05,
fg=(1.0, 1.0, 1.0, 1.0),
align=TextNode.ACenter)
self._piePowerTitle.setBin('fixed', 0)
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def _initScore(self):
self._scoreLabel = OnscreenText(
text='0',
pos=PartyGlobals.CogActivityScorePos,
scale=PartyGlobals.TugOfWarTextWordScale,
fg=(1.0, 1.0, 0.0, 1.0),
align=TextNode.ARight,
font=ToontownGlobals.getSignFont(),
mayChange=True)
self._scoreTitle = OnscreenText(text=TTLocalizer.PartyCogGuiScoreLabel,
pos=PartyGlobals.CogActivityScoreTitle,
scale=0.05,
fg=(1.0, 1.0, 1.0, 1.0),
align=TextNode.ARight)
self._scoreLabel.hide()
self._scoreTitle.hide()
def _initCogTracker(self):
self._cogTracker = PartyCogTrackerGui()
def _initSpamWarning(self):
self._spamWarning = OnscreenText(
text=TTLocalizer.PartyCogGuiSpamWarning,
scale=0.15,
fg=(1.0, 1.0, 0, 1.0),
shadow=(0, 0, 0, 0.62),
mayChange=False,
pos=(0, 0.33))
self._spamWarning.hide()
def showScore(self):
self._scoreLabel.show()
self._scoreTitle.show()
def hideScore(self):
self._scoreLabel.hide()
self._scoreTitle.hide()
def setScore(self, score=0):
self._scoreLabel['text'] = str(score)
def resetPiePowerMeter(self):
self._piePowerMeter['value'] = 0
def showPiePowerMeter(self):
self._piePowerMeter.show()
self._piePowerTitle.show()
def hidePiePowerMeter(self):
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def updatePiePowerMeter(self, value):
self._piePowerMeter['value'] = value
def getPiePowerMeterValue(self):
return self._piePowerMeter['value']
def hideSpamWarning(self):
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.hide()
def showSpamWarning(self):
if self._spamWarning.isHidden():
self._spamWarning.show()
taskMgr.remove(self._spamWarningIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(
self._spamWarning, ''),
Wait(PartyGlobals.CogActivitySpamWarningShowTime),
Func(self.hideSpamWarning),
name=self._spamWarningIvalName,
autoFinish=1).start()
def hide(self):
self.hidePiePowerMeter()
self.hideScore()
self.hideSpamWarning()
self.hideControls()
def disableToontownHUD(self):
base.localAvatar.hideName()
base.localAvatar.laffMeter.hide()
base.setCellsActive(base.bottomCells + [base.rightCells[1]], False)
def enableToontownHUD(self):
base.localAvatar.showName()
base.localAvatar.laffMeter.show()
base.setCellsActive(base.bottomCells + [base.rightCells[1]], True)
def setTeam(self, team):
self.team = team
if team == 0:
self._cogTracker.frame.setR(180)
self._piePowerMeter['barColor'] = PartyGlobals.CogActivityColors[team]
def startTrackingCogs(self, cogs):
self.cogs = cogs
taskMgr.add(self.trackCogs, 'trackCogs')
def trackCogs(self, task):
if self.cogs is None:
return
self._updateVictoryBar()
for i, cog in enumerate(self.cogs):
self._cogTracker.updateCog(i, cog, self.team)
return task.cont
def _updateVictoryBar(self):
if not (hasattr(self, '_victoryBalanceBar')
and self._victoryBalanceBar):
return
netDistance = 0
for cog in self.cogs:
netDistance = netDistance + cog.targetDistance
teamDistance = netDistance / 6.0
self._victoryBalanceBarOrange.setScale(
PartyGlobals.CogActivityBarStartScale +
teamDistance * 10 * PartyGlobals.CogActivityBarUnitScale, 1.0, 1.0)
self._victoryBalanceBarPie.setX(
PartyGlobals.CogActivityVictoryBarPiePos[0] +
teamDistance * 10 * PartyGlobals.CogActivityBarPieUnitMove)
self._victoryBalanceBarPie.setY(
PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(
PartyGlobals.CogActivityVictoryBarPiePos[2])
if teamDistance > 0.0:
self._victoryBalanceBarArrow.setColor(
PartyGlobals.CogActivityColors[1])
elif teamDistance < 0.0:
self._victoryBalanceBarArrow.setColor(
PartyGlobals.CogActivityColors[0])
else:
self._victoryBalanceBarArrow.setColor(VBase4(1.0, 1.0, 1.0, 1.0))
def stopTrackingCogs(self):
taskMgr.remove('trackCogs')
def showAttackControls(self):
if self._attackKeys.isHidden():
self._attackKeys.show()
taskMgr.remove(self._attackIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(self._attackKeys,
'',
scale=0.15),
Wait(PartyGlobals.CogActivityControlsShowTime),
Func(self.hideAttackControls),
name=self._attackIvalName,
autoFinish=1).start()
def showMoveControls(self):
if self._moveKeys.isHidden() and not self._attackKeys.isHidden():
self._moveKeys.show()
taskMgr.remove(self._moveIvalName)
Sequence(ToontownIntervals.getPulseLargerIval(self._moveKeys,
'',
scale=0.15),
Wait(PartyGlobals.CogActivityControlsShowTime),
Func(self.hideMoveControls),
name=self._moveIvalName,
autoFinish=1).start()
def hideAttackControls(self):
taskMgr.remove(self._attackIvalName)
if hasattr(self, '_attackKeys') and self._attackKeys:
self._attackKeys.hide()
def hideMoveControls(self):
taskMgr.remove(self._moveIvalName)
if hasattr(self, '_moveKeys') and self._moveKeys:
self._moveKeys.hide()
def hideControls(self):
self.hideMoveControls()
self.hideAttackControls()
class PartyCogActivityGui(DirectObject):
notify = directNotify.newCategory("PartyCogActivityGui")
def __init__(self):
DirectObject.__init__(self)
self._piePowerMeter = None
self._victoryBalanceBar = None
self._scoreLabel = None
self._cogTracker = None
self._piePowerTitle = None
self._victoryBalanceTitle = None
self._scoreTitle = None
self._spamWarning = None
self._spamWarningIvalName = "PartyCogActivityGui-SpamWarning"
def load(self):
self._initPiePowerMeter()
self._initScore()
self._initCogTracker()
self._initSpamWarning()
self._initControlGui()
self._initVictoryBalanceBar()
def unload(self):
if self._cogTracker is not None:
self._cogTracker.destory()
self._cogTracker = None
if self._piePowerMeter is not None:
self._piePowerMeter.destroy()
self._piePowerMeter = None
if self._piePowerTitle is not None:
self._piePowerTitle.destroy()
self._piePowerTitle = None
if self._scoreLabel is not None:
self._scoreLabel.destroy()
self._scoreLabel = None
if self._scoreTitle is not None:
self._scoreTitle.destroy()
self._scoreTitle = None
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.destroy()
self._spamWarning = None
if hasattr(self, '_attackKeys'):
self._attackKeys.detachNode()
del self._attackKeys
if hasattr(self, '_moveKeys'):
self._moveKeys.detachNode()
del self._moveKeys
if self._victoryBalanceBar:
self._victoryBalanceBar.detachNode()
self._victoryBalanceBar = None
if self._victoryBalanceBarOrange:
self._victoryBalanceBarOrange.detachNode()
self._victoryBalanceBarOrange = None
if self._victoryBalanceBarPie:
self._victoryBalanceBarPie.detachNode()
self._victoryBalanceBarPie = None
if self._victoryBalanceBarArrow:
self._victoryBalanceBarArrow.detachNode()
self._victoryBalanceBarArrow = None
def _initVictoryBalanceBar(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
victoryBalanceBar = loader.loadModel("phase_13/models/parties/tt_m_gui_pty_pieToss_balanceBar")
self._victoryBalanceBar = victoryBalanceBar.find("**/*tt_t_gui_pty_pieToss_balanceBarBG")
self._victoryBalanceBar.reparentTo(aspect2d)
self._victoryBalanceBar.setBin("fixed", 0)
self._victoryBalanceBar.setPos(PartyGlobals.CogActivityVictoryBarPos)
self._victoryBalanceBar.setScale(1)
self._victoryBalanceBarOrange = victoryBalanceBar.find("**/*tt_t_gui_pty_pieToss_balanceBarOrange")
self._victoryBalanceBarOrange.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarOrange.setBin("fixed", 1)
self._victoryBalanceBarOrange.setPos(PartyGlobals.CogActivityVictoryBarOrangePos)
self._victoryBalanceBarOrange.setScale(PartyGlobals.CogActivityBarStartScale, 1.0, 1.0)
self._victoryBalanceBarPie = victoryBalanceBar.find("**/*tt_t_gui_pty_pieToss_balanceBarPie")
self._victoryBalanceBarPie.reparentTo(self._victoryBalanceBar)
self._victoryBalanceBarPie.setBin("fixed", 2)
self._victoryBalanceBarPie.setX(PartyGlobals.CogActivityVictoryBarPiePos[0])
self._victoryBalanceBarPie.setY(PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(PartyGlobals.CogActivityVictoryBarPiePos[2])
self._victoryBalanceBarPie.setScale(PartyGlobals.CogActivityBarPieScale)
self._victoryBalanceBarArrow = victoryBalanceBar.find("**/*tt_t_gui_pty_pieToss_balanceArrow")
self._victoryBalanceBarArrow.reparentTo(self._victoryBalanceBarPie)
self._victoryBalanceBarArrow.setBin("fixed", 2)
self._victoryBalanceBarArrow.setPos(PartyGlobals.CogActivityVictoryBarArrow)
self._victoryBalanceBarArrow.setScale(1/PartyGlobals.CogActivityBarPieScale)
# self._victoryBalanceBar = DirectWaitBar(
# frameSize = (-h, h, -w, w),
# relief = DGG.SUNKEN,
# frameColor = PartyGlobals.CogActivityColors[0],
# borderWidth = (0.0, 0.0),
# barColor = PartyGlobals.CogActivityColors[1],
# pos = PartyGlobals.CogActivityVictoryBarPos,
# hpr = (0.0, 0.0, -90.0),
# )
# self._victoryBalanceBar.setBin("fixed", 0)
# self._victoryBalanceBar["value"] = 50
def _initControlGui(self):
self._attackIvalName = "PartyCogActivityGui-attackKeys"
self._moveIvalName = "PartyCogActivityGui-moveKeys"
pieTossControls = loader.loadModel("phase_13/models/parties/tt_m_gui_pty_pieToss_controls")
self._attackKeys = pieTossControls.find("**/*control*")
self._moveKeys = pieTossControls.find("**/*arrow*")
self._moveKeys.reparentTo(aspect2d)
self._moveKeys.setPos(1.0, 0.0, -0.435)
self._moveKeys.setScale(0.15)
self._attackKeys.reparentTo(aspect2d)
self._attackKeys.setPos(0.85, 0.0, -0.45)
self._attackKeys.setScale(0.15)
self._moveKeys.hide()
self._attackKeys.hide()
def _initPiePowerMeter(self):
h = PartyGlobals.CogActivityPowerMeterHeight / 2.0
w = PartyGlobals.CogActivityPowerMeterWidth / 2.0
self._piePowerMeter = DirectWaitBar(
frameSize = (-h, h, -w, w),
relief = DGG.GROOVE,
frameColor = (0.9, 0.9, 0.9, 1.0),
borderWidth = (0.01, 0.01),
barColor = PartyGlobals.CogActivityColors[0],
pos = PartyGlobals.CogActivityPowerMeterPos,
hpr = (0.0, 0.0, -90.0),
)
self._piePowerMeter.setBin("fixed", 0)
self._piePowerTitle = OnscreenText(
text=TTLocalizer.PartyCogGuiPowerLabel,
pos=PartyGlobals.CogActivityPowerMeterTextPos,
scale=0.05,
fg=(1.0, 1.0, 1.0, 1.0),
align=TextNode.ACenter,
)
self._piePowerTitle.setBin("fixed", 0)
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def _initScore(self):
self._scoreLabel = OnscreenText(
text="0",
pos=PartyGlobals.CogActivityScorePos,
scale=PartyGlobals.TugOfWarTextWordScale,
#fg=base.localAvatar.style.getHeadColor(),
#fg=PartyGlobals.TeamActivityStatusColor,
fg=(1.0, 1.0, 0.0, 1.0),
align=TextNode.ARight,
font=ToontownGlobals.getSignFont(),
mayChange=True,
)
self._scoreTitle = OnscreenText(
text=TTLocalizer.PartyCogGuiScoreLabel,
pos=PartyGlobals.CogActivityScoreTitle,
scale=0.05,
fg=(1.0, 1.0, 1.0, 1.0),
#fg=(0.0, 0.0, 0.0, 1.0),
align=TextNode.ARight,
)
self._scoreLabel.hide()
self._scoreTitle.hide()
def _initCogTracker(self):
self._cogTracker = PartyCogTrackerGui()
def _initSpamWarning(self):
self._spamWarning = OnscreenText(
text = TTLocalizer.PartyCogGuiSpamWarning,
scale = 0.15,
fg = (1.0, 1.0, 0, 1.0),
shadow = (0, 0, 0, 0.62),
mayChange = False,
pos = (0, 0.33)
)
self._spamWarning.hide()
def showScore(self):
self._scoreLabel.show()
self._scoreTitle.show()
def hideScore(self):
self._scoreLabel.hide()
self._scoreTitle.hide()
def setScore(self, score=0):
self._scoreLabel["text"] = str(score)
def resetPiePowerMeter(self):
self._piePowerMeter["value"] = 0
def showPiePowerMeter(self):
self._piePowerMeter.show()
self._piePowerTitle.show()
def hidePiePowerMeter(self):
self._piePowerMeter.hide()
self._piePowerTitle.hide()
def updatePiePowerMeter(self, value):
self._piePowerMeter["value"] = value
# Uncomment this part if you want the bar to by a psychedelic change of colors.
# self._piePowerMeter["barColor"] = (
# min(value * 2, 100) / 100.0,
# min((100 - value) * 2, 100) / 100.0,
# 0.0,
# 1.0
# )
def getPiePowerMeterValue(self):
return self._piePowerMeter["value"]
def hideSpamWarning(self):
taskMgr.remove(self._spamWarningIvalName)
if self._spamWarning:
self._spamWarning.hide()
def showSpamWarning(self):
if self._spamWarning.isHidden():
self._spamWarning.show()
taskMgr.remove(self._spamWarningIvalName)
Sequence(
# Boing it in to grab attention.
ToontownIntervals.getPulseLargerIval(self._spamWarning, ""),
# Let it sit on-screen for a while.
Wait(PartyGlobals.CogActivitySpamWarningShowTime),
# Then get rid of it.
Func(self.hideSpamWarning),
name=self._spamWarningIvalName,
autoFinish=1
).start()
def hide(self):
self.hidePiePowerMeter()
self.hideScore()
self.hideSpamWarning()
self.hideControls()
def disableToontownHUD(self):
base.localAvatar.hideName()
base.localAvatar.laffMeter.hide()
base.setCellsAvailable(base.bottomCells + [base.rightCells[1]], False)
def enableToontownHUD(self):
base.localAvatar.showName()
base.localAvatar.laffMeter.show()
base.setCellsAvailable(base.bottomCells + [base.rightCells[1]], True)
def setTeam(self, team):
self.team = team
if team == 0:
self._cogTracker.frame.setR(180)
# The old default power bar color was blue, but that was confusing because there's a blue team.
# So the idea here is to make the color the same as the player's team.
self._piePowerMeter["barColor"] = PartyGlobals.CogActivityColors[team]
def startTrackingCogs(self, cogs):
self.cogs = cogs
taskMgr.add(self.trackCogs, "trackCogs")
def trackCogs(self, task):
if self.cogs is None:
return
self._updateVictoryBar()
for i, cog in enumerate(self.cogs):
self._cogTracker.updateCog(i,cog,self.team)
return task.cont
def _updateVictoryBar(self):
if not ( hasattr(self, "_victoryBalanceBar") and self._victoryBalanceBar):
return
netDistance = 0
for cog in self.cogs:
netDistance = netDistance + cog.targetDistance
teamDistance = netDistance/6.0
self._victoryBalanceBarOrange.setScale(PartyGlobals.CogActivityBarStartScale + \
(teamDistance * 10 * PartyGlobals.CogActivityBarUnitScale),1.0,1.0)
#self._victoryBalanceBar["value"] = 50 + (netDistance/6 * 100)
self._victoryBalanceBarPie.setX(PartyGlobals.CogActivityVictoryBarPiePos[0] + \
(teamDistance * 10 *PartyGlobals.CogActivityBarPieUnitMove))
self._victoryBalanceBarPie.setY(PartyGlobals.CogActivityVictoryBarPiePos[1])
self._victoryBalanceBarPie.setZ(PartyGlobals.CogActivityVictoryBarPiePos[2])
if teamDistance>0.0:
self._victoryBalanceBarArrow.setColor(PartyGlobals.CogActivityColors[1])
elif teamDistance<0.0:
self._victoryBalanceBarArrow.setColor(PartyGlobals.CogActivityColors[0])
else:
self._victoryBalanceBarArrow.setColor(VBase4(1.0,1.0,1.0,1.0))
def stopTrackingCogs(self):
taskMgr.remove("trackCogs")
def showAttackControls(self):
if self._attackKeys.isHidden():
self._attackKeys.show()
taskMgr.remove(self._attackIvalName)
Sequence(
ToontownIntervals.getPulseLargerIval(self._attackKeys, "", scale = 0.15),
Wait(PartyGlobals.CogActivityControlsShowTime),
Func(self.hideAttackControls),
name=self._attackIvalName,
autoFinish=1
).start()
def showMoveControls(self):
if self._moveKeys.isHidden() and not self._attackKeys.isHidden():
self._moveKeys.show()
taskMgr.remove(self._moveIvalName)
Sequence(
ToontownIntervals.getPulseLargerIval(self._moveKeys, "", scale = 0.15),
Wait(PartyGlobals.CogActivityControlsShowTime),
Func(self.hideMoveControls),
name=self._moveIvalName,
autoFinish=1
).start()
def hideAttackControls(self):
taskMgr.remove(self._attackIvalName)
if hasattr(self, "_attackKeys") and self._attackKeys:
self._attackKeys.hide()
def hideMoveControls(self):
taskMgr.remove(self._moveIvalName)
if hasattr(self, "_moveKeys") and self._moveKeys:
self._moveKeys.hide()
def hideControls(self):
self.hideMoveControls()
self.hideAttackControls()
