87 lines
3.7 KiB
Python
87 lines
3.7 KiB
Python
from panda3d.core import CSDefault, NodePath, TransformState, TransparencyAttrib, Vec4
|
|
from direct.interval.IntervalGlobal import Parallel, Sequence, Func, Wait
|
|
|
|
|
|
class HolidayDecorator:
|
|
|
|
def __init__(self):
|
|
self.dnaStore = base.cr.playGame.dnaStore
|
|
self.swapIval = None
|
|
|
|
def exit(self):
|
|
if (self.swapIval is not None) and self.swapIval.isPlaying():
|
|
self.swapIval.finish()
|
|
|
|
def decorate(self):
|
|
# Load the specified seasonal storage file
|
|
self.updateHoodDNAStore()
|
|
self.swapIval = self.getSwapVisibleIval()
|
|
if self.swapIval:
|
|
self.swapIval.start()
|
|
|
|
def undecorate(self):
|
|
# if there are any other decoration holidays running
|
|
holidayIds = base.cr.newsManager.getDecorationHolidayId()
|
|
if len(holidayIds) > 0:
|
|
self.decorate()
|
|
return
|
|
# Reload the regular storage file
|
|
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):
|
|
# Load the specified storage files for this hood to overwrite
|
|
# DNA storage with seasonal specific files
|
|
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
|
|
# Update all visible holiday props
|
|
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
|
|
# Add new
|
|
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)
|
|
# Set transform to match old node paths transform
|
|
if np.hasTag('transformIndex'):
|
|
index = int(np.getTag('transformIndex'))
|
|
transform = loader.holidayPropTransforms.get(index, TransformState.makeIdentity())
|
|
# Position relative to empty node path *just in case* render not top of scene graph
|
|
newNP.setTransform(NodePath(), transform)
|
|
|
|
newNP.setTag('transformIndex', 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
|