Replace FramesToCompare enum by MaxNumFrames

[alexxy/gromacs.git] / src / programs / mdrun / tests / trajectorycomparison.cpp

diff --git a/src/programs/mdrun/tests/trajectorycomparison.cpp b/src/programs/mdrun/tests/trajectorycomparison.cpp
index b4573bbeca10de29f28e1da3d6f0b4ce72da0ad7..16a8a0980c40ec534efc29944221e65db4e76081 100644 (file)
--- a/src/programs/mdrun/tests/trajectorycomparison.cpp
+++ b/src/programs/mdrun/tests/trajectorycomparison.cpp
@@ -252,7 +252,7 @@ TrajectoryComparison::TrajectoryComparison(const TrajectoryFrameMatchSettings& m
 
 void TrajectoryComparison::operator()(const TrajectoryFrame& reference, const TrajectoryFrame& test) const
 {
-    if (matchSettings_.framesToCompare == FramesToCompare::OnlyFirstFrame && firstFrameHasBeenCompared_)
+    if (numComparedFrames_ >= matchSettings_.maxNumTrajectoryFrames)
     {
         // Nothing should be compared
         return;
@@ -265,7 +265,7 @@ void TrajectoryComparison::operator()(const TrajectoryFrame& reference, const Tr
     compareCoordinates(reference, test, matchSettings_, tolerances_.coordinates);
     compareVelocities(reference, test, matchSettings_, tolerances_.velocities);
     compareForces(reference, test, matchSettings_, tolerances_.forces);
-    firstFrameHasBeenCompared_ = true;
+    numComparedFrames_++;
 }
 
 } // namespace test