AI-Project/survival/generators/resource_generator.py

35 lines
1.3 KiB
Python
Raw Normal View History

2021-03-28 18:42:04 +02:00
import random
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-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()
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))
col = OnCollisionComponent()
2021-03-28 18:42:04 +02:00
self.world.add_component(obj, pos)
self.world.add_component(obj, sprite)
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