/*** Main data structures ***/
Table2 locationTable;  // Name and lab numbers of everyone
Scientist[] weitzlab;  // An array of "Scientists"
String roomNames[];    // The "names" of the rooms, in numerical order
boolean[][] fullMatrix;  // Rows are rooms, columns are scientsts
int pplRooms[];  // The number of people in a room

/*** Highlighting variables ***/
boolean hiNames[];   // Whether to highlight a name
boolean hiRooms[];      // Whether to highlight a room
boolean boolRooms;     // If true, highlight all the rooms for a given scientist
boolean boolSci;    // If true, highlight all the people and rooms in a group
boolean boolBWG, boolMF, boolSC;  // Whether the particular research group is active
boolean boolPD, boolGrad, boolVisit, boolUG, boolOther;

/*** Data intput/output ***/
PImage mapImage;        // The map of the Wetiz labs
String[] rows;        // An array use for reading from the file
int numScientists;    // The total number of scientists
int numRooms;        // The toal number of rooms
int hMax;          // needed for determining mouse selections
PFont plotFont;
PFont plotFontMed;
PFont plotFontBig;
int radius = 36;
float closestDist = MAX_FLOAT;

void setup() {
  println("Loading room info...");
  size(820, 540);
  mapImage = loadImage("WeitzMap2.jpg");
  locationTable = new Table2("locations.tsv");
  numRooms = locationTable.getRowCount();
  roomNames = new String[numRooms];
  rows = loadStrings("WeitzLocationData.csv");
  numScientists = rows.length;
  hiRooms = new boolean[numRooms];
  hiNames = new boolean[numScientists];
  fullMatrix = new boolean[numRooms][numScientists];
  pplRooms = new int[numRooms];
  boolRooms = false;

  for (int i=0; i < numRooms; i++) {
    String rn = locationTable.getRowName(i);
    print(i + ". " + rn + ", ");
    roomNames[i] = rn;
    for (int j = 0; j < numScientists; j++) {
      fullMatrix[i][j] = false;
    }
    hiRooms[i] = false;
    pplRooms[i] = 0;
  }

  readInput();
}

void readInput() {
  println();
  println("Reading scientist info...");
  weitzlab = new Scientist[numScientists];
  String sciLastName;
  String sciFirstName;
  String sciCategory;
  String[] sciRooms;
  String[] sciGroups;
  for (int i=0; i<numScientists; i++) {
    hiNames[i] = false;
    String[] cols = split(rows[i], ',');
    sciLastName = cols[0];
    sciFirstName = cols[1];
    sciCategory = cols[2];
    sciRooms = split(cols[3], ';');
    sciGroups = split(cols[4], ';');
    Scientist sci = new Scientist(sciLastName, sciFirstName, sciCategory, sciRooms, sciGroups);
    weitzlab[i] = sci;
    for (int j = 0; j < sciRooms.length; j++) {
      for (int k = 0; k < roomNames.length; k++) {
        if (match(roomNames[k], trim(sciRooms[j])) != null) {
          fullMatrix[k][i] = true;  // Populate the matrix
          pplRooms[k]++;
        }
      }
    }
  }
}

class Scientist {
  String lastName;
  String firstName;
  String category;
  String[] rooms;
  String[] groups;
  
  Scientist(String sciLastName, String sciFirstName, String sciCategory, String[] sciRooms, String[] sciGroups) {
    lastName = sciLastName;
    firstName = sciFirstName;
    category = sciCategory;
    rooms = sciRooms;
    groups = sciGroups;
  }
}

void draw() {
  image(mapImage, 0, 0);
 
  drawScientists();
  drawRooms();
  drawGroupLegend();
}

