106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QLabel, QPushButton, QGridLayout, QTreeWidget, QTreeWidgetItem, QComboBox
|
|
from dusk.defs import CHUNK_WIDTH, CHUNK_HEIGHT, CHUNK_DEPTH, TILE_SHAPES
|
|
|
|
class ChunkPanel(QWidget):
|
|
def __init__(self, parent):
|
|
super().__init__(parent)
|
|
self.parent = parent
|
|
layout = QVBoxLayout(self)
|
|
self.chunkInfoLabel = QLabel("Tile Information")
|
|
layout.addWidget(self.chunkInfoLabel)
|
|
|
|
# Nav buttons
|
|
grid = QGridLayout()
|
|
self.btnUp = QPushButton("U")
|
|
self.btnN = QPushButton("N")
|
|
self.btnDown = QPushButton("D")
|
|
self.btnW = QPushButton("W")
|
|
self.btnS = QPushButton("S")
|
|
self.btnE = QPushButton("E")
|
|
|
|
grid.addWidget(self.btnUp, 0, 0)
|
|
grid.addWidget(self.btnN, 0, 1)
|
|
grid.addWidget(self.btnDown, 0, 2)
|
|
grid.addWidget(self.btnW, 1, 0)
|
|
grid.addWidget(self.btnS, 1, 1)
|
|
grid.addWidget(self.btnE, 1, 2)
|
|
layout.addLayout(grid)
|
|
|
|
# Tile shape dropdown
|
|
self.tileShapeDropdown = QComboBox()
|
|
self.tileShapeDropdown.addItems(TILE_SHAPES.keys())
|
|
self.tileShapeDropdown.setToolTip("Tile Shape")
|
|
layout.addWidget(self.tileShapeDropdown)
|
|
|
|
# Add expandable tree list
|
|
self.tree = QTreeWidget()
|
|
self.tree.setHeaderLabel("Chunks")
|
|
self.tree.expandAll() # Expand by default, remove if you want collapsed
|
|
layout.addWidget(self.tree) # Removed invalid stretch factor
|
|
|
|
# Add stretch so tree expands
|
|
layout.setStretchFactor(grid, 0)
|
|
layout.setStretchFactor(self.tree, 1)
|
|
|
|
# Event subscriptions
|
|
self.btnN.clicked.connect(lambda: self.parent.map.moveRelative(0, -1, 0))
|
|
self.btnS.clicked.connect(lambda: self.parent.map.moveRelative(0, 1, 0))
|
|
self.btnE.clicked.connect(lambda: self.parent.map.moveRelative(1, 0, 0))
|
|
self.btnW.clicked.connect(lambda: self.parent.map.moveRelative(-1, 0, 0))
|
|
self.btnUp.clicked.connect(lambda: self.parent.map.moveRelative(0, 0, 1))
|
|
self.btnDown.clicked.connect(lambda: self.parent.map.moveRelative(0, 0, -1))
|
|
self.parent.map.onMapData.sub(self.onMapData)
|
|
self.parent.map.onPositionChange.sub(self.onPositionChange)
|
|
self.tileShapeDropdown.currentTextChanged.connect(self.onTileShapeChanged)
|
|
|
|
# For each chunk
|
|
for chunk in self.parent.map.chunks.values():
|
|
# Create tree element
|
|
item = QTreeWidgetItem(self.tree, ["Chunk ({}, {}, {})".format(chunk.x, chunk.y, chunk.z)])
|
|
chunk.chunkPanelTree = item
|
|
chunk.chunkPanelTree.setExpanded(True)
|
|
item.setData(0, 0, chunk) # Store chunk reference
|
|
|
|
chunk.onChunkData.sub(self.onChunkData)
|
|
|
|
def onMapData(self, data):
|
|
pass
|
|
|
|
def onPositionChange(self, pos):
|
|
self.updateChunkList()
|
|
|
|
tile = self.parent.map.getTileAtWorldPos(*self.parent.map.position)
|
|
if tile is None:
|
|
return
|
|
|
|
key = "TILE_SHAPE_NULL"
|
|
for k, v in TILE_SHAPES.items():
|
|
if v != tile.shape:
|
|
continue
|
|
key = k
|
|
break
|
|
self.tileShapeDropdown.setCurrentText(key)
|
|
|
|
def onTileShapeChanged(self, shape_key):
|
|
tile = self.parent.map.getTileAtWorldPos(*self.parent.map.position)
|
|
|
|
if tile is None or shape_key not in TILE_SHAPES:
|
|
return
|
|
tile.setShape(TILE_SHAPES[shape_key])
|
|
|
|
def updateChunkList(self):
|
|
# Clear existing items
|
|
currentChunk = self.parent.map.getChunkAtWorldPos(*self.parent.map.position)
|
|
|
|
# Example tree items
|
|
for chunk in self.parent.map.chunks.values():
|
|
title = "Chunk ({}, {}, {})".format(chunk.x, chunk.y, chunk.z)
|
|
if chunk == currentChunk:
|
|
title += " [C]"
|
|
if chunk.isDirty():
|
|
title += " *"
|
|
item = chunk.chunkPanelTree
|
|
item.setText(0, title)
|
|
|
|
def onChunkData(self, chunk):
|
|
self.updateChunkList() |