1 files changed, 6 insertions, 2 deletions

diff --git a/src/explicitsimulation.cpp b/src/explicitsimulation.cpp
index 98df0b0..95b721b 100644
--- a/src/explicitsimulation.cpp
+++ b/src/explicitsimulation.cpp
@@ -13,12 +13,13 @@ using namespace Bu;
 
 Genetic::ExplicitSimulation::ExplicitSimulation( Genetic::Operator *pOper,
                 Genetic::FitnessFunction *pFunc, int iThreads, int iPopSize,
-                float fKeep, float fRandom, bool bKeepBest ) :
+                float fKeep, float fRandom, bool bKeepBest, bool bRecalcSurvivors ) :
         pOper( pOper ),
         iPopSize( iPopSize ),
         fKeep( fKeep ),
         fRandom( fRandom ),
         bKeepBest( bKeepBest ),
+        bRecalcSurvivors( bRecalcSurvivors ),
         bRunning( true )
 {
         for( int j = 0; j < iPopSize; j++ )
@@ -102,7 +103,10 @@ void Genetic::ExplicitSimulation::timestep()
 
         // Refill the population
         mFitness.lock();
-        hFitness = hTempFitness;
+        if( bRecalcSurvivors )
+                hFitness.clear();
+        else
+                hFitness = hTempFitness;
         mFitness.unlock();
         xPop.clear();
         xPop.timestep();