toontown-just-works/toontown/town/TownBattleSOSPetInfoPanel.py
2024-07-07 18:08:39 -05:00

160 lines
8.3 KiB
Python

from panda3d.core import *
from direct.fsm import StateData
from direct.gui.DirectGui import *
from toontown.toonbase import TTLocalizer
from toontown.pets import Pet, PetTricks, PetDetailPanel
from toontown.speedchat import TTSCPetTrickMenu
from otp.speedchat import SpeedChatGlobals, SCSettings
from otp.otpbase import OTPLocalizer
class TownBattleSOSPetInfoPanel(StateData.StateData):
def __init__(self, doneEvent):
StateData.StateData.__init__(self, doneEvent)
def load(self):
gui = loader.loadModel('phase_3.5/models/gui/PetControlPannel')
guiScale = 0.116
guiPos = (0, 0, 0)
self.frame = DirectFrame(image=gui, scale=guiScale, pos=guiPos, relief=None)
self.frame.hide()
disabledImageColor = Vec4(0.6, 0.6, 0.6, 1)
text0Color = Vec4(1, 1, 1, 1)
text1Color = Vec4(0.5, 1, 0.5, 1)
text2Color = Vec4(1, 1, 0.5, 1)
text3Color = Vec4(0.6, 0.6, 0.6, 1)
self.closeButton = DirectButton(parent=self.frame, image=(gui.find('**/CancelButtonUp'), gui.find('**/CancelButtonDown'), gui.find('**/CancelButtonRollover')), relief=None, command=self.__handleClose)
self.feedButton = DirectButton(parent=self.frame, image=(gui.find('**/ButtonFeedUp'),
gui.find('**/ButtonFeedDown'),
gui.find('**/ButtonFeedRollover'),
gui.find('**/ButtonFeedUp')), geom=gui.find('**/PetControlFeedIcon'), image3_color=disabledImageColor, relief=None, text=TTLocalizer.PetPanelFeed, text_scale=0.5, text0_fg=text0Color, text1_fg=text1Color, text2_fg=text2Color, text3_fg=text3Color, text_pos=(-0.5, 2.8), text_align=TextNode.ALeft)
self.feedButton['state'] = DGG.DISABLED
self.callButton = DirectButton(parent=self.frame, image=(gui.find('**/ButtonGoToUp'),
gui.find('**/ButtonGoToDown'),
gui.find('**/ButtonGoToRollover'),
gui.find('**/ButtonGoToUp')), geom=gui.find('**/PetControlGoToIcon'), image3_color=disabledImageColor, relief=None, text=TTLocalizer.PetPanelCall, text0_fg=text0Color, text1_fg=text1Color, text2_fg=text2Color, text3_fg=text3Color, text_scale=0.5, text_pos=(-0.5, 1.3), text_align=TextNode.ALeft)
self.callButton['state'] = DGG.DISABLED
self.scratchButton = DirectButton(parent=self.frame, image=(gui.find('**/ButtonScratchUp'),
gui.find('**/ButtonScratchDown'),
gui.find('**/ButtonScratchRollover'),
gui.find('**/ButtonScratchUp')), geom=gui.find('**/PetControlScratchIcon'), image3_color=disabledImageColor, relief=None, text=TTLocalizer.PetPanelScratch, text0_fg=text0Color, text1_fg=text1Color, text2_fg=text2Color, text3_fg=text3Color, text_scale=0.5, text_pos=(-0.5, 2.05), text_align=TextNode.ALeft)
self.scratchButton['state'] = DGG.DISABLED
self.callOwnerButton = DirectButton(parent=self.frame, image=(gui.find('**/PetControlToonButtonUp'), gui.find('**/PetControlToonButtonDown'), gui.find('**/PetControlToonButtonRollover')), geom=gui.find('**/PetControlToonIcon'), geom3_color=disabledImageColor, relief=None, image3_color=disabledImageColor, text=('',
TTLocalizer.PetPanelOwner,
TTLocalizer.PetPanelOwner,
''), text_fg=text2Color, text_shadow=(0, 0, 0, 1), text_scale=0.35, text_pos=(0.3, 1.1), text_align=TextNode.ACenter, command=self.__handleDetail)
self.callOwnerButton['state'] = DGG.DISABLED
self.detailButton = DirectButton(parent=self.frame, image=(gui.find('**/PetControlToonButtonUp1'), gui.find('**/PetControlToonButtonDown1'), gui.find('**/PetControlToonButtonRollover1')), geom=gui.find('**/PetBattleIcon'), geom3_color=disabledImageColor, relief=None, pos=(0, 0, 0), image3_color=disabledImageColor, text=('',
TTLocalizer.PetPanelDetail,
TTLocalizer.PetPanelDetail,
''), text_fg=text2Color, text_shadow=(0, 0, 0, 1), text_scale=0.35, text_pos=(0.3, 1.1), text_align=TextNode.ACenter, command=self.__handleDetail)
self.detailButton['state'] = DGG.NORMAL
gui.removeNode()
self.nameLabel = None
self.trickMenu = TTSCPetTrickMenu.TTSCPetTrickMenu()
self.settings = SCSettings.SCSettings(eventPrefix='')
self.trickMenu.privSetSettingsRef(self.settings)
self.trickMenuEventName = self.trickMenu.getEventName(SpeedChatGlobals.SCStaticTextMsgEvent)
self.trickMenu.setScale(0.055)
self.trickMenu.setBin('gui-popup', 0)
self.trickMenu.finalizeAll()
localAvatar.chatMgr.chatInputSpeedChat.whisperAvatarId = None
self.petDetailPanel = None
return
def unload(self):
self.frame.destroy()
del self.frame
self.frame = None
if hasattr(self, 'petView'):
self.petView.removeNode()
del self.petView
if hasattr(self, 'petModel'):
self.petModel.delete()
del self.petModel
del self.closeButton
del self.feedButton
del self.callButton
del self.scratchButton
del self.callOwnerButton
del self.detailButton
self.trickMenu.destroy()
del self.trickMenu
del self.petDetailPanel
return
def enter(self, petProxyId):
self.petProxyId = petProxyId
if petProxyId not in base.cr.doId2do:
self.notify.warning('petProxyId %s not in doId2do!' % petProxyId)
return
self.petProxy = base.cr.doId2do[petProxyId]
self.__fillPetInfo(self.petProxy)
self.frame.show()
self.accept(self.trickMenuEventName, self.__handleTrickMenuEvent)
self.trickMenu.reparentTo(aspect2dp, DGG.FOREGROUND_SORT_INDEX)
localAvatar.chatMgr.chatInputSpeedChat.whisperAvatarId = None
self.detailButton['state'] = DGG.NORMAL
return
def exit(self):
self.ignore(self.trickMenuEventName)
self.trickMenu.reparentTo(hidden)
self.petProxy = None
if self.petDetailPanel != None:
self.petDetailPanel.cleanup()
self.petDetailPanel = None
self.frame.hide()
return
def __handleTrickMenuEvent(self, textId):
if textId in PetTricks.ScId2trickId:
trickId = PetTricks.ScId2trickId[textId]
doneStatus = {'mode': 'OK',
'trickId': trickId}
messenger.send(self.doneEvent, [doneStatus])
self.detailButton['state'] = DGG.NORMAL
def __handleClose(self):
doneStatus = {'mode': 'Back'}
messenger.send(self.doneEvent, [doneStatus])
def __handleCall(self):
doneStatus = {'mode': 'OK',
'trickId': 0}
messenger.send(self.doneEvent, [doneStatus])
def __handleDetailDone(self):
if self.petDetailPanel != None:
self.petDetailPanel.cleanup()
self.petDetailPanel = None
self.detailButton['state'] = DGG.NORMAL
return
def __handleDetail(self):
self.petDetailPanel = PetDetailPanel.PetDetailPanel(pet=self.petProxy, closeCallback=self.__handleDetailDone, parent=self.frame)
self.detailButton['state'] = DGG.DISABLED
def __fillPetInfo(self, avatar):
self.notify.debug('__fillPetInfo(): doId=%s' % avatar.doId)
if self.nameLabel == None:
self.petView = self.frame.attachNewNode('petView')
self.petView.setPos(0, 0, 5.4)
self.petModel = Pet.Pet(forGui=1)
self.petModel.setDNA(avatar.getDNA())
self.petModel.fitAndCenterHead(3.575, forGui=1)
self.petModel.reparentTo(self.petView)
self.petModel.enterNeutralHappy()
self.petModel.startBlink()
self.petModel.setScale(0.75)
self.nameLabel = DirectLabel(parent=self.frame, pos=(0, 0, 5.2), relief=None, text=avatar.getName(), text_font=avatar.getFont(), text_fg=Vec4(0, 0, 0, 1), text_pos=(0, 0), text_scale=0.4, text_wordwrap=7.5, text_shadow=(1, 1, 1, 1))
self.stateLabel = DirectLabel(parent=self.frame, pos=(0.7, 0, 3.5), relief=None, text='', text_font=avatar.getFont(), text_fg=Vec4(0, 0, 0, 1), text_scale=0.4, text_wordwrap=7.5, text_shadow=(1, 1, 1, 1))
self.__refreshPetInfo(avatar)
return
def __refreshPetInfo(self, avatar):
self.notify.debug('__refreshPetInfo(): doId=%s' % avatar.doId)
avatar.updateOfflineMood()
mood = avatar.getDominantMood()
self.stateLabel['text'] = TTLocalizer.PetMoodAdjectives[mood]
self.nameLabel['text'] = avatar.getName()