OOPSE/libmdtools/RigidBodyVisitor.hpp

#ifndef _RIGIDBODYVISITOR_H_
#define _RIGIDBODYVISITOR_H_
#include <iostream>
#include <set>
#include <string>
#include "BaseVisitor.hpp"
#include "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:	1126
Committed:	Tue Apr 20 05:39:38 2004 UTC (20 years, 2 months ago) by tim
File size:	1254 byte(s)
Log Message:	DUMP2XYZ 0.99 version
#	Content
1	#ifndef _RIGIDBODYVISITOR_H_
2	#define _RIGIDBODYVISITOR_H_
3	#include <iostream>
4	#include <set>
5	#include <string>
6	#include "BaseVisitor.hpp"
7	#include "GenericData.hpp"
8
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