import traer.physics.*;
import traer.animation.*;

final float NODE_SIZE = 10;
final float EDGE_LENGTH = 20;
final float EDGE_STRENGTH = 0.12;
final float SPACER_STRENGTH = 1000*2;

final float K=0.1;
int selectedpoint = 0;
float scale=1.0;

ParticleSystem physics;
Smoother3D centroid;

int disptext[]={
  1,0};
String displaystring;

String names[] = new String[3000];
int firstfree=0;
int w[][] = new int[3000][3];
int wfirstfree=0; 
int type[]=new int[3000];
int posts[]=new int[3000];
int npart=0;
int conv[]=new int[3000];
int convb[]=new int[3000];
int nspring=0;
int conv2[]=new int[3000];
int locked[]=new int[3000];

void setup()
{
  size( 600, 600 );
  smooth();
  // frameRate( 24 );
  strokeWeight( 2 );
  ellipseMode( CENTER );       

  physics = new ParticleSystem( 0, 0.25 );
  centroid = new Smoother3D( 0.8 );
  initialize();

  PFont font;
  font  = loadFont("UniversLTStd-Light-48.vlw");
  textFont(font, 14); 
  textAlign(CENTER);

loadmodel("1.html");
displaystring="Extropy list Jan 2007";
}

void deletemodel()
{
  physics.clear();
 selectedpoint = 0;

 names = new String[3000];
 firstfree=0;
 w = new int[3000][3];
 wfirstfree=0; 
 type=new int[3000];
 posts=new int[3000];
 npart=0;
 conv=new int[3000];
 convb=new int[3000];
 nspring=0;
 conv2=new int[3000];
 locked=new int[3000];
}


void loadmodel(String filename)
{
   //--------------------------
  // String lines[] = loadStrings("http://lists.extropy.org/pipermail/extropy-chat/2007-January/date.html");
  String lines[] = loadStrings(filename);

  for (int i=0; i < lines.length; i++) 
  {
    int index = lines[i].indexOf("<LI>");
    if (index != -1)  
    {
      int index2=lines[i].indexOf("[extropy-chat] ");
      String subject = lines[i].substring(index2+15,lines[i].length());
      int si= insert(subject);
      String name = lines[i+2].substring(3,lines[i+2].length());
      int ni = insert(name);
      winsert(ni,si);
      type[ni]=0;
      type[si]=1;
    }
  }

  for (int i=0; i < firstfree; i++)
  {
    int s=0;
    for (int j=0; j < wfirstfree; j++)
    { 
      if ((w[j][0]==i || w[j][1]==i ) && (w[j][2]>1)) s=s+1;
    }

    if (s>0)
    {
      Particle p = physics.makeParticle();
      p.moveTo( random( -1, 1 ), random( -1, 1 ), 0 );
      // p.setMass(1+.2*posts[i]);
      conv[i]=npart;
      convb[npart]=i;
      npart+=1;
    }
  }
  for (int i=0; i < wfirstfree; i++)
  { 
    if (w[i][2]>1)
    {
      Particle p = physics.getParticle( conv[w[i][0]] );
      Particle q = physics.getParticle( conv[w[i][1]] );
      conv2[nspring]=i;
      nspring++;
      // addSpacersToNode( p, q );
      physics.makeSpring( p, q, EDGE_STRENGTH, EDGE_STRENGTH, EDGE_LENGTH*5/w[i][2] );
    } 
  }

  for ( int i = 0; i < physics.numberOfParticles(); ++i )
  {
    Particle p = physics.getParticle( i );
    for ( int j = 0; j < physics.numberOfParticles(); ++j )
    {
      Particle q = physics.getParticle( j );
      if (q != p)
        physics.makeAttraction( p, q, -SPACER_STRENGTH, 20 );
    }
  }
  //-------------------------- 
}

int insert(String s)
{
  int i;
  for (i=0; i<firstfree; i++)
    if (names[i].equals(s)) break; 
  names[i]=s;
  posts[i]++;
  if (i == firstfree) firstfree++;
  return i;
}

void winsert(int x, int y)
{
  int i;
  for (i=0; i<wfirstfree; i++)
    if (w[i][0]==x && w[i][1]==y) break;
  w[i][0]=x;
  w[i][1]=y;
  w[i][2]=w[i][2]+1;
  if (i == wfirstfree) wfirstfree++; 
}

void draw()
{
  physics.tick( 1.0 ); 
  centroid.tick();
  background( 255 );
  translate( width/2 , height/2 );
  scale( centroid.z() );
  translate( -centroid.x(), -centroid.y() );

  if (mousePressed == true) {
    Particle p = physics.getParticle( selectedpoint );
    p.moveTo( (mouseX-width/2 )/scale + centroid.x(), (mouseY-height/2 )/scale + centroid.y(), 0 );
  }

  for ( int i = 0; i < physics.numberOfParticles(); ++i )
  {
    Particle p = physics.getParticle( i );
    p.moveTo(min(max(p.position().x(),-width/2),width/2),min(max(p.position().y(),-height/2),height/2),0);
  }

  drawNetwork();  
}

