59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
import pygame
|
|
|
|
from survival.ai.genetic_algorithm import GeneticAlgorithm
|
|
from survival.components.inventory_component import InventoryComponent
|
|
from survival.game.game_map import GameMap
|
|
from survival.generators.building_generator import BuildingGenerator
|
|
from survival.generators.player_generator import PlayerGenerator
|
|
from survival.generators.resource_generator import ResourceGenerator
|
|
from survival.generators.world_generator import WorldGenerator
|
|
from survival.settings import SCREEN_WIDTH, SCREEN_HEIGHT, MUTATE_NETWORKS, LEARN
|
|
from survival.systems.draw_system import DrawSystem
|
|
from survival.systems.neural_system import NeuralSystem
|
|
|
|
|
|
class Game:
|
|
def __init__(self):
|
|
self.world_generator = WorldGenerator(win, self.reset)
|
|
self.game_map, self.world, self.camera = self.world_generator.create_world()
|
|
self.run = True
|
|
if LEARN and MUTATE_NETWORKS:
|
|
self.genetic_algorithm = GeneticAlgorithm(self.world.get_processor(NeuralSystem), self.finish_training)
|
|
|
|
def reset(self):
|
|
if LEARN and MUTATE_NETWORKS:
|
|
self.genetic_algorithm.train()
|
|
self.world_generator.reset_world()
|
|
|
|
def finish_training(self):
|
|
self.run = False
|
|
|
|
def update(self, ms):
|
|
events = pygame.event.get()
|
|
|
|
for event in events:
|
|
if event.type == pygame.QUIT:
|
|
self.run = False
|
|
if pygame.key.get_pressed()[pygame.K_DELETE]:
|
|
self.reset()
|
|
|
|
win.fill((0, 0, 0))
|
|
self.game_map.draw(self.camera)
|
|
self.world.process(ms)
|
|
pygame.display.update()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
pygame.init()
|
|
|
|
pygame.font.init()
|
|
|
|
win = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
|
|
pygame.display.set_caption("AI Project")
|
|
|
|
clock = pygame.time.Clock()
|
|
game = Game()
|
|
|
|
while game.run:
|
|
game.update(clock.tick(500))
|