You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
8.9 KiB
127 lines
8.9 KiB
from enne2engine.sdl2_wrapper import SDL2Wrapper |
|
from enne2engine.pyglet_wrapper import PygletWrapper |
|
from enne2engine.controls import UserControls |
|
import sys |
|
import os |
|
from Entities.Units.marine import Marine |
|
|
|
|
|
class GameEngine(UserControls): |
|
def __init__(self): |
|
super().__init__() |
|
if "--pyglet" in sys.argv: |
|
self.graphics = PygletWrapper() |
|
else: |
|
self.graphics = SDL2Wrapper(self) |
|
self.map = [ |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': True, 'tile': "landscapeTiles_066" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': True, 'tile': "landscapeTiles_066" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': True, 'tile': "landscapeTiles_066" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_001" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }], |
|
[{ 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': False, 'tile': "landscapeTiles_067" }, { 'wall': True, 'tile': "landscapeTiles_066" }] |
|
] |
|
self.frame_time = 0 |
|
self.cursor_pos = (0, 0) |
|
|
|
self.load_assets() |
|
self.entities = [] |
|
self.entities_positions = {} |
|
|
|
def run(self): |
|
running = True |
|
# Set a custom scale if needed |
|
self.graphics.set_scaling_factor(0.5) # 50% scale |
|
|
|
self.entities.append(Marine("knight", 0, 0, "idle", 1, 1, self)) |
|
self.graphics.create_background(self.map, "tiles") |
|
while running: |
|
perf_counter = self.graphics.get_perf_counter() |
|
for entity in self.entities: |
|
self.entities_positions[(entity.x, entity.y)] = entity |
|
self.graphics.create_background(self.map, "tiles") |
|
event = self.graphics.handle_events() |
|
if event: |
|
#print(f"Event detected: {event}") |
|
if event.startswith("MOUSEDOWN"): |
|
pass |
|
elif event.startswith("SELECTION"): |
|
# Handle multiple unit selection |
|
self.select_units_in_area(event) |
|
elif event.startswith("MOUSEUP"): |
|
self.select_entity_at_cursor() |
|
self.handle_events("keymap_game", event) |
|
running = False if event == "QUIT" else True |
|
self.graphics.clear_screen() |
|
self.graphics.render_background() |
|
#self.graphics.render_tile(spritesheet_name="tiles", tile="landscapeTiles_064", x=0, y=0) |
|
#self.graphics.draw_square(0,0) |
|
self.cursor_pos = self.graphics.draw_cursor() |
|
# Draw the selection rectangle if selecting |
|
if self.graphics.is_mouse_button_pressed(1): |
|
self.graphics.draw_selection_rectangle() |
|
for entity in self.entities: |
|
entity.update() |
|
self.graphics.render_sprites() |
|
self.graphics.update_status(f"Frame time: {round(self.frame_time)}ms - FPS: {round(1000/self.frame_time if self.frame_time != 0 else 1)}") |
|
self.graphics.present_renderer() |
|
self.frame_time = self.graphics.get_frame_time(perf_counter) |
|
self.graphics.delay_frame(self.frame_time,50) |
|
self.graphics.quit() |
|
|
|
def set_cursor(self, x, y): |
|
self.graphics.cursor = (x, y) |
|
|
|
def load_assets(self): |
|
self.graphics.load_tilesheet("tiles", "assets/tiles/landscapeTiles_sheet.png") |
|
for dir in os.listdir("assets/KnightBasic"): |
|
for file in os.listdir(f"assets/KnightBasic/{dir}"): |
|
if file.endswith(".json"): |
|
self.graphics.load_spritesheet(file[:-5].lower(), f"assets/KnightBasic/{dir}/{file}") |
|
|
|
def select_entity_at_cursor(self): |
|
cursor_x, cursor_y = self.cursor_pos |
|
print(f"Cursor position: {cursor_x}, {cursor_y}") |
|
# First deselect all entities |
|
for entity in self.entities: |
|
entity.selected = False |
|
|
|
# Then select the entity at cursor position, if any |
|
entity = self.entities_positions.get((cursor_x, cursor_y)) |
|
if entity: |
|
entity.select_unit() |
|
print(f"Selected entity at cursor: {entity.asset} at position {entity.x}, {entity.y}") |
|
else: |
|
print("No entity selected at cursor position.") |
|
|
|
def select_units_in_area(self, selection_event): |
|
"""Select all units within the specified selection area.""" |
|
# Parse selection coordinates |
|
_, start_x, start_y, end_x, end_y = selection_event.split(":") |
|
start_x, start_y, end_x, end_y = int(start_x), int(start_y), int(end_x), int(end_y) |
|
|
|
# Calculate selection rectangle in screen coordinates |
|
min_x = min(start_x, end_x) |
|
max_x = max(start_x, end_x) |
|
min_y = min(start_y, end_y) |
|
max_y = max(start_y, end_y) |
|
|
|
# First deselect all entities |
|
for entity in self.entities: |
|
entity.selected = False |
|
|
|
# Select entities within the rectangle |
|
for entity in self.entities: |
|
# Convert entity position to screen coordinates |
|
screen_x, screen_y = self.graphics.iso_transform(entity.x, entity.y) |
|
|
|
# Check if entity is within selection rectangle |
|
if min_x <= screen_x <= max_x and min_y <= screen_y <= max_y: |
|
entity.select_unit() |
|
print(f"Selected entity in area: {entity.asset} at position {entity.x}, {entity.y}") |
|
|
|
if __name__ == "__main__": |
|
engine = GameEngine() |
|
engine.run() |