void drawNetwork()
{   
  
    fill( 1,1,1 );
      text(displaystring, 0, -height/2+20); 
      
  // draw edges 
  colorMode(HSB, 1);
  beginShape( LINES );
  for ( int i = 0; i < physics.numberOfSprings(); ++i )
  {
    stroke( 0,1-2/(1+w[conv2[i]][2]),.5,.5 );
    strokeWeight(sqrt(w[conv2[i]][2]));
    Spring e = physics.getSpring( i );
    Particle a = e.getOneEnd();
    Particle b = e.getTheOtherEnd();
    vertex( a.position().x(), a.position().y() );
    vertex( b.position().x(), b.position().y() );
  }
  endShape();  
  colorMode(RGB, 255);
  // draw vertices

  noStroke();
  for ( int i = 0; i < physics.numberOfParticles(); ++i )
  {
    if (type[convb[i]]==0)
      fill(0,50, 260,50 );
    else
      fill( 204, 102, 0,50 );


    Particle v = physics.getParticle( i );
    float siz = NODE_SIZE*posts[convb[i]]*K;
    if (v.isFree())
      ellipse( v.position().x(), v.position().y(), siz, siz );
    else
      rect( v.position().x()-siz/2, v.position().y()-siz/2, siz, siz );
    fill( 102, 102, 102 );

    if (names[convb[i]]!=null & disptext[type[convb[i]]]==1)
      text(names[convb[i]], v.position().x(), v.position().y()); 
  }
}

void initialize()
{
  physics.clear();
  physics.makeParticle();
  centroid.setValue( 0, 0, 1.0 );
}


void addSpacersToNode( Particle p, Particle r )
{
  for ( int i = 0; i < physics.numberOfParticles(); ++i )
  {
    Particle q = physics.getParticle( i );
    if ( p != q && p != r )
      physics.makeAttraction( p, q, -SPACER_STRENGTH, 20 );
  }
}

void makeEdgeBetween( Particle a, Particle b )
{
  physics.makeSpring( a, b, EDGE_STRENGTH, EDGE_STRENGTH, EDGE_LENGTH );
}

void addNode()
{ 
  Particle p = physics.makeParticle();
  Particle q = physics.getParticle( (int)random( 0, physics.numberOfParticles()-1) );
  while ( q == p )
    q = physics.getParticle( (int)random( 0, physics.numberOfParticles()-1) );
  addSpacersToNode( p, q );
  makeEdgeBetween( p, q );
  p.moveTo( q.position().x() + random( -1, 1 ), q.position().y() + random( -1, 1 ), 0 );
}

void mousePressed() {
  float dist=10000;
  for ( int i = 0; i < physics.numberOfParticles(); ++i )
  {
    Particle p = physics.getParticle( i );
    float dx=abs(p.position().x()-(mouseX-width/2)/scale - centroid.x() );
    if (dx<dist)
    {
      float dy=abs(p.position().y()-(mouseY-height/2)/scale -  centroid.y() );
      if (dy<dist)
        if (dx*dx+dy*dy<dist*dist)
        {
          dist = sqrt(dx*dx+dy*dy);
          selectedpoint = i;
        }
    }
  }

  if (mouseButton == RIGHT) 
  {
    Particle p = physics.getParticle( selectedpoint );
    if (p.isFixed())
      p.makeFree();
    else
      p.makeFixed();
  }
}

void keyPressed() 
{
  if (key == 'p') disptext[0] = 1-disptext[0];
  if (key == 's') disptext[1] = 1-disptext[1];
  if (key == 'r')
  { 
    for ( int i = 0; i < physics.numberOfParticles(); ++i )
      physics.getParticle(i).makeFree();
  }

  if (key=='z')
  {
    scale+=0.5;
    centroid.setZTarget(scale);
  }
  if (key=='x')
  {
    scale=max(scale-0.5,0.5);
    centroid.setZTarget(scale);
  }

  if (key=='c')
  {
    scale=1;
    centroid.setTarget(0,0,scale);
  }

  if (keyCode == UP)
    centroid.setYTarget(centroid.y()-.5*height/scale);

  if (keyCode == DOWN)
    centroid.setYTarget(centroid.y()+.5*height/scale);

  if (keyCode == LEFT)
    centroid.setXTarget(centroid.x()-.5*width/scale);

  if (keyCode == RIGHT)
    centroid.setXTarget(centroid.x()+.5*width/scale);
    
  if (key=='1')
  {
    deletemodel();
    loadmodel("12.html");
        displaystring="Extropy list Dec 2006";
  }
    if (key=='2')
  {
    deletemodel();
    loadmodel("11.html");
        displaystring="Extropy list Nov 2006";
  }
      if (key=='3')
  {
    deletemodel();
    loadmodel("10.html");
        displaystring="Extropy list Oct 2006";
  }
      if (key=='4')
  {
    deletemodel();
    loadmodel("9.html");
        displaystring="Extropy list Sep 2006";
  }
      if (key=='5')
  {
    deletemodel();
    loadmodel("8.html");
        displaystring="Extropy list Aug 2006";
  }
  if (key=='6')
  {
    deletemodel();
    loadmodel("7.html");
        displaystring="Extropy list Jul 2006";
  }
  if (key=='7')
  {
    deletemodel();
    loadmodel("6.html");
        displaystring="Extropy list Jun 2006";
  }
  if (key=='8')
  {
    deletemodel();
    loadmodel("5.html");
        displaystring="Extropy list May 2006";
  }
  if (key=='9')
  {
    deletemodel();
    loadmodel("4.html");
    displaystring="Extropy list Apr 2006";
  }
  
    if (key=='0')
  {
    deletemodel();
    loadmodel("1.html");
    displaystring="Extropy list Jan 2007";
  }
}