Adding select all/unselect all menu items.

sepinf-inc · Sep 5, 2023 · 4b606ea · 4b606ea
1 parent a089532
commit 4b606ea
Showing 1 changed file with 35 additions and 1 deletion.
diff --git a/iped-app/src/main/java/iped/app/timelinegraph/popups/LegendItemPopupMenu.java b/iped-app/src/main/java/iped/app/timelinegraph/popups/LegendItemPopupMenu.java
@@ -30,6 +30,9 @@ public class LegendItemPopupMenu extends JPopupMenu implements ActionListener {
     JMenuItem filter;
     JMenuItem unfilter;
 
+    JMenuItem selectAll;
+    JMenuItem unselectAll;
+
     JMenuItem checkItems;
     JMenuItem uncheckItems;
 
@@ -51,7 +54,15 @@ public LegendItemPopupMenu(IpedChartPanel ipedChartPanel) {
         unfilter = new JMenuItem(Messages.getString("TimeLineGraph.unfilterEventFromResultSet", "Unfilter"));
         unfilter.addActionListener(this);
         add(unfilter);
+
+        selectAll = new JMenuItem("Select all event types");
+        selectAll.addActionListener(this);
+        add(selectAll);
 
+        unselectAll = new JMenuItem("Unselect all event types");
+        unselectAll.addActionListener(this);
+        add(unselectAll);
+
         checkItems = new JMenuItem("Check corresponding events items");
         checkItems.addActionListener(this);
         add(checkItems);
@@ -198,14 +209,25 @@ public void actionPerformed(ActionEvent e) {
             }
             ipedChartPanel.filterSelection();
         }
-        
+
         //IMPORTANT
         if (e.getSource() == checkItems) {
             checkEventItems(selLegends);
         }
         if (e.getSource() == uncheckItems) {
             unselectEvents(selLegends);
         }
+
+        if (e.getSource() == selectAll) {
+            JList list = ipedChartPanel.getIpedChartsPanel().getLegendList();
+            list.addSelectionInterval(0, list.getModel().getSize());
+        }
+
+        if (e.getSource() == unselectAll) {
+            JList list = ipedChartPanel.getIpedChartsPanel().getLegendList();
+            list.removeSelectionInterval(0, list.getModel().getSize());
+        }
+
     }
 
     @Override
@@ -219,6 +241,18 @@ public void show(Component invoker, int x, int y){
         boolean selectionContainsSelect = false;
         boolean selectionContainsUnselect = false;
 
+        if (selLegends.isEmpty()) {
+            unselectAll.setEnabled(false);
+        } else {
+            unselectAll.setEnabled(true);
+        }
+
+        if (selLegends.size() >= list.getModel().getSize()) {
+            selectAll.setEnabled(false);
+        } else {
+            selectAll.setEnabled(true);
+        }
+
         for (Iterator iterator = selLegends.iterator(); iterator.hasNext();) {
             LegendItemBlockContainer legendItemBlockContainer = (LegendItemBlockContainer) iterator.next();