63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
from survival import esper, GameMap
|
|
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 AutomationComponent:
|
|
pass
|
|
|
|
|
|
class AutomationSystem(esper.Processor):
|
|
def __init__(self, game_map: GameMap):
|
|
self.game_map = game_map
|
|
|
|
def process(self, dt):
|
|
for ent, (automation, pos) in self.world.get_components(AutomationComponent, PositionComponent):
|
|
if self.world.has_component(ent, PathfindingComponent):
|
|
continue
|
|
|
|
resource = self.detect_closest_resource(pos, ent)
|
|
if resource is None:
|
|
# TODO: Check if target position is not out of map bounds
|
|
self.world.add_component(ent, PathfindingComponent((pos.grid_position[0] * 32 + 64, pos.grid_position[1] * 32 + 64)))
|
|
# Move somewhere else
|
|
else:
|
|
target = self.world.component_for_entity(resource, PositionComponent).grid_position
|
|
self.world.add_component(ent, PathfindingComponent((target[0] * 32, target[1] * 32), True))
|
|
# Go collect target resource
|
|
|
|
def detect_closest_resource(self, position: PositionComponent, target_entity: int):
|
|
entity_position = position.grid_position
|
|
x_range = [entity_position[0] - 5, entity_position[0] + 5]
|
|
y_range = [entity_position[1] - 5, entity_position[1] + 5]
|
|
|
|
# Check if range is not out of map bounds
|
|
if x_range[0] < 0:
|
|
x_range[0] = 0
|
|
if x_range[1] >= self.game_map.width:
|
|
x_range[1] = self.game_map.width - 1
|
|
if y_range[0] < 0:
|
|
y_range[0] = 0
|
|
if y_range[1] >= self.game_map.height:
|
|
y_range[1] = self.game_map.height - 1
|
|
|
|
found_resource = [-1, 200000]
|
|
|
|
for y in range(y_range[0], y_range[1]):
|
|
for x in range(x_range[0], x_range[1]):
|
|
ent = self.game_map.get_entity([x, y])
|
|
if ent == target_entity:
|
|
continue
|
|
if ent is not None and self.world.has_component(ent, ResourceComponent):
|
|
res_position = self.world.component_for_entity(ent, PositionComponent).grid_position
|
|
distance = abs(entity_position[0] - res_position[0]) + abs(entity_position[1] - res_position[1])
|
|
if found_resource[1] > distance:
|
|
found_resource = [ent, distance]
|
|
|
|
if found_resource[0] == -1:
|
|
return None
|
|
else:
|
|
return found_resource[0]
|