Add Diversity Processor Calculator iterator

lczech · May 29, 2024 · 7240f36 · 7240f36
1 parent 69a4531
commit 7240f36
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/lib/genesis/population/function/diversity_pool_processor.hpp b/lib/genesis/population/function/diversity_pool_processor.hpp
@@ -273,6 +273,20 @@ class DiversityPoolProcessor
         return calculators_;
     }
 
+    // -------------------------------------------------------------------------
+    //     Calculator Iterator
+    // -------------------------------------------------------------------------
+
+    std::vector<DiversityPoolCalculator>::iterator begin()
+    {
+        return calculators_.begin();
+    }
+
+    std::vector<DiversityPoolCalculator>::iterator end()
+    {
+        return calculators_.end();
+    }
+
     // -------------------------------------------------------------------------
     //     Internal Members
     // -------------------------------------------------------------------------

diff --git a/test/src/population/diversity.cpp b/test/src/population/diversity.cpp
@@ -539,6 +539,11 @@ void test_diversity_fuzzy_run_( std::vector<Variant> const& data )
         window_average_policy, settings, pool_sizes
     );
 
+    // Set some settings, to test the iterator
+    for( auto& calc : processor ) {
+        calc.enable_tajima_d( true );
+    }
+
     // Run the data
     for( auto const& variant : data ) {
         processor.process( variant );