2021-03-28 18:42:04 +02:00
|
|
|
import random
|
|
|
|
|
2021-05-23 14:33:21 +02:00
|
|
|
from survival import GameMap
|
2021-05-21 20:14:00 +02:00
|
|
|
from survival.components.OnCollisionComponent import OnCollisionComponent
|
2021-03-28 18:42:04 +02:00
|
|
|
from survival.components.position_component import PositionComponent
|
|
|
|
from survival.components.sprite_component import SpriteComponent
|
2021-05-23 14:33:21 +02:00
|
|
|
from survival.esper import World
|
2021-03-28 18:56:32 +02:00
|
|
|
from survival.settings import RESOURCES_AMOUNT
|
2021-03-28 18:42:04 +02:00
|
|
|
|
|
|
|
|
|
|
|
class ResourceGenerator:
|
|
|
|
def __init__(self, world, game_map):
|
|
|
|
self.world = world
|
|
|
|
self.map = game_map
|
|
|
|
|
|
|
|
def generate_resources(self):
|
2021-03-28 18:56:32 +02:00
|
|
|
for x in range(RESOURCES_AMOUNT):
|
2021-03-28 18:42:04 +02:00
|
|
|
obj = self.world.create_entity()
|
2021-03-29 17:24:43 +02:00
|
|
|
sprites = ['apple.png', 'water.png', 'wood.png']
|
2021-03-28 18:42:04 +02:00
|
|
|
|
|
|
|
empty_grid_pos = self.get_empty_grid_position()
|
|
|
|
empty_pos = [empty_grid_pos[0] * 32, empty_grid_pos[1] * 32]
|
|
|
|
|
|
|
|
pos = PositionComponent(empty_pos, empty_grid_pos)
|
|
|
|
sprite = SpriteComponent(random.choice(sprites))
|
2021-05-21 20:14:00 +02:00
|
|
|
col = OnCollisionComponent()
|
2021-05-23 14:33:21 +02:00
|
|
|
col.addCallback(self.remove_resource, world=self.world, game_map=self.map, entity=obj)
|
2021-03-28 18:42:04 +02:00
|
|
|
self.world.add_component(obj, pos)
|
|
|
|
self.world.add_component(obj, sprite)
|
2021-05-21 20:14:00 +02:00
|
|
|
self.world.add_component(obj, col)
|
2021-03-28 18:42:04 +02:00
|
|
|
self.map.add_entity(obj, pos)
|
|
|
|
|
|
|
|
def get_empty_grid_position(self):
|
|
|
|
free_pos = [random.randrange(self.map.width), random.randrange(self.map.height)]
|
|
|
|
while self.map.is_colliding(free_pos):
|
|
|
|
free_pos = [random.randrange(self.map.width), random.randrange(self.map.height)]
|
|
|
|
return free_pos
|
2021-05-23 14:33:21 +02:00
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def remove_resource(world: World, game_map: GameMap, entity: int):
|
|
|
|
pos = world.component_for_entity(entity, PositionComponent)
|
|
|
|
game_map.remove_entity(pos.grid_position)
|
|
|
|
world.delete_entity(entity, immediate=True)
|