From b7b70db581981582bfc567f2d80d27f2bb533503 Mon Sep 17 00:00:00 2001 From: sjplimp Date: Fri, 17 Oct 2008 15:49:13 +0000 Subject: [PATCH] git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@2173 f3b2605a-c512-4ea7-a41b-209d697bcdaa --- src/fix_line_force.cpp | 8 ++++++++ src/fix_plane_force.cpp | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/src/fix_line_force.cpp b/src/fix_line_force.cpp index d6750b4d01..530556b380 100644 --- a/src/fix_line_force.cpp +++ b/src/fix_line_force.cpp @@ -11,6 +11,7 @@ See the README file in the top-level LAMMPS directory. ------------------------------------------------------------------------- */ +#include "math.h" #include "string.h" #include "stdlib.h" #include "fix_line_force.h" @@ -30,6 +31,13 @@ FixLineForce::FixLineForce(LAMMPS *lmp, int narg, char **arg) : xdir = atof(arg[3]); ydir = atof(arg[4]); zdir = atof(arg[5]); + + double len = sqrt(xdir*xdir + ydir*ydir + zdir*zdir); + if (len == 0.0) error->all("Illegal fix lineforce command"); + + xdir /= len; + ydir /= len; + zdir /= len; } /* ---------------------------------------------------------------------- */ diff --git a/src/fix_plane_force.cpp b/src/fix_plane_force.cpp index e4b7d98b72..873db6e165 100644 --- a/src/fix_plane_force.cpp +++ b/src/fix_plane_force.cpp @@ -11,6 +11,7 @@ See the README file in the top-level LAMMPS directory. ------------------------------------------------------------------------- */ +#include "math.h" #include "string.h" #include "stdlib.h" #include "fix_plane_force.h" @@ -30,6 +31,13 @@ FixPlaneForce::FixPlaneForce(LAMMPS *lmp, int narg, char **arg) : xdir = atof(arg[3]); ydir = atof(arg[4]); zdir = atof(arg[5]); + + double len = sqrt(xdir*xdir + ydir*ydir + zdir*zdir); + if (len == 0.0) error->all("Illegal fix planeforce command"); + + xdir /= len; + ydir /= len; + zdir /= len; } /* ---------------------------------------------------------------------- */