//visualisation of info from the wikileaks iraq war logs
//Jim Anning @jimanning December 2010


import org.json.*;
import java.util.Calendar;
JSONObject deathData;
ArrayList deaths;
int ballWidth = 48;
int framesPerDay = 3;
int currentFrame = 1;
int numHandles = 2;
int days = 0;
RectButton ce, ea, eh, fa, nc, ff, o;
float lonConvert = 561/12;
float latConvert = 480/8.55;
float fadeSpeed = 5;
boolean clickable = true;
boolean locked = false;
Calendar currentDate = Calendar.getInstance();
PImage iraqmap;
String currentDateStr = "";
String[] monthnames = new String[12];
String y = "";
String m = "";
String d = "";
Handle[] handles;
void setup() {
  size(640, 480);
 // frameRate(60);
  smooth();
 handles = new Handle[2];
 int hsize = 10;
  for(int i=0; i<numHandles; i++) {
    handles[i] = new Handle(10, 70+i*50, 90, 10, handles);
  } 
  
  handles[1].length = 10;
monthnames[0] = "Jan";
monthnames[1] = "Feb";
monthnames[2] = "Mar";
monthnames[3] = "Apr";
monthnames[4] = "May";
monthnames[5] = "Jun";
monthnames[6] = "Jul";
monthnames[7] = "Aug";
monthnames[8] = "Sep";
monthnames[9] = "Oct";
monthnames[10] = "Nov";
monthnames[11] = "Dec";


  color buttoncolor = color(153);
  color highlight = color(100);
  ce = new RectButton(450, 90, 10, buttoncolor, highlight, "Criminal Event");
  ea = new RectButton(450, 110, 10, buttoncolor, highlight, "Enemy Action");
  eh = new RectButton(450, 130, 10, buttoncolor, highlight, "Explosive Hazard");
  fa = new RectButton(450, 150, 10, buttoncolor, highlight, "Friendly Action");
  nc = new RectButton(450, 170, 10, buttoncolor, highlight, "Non_combat Event");
  ff = new RectButton(450, 190, 10, buttoncolor, highlight, "Friendly Fire");
  o = new RectButton(450, 210, 10, buttoncolor, highlight, "Other");
        
  

  iraqmap = loadImage("http://jimanning.com/iraq/iraqmap2.jpg");
  image(iraqmap, 0, 0);
  currentDate.set(2004, 0, 1);

  
  String dataString = join( loadStrings("http://jimanning.com/iraq/iraq.json"), "");
  try{
    deathData = new JSONObject(dataString);
    
  }catch (JSONException e) {
    //println ("There was an error parsing the JSONObject.");
  };
 //println(deathData.toString());
   //println(deathData.toString);

  // Create an empty ArrayList
  deaths = new ArrayList();
  
  // Start by adding one element
  //deaths.add(new Death(width/2, 0, ballWidth));
}

