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


import java.util.*;



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


Sliders parameters;

Pt      a, b, c, d, e, f;

Obj     obj;

int     nIteration = 0;

void setup(){
    size(800,400,OPENGL);
  Anar.init(this);

  Scene.autoSeek = false;

  obj = new Obj();

  a = Anar.Pt( -30,30);
  b = Anar.Pt( -25,0);
  c = Anar.Pt( -30, -30);
  d = Anar.Pt(30, -30);
  e = Anar.Pt(25,0);
  f = Anar.Pt(30,30);

  initForm();
  initSpacePoints();

  Pts.globalRender = new RenderPtsAll();

}


void reset(){
  obj.faces.clear();
  obj.pts.ptList.clear();

  initForm();
  initSpacePoints();

  nIteration = 0;
}

void initForm(){


  Pts facePts = new Pts();
  facePts.add(a);
  facePts.add(b);
  facePts.add(c);
  facePts.add(d);
  facePts.add(e);
  facePts.add(f);
  Face f = new Face(facePts);

  obj.add(f);

  parameters = new Sliders(a);
  // println(parameters);

}

void initSpacePoints(){

  float maxX = 80;
  float maxY = 80;
  float maxZ = 400;

  for (int i = 0; i<200; i++){
    // obj.add(Anar.Pt( (float)Math.random()*2*maxX - maxX,
    // (float)Math.random()*2*maxY - maxY,
    // (float)Math.random()*maxZ));
    obj.add(Anar.Pt((float)Math.sin(i/(float)Math.PI)*maxX,(float)Math.cos(i/(float)Math.PI)*maxY,i/2));
  }

}

void spaceColonyStep(){

  float actionDist = 60;
  float deleteDist = 50;

  float coeff = 5;
  float zz = 5;


  // get last face
  Face f = obj.faces.get(obj.faces.size()-1);

  // make a translated copy
  Face newFace = new Face();

  Iterator it = f.iterator();
  while (it.hasNext()){
    Pt pp = (Pt)it.next();
    Pt ppp = Anar.Pt(pp.x(),pp.y(),pp.z());
    ppp.translate(0,0,zz);
    newFace.add(ppp);

  }
  obj.add(newFace);

  // now perform spaceColony attraction on all the vertices in the object
  // for each space point select nearest and attract it.
  Pts newSpacePoints = new Pts();

  ArrayList listT = new ArrayList();

  it = obj.pts.iterator();
  while (it.hasNext()){
    Pt spaceP = (Pt)it.next();
    boolean delete = false;

    int index = 0;

    ListIterator itF = obj.faces.listIterator();
    while (itF.hasNext()){
      f = (Face)itF.next();
      Iterator itP = f.iterator();
      while (itP.hasNext()){
        Pt faceP = (Pt)itP.next();
        // first round populate transform list
        if(listT.size()<index+1){
          listT.add(new AllTransform());

        }
        if(spaceP.length(faceP)<actionDist){

          ((AllTransform)listT.get(index)).add(new Translate(Anar.Pt( (spaceP.x()-faceP.x())/(float)Math.pow(spaceP.length(faceP),2),
              (spaceP.y()-faceP.y())/(float)Math.pow(spaceP.length(faceP),2),0)));
          if(faceP.length(spaceP)<deleteDist)
            delete = true;
        }

        index++;

      } // end while pt
    }// end while face

    if( !delete)
      newSpacePoints.add(spaceP);

  } // end while space point

  // now update transform on all points
  int index = 0;
  Iterator itF = obj.faces.iterator();
  while (itF.hasNext()){
    f = (Face)itF.next();
    Iterator itP = f.iterator();
    while (itP.hasNext()){
      Pt faceP = (Pt)itP.next();

      Pt v = ((AllTransform)listT.get(index)).toUnit().getVertex();
      faceP.translate(coeff*v.x(),coeff*v.y(),coeff*v.z());

      index++;

    } // end while pt
  }// end while face


  println("Space points "+newSpacePoints.numOfPts());

  Obj newObj = new Obj();

  itF = obj.faces.iterator();
  while (itF.hasNext()){
    newObj.add((Face)itF.next());
  }

  it = newSpacePoints.iterator();
  while (it.hasNext()){
    newObj.add((Pt)it.next());
  }

  obj = newObj;

}


