class OccupationsCsv {
  ArrayList names;
  ArrayList codes;
  ArrayList wages;
  ArrayList j2010;
  ArrayList j2020;
  ArrayList parent;

  public OccupationsCsv(String filename) {
    names = new ArrayList();
    depth = new ArrayList();
    codes = new ArrayList();
    wages = new ArrayList();
    j2010 = new ArrayList();
    j2020 = new ArrayList();
    parent = new ArrayList();
    String[] data = loadStrings(filename);

    int[] par = new int[6];

    for (int i=1;i<data.length;i++) {
      String[] parts = data[i].split("\t");
      if (parts.length < 3) continue;
      String title = parts[0].replaceAll("\"", "");
      int dep = parts[1].replaceAll("\"", "").replaceAll(",", "").replaceAll("[.].*", "").trim();
      String code  = parts[2].replaceAll("\"", "");
      int jobs2010 = parts[3].replaceAll("\"", "").replaceAll(",", "").replaceAll("[.].*", "").trim();
      int jobs2020 = parts[4].replaceAll("\"", "").replaceAll(",", "").replaceAll("[.].*", "").trim();
      int wage = parts[5].replaceAll("\"", "").replaceAll(",", "").replaceAll("[.].*", "").trim();

      names.add(title);
      depth.add(dep);
      codes.add(code);
      j2010.add(jobs2010);
      j2020.add(jobs2020);
      wages.add(wage);
      if (dep==0) {
        parent.add(-1);
      }
      else {
        parent.add(par[dep-1]);
      }

      par[dep] = names.size()-1;
    }
  }
}

OccupationsCsv d;
RTree t;

class RNode {
  color c;
  int x, y, r = 50; //R is actually diameter
  String banner;
  ArrayList children = new ArrayList();
  public boolean isMouseOver(int mouseX, int mouseY) {
    float d = sqrt( Math.pow(mouseX-x, 2) + Math.pow(mouseY-y, 2));
    if (d < r/2 || d <= 1) {

      return true;
    }
    return false;
  }
}

class RTree {
  HashMap nodes;
  public RTree() {
    nodes = new HashMap();
  }

  public void draw() {
    int centerX = width/2;
    int centerY = height/2;
    RNode root = nodes.get(0);
    root.x = centerX;
    root.y = centerY;
    int dr = 70;

    fill(0);
    ellipse(width/2, height/2, root.r, root.r);

    float theta = 0;
    int drd = 20;
    for (int i=0;i<root.children.size();i++) {
      theta = theta + (2*PI / root.children.size());
      int x = centerX + dr*sin(theta);
      int y = centerY + dr*cos(theta);
      int h = i*1.0/root.children.size() * 100;
      nodes.get(root.children.get(i)).x = x;
      nodes.get(root.children.get(i)).y = y;
      nodes.get(root.children.get(i)).c = color(h, 100, 100);

      int r = nodes.get(root.children.get(i)).r; 
      fill(color(h, 100, 100));
      line(centerX, centerY, x, y);
      ellipse(x, y, r, r); 
      rdraw(2, root.children.get(i), theta, x, y, 70+drd);
      drd = -drd;
    }
  }

  public void rdraw(int depth, int parent, float theta, int centerX, int centerY, int dr) {
    RNode root = nodes.get(parent);
    float dTheta = PI/2*2/depth / root.children.size();
    theta = theta - dTheta * (root.children.size()/2);
    int drd = 20;
    for (int i=0;i<root.children.size();i++) {
      int x = centerX + dr*sin(theta);
      int y = centerY + dr*cos(theta);
      int r = nodes.get(root.children.get(i)).r;
      nodes.get(root.children.get(i)).x = x;
      nodes.get(root.children.get(i)).y = y;
      line(centerX, centerY, x, y);
      ellipse(x, y, r, r); 
      theta = theta + dTheta;


      if (r > 10) {
        rdraw(depth + 1, root.children.get(i), theta, x, y, dr+drd);
        drd = -drd;
      }
    }
  }

  void mouseMove(int mouseX, int mouseY) {
    colorMode(RGB); 
    fill(255, 255, 255);
    noStroke();
    rect(0, 0, width, 50);
    fill(0);
    colorMode(HSB, 100, 100, 100);
    for (int i=0;i<nodes.size();i++) {
      if (nodes.get(i).isMouseOver(mouseX, mouseY)==true) {
        fill(0);
        text(nodes.get(i).banner, 30, 30);
        colorMode(HSB, 100, 100, 100);
      }
    }
  }
}
PFont pfb;

void setup() {
  pfb = createFont("Beteckna", 32);
  size(800, 800);
  background(233);
  fill(255);
  noStroke();
  rect(0, 700, width, 100);
  fill(0);
  textFont(pfb);
  text("2010 - 2020 BLS Job Growth Estimates", 30, 750);
  textFont(createFont("Arial", 12));
  stroke(0);
  colorMode(HSB, 100, 100, 100); 

  fill(0);
  d = new OccupationsCsv("occupation.txt"); 
  t = new RTree();
  for (int i=0;i<d.names.size();i++) {
    RNode n = new RNode();
    int newjobs = (d.j2020.get(i) - d.j2010.get(i));
    int percentnew = round((d.j2020.get(i) - d.j2010.get(i))/d.j2010.get(i)*100);
    n.banner = d.names.get(i) + " "+newjobs+" thousand new jobs ("+percentnew+"% growth) Median Salary: "+d.wages.get(i);
    n.r = sqrt(d.j2020.get(i) - d.j2010.get(i))/2;
    int parent = d.parent.get(i);
    if (parent >= 0) {
      t.nodes.get(parent).children.add(i);
    }
    t.nodes.put(i, n);
  }
  t.draw();
}

void draw() {
  t.mouseMove(mouseX, mouseY);
}