diff --git a/src/fix_enforce2d.cpp b/src/fix_enforce2d.cpp index fff0f53535..31a3e0d31a 100644 --- a/src/fix_enforce2d.cpp +++ b/src/fix_enforce2d.cpp @@ -15,6 +15,7 @@ #include "fix_enforce2d.h" #include "atom.h" #include "update.h" +#include "domain.h" #include "respa.h" #include "error.h" @@ -41,6 +42,14 @@ int FixEnforce2D::setmask() /* ---------------------------------------------------------------------- */ +void FixEnforce2D::init() +{ + if (domain->dimension == 3) + error->all("Should not use fix enforce2d with 3d simulation"); +} + +/* ---------------------------------------------------------------------- */ + void FixEnforce2D::setup(int vflag) { if (strcmp(update->integrate_style,"verlet") == 0) diff --git a/src/fix_enforce2d.h b/src/fix_enforce2d.h index b1b9440e72..f8f2a415c8 100644 --- a/src/fix_enforce2d.h +++ b/src/fix_enforce2d.h @@ -22,6 +22,7 @@ class FixEnforce2D : public Fix { public: FixEnforce2D(class LAMMPS *, int, char **); int setmask(); + void init(); void setup(int); void min_setup(int); void post_force(int);