from pandac.PandaModules import Vec3, Vec4, Point3, TextNode, VBase4, NodePath from direct.gui.DirectGui import DirectFrame, DirectButton, DirectLabel, DirectScrolledList, DirectCheckButton from direct.gui import DirectGuiGlobals from direct.showbase import DirectObject from direct.showbase import PythonUtil from direct.task.Task import Task from toontown.toonbase import ToontownGlobals from toontown.toonbase import TTLocalizer from toontown.parties import PartyGlobals from toontown.parties.PartyInfo import PartyInfo from toontown.parties import PartyUtils class PartyEditorGridElement(DirectButton): notify = directNotify.newCategory('PartyEditorGridElement') def __init__(self, partyEditor, id, isDecoration, checkSoldOutAndPaidStatusAndAffordability, **kw): self.partyEditor = partyEditor self.id = id self.isDecoration = isDecoration self.checkSoldOutAndPaidStatusAndAffordability = checkSoldOutAndPaidStatusAndAffordability if self.isDecoration: self.name = TTLocalizer.PartyDecorationNameDict[self.id]['editor'] colorList = ((1.0, 1.0, 1.0, 1.0), (0.0, 0.0, 1.0, 1.0), (0.0, 1.0, 1.0, 1.0), (0.5, 0.5, 0.5, 1.0)) self.geom = self.partyEditor.partyPlanner.gui.find('**/%s' % PartyGlobals.DecorationInformationDict[self.id]['gridAsset']) else: self.name = TTLocalizer.PartyActivityNameDict[self.id]['editor'] colorList = ((1.0, 1.0, 1.0, 1.0), (0.0, 1.0, 0.0, 1.0), (1.0, 1.0, 0.0, 1.0), (0.5, 0.5, 0.5, 1.0)) self.geom = self.partyEditor.partyPlanner.gui.find('**/%s' % PartyGlobals.ActivityInformationDict[self.id]['gridAsset']) optiondefs = (('geom', self.geom, None), ('geom_scale', 1.0, None), ('geom_color', colorList[0], None), ('geom1_color', colorList[0], None), ('geom2_color', colorList[0], None), ('geom3_color', colorList[0], None), ('relief', None, None)) self.defineoptions(kw, optiondefs) DirectButton.__init__(self, self.partyEditor.parent) self.initialiseoptions(PartyEditorGridElement) self.setName('%sGridElement' % self.name) self.bind(DirectGuiGlobals.B1PRESS, self.clicked) self.bind(DirectGuiGlobals.B1RELEASE, self.released) self.bind(DirectGuiGlobals.ENTER, self.mouseEnter) self.bind(DirectGuiGlobals.EXIT, self.mouseExit) self.uprightNodePath = NodePath('%sUpright' % self.name) self.uprightNodePath.reparentTo(self) rollOverZOffset = self.getGridSize()[1] / 30.0 self.rolloverTitle = DirectLabel(relief=None, parent=self.uprightNodePath, pos=Point3(0.0, 0.0, rollOverZOffset), text=self.name, text_fg=(1.0, 1.0, 1.0, 1.0), text_shadow=(0.0, 0.0, 0.0, 1.0), text_scale=0.075) self.rolloverTitle.stash() self.stash() self.overValidSquare = False self.lastValidPosition = None self.setColorScale(0.9, 0.9, 0.9, 0.7) self.setTransparency(True) self.mouseOverTrash = False self.centerGridSquare = None return def getCorrectRotation(self): r = self.getR() if r == 90.0: r = 270.0 elif r == 270.0: r = 90.0 if self.id == PartyGlobals.ActivityIds.PartyCannon: return PartyUtils.convertDegreesToPartyGrid(r + 180.0) return PartyUtils.convertDegreesToPartyGrid(r) def getDecorationTuple(self, x, y): return (self.id, self.centerGridSquare.x, PartyGlobals.PartyEditorGridSize[1] - 1 - self.centerGridSquare.y, self.getCorrectRotation()) def getActivityTuple(self, x, y): return (self.id, self.centerGridSquare.x, PartyGlobals.PartyEditorGridSize[1] - 1 - self.centerGridSquare.y, self.getCorrectRotation()) def attach(self, mouseEvent): PartyEditorGridElement.notify.debug('attached grid element %s' % self.name) taskMgr.remove('gridElementDragTask%s' % self.name) vWidget2render2d = self.getPos(render2d) vMouse2render2d = Point3(mouseEvent.getMouse()[0], 0, mouseEvent.getMouse()[1]) taskMgr.add(self.elementDragTask, 'gridElementDragTask%s' % self.name) self.unstash() self.rolloverTitle.unstash() self.uprightNodePath.reparentTo(self) self.setPosHprToDefault() def elementDragTask(self, state): mwn = base.mouseWatcherNode if mwn.hasMouse(): vMouse2render2d = Point3(mwn.getMouse()[0], 0, mwn.getMouse()[1]) newPos = vMouse2render2d if newPos[0] > PartyGlobals.PartyEditorGridBounds[0][0] and newPos[0] < PartyGlobals.PartyEditorGridBounds[1][0] and newPos[2] < PartyGlobals.PartyEditorGridBounds[0][1] and newPos[2] > PartyGlobals.PartyEditorGridBounds[1][1]: centerGridSquare = self.snapToGrid(newPos) if centerGridSquare is not None: self.centerGridSquare = centerGridSquare if not self.overValidSquare: self.setOverValidSquare(True) if self.mouseOverTrash: self.setOverTrash(False) return Task.cont if self.id != PartyGlobals.ActivityIds.PartyClock and newPos[0] > PartyGlobals.PartyEditorTrashBounds[0][0] and newPos[0] < PartyGlobals.PartyEditorTrashBounds[1][0] and newPos[2] < PartyGlobals.PartyEditorTrashBounds[0][1] and newPos[2] > PartyGlobals.PartyEditorTrashBounds[1][1]: if not self.mouseOverTrash: self.setOverTrash(True) elif self.mouseOverTrash: self.setOverTrash(False) self.setPos(render2d, newPos) if self.overValidSquare: self.setOverValidSquare(False) return Task.cont def setOverTrash(self, value): self.mouseOverTrash = value if value: self.partyEditor.trashCanButton['state'] = DirectGuiGlobals.DISABLED self.setColorScale(1.0, 0.0, 0.0, 1.0) else: self.partyEditor.trashCanButton['state'] = DirectGuiGlobals.NORMAL self.setColorScale(0.9, 0.9, 0.9, 0.7) def setOverValidSquare(self, value): self.overValidSquare = value if value: self.setColorScale(1.0, 1.0, 1.0, 1.0) else: self.setColorScale(0.9, 0.9, 0.9, 0.7) def removeFromGrid(self): if self.centerGridSquare is not None: self.partyEditor.partyEditorGrid.removeElement(self.centerGridSquare, self.getGridSize()) self.setOverValidSquare(False) self.lastValidPosition = None self.stash() return def snapToGrid(self, newPos): gridSquare = self.getGridSquareFromPosition(newPos) if gridSquare == None: self.setPosHprToDefault() self.setPos(render2d, newPos) return elif not self.partyEditor.partyEditorGrid.checkGridSquareForAvailability(gridSquare, self.getGridSize()): self.setPos(render2d, newPos) return self.setPosHprBasedOnGridSquare(gridSquare) return gridSquare def getGridSize(self): if self.isDecoration: return PartyGlobals.DecorationInformationDict[self.id]['gridsize'] else: return PartyGlobals.ActivityInformationDict[self.id]['gridsize'] def setPosHprToDefault(self): self.setR(0.0) self.uprightNodePath.setR(0.0) def setPosHprBasedOnGridSquare(self, gridSquare): gridPos = gridSquare.getPos() if self.getGridSize()[0] % 2 == 0: gridPos.setX(gridPos[0] + PartyGlobals.PartyEditorGridSquareSize[0] / 2.0) if self.getGridSize()[1] % 2 == 0: gridPos.setZ(gridPos[2] + PartyGlobals.PartyEditorGridSquareSize[1] / 2.0) if self.id != PartyGlobals.ActivityIds.PartyFireworks: if gridPos[0] > PartyGlobals.PartyEditorGridCenter[0] + PartyGlobals.PartyEditorGridRotateThreshold: self.setR(90.0) self.uprightNodePath.setR(-90.0) elif gridPos[0] < PartyGlobals.PartyEditorGridCenter[0] - PartyGlobals.PartyEditorGridRotateThreshold: self.setR(270.0) self.uprightNodePath.setR(-270.0) elif gridPos[2] < PartyGlobals.PartyEditorGridCenter[1] - PartyGlobals.PartyEditorGridRotateThreshold: self.setR(180.0) self.uprightNodePath.setR(-180.0) else: self.setR(0.0) self.uprightNodePath.setR(0.0) else: self.setR(270.0) self.uprightNodePath.setR(-270.0) self.setPos(render2d, gridPos) self.lastValidPosition = gridPos def getGridSquareFromPosition(self, newPos): localX = newPos[0] - PartyGlobals.PartyEditorGridBounds[0][0] localY = newPos[2] - PartyGlobals.PartyEditorGridBounds[1][1] x = int(localX / PartyGlobals.PartyEditorGridSquareSize[0]) y = int(localY / PartyGlobals.PartyEditorGridSquareSize[1]) y = PartyGlobals.PartyEditorGridSize[1] - 1 - y return self.partyEditor.partyEditorGrid.getGridSquare(x, y) def detach(self, mouseEvent): taskMgr.remove('gridElementDragTask%s' % self.name) self.rolloverTitle.stash() if self.overValidSquare: self.partyEditor.partyEditorGrid.registerNewElement(self, self.centerGridSquare, self.getGridSize()) self.partyEditor.updateCostsAndBank() self.partyEditor.handleMutuallyExclusiveActivities() elif self.lastValidPosition is not None: if self.mouseOverTrash: self.partyEditor.trashCanButton['state'] = DirectGuiGlobals.NORMAL self.lastValidPosition = None self.partyEditor.updateCostsAndBank() self.stash() else: self.setPos(render2d, self.lastValidPosition) self.setOverValidSquare(True) self.partyEditor.partyEditorGrid.registerNewElement(self, self.centerGridSquare, self.getGridSize()) self.partyEditor.updateCostsAndBank() self.partyEditor.handleMutuallyExclusiveActivities() else: self.stash() self.checkSoldOutAndPaidStatusAndAffordability() return def placeInPartyGrounds(self, desiredXY = None): self.centerGridSquare = self.partyEditor.partyEditorGrid.getClearGridSquare(self.getGridSize(), desiredXY) if self.centerGridSquare is not None: self.setOverValidSquare(True) self.unstash() self.setPosHprBasedOnGridSquare(self.centerGridSquare) self.partyEditor.partyEditorGrid.registerNewElement(self, self.centerGridSquare, self.getGridSize()) self.partyEditor.updateCostsAndBank() self.partyEditor.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsPartyGrounds self.checkSoldOutAndPaidStatusAndAffordability() return True else: return False return def clicked(self, mouseEvent): PartyEditorGridElement.notify.debug('clicked grid element %s' % self.name) if self.centerGridSquare is not None: self.attach(mouseEvent) self.partyEditor.partyEditorGrid.removeElement(self.centerGridSquare, self.getGridSize()) return def released(self, mouseEvent): PartyEditorGridElement.notify.debug('released grid element %s' % self.name) self.detach(mouseEvent) def mouseEnter(self, mouseEvent): parent = self.getParent() self.reparentTo(parent) self.rolloverTitle.unstash() def mouseExit(self, mouseEvent): self.rolloverTitle.stash() def destroy(self): self.unbind(DirectGuiGlobals.B1PRESS) self.unbind(DirectGuiGlobals.B1RELEASE) self.unbind(DirectGuiGlobals.ENTER) self.unbind(DirectGuiGlobals.EXIT) DirectButton.destroy(self)