Add edges and links to table view

CoreGraphFramework/src/au/gov/asd/tac/constellation/graph/processing/GraphRecordStoreUtilities.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2010-2022 Australian Signals Directorate
+ * Copyright 2010-2024 Australian Signals Directorate
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -70,7 +70,9 @@ private GraphRecordStoreUtilities() {
     public static final String DIRECTED_KEY = "[directed]<string>";
     public static final String COMPLETE_WITH_SCHEMA_KEY = "[complete_with_schema]<string>";
     public static final String DELETE_KEY = "[delete]<string>";
-
+    public static final String LINK_LOW = "low.";
+    public static final String LINK_HIGH = "high.";
+
     private static final String SELECTED_ATTRIBUTE_NAME = "selected";
     private static final String FALSE = "false";
     private static final String NUMBER_STRING_STRING_FORMAT = "%d:%s:%s";
@@ -86,7 +88,13 @@ private GraphRecordStoreUtilities() {
             "destination.Identifier",
             "destination.Label",
             "source.Type",
-            "destination.Type"
+            "destination.Type",
+            "low.Identifier",
+            "low.Type",
+            "low.Label",
+            "high.Identifier",
+            "high.Type",
+            "high.Label"
     );
     private static final List<String> ApprovedTypes = SchemaVertexTypeUtilities.getTypes().stream().map(i -> i.getName()).collect(Collectors.toList());
 

CoreTableView/src/au/gov/asd/tac/constellation/views/tableview/TableViewTopComponent.java

@@ -143,7 +143,9 @@ public TableViewTopComponent() {
         // the selection. The table also needs to have its element type set to TRANSACTION
         addAttributeValueChangeHandler(VisualConcept.TransactionAttribute.SELECTED, graph -> {
             if (needsUpdate() && currentState != null
-                    && currentState.getElementType() == GraphElementType.TRANSACTION) {
+                    && (currentState.getElementType() == GraphElementType.TRANSACTION
+                    || currentState.getElementType() == GraphElementType.EDGE
+                    || currentState.getElementType() == GraphElementType.LINK)) {
                 if (currentState.isSelectedOnly()) {
                     executorService.submit(new TriggerDataUpdateTask(pane, graph, getCurrentState()));
                 } else {
@@ -335,9 +337,9 @@ protected TablePane createContent() {
 
     @Override
     protected String createStyle() {
-        return JavafxStyleManager.isDarkTheme() 
-            ? "resources/table-view-dark.css" 
-            : "resources/table-view-light.css";
+        return JavafxStyleManager.isDarkTheme()
+                ? "resources/table-view-dark.css"
+                : "resources/table-view-light.css";
     }
 
     @Override

CoreTableView/src/au/gov/asd/tac/constellation/views/tableview/components/ElementTypeContextMenu.java

@@ -0,0 +1,153 @@
+/*
+ * Copyright 2010-2024 Australian Signals Directorate
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package au.gov.asd.tac.constellation.views.tableview.components;
+
+import au.gov.asd.tac.constellation.graph.GraphElementType;
+import au.gov.asd.tac.constellation.plugins.PluginExecution;
+import au.gov.asd.tac.constellation.views.tableview.TableViewTopComponent;
+import au.gov.asd.tac.constellation.views.tableview.plugins.UpdateStatePlugin;
+import au.gov.asd.tac.constellation.views.tableview.state.TableViewState;
+import javafx.event.ActionEvent;
+import javafx.event.EventHandler;
+import javafx.scene.control.ContextMenu;
+import javafx.scene.control.CustomMenuItem;
+import javafx.scene.control.Label;
+
+/**
+ *
+ * @author Quasar985
+ */
+public class ElementTypeContextMenu {
+
+    private static final String TRANSACTION = "Transactions";
+    private static final String VERTEX = "Vertices";
+    private static final String EDGE = "Edges";
+    private static final String LINK = "Links";
+
+    private final Table table;
+
+    private ContextMenu contextMenu;
+
+    private CustomMenuItem transactionsMenu;
+    private CustomMenuItem verticesMenu;
+    private CustomMenuItem edgesMenu;
+    private CustomMenuItem linksMenu;
+
+    /**
+     * Creates a new column visibility context menu.
+     *
+     * @param table the table that this menu was will be associated to
+     */
+    public ElementTypeContextMenu(final Table table) {
+        this.table = table;
+    }
+
+    /**
+     * Initializes the column visibility context menu. Until this method is
+     * called, all menu UI components will be null.
+     */
+    public void init() {
+        contextMenu = new ContextMenu();
+
+        transactionsMenu = createCustomMenu(TRANSACTION, e -> {
+            if (getTableViewTopComponent().getCurrentState() != null) {
+                final TableViewState newState = new TableViewState(getTableViewTopComponent().getCurrentState());
+                newState.setElementType(GraphElementType.TRANSACTION);
+
+                PluginExecution.withPlugin(
+                        new UpdateStatePlugin(newState)
+                ).executeLater(getTableViewTopComponent().getCurrentGraph());
+            }
+            e.consume();
+        });
+
+        verticesMenu = createCustomMenu(VERTEX, e -> {
+            if (getTableViewTopComponent().getCurrentState() != null) {
+                final TableViewState newState = new TableViewState(getTableViewTopComponent().getCurrentState());
+                newState.setElementType(GraphElementType.VERTEX);
+
+                PluginExecution.withPlugin(
+                        new UpdateStatePlugin(newState)
+                ).executeLater(getTableViewTopComponent().getCurrentGraph());
+            }
+            e.consume();
+        });
+
+        edgesMenu = createCustomMenu(EDGE, e -> {
+            if (getTableViewTopComponent().getCurrentState() != null) {
+                final TableViewState newState = new TableViewState(getTableViewTopComponent().getCurrentState());
+                newState.setElementType(GraphElementType.EDGE);
+
+                PluginExecution.withPlugin(
+                        new UpdateStatePlugin(newState)
+                ).executeLater(getTableViewTopComponent().getCurrentGraph());
+            }
+            e.consume();
+        });
+
+        linksMenu = createCustomMenu(LINK, e -> {
+            if (getTableViewTopComponent().getCurrentState() != null) {
+                final TableViewState newState = new TableViewState(getTableViewTopComponent().getCurrentState());
+                newState.setElementType(GraphElementType.LINK);
+
+                PluginExecution.withPlugin(
+                        new UpdateStatePlugin(newState)
+                ).executeLater(getTableViewTopComponent().getCurrentGraph());
+            }
+            e.consume();
+        });
+
+        contextMenu.getItems().addAll(transactionsMenu, verticesMenu, edgesMenu, linksMenu);
+    }
+
+    /**
+     * Create a custom menu item that will be added to menu buttons on the
+     * context menu. Sets the associated text and adds a listener for when it is
+     * clicked.
+     *
+     * @param title the title to be associated to the menu item
+     * @param handler the action handler that will be called when the menu item
+     * is clicked
+     * @return the created menu item
+     */
+    private CustomMenuItem createCustomMenu(final String title,
+            final EventHandler<ActionEvent> handler) {
+        final CustomMenuItem menuItem = new CustomMenuItem(new Label(title));
+
+        menuItem.setHideOnClick(false);
+        menuItem.setOnAction(handler);
+
+        return menuItem;
+    }
+
+    /**
+     * Convenience method for accessing the table view top component.
+     *
+     * @return the table view top component
+     */
+    private TableViewTopComponent getTableViewTopComponent() {
+        return table.getParentComponent().getParentComponent();
+    }
+
+    /**
+     * Gets the element type context menu.
+     *
+     * @return the element type context menu
+     */
+    public ContextMenu getContextMenu() {
+        return contextMenu;
+    }
+}