133 lines
5.0 KiB
Python
133 lines
5.0 KiB
Python
import pygame as pg
|
|
from game.util import load_image, load_images
|
|
from game.tilemap import Tilemap
|
|
|
|
RENDER_SCALE = 2.0
|
|
WINDOW_SIZE=(640, 480)
|
|
|
|
class Editor():
|
|
def __init__(self):
|
|
self.screen = pg.display.set_mode(WINDOW_SIZE)
|
|
self.clock = pg.time.Clock()
|
|
|
|
self.width, self.height = pg.display.get_surface().get_size()
|
|
self.tilemap = Tilemap(self)
|
|
self.display = pg.Surface((self.width/4*3, self.height/6*5))
|
|
|
|
self.map = 'test'
|
|
self.path = 'assets/maps/' + str(self.map) + '.json'
|
|
|
|
try:
|
|
self.tilemap.load(self.path)
|
|
except FileNotFoundError:
|
|
print('File not Found')
|
|
|
|
self.scroll = [0, 0]
|
|
|
|
self.assets = {
|
|
"base": load_images("/tiles/test")
|
|
}
|
|
|
|
self.tile_list = list(self.assets)
|
|
|
|
self.tile_group = 0
|
|
self.tile_variant = 0
|
|
|
|
self.movment = [False, False, False, False]
|
|
|
|
self.clicking = False
|
|
self.right_clicking = False
|
|
self.shift = False
|
|
self.ongrid = True
|
|
|
|
def run(self):
|
|
self.running = True
|
|
while self.running:
|
|
self.clock.tick(60)
|
|
self.update()
|
|
self.draw()
|
|
return self.running
|
|
|
|
def update(self):
|
|
self.mpos = pg.mouse.get_pos()
|
|
self.mpos = (self.mpos[0] / RENDER_SCALE, self.mpos[1] / RENDER_SCALE)
|
|
self.tile_pos = (int((self.mpos[0] + self.scroll[0]) // self.tilemap.tile_size), int((self.mpos[1] + self.scroll[1]) // self.tilemap.tile_size))
|
|
|
|
self.scroll[0] += (self.movment[1] - self.movment[0]) * 2
|
|
self.scroll[1] += (self.movment[3] - self.movment[2]) * 2
|
|
|
|
if self.clicking and self.ongrid:
|
|
self.tilemap.tilemap[str(self.tile_pos[0]) + ';' + str(self.tile_pos[1])] = {'type': self.tile_list[self.tile_group], 'variant': self.tile_variant, 'pos': self.tile_pos}
|
|
if self.right_clicking:
|
|
tile_loc = str(self.tile_pos[0]) + ';' + str(self.tile_pos[1])
|
|
if tile_loc in self.tilemap.tilemap:
|
|
del self.tilemap.tilemap[tile_loc]
|
|
|
|
self.current_tile_img = self.assets[self.tile_list[self.tile_group]][self.tile_variant].copy()
|
|
|
|
for event in pg.event.get():
|
|
if event.type == pg.QUIT:
|
|
self.running = False
|
|
if event.type == pg.MOUSEBUTTONDOWN:
|
|
if event.button == 1:
|
|
self.clicking = True
|
|
if event.button == 3:
|
|
self.right_clicking = True
|
|
if self.shift:
|
|
if event.button == 4:
|
|
self.tile_variant = (self.tile_variant - 1) % len(self.assets[self.tile_list[self.tile_group]])
|
|
if event.button == 5:
|
|
self.tile_variant = (self.tile_variant + 1) % len(self.assets[self.tile_list[self.tile_group]])
|
|
else:
|
|
if event.button == 4:
|
|
self.tile_group = (self.tile_group - 1) % len(self.tile_list)
|
|
self.tile_variant = 0
|
|
if event.button == 5:
|
|
self.tile_group = (self.tile_group + 1) % len(self.tile_list)
|
|
self.tile_variant = 0
|
|
if event.type == pg.MOUSEBUTTONUP:
|
|
if event.button == 1:
|
|
self.clicking = False
|
|
if event.button == 3:
|
|
self.right_clicking = False
|
|
|
|
if event.type == pg.KEYDOWN:
|
|
if event.key == pg.K_a:
|
|
self.movment[0] = True
|
|
if event.key == pg.K_d:
|
|
self.movment[1] = True
|
|
if event.key == pg.K_w:
|
|
self.movment[2] = True
|
|
if event.key == pg.K_s:
|
|
self.movment[3] = True
|
|
if event.key == pg.K_LSHIFT:
|
|
self.shift = True
|
|
if event.key == pg.K_o:
|
|
self.tilemap.save(self.path)
|
|
if event.type == pg.KEYUP:
|
|
if event.key == pg.K_a:
|
|
self.movment[0] = False
|
|
if event.key == pg.K_d:
|
|
self.movment[1] = False
|
|
if event.key == pg.K_w:
|
|
self.movment[2] = False
|
|
if event.key == pg.K_s:
|
|
self.movment[3] = False
|
|
if event.key == pg.K_LSHIFT:
|
|
self.shift = False
|
|
|
|
def draw(self):
|
|
self.display.fill((0, 0, 0))
|
|
|
|
render_scroll = (int(self.scroll[0]), int(self.scroll[1]))
|
|
|
|
self.display.blit(self.current_tile_img, (10, 10))
|
|
|
|
self.tilemap.render(self.display, offset=render_scroll)
|
|
if self.ongrid:
|
|
self.display.blit(self.current_tile_img, (self.tile_pos[0] * self.tilemap.tile_size - self.scroll[0], self.tile_pos[1] * self.tilemap.tile_size - self.scroll[1]))
|
|
else:
|
|
self.display.blit(self.current_tile_img, self.mpos)
|
|
|
|
self.screen.blit(pg.transform.scale(self.display, self.screen.get_size()))
|
|
pg.display.update() |