void spaceColonyLastStep(){

  float actionDist = 60;
  float deleteDist = 50;

  float coeff = 1;
  float zz = 2;


  // get last face
  Face f = obj.faces.get(obj.faces.size()-1);

  // make a translated copy
  Face newFace = new Face();

  Iterator it = f.iterator();
  while (it.hasNext()){
    Pt pp = (Pt)it.next();
    Pt ppp = Anar.Pt(pp.x(),pp.y(),pp.z());
    ppp.translate(0,0,zz);
    newFace.add(ppp);

  }
  obj.add(newFace);

  // now perform spaceColony attraction on the vertices of the last facein the
  // object
  // for each space point select nearest and attract it.
  Pts newSpacePoints = new Pts();

  ArrayList listT = new ArrayList();

  it = obj.pts.iterator();
  while (it.hasNext()){
    Pt spaceP = (Pt)it.next();
    boolean delete = false;

    int index = 0;

    f = obj.faces.get(obj.faces.size()-1);
    Iterator itP = f.iterator();
    while (itP.hasNext()){
      Pt faceP = (Pt)itP.next();
      // first round populate transform list
      if(listT.size()<index+1){
        listT.add(new AllTransform());

      }
      if(spaceP.length(faceP)<actionDist){

        ((AllTransform)listT.get(index)).add(new Translate(Anar.Pt( (spaceP.x()-faceP.x())/(float)Math.pow(spaceP.length(faceP),2), (spaceP.y()-faceP.y())
            /(float)Math.pow(spaceP.length(faceP),2),0)));
        if(faceP.length(spaceP)<deleteDist)
          delete = true;
      }

      index++;

    } // end while pt

    if( !delete)
      newSpacePoints.add(spaceP);

  } // end while space point

  // now update transform on all points
  int index = 0;
  f = obj.faces.get(obj.faces.size()-1);
  Iterator itP = f.iterator();
  while (itP.hasNext()){
    Pt faceP = (Pt)itP.next();

    Pt v = ((AllTransform)listT.get(index)).toUnit().getVertex();
    faceP.translate(coeff*v.x(),coeff*v.y(),coeff*v.z());

    index++;

  } // end while pt


  println("Space points "+newSpacePoints.numOfPts());

  Obj newObj = new Obj();

  Iterator itF = obj.faces.iterator();
  while (itF.hasNext()){
    newObj.add((Face)itF.next());
  }

  it = newSpacePoints.iterator();
  while (it.hasNext()){
    newObj.add((Pt)it.next());
  }

  obj = newObj;

}



  ArrayList trans;

  AllTransform(){
    trans = new ArrayList();
  }

  void add(Translate t){
    trans.add(t);
  }

  Translate toUnit(){
    Pt v = Anar.Pt(0,0,0);

    Iterator it = trans.iterator();
    while (it.hasNext()){
      Pt p = ((Translate)it.next()).getVertex();
      v.set(v.x()+p.x(),v.y()+p.y(),v.z()+p.z());
    }

    // unit length
    if(v.length()>0)
      v.set(v.x()/v.length(),v.y()/v.length(),v.z()/v.length());

    return new Translate(v);
  }

}

