class PartyEditor(DirectObject, FSM):
"""
This class creates the grid and scrolled list needed for players to
drag and drop activities and decorations onto their party grounds.
"""
notify = directNotify.newCategory("PartyEditor")
def __init__(self, partyPlanner, parent):
FSM.__init__(self, self.__class__.__name__)
self.partyPlanner = partyPlanner
self.parent = parent
self.partyEditorGrid = PartyEditorGrid(self)
self.currentElement = None
self.defaultTransitions = {
"Hidden": ["Idle", "Cleanup"],
"Idle": ["DraggingElement", "Hidden", "Cleanup"],
"DraggingElement":
["Idle", "DraggingElement", "Hidden", "Cleanup"],
"Cleanup": [],
}
self.initElementList()
self.initPartyClock()
self.initTrashCan()
def initElementList(self):
self.activityIconsModel = loader.loadModel(
"phase_4/models/parties/eventSignIcons")
self.decorationModels = loader.loadModel(
"phase_4/models/parties/partyDecorations")
pos = self.partyPlanner.gui.find(
"**/step_05_activitiesIcon_locator").getPos()
self.elementList = DirectScrolledList(
parent=self.parent,
relief=None,
# inc and dec are DirectButtons
decButton_image=(
self.partyPlanner.gui.find("**/activitiesButtonUp_up"),
self.partyPlanner.gui.find("**/activitiesButtonUp_down"),
self.partyPlanner.gui.find("**/activitiesButtonUp_rollover"),
self.partyPlanner.gui.find("**/activitiesButtonUp_inactive"),
),
decButton_relief=None,
decButton_pos=(-0.05, 0.0, -0.38),
incButton_image=(
self.partyPlanner.gui.find("**/activitiesButtonDown_up"),
self.partyPlanner.gui.find("**/activitiesButtonDown_down"),
self.partyPlanner.gui.find("**/activitiesButtonDown_rollover"),
self.partyPlanner.gui.find("**/activitiesButtonDown_inactive"),
),
incButton_relief=None,
incButton_pos=(-0.05, 0.0, -0.94),
# itemFrame is a DirectFrame
itemFrame_pos=(pos[0], pos[1], pos[2] + 0.04),
itemFrame_relief=None,
# each item is a button with text on it
numItemsVisible=1,
items=[],
)
for activityId in PartyGlobals.PartyEditorActivityOrder:
if activityId in PartyGlobals.VictoryPartyActivityIds:
holidayIds = base.cr.newsManager.getHolidayIdList()
if ToontownGlobals.VICTORY_PARTY_HOLIDAY in holidayIds:
pele = PartyEditorListElement(self, activityId)
self.elementList.addItem(pele)
elif activityId in PartyGlobals.VictoryPartyReplacementActivityIds:
holidayIds = base.cr.newsManager.getHolidayIdList()
if not ToontownGlobals.VICTORY_PARTY_HOLIDAY in holidayIds:
pele = PartyEditorListElement(self, activityId)
self.elementList.addItem(pele)
else:
pele = PartyEditorListElement(self, activityId)
self.elementList.addItem(pele)
if activityId == PartyGlobals.ActivityIds.PartyClock:
self.partyClockElement = pele
for decorationId in PartyGlobals.DecorationIds:
decorName = PartyGlobals.DecorationIds.getString(decorationId)
if (decorName == "HeartTarget") \
or (decorName == "HeartBanner") \
or (decorName == "FlyingHeart"):
holidayIds = base.cr.newsManager.getHolidayIdList()
if ToontownGlobals.VALENTINES_DAY in holidayIds:
pele = PartyEditorListElement(self,
decorationId,
isDecoration=True)
self.elementList.addItem(pele)
elif decorationId in PartyGlobals.VictoryPartyDecorationIds:
holidayIds = base.cr.newsManager.getHolidayIdList()
if ToontownGlobals.VICTORY_PARTY_HOLIDAY in holidayIds:
pele = PartyEditorListElement(self,
decorationId,
isDecoration=True)
self.elementList.addItem(pele)
elif decorationId in PartyGlobals.VictoryPartyReplacementDecorationIds:
holidayIds = base.cr.newsManager.getHolidayIdList()
if not ToontownGlobals.VICTORY_PARTY_HOLIDAY in holidayIds:
pele = PartyEditorListElement(self,
decorationId,
isDecoration=True)
self.elementList.addItem(pele)
else:
pele = PartyEditorListElement(self,
decorationId,
isDecoration=True)
self.elementList.addItem(pele)
self.elementList.refresh()
self.elementList['command'] = self.scrollItemChanged
def initPartyClock(self):
self.partyClockElement.buyButtonClicked((8, 7))
def initTrashCan(self):
trashcanGui = loader.loadModel("phase_3/models/gui/trashcan_gui")
self.trashCanButton = DirectButton(
parent=self.parent,
relief=None,
pos=Point3(*PartyGlobals.TrashCanPosition),
scale=PartyGlobals.TrashCanScale,
geom=(
trashcanGui.find("**/TrashCan_CLSD"),
trashcanGui.find("**/TrashCan_OPEN"),
trashcanGui.find("**/TrashCan_RLVR"),
trashcanGui.find("**/TrashCan_RLVR"),
),
command=self.trashCanClicked,
)
self.trashCanButton.bind(DirectGuiGlobals.ENTER, self.mouseEnterTrash)
self.trashCanButton.bind(DirectGuiGlobals.EXIT, self.mouseExitTrash)
self.mouseOverTrash = False
self.oldInstructionText = ""
self.trashCanLastClickedTime = 0
def scrollItemChanged(self):
if not self.elementList["items"]:
# we are probably closing the gui, do nothing
return
self.currentElement = self.elementList["items"][
self.elementList.getSelectedIndex()]
self.elementList["items"][
self.elementList.getSelectedIndex()].elementSelectedFromList()
if self.elementList["items"][
self.elementList.getSelectedIndex()].isDecoration:
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsClickedElementDecoration
else:
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsClickedElementActivity
def listElementClicked(self):
self.request("DraggingElement")
def listElementReleased(self):
self.request("Idle", True)
def trashCanClicked(self):
currentTime = time.time()
# Check for double click, if so, clear the party grounds
if currentTime - self.trashCanLastClickedTime < 0.2:
self.clearPartyGrounds()
self.trashCanLastClickedTime = time.time()
def clearPartyGrounds(self):
for item in self.elementList["items"]:
item.clearPartyGrounds()
self.initPartyClock()
if self.currentElement:
self.currentElement.checkSoldOutAndPaidStatusAndAffordability()
def buyCurrentElement(self):
if self.currentElement:
purchaseSuccessful = self.currentElement.buyButtonClicked()
if purchaseSuccessful:
# The buying and placement of the item was successful
self.handleMutuallyExclusiveActivities()
pass
else:
# The buying and placement of the item was not successful
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsNoRoom
def mouseEnterTrash(self, mouseEvent):
self.mouseOverTrash = True
self.oldInstructionText = self.partyPlanner.instructionLabel["text"]
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsTrash
def mouseExitTrash(self, mouseEvent):
self.mouseOverTrash = False
self.partyPlanner.instructionLabel["text"] = self.oldInstructionText
### FSM Methods ###
def enterHidden(self):
PartyEditor.notify.debug("Enter Hidden")
def exitHidden(self):
PartyEditor.notify.debug("Exit Hidden")
def enterIdle(self, fromDragging=False):
PartyEditor.notify.debug("Enter Idle")
if not fromDragging:
self.elementList.scrollTo(0)
self.elementList["items"][0].elementSelectedFromList()
self.currentElement = self.elementList["items"][
self.elementList.getSelectedIndex()]
self.currentElement.checkSoldOutAndPaidStatusAndAffordability()
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsIdle
self.updateCostsAndBank()
self.handleMutuallyExclusiveActivities()
def handleMutuallyExclusiveActivities(self):
"""Smartly removed the older activity and inform the user."""
mutSet = self.getMutuallyExclusiveActivities()
if not mutSet:
return
# our mutset doesn't tell us which one is older
currentActivities = self.partyEditorGrid.getActivitiesElementsOnGrid()
lastActivity = self.partyEditorGrid.lastActivityIdPlaced
for act in currentActivities:
if (act.id in mutSet) and not (lastActivity == act.id):
act.removeFromGrid()
removedName = TTLocalizer.PartyActivityNameDict[
act.id]["editor"]
addedName = TTLocalizer.PartyActivityNameDict[lastActivity][
"editor"]
instr = TTLocalizer.PartyPlannerEditorInstructionsRemoved % \
{"removed" : removedName, "added" : addedName}
self.partyPlanner.instructionLabel["text"] = instr
self.updateCostsAndBank()
# deliberately no break here, in case they manage to
# get 3 jukeboxes into the editor somehow
def getMutuallyExclusiveActivities(self):
"""Return the set of activities on the grid that are mutually exclusive, None otherwise."""
# create a set of activity Ids
currentActivities = self.partyEditorGrid.getActivitiesOnGrid()
actSet = Set([])
for act in currentActivities:
actSet.add(act[0])
result = None
for mutuallyExclusiveTuples in PartyGlobals.MutuallyExclusiveActivities:
mutSet = Set(mutuallyExclusiveTuples)
inter = mutSet.intersection(actSet)
if len(inter) > 1:
result = inter
break
return result
def updateCostsAndBank(self):
"""
We need to update the total cost of the party and what they will have
left in their bank.
"""
currentActivities = self.partyEditorGrid.getActivitiesOnGrid()
currentDecorations = self.partyEditorGrid.getDecorationsOnGrid()
newCost = 0
for elementTuple in currentActivities:
newCost += PartyGlobals.ActivityInformationDict[
elementTuple[0]]["cost"]
for elementTuple in currentDecorations:
newCost += PartyGlobals.DecorationInformationDict[
elementTuple[0]]["cost"]
self.partyPlanner.costLabel[
"text"] = TTLocalizer.PartyPlannerTotalCost % newCost
if len(currentActivities) > 0 or len(currentDecorations) > 0:
self.partyPlanner.setNextButtonState(enabled=True)
else:
self.partyPlanner.setNextButtonState(enabled=False)
self.partyPlanner.totalCost = newCost
self.partyPlanner.beanBank["text"] = str(
int(self.partyPlanner.totalMoney - self.partyPlanner.totalCost))
def exitIdle(self):
PartyEditor.notify.debug("Exit Idle")
def enterDraggingElement(self):
PartyEditor.notify.debug("Enter DraggingElement")
if self.currentElement.isDecoration:
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsDraggingDecoration
else:
self.partyPlanner.instructionLabel[
"text"] = TTLocalizer.PartyPlannerEditorInstructionsDraggingActivity
def exitDraggingElement(self):
PartyEditor.notify.debug("Exit DraggingElement")
def enterCleanup(self):
PartyEditor.notify.debug("Enter Cleanup")
self.partyEditorGrid.destroy()
self.elementList.removeAndDestroyAllItems()
self.elementList.destroy()
self.trashCanButton.unbind(DirectGuiGlobals.ENTER)
self.trashCanButton.unbind(DirectGuiGlobals.EXIT)
self.trashCanButton.destroy()
def exitCleanup(self):
PartyEditor.notify.debug("Exit Cleanup")
class PartyEditor(DirectObject, FSM):
notify = directNotify.newCategory('PartyEditor')
def __init__(self, partyPlanner, parent):
FSM.__init__(self, self.__class__.__name__)
self.partyPlanner = partyPlanner
self.parent = parent
self.partyEditorGrid = PartyEditorGrid(self)
self.currentElement = None
self.defaultTransitions = {'Hidden': ['Idle', 'Cleanup'],
'Idle': ['DraggingElement', 'Hidden', 'Cleanup'],
'DraggingElement': ['Idle',
'DraggingElement',
'Hidden',
'Cleanup'],
'Cleanup': []}
self.initElementList()
self.initPartyClock()
self.initTrashCan()
return
def initElementList(self):
self.activityIconsModel = loader.loadModel('phase_4/models/parties/eventSignIcons')
self.decorationModels = loader.loadModel('phase_4/models/parties/partyDecorations')
pos = self.partyPlanner.gui.find('**/step_05_activitiesIcon_locator').getPos()
self.elementList = DirectScrolledList(parent=self.parent, relief=None, decButton_image=(self.partyPlanner.gui.find('**/activitiesButtonUp_up'),
self.partyPlanner.gui.find('**/activitiesButtonUp_down'),
self.partyPlanner.gui.find('**/activitiesButtonUp_rollover'),
self.partyPlanner.gui.find('**/activitiesButtonUp_inactive')), decButton_relief=None, decButton_pos=(-0.05, 0.0, -0.38), incButton_image=(self.partyPlanner.gui.find('**/activitiesButtonDown_up'),
self.partyPlanner.gui.find('**/activitiesButtonDown_down'),
self.partyPlanner.gui.find('**/activitiesButtonDown_rollover'),
self.partyPlanner.gui.find('**/activitiesButtonDown_inactive')), incButton_relief=None, incButton_pos=(-0.05, 0.0, -0.94), itemFrame_pos=(pos[0], pos[1], pos[2] + 0.04), itemFrame_relief=None, numItemsVisible=1, items=[])
holidayIds = base.cr.newsManager.getHolidayIdList()
isWinter = ToontownGlobals.WINTER_DECORATIONS in holidayIds or ToontownGlobals.WACKY_WINTER_DECORATIONS in holidayIds
isVictory = ToontownGlobals.VICTORY_PARTY_HOLIDAY in holidayIds
isValentine = ToontownGlobals.VALENTINES_DAY in holidayIds
for activityId in PartyGlobals.PartyEditorActivityOrder:
if not isVictory and activityId in PartyGlobals.VictoryPartyActivityIds or not isWinter and activityId in PartyGlobals.WinterPartyActivityIds or not isValentine and activityId in PartyGlobals.ValentinePartyActivityIds:
pass
elif isVictory and activityId in PartyGlobals.VictoryPartyReplacementActivityIds or isWinter and activityId in PartyGlobals.WinterPartyReplacementActivityIds or isValentine and activityId in PartyGlobals.ValentinePartyReplacementActivityIds:
pass
else:
pele = PartyEditorListElement(self, activityId)
self.elementList.addItem(pele)
if activityId == PartyGlobals.ActivityIds.PartyClock:
self.partyClockElement = pele
for decorationId in PartyGlobals.DecorationIds:
if not isVictory and decorationId in PartyGlobals.VictoryPartyDecorationIds or not isWinter and decorationId in PartyGlobals.WinterPartyDecorationIds or not isValentine and decorationId in PartyGlobals.ValentinePartyDecorationIds:
pass
elif isVictory and decorationId in PartyGlobals.VictoryPartyReplacementDecorationIds or isValentine and decorationId in PartyGlobals.ValentinePartyReplacementDecorationIds:
pass
elif decorationId in PartyGlobals.TTIUnreleasedDecor:
pass
else:
pele = PartyEditorListElement(self, decorationId, isDecoration=True)
self.elementList.addItem(pele)
self.elementList.refresh()
self.elementList['command'] = self.scrollItemChanged
return
def initPartyClock(self):
self.partyClockElement.buyButtonClicked((8, 7))
def initTrashCan(self):
trashcanGui = loader.loadModel('phase_3/models/gui/trashcan_gui')
self.trashCanButton = DirectButton(parent=self.parent, relief=None, pos=Point3(*PartyGlobals.TrashCanPosition), scale=PartyGlobals.TrashCanScale, geom=(trashcanGui.find('**/TrashCan_CLSD'),
trashcanGui.find('**/TrashCan_OPEN'),
trashcanGui.find('**/TrashCan_RLVR'),
trashcanGui.find('**/TrashCan_RLVR')), command=self.trashCanClicked)
self.trashCanButton.bind(DirectGuiGlobals.ENTER, self.mouseEnterTrash)
self.trashCanButton.bind(DirectGuiGlobals.EXIT, self.mouseExitTrash)
self.mouseOverTrash = False
self.oldInstructionText = ''
self.trashCanLastClickedTime = 0
return
def scrollItemChanged(self):
if not self.elementList['items']:
return
self.currentElement = self.elementList['items'][self.elementList.getSelectedIndex()]
self.elementList['items'][self.elementList.getSelectedIndex()].elementSelectedFromList()
if self.elementList['items'][self.elementList.getSelectedIndex()].isDecoration:
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsClickedElementDecoration
else:
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsClickedElementActivity
def listElementClicked(self):
self.request('DraggingElement')
def listElementReleased(self):
self.request('Idle', True)
def trashCanClicked(self):
currentTime = time.time()
if currentTime - self.trashCanLastClickedTime < 0.2:
self.clearPartyGrounds()
self.trashCanLastClickedTime = time.time()
def clearPartyGrounds(self):
for item in self.elementList['items']:
item.clearPartyGrounds()
self.initPartyClock()
if self.currentElement:
self.currentElement.checkSoldOutAndPaidStatusAndAffordability()
def buyCurrentElement(self):
if self.currentElement:
purchaseSuccessful = self.currentElement.buyButtonClicked()
if purchaseSuccessful:
self.handleMutuallyExclusiveActivities()
else:
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsNoRoom
def mouseEnterTrash(self, mouseEvent):
self.mouseOverTrash = True
self.oldInstructionText = self.partyPlanner.instructionLabel['text']
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsTrash
def mouseExitTrash(self, mouseEvent):
self.mouseOverTrash = False
self.partyPlanner.instructionLabel['text'] = self.oldInstructionText
def enterHidden(self):
PartyEditor.notify.debug('Enter Hidden')
def exitHidden(self):
PartyEditor.notify.debug('Exit Hidden')
def enterIdle(self, fromDragging = False):
PartyEditor.notify.debug('Enter Idle')
if not fromDragging:
self.elementList.scrollTo(0)
self.elementList['items'][0].elementSelectedFromList()
self.currentElement = self.elementList['items'][self.elementList.getSelectedIndex()]
self.currentElement.checkSoldOutAndPaidStatusAndAffordability()
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsIdle
self.updateCostsAndBank()
self.handleMutuallyExclusiveActivities()
def handleMutuallyExclusiveActivities(self):
mutSet = self.getMutuallyExclusiveActivities()
if not mutSet:
return
currentActivities = self.partyEditorGrid.getActivitiesElementsOnGrid()
lastActivity = self.partyEditorGrid.lastActivityIdPlaced
for act in currentActivities:
if act.id in mutSet and not lastActivity == act.id:
act.removeFromGrid()
removedName = TTLocalizer.PartyActivityNameDict[act.id]['editor']
addedName = TTLocalizer.PartyActivityNameDict[lastActivity]['editor']
instr = TTLocalizer.PartyPlannerEditorInstructionsRemoved % {'removed': removedName,
'added': addedName}
self.partyPlanner.instructionLabel['text'] = instr
self.updateCostsAndBank()
def getMutuallyExclusiveActivities(self):
currentActivities = self.partyEditorGrid.getActivitiesOnGrid()
actSet = Set([])
for act in currentActivities:
actSet.add(act[0])
result = None
for mutuallyExclusiveTuples in PartyGlobals.MutuallyExclusiveActivities:
mutSet = Set(mutuallyExclusiveTuples)
inter = mutSet.intersection(actSet)
if len(inter) > 1:
result = inter
break
return result
def updateCostsAndBank(self):
currentActivities = self.partyEditorGrid.getActivitiesOnGrid()
currentDecorations = self.partyEditorGrid.getDecorationsOnGrid()
newCost = 0
for elementTuple in currentActivities:
newCost += PartyGlobals.ActivityInformationDict[elementTuple[0]]['cost']
for elementTuple in currentDecorations:
newCost += PartyGlobals.DecorationInformationDict[elementTuple[0]]['cost']
self.partyPlanner.costLabel['text'] = TTLocalizer.PartyPlannerTotalCost % newCost
if len(currentActivities) > 0 or len(currentDecorations) > 0:
self.partyPlanner.setNextButtonState(enabled=True)
else:
self.partyPlanner.setNextButtonState(enabled=False)
self.partyPlanner.totalCost = newCost
self.partyPlanner.beanBank['text'] = str(int(self.partyPlanner.totalMoney - self.partyPlanner.totalCost))
def exitIdle(self):
PartyEditor.notify.debug('Exit Idle')
def enterDraggingElement(self):
PartyEditor.notify.debug('Enter DraggingElement')
if self.currentElement.isDecoration:
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsDraggingDecoration
else:
self.partyPlanner.instructionLabel['text'] = TTLocalizer.PartyPlannerEditorInstructionsDraggingActivity
def exitDraggingElement(self):
PartyEditor.notify.debug('Exit DraggingElement')
def enterCleanup(self):
PartyEditor.notify.debug('Enter Cleanup')
self.partyEditorGrid.destroy()
self.elementList.removeAndDestroyAllItems()
self.elementList.destroy()
self.trashCanButton.unbind(DirectGuiGlobals.ENTER)
self.trashCanButton.unbind(DirectGuiGlobals.EXIT)
self.trashCanButton.destroy()
def exitCleanup(self):
PartyEditor.notify.debug('Exit Cleanup')