refactor: simplify legend panel using EdgeType enum loop

- Added legendImagePath field to EdgeType enum
- Replaced 60+ lines of manual legend construction with a 10-line loop
- Legend now automatically stays in sync when new EdgeType values are added
- No behavioral change

Author: sauravbhattacharya001

Gvisual/src/gvisual/EdgeType.java

@@ -18,23 +18,25 @@
  */
 public enum EdgeType {
 
-    FRIEND      ("f",  "friend",            Color.GREEN,             10, 2),
-    CLASSMATE   ("c",  "Classmate",         Color.BLUE,              30, 1),
-    FAMILIAR    ("fs", "Familiar Stranger",  Color.GRAY,              2, 1),
-    STRANGER    ("s",  "Stranger",           Color.RED,               2, 2),
-    STUDY_GROUP ("sg", "Study Groups",       Color.ORANGE,           20, 1);
+    FRIEND      ("f",  "friend",            Color.GREEN,  "images/green.jpg",  10, 2),
+    CLASSMATE   ("c",  "Classmate",         Color.BLUE,   "images/blue.jpg",   30, 1),
+    FAMILIAR    ("fs", "Familiar Stranger",  Color.GRAY,  "images/gray.jpg",    2, 1),
+    STRANGER    ("s",  "Stranger",           Color.RED,   "images/red.jpg",     2, 2),
+    STUDY_GROUP ("sg", "Study Groups",       Color.ORANGE,"images/yellow.jpg", 20, 1);
 
     private final String code;
     private final String displayLabel;
     private final Color color;
+    private final String legendImagePath;
     private final int defaultDurationThreshold;
     private final int defaultMeetingThreshold;
 
-    EdgeType(String code, String displayLabel, Color color,
+    EdgeType(String code, String displayLabel, Color color, String legendImagePath,
              int defaultDurationThreshold, int defaultMeetingThreshold) {
         this.code = code;
         this.displayLabel = displayLabel;
         this.color = color;
+        this.legendImagePath = legendImagePath;
         this.defaultDurationThreshold = defaultDurationThreshold;
         this.defaultMeetingThreshold = defaultMeetingThreshold;
     }
@@ -54,6 +56,11 @@ public Color getColor() {
         return color;
     }
 
+    /** Returns the path to the legend colour-swatch image. */
+    public String getLegendImagePath() {
+        return legendImagePath;
+    }
+
     /** Returns the default duration-of-meeting threshold (minutes). */
     public int getDefaultDurationThreshold() {
         return defaultDurationThreshold;

Gvisual/src/gvisual/Main.java

@@ -637,62 +637,17 @@ public final void initializeLegendSpace(){
 
         JLabel legendHeading = new JLabel("Legend for the Graph");
 
-        String blueImgPath = "images/blue.jpg";
-        String redImgPath = "images/red.jpg";
-        String greenImgPath = "images/green.jpg";
-        String yellowImgPath = "images/yellow.jpg";
-        String grayImgPath = "images/gray.jpg";
-
-        Icon blue = new ImageIcon(blueImgPath);
-        Icon red = new ImageIcon(redImgPath);
-        Icon green = new ImageIcon(greenImgPath);
-        Icon yellow = new ImageIcon(yellowImgPath);
-        Icon gray = new ImageIcon(grayImgPath);
-
-        JLabel fLabel = new JLabel("Friend");
-        JLabel fColor= new JLabel(green);
-        JLabel fsLabel= new JLabel("Familiar Stranger");
-        JLabel fsColor= new JLabel(gray);
-        JLabel sLabel= new JLabel("Stranger");
-        JLabel sColor= new JLabel(red);
-        JLabel cLabel= new JLabel("Classmate");
-        JLabel cColor = new JLabel(blue);
-        JLabel sgLabel= new JLabel("Study Group");
-        JLabel sgColor= new JLabel(yellow);
-
-        Box HBox[] = new Box[5];
-        Box VBox[] = new Box[1];
-
-        VBox[0]= Box.createVerticalBox();
-
-        HBox[0]= Box.createHorizontalBox();
-        HBox[1]= Box.createHorizontalBox();
-        HBox[2]= Box.createHorizontalBox();
-        HBox[3]= Box.createHorizontalBox();
-        HBox[4]= Box.createHorizontalBox();
-
-
-        HBox[0].add(fColor);
-        HBox[1].add(fsColor);
-        HBox[2].add(sColor);
-        HBox[3].add(cColor);
-        HBox[4].add(sgColor);
-
-        HBox[0].add(fLabel);
-        HBox[1].add(fsLabel);
-        HBox[2].add(sLabel);
-        HBox[3].add(cLabel);
-        HBox[4].add(sgLabel);
-
-        VBox[0].add(HBox[0]);
-        VBox[0].add(HBox[1]);
-        VBox[0].add(HBox[2]);
-        VBox[0].add(HBox[3]);
-        VBox[0].add(HBox[4]);
+        Box legendBox = Box.createVerticalBox();
+        for (EdgeType type : EdgeType.values()) {
+            Icon icon = new ImageIcon(type.getLegendImagePath());
+            Box row = Box.createHorizontalBox();
+            row.add(new JLabel(icon));
+            row.add(new JLabel(type.getDisplayLabel()));
+            legendBox.add(row);
+        }
 
         JSplitPane splitPane = new JSplitPane(JSplitPane.VERTICAL_SPLIT,
-                legendHeading, VBox[0]);
-
+                legendHeading, legendBox);
 
         legendPanel.add(splitPane);