AI-Project/survival/generators/resource_generator.py

import random

from enum import Enum
from survival import GameMap
from survival.components.OnCollisionComponent import OnCollisionComponent
from survival.components.inventory_component import InventoryComponent
from survival.components.position_component import PositionComponent
from survival.components.resource_component import ResourceComponent
from survival.components.sprite_component import SpriteComponent
from survival.esper import World
from survival.settings import RESOURCES_AMOUNT


class ResourceType(Enum):
    FOOD = 1
    WATER = 2
    WOOD = 3


class ResourceGenerator:
    def __init__(self, world, game_map):
        self.world = world
        self.map = game_map

    def generate_resources(self, player: int):
        for x in range(RESOURCES_AMOUNT):
            obj = self.world.create_entity()
            sprites = {
                ResourceType.FOOD: 'apple.png',
                ResourceType.WATER: 'water.png',
                ResourceType.WOOD: 'wood.png'
            }

            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)
            resource_type = random.choice(list(ResourceType))
            sprite = SpriteComponent(sprites[resource_type])
            col = OnCollisionComponent()
            col.addCallback(self.remove_resource, world=self.world, game_map=self.map, entity=obj, player=player)
            self.world.add_component(obj, pos)
            self.world.add_component(obj, sprite)
            self.world.add_component(obj, col)
            self.world.add_component(obj, ResourceComponent(resource_type))
            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

    @staticmethod
    def remove_resource(world: World, game_map: GameMap, entity: int, player: int):
        pos = world.component_for_entity(entity, PositionComponent)
        resource = world.component_for_entity(entity, ResourceComponent)
        inventory = world.component_for_entity(player, InventoryComponent)
        inventory.add_item(resource.resource_type, 1)
        game_map.remove_entity(pos.grid_position)
        world.delete_entity(entity, immediate=True)