testing next gen

Author: Dpbm

game/players/ai_player.cpp

@@ -14,6 +14,14 @@ using Utils::get_angle;
 namespace Players{
   AIPlayer::AIPlayer(uint8_t board_w, uint8_t board_h) : Player(board_w, board_h){
     this->setup_nn();
+    this->setup_chromosome(); 
+    this->load_genes_into_weights();
+  }
+  
+  AIPlayer::AIPlayer(uint8_t board_w, uint8_t board_h, Chromosome* chromosome) : Player(board_w, board_h){
+    this->setup_nn();
+    this->chromosome = chromosome;
+    this->load_genes_into_weights();
   }
 
   void AIPlayer::setup_nn(){
@@ -25,9 +33,6 @@ namespace Players{
 
     this->nn->get_layer(1)->relu(); 
     this->nn->get_output_layer()->softmax();
-    
-    this->setup_chromosome(); 
-    this->load_genes_into_weights();
   }
 
   void AIPlayer::setup_chromosome(){

game/players/ai_player.h

@@ -18,6 +18,7 @@ namespace Players{
   class AIPlayer : public Player {
     public:
       AIPlayer(uint8_t board_w, uint8_t board_h);
+      AIPlayer(uint8_t board_w, uint8_t board_h, Chromosome* chromosome);
       ~AIPlayer();
 
       void load_genes_into_weights();

genetic/population.cpp

@@ -22,11 +22,13 @@ namespace Genetic{
   Population::Population(uint16_t total, uint8_t board_w, uint8_t board_h, uint8_t total_food){
     this->total_ind = total;
     this->total_food = total_food;
+    this->board_h = board_h;
+    this->board_w = board_w;
+    
     this->generate_food_positions(total_food, board_w, board_h);
-  
     vec2 first_food_pos = this->food_positions.at(0);
-
-    for(size_t i = 0; i < total; i++){
+    
+    for(size_t i = 0; i < this->total_ind; i++){
       Individual* ind = new Individual;
       ind->board = new Board(board_w, board_h);
       ind->player = new AIPlayer(board_w, board_h);
@@ -134,19 +136,45 @@ namespace Genetic{
 
       this->update_individual_food_position(ind);
     }
-    if(this->total_alive == 0)
-      this->gen ++;
   }
 
   void Population::next_gen(){
+    this->gen++;
     Individual** parents = this->select_parents();
     Chromosome* offspring = this->generate_offspring(parents[0]->player->get_chromossome(), parents[1]->player->get_chromossome());
-  
-    //reset individuals
-    //replicate
-    //mutate
-    //reset foods
-    //clear pointers
+    Gene* offspring_genes = offspring->get_genes();
+
+    uint64_t offspring_ch_size = offspring->get_size();
+
+    this->clear();
+    this->food_positions.clear();
+    this->generate_food_positions(total_food, board_w, board_h);
+    vec2 first_food_pos = this->food_positions.at(0);
+    
+    this->individuals.clear();
+    for(size_t i = 0; i < this->total_ind; i++){
+      Individual* ind = new Individual;
+      ind->board = new Board(board_w, board_h);
+      
+      if(i == 0)
+        ind->player = new AIPlayer(board_w, board_h, offspring);
+      else{
+        Chromosome* player_chromosome = new Chromosome(offspring_ch_size);
+        player_chromosome->copy_genes(offspring_genes);
+        player_chromosome->mutate(0.02);
+
+        ind->player = new AIPlayer(board_w, board_h, player_chromosome);
+      }
+
+      ind->board->set_food_pos(first_food_pos.x, first_food_pos.y); 
+      ind->board->add_player(ind->player);
+      ind->fitness = 0;
+      ind->same_dir_counter = 0;
+      ind->index = i;
+      ind->las_dir = ind->player->get_dir();
+
+      this->individuals.push_back(ind); 
+    }
     delete parents;
   }
 
@@ -231,11 +259,15 @@ namespace Genetic{
     return this->get_best_individual()->fitness;
   }
 
-  Population::~Population(){
+  void Population::clear(){
     for(Individual* ind : this->individuals){
       delete ind->board;
       delete ind->player;
       delete ind;
     }   
   }
+
+  Population::~Population(){
+    this->clear();
+  }
 };

genetic/population.h

@@ -38,6 +38,7 @@ namespace Genetic {
       int64_t get_best_fitness();
       ~Population();
 
+      
       //for tests (these ones could be private)
       Individual** select_parents();
       void append_individual(Individual* ind);
@@ -50,11 +51,14 @@ namespace Genetic {
       uint16_t total_ind = 0;
       uint16_t total_alive = 0;
       uint16_t total_win = 0;
+      uint8_t board_w = 0;
+      uint8_t board_h = 0;
       vector<Individual*> individuals;
       vector<vec2> food_positions; 
 
       Individual* get_best_individual();
       void generate_food_positions(uint8_t total, uint8_t w, uint8_t h);
       void update_individual_food_position(Individual *ind);
+      void clear();
   };
 };