69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
from toontown.toonbase import ToontownGlobals
|
|
from direct.interval.IntervalGlobal import Parallel, Sequence, Func, Wait
|
|
from panda3d.core import Vec4, CSDefault, TransformState, NodePath, TransparencyAttrib
|
|
from panda3d.toontown import loadDNAFile
|
|
|
|
class HolidayDecorator:
|
|
|
|
def __init__(self):
|
|
self.dnaStore = base.cr.playGame.dnaStore
|
|
self.swapIval = None
|
|
return
|
|
|
|
def exit(self):
|
|
if self.swapIval is not None and self.swapIval.isPlaying():
|
|
self.swapIval.finish()
|
|
return
|
|
|
|
def decorate(self):
|
|
self.updateHoodDNAStore()
|
|
self.swapIval = self.getSwapVisibleIval()
|
|
if self.swapIval:
|
|
self.swapIval.start()
|
|
|
|
def undecorate(self):
|
|
holidayIds = base.cr.newsManager.getDecorationHolidayId()
|
|
if len(holidayIds) > 0:
|
|
self.decorate()
|
|
return
|
|
storageFile = base.cr.playGame.hood.storageDNAFile
|
|
if storageFile:
|
|
loadDNAFile(self.dnaStore, storageFile, CSDefault)
|
|
self.swapIval = self.getSwapVisibleIval()
|
|
if self.swapIval:
|
|
self.swapIval.start()
|
|
|
|
def updateHoodDNAStore(self):
|
|
hood = base.cr.playGame.hood
|
|
holidayIds = base.cr.newsManager.getDecorationHolidayId()
|
|
for holiday in holidayIds:
|
|
for storageFile in hood.holidayStorageDNADict.get(holiday, []):
|
|
loadDNAFile(self.dnaStore, storageFile, CSDefault)
|
|
|
|
def getSwapVisibleIval(self, wait = 5.0, tFadeOut = 3.0, tFadeIn = 3.0):
|
|
loader = base.cr.playGame.hood.loader
|
|
npl = render.findAllMatches('**/=DNARoot=holiday_prop;+s')
|
|
p = Parallel()
|
|
for i in range(npl.getNumPaths()):
|
|
np = npl.getPath(i)
|
|
np.setTransparency(TransparencyAttrib.MDual, 1)
|
|
if not np.hasTag('DNACode'):
|
|
continue
|
|
dnaCode = np.getTag('DNACode')
|
|
dnaNode = self.dnaStore.findNode(dnaCode)
|
|
if dnaNode.isEmpty():
|
|
continue
|
|
newNP = dnaNode.copyTo(np.getParent())
|
|
newNP.setTag('DNARoot', 'holiday_prop')
|
|
newNP.setTag('DNACode', dnaCode)
|
|
newNP.setColorScale(1, 1, 1, 0)
|
|
newNP.setTransparency(TransparencyAttrib.MDual, 1)
|
|
if np.hasTag('transformIndex'):
|
|
index = int(np.getTag('transformIndex'))
|
|
transform = loader.holidayPropTransforms.get(index, TransformState.makeIdentity())
|
|
newNP.setTransform(NodePath(), transform)
|
|
newNP.setTag('transformIndex', repr(index))
|
|
s = Sequence(Wait(wait), np.colorScaleInterval(tFadeOut, Vec4(1, 1, 1, 0), startColorScale=Vec4(1, 1, 1, 1), blendType='easeInOut'), Func(np.detachNode), Func(np.clearTransparency), newNP.colorScaleInterval(tFadeOut, Vec4(1, 1, 1, 1), startColorScale=Vec4(1, 1, 1, 0), blendType='easeInOut'), Func(newNP.clearTransparency), Func(newNP.clearColorScale))
|
|
p.append(s)
|
|
|
|
return p
|