Use enum class for nbnxm locality

[alexxy/gromacs.git] / src / gromacs / nbnxm / atomdata.h

diff --git a/src/gromacs/nbnxm/atomdata.h b/src/gromacs/nbnxm/atomdata.h
index ebfeee5bfbb35ba376505d401e48b61f2db74feb..bd210ea01444e107ed783a131748831213e86f08 100644 (file)
--- a/src/gromacs/nbnxm/atomdata.h
+++ b/src/gromacs/nbnxm/atomdata.h
@@ -42,6 +42,8 @@
 #include "gromacs/utility/basedefinitions.h"
 #include "gromacs/utility/real.h"
 
+#include "locality.h"
+
 namespace gmx
 {
 class MDLogger;
@@ -94,7 +96,7 @@ void nbnxn_atomdata_copy_shiftvec(gmx_bool          dynamic_box,
  * FillLocal tells if the local filler particle coordinates should be zeroed.
  */
 void nbnxn_atomdata_copy_x_to_nbat_x(const nbnxn_search  *nbs,
-                                     int                  locality,
+                                     Nbnxm::AtomLocality  locality,
                                      gmx_bool             FillLocal,
                                      rvec                *x,
                                      nbnxn_atomdata_t    *nbat,
@@ -102,7 +104,7 @@ void nbnxn_atomdata_copy_x_to_nbat_x(const nbnxn_search  *nbs,
 
 /* Add the forces stored in nbat to f, zeros the forces in nbat */
 void nbnxn_atomdata_add_nbat_f_to_f(nbnxn_search           *nbs,
-                                    int                     locality,
+                                    Nbnxm::AtomLocality     locality,
                                     nbnxn_atomdata_t       *nbat,
                                     rvec                   *f,
                                     gmx_wallcycle          *wcycle);