eval_genomes.py

import pygame, random, os, sys, neat
from classes.background import Background
from classes.bird import Bird
from classes.pipe import Pipe
pygame.init()
pygame.mixer.init()

SCREEN_WIDTH, SCREEN_HEIGHT = 500, 768
SCREEN = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
FONT = pygame.font.Font('./assets/flappy.ttf', 72)
pygame.display.set_caption("NEAT - Flappy Bird")

flap_sound = pygame.mixer.Sound("./assets/bird/wing.mp3")
point_sound = pygame.mixer.Sound("./assets/point.mp3")

def display_score(score):
    score_img = FONT.render("{}".format(score), True, (255, 255, 255))
    score_rect = score_img.get_rect()
    score_rect.center = (SCREEN_WIDTH // 2, 100)
    SCREEN.blit(score_img, score_rect)

def remove_bird(i, genomes, nets, score):
    genomes[i].fitness += score
    Bird.birds.pop(i)
    genomes.pop(i)
    nets.pop(i)

def eval_genomes(genomes, config):
    run = True
    clock = pygame.time.Clock()

    # Initialize a background
    bg = Background(SCREEN_WIDTH, SCREEN_HEIGHT)

    score = 0

    # Setup genomes
    ge = []
    nets = []
    for genome_id, genome in genomes:
        Bird.birds.append(Bird(SCREEN_WIDTH // 2, SCREEN_HEIGHT // 2, Bird.COLORS[genome_id % 6]))
        ge.append(genome)
        nets.append(neat.nn.FeedForwardNetwork.create(genome, config))
        genome.fitness = 0

    while run:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
                pygame.quit()

        if len(Bird.birds) == 0:
            break

        dt = 1 / 60
        SCREEN.fill((255, 255, 255))

        if len(Pipe.pipes) == 0 or Pipe.pipes[-1].right_x() < SCREEN_WIDTH - 300:
            bottom_y = random.randint(300, SCREEN_HEIGHT - 200)
            top_y = random.randint(100, bottom_y - 200)
            pipe = Pipe(SCREEN_WIDTH, bottom_y, top_y)

        for i, bird in enumerate(Bird.birds):
            for pipe in Pipe.pipes:
                if pipe.right_x() >= SCREEN_WIDTH // 2:
                    closest_pipe = pipe
                    break
            output = nets[i].activate((bird.rect.y, abs(closest_pipe.top_pipe_y() - bird.rect.y), abs(closest_pipe.bottom_pipe_y() - bird.rect.y)))
            if output[0] > 0.5:
                bird.jump()

        # Update and draw the background
        bg.update(dt)
        bg.draw(SCREEN)

        # Update and draw the birds
        for i, bird in enumerate(Bird.birds):
            bird.update(dt)

            # Collisions 
            for pipe in Pipe.pipes:
                if pipe.collide(bird):
                    remove_bird(i, ge, nets, score)
                if bird.rect.bottom < 0 or bird.rect.top  > SCREEN_HEIGHT:
                    remove_bird(i, ge, nets, score)

            bird.draw(SCREEN)

        for pipe in Pipe.pipes:
            pipe.update(dt)
            pipe.draw(SCREEN)

            if pipe.right_x() < SCREEN_WIDTH // 2 and not pipe.scored:
                score += 1
                pipe.scored = True
                point_sound.play()

        display_score(score)

        pygame.display.update()
        clock.tick(60)
    Pipe.pipes = []

def run(config_path):
    config = neat.config.Config(
        neat.DefaultGenome,
        neat.DefaultReproduction,
        neat.DefaultSpeciesSet,
        neat.DefaultStagnation,
        config_path
    )
    pop = neat.Population(config)
    pop.run(eval_genomes, 50)

if __name__ == "__main__":
    if getattr(sys, 'frozen', False):
        local_dir = os.path.dirname(sys.executable)
    elif __file__:
        local_dir = os.path.dirname(__file__)
    config_path = os.path.join(local_dir, 'config.txt')
    run(config_path)