import treemap.*;

FolderItem rootItem;
FileItem rolloverItem;
PFont font;

public void setup(  ) {
  size(1024, 768);
  rectMode(CORNERS);
  cursor(CROSS);
  font = createFont("SansSerif", 13);
  setRoot(new File("C:\\Users\\Robert\\code\\"));
}

void setRoot(File folder) {
  FolderItem tm = new FolderItem(null, folder, 0, 0);
  tm.setBounds(0, 0, width-1, height-1);
  tm.contentsVisible = true;
  rootItem = tm;
}

void draw(  ) {
  background(0);
  textFont(font);rolloverItem = null;

  if (rootItem != null) {
    rootItem.draw(  );
  }
  if (rolloverItem != null) {
    rolloverItem.drawTitle(  );
  }
}

void mousePressed(  ) {
  if (rootItem != null) {
    rootItem.mousePressed(  );
  }
}