[ViewVC] Diff of: group/trunk/OOPSE/utils/MoLocator.cpp

Comparing trunk/OOPSE/utils/MoLocator.cpp (file contents):
Revision 501 by mmeineke, Tue Apr 15 21:20:35 2003 UTC vs.
Revision 504 by mmeineke, Thu Apr 17 21:54:18 2003 UTC

-+
+#include <iostream>
-+
-+
+#include <cstdlib>
-+
+#include <cmath>
-+
-+
+#include "simError.h"
-+
+#include "MoLocator.hpp"
-+
-+
-+
+MoLocator::MoLocator( MoleculeStamp* theStamp ){
-+
-+
+  myStamp = theStamp;
-+
+  nAtoms = myStamp->getNAtoms();
-+
-+
+  myCoords = NULL;
-+
-+
+  calcRefCoords();
-+
+}
-+
-+
+MoLocator::~MoLocator(){
-+
-+
+  if( myCoords != NULL ) delete[] myCoords;
-+
+}
-+
-+
+void MoLocator::placeMol( double pos[3], double A[3][3], Atom** atomArray,
-+
+                          int atomIndex ){
-+
-+
+  int i,j,k;
-+
+  double r[3];
-+
+  double ux, uy, uz, u, uSqr;
-+
-+
+  AtomStamp* currAtom;
-+
+  DirectionalAtom* dAtom;
-+
-+
+  for(i=0; i<nAtoms; i++){
-+
-+
+    currAtom = myStamp->getAtom( i );
-+
+    j = atomIndex+i;
-+
-+
+    if( currAtom->haveOrientation()){
-+
-+
+      dAtom = new DirectionalAtom( j );
-+
+      atomArray[j] = dAtom;
-+
-+
+      ux = currAtom->getOrntX();
-+
+      uy = currAtom->getOrntY();
-+
+      uz = currAtom->getOrntZ();
-+
-+
+      uSqr = (ux * ux) + (uy * uy) + (uz * uz);
-+
-+
+      u = sqrt( uSqr );
-+
+      ux = ux / u;
-+
+      uy = uy / u;
-+
+      uz = uz / u;
-+
-+
+      dAtom->setSUx( ux );
-+
+      dAtom->setSUy( uy );
-+
+      dAtom->setSUz( uz );
-+
-+
+      dAtom->setA( A );
-+
-+
+      dAtom->setJx( 0.0 );
-+
+      dAtom->setJy( 0.0 );
-+
+      dAtom->setJz( 0.0 );
-+
-+
+    }
-+
+    else{
-+
+      atomArray[j] = new GeneralAtom( j );
-+
+    }
-+
-+
+    atomArray[j]->setType( currAtom->getType() );
-+
-+
+    for(k=0; k<3; k++) r[k] = myCoords[(i*3)+k];
-+
-+
+    rotMe( r, A );
-+
-+
+    for(k=0; k<3; k++) r[k] += pos[k];
-+
-+
+    atomArray[j]->setX( r[0] );
-+
+    atomArray[j]->setY( r[1] );
-+
+    atomArray[j]->setZ( r[2] );
-+
-+
+    atomArray[j]->set_vx( 0.0 );
-+
+    atomArray[j]->set_vy( 0.0 );
-+
+    atomArray[j]->set_vz( 0.0 );
-+
+  }
-+
+}
-+
-+
+void MoLocator::calcRefCoords( void ){
-+
-+
+  int i,j,k;
-+
+  AtomStamp* currAtom;
-+
+  double centerX, centerY, centerZ;
-+
+  double smallX, smallY, smallZ;
-+
+  double bigX, bigY, bigZ;
-+
+  double dx, dy, dz;
-+
+  double dsqr;
-+
-+
-+
+  centerX = 0.0;
-+
+  centerY = 0.0;
-+
+  centerZ = 0.0;
-+
+  for(i=0; i<nAtoms; i++){
-+
-+
+    currAtom = myStamp->getAtom(i);
-+
+    if( !currAtom->havePosition() ){
-+
+      sprintf( painCave.errMsg,
-+
+               "MoLocator error.\n"
-+
+               "  Component %s, atom %s does not have a position specified.\n"
-+
+               "  This means MoLocator cannot initalize it's position.\n",
-+
+               myStamp->getID(),
-+
+               currAtom->getType() );
-+
+      painCave.isFatal = 1;
-+
+      simError();
-+
+    }
-+
-+
-+
+    centerX += currAtom->getPosX();
-+
+    centerY += currAtom->getPosY();
-+
+    centerZ += currAtom->getPosZ();
-+
+  }
-+
-+
+  centerX /= nAtoms;
-+
+  centerY /= nAtoms;
-+
+  centerZ /= nAtoms;
-+
-+
+  myCoords = new double[nAtoms*3];
-+
-+
+  j = 0;
-+
+  for(i=0; i<nAtoms; i++){
-+
-+
+    currAtom = myStamp->getAtom(i);
-+
+    j = i*3;
-+
-+
+    myCoords[j]   = currAtom->getPosX() - centerX;
-+
+    myCoords[j+1] = currAtom->getPosY() - centerY;
-+
+    myCoords[j+2] = currAtom->getPosZ() - centerZ;
-+
+  }
-+
-+
+  smallX = myCoords[0];
-+
+  smallY = myCoords[1];
-+
+  smallZ = myCoords[2];
-+
-+
+  bigX = myCoords[0];
-+
+  bigY = myCoords[1];
-+
+  bigZ = myCoords[2];
-+
-+
+  j=0;
-+
+  for(i=1; i<nAtoms; i++){
-+
+    j= i*3;
-+
-+
+    if( myCoords[j]   < smallX ) smallX = myCoords[j];
-+
+    if( myCoords[j+1] < smallY ) smallY = myCoords[j+1];
-+
+    if( myCoords[j+2] < smallZ ) smallZ = myCoords[j+2];
-+
-+
+    if( myCoords[j]   > bigX ) bigX = myCoords[j];
-+
+    if( myCoords[j+1] > bigY ) bigY = myCoords[j+1];
-+
+    if( myCoords[j+2] > bigZ ) bigZ = myCoords[j+2];
-+
+  }
-+
-+
-+
+  dx = bigX - smallX;
-+
+  dy = bigY - smallY;
-+
+  dz = bigZ - smallZ;
-+
-+
+  dsqr = (dx * dx) + (dy * dy) + (dz * dz);
-+
+  maxLength = sqrt( dsqr );
-+
+}
-+
-+
+void MoLocator::rotMe( double r[3], double A[3][3] ){
-+
-+
+  double rt[3];
-+
+  int i,j;
-+
-+
+  for(i=0; i<3; i++) rt[i] = r[i];
-+
-+
+  for(i=0; i<3; i++){
-+
+    r[i] = 0.0;
-+
+    for(j=0; j<3; j++){
-+
+      r[i] += A[i][j] * rt[j];
-+
+    }
-+
+  }
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing trunk/OOPSE/utils/MoLocator.cpp (file contents): Revision 501 by mmeineke, Tue Apr 15 21:20:35 2003 UTC vs. Revision 504 by mmeineke, Thu Apr 17 21:54:18 2003 UTC

Diff Legend

Comparing trunk/OOPSE/utils/MoLocator.cpp (file contents):
Revision 501 by mmeineke, Tue Apr 15 21:20:35 2003 UTC vs.
Revision 504 by mmeineke, Thu Apr 17 21:54:18 2003 UTC