Refactor

tools/editor/maptool.py

@@ -0,0 +1,143 @@
+import os
+from PyQt5.QtWidgets import QMainWindow, QWidget, QHBoxLayout, QMessageBox
+from PyQt5.QtCore import Qt
+from tools.editor.map.glwidget import GLWidget
+from tools.editor.map.mapleftpanel import MapLeftPanel
+from tools.editor.map.mapinfopanel import MapInfoPanel
+from tools.editor.map.menubar import MapMenubar
+from tools.editor.map.statusbar import StatusBar
+from tools.dusk.map import Map
+from tools.dusk.defs import CHUNK_WIDTH, CHUNK_HEIGHT, CHUNK_DEPTH, TILE_SHAPE_NULL, TILE_SHAPE_FLOOR
+from tools.editor.map.selectbox import SelectBox
+from tools.editor.map.camera import Camera
+from tools.editor.map.grid import Grid
+
+class MapWindow(QMainWindow):
+  def __init__(self):
+    super().__init__()
+
+    self.insertPressed = False
+    self.deletePressed = False
+
+    # Subclasses
+    self.map = Map(self)
+    self.camera = Camera(self)
+    self.grid = Grid()
+    self.selectBox = SelectBox(self)
+
+    # Window setup
+    self.setWindowTitle("Dusk Map Editor")
+    self.resize(1600, 900)
+
+    # Menubar (TESTING)
+    self.menubar = MapMenubar(self)
+
+    central = QWidget()
+    self.setCentralWidget(central)
+    mainLayout = QHBoxLayout(central)
+
+    # Left panel (tabs + nav buttons)
+    self.leftPanel = MapLeftPanel(self)
+    self.leftPanel.setFixedWidth(350)
+    mainLayout.addWidget(self.leftPanel)
+
+    # Center panel (GLWidget + controls)
+    self.glWidget = GLWidget(self)
+    mainLayout.addWidget(self.glWidget, stretch=3)
+
+    # Right panel (MapInfoPanel)
+    self.mapInfoPanel = MapInfoPanel(self)
+    rightWidget = self.mapInfoPanel
+    rightWidget.setFixedWidth(250)
+    mainLayout.addWidget(rightWidget)
+
+    # Status bar
+    self.statusBar = StatusBar(self)
+    self.setStatusBar(self.statusBar)
+
+    self.installEventFilter(self)
+    self.installEventFilterRecursively(self)
+
+  def installEventFilterRecursively(self, widget):
+    for child in widget.findChildren(QWidget):
+      child.installEventFilter(self)
+      self.installEventFilterRecursively(child)
+
+  def closeEvent(self, event):
+    if not self.map.isDirty():
+      event.accept()
+      return
+    
+    reply = QMessageBox.question(
+      self,
+      "Unsaved Changes",
+      "You have unsaved changes. Do you want to save before closing?",
+      QMessageBox.Save | QMessageBox.Discard | QMessageBox.Cancel,
+      QMessageBox.Save
+    )
+    if reply == QMessageBox.Save:
+      self.map.save()
+    elif reply == QMessageBox.Cancel:
+      event.ignore()
+      return
+    event.accept()
+
+  def eventFilter(self, obj, event):
+    if event.type() == event.KeyPress:
+      amtX, amtY, amtZ = 0, 0, 0
+      
+      key = event.key()
+      if key == Qt.Key_Left:
+        amtX = -1
+      elif key == Qt.Key_Right:
+        amtX = 1
+      elif key == Qt.Key_Up:
+        amtY = -1
+      elif key == Qt.Key_Down:
+        amtY = 1
+      elif key == Qt.Key_PageUp:
+        amtZ = 1
+      elif key == Qt.Key_PageDown:
+        amtZ = -1
+
+      if event.modifiers() & Qt.ShiftModifier:
+        amtX *= CHUNK_WIDTH
+        amtY *= CHUNK_HEIGHT
+        amtZ *= CHUNK_DEPTH
+
+      if amtX != 0 or amtY != 0 or amtZ != 0:
+        self.map.moveRelative(amtX, amtY, amtZ)
+        if self.insertPressed:
+          tile = self.map.getTileAtWorldPos(*self.map.position)
+          if tile is not None and tile.shape == TILE_SHAPE_NULL:
+            tile.setShape(TILE_SHAPE_FLOOR)
+        if self.deletePressed:
+          tile = self.map.getTileAtWorldPos(*self.map.position)
+          if tile is not None:
+            tile.setShape(TILE_SHAPE_NULL)
+        event.accept()
+        return True
+      
+      if key == Qt.Key_Delete:
+        tile = self.map.getTileAtWorldPos(*self.map.position)
+        self.deletePressed = True
+        if tile is not None:
+          tile.setShape(TILE_SHAPE_NULL)
+        event.accept()
+        return True
+      if key == Qt.Key_Insert:
+        tile = self.map.getTileAtWorldPos(*self.map.position)
+        self.insertPressed = True
+        if tile is not None and tile.shape == TILE_SHAPE_NULL:
+          tile.setShape(TILE_SHAPE_FLOOR)
+        event.accept()
+    elif event.type() == event.KeyRelease:
+      key = event.key()
+      if key == Qt.Key_Delete:
+        self.deletePressed = False
+        event.accept()
+        return True
+      if key == Qt.Key_Insert:
+        self.insertPressed = False
+        event.accept()
+    return super().eventFilter(obj, event)