More bugfixes for empty occurrences

pbloem · Aug 17, 2016 · f76453c · f76453c
1 parent 143c1ce
commit f76453c
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/src/main/java/nl/peterbloem/motive/MotifModel.java b/src/main/java/nl/peterbloem/motive/MotifModel.java
@@ -291,9 +291,12 @@ public static List<D> subbedDegrees(
 		rest.add("multi-edges", OnlineModel.storeIntegers(additions)); 
 
 		// * check for any disconnected nodes and add 0s
-		int expSize = graph.size() - occurrences.size() * (occurrences.get(0).size() - 1);
-		while(degrees.size() < expSize)
-			degrees.add(new D(0, 0));
+		if(! occurrences.isEmpty())
+		{	
+			int expSize = graph.size() - occurrences.size() * (occurrences.get(0).size() - 1);
+			while(degrees.size() < expSize)
+				degrees.add(new D(0, 0));
+		}
 
 		return degrees;
 	}
@@ -432,9 +435,12 @@ public static List<Integer> subbedDegrees(
 		rest.add("multi-edges", OnlineModel.storeIntegers(additions)); 
 
 		// * check for any disconnected nodes and add 0s
-		int expSize = graph.size() - occurrences.size() * (occurrences.get(0).size() - 1);
-		while(result.size() < expSize)
-			result.add(0);
+		if(! occurrences.isEmpty())
+		{
+			int expSize = graph.size() - occurrences.size() * (occurrences.get(0).size() - 1);
+			while(result.size() < expSize)
+				result.add(0);
+		}
 
 		return result;
 	}