src/visitors/RigidBodyVisitor.hpp

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

using namespace std;

namespace oopse {
    
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();
};

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