After simplification, build recycling queue from simplification results

if simulating neutral mutations.
molpopgen · Jan 26, 2021 · afcfe74 · afcfe74
1 parent 67bc52a
commit afcfe74
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/fwdpy11/src/evolve_population/with_tree_sequences.cc b/fwdpy11/src/evolve_population/with_tree_sequences.cc
@@ -504,8 +504,20 @@ evolve_with_tree_sequences(
                                             end(simplification_rv.second),
                                             std::numeric_limits<std::size_t>::max()),
                                 end(simplification_rv.second));
-                            genetics.mutation_recycling_bin = fwdpp::ts::make_mut_queue(
-                                pop.mcounts, pop.mcounts_from_preserved_nodes);
+                            if (simulating_neutral_variants)
+                                {
+                                    genetics.mutation_recycling_bin
+                                        = fwdpp::ts::make_mut_queue(
+                                            simplification_rv.second,
+                                            pop.mutations.size());
+                                }
+                            else
+                                {
+                                    genetics.mutation_recycling_bin
+                                        = fwdpp::ts::make_mut_queue(
+                                            pop.mcounts,
+                                            pop.mcounts_from_preserved_nodes);
+                                }
                         }
                     else
                         {