fixed DegreeModelComponent to work for directed networks

narnolddd · Mar 7, 2023 · 353861a · 353861a
1 parent 611ae1c
commit 353861a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/feta/objectmodels/components/DegreeModelComponent.java b/feta/objectmodels/components/DegreeModelComponent.java
@@ -57,6 +57,20 @@ public void updateNormalisation(UndirectedNetwork net, HashSet<Integer> availabl
         }
     }
 
+    public void updateNormalisation(DirectedNetwork net, HashSet<Integer> availableNodes, int chosenNode) {
+        if (!random_) {
+            if (useInDegree_) {
+                tempConstant_ -= net.getInDegree(chosenNode);
+            } else {
+                tempConstant_ -= net.getOutDegree(chosenNode);
+            }
+        }
+        if (random_ || tempConstant_==0) {
+            random_=true;
+            tempConstant_=availableNodes.size();
+        }
+    }
+
     public void calcNormalisation(UndirectedNetwork network, int [] removed) {
         int degSum = 0;
         for (int j : removed) {