Fix disappearing cow bug. If an export is not allowed, we have to mak…

…e sure the cows aren't removed from source herd still.
mrc-ide · Aug 19, 2024 · fba96e4 · fba96e4
1 parent 0eb6fb0
commit fba96e4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/inst/dust/cows.cpp b/inst/dust/cows.cpp
@@ -210,6 +210,11 @@ class cows {
           internal.import_E[i_dst] += internal.export_E[i_src];
           internal.import_I[i_dst] += internal.export_I[i_src];
           internal.import_R[i_dst] += internal.export_R[i_src];
+        } else {
+          internal.export_S[i_src] = 0;
+          internal.export_E[i_src] = 0;
+          internal.export_I[i_src] = 0;
+          internal.export_R[i_src] = 0;
         }
       }
     }

diff --git a/src/cows.cpp b/src/cows.cpp