from toontown.estate import DistributedPlantBase from direct.directnotify import DirectNotifyGlobal from toontown.estate import FlowerBase from toontown.estate import GardenGlobals from toontown.toontowngui import TTDialog from toontown.toonbase import TTLocalizer DIRT_AS_WATER_INDICATOR = True DIRT_MOUND_HEIGHT = 0.3 class DistributedFlower(DistributedPlantBase.DistributedPlantBase, FlowerBase.FlowerBase): notify = DirectNotifyGlobal.directNotify.newCategory('DistributedFlower') def __init__(self, cr): DistributedPlantBase.DistributedPlantBase.__init__(self, cr) FlowerBase.FlowerBase.__init__(self, 49, 0) self.confirmDialog = None self.stickUp = 1.07 if DIRT_AS_WATER_INDICATOR: self.stickUp += DIRT_MOUND_HEIGHT self.collSphereRadius = 2.2 self.shadowScale = 0.5 self.collSphereOffset = 0.0 self.dirtMound = None self.sandMound = None self.resultDialog = None return def delete(self): DistributedPlantBase.DistributedPlantBase.delete(self) del self.dirtMound del self.sandMound def setTypeIndex(self, typeIndex): DistributedPlantBase.DistributedPlantBase.setTypeIndex(self, typeIndex) self.setSpecies(typeIndex) def showWiltOrBloom(self): if not self.model: return nodePath = self.model desat = None flowerColorIndex = GardenGlobals.PlantAttributes[self.getSpecies()]['varieties'][self.getVariety()][1] colorTuple = GardenGlobals.FlowerColors[flowerColorIndex] useWilted = self.waterLevel < 0 wilt = nodePath.find('**/*wilt*') bloom = nodePath.find('**/*bloom*') if useWilted: wilt.show() desat = wilt.find('**/*desat*') bloom.hide() leaves = wilt.findAllMatches('**/*leaf*') for leafIndex in xrange(leaves.getNumPaths()): leaf = leaves.getPath(leafIndex) leaf.setColorScale(1.0, 0.3, 0.1, 1.0) else: bloom.show() desat = bloom.find('**/*desat*') wilt.hide() if desat and not desat.isEmpty(): desat.setColorScale(colorTuple[0], colorTuple[1], colorTuple[2], 1.0) elif not self.isSeedling(): nodePath.setColorScale(colorTuple[0], colorTuple[1], colorTuple[2], 1.0) return def loadModel(self): DistributedPlantBase.DistributedPlantBase.loadModel(self) self.showWiltOrBloom() self.model.setH(180) flowerScale = 2.0 invFlowerScale = 1.0 / flowerScale self.model.setScale(flowerScale) if DIRT_AS_WATER_INDICATOR: dirtMoundScale = invFlowerScale * 0.73 self.dirtMound = loader.loadModel('phase_5.5/models/estate/dirt_mound') self.dirtMound.reparentTo(self.model) self.dirtMound.setScale(dirtMoundScale) self.dirtMound.setZ(self.dirtMound.getZ() - DIRT_MOUND_HEIGHT / 2.0) self.sandMound = loader.loadModel('phase_5.5/models/estate/sand_mound') self.sandMound.reparentTo(self.model) self.sandMound.setScale(dirtMoundScale) self.sandMound.setZ(self.sandMound.getZ() - DIRT_MOUND_HEIGHT / 2.0) self.adjustWaterIndicator() def handlePicking(self): messenger.send('wakeup') fullName = GardenGlobals.getFlowerVarietyName(self.species, self.variety) if self.isWilted(): self.confirmDialog = TTDialog.TTDialog(style=TTDialog.YesNo, text=TTLocalizer.ConfirmWiltedFlower % {'plant': fullName}, command=self.confirmCallback) elif not self.isFruiting(): self.confirmDialog = TTDialog.TTDialog(style=TTDialog.YesNo, text=TTLocalizer.ConfirmUnbloomingFlower % {'plant': fullName}, command=self.confirmCallback) elif base.localAvatar.isFlowerBasketFull(): self.confirmDialog = TTDialog.TTDialog(style=TTDialog.CancelOnly, text=TTLocalizer.ConfirmBasketFull, command=self.confirmCallback) else: shovel = base.localAvatar.shovel skill = base.localAvatar.shovelSkill shovelPower = GardenGlobals.getShovelPower(shovel, skill) giveSkillUp = True beansRequired = GardenGlobals.getNumBeansRequired(self.species, self.variety) if not shovelPower == beansRequired: giveSkillUp = False if giveSkillUp: if skill == GardenGlobals.getMaxShovelSkill(): text = (TTLocalizer.ConfirmMaxedSkillFlower % {'plant': fullName},) else: text = TTLocalizer.ConfirmSkillupFlower % {'plant': fullName} self.confirmDialog = TTDialog.TTDialog(style=TTDialog.YesNo, text=text, command=self.confirmCallback) else: self.confirmDialog = TTDialog.TTDialog(style=TTDialog.YesNo, text=TTLocalizer.ConfirmNoSkillupFlower % {'plant': fullName}, command=self.confirmCallback) self.confirmDialog.show() base.localAvatar.setInGardenAction(self) base.cr.playGame.getPlace().detectedGardenPlotUse() def confirmCallback(self, value): self.notify.debug('value=%d' % value) self.confirmDialog.destroy() self.confirmDialog = None if value > 0: self.doPicking() else: self.finishInteraction() return def doPicking(self): if not self.canBePicked(): self.notify.debug("I don't own this flower, just returning") return base.localAvatar.showGardeningGui() base.localAvatar.removeShovelRelatedDoId(self.doId) base.localAvatar.setInGardenAction(self) base.cr.playGame.getPlace().detectedGardenPlotUse() self.sendUpdate('removeItem', []) def setWaterLevel(self, waterLevel): DistributedPlantBase.DistributedPlantBase.setWaterLevel(self, waterLevel) self.showWiltOrBloom() if self.model: self.adjustWaterIndicator() def setGrowthLevel(self, growthLevel): origGrowthLevel = self.growthLevel self.growthLevel = growthLevel if origGrowthLevel > -1: self.loadModel() self.makeMovieNode() def makeMovieNode(self): self.movieNode = self.rotateNode.attachNewNode('moviePos') self.movieNode.setPos(0, 3, 0) self.movieNode.setH(180) self.stick2Ground() def setupShadow(self): if DIRT_AS_WATER_INDICATOR: pass else: DistributedPlantBase.DistributedPlantBase.setupShadow(self) self.setShadowHeight(-(self.stickUp + 1)) def adjustWaterIndicator(self): if DIRT_AS_WATER_INDICATOR: if self.dirtMound: curWaterLevel = self.waterLevel if curWaterLevel > self.maxWaterLevel: curWaterLevel = self.maxWaterLevel if curWaterLevel > 0: darkestColorScale = 0.4 lightestColorScale = 1.0 scaleRange = lightestColorScale - darkestColorScale scaleIncrement = scaleRange / self.maxWaterLevel darker = lightestColorScale - scaleIncrement * curWaterLevel self.dirtMound.setColorScale(darker, darker, darker, 1.0) self.sandMound.hide() self.dirtMound.show() else: self.sandMound.show() self.dirtMound.hide() elif self.model: color = float(self.waterLevel) / self.maxWaterLevel self.dropShadow.setColor(0.0, 0.0, 0.0, color) def doResultDialog(self): self.startInteraction() flowerName = GardenGlobals.getFlowerVarietyName(self.species, self.variety) stringToShow = TTLocalizer.getResultPlantedSomethingSentence(flowerName) self.resultDialog = TTDialog.TTDialog(style=TTDialog.Acknowledge, text=stringToShow, command=self.resultsCallback) def resultsCallback(self, value): self.notify.debug('value=%d' % value) if self.resultDialog: self.resultDialog.destroy() self.resultDialog = None self.finishInteraction() return