int WIDTH=799;
int HEIGHT=490;
int BLOB_R=5;
float ECOL=0.9;
float EDRAG=0.999;
float EG=0.0;
int N;
int NS;
long CT=0;
int MODE=0;

float[] PX = new float[4];
float[] PY = new float[4];
float[] PX_OLD = new float[4];
float[] PY_OLD = new float[4];

int STICKA[] = new int[5];
int STICKB[] = new int[5];
float STICKLENGTH[] = new float[5];


boolean RUNNING=true;

boolean GRAVITY=false;

void setup() {
  size(800, 550, P2D);
  reset();
}

void draw() {

background(0,0,0,0);
noFill();
stroke(255);
rect(0,0,WIDTH,HEIGHT);

renderbuttons();
if(CT++>3000) {reset(); RUNNING=false;} //fuze

if (RUNNING) {
updatepoints();
renderpoints();
}


}

void renderpoints() {

stroke(255,0,0);
for(int i=0;i<NS;i++) {
line(PX[STICKA[i]],PY[STICKA[i]],PX[STICKB[i]],PY[STICKB[i]]); 
}
  
  
stroke(164,164,164); 
fill(255,255,0);
for (int i=0;i<N;i++)
{
ellipse(PX[i],PY[i],BLOB_R*2,BLOB_R*2);  
}
  
}

void updatepoints() {
 
 float VX,VY;

for (int i=0;i<N;i++)
{

  
  VX=(PX[i]-PX_OLD[i])*EDRAG;
  VY=(PY[i]-PY_OLD[i])*EDRAG;
  PX_OLD[i]=PX[i];
  PY_OLD[i]=PY[i];
  PX[i]+=VX;
  PY[i]+=VY;
  PY[i]+=EG;


if (PX[i]>(WIDTH-BLOB_R)) 
 {
   PX[i]=(WIDTH-BLOB_R);
   PX_OLD[i]=(PX[i]+(VX*ECOL));
 }
 else if (PX[i]<BLOB_R)
  {
    PX[i]=BLOB_R;
    PX_OLD[i]=(PX[i]+(VX*ECOL));
  }
  
  if (PY[i]>(HEIGHT-BLOB_R)) 
 {
   PY[i]=(HEIGHT-BLOB_R);
   PY_OLD[i]=(PY[i]+(VY*ECOL));
 }
 else if (PY[i]<BLOB_R)
  {
    PY[i]=BLOB_R;
    PY_OLD[i]=(PY[i]+(VY*ECOL));
  }
  
} 

}

void reset() {

//always get first point  
PX[0]=random(50,WIDTH-50);
PY[0]=random(50,100);
PX_OLD[0]=PX[0]+(random(-150,150)/10);
PY_OLD[0]=PY[0]+(random(-150,150)/10);

//then add at least two corners
for(int i=1;i<=3;i++) {
PX[i]=PX[0]+random(-30,30);
PY[i]=PY[0]+random(-30,30);
PX_OLD[i]=PX[i]+(random(-50,50)/10);
PY_OLD[i]=PY[i]+(random(-50,50)/10);
}




  switch(MODE) {
  case 0: N=2;NS=1;
   STICKA[0]=0;STICKB[0]=1;STICKLENGTH[0]=calclength(0,1);
   break;
  
  case 1: N=3;NS=3;
   STICKA[0]=0;STICKB[0]=1;STICKLENGTH[0]=calclength(0,1);
   STICKA[1]=0;STICKB[1]=2;STICKLENGTH[1]=calclength(0,2);
   STICKA[2]=1;STICKB[2]=2;STICKLENGTH[2]=calclength(1,2);
   break;
  
  case 2: N=4;NS=4;
   STICKA[0]=0;STICKB[0]=1;STICKLENGTH[0]=calclength(0,1);
   STICKA[1]=0;STICKB[1]=2;STICKLENGTH[1]=calclength(0,2);
   STICKA[2]=1;STICKB[2]=3;STICKLENGTH[2]=calclength(1,3);
   STICKA[3]=2;STICKB[3]=3;STICKLENGTH[3]=calclength(2,3);
   break;
  
  case 3: N=4;NS=5;
   STICKA[0]=0;STICKB[0]=1;STICKLENGTH[0]=calclength(0,1);
   STICKA[1]=0;STICKB[1]=2;STICKLENGTH[1]=calclength(0,2);
   STICKA[2]=1;STICKB[2]=3;STICKLENGTH[2]=calclength(1,3);
   STICKA[3]=2;STICKB[3]=3;STICKLENGTH[3]=calclength(2,3);
   STICKA[4]=0;STICKB[4]=3;STICKLENGTH[4]=calclength(0,3);
   break;
  } 


RUNNING=false;
CT=0;
}

float calclength(int p1, int p2) {
float dx2,dy2;
dx2=pow(PX[p1]-PX[p2],2);
dy2=pow(PY[p1]-PY[p2],2);
return sqrt(dx2+dy2);
}

void renderbuttons() {
  textSize(16);
  textAlign(CENTER, CENTER);
  stroke(164,164,164);
  String label;
  
  if(RUNNING) {fill(128,110,0); label="RESET";} else {fill(100,100,100); label="Start";}
  rect(0,510,100,30,4);
  fill(255);
  text(label,50,525);
  
  if(GRAVITY) {fill(128,110,0); label="Gravity"; EG=0.5;} else {fill(100,100,100); label="No Gravity"; EG=0.0;}
  rect(110,510,100,30,4);
  fill(255);
  text(label,160,525);
 
 String l;
 for (int i=0;i<=3;i++)
 {

  switch(i) {
  case 0: l="Bar";break;
  case 1: l="Triangle";break;
  case 2: l="Quad";break;
  case 3: l="Braced";break;
  } 
 
  if (i==MODE) {fill(128,110,0);} else {fill(100,100,100);}
  rect(400+(i*90),510,80,30,4);
  fill(255);
  text(l,440+(i*90),525);
 }  
 
  textSize(20);
  textAlign(RIGHT, CENTER);
  stroke(164,164,164);
  text("Shape: ",400,525);
}

void mouseClicked() {
 if ( mouseY>=510 && mouseY<=540 && mouseX>0 && mouseX<100) {RUNNING=!RUNNING; if(!RUNNING) {reset();}}
 if ( mouseY>=510 && mouseY<=540 && mouseX>110 && mouseX<210) {GRAVITY=!GRAVITY;}
 if ( mouseY>=510 && mouseY<=540 && mouseX>400 && mouseX<750) {MODE=int((mouseX-400)/90); reset();}
}