Poodletooth-iLand/toontown/toon/DistributedNPCGlove.py

192 lines
7.1 KiB
Python
Raw Normal View History

from direct.fsm import ClassicFSM, State
from direct.gui.DirectGui import *
from direct.task.Task import Task
from pandac.PandaModules import *
import time
from DistributedNPCToonBase import *
from toontown.chat.ChatGlobals import *
from toontown.effects import DustCloud
from toontown.nametag.NametagGlobals import *
2015-03-08 13:01:05 +00:00
from toontown.toonbase import TTLocalizer, ToontownGlobals
from toontown.toon.ToonDNA import allColorsList
def getDustCloud(toon):
dustCloud = DustCloud.DustCloud(fBillboard=0)
2015-05-11 19:05:59 +00:00
dustCloud.setBillboardAxis(2.0)
dustCloud.setZ(3)
dustCloud.setScale(0.4)
dustCloud.createTrack()
return Sequence(Wait(0.5), Func(dustCloud.reparentTo, toon), dustCloud.track, Func(dustCloud.destroy))
class DistributedNPCGlove(DistributedNPCToonBase):
def __init__(self, cr):
DistributedNPCToonBase.__init__(self, cr)
self.nextCollision = 0
self.fsm = ClassicFSM.ClassicFSM(
'NPCGlove',
[
State.State('off', self.enterOff, self.exitOff, ['pickColor']),
State.State('pickColor', self.enterPickColor, self.exitPickColor, ['off'])
], 'off', 'off')
self.fsm.enterInitialState()
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.title = None
self.notice = None
self.color = None
self.buyButton = None
self.cancelButton = None
self.leftButton = None
self.rightButton = None
self.index = 0
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.gui = loader.loadModel('phase_3/models/gui/tt_m_gui_mat_mainGui')
self.shuffleArrowUp = self.gui.find('**/tt_t_gui_mat_shuffleArrowUp')
self.shuffleArrowDown = self.gui.find('**/tt_t_gui_mat_shuffleArrowDown')
self.shuffleUp = self.gui.find('**/tt_t_gui_mat_shuffleUp')
self.shuffleDown = self.gui.find('**/tt_t_gui_mat_shuffleDown')
def disable(self):
self.ignoreAll()
2015-03-08 13:01:05 +00:00
self.destroyGui()
self.nextCollision = 0
DistributedNPCToonBase.disable(self)
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
def destroyGui(self):
for element in [self.title, self.notice, self.color, self.buyButton, self.cancelButton, self.leftButton, self.rightButton]:
if element:
element.destroy()
element = None
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.index = 0
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
def createGui(self):
self.title = DirectLabel(aspect2d, relief=None, text=TTLocalizer.GloveGuiTitle,
2015-03-08 13:30:47 +00:00
text_fg=(0, 1, 0, 1), text_scale=0.15, text_font=ToontownGlobals.getSignFont(),
2015-03-08 13:01:05 +00:00
pos=(0, 0, -0.30), text_shadow=(1, 1, 1, 1))
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.notice = DirectLabel(aspect2d, relief=None, text=TTLocalizer.GloveGuiNotice % ToontownGlobals.GloveCost,
text_fg=(1, 0, 0, 1), text_scale=0.11, text_font=ToontownGlobals.getSignFont(),
pos=(0, 0, -0.45), text_shadow=(1, 1, 1, 1))
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.color = DirectLabel(aspect2d, relief=None, text='',
text_scale=0.11, text_font=ToontownGlobals.getSignFont(),
pos=(0, 0, -0.70), text_shadow=(1, 1, 1, 1))
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.buyButton = DirectButton(aspect2d, relief=None, image=(self.shuffleUp, self.shuffleDown),
text=TTLocalizer.GloveGuiBuy, text_font=ToontownGlobals.getInterfaceFont(),
text_scale=0.11, text_pos=(0, -0.02), pos=(-0.60, 0, -0.90), text_fg=(1, 1, 1, 1),
text_shadow=(0, 0, 0, 1), command=self.handleBuy)
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.cancelButton = DirectButton(aspect2d, relief=None, image=(self.shuffleUp, self.shuffleDown),
text=TTLocalizer.GloveGuiCancel, text_font=ToontownGlobals.getInterfaceFont(),
text_scale=0.11, text_pos=(0, -0.02), pos=(0.60, 0, -0.90), text_fg=(1, 1, 1, 1),
text_shadow=(0, 0, 0, 1), command=self.leave)
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.leftButton = DirectButton(aspect2d, relief=None, image=(self.shuffleArrowUp, self.shuffleArrowDown),
pos=(-0.60, 0, -0.66), command=self.handleSetIndex, extraArgs=[-1])
2015-03-08 13:01:05 +00:00
self.rightButton = DirectButton(aspect2d, relief=None, image=(self.shuffleArrowUp, self.shuffleArrowDown),
pos=(0.60, 0, -0.66), scale=-1, command=self.handleSetIndex, extraArgs=[1])
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
self.updateGuiByIndex()
def handleSetIndex(self, offset):
newIndex = self.index + offset
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
if newIndex > -1 and newIndex < len(TTLocalizer.NumToColor):
self.index = newIndex
self.updateGuiByIndex()
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
def updateGuiByIndex(self):
self.color['text'] = TTLocalizer.NumToColor[self.index]
self.color['text_fg'] = allColorsList[self.index]
2015-05-11 19:05:59 +00:00
2015-03-08 13:01:05 +00:00
def handleBuy(self):
self.d_requestTransformation(self.index)
2015-05-11 19:05:59 +00:00
def initToonState(self):
self.setAnimState('neutral', 1.05, None, None)
self.setPosHpr(101, -14, 4, -305, 0, 0)
def getCollSphereRadius(self):
return 1.0
def handleCollisionSphereEnter(self, collEntry):
self.currentTime = time.time()
2015-03-08 13:01:05 +00:00
if self.nextCollision <= self.currentTime:
self.fsm.request('pickColor')
2015-03-08 13:01:05 +00:00
self.nextCollision = self.currentTime + 2
def enterOff(self):
pass
def exitOff(self):
pass
def enterPickColor(self):
base.cr.playGame.getPlace().setState('stopped')
2015-03-08 13:07:29 +00:00
taskMgr.doMethodLater(45, self.leave, 'npcSleepTask-%s' % self.doId)
2015-03-08 13:01:05 +00:00
self.setChatAbsolute('', CFSpeech)
2015-05-11 19:05:59 +00:00
if base.localAvatar.getTotalMoney() < ToontownGlobals.GloveCost:
2015-03-08 13:01:05 +00:00
self.setChatAbsolute(self.getMessageById(2), CFSpeech|CFTimeout)
self.reset()
else:
self.popupPickColorGUI()
def exitPickColor(self, task=None):
taskMgr.remove('npcSleepTask-%s' % self.doId)
2015-03-08 13:01:05 +00:00
taskMgr.doMethodLater(0.5, self.reset, 'avatarRecover-%s-%s' % (self.doId, base.localAvatar.doId))
2015-05-11 19:05:59 +00:00
if task is not None:
return task.done
def popupPickColorGUI(self):
self.setChatAbsolute('', CFSpeech)
2015-03-08 13:23:14 +00:00
self.setChatAbsolute(TTLocalizer.GlovePickColorMessage, CFSpeech|CFTimeout)
base.setCellsActive(base.bottomCells, 0)
2015-03-08 13:01:05 +00:00
self.createGui()
def getMessageById(self, response):
if response == 1:
return TTLocalizer.GloveSameColorMessage
elif response == 2:
return TTLocalizer.GloveNoMoneyMessage
else:
return TTLocalizer.GloveSuccessMessage
def doTransformation(self, avId, response):
av = self.cr.doId2do.get(avId)
2015-05-11 19:05:59 +00:00
if not av:
return
2015-05-11 19:05:59 +00:00
if response == 3:
getDustCloud(av).start()
self.setChatAbsolute('', CFSpeech)
self.setChatAbsolute(self.getMessageById(response), CFSpeech|CFTimeout)
def d_requestTransformation(self, color):
self.sendUpdate('requestTransformation', [color])
self.reset()
2015-03-08 23:06:02 +00:00
def leave(self, task=None):
2015-03-08 13:01:05 +00:00
self.setChatAbsolute('', CFSpeech)
self.setChatAbsolute(TTLocalizer.GloveByeMessage, CFSpeech|CFTimeout)
2015-03-08 23:06:02 +00:00
self.reset(task)
2015-05-11 19:05:59 +00:00
def reset(self, task=None):
self.fsm.request('off')
base.cr.playGame.getPlace().setState('walk')
base.setCellsActive(base.bottomCells, 1)
2015-03-08 13:08:57 +00:00
self.destroyGui()
if task is not None:
2015-05-11 19:05:59 +00:00
return task.done