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

lib/poems/body.cpp

@@ -0,0 +1,137 @@
+/*
+ *_________________________________________________________________________*
+ *      POEMS: PARALLELIZABLE OPEN SOURCE EFFICIENT MULTIBODY SOFTWARE     *
+ *      DESCRIPTION: SEE READ-ME                                           *
+ *      FILE NAME: body.cpp                                                *
+ *      AUTHORS: See Author List                                           * 
+ *      GRANTS: See Grants List                                            *
+ *      COPYRIGHT: (C) 2005 by Authors as listed in Author's List          *
+ *      LICENSE: Please see License Agreement                              *
+ *      DOWNLOAD: Free at www.rpi.edu/~anderk5                             *
+ *      ADMINISTRATOR: Prof. Kurt Anderson                                 *
+ *                     Computational Dynamics Lab                          *
+ *                     Rensselaer Polytechnic Institute                    *
+ *                     110 8th St. Troy NY 12180                           * 
+ *      CONTACT:        anderk5@rpi.edu                                    *
+ *_________________________________________________________________________*/
+ 
+#include "bodies.h"
+#include "point.h"
+
+using namespace std;
+
+Body::Body()
+{
+  inertia.Zeros();
+  mass = 0;
+  a_t.Zeros();
+}
+
+Body::~Body()
+{
+  points.DeleteValues();
+}
+
+bool Body::ReadIn(istream& in){
+  return (ReadInBodyData(in) && ReadInPoints(in));
+}
+
+void Body::WriteOut(ostream& out){
+  // write out header <type> <name>
+  out << GetType() << ' ' << GetName() << endl;
+
+  // write out body specific data
+  WriteOutBodyData(out);
+
+  // write out points
+  WriteOutPoints(out);
+
+}
+
+bool Body::ReadInPoints(istream& in){
+  // get numfixed points
+  int numpoints;
+  int index;
+  Point* point;
+  int pointtype;
+  char pointname[256];
+
+  in >> numpoints;
+
+  for(int i=points.GetNumElements();i<numpoints;i++){
+    // error check
+    in >> index;
+    if(index != i){
+      cerr << "Invalid file format" << endl;
+      return false;
+    }
+
+    in >> pointtype >> pointname;
+    point = NewPoint(pointtype);
+
+    if(!point){
+      cerr << "Unrecognized point type '" << pointtype << endl;
+      return false;
+    }
+
+    // add the point
+    AddPoint(point);
+
+    // set generic point info
+    point->ChangeName(pointname);
+
+    // read in the rest of its data
+    if(!point->ReadIn(in)) return false;
+  }
+  return true;
+}
+
+void Body::WriteOutPoints(std::ostream& out){
+  int numpoints = points.GetNumElements();
+
+  out << numpoints << endl;
+
+  // list element pointer
+  ListElement<Point>* ele = points.GetHeadElement();
+
+  // set the ID of each point
+  for(int i=0;i<numpoints;i++){
+    ele->value->SetID(i);
+    out << i << ' ';
+    ele->value->WriteOut(out);
+
+    // increment list element pointer
+    ele = ele->next;
+  }
+  out << endl;
+}
+
+Point* Body::GetPoint(int p) {
+  return points(p);
+}
+
+void Body::AddJoint(Joint* joint){
+  joints.Append(joint);
+}
+
+void Body::AddPoint(Point* point){
+  points.Append(point);
+}
+
+//
+// global body functions
+//
+
+Body* NewBody(int type){
+  switch( BodyType(type) )
+    {
+      case INERTIALFRAME :  // The inertial reference frame
+        return new InertialFrame;
+      case RIGIDBODY :  // A Rigid Body
+        return new RigidBody;
+      case PARTICLE :  // A Particle
+        return new Particle;
+      default  :  // error
+        return 0;
+    }
+}