import processing.opengl.*;
import anar.*;





/*
 * Example for Anar library by Guillaume LaBelle + Julien Nembrini
 * http://anar.ch
 */


Obj myObject;

Pts line;
Pts copy;

void setup(){

  // size(screen.width,screen.height, P3D);
    size(1000,500,OPENGL);
  // size(1000,500, P3D);
  background(55,33,6);

  myObject = new Obj();


  line = new Pts();

  Pt o = Anar.PtNull(0,0,0);
  // Pt a = Anar.PtNull(40,0,40);
  Pt b = Anar.PtNull(80,0,0);
  // Pt c = Anar.PtNull(80,0,80);
  // Pt d = Anar.PtNull(80,100,80);

  line.add(o);
  // line.add(a);
  line.add(b);
  // line.add(c);
  // line.add(d);

  // for (int i=0; i<100; i++) {
  // line.add(Anar.PtRnd(100,100,100));
  // }


  myObject.add(line);

  Anar.init(this);

  Anar.drawAxis(true);
  Anar.sliders(b);
  Anar.camTarget(myObject);

  copy();
  myObject.add(copy);


  Pts.globalRender = new RenderPtsAll();
  Pt.globalRender = new RenderPtShapeOriented(new AColor(100),new AColor(255,0,0,150),Anar.scene);


}


void subdivide(){

  // find midpoints of edges
  Pts midPts = new Pts();
  for (int i = 0; i<line.numOfPts()-1; i++){
    Pts tmp = new Pts(line.pt(i),line.pt(i+1));
    midPts.add(new PtBary(tmp));
  }


  // average old points
  Pts newPts = new Pts();
  for (int i = 1; i<line.numOfPts()-1; i++){
    Pts tmp = new Pts(midPts.pt(i-1),midPts.pt(i));
    newPts.add(new PtBary(tmp));
  }

  // construct new line
  Pts newLine = new Pts();

  newLine.add(line.pt(0));
  for (int i = 0; i<newPts.numOfPts(); i++){
    newLine.add(midPts.pt(i));
    newLine.add(newPts.pt(i));
  }
  newLine.add(midPts.ptEnd());
  newLine.add(line.ptEnd());

  line = newLine;

  println(line.numOfPts());

  myObject = new Obj();
  myObject.add(line);
  Anar.slidersReset();
  Anar.sliders(line);

}


/**
 * creates a copy of line loosing parametric relationships
 */
void copy(){
  copy = new Pts();
  for (int i = 0; i<line.numOfPts(); i++){
    Pt copyP = Anar.Pt(line.pt(i).x(),line.pt(i).y(),line.pt(i).z());
    copy.add(copyP);
  }
  copy.stroke(0,0,255);

  myObject.add(copy);
}


/**
 * updates the copy using euler
 */
void update(){
  float k = 10.0f;
  // assume constant length between original nodes
  float length = line.pt(0).length(line.pt(1));

  float m = 1/(float)copy.numOfPts();

  for (int i = 1; i<copy.numOfPts()-1; i++){

    Pt p = copy.pt(i);
    Pt pPrev = copy.pt(i-1);
    Pt pNext = copy.pt(i+1);

    Vertex sum = new Vertex(0,0,0);
    // compute force with previous node
    float f = k* (p.length(pPrev));
    // float f = k*(p.length(pPrev)-length);
    Vertex direct = (new Vertex(pPrev).minus(new Vertex(p))).normalize();
    direct = (Vertex)direct.multiply(f);

    sum = sum.plus(direct);

    // compute force with previous node
    f = k* (p.length(pNext));
    // f = k*(p.length(pNext)-length);
    direct = (Vertex) (new Vertex(pNext).minus(new Vertex(p))).normalize();
    direct = (Vertex)direct.multiply(f);

    sum = sum.plus(direct);

    // compute gravity with mass = 1/N
    Vertex grav = new Vertex(0,0, -9.81f*m);
    sum = sum.plus(grav);

    // euler step
    float h = 0.01f;
    sum = (Vertex)sum.multiply(h);

    p.set(p.x()+sum.x(),p.y()+sum.y(),p.z()+sum.z());

  }
}


void draw(){
  background(255);
  myObject.draw();
  update();
}


void keyPressed(){

  switch(key){
    case 'q':
      subdivide();
      copy();
    break;
    case ' ':
    break;
    case 'w':
    break;
    case 'm':
    break;
    case 'n':
    break;
    case 'e':
    break;
    case 'p':
      Scene.autoSeek = false;
    break;
    case 'o':
    break;
    case 'r':
    break;
    case 't':
    break;
    case 'a':
      TextIO.write( ((Object)this).getClass().getName()+".lsp",myObject.toAutocad());
    break;
    case 'b':
    break;
    case 'f':
      save("linearSubdiv.jpg");
    break;


  }

}