// Evolutionary tree example program - (c) axel Kilian 2004
// feel free to use the code as long as you keep the copyrigh note at the top of the code as it is here 
//
//

BFont metaBold;
float[][] population;
float[][] toptenpopulation;
int genlength;
int popsize;
float mapx;
float mapy;
float rotatepos2;
float rotatepos;
float targetvol;
float weightV;
int topten;
float weightX;
float targetX;
float weightY;
float targetY;
float weightZ;
float targetZ;
int genomelength;
int readsequence;

void setup() {
  colorMode(RGB);
  mapx = 0.7;
  mapy = 0.0;
  popsize = 100;
  genomelength = 27;
  topten = (int)(popsize/10);
  readsequence = 2;
  population = new float[popsize][genomelength];
  toptenpopulation = new float[topten][genomelength];

mapx=0.7;

  for (int i = 0;i<topten;i++) {
    for (int j = 0;j<genomelength;j++) {
      toptenpopulation[i][j] = 2;
    }
  }
  createPop();
  for (int i = 0;i<(popsize);i++)
  {
    population[i][0] = 1;
    population[i][1] = 1;

    population[i][2] = 1;
    population[i][3] = 1;

    population[i][4] = 1;
    population[i][5] = 1;

    population[i][6] = 1;
    population[i][7] = 1;

    population[i][8] = 1;
    population[i][9] = 1;

    population[i][10] = 1;
    population[i][11] = 1;

    population[i][12] = 1;
    population[i][13] = 1;

    population[i][14] = 1;
    population[i][15] = 1;

    population[i][16] = 1;
    population[i][17] = 1;

    population[i][18] = 1;
    population[i][19] = 1;

    population[i][20] = 1;
    population[i][21] = 1;

    population[i][22] = 1;
    population[i][23] = 1;

    population[i][24] = 1;
    population[i][25] = 1;

    population[i][26] = 0;
  }

  targetvol = 1500;
  weightV = 1;

  size(800, 800);
  background(0);
  metaBold = loadFont("Meta-Bold.vlw.gz");
  textFont(metaBold, 5);
  weightX=0;
}

void loop() {

  background(255);
  fill(160,20,20);
  rect(20,40,100,10);
  rect(20,60,100,10);

  fill(20,160,20);
  rect(20,80,100,10);
  rect(20,100,100,10);

  fill(20,20,160);
  rect(20,120,100,10);
  rect(20,140,100,10);

  fill(160,160,160);
  rect(20,160,100,10);
  rect(20,180,100,10);

  textFont(metaBold, 20);
  fill(160,30,10);
  text("X",10,50);
  text(targetX,120,50);
  text(weightX,120,70);
  fill(10,160,10);
  text("Y",10,90);
  text(targetY,120,90);
  text(weightY,120,110);
  fill(10,20,160);
  text("Z",10,130);
  text(targetZ,120,130);
  text(weightZ,120,150);

  fill(160,160,160);
  text("V",10,170);
  text(targetvol,120,170);
  text(weightV,120,190);

  textFont(metaBold, 5);
  color boxC = color(0,0,0);

  for (int i = 0;i<(popsize/10);i++)
  {
    createPop();
    for (int j = 0;j<10;j++)
    {
      push();
      translate(420,400,100);
      rotateX(mapx);
      rotateZ(mapy);
      stroke(204, 102, 0);
      line(-293,-235,0,230,-235,0);
      line(-293,-287,0,230,-287,0);
      line(-293,-235,0,-293,-287,0);
      line(230,-235,0,230,-287,0);
      stroke(0.1,0.2,0.2);
      translate((j*52)-(5*52),(i*52)-((popsize/20)*52),0);
      fill(0.9,0.5,0.9);
      if (((i*10)+j)<topten) {
        String sd = nf(population[(i*10)+j][genomelength-1], 3, 2);
        text(sd, -5, -10);       
        boxC = color(100,10,10);
      }
      else boxC = color(200,200,200);
      drawbox(population[(i*10)+j],boxC,(i*10)+j);
      pop();
    }
  }

  /*
  for (int i = 0;i<topten;i++)
  {
    push();
    translate(300,300,400);
    rotateX(mapx);
    rotateY(mapy);
    translate(i*22,-50,0);
    //fill(0.9,0.5,0.9);
    //String sd = nf(toptenpopulation[i][9], 3, 2);
    //text(sd, -5, -13);
    drawbox(population[i]);
    //box(10);
    pop();
  }
  */

}

