ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@481 f3b2605a-c512-4ea7-a41b-209d697bcdaa

Browse Source
This commit is contained in:
sjplimp
2007-04-13 15:23:45 +00:00
parent e9dbae3501
commit 25c1766368
4 changed files with 18 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/fix_enforce2d.cpp
View File

@ -41,13 +41,6 @@ int FixEnforce2D::setmask()
/* ---------------------------------------------------------------------- */
void FixEnforce2D::init()
{
granular = atom->check_style("granular");
}
/* ---------------------------------------------------------------------- */
void FixEnforce2D::setup()
{
if (strcmp(update->integrate_style,"verlet") == 0)
@ -74,7 +67,9 @@ void FixEnforce2D::min_setup()
void FixEnforce2D::post_force(int vflag)
{
double **v = atom->v;
double **omega = atom->omega;
double **f = atom->f;
double **torque = atom->torque;
int *mask = atom->mask;
int nlocal = atom->nlocal;
@ -84,17 +79,21 @@ void FixEnforce2D::post_force(int vflag)
f[i][2] = 0.0;
}
// for granular systems, zero xy rotational componenets
// for omega/torque systems, zero xy rotation
if (granular) {
double **phiv = atom->phiv;
double **phia = atom->phia;
if (omega) {
for (int i = 0; i < nlocal; i++)
if (mask[i] & groupbit) {
phiv[i][0] = 0.0;
phiv[i][1] = 0.0;
phia[i][0] = 0.0;
phia[i][1] = 0.0;
omega[i][0] = 0.0;
omega[i][1] = 0.0;
}
}
if (torque) {
for (int i = 0; i < nlocal; i++)
if (mask[i] & groupbit) {
torque[i][0] = 0.0;
torque[i][1] = 0.0;
}
}
}