src/visitors/RigidBodyVisitor.hpp

#ifndef _RIGIDBODYVISITOR_H_
#define _RIGIDBODYVISITOR_H_
#include <iostream>
#include <set>
#include <string>
#include "visitors/BaseVisitor.hpp"
#include "utils/GenericData.hpp"

using namespace std;

class BaseRigidBodyVisitor: public BaseVisitor{
  public:

  protected:
     BaseRigidBodyVisitor(SimInfo* info) : BaseVisitor(){ this->info = info;}

      SimInfo* info;
};


//LipidHeadVisitor  adds a pesudo atom into rigidbody which holds a dipole moment
class LipidHeadVisitor : public BaseRigidBodyVisitor{
  public:
     LipidHeadVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "LipidHeadVisitor";}

    virtual void visit(Atom* atom) {}
    virtual void visit(DirectionalAtom* datom) {}
    virtual void visit(RigidBody* rb);
    
    void addLipidHeadName(const string& name);
    virtual const string toString();
    
  protected:
     bool canVisit(const string& name);

     set<string> lipidHeadName;
};

class RBCOMVisitor : public BaseRigidBodyVisitor{
  public:
    RBCOMVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "RBCOMVisitor";}

    virtual void visit(Atom* atom) {}
    virtual void visit(DirectionalAtom* datom) {}
    virtual void visit(RigidBody* rb);

    virtual const string toString();
};
#endif
Revision:	3
Committed:	Fri Sep 24 16:27:58 2004 UTC (21 years, 1 month ago) by tim
File size:	1269 byte(s)
Log Message:	change the #include in source files
#	User	Rev	Content
1	gezelter	2	#ifndef _RIGIDBODYVISITOR_H_
2			#define _RIGIDBODYVISITOR_H_
3			#include <iostream>
4			#include <set>
5			#include <string>
6	tim	3	#include "visitors/BaseVisitor.hpp"
7			#include "utils/GenericData.hpp"
8	gezelter	2
9			using namespace std;
10
11			class BaseRigidBodyVisitor: public BaseVisitor{
12			public:
13
14			protected:
15			BaseRigidBodyVisitor(SimInfo* info) : BaseVisitor(){ this->info = info;}
16
17			SimInfo* info;
18			};
19
20
21			//LipidHeadVisitor adds a pesudo atom into rigidbody which holds a dipole moment
22			class LipidHeadVisitor : public BaseRigidBodyVisitor{
23			public:
24			LipidHeadVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "LipidHeadVisitor";}
25
26			virtual void visit(Atom* atom) {}
27			virtual void visit(DirectionalAtom* datom) {}
28			virtual void visit(RigidBody* rb);
29
30			void addLipidHeadName(const string& name);
31			virtual const string toString();
32
33			protected:
34			bool canVisit(const string& name);
35
36			set<string> lipidHeadName;
37			};
38
39			class RBCOMVisitor : public BaseRigidBodyVisitor{
40			public:
41			RBCOMVisitor(SimInfo* info) : BaseRigidBodyVisitor(info){ visitorName = "RBCOMVisitor";}
42
43			virtual void visit(Atom* atom) {}
44			virtual void visit(DirectionalAtom* datom) {}
45			virtual void visit(RigidBody* rb);
46
47			virtual const string toString();
48			};
49			#endif