bldg title labels

Author: drewc5131

toontown/leveleditor/LevelEditor.py

@@ -130,6 +130,7 @@ def __init__(self):
         self.visitedEdges = []
 
         self.zoneLabels = []
+        self.bldgLabels = []
         self.animPropDict = {}
 
         self.collisionsToggled = False
@@ -3308,6 +3309,35 @@ def clearZoneLabels(self):
         for label in self.zoneLabels:
             label.removeNode()
         self.zoneLabels = []
+        
+    def labelBldgs(self):
+        ''' Draws a text label above Landmark bldgs displaying their title and block # '''
+        self.clearBldgLabels()
+        for bldg in self.NPToplevel.findAllMatches('**/*tb*:toon_landmark*'):
+            dnanode = self.findDNANode(bldg)
+            block = self.getBlockFromName(dnanode.getName())
+            title = dnanode.getTitle()
+            fg = (0.8, 0.4, 0.2, 1)
+            if not title: 
+                title = '(unnamed)'
+                fg = (1.0, 0.1, 0.1, 1)
+            label = DirectGui.DirectLabel(text = f"TB{block}\n{title}",
+                                          text_font = ToontownGlobals.getSignFont(),
+                                          text_fg = fg, text_wordwrap = 20,
+                                          parent = bldg,
+                                          relief = None, scale = 4)
+            label.setBillboardPointWorld()
+            label.setDepthWrite(False)
+            label.setDepthTest(not self.labelsOnTop)
+            if not bldg.find('**/**front').getBounds().isEmpty():
+                center = bldg.find('**/**front').getBounds().getCenter()
+                label.setPos(center[0], center[1], 50)
+            self.bldgLabels.append(label)
+            
+    def clearBldgLabels(self):
+        for label in self.bldgLabels:
+            label.removeNode()
+        self.bldgLabels = []
 
     def getBlockFromName(self, name):
         block = name[2:name.find(':')]

toontown/leveleditor/LevelEditorPanel.py

@@ -375,6 +375,15 @@ def __init__(self, levelEditor, parent = None, **kw):
                     landmarkBuildingsPage, width = 24,
                     textvariable = self.landmarkBuildingNameString)
             self.landmarkBuildingNameBox.pack(expand = 0, fill = X)
+            
+        self.bldgLabels = IntVar()
+        self.bldgLabels.set(0)
+        self.bldgLabelsButton = ttk.Checkbutton(
+                landmarkBuildingsPage,
+                text = 'Show Bldg Labels', width = 20,
+                variable = self.bldgLabels,
+                command = self.toggleBldgLabels)
+        self.bldgLabelsButton.pack(side = LEFT, expand = 1, fill = X)
 
         # ANIMATED BUILDINGS
         Label(animBuildingsPage, text = 'Animated Buildings',
@@ -988,7 +997,7 @@ def toggleZoneLabels(self):
 
     def toggleZoneLabelsOnTop(self):
         self.levelEditor.labelsOnTop = self.fLabelOnTop.get()
-        for lbl in self.levelEditor.zoneLabels:
+        for lbl in self.levelEditor.zoneLabels + self.levelEditor.bldgLabels:
             lbl.setDepthTest(not self.fLabelOnTop.get())
         for lbl in self.levelEditor.NPToplevel.findAllMatches('**/suit_point_label_*'):
             lbl.setDepthTest(not self.fLabelOnTop.get())
@@ -1001,6 +1010,12 @@ def togglePathLabels(self):
             for lbl in self.levelEditor.NPToplevel.findAllMatches('**/suit_point_label_*'):
                 lbl.hide()
 
+    def toggleBldgLabels(self):
+        if self.bldgLabels.get():
+            self.levelEditor.labelBldgs()
+        else:
+            self.levelEditor.clearBldgLabels()
+
     def toggleXyzSnap(self):
         base.direct.grid.setXyzSnap(self.fXyzSnap.get())