void weavingIteration(){

  nIteration++;
  println(nIteration);

  float z = 2f;

  ArrayList newFaces = new ArrayList();

  for (int k = 0; k<obj.numOfFaces(); k++){
    Face f = obj.face(k);

    // float param;
    // if (positive) param = z / (nIteration * nIteration) ;
    // else param = - z / (nIteration * nIteration) ;

    float param = z;

    Pts pp = new Pts();
    Pts subdiv = new Pts();

    // layout of pp in quad
    // 3 6 7 2
    //			 
    //			 
    // 0 4 5 1

    // get the vertices of the triangle
    pp.add(f.pt(0));
    pp.add(f.pt(1));
    pp.add(f.pt(2));
    pp.add(f.pt(3));

    // get the middle points
    Pts AB = new PtsMid(pp.pt(0),pp.pt(1),3);
    Pts BC = new PtsMid(pp.pt(1),pp.pt(2),3);
    Pts CD = new PtsMid(pp.pt(2),pp.pt(3),3);
    Pts DA = new PtsMid(pp.pt(3),pp.pt(0),3);

    pp.add(Anar.Pt(AB.pt(1)));
    pp.add(Anar.Pt(AB.pt(2)));

    pp.add(Anar.Pt(CD.pt(2)));
    pp.add(Anar.Pt(CD.pt(1)));

    // interior points
    Pts AB1CD2 = new PtsMid(pp.pt(4),pp.pt(6),3);
    Pts AB2CD1 = new PtsMid(pp.pt(5),pp.pt(7),3);

    // layout of the quad subdivision in subdiv
    // up down
    // 0 1 2 3
    // 4 5 6 7 16 17 18 19
    // 8 9 10 11 20 21 22 23
    // 12 13 14 15

    // 1st horizontal
    // 3 6 7 2
    subdiv.add(pp.pt(3));
    subdiv.add(pp.pt(6));
    subdiv.add(pp.pt(7));
    subdiv.add(pp.pt(2));

    // 2nd horizontal
    subdiv.add(Anar.Pt(DA.pt(1)));

    subdiv.add(Anar.Pt(AB1CD2.pt(2)));
    subdiv.add(Anar.Pt(AB2CD1.pt(2)));

    subdiv.add(Anar.Pt(BC.pt(2)));

    // 3rd horizontal
    subdiv.add(Anar.Pt(DA.pt(2)));

    subdiv.add(Anar.Pt(AB1CD2.pt(1)));
    subdiv.add(Anar.Pt(AB2CD1.pt(1)));

    subdiv.add(Anar.Pt(BC.pt(1)));


    // 4th horizontal
    // 0 4 5 1
    subdiv.add(pp.pt(0));
    subdiv.add(pp.pt(4));
    subdiv.add(pp.pt(5));
    subdiv.add(pp.pt(1));


    // duplicate points
    // 2nd horizontal
    subdiv.add(Anar.Pt(DA.pt(1)));

    subdiv.add(Anar.Pt(AB1CD2.pt(2)));
    subdiv.add(Anar.Pt(AB2CD1.pt(2)));

    subdiv.add(Anar.Pt(BC.pt(2)));

    // 3rd horizontal
    subdiv.add(Anar.Pt(DA.pt(2)));

    subdiv.add(Anar.Pt(AB1CD2.pt(1)));
    subdiv.add(Anar.Pt(AB2CD1.pt(1)));

    subdiv.add(Anar.Pt(BC.pt(1)));


    // up
    for (int i = 4; i<12; i++){
      Pt p = subdiv.pt(i);
      p.translate(0f,0f,param);
    }

    // down
    for (int i = 16; i<24; i++){
      Pt p = subdiv.pt(i);
      p.translate(0f,0f, -param);
    }

    // add first batch (regular ones)
    int[] angle = {0, 2, 5, 8, 10, 16, 18};
    // int[] angle = {0, 2, 5, 8, 10};
    for (int i = 0; i<angle.length; i++){
      Pts fps = new Pts();
      fps.add(subdiv.pt(angle[i]+4));
      fps.add(subdiv.pt(angle[i]+5));
      fps.add(subdiv.pt(angle[i]+1));
      fps.add(subdiv.pt(angle[i]));
      newFaces.add(new Face(fps));
    }

    // add second batch (irregular ones)
    Pts fps = new Pts();
    fps.add(subdiv.pt(17));
    fps.add(subdiv.pt(18));
    fps.add(subdiv.pt(2));
    fps.add(subdiv.pt(1));
    newFaces.add(new Face(fps));

    fps = new Pts();
    fps.add(subdiv.pt(13));
    fps.add(subdiv.pt(14));
    fps.add(subdiv.pt(22));
    fps.add(subdiv.pt(21));
    newFaces.add(new Face(fps));


  }

  obj = new Obj();

  for (int i = 0; i<newFaces.size(); i++)
    obj.add((Face)newFaces.get(i));

  Anar.camTarget(obj);

}


void draw(){
  background(153);

  // ParametricValue v = parameters.pt(0);
  // v.set(v.pt()+ random(10)-5);

  obj.draw();
  if(drawPoints)
    obj.pts.draw();
}


boolean positive   = false;
boolean drawPoints = false;

void keyPressed(){

  switch(key){
    case 'q':
      spaceColonyStep();
    break;
    case 'w':
      spaceColonyLastStep();
    break;
    case 'e':
      reset();
    break;
    case 'r':
      drawPoints = !drawPoints;
    break;
    case 't':
      weavingIteration();
    break;


    case 'a':
      anar.tools.TextIO.write( ((Object)this).getClass().getName()+".lsp",obj.toAutocad());
    break;


  }


}