void drawScientists() {
  plotFont = createFont("SansSerif", 8);
  plotFontBig = createFont("SansSerif", 14);

  noStroke();
  int height = 540/numScientists;
  hMax = height*numScientists;
  fill(0, 0, 0);
  rect(720, 0, 820, 540);
  for (int i=0; i<numScientists; i++) {
    rect(720, i*height, 820, (i+1)*height);
    textAlign(LEFT);
    if (hiNames[i] && boolRooms) {
      fill(255, 0, 0);
      String[] sciRooms = weitzlab[i].rooms;
      int numRooms = sciRooms.length;
      String[] sciGroups = weitzlab[i].groups;
      int numGroups = sciGroups.length;
      String sciCat = weitzlab[i].category;
      for (int j=0; j<numRooms; j++) {
        String strRoom = trim(sciRooms[j]);
        int ri =  locationTable.getRowIndex(strRoom);
        if (ri > 0) {
          hiRooms[ri] = true;
        }
      }
      for (int j=0; j<numGroups; j++) {
        if (sciGroups[j].equals("BWG")) boolBWG = true;
        if (sciGroups[j].equals("MF")) boolMF = true;
        if (sciGroups[j].equals("SC")) boolSC = true;
        // if (boolBWG) println("Moo!");
      }
      if (sciCat.equals("visitor")) boolVisit = true;
      else if (sciCat.equals("post-doc")) boolPD = true;
      else if (sciCat.equals("grad student")) boolGrad = true;
      else if (sciCat.equals("undergraduate")) boolUG = true;
      else boolOther = true;
      
      textFont(plotFontBig);
      text(weitzlab[i].lastName + ", " + weitzlab[i].firstName, 520, 280);
    }
   else if (hiNames[i] && boolSci) {
      fill(255, 0, 0);
   } else {
      fill(128, 128, 128);
   }
   textFont(plotFont);
   text(weitzlab[i].lastName + ", " + weitzlab[i].firstName, 730, i*height);
   fill(0, 0, 0);
  }
}

void drawRooms() {
  ellipseMode(CENTER);
  textAlign(CENTER);
  plotFontMed = createFont("SansSerif", 10);
  textFont(plotFontMed);

  for (int i=0; i<numRooms; i++) {
    int xPos = locationTable.getInt(i, 1);
    int yPos = locationTable.getInt(i, 2);
    if (hiRooms[i]) { 
      fill(256, 0, 0); 
      if (boolSci) {
        textFont(plotFontBig);
        textAlign(LEFT);
        text(pplRooms[i] + " scientist(s)", 520, 280);
        textFont(plotFontMed);
        textAlign(CENTER);
      }
    } 
    else { fill(128, 128, 128); }
    ellipse(xPos, yPos, radius, radius);
    fill(0, 0, 0);
    text(locationTable.getRowName(i), xPos, yPos+4);
  }
}

void drawGroupLegend() {
  plotFontBig = createFont("SansSerif", 14);
  textFont(plotFontBig);
  textAlign(LEFT);
  
  if (boolBWG) { fill(255, 0, 255); } 
  else { fill(128, 128, 128); }
  text("Biophysics Working Group", 520, 320);
  if (boolMF) { fill(255, 0, 255); } 
  else { fill(128, 128, 128); }
  text("Microfluidics", 520, 340);
  if (boolSC) { fill(255, 0, 255); } 
  else { fill(128, 128, 128); } 
  text("Super-colloiders", 520, 360);
  
  if (boolVisit) { fill(255, 255, 0); }
  else { fill(128, 128, 128); }
  text("Visitor", 520, 400);
  if (boolPD) { fill(255, 255, 0); }
  else { fill(128, 128, 128); }
  text("Post-doc", 520, 420);
  if (boolGrad) { fill(255, 255, 0); }
  else { fill(128, 128, 128); }
  text("Grad Student", 520, 440);
  if (boolUG) { fill(255, 255, 0); }
  else { fill(128, 128, 128); }
  text("Undergraduate", 520, 460);
 if (boolOther) { fill(255, 255, 0); }
  else { fill(128, 128, 128); }
  text("Other", 520, 480);
  
}

