void display()
{
  if(displayMode==1)
    drawLattice(rawr);  
  if(displayMode==2)
  {
//    rawr.drawLattice();
    drawCells(rawr);  
  }
  if(displayMode==3)
    drawLines(rawr);      
}

void ellipse(position p,float d)
{
  ellipse(p.x,p.y,d,d);
}

void drawArrow(position p1,position p2)
{
  position corner=new position(new position(0,0));
  position bLeft=new position(corner.displace(140,4));
  position bRight=new position(corner.displace(220,4));
  float heading=radians(360-getHeading(p1,p2));  
  push();
  line(p1.x,p1.y,p2.x,p2.y);
  translate(p2.x,p2.y);
  rotate(heading);
  beginShape(POLYGON);
  vertex(corner.x,corner.y);
  vertex(bLeft.x,bLeft.y);
  vertex(bRight.x,bRight.y);
  endShape();
  pop();
}



void drawLattice(lattice phen)
{
  noFill();
  float t=phen.tension*200;
  for(int i=0;i<phen.nodes.length;i++)
  {
    for(int s=0;s<phen.nodes[i].c.output.length;s++)
    {
      float a=255-(dist(phen.nodes[i].p,phen.nodes[phen.nodes[i].c.output[s]].p)/25*255);
      if(a>15)
      {
        stroke(255,255,255,a);
        line(phen.nodes[i].x(),phen.nodes[i].y(),phen.nodes[phen.nodes[i].c.output[s]].x(),phen.nodes[phen.nodes[i].c.output[s]].y());  
      }
    }
    for(int v=i;v<phen.nodes.length;v++)
    {
      boolean doCollide=true;              
      if(dist(phen.nodes[i].p,phen.nodes[v].p)<phen.collisionDistance)
      {
        for(int out=0;out<phen.nodes[i].c.output.length;out++)
        {
          if(phen.nodes[i].c.output[out]==v)
          {
            doCollide=false;
            break;                    
          }
        }              
        for(int in=0;in<phen.nodes[i].c.input.length;in++)
        {
          if(phen.nodes[i].c.input[in]==v)
          {
            doCollide=false;              
            break;
          }
        }      
        if(doCollide)
        {
          float a=150-(dist(phen.nodes[i].p,phen.nodes[v].p)/150*255);        
          if(a>120)
          {
            stroke(150+a,125,115,a);
            line(phen.nodes[i].x(),phen.nodes[i].y(),phen.nodes[v].x(),phen.nodes[v].y());
          }
        }
      }
    }
    float vel=phen.nodes[i].velocity.m;
    stroke(222,222,222+t);        
    ellipse(phen.nodes[i].x(),phen.nodes[i].y(),5+constrain((vel*1.5),0,4),5+constrain((vel*1.5),0,4));
  }
}
void drawCells(lattice phen)
{
  for(int i=0;i<phen.nodes.length;i++)
  {   
    noFill();      
    if(phen.nodes[i].n.firing)
    {
      stroke(255,0,0);
      ellipse(phen.nodes[i].p,13);      
    }
    else
    {
      noStroke();
      float ra=(phen.nodes[i].n.accumulated+1)/2;
      ra=constrain(ra,0,1);
      float rt=(phen.nodes[i].n.threshold+1)/2;
      ra=constrain(ra,0,1);        
      fill(150,180,250,ra*200);        
      ellipse(phen.nodes[i].p,ra*10);
      
      fill(255,50,50,rt*200); 
      ellipse(phen.nodes[i].p,rt*8);                
      noFill();
    
      if(phen.nodes[i].isNeuron&&!phen.nodes[i].isLightSensor)        
        stroke(80,120,200);
      else
      if(phen.nodes[i].isMuscle)
        stroke(220,160,80);
      else      
      if(phen.nodes[i].isLightSensor)
        stroke(150,220,70);
      else
        stroke(90);
      ellipse(phen.nodes[i].p,12);      

    }
    if(phen.nodes[i].isNeuron)
    {
      int le=phen.nodes[i].n.output.length;
      for(int o=0;o<le;o++)
      {
        int id=phen.nodes[i].n.output[o].n.id;      
        vector vec=new vector(phen.nodes[i].p,phen.nodes[id].p);
        vec.m*=.8;  
        if(phen.nodes[i].n.firing)
        {
          stroke(240,0,0);
          fill(240,0,0);                  
        }
        else
        {
          stroke(220,240,225,100);
          fill(220,240,225,120);                            
        }
        drawArrow(vec.p,vec.ep());        
      }        
    }
    if(phen.nodes[i].isMuscle)
    {  
      stroke(220,35,15,150);          
      position p1=phen.nodes[i].p;    
      int lo=phen.nodes[i].c.output.length;          
      for(int o=0;o<lo;o++)
      {
        int index=phen.nodes[i].c.output[o];
        if(phen.nodes[index].isMuscle)
        {
          position p2=phen.nodes[index].p;
          line(p1.x,p1.y,p2.x,p2.y);  
        }
      }
      int li=phen.nodes[i].c.input.length;    
      for(int in=0;in<li;in++)
      {
        int index=phen.nodes[i].c.input[in];      
        if(phen.nodes[index].isMuscle)
        {
          position p2=phen.nodes[index].p;
          line(p1.x,p1.y,p2.x,p2.y);  
        }
      }      
    }
  }
}
void drawLines(lattice phen)
{
  noFill();
  float t=phen.tension*200;
  for(int i=0;i<phen.nodes.length;i++)
  {
    for(int s=0;s<phen.nodes[i].c.output.length;s++)
    {
      float a=255-(dist(phen.nodes[i].p,phen.nodes[phen.nodes[i].c.output[s]].p)/20*255);
      stroke(220,220,20+t,a);
      line(phen.nodes[i].x(),phen.nodes[i].y(),phen.nodes[phen.nodes[i].c.output[s]].x(),phen.nodes[phen.nodes[i].c.output[s]].y());  
      noFill();
      ellipse(phen.nodes[i].p,4);    
    }
  }  
}