float a;
float dy = 16;
int k = 0;
int detectorAngles = 2;
int ellipsex;
int ellipsey;
int ellipseflag = 0;
int numvertices = 3;
int inc = 0, vertinc = 0;
int numRays;
float[] polydata = new float[20]; // max of 10 vertices
int[][] testRay;
//Reconstruction[] backProjection = new Reconstruction[detectorAngles];
Reconstruction[] backProjection = new Reconstruction[10];
int buttonOffsetX = 256;
int buttonOffsetY;
int mouseCounter = 0;
int selectedBtnP, selectedBtnD;

color currentcolor;
RectButton prect1, prect2, prect3, prect4, prect5, prect6, prect7, prect8;
RectButton drect1, drect2, drect3, drect4, drect5, drect6, drect7, drect8;

boolean locked = false;

void setup() 
{
  size(1536, 512);
  stroke(255);
  smooth();
  numRays = height;
  a = height;
//  detectorAngles = 9;
//  numvertices = 3;
//  testRay = new int[detectorAngles][numRays+1];
  testRay = new int[9][numRays+1];
  background(0);
  
  color baseColor = color(102);
  currentcolor = baseColor;

  PFont font;
  font = loadFont("ArialMT-12.vlw"); 
  textFont(font, 12);
  // Define and create rectangle button
  color buttoncolor = color(102);
  color highlight = color(51); 
  buttonOffsetY = 4;
  prect1 = new RectButton(buttonOffsetX, buttonOffsetY, 20, buttoncolor, highlight,3);
  prect2 = new RectButton(buttonOffsetX+24, buttonOffsetY, 20, buttoncolor, highlight,4);
  prect3 = new RectButton(buttonOffsetX+48, buttonOffsetY, 20, buttoncolor, highlight,5);
  prect4 = new RectButton(buttonOffsetX+72, buttonOffsetY, 20, buttoncolor, highlight,6);
  prect5 = new RectButton(buttonOffsetX+96, buttonOffsetY, 20, buttoncolor, highlight,7);
  prect6 = new RectButton(buttonOffsetX+120, buttonOffsetY, 20, buttoncolor, highlight,8);
  prect7 = new RectButton(buttonOffsetX+144, buttonOffsetY, 20, buttoncolor, highlight,9);
  prect8 = new RectButton(buttonOffsetX+168, buttonOffsetY, 20, buttoncolor, highlight,10);

  buttonOffsetY = 28;
  drect1 = new RectButton(buttonOffsetX, buttonOffsetY, 20, buttoncolor, highlight,2);
  drect2 = new RectButton(buttonOffsetX+24, buttonOffsetY, 20, buttoncolor, highlight,3);
  drect3 = new RectButton(buttonOffsetX+48, buttonOffsetY, 20, buttoncolor, highlight,4);
  drect4 = new RectButton(buttonOffsetX+72, buttonOffsetY, 20, buttoncolor, highlight,5);
  drect5 = new RectButton(buttonOffsetX+96, buttonOffsetY, 20, buttoncolor, highlight,6);
  drect6 = new RectButton(buttonOffsetX+120, buttonOffsetY, 20, buttoncolor, highlight,7);
  drect7 = new RectButton(buttonOffsetX+144, buttonOffsetY, 20, buttoncolor, highlight,8);
  drect8 = new RectButton(buttonOffsetX+168, buttonOffsetY, 20, buttoncolor, highlight,9);
}

void mouseReleased() {
  int m;
    mouseCounter = mouseCounter + 1;
//    print("mouseCounter = "+mouseCounter+"\n");
    if (mouseCounter > 2 && vertinc <= numvertices){
        polydata[2*vertinc] = mouseX-width/6;
        polydata[2*vertinc+1] = mouseY-height/2;
        vertinc = vertinc + 1;
        if(vertinc > 1){
            stroke(255);
          for (m = 1; m<vertinc;m++){
            line(polydata[2*(m-1)]+width/6,polydata[2*(m-1)+1]+height/2,polydata[2*m]+width/6,polydata[2*m+1]+height/2);
          }
        }
    }
    if (vertinc == numvertices ){
      ellipseflag = 1;
    }  
}

void Polygon(){
    beginShape();
    for(int m = 0; m <numvertices; m=m+1) {
         vertex(polydata[2*m],polydata[2*m+1]);
    }
    endShape(CLOSE);
}

