Habitat.pde

class Habitat {
  LifeTime life;
  ArrayList<Snake> snakes;
  FloatList fitnesses;
  Point currentSnakePosition;
  int currentSnake = 0, currentGeneration = 0;
  int population;
  
  Habitat(){
    currentSnakePosition = new Point(int(random(safetyMargin, size[0]-safetyMargin)), int(random(safetyMargin, size[1]-safetyMargin)));
    life = new LifeTime(5, -1);
    snakes = new ArrayList<Snake>(1);
    population = 1;
    snakes.add(new Snake());
    life.setSpecimen(snakes.get(0));
  };
  Habitat(int numOfMice, int numOfSeconds) {
    currentSnakePosition = new Point(int(random(safetyMargin, size[0]-safetyMargin)), int(random(safetyMargin, size[1]-safetyMargin)));
    life = new LifeTime(numOfMice, numOfSeconds);
    snakes = new ArrayList<Snake>(1);
    snakes.add(new Snake());
    population = 1;
    fitnesses = new FloatList(1);
    life.setSpecimen(snakes.get(0));
  };
  Habitat(int numOfSnakes, int numOfMice, int numOfSeconds) {
    currentSnakePosition = new Point(int(random(safetyMargin, size[0]-safetyMargin)), int(random(safetyMargin, size[1]-safetyMargin)));
    life = new LifeTime(numOfMice, numOfSeconds);
    snakes = new ArrayList<Snake>(numOfSnakes);
    population = numOfSnakes;
    fitnesses = new FloatList(numOfSnakes);
    for(int i=0; i<numOfSnakes; i++){
      snakes.add(new Snake());
    }
    life.setSpecimen(snakes.get(0));
  };
  
  void update() {
    if(life.update(currentGeneration % drawGenerationEvery == 0)) {
      if(currentGeneration % drawGenerationEvery == 0){
        frameRate(30);
      } else{
        frameRate(30*1000);
      }
      fitnesses.append(life.popFitness());
      life.setSpecimen(snakes.get(currentSnake));
      life.startTiming();
      snakes.get(currentSnake).setPosition(currentSnakePosition);
      currentSnake++;
    }
    
    if(snakes.size() <= currentSnake) {
      if(saveCurrentGen){  
        String fileName = 'G'+str(currentGeneration)+"_"+str(currentSnake);
        saveGeneration(savePath + fileName);
        println(fileName + " saved");
        saveCurrentGen = false;
      }
      if(loadNextGen){
        currentGeneration = 0;
        loadNextGen = false;
        loadGeneration();
      } else {
        nextGeneration();
        life.NewMousePositions();
        currentGeneration++;
      }
    }
    
    writeGenInfo();
  };
  
  void loadGeneration(){
    BufferedReader in = createReader(loadGen.getAbsolutePath());
    snakes.clear();
    String line = null;
    try{
      while((line = in.readLine()) != null)
        snakes.add(new Snake(line));
      in.close();
    } catch(IOException e){
      e.printStackTrace();
    }
  }
  
  void saveGeneration(String fName){ //<>// //<>//
    PrintWriter out = createWriter(fName);
    for(Snake snek : snakes)
      out.println(snek.Serialize());
    out.flush();
    out.close();
  };
  
  void nextGeneration() {
    currentSnakePosition = new Point(int(random(safetyMargin, size[0]-safetyMargin)), int(random(safetyMargin, size[1]-safetyMargin)));
    
    double fitnessSum=0, curSum, fitnessRoll;
    for(int i=0; i<fitnesses.size(); i++)
      if(fitnesses.get(i) < 1e-3)
        fitnesses.set(i, 0.0);
    fitnessSum = fitnesses.sum();
    ArrayList<Snake> newGeneration = new ArrayList<Snake>();
    int childrenSum=0, childrenCount[] = new int[fitnesses.size()];
    
    print(fitnessSum + "\n");
    if(fitnessSum <= 1e-3) {
      print("Generation " + currentGeneration + " unusable\n");
      while(newGeneration.size() < population) {
        newGeneration.add(new Snake());
      }
    }
    else {
      while(childrenSum < population) {
        if(fitnessSum > 1e-3) {
          fitnessRoll = random((float)fitnessSum);
          curSum = 0;
          for(int i=0; i<snakes.size(); i++){
            curSum = curSum + fitnesses.get(i);
            if(curSum >= fitnessRoll) {
              if(childrenCount[i] == 0)
                childrenSum++;
              childrenCount[i]++;
              childrenSum++;
              //newGeneration.add(snakes.get(i));
              //snakes.get(i).Coords.clear();
              break;
            }
          }
        } 
        else {
          int i = (int)random(snakes.size());
          if(childrenCount[i] == 0)
            childrenSum++;
          childrenCount[i]++;
          childrenSum++;
          //newGeneration.add(snakes.get(i));
          //snakes.get(i).Coords.clear();
        }
      }
      
      for(int i=0; i<fitnesses.size(); i++) {
        if(childrenCount[i] > 0) {
          newGeneration.add(snakes.get(i));
          while(childrenCount[i]--> 0) {
            newGeneration.add(new Snake(snakes.get(i)));
          }
        }
      }
      
      /*int parent_index = 0;
      while(newGeneration.size() < population) {
        newGeneration.add(new Snake(newGeneration.get(parent_index))); 
        parent_index += 1;
      }*/
    }
     
    fitnesses.clear(); 
    snakes = newGeneration;
    currentSnake = 0;
  };
  
  void writeGenInfo() {
    String cS = 'G'+str(currentGeneration)+" #"+str(currentSnake);
    rectMode(CORNERS);  
    fill(200);  
    rect(0, size[1]-25, 75, size[1]);  
    fill(50);  
    textSize(14);
    textAlign(CENTER, CENTER);
    text(cS, 0, size[1]-25, 75, size[1]);
    noFill();
  };
  
};