import traer.physics.*;
import processing.video.*;

MovieMaker mm;  // Declare MovieMaker object
ParticleSystem physics;
int SIZE=20; 
int DIST=57;
int MASS=15;
Particle p, p2;
int pn;
Particle anchor;
Spring s, s2;
String lines[];
String[] list;
String delim=";";
Float g=0.0;
boolean rec=false;
void setup()
{
  lines = loadStrings("liste.csv");
  size( 750, 600 );
  smooth();
  fill( 0 );
  ellipseMode( CENTER );
  physics = new ParticleSystem( 1, 0.05 );
  //Obtain maximum Number of Particles
  for (int i=0; i < lines.length; i++) {
    list = split(lines[i], delim);

    if (pn<int(list[0])){
      pn=int(list[0]);
    }
    if (pn<int(list[1])) {
      pn=int(list[1]);
    }

  } 
  for (int i=1;i<pn+2;i++) {

    p = physics.makeParticle( MASS, width/2+sin(i), height/2+cos(i), 0);
  }
  physics.getParticle(1).makeFixed();
  for (int i=1;i<pn+1;i++) {
      p2=physics.getParticle(i);       
    for (int j=1;j<pn+1;j++) {
      p=physics.getParticle(j);
      physics.makeAttraction( p , p2, -MASS/2, DIST/2); 
    }
  }  

  for (int i=0; i < lines.length; i++) {
    list = split(lines[i], delim);  
    p=physics.getParticle(int(list[0]));
    p2=physics.getParticle(int(list[1]));
    s = physics.makeSpring(p, p2 , 0.6, 1, DIST );
  }

  physics.setGravity(g);
}

void mouseDragged()
{
  anchor.makeFixed();
  anchor.position().set( mouseX, mouseY, 0 );

}

void mousePressed()
{
  anchor=physics.getParticle(1);
  for (int i=1;i<pn+1;i++) {
    p=physics.getParticle(i);
    if (abs(p.position().x()-mouseX)<SIZE) {
      if (abs(p.position().y()-mouseY)<SIZE) {
        anchor=p;
      }
    }
  }
}

void keyPressed() {
  if (key == 'r') {
    for (int i=1;i<pn+1;i++) {
      p=physics.getParticle(i);
      p.makeFree(); 
    }
  }
  if (key == 'f') {
    for (int i=1;i<pn+1;i++) {
      p=physics.getParticle(i);
      p.makeFixed(); 
    }
  }
  if (key == 'd') {
    
    for (int i=1;i<pn+1;i++) {
      p=physics.getParticle(i);
      p.makeFree(); 
      p.position().set(width/2+sin(i), height/2+cos(i), 0  );
    }
    physics.getParticle(1).makeFixed();
  }  
}


void mouseReleased()
{

  anchor.makeFixed(); 
}

void draw()
{
  String s;
  physics.tick();

  strokeCap(ROUND);
  background( color(0, 0, 0) );

  for (int i=0; i < lines.length; i++) {
    list = split(lines[i], delim); 
    p= physics.getParticle(int(list[0]));
    p2=physics.getParticle(int(list[1]));      
    stroke(getC(list[2]), 90);
    strokeWeight(4);
    line( p.position().x(), p.position().y(), p2.position().x(), p2.position().y() );

  }

  for (int i=1;i<pn+1;i++) {
    p=physics.getParticle(i);
    strokeWeight(2);
    
    s=getType(i);
    if (s.equals("")){

    } 
    else { 
      fill(getC(s), 55);
      stroke(getC(s), 100);
      if (p.isFixed()) {
        fill(getC(s), 100);
      }
      ellipse( p.position().x(), p.position().y(), SIZE, SIZE );
      if (p.position().x()>=width) {
        revV(p);
      }
      if (p.position().y()>=height) {
        revV(p);
      }
      if (p.position().x()<=0) {
        revV(p);
      }
      if (p.position().y()<=0) {
        revV(p);
      }       
    }
  } 
}

String getType(int index) {
  boolean bFound=false;  
  String s ="";
  int i=0;
  while((i<lines.length) && (s.equals(""))) {
    list = split(lines[i], delim);
    if (int(list[0])==index) {
      s =list[4];
    }
    if (int(list[1])==index) {
      s= list[6];
    }
    i++;
  }

  return s;
}

void revV(Particle p) {
  p.velocity().set( -p.velocity().x(), -p.velocity().y(), -p.velocity().z()  );
}

int getC(String sIN) {
  int c=color(113, 82, 63);

  if (sIN.equals("Projects")) {
    c= color(255, 255, 255);
  }  
  if (sIN.equals("VBE")) {
    c= color(255, 255, 255);
  }   
  if (sIN.equals("Table")) {
    c= color(255, 0, 0);
  }
  if (sIN.equals("Query")) {
    c= color(200, 155, 0);
  }    
  if (sIN.equals("Recordset")) {
    c= color(255, 0, 0);
  }  

  if (sIN.equals("Module")) {
    c= color(0, 255, 0);
  } 
  if (sIN.equals("Form")) {
    c= color(0, 255, 0);
  }
  if (sIN.equals("Call")) {
    c= color(0, 155, 0);
  }
  if (sIN.equals("Code")) {
    c= color(0, 155, 155);
  }
  if (sIN.equals("Execute")) {
    c= color(0, 155, 0);
  }  



  return c;
}