void draw() 
{
  float angle;
  float xtest, ytest;
  int insidetest = 0;
  int jj;
  int numRays = 2*int(height) / int(dy);

  fill(0);
  rect(0,0,width/3,height);
  line(2*width/3,0,2*width/3,height);

  update(mouseX, mouseY);
  fill(255);
  text("Number of polygon vertices",90,20);
  prect1.display(selectedBtnP);
  prect2.display(selectedBtnP);
  prect3.display(selectedBtnP);
  prect4.display(selectedBtnP);
  prect5.display(selectedBtnP);
  prect6.display(selectedBtnP);
  prect7.display(selectedBtnP);
  prect8.display(selectedBtnP);
  fill(255);
  text("Number of detector angles",90,42);
  drect1.display(selectedBtnD);
  drect2.display(selectedBtnD);
  drect3.display(selectedBtnD);
  drect4.display(selectedBtnD);
  drect5.display(selectedBtnD);
  drect6.display(selectedBtnD);
  drect7.display(selectedBtnD);
  drect8.display(selectedBtnD);

//  print("numvertices = "+numvertices+", detectorAngles = "+detectorAngles+"\n");
  if (ellipseflag==0){
//    print("Use the mouse to select the ellipse size, referenced to top left corner\n");
    fill(100,100,100,10);
    ellipse(width/6,height/2,192,192);
  }

  if (mouseCounter == numvertices + 3) {
      Polygon();
      mouseCounter = 0;
  }
  if (ellipseflag==1){   
      if (k == detectorAngles)
      {
        k = 0;
        noLoop();
      }
    if(k<detectorAngles){
      translate(width/6, height/2);
      fill(255);
      Polygon();
      angle = k*PI/detectorAngles;
      rotate(angle);
      line(-width/12, a/4, width/12, a/4);  
      for (int g = 0; g < 64; g=g+1){
        xtest = -width/12 + g/64.0 * width/6;
        ytest = a/4;      
        insidetest=insidetest + insidePolygon(xtest,ytest,polydata,angle);        
      }
    }
    testRay[k][inc] = insidetest;
    a = a - dy;
    inc = inc + 1;
    if (a < -height) { 
      a = height; 
      inc = 0;
      if(k<=detectorAngles){ 
        backProjection[k] = new Reconstruction();
        rotate(-k*PI/detectorAngles);
        translate(width/3, 0*height/2);
        backProjection[k].display(0);
        rotate(-k*PI/detectorAngles);
        translate(width/3, 0*height/2);
        backProjection[k].display(1);
      }
      k = (k + 1);
    }
  }
}

int insidePolygon(float x,float y, float poly[],float rotangle) {
  int id1,id2;
  int i, j, c = 0;
  float[] polyr = new float[2*numvertices];
  
  for(int n = 0; n < numvertices; n=n+1){
    polyr[2*n] = poly[2*n]*cos(-rotangle) - poly[2*n+1]*sin(-rotangle);
    polyr[2*n+1] = poly[2*n]*sin(-rotangle) + poly[2*n+1]*cos(-rotangle);
  }    

  for (i = 0, j = numvertices-1; i < numvertices; j = i++) {
    id1=i*2;
    id2=j*2;
    if ((((polyr[id1+1] <= y) && (y < polyr[id2+1])) || ((polyr[id2+1] <= y) && (y < polyr[id1+1]))) &&
	     (x < (polyr[id2] - polyr[id1]) * (y - polyr[id1+1]) / (polyr[id2+1] - polyr[id1+1]) + polyr[id1]))
	     c = (c+1)%2;
  }
  if(c==1){ 
    stroke(255,0,0);
    fill(0,0,0);
    ellipse(x,y,4,4);
    stroke(255);
    return 2;
  } else {
    return 0;
  }
}


void update(int x, int y)
{
  if(locked == false) {
    prect1.update();
    prect2.update();
    prect3.update();
    prect4.update();
    prect5.update();
    prect6.update();
    prect7.update();
    prect8.update();
    drect1.update();
    drect2.update();
    drect3.update();
    drect4.update();
    drect5.update();
    drect6.update();
    drect7.update();
    drect8.update();
  } 
  else {
    locked = false;
  }

  if(mousePressed) {
    if(prect1.pressed()) {
      currentcolor = prect1.basecolor;
      numvertices = 3;
    } 
    else if(prect2.pressed()) {
      currentcolor = prect2.basecolor;
      numvertices = 4;
    }
    else if(prect3.pressed()) {
      currentcolor = prect3.basecolor;
      numvertices = 5;
    }
    else if(prect4.pressed()) {
      currentcolor = prect4.basecolor;
      numvertices = 6;
    }
    else if(prect5.pressed()) {
      currentcolor = prect5.basecolor;
      numvertices = 7;
    }
    else if(prect6.pressed()) {
      currentcolor = prect6.basecolor;
      numvertices = 8;
    }
    else if(prect7.pressed()) {
      currentcolor = prect7.basecolor;
      numvertices = 9;
    }
    else if(prect8.pressed()) {
      currentcolor = prect8.basecolor;
      numvertices = 10;
    }
    if(drect1.pressed()) {
      currentcolor = drect1.basecolor;
      detectorAngles = 2;
  } 
    else if(drect2.pressed()) {
      currentcolor = drect2.basecolor;
      detectorAngles = 3;
    }
    else if(drect3.pressed()) {
      currentcolor = drect3.basecolor;
      detectorAngles = 4;
    }
    else if(drect4.pressed()) {
      currentcolor = drect4.basecolor;
      detectorAngles = 5;
    }
    else if(drect5.pressed()) {
      currentcolor = drect5.basecolor;
      detectorAngles = 6;
    }
    else if(drect6.pressed()) {
      currentcolor = drect6.basecolor;
      detectorAngles = 7;
    }
    else if(drect7.pressed()) {
      currentcolor = drect7.basecolor;
      detectorAngles = 8;
    }
    else if(drect8.pressed()) {
      currentcolor = drect8.basecolor;
      detectorAngles = 9;
    }
  }
}