void mouseClicked() {
  // Clear the array of highlighted scienists
  closestDist = MAX_FLOAT;
  for (int i=0; i < numScientists; i++) {hiNames[i] = false; }
  for (int i=0; i < numRooms; i++) { hiRooms[i] = false; }
  boolSci = false;
  boolRooms = false;
  boolBWG = false;
  boolMF = false;
  boolSC = false;
  boolVisit = false;
  boolPD = false;
  boolGrad = false;
  boolUG = false;
  boolOther = false;
    
  // Check to see if the mouse is hovering over the scientist list  
  if((mouseX > 720) && (mouseY < hMax)) {
    boolRooms = true;
    int height = 540/numScientists;
    int hiNum = mouseY/height;
    hiNames[hiNum] = true;
  }
  
  // Otherwise, see which room is selected
  else {
    String currentRoom = "none";
    int currentRoomIndex = -1;

    for (int j=0; j<roomNames.length; j++) {
      String strRoom = roomNames[j];
      int ri =  locationTable.getRowIndex(strRoom);
      if (ri > 0) {
        int xPos = locationTable.getInt(ri, 1);
        int yPos = locationTable.getInt(ri, 2);
        float d = dist(xPos, yPos, mouseX, mouseY);
        if ((d < radius) && (d < closestDist)) {
          closestDist = d;  
          currentRoom = strRoom;
          currentRoomIndex = ri;
          hiRooms[j] = true;
        }
      }
    }
    if (currentRoomIndex > -1) {
      boolSci = true;
      for (int j = 0; j < numScientists; j++) {
        if (fullMatrix[currentRoomIndex][j]) {
          hiNames[j] = true;
        }
      }
    }
    else if (currentRoomIndex < 0) {
      boolSci = true;
      if ((mouseX > 520) && (mouseX < 700)) {
        if ((mouseY > 310) && (mouseY < 320)) {
          boolBWG = true;
        } else if ((mouseY > 330) && (mouseY < 350)) {
          boolMF = true;
        } else if ((mouseY > 350) && (mouseY < 370)) {
          boolSC = true;
        } else {
          boolBWG = false;
          boolMF = false;
          boolSC = false;
        }
        for (int j = 0; j < numScientists; j++) {
          if ((boolBWG) && (weitzlab[j].groups[0].equals("BWG"))) hiNames[j] = true;
          else if ((boolMF) && (weitzlab[j].groups[0].equals("MF"))) hiNames[j] = true;
          else if ((boolSC) && (weitzlab[j].groups[0].equals("SC"))) hiNames[j] = true;
        }
        if ((mouseY > 390) && (mouseY < 410)) {
          boolVisit = true;
        } else if ((mouseY > 410) && (mouseY < 430)) {
          boolPD = true;
        } else if ((mouseY > 430) && (mouseY < 450)) {
          boolGrad = true;  
        } else if ((mouseY > 450) && (mouseY < 470)) {
          boolUG = true;  
        } else if ((mouseY > 470) && (mouseY < 490)) {
          boolOther = true;  
        }
        for (int j = 0; j < numScientists; j++) {
          if ((boolVisit) && (weitzlab[j].category.equals("visitor"))) hiNames[j] = true;
          else if ((boolPD) && (weitzlab[j].category.equals("post-doc"))) hiNames[j] = true;
          else if ((boolGrad) && (weitzlab[j].category.equals("grad student"))) hiNames[j] = true;
          else if ((boolUG) && (weitzlab[j].category.equals("undergraduate"))) hiNames[j] = true;
          else if (boolOther && !(weitzlab[j].category.equals("visitor") ||
          (weitzlab[j].category.equals("post-doc")) ||
          (weitzlab[j].category.equals("grad student")) ||
          (weitzlab[j].category.equals("undergraduate")))) hiNames[j] = true;
        }
      }
    }
  }
}