Rework state checkpoint enum handling

[alexxy/gromacs.git] / src / gromacs / domdec / distribute.cpp

diff --git a/src/gromacs/domdec/distribute.cpp b/src/gromacs/domdec/distribute.cpp
index efa2611c1b6742c1f408560682bc466805f8437b..9adb561d415741a49d1ff1bb93507fb5334624bc 100644 (file)
--- a/src/gromacs/domdec/distribute.cpp
+++ b/src/gromacs/domdec/distribute.cpp
@@ -262,15 +262,15 @@ static void dd_distribute_state(gmx_domdec_t* dd, const t_state* state, t_state*
 
     state_change_natoms(state_local, dd->comm->atomRanges.numHomeAtoms());
 
-    if (state_local->flags & (1 << estX))
+    if (state_local->flags & enumValueToBitMask(StateEntry::X))
     {
         distributeVec(dd, DDMASTER(dd) ? state->x : gmx::ArrayRef<const gmx::RVec>(), state_local->x);
     }
-    if (state_local->flags & (1 << estV))
+    if (state_local->flags & enumValueToBitMask(StateEntry::V))
     {
         distributeVec(dd, DDMASTER(dd) ? state->v : gmx::ArrayRef<const gmx::RVec>(), state_local->v);
     }
-    if (state_local->flags & (1 << estCGP))
+    if (state_local->flags & enumValueToBitMask(StateEntry::Cgp))
     {
         distributeVec(dd, DDMASTER(dd) ? state->cg_p : gmx::ArrayRef<const gmx::RVec>(), state_local->cg_p);
     }