Implement third feature.

theViz343 · Oct 8, 2023 · 86547b1 · 86547b1
1 parent 64a24a0
commit 86547b1
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/src/main/java/GraphData.java b/src/main/java/GraphData.java
@@ -64,13 +64,36 @@ public void addNodes(String[] labels) {
             addNode(label);
         }
     }
+
+    public void addEdge(String srcLabel, String dstLabel) {
+        boolean srcnodeexisting = graphObject.vertexSet().stream().anyMatch(v -> Objects.equals(v, srcLabel));
+        boolean dstnodeexisting = graphObject.vertexSet().stream().anyMatch(v -> Objects.equals(v, dstLabel));
+        DefaultEdge edgeexisting = graphObject.getEdge(srcLabel, dstLabel);
+        if (edgeexisting!=null) {
+            System.out.println("Edge "+ edgeexisting + " already exists!");
+            return;
+        }
+        if (!srcnodeexisting) {
+            System.out.println("Node "+ srcLabel+" does not exist!");
+        } else if (!dstnodeexisting) {
+            System.out.println("Node "+ dstLabel+" does not exist!");
+        } else {
+            graphObject.addEdge(srcLabel, dstLabel);
+        }
+    }
     public static void main(String[] args) throws IOException {
         GraphData graphApi = new GraphData();
         graphApi.parseGraph("src/main/example.dot");
         System.out.println(graphApi.toString());
         graphApi.outputGraph("src/main/output.txt");
         graphApi.addNodes(new String[]{"Z", "X"});
         System.out.println(graphApi.toString());
+        graphApi.addEdge("Z", "X");
+        System.out.println(graphApi.toString());
+        graphApi.addEdge("Z", "X");
+        System.out.println(graphApi.toString());
+
+