void draw() {
  background(0);
  
   image(iraqmap, 0, 0);
   fill(128);
   rect(10,5,100,10);
   fill(30);
   for(int i=1; i<6; i++){
   rect(10+i*100/6, 5, 1, 10);
   }
   
   
   fill(255);
   
  // text(currentDateStr, 30, 30);
   
   text(d,15,30);
   text(monthnames[currentDate.get(Calendar.MONTH)], 35,30);
   text("20"+y, 63, 30);
   
   ellipse(10+(100*(float)days/2190), 9, 10, 10);
   
   //do title
   
   text("Deaths in Iraq 2004-2009 from Wikileaks info", 330,20);
   
   //do legend
   
   fill(0,255,0);
   ellipse(40, 375, 10, 10);
   text("Iraqi", 60, 380);
   fill(115,115,255);
   ellipse(40,395, 10, 10);
   text("Coalition", 60, 400);
   fill(255, 0, 0);
   ellipse(40,415, 10,10);
   text("Enemy", 60, 420);
   fill(255, 255, 0);
   ellipse(40, 435, 10, 10);
   text("Civilian", 60, 440);
   
  ce.display();
  ea.display();
  eh.display();
  fa.display();
  nc.display();
  ff.display();
  o.display(); 
   
  for(int i=0; i<numHandles; i++) {
    handles[i].update();
    handles[i].display();
  }
  
  //draw handler labels
  
  text("Speed", 10, 90);
  text("Fade length", 10, 140);
  
  framesPerDay= floor((float)(90- handles[0].length) / 9);
  fadeSpeed = floor((float)(90- handles[1].length)/10);
  
  if (framesPerDay < 1) framesPerDay = 1;
  if (fadeSpeed < 1) fadeSpeed = 1;
  
  update(mouseX, mouseY);
  
  if (currentFrame % framesPerDay == 0){
    
    //whats our current date
    y = currentDate.get(Calendar.YEAR)+"";
    m = (currentDate.get(Calendar.MONTH)+1)+"";
    d = currentDate.get(Calendar.DATE)+"";
    
    if (m.length() == 1) m = "0"+m;
    if (d.length() == 1) d = "0"+d;
    y = y.substring(2);
  
  
  
  
  
    currentDateStr = d+"-"+m+"-"+y;
    // do we need to add some points to plot
    
   
    
    try{
      JSONArray todaysDeaths = deathData.getJSONArray(currentDateStr);
    //  println(todaysDeaths.toString());
      int i = todaysDeaths.length();
      //println("Incidents for "+currentDateStr+" = "+i);
      for (int j=0; j<i; j++){
       JSONObject incident = todaysDeaths.getJSONObject(j);
       String type = incident.getString("t");
       
       //filter whether to display or not
       
       boolean disp = false;
       if (type.equals("CE") && ce.switchedOn) disp = true;
       if (type.equals("EA") && ea.switchedOn) disp = true;
       if (type.equals("EH") && eh.switchedOn) disp = true;
       if (type.equals("FA") && fa.switchedOn) disp = true;
       if (type.equals("NC") && nc.switchedOn) disp = true;
       if (type.equals("FF") && ff.switchedOn) disp = true;
       if (type.equals("O") && o.switchedOn) disp = true;
       
       if (disp){
         
         
         int iD = incident.getInt("i"); //iraq
         int aD = incident.getInt("a"); //coalition
         int cD = incident.getInt("c"); //civilian
         int eD = incident.getInt("e"); //enemy
       
        //println(incident.toString());
         JSONObject geo = incident.getJSONObject("geo");
      //println(geo.toString());
         Double lat = geo.getDouble("lat");
         Double lon = geo.getDouble("lon");
      //println(iD + " " + aD + " " +cD +" "+lat+" "+lon); 
        //create the new death object
    
         // int total = iD+aD+cD;
        int xpos=(int)((lon-38)*lonConvert);
        int ypos=(int)((37.45-lat)*latConvert);
    //println ("x="+xpos+" y="+ypos+" total="+total);
  
        if (iD != 0) deaths.add(new Death(xpos, ypos, iD, 0, 255, 0)); // iraqi army green
        if (aD != 0) deaths.add(new Death(xpos, ypos, aD, 115, 115, 255)); //coallition blue
        if (cD != 0) deaths.add(new Death(xpos, ypos, cD, 255, 255, 0)); //civilians yellow
        if (eD != 0) deaths.add(new Death(xpos, ypos, eD, 255, 0, 0)); //enemy red
        //deaths.add(new Death(xpos, ypos, total, "i"));  
       } 
      
    }  
    
      
    }catch (JSONException e) {
    //println ("There was an error parsing the JSONObject.");
  };
    
    
    
  
    
    
    
    //increment the day by 1
    //println(currentDateStr);
    //currentDateStr != "01-02-04"
    if (currentDateStr.equals("01-01-10")){
      //reset date
      currentDate.set(2004, 0, 1);
      currentFrame=1;
      days = 0;
    }else{
      days++;
      currentDate.add(Calendar.DAY_OF_YEAR, 1);
    }
   
  }
  

  // With an array, we say balls.length, with an ArrayList, we say balls.size()
  // The length of an ArrayList is dynamic
  // Notice how we are looping through the ArrayList backwards
  // This is because we are deleting elements from the list  
  for (int i = deaths.size()-1; i >= 0; i--) { 
    // An ArrayList doesn't know what it is storing so we have to cast the object coming out
    Death death = (Death) deaths.get(i);
   // ball.move();
   noStroke();
    death.display();
    if (death.finished()) {
      // Items can be deleted with remove()
      deaths.remove(i);
    }
    
  }  
 
 currentFrame++;
  
}






class Death {
  
  float x;
  float y;
  int r;
  int g;
  int b;
  float w;
  float life = 255;
  float opacity = 10;
  
  Death(float tempX, float tempY, float tempW, int tr, int tg, int tb) {
    x = tempX;
    y = tempY;
    w = tempW;
    r = tr;
    g = tg;
    b = tb;
   
  }
  
 
 
  boolean finished() {
    // Balls fade out
   
    if (life > 230) {
      life = life-10;
   
    }else{
    
      life=life-fadeSpeed;
  }
      
     
   
    if (life < 10) {
      return true;
    } else {
      return false;
    }
  }
  
  void display() {
    // Display the circle
    
    fill(r,g,b,life);
    
    //stroke(0,life);
    ellipse(x,y,5+(w/3),5+(w/3));
  }
}  

class Button
{
   
