zero total bug

alarm-redist · Jun 5, 2024 · 1ec4dd8 · 1ec4dd8
1 parent c62cef4
commit 1ec4dd8
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/segregation.cpp b/src/segregation.cpp
@@ -59,12 +59,19 @@ NumericVector segregationcalc(NumericMatrix distmat,
 
       }
 
-      // Get district proportions
-      double p = (double)gpop / tpop;
+      if (tpop == 0.0) {
 
-      // Add to dissimilarity index
-      dissim += (double)d * tpop * std::abs(p - P);
+        dissim += 0;
 
+      } else {
+
+        // Get district proportions
+        double p = (double)gpop / tpop;
+
+        // Add to dissimilarity index
+        dissim += (double)d * tpop * std::abs(p - P);
+
+      }
     }
 
     // Add to vector