Homec4science
Diffusion lammps (efficient_neuronet) Browse

lammps/lib/poems/body.cpp
d114fa1315a1efficient_neuronet

Last Change
Raw File

body.cpp
View Options

/*
*_________________________________________________________________________*
* 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;
}
}
c4science · Help