Notes

/**
* Creates legend shape based on category and song attribute passed
*
* @param category
* Category attributes to display
* @param song
* Song attribute glyphs are sorted by
*
*/
public void updateLegend(String category, String song) {
String sub1 = "";
String sub2 = "";
String sub3 = "";
String sub4 = "";

if (category == "Hobby") {
sub1 = "Reading";
sub2 = "Art";
sub3 = "Sports";
sub4 = "Music";
}
else if (category == "Major") {
sub1 = "Comp Sci";
sub2 = "Other Eng";
sub3 = "Math/CMDA";
sub4 = "Other";
}
else if (category == "Region") {
sub1 = "Northeast";
sub2 = "Southeast";
sub3 = "Other US";
sub4 = "Outside US";
}

// Legend descriptions
int legendW = 120;
int legendH = 200;
int betweenY = 20;
int betweenX = 5;
int startX = 850;
int startY = 280;

legend = new Shape[13];
legend[0] = new TextShape(startX, startY, category + " Legend",
Color.BLACK);
legend[1] = new TextShape(startX, legend[0].getY() + betweenY, sub1,
Color.MAGENTA);
legend[2] = new TextShape(startX, legend[1].getY() + betweenY, sub2,
Color.BLUE);
legend[3] = new TextShape(startX, legend[2].getY() + betweenY, sub3,
Color.YELLOW);
legend[4] = new TextShape(startX, legend[3].getY() + betweenY, sub4,
Color.GREEN);

int centerW = startX - betweenX + legendW / 2;
legend[5] = new TextShape(0, legend[4].getY() + betweenY, "Song "
+ song, Color.BLACK);
legend[5].moveTo(centerW - legend[5].getWidth() / 2, legend[5].getY());

legend[8] = new Shape(centerW - AXIS_W / 2, legend[5].getY() + betweenY,
AXIS_W, AXIS_H, Color.BLACK);

int axisCenterY = legend[8].getY() + AXIS_H / 2;
legend[6] = new TextShape(0, 0, "Heard", Color.BLACK);
legend[7] = new TextShape(0, 0, "Likes", Color.BLACK);
legend[6].move(centerW - AXIS_W / 2 - betweenX - legend[6].getWidth(),
axisCenterY - legend[6].getHeight() / 2);
legend[7].move(centerW + AXIS_W / 2 + betweenX / 2, axisCenterY
- legend[7].getHeight() / 2);

// legend outline
startX = startX - betweenX;
startY = startY - betweenX;
int thickness = 3;
legend[9] = new Shape(startX, startY, legendW, thickness, Color.BLACK);
legend[10] = new Shape(startX, startY + legendH - thickness, legendW,
thickness, Color.BLACK);
legend[11] = new Shape(startX, startY, thickness, legendH, Color.BLACK);
legend[12] = new Shape(startX + legendW - thickness, startY, thickness,
legendH, Color.BLACK);

for (int k = 0; k < legend.length; k++) {
if (k < 8) {
legend[k].setBackgroundColor(Color.WHITE);
}
window.addShape(legend[k]);
}
}