103 lines
2.5 KiB
Python
103 lines
2.5 KiB
Python
import json
|
|
import os
|
|
from dusk.event import Event
|
|
from dusk.defs import CHUNK_WIDTH, CHUNK_HEIGHT, CHUNK_DEPTH, CHUNK_VERTEX_COUNT_MAX, TILE_SHAPE_NULL
|
|
from editortool.map.tile import Tile
|
|
from editortool.map.vertexbuffer import VertexBuffer
|
|
from OpenGL.GL import *
|
|
|
|
class Chunk:
|
|
def __init__(self, map, x, y, z):
|
|
self.map = map
|
|
self.x = x
|
|
self.y = y
|
|
self.z = z
|
|
self.current = {}
|
|
self.original = {}
|
|
self.onChunkData = Event()
|
|
self.dirty = False
|
|
|
|
self.tiles = {}
|
|
self.vertexBuffer = VertexBuffer()
|
|
|
|
tileIndex = 0
|
|
for tz in range(CHUNK_DEPTH):
|
|
for ty in range(CHUNK_HEIGHT):
|
|
for tx in range(CHUNK_WIDTH):
|
|
self.tiles[tileIndex] = Tile(self, tx, ty, tz, tileIndex)
|
|
tileIndex += 1
|
|
|
|
# Update vertices
|
|
self.tileUpdateVertices()
|
|
|
|
def tileUpdateVertices(self):
|
|
self.vertexBuffer.clear()
|
|
for tile in self.tiles.values():
|
|
tile.buffer(self.vertexBuffer)
|
|
self.vertexBuffer.buildData()
|
|
|
|
def load(self):
|
|
fname = self.getFilename()
|
|
if not fname or not os.path.exists(fname):
|
|
self.new()
|
|
return
|
|
try:
|
|
with open(fname, 'r') as f:
|
|
data = json.load(f)
|
|
|
|
if not 'shapes' in data:
|
|
data['shapes'] = []
|
|
|
|
# For each tile.
|
|
for tile in self.tiles.values():
|
|
tile.load(data)
|
|
|
|
self.tileUpdateVertices()
|
|
self.dirty = False
|
|
self.onChunkData.invoke(self)
|
|
except Exception as e:
|
|
raise RuntimeError(f"Failed to load chunk file: {e}")
|
|
|
|
def save(self):
|
|
if not self.isDirty():
|
|
return
|
|
|
|
dataOut = {
|
|
'shapes': []
|
|
}
|
|
|
|
for tile in self.tiles.values():
|
|
dataOut['shapes'].append(tile.shape)
|
|
|
|
fname = self.getFilename()
|
|
if not fname:
|
|
raise ValueError("No filename specified for saving chunk.")
|
|
try:
|
|
with open(fname, 'w') as f:
|
|
json.dump(dataOut, f)
|
|
self.dirty = False
|
|
self.onChunkData.invoke(self)
|
|
except Exception as e:
|
|
raise RuntimeError(f"Failed to save chunk file: {e}")
|
|
|
|
def new(self):
|
|
for tile in self.tiles.values():
|
|
tile.shape = TILE_SHAPE_NULL
|
|
|
|
self.tileUpdateVertices()
|
|
self.dirty = False
|
|
self.onChunkData.invoke(self)
|
|
|
|
def isDirty(self):
|
|
return self.dirty
|
|
|
|
def getFilename(self):
|
|
if not self.map or not hasattr(self.map, 'getMapDirectory'):
|
|
return None
|
|
dir_path = self.map.getMapDirectory()
|
|
if dir_path is None:
|
|
return None
|
|
return f"{dir_path}/{self.x}_{self.y}_{self.z}.json"
|
|
|
|
def draw(self):
|
|
self.vertexBuffer.draw() |