Exception handled when working with directed graphs.

mirsamantajbakhsh · Feb 14, 2015 · c00b67c · c00b67c
1 parent b6be992
commit c00b67c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/org/urmiauniversity/it/mst/cpm/CPM.java b/src/org/urmiauniversity/it/mst/cpm/CPM.java
@@ -204,7 +204,11 @@ public void execute(GraphModel gm, AttributeModel am) {
         for (Node vi : nodes) {
             for (Node vj : nodes) {
                 if ((vi != vj) && (getSharedNodes(vi, vj) == k - 1)) {
-                    edges.add(gm.factory().newEdge(vi, vj, 1.0f, false));
+                    if (g.getGraphModel().isDirected()) {
+                        edges.add(gm.factory().newEdge(vi, vj, 1.0f, true));
+                    } else {
+                        edges.add(gm.factory().newEdge(vi, vj, 1.0f, false));
+                    }
                 }
             }
         }