void mouseDragged() {

  if ((mouseX<120) && (mouseX>20) && (mouseY<50) && (mouseY>40)) {
    targetX += (mouseX-pmouseX)*0.1 ;
    if (targetX<0) targetX=0;

  }
  else if ((mouseX<120) && (mouseX>20) && (mouseY<70) && (mouseY>60)) {
    weightX += (mouseX-pmouseX)*1 ;
    if (weightX<0) weightX=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<90) && (mouseY>80)) {
    targetY += (mouseX-pmouseX)*0.1 ;
    if (targetY<0) targetY=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<110) && (mouseY>100)) {
    weightY += (mouseX-pmouseX)*1 ;
    if (weightY<0) weightY=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<130) && (mouseY>120)) {
    targetZ += (mouseX-pmouseX)*0.1 ;
    if (targetZ<0) targetZ=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<150) && (mouseY>140)) {
    weightZ += (mouseX-pmouseX)*1 ;
    if (weightZ<0) weightZ=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<170) && (mouseY>160)) {
    targetvol += (mouseX-pmouseX)*4 ;
    if (targetvol<0) targetvol=0;
  }

  else if ((mouseX<120) && (mouseX>20) && (mouseY<190) && (mouseY>180)) {
    weightV += (mouseX-pmouseX)*1 ;
    if (weightV<0) weightV=0;
  }

   else {
    rotatepos2 += (mouseX-pmouseX);
    rotatepos -= (mouseY-pmouseY);
    mapx = (rotatepos/float(height) * TWO_PI);
    mapy = (rotatepos2/float(height) * TWO_PI);
  }
}

void createPop() {
  for (int i = 0;i<popsize;i++)
  {
    for(int s=0;s<genomelength;s++) {
      population[i][s] = (int)random(2);
    }

    //println(population[i][0]+ " " + population[i][1]+ " " +population[i][2]+ " " +population[i][3]+ " " +population[i][4]+" " +population[i][5]+" " +population[i][6]+" " +population[i][7]+" " +population[i][8]);
    //population[i][9] =
    checkfitness(i);
  }
  float testtemp=0;
  if (frame!=0) {
    for(int s=0;s<topten;s++)
    {
      for(int t=0;t<genomelength;t++)
      {
        population[s][t] = toptenpopulation[s][t];
      }
    }
  }
  quicksort(0,popsize-1);

  for(int s=0;s<topten;s++)
  {
    for(int t=0;t<genomelength;t++)
    {
      toptenpopulation[s][t] = population[s][t];
    }
  }
  crossover();
  pointMutate();
}

void drawbox(float[] indiv, color theC, int num) 
{
  int[] dx = new int[(genomelength-1)/2];
  for(int g =0;g<((genomelength-1)/2);g++) {
    for(int i=(g*readsequence);i<((((g+1)*readsequence)));i++) {
      dx[g] += (int)indiv[i]*pow(2,(i-(g*readsequence)));
    }
  }
  makeboxes(dx,num);
}

void drawit(float stepB, float sizeB) {
  stroke(10,100,10);
  fill(10,80,10);
  //box(3);

  translate(0,0,stepB);
  box(sizeB);
  translate(0,0,stepB);
  box(sizeB);
  translate(0,0,stepB);
  box(sizeB);
  translate(0,0,stepB);
  box(sizeB);
}

void makeboxes(int[] instr, int b)
{
  float score=0;
  float sizeB = 3;
  float stepB = 4;
  int l = 4;
  fill(70,10,10);
  box(5);push();
  rotateY(0);
  drawit(stepB,sizeB);
  score+=9;
  for(int i =1;i<instr[0]+1;i++)
  {
    push();
    if((instr[0]-1)!=0) {
      rotateZ( ((3*PI)/(instr[0]-1)) *i);
      rotateY((HALF_PI/(instr[0]-1))*(i-1)-(HALF_PI/2.0));
    }
    sizeB = 2.5;
    stepB = 3.5;
    drawit(stepB,sizeB);
    score+=9;
    for(int w=0;w<instr[i];w++)
    {
      push();
      sizeB = 2.0;
      stepB = 3.0;
      if((instr[i]-1)!=0) {
        rotateZ( ((PI/2.0)/(instr[i]-1)) *w);
        rotateY((HALF_PI/(instr[i]-1))*(w)-(HALF_PI/2.0));
      }
      drawit(stepB,sizeB);
      score+=9;
      for(int q=0;q<instr[l];q++)
      {
        push();
        sizeB = 1.5;
        stepB = 2;
        if ((instr[l]-1)!=0) {
          rotateZ( ((PI/2.0)/(instr[l]-1)) *q);
          rotateY((HALF_PI/(instr[l]-1))*(q)-(HALF_PI/2.0));
        }
        drawit(stepB,sizeB);
        score+=9;
        pop();
      }
      l+=1;
      pop();
    }
    pop();
  }
  pop();
  population[b][26] = abs((targetvol - score))*weightV;
  //println(population[b][26]);
}

void crossover() {
  for(int i = 0;i<topten;i+=2) {
    int mutationpoint = (int)random(0,genomelength-1);
    float[] one_L = new float[mutationpoint];
    float[] one_R = new float[genomelength-1-mutationpoint];
    float[] two_L = new float[mutationpoint];
    float[] two_R = new float[genomelength-1-mutationpoint];
    //println(mutationpoint + " 1 " + (9-mutationpoint));

    for(int o = 0; o< mutationpoint;o++) {
      one_L[o] = toptenpopulation[i][o];
      two_L[o] = toptenpopulation[i+1][o];
    }
    //println(mutationpoint + " 2 " + (9-mutationpoint));
    for(int o = mutationpoint; o< genomelength-1;o++) {
      one_R[o-mutationpoint] = toptenpopulation[i][o];
      two_R[o-mutationpoint] = toptenpopulation[i+1][o];
    }
    for(int o = 0; o< mutationpoint;o++) {
      toptenpopulation[i][o] = two_L[o];
      toptenpopulation[i+1][o] = one_L[o];
    }
    for(int o = mutationpoint; o<genomelength-1;o++) {
      toptenpopulation[i][o] = two_R[o-mutationpoint];
      toptenpopulation[i+1][o] = one_R[o-mutationpoint];
    }
  }

  for(int w = 0;w<topten;w+=2) {
    int[] dx = new int[(genomelength-1)/2];
    for(int g =0;g<((genomelength-1)/2);g++) {
      for(int i=(g*readsequence);i<((((g+1)*readsequence)));i++) {
        dx[g] += (int)toptenpopulation[w][i]*pow(2,(i-(g*readsequence)));
      }
    }
    //toptenpopulation[i][26] = abs((targetX-dx)*weightX)+abs((targetY-dy)*weightY)+abs((targetZ-dz)*weightZ)+(float)abs(((targetvol-(dx*dy*dz))*weightV));
  }
}

void pointMutate() {
  for(int i = 0;i<topten;i++) {
    if (5==(random(0,25))) {
      int mutationpoint = (int)random(0,25);
      toptenpopulation[i][mutationpoint] = (toptenpopulation[i][mutationpoint]+1)%1;
    }

  }
}

void checkfitness2(int check) {
  for(int w = 0;w<topten;w+=2) {
    int[] dx = new int[(genomelength-1)/2];
    for(int g =0;g<((genomelength-1)/2);g++) {
      for(int i=(g*readsequence);i<((((g+1)*readsequence)));i++) {
        dx[g] += (int)population[check][i]*pow(2,(i-(g*readsequence)));
      }
    }
    // population[check][genomelength-1] = abs((targetX-dx)*weightX)+abs((targetY-dy)*weightY)+abs((targetZ-dz)*weightZ)+(float)abs(((targetvol-(dx*dy*dz))*weightV));

  }
}

void checkfitness(int check) {
 int[] dx = new int[(genomelength-1)/2];
  for(int g =0;g<((genomelength-1)/2);g++) {
    for(int i=(g*readsequence);i<((((g+1)*readsequence)));i++) {
      dx[g] += (int)population[check][i]*pow(2,(i-(g*readsequence)));
    }
  }
  int l=0;
float score=0;
  score+=9;
  for(int i =1;i<dx[0]+1;i++)
  {
    score+=9;
    for(int w=0;w<dx[i];w++)
    {
      score+=9;
      for(int q=0;q<dx[l];q++)
      {
        score+=9;
     
      }
      l+=1;
    }
  }
  population[check][26] = abs((targetvol - score))*weightV;
}



void quicksort( int first, int last ) {
  int lo  = first;
  int hi  = last;
  int mid = ( first + last ) / 2;

  if( first < last ) {
    do {
      while ( ( lo < last  ) && ( compare( lo, mid ) < 0 ) ) ++lo;
      while ( ( hi > first ) && ( compare( hi, mid ) > 0 ) ) --hi;
      if ( lo <= hi ) {
        swap( lo, hi );
        ++lo;
        --hi;
      }
    } while( lo < hi );
    if ( first < hi   ) quicksort( first, hi );
    if ( lo    < last ) quicksort( lo,  last );
  }
}

void swap( int ia, int ib ) {
  float[] temp = new float[genomelength];
  for(int i = 0; i<genomelength;i++)
  {
    temp[i] = population[ia][i];
  }
  for(int i = 0; i<genomelength;i++)
  {
    population[ia][i] = population[ib][i];
  }
  for(int i = 0; i<genomelength;i++)
  {
    population[ib][i] = temp[i] ;
  }
}

int compare( int ia, int ib ) {
  float alpha = population[ia][genomelength-1];
  float omega = population[ib][genomelength-1];
  return ( alpha > omega ) ? 1 : -1;
}