main.py

# IMPORTS
import pygame
import neat
import random
import time
import os
pygame.font.init()

# Window dimensions
WINDOW_WIDTH = 500
WINDOW_HEIGHT = 800


def LoadImage(file_name: str, doubled:bool = True):
    if doubled:
        return pygame.transform.scale2x(pygame.image.load(os.path.join('Assets', file_name)))
    else:
        return pygame.image.load(os.path.join('Assets', file_name))

def RandomColor(lower_bound = None, upper_bound = None):
        if not lower_bound:
            l_bound = 0
        else:
            l_bound = lower_bound

        if not upper_bound:
            u_bound = 255
        else:
            u_bound = lower_bound

        return random.randint(l_bound, u_bound), random.randint(l_bound, u_bound), random.randint(l_bound, u_bound)


BIRD_IMGS = [ LoadImage('bird1.png'), LoadImage('bird2.png'), LoadImage('bird3.png')]
PIPE_IMG = LoadImage('pipe.png')
FLOOR_IMG = LoadImage('base.png')
BG_IMG = LoadImage('bg.png')

STAT_FONT = pygame.font.SysFont("comicsans", 50)
class Bird:
    ### CONSTANTS FOR ALL AGENTS
    # Animation frames of bird
    IMGS = BIRD_IMGS
    # Tilt range of bird
    MAX_ROTATION = 25
    # Tilt speed of bird
    ROT_VEL = 20
    # Flapping speed of bird
    ANIMATION_TIME = 5

    def __init__(self, pos_horizontal, pos_vertical):
        ### AGENT PROPETIES
        # Position of agent
        self.pos_horizontal = pos_horizontal
        self.pos_vertical = pos_vertical
        # Tilt of agent
        self.tilt = 0
        # Animation tick counter
        self.tick_count = 0
        # ???
        self.vel = 0
        # Current height of jump
        self.height = 0
        # How many times the frame has been shown
        self.img_count = 0
        # Current animation frame
        self.img = self.IMGS[0]
        # Tint of the agent
        self.r, self.g, self.b = RandomColor(100)
        self.tint = (self.r, self.g, self.b)

    def Jump(self):
        # Set initial velocity of jump
        self.vel = -10.5
        # Reset tick of animation
        self.tick_count = 0
        # Reset height to current bird position
        self.height = self.pos_vertical

    def Move(self):
        # Increment tick
        self.tick_count += 1
        # Jump physics
        displacement  = self.vel*self.tick_count + 1.5*self.tick_count**2

        # Limit falling speed
        if displacement >= 16:
            displacement = 16
        elif displacement < 0:
            displacement -= 2

        # Apply the displacement
        self.pos_vertical += displacement

        # If we are going upwards
        if displacement < 0 or self.pos_vertical < self.height + 50:
            # Rotate the agent to the most upward rotation
            # !!! I think this 'if' might be deletable.
            if self.tilt < self.MAX_ROTATION:
                self.tilt = self.MAX_ROTATION
        # Else we must be going downwards
        else:
            # Rotate the agent downwards until we hit the straight down
            if self.tilt > -90: # add STRAIGHT_DOWN ?
                self.tilt -= self.ROT_VEL

    def Draw(self, win):
        self.img_count += 1
        
        # Not efficient but works
        # Goddamn it feels horrible
        # Pls change it in the future
        if self.img_count < self.ANIMATION_TIME:
            self.img = self.IMGS[0]
        elif self.img_count < self.ANIMATION_TIME*2:
            self.img = self.IMGS[1]
        elif self.img_count < self.ANIMATION_TIME*3:
            self.img = self.IMGS[2]
        elif self.img_count < self.ANIMATION_TIME*4:
            self.img = self.IMGS[1]
        elif self.img_count < self.ANIMATION_TIME*5:
            self.img = self.IMGS[0]
            self.img_count = 0

        # If falling down
        if self.tilt <= -80:
            # Make animation fly upwards when agent finally flaps
            self.img = self.IMGS[1]
            self.img_count = self.ANIMATION_TIME*2

        # Tint the agent
        self.img = self.img.convert_alpha()
        self.img.fill((self.r, self.g, self.b, 255), special_flags=pygame.BLEND_RGBA_MULT)

        # Stackoverflow code. How does it work?
        # Rotate the agent image around it's center
        rotated_image = pygame.transform.rotate(self.img, self.tilt)
        new_rect = rotated_image.get_rect(center=self.img.get_rect(topleft = (self.pos_horizontal, self.pos_vertical)).center)


        # Blit the rotated agent(Bird) onto the screen
        win.blit(rotated_image, new_rect.topleft)

    def GetMask(self):
        return pygame.mask.from_surface(self.img)

class Pipe:
    ### CONSTANTS
    # Distance between the generated pipes
    GAP = 200
    # Velocity of pipes
    VEL = 5

    # This class actually holds both TOP and BOTTOM pipe

    def __init__(self, pos_horizontal):
        # Horizontal placement of pipe. Initiation usually to right end of the window
        self.pos_horizontal = pos_horizontal
        # Think about is as a gap pposition
        self.height = 0

        # Vertical placement anchors for top and bottom pipe
        self.top = 0
        self.bottom = 0

        # Pipe images
        self.PIPE_BOTTOM = PIPE_IMG
        self.PIPE_TOP = pygame.transform.flip(PIPE_IMG, False, True) # Worth moving to globals?

        # Whether the agent passed thru
        self.passed = False

        self.SetHeight()

    # Set positions of top and bottom pipe
    def SetHeight(self):
        # Pick the pipe gap position randomly
        self.height = random.randrange(50, 450)
        # Set the pipe vertical anchors
        self.top = self.height - self.PIPE_TOP.get_height()
        self.bottom = self.height + self.GAP

    # Change the pipes horizontal position in relation to VEL
    def Move(self):
        self.pos_horizontal -= self.VEL

    # Blit both top and bottom pipe onto the window passed in the argument
    def Draw(self, win):
        win.blit(self.PIPE_TOP, (self.pos_horizontal, self.top))
        win.blit(self.PIPE_BOTTOM, (self.pos_horizontal, self.bottom))

    # Pixel Perfect collisions using masks
    def Collide(self, bird:Bird):
        bird_mask = bird.GetMask()
        top_mask = pygame.mask.from_surface(self.PIPE_TOP)
        bottom_mask = pygame.mask.from_surface(self.PIPE_BOTTOM)

        # Calculate the distance between the bird and the pipes. (VECTOR)
        top_offset = (self.pos_horizontal - bird.pos_horizontal, self.top - round(bird.pos_vertical))
        bottom_offset = (self.pos_horizontal - bird.pos_horizontal, self.bottom - round(bird.pos_vertical))

        # Check for overlaps
        b_point = bird_mask.overlap(bottom_mask, bottom_offset)
        t_point = bird_mask.overlap(top_mask, top_offset)

        # If b_point != None, then there is a collision
        if t_point or b_point:
            return True

        # Else no collision
        return False


class Floor:
    ### CONSTANTS
    # Velocity of the ground moving. Should be equal to the velocity of pipes
    VEL = Pipe.VEL # Will it work? Do i need to make it static somehow like in c#
    # Width of the base image
    WIDTH = FLOOR_IMG.get_width()
    # Sprite of the base
    IMG = FLOOR_IMG

    def __init__(self, pos_vertical):
        self.pos_vertical = pos_vertical
        self.anchor_zero = 0
        self.anchor_one = self.WIDTH

    def Move(self):
        # Translate the anchors by velocity
        self.anchor_zero -= self.VEL
        self.anchor_one -= self.VEL

        # Interlace the floors
        if self.anchor_zero + self.WIDTH < 0:
            self.anchor_zero = self.anchor_one + self.WIDTH

        if self.anchor_one + self.WIDTH < 0:
            self.anchor_one = self.anchor_zero + self.WIDTH

    # Basic blit to the screen
    def Draw(self, win):
        win.blit(self.IMG, (self.anchor_zero, self.pos_vertical))
        win.blit(self.IMG, (self.anchor_one, self.pos_vertical))

    # Pixel perfect collisions
    def Collide(self, bird: Bird):

        '''
        # Create masks
        bird_mask = bird.GetMask()
        floor_mask = pygame.mask.from_surface(self.IMG)

        # Calculate offset
        floor_zero_offset = (round(self.anchor_zero - bird.pos_horizontal), round(self.pos_vertical - bird.pos_vertical))
        floor_one_offset = (round(self.anchor_one - bird.pos_horizontal), round(self.pos_vertical - bird.pos_vertical))

        # Check overlap
        overlap_0 = bird_mask.overlap(floor_mask, floor_zero_offset)
        overlap_1 = bird_mask.overlap(floor_mask, floor_one_offset)

        # Pixel or more overlaps - COLLISION
        if overlap_0 or overlap_1:
            return True

        # Else - NO COLLISION
        else:
            return False
        '''
        # Collision with floor or ceiling
        if bird.pos_vertical + bird.img.get_height() >= self.pos_vertical or bird.pos_vertical < 0:
            return True

        else:
            return False



        



def DrawWindow(win, birds, pipes, floor, score):
    win.blit(BG_IMG, (0,0))

    # Take care of bird
    for bird in birds:
        bird.Draw(win)

    # Floor
    floor.Draw(win)

    # Pipes
    for pipe in pipes:
        pipe.Draw(win)

    # Score
    text = STAT_FONT.render("Score: " + str(score), True, pygame.Color('black'))
    win.blit(text, (WINDOW_WIDTH - 10 - text.get_width(), 10))

    pygame.display.update()
    

def Main(genomes, config):
    nets  = []
    ge = []
    birds = []

    for _, g in genomes:
        net = neat.nn.FeedForwardNetwork.create(g, config)
        nets.append(net)
        birds.append(Bird(230, 350))
        g.fitness = 0
        ge.append(g)

    PIPE_DISTANCING = 600
    FPS = 60
    floor = Floor(730)
    pipes = [Pipe(PIPE_DISTANCING)]
    score = 0

    win = pygame.display.set_mode( (WINDOW_WIDTH, WINDOW_HEIGHT) )
    clock = pygame.time.Clock()

    run = True

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

        pipe_ind = 0
        if len(birds) > 0:
            if len(pipes) > 1 and birds[0].pos_horizontal > pipes[0].pos_horizontal + pipes[0].PIPE_TOP.get_width():
                pipe_ind = 1
        else:
            run = False
            break

        for x, bird in enumerate(birds):
            bird.Move()
            ge[x].fitness += 0.1

            output = nets[x].activate((bird.pos_vertical, abs(bird.pos_vertical - pipes[pipe_ind].height), abs(bird.pos_vertical - pipes[pipe_ind].bottom)))

            if output[0] > 0.5:
                bird.Jump()


        # Pipes
        rem = []
        add_pipe = False
        for pipe in pipes:
            for x, bird in enumerate(birds):
                if pipe.Collide(bird):
                    ge[x].fitness -= 1
                    birds.pop(x)                    
                    nets.pop(x)
                    ge.pop(x)

                if not pipe.passed and pipe.pos_horizontal < bird.pos_horizontal:
                    pipe.passed = True
                    add_pipe = True

            if pipe.pos_horizontal + pipe.PIPE_TOP.get_width() < 0:
                rem.append(pipe)

            pipe.Move()

        if add_pipe:
            score += 1
            for g in ge:
                g.fitness += 5
                
            pipes.append(Pipe(PIPE_DISTANCING))

        for r in rem:
            pipes.remove(r)

        for x, bird in enumerate(birds):
            if floor.Collide(bird):
                birds.pop(x)
                nets.pop(x)
                ge.pop(x)

        floor.Move()
        
            
        DrawWindow(win, birds, pipes, floor, score)

    

def Run(config_path):
    # Set configuration file
    config = neat.config.Config(neat.DefaultGenome, neat.DefaultReproduction, neat.DefaultSpeciesSet, neat.DefaultStagnation, config_path)

    # Set population
    p = neat.Population(config)

    # Add reported for stats
    p.add_reporter(neat.StdOutReporter(True))
    stats = neat.StatisticsReporter()
    p.add_reporter(stats)

    # Set the fitness function
    winner = p.run(Main, 50)


if __name__ == "__main__":
    local_dir = os.path.dirname(__file__)
    config_path = os.path.join(local_dir, "config-feedforward.txt")
    Run(config_path)