Poodletooth-iLand/toontown/shtiker/SummonCogDialog.py

177 lines
8.2 KiB
Python
Raw Normal View History

2015-03-03 22:10:12 +00:00
from direct.gui.DirectGui import *
from pandac.PandaModules import *
from direct.directnotify import DirectNotifyGlobal
from direct.fsm import StateData
from toontown.toonbase import TTLocalizer
from toontown.toontowngui import TTDialog
from toontown.toonbase import ToontownGlobals
from toontown.suit import SuitDNA
from toontown.suit import Suit
from toontown.battle import SuitBattleGlobals
from toontown.toon import NPCToons
TTL = TTLocalizer
class SummonCogDialog(DirectFrame, StateData.StateData):
notify = DirectNotifyGlobal.directNotify.newCategory('SummonCogDialog')
notify.setInfo(True)
def __init__(self, suitIndex):
DirectFrame.__init__(self, parent=aspect2dp, pos=(0, 0, 0.3), relief=None, image=DGG.getDefaultDialogGeom(), image_scale=(1.6, 1, 0.7), image_pos=(0, 0, 0.18), image_color=ToontownGlobals.GlobalDialogColor, text=TTL.SummonDlgTitle, text_scale=0.12, text_pos=(0, 0.4), borderWidth=(0.01, 0.01), sortOrder=NO_FADE_SORT_INDEX)
StateData.StateData.__init__(self, 'summon-cog-done')
self.initialiseoptions(SummonCogDialog)
self.suitIndex = suitIndex
base.summonDialog = self
self.popup = None
self.suitName = SuitDNA.suitHeadTypes[self.suitIndex]
self.suitFullName = SuitBattleGlobals.SuitAttributes[self.suitName]['name']
return
def unload(self):
if self.isLoaded == 0:
return None
self.isLoaded = 0
self.exit()
DirectFrame.destroy(self)
return None
def load(self):
if self.isLoaded == 1:
return None
self.isLoaded = 1
gui = loader.loadModel('phase_3/models/gui/dialog_box_buttons_gui')
guiButton = loader.loadModel('phase_3/models/gui/quit_button')
self.head = Suit.attachSuitHead(self, self.suitName)
z = self.head.getZ()
self.head.setPos(-0.4, -0.1, z + 0.2)
self.suitLabel = DirectLabel(parent=self, relief=None, text=self.suitFullName, text_font=ToontownGlobals.getSuitFont(), pos=(-0.4, 0, 0), scale=0.07)
closeButtonImage = (gui.find('**/CloseBtn_UP'), gui.find('**/CloseBtn_DN'), gui.find('**/CloseBtn_Rllvr'))
buttonImage = (guiButton.find('**/QuitBtn_UP'), guiButton.find('**/QuitBtn_DN'), guiButton.find('**/QuitBtn_RLVR'))
disabledColor = Vec4(0.5, 0.5, 0.5, 1)
self.summonSingleButton = DirectButton(parent=self, relief=None, text=TTL.SummonDlgButton1, image=buttonImage, image_scale=(1.7, 1, 1), image3_color=disabledColor, text_scale=0.06, text_pos=(0, -0.01), pos=(0.3, 0, 0.25), command=self.issueSummons, extraArgs=['single'])
self.summonBuildingButton = DirectButton(parent=self, relief=None, text=TTL.SummonDlgButton2, image=buttonImage, image_scale=(1.7, 1, 1), image3_color=disabledColor, text_scale=0.06, text_pos=(0, -0.01), pos=(0.3, 0, 0.125), command=self.issueSummons, extraArgs=['building'])
self.summonInvasionButton = DirectButton(parent=self, relief=None, text=TTL.SummonDlgButton3, image=buttonImage, image_scale=(1.7, 1, 1), image3_color=disabledColor, text_scale=0.06, text_pos=(0, -0.01), pos=(0.3, 0, 0.0), command=self.issueSummons, extraArgs=['invasion'])
self.statusLabel = DirectLabel(parent=self, relief=None, text='', text_wordwrap=12, pos=(0.3, 0, 0.25), scale=0.07)
self.cancel = DirectButton(parent=self, relief=None, image=closeButtonImage, pos=(0.7, 0, -0.1), command=self.__cancel)
gui.removeNode()
guiButton.removeNode()
self.hide()
return
def enter(self):
if self.isEntered == 1:
return None
self.isEntered = 1
if self.isLoaded == 0:
self.load()
self.disableButtons()
self.enableButtons()
self.popup = None
base.transitions.fadeScreen(0.5)
self.show()
return
def exit(self):
if self.isEntered == 0:
return None
self.isEntered = 0
self.cleanupDialogs()
base.transitions.noTransitions()
self.ignoreAll()
self.hide()
messenger.send(self.doneEvent, [])
return None
def cleanupDialogs(self):
self.head = None
if self.popup != None:
self.popup.cleanup()
self.popup = None
return
def cogSummonsDone(self, returnCode, suitIndex, buildingId):
self.cancel['state'] = DGG.NORMAL
if self.summonsType == 'single':
if returnCode == 'success':
self.statusLabel['text'] = TTL.SummonDlgSingleSuccess
elif returnCode == 'badlocation':
self.statusLabel['text'] = TTL.SummonDlgSingleBadLoc
elif returnCode == 'fail':
self.statusLabel['text'] = TTL.SummonDlgInvasionFail
elif self.summonsType == 'building':
if returnCode == 'success':
building = base.cr.doId2do.get(buildingId)
dnaStore = base.cr.playGame.dnaStore
buildingTitle = dnaStore.getTitleFromBlockNumber(building.block)
buildingInteriorZone = building.zoneId + 500 + building.block
npcName = TTLocalizer.SummonDlgShopkeeper
npcId = NPCToons.zone2NpcDict.get(buildingInteriorZone)
if npcId:
npcName = NPCToons.getNPCName(npcId[0])
if buildingTitle:
self.statusLabel['text'] = TTL.SummonDlgBldgSuccess % (npcName, buildingTitle)
else:
self.statusLabel['text'] = TTL.SummonDlgBldgSuccess2
elif returnCode == 'badlocation':
self.statusLabel['text'] = TTL.SummonDlgBldgBadLoc
elif returnCode == 'fail':
self.statusLabel['text'] = TTL.SummonDlgInvasionFail
elif self.summonsType == 'invasion':
if returnCode == 'success':
self.statusLabel['text'] = TTL.SummonDlgInvasionSuccess
elif returnCode == 'busy':
self.statusLabel['text'] = TTL.SummonDlgInvasionBusy % self.suitFullName
elif returnCode == 'fail':
self.statusLabel['text'] = TTL.SummonDlgInvasionFail
def hideSummonButtons(self):
self.summonSingleButton.hide()
self.summonBuildingButton.hide()
self.summonInvasionButton.hide()
def issueSummons(self, summonsType):
if summonsType == 'single':
text = TTL.SummonDlgSingleConf
elif summonsType == 'building':
text = TTL.SummonDlgBuildingConf
elif summonsType == 'invasion':
text = TTL.SummonDlgInvasionConf
text = text % self.suitFullName
def handleResponse(resp):
self.popup.cleanup()
self.popup = None
self.reparentTo(self.getParent(), NO_FADE_SORT_INDEX)
base.transitions.fadeScreen(0.5)
if resp == DGG.DIALOG_OK:
self.notify.info('issuing %s summons for %s' % (summonsType, self.suitIndex))
self.accept('cog-summons-response', self.cogSummonsDone)
self.summonsType = summonsType
self.doIssueSummonsText()
base.localAvatar.d_reqCogSummons(self.summonsType, self.suitIndex)
self.hideSummonButtons()
self.cancel['state'] = DGG.DISABLED
return
self.reparentTo(self.getParent(), 0)
self.popup = TTDialog.TTDialog(parent=aspect2dp, style=TTDialog.YesNo, text=text, fadeScreen=1, command=handleResponse)
def doIssueSummonsText(self):
self.disableButtons()
self.statusLabel['text'] = TTL.SummonDlgDelivering
def disableButtons(self):
self.summonSingleButton['state'] = DGG.DISABLED
self.summonBuildingButton['state'] = DGG.DISABLED
self.summonInvasionButton['state'] = DGG.DISABLED
def enableButtons(self):
if base.localAvatar.hasCogSummons(self.suitIndex, 'single'):
self.summonSingleButton['state'] = DGG.NORMAL
if base.localAvatar.hasCogSummons(self.suitIndex, 'building'):
self.summonBuildingButton['state'] = DGG.NORMAL
if base.localAvatar.hasCogSummons(self.suitIndex, 'invasion'):
self.summonInvasionButton['state'] = DGG.NORMAL
def __cancel(self):
self.exit()