  int x, y;
  int size;
  color basecolor, highlightcolor;
  color currentcolor;
  boolean over = false;
  boolean pressed = false;   
  boolean switchedOn = true;
  void update() 
  {
    if(over()) {
      currentcolor = highlightcolor;
    } 
    else {
      currentcolor = basecolor;
    }
  }

  boolean pressed() 
  {
    if(over) {
      locked = true;
      return true;
    } 
    else {
      locked = false;
      return false;
    }    
  }

  boolean over() 
  { 
    return true; 
  }

  boolean overRect(int x, int y, int width, int height) 
  {
    if (mouseX >= x && mouseX <= x+width && 
      mouseY >= y && mouseY <= y+height) {
      return true;
    } 
    else {
      return false;
    }
  }



}


class RectButton extends Button
{
   String txt;
  RectButton(int ix, int iy, int isize, color icolor, color ihighlight, String itxt) 
  {
    x = ix;
    y = iy;
    size = isize;
    basecolor = icolor;
    highlightcolor = ihighlight;
    currentcolor = basecolor;
    txt = itxt;
  }

  boolean over() 
  {
    if( overRect(x, y, size, size) ) {
      over = true;
      return true;
    } 
    else {
      over = false;
      return false;
    }
  }

  void display() 
  {
    stroke(255);
    strokeWeight(1);
    fill(currentcolor);
    rect(x, y, size, size);
    fill(128);
    text(txt, x+20, y+10);
   
   
   if(switchedOn){
     stroke(0);
     strokeWeight(2);
     line(x+1,y+1,x+size-1, y+size-1);
     line(x+size-1, y+1, x+1, y+size-1);
     
   }
  }
}

void update(int x, int y)
{
  if(locked == false) {
    ce.update();
    ea.update();
    eh.update();
    fa.update();
    nc.update();
    ff.update();
    o.update();
              
    
  } 
  else {
    locked = false;
  }
  
 
  if(mousePressed) {
    if(ce.pressed() && clickable) {
      ce.switchedOn = !ce.switchedOn;
      
    } 
    else if(ea.pressed() && clickable) {
      ea.switchedOn = !ea.switchedOn;
    }else if(eh.pressed() && clickable){
       eh.switchedOn = !eh.switchedOn;
    }else if(fa.pressed() && clickable){
       fa.switchedOn = !fa.switchedOn;
    }else if(nc.pressed() && clickable){
       nc.switchedOn = !nc.switchedOn;
    }else if(ff.pressed() && clickable){
       ff.switchedOn = !ff.switchedOn;
    }else if(o.pressed() && clickable){
       o.switchedOn = !o.switchedOn;
    }
    
    clickable = false;
  }else{
    clickable = true;
  }
}

class Handle
{
  int x, y;
  int boxx, boxy;
  int length;
  int size;
  int fullLength;
  boolean over;
  boolean press;
  boolean locked = false;
  boolean otherslocked = false;
  Handle[] others;
  
  Handle(int ix, int iy, int il, int is, Handle[] o)
  {
    x = ix;
    y = iy;
    length = il;
    size = is;
    fullLength = length;
    boxx = x+length - size/2;
    boxy = y - size/2;
    others = o;
  }
  
  void update() 
  {
    boxx = x+length;
    boxy = y - size/2;
    
    for(int i=0; i<others.length; i++) {
      if(others[i].locked == true) {
        otherslocked = true;
        break;
      } else {
        otherslocked = false;
      }  
    }
    
    if(otherslocked == false) {
      over();
      press();
    }
    
    if(press) {
      length = lock(mouseX-10-size/2, 0, fullLength);
 
     
      //println(length);
      
    }
  }
  
  void over()
  {
    if(overRect(boxx, boxy, size, size)) {
      over = true;
    } else {
      over = false;
    }
  }
  
  void press()
  {
    if(over && mousePressed || locked) {
      press = true;
      locked = true;
    } else {
      press = false;
    }
  }
  
  void release()
  {
    locked = false;
  }
  
  void display() 
  {
    stroke(128);
    line(x, y, x+length, y);
    fill(255);
    stroke(128);
    rect(boxx, boxy, size, size);
    if(over || press) {
      line(boxx, boxy, boxx+size, boxy+size);
      line(boxx, boxy+size, boxx+size, boxy);
    }

  }
}

boolean overRect(int x, int y, int width, int height) 
{
  if (mouseX >= x && mouseX <= x+width && 
      mouseY >= y && mouseY <= y+height) {
    return true;
  } else {
    return false;
  }
}

int lock(int val, int minv, int maxv) 
{ 
  return  min(max(val, minv), maxv); 
} 

void mouseReleased() 
{
  for(int i=0; i<numHandles; i++) {
    handles[i].release();
  }
}