41 lines
2.0 KiB
Python
41 lines
2.0 KiB
Python
import pygame
|
|
|
|
from survival import esper, GameMap
|
|
from survival.components.direction_component import DirectionChangeComponent
|
|
from survival.components.input_component import InputComponent
|
|
from survival.components.moving_component import MovingComponent
|
|
from survival.components.pathfinding_component import PathfindingComponent
|
|
from survival.components.position_component import PositionComponent
|
|
from survival.components.resource_component import ResourceComponent
|
|
|
|
|
|
class InputSystem(esper.Processor):
|
|
def __init__(self, camera, game_map: GameMap):
|
|
self.camera = camera
|
|
self.game_map = game_map
|
|
|
|
def process(self, dt):
|
|
for ent, (inp, pos) in self.world.get_components(InputComponent, PositionComponent):
|
|
keys = pygame.key.get_pressed()
|
|
mouse = pygame.mouse.get_pressed(3)
|
|
if mouse[0] == 1:
|
|
pos = pygame.mouse.get_pos()
|
|
pos = (pos[0] - self.camera.camera.left, pos[1] - self.camera.camera.top)
|
|
if not self.world.has_component(ent, PathfindingComponent):
|
|
target_ent = self.game_map.get_entity([int(pos[0] / 32), int(pos[1]/ 32)])
|
|
if target_ent is not None and self.world.has_component(target_ent, ResourceComponent):
|
|
self.world.add_component(ent, PathfindingComponent(pos))
|
|
else:
|
|
self.world.add_component(ent, PathfindingComponent(pos))
|
|
|
|
if self.world.has_component(ent, MovingComponent):
|
|
continue
|
|
if keys[pygame.K_LEFT]:
|
|
if not self.world.has_component(ent, DirectionChangeComponent):
|
|
self.world.add_component(ent, DirectionChangeComponent(pos.rotate_left()))
|
|
elif keys[pygame.K_RIGHT]:
|
|
if not self.world.has_component(ent, DirectionChangeComponent):
|
|
self.world.add_component(ent, DirectionChangeComponent(pos.rotate_right()))
|
|
elif keys[pygame.K_UP]:
|
|
self.world.add_component(ent, MovingComponent())
|