143 lines
4.2 KiB
Python
143 lines
4.2 KiB
Python
import os
|
|
from PyQt5.QtWidgets import QMainWindow, QWidget, QHBoxLayout, QMessageBox
|
|
from PyQt5.QtCore import Qt
|
|
from editortool.map.glwidget import GLWidget
|
|
from editortool.map.chunkpanel import ChunkPanel
|
|
from editortool.map.mapinfopanel import MapInfoPanel
|
|
from editortool.map.menubar import MapMenubar
|
|
from editortool.map.statusbar import StatusBar
|
|
from editortool.map.map import Map
|
|
from dusk.defs import CHUNK_WIDTH, CHUNK_HEIGHT, CHUNK_DEPTH, TILE_SHAPE_NULL, TILE_SHAPE_FLOOR
|
|
from editortool.map.selectbox import SelectBox
|
|
from editortool.map.camera import Camera
|
|
from editortool.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(1280, 720)
|
|
|
|
# Menubar (TESTING)
|
|
self.menubar = MapMenubar(self)
|
|
|
|
central = QWidget()
|
|
self.setCentralWidget(central)
|
|
mainLayout = QHBoxLayout(central)
|
|
|
|
# Left panel (ChunkPanel)
|
|
self.chunkPanel = ChunkPanel(self)
|
|
self.chunkPanel.setFixedWidth(200)
|
|
mainLayout.addWidget(self.chunkPanel)
|
|
|
|
# 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) |