Update main.py

main.py

@@ -7,8 +7,6 @@ POPULATION_SIZE = 20
 MUTATION_RATE = 0.5
 POPULATION_BEST = 0.2
 FRAME_RATE = 160 #TODO: FIX THE INCORRECT FRAME RATE CORRELATION
-BARRIER_SPEED = 10
-BARRIER_DELAY = 75
 
 pygame.init()
 
@@ -34,11 +32,8 @@ def init_game():
     grass = Background("images/Grass.jpg", 0, 0)
     grass.set_size(WIDTH, HEIGHT)
 
-<<<<<<< Updated upstream
-=======
     BARRIER_SPEED = 25
 
->>>>>>> Stashed changes
     gameobjects = []
     
     barriers = []
@@ -51,7 +46,7 @@ def init_game():
         horse.set_vspeed(0)
         horse.frame_counter = 0
         horse.fitness = 0
-    spawner = Spawner("images/Barrier.png", BARRIER_DELAY)
+    spawner = Spawner("images/Barrier.png", 125)
     new_barrier = spawner.spawn()
     barriers.append(new_barrier)
     gameobjects.extend(horses)
@@ -70,12 +65,12 @@ def get_features(horse):
                last_barrier.rect.bottomright[0], last_barrier.rect.bottomright[1], 
                horse.rect.topleft[0], horse.rect.topleft[1],
                horse.rect.bottomright[0], horse.rect.bottomright[1],
-               0, HEIGHT, BARRIER_SPEED]
+               HEIGHT, BARRIER_SPEED]
     return features 
 
+genecticAlg = GeneticAlgorithm(POPULATION_SIZE, MUTATION_RATE, POPULATION_BEST, 10)
 horses = [Horse("images/Horse_1.png", 50, HEIGHT/2 + 0 * i) for i in range(POPULATION_SIZE)]
 init_game()
-genecticAlg = GeneticAlgorithm(POPULATION_SIZE, MUTATION_RATE, POPULATION_BEST, len(get_features(horses[0])))
 while True:
     for event in pygame.event.get():
         if event.type == pygame.QUIT:  # Handle window close event
@@ -120,6 +115,7 @@ while True:
             if horse.rect.colliderect(upper_bound_rect) or horse.rect.colliderect(lower_bound_rect):
                 horse.stop()
             horse.count_fitness()
+            print(str(horse.color) + ' ' + str(horse.fitness))
                 
 
     for object in gameobjects: