OOPSE/libmdtools/OtherVisitor.hpp

#ifndef _OTHERVISITOR_H_
#define _OTHERVISITOR_H_
#include <set>
#include <string>
#include <vector>

#include "BaseVisitor.hpp"
#include "StuntDouble.hpp"
using namespace std;

//IgnoreVisitor will turn on the ignoring flag of the stuntdouble
class IgnoreVisitor : public BaseVisitor{
  public:
    IgnoreVisitor() : BaseVisitor() {visitorName = "IgnoreVisitor";}

    virtual void visit(Atom* atom);
    virtual void visit(DirectionalAtom* datom);
    virtual void visit(RigidBody* rb);
    
    virtual const string toString();

    void addIgnoreType(const string& type) {itList.insert(type);}
    
  protected:
    bool isIgnoreType(const string& name);
    void internalVisit(StuntDouble* sd);
    set<string> itList; //ignore type list;
};


class WrappingVisitor : public BaseVisitor{
  public:
    WrappingVisitor(SimInfo* info) : BaseVisitor() {
      this->info = info;
      visitorName = "WrappingVisitor";
    }
    virtual void visit(Atom* atom);
    virtual void visit(DirectionalAtom* datom);
    virtual void visit(RigidBody* rb);

    virtual const string toString();

  protected:
    void internalVisit(StuntDouble* sd);
    SimInfo* info;
};


class IntVec3 {
  public:
    IntVec3(){}
    IntVec3(int i, int j, int k){
      vec[0] = i;
      vec[1] = j;
      vec[2] = k;
    }
    
    int vec[3];
    int& operator[](int index) {return vec[index];}  
};

class ReplicateVisitor : public BaseVisitor{
  public:
    ReplicateVisitor(SimInfo* info, IntVec3 opt);
    virtual void visit(Atom* atom);
    virtual void visit(DirectionalAtom* datom);
    virtual void visit(RigidBody* rb);

    virtual const string toString();
  protected:
    void internalVisit(StuntDouble* sd);
    void replicate(vector<AtomInfo*>& infoList,  AtomData* data, double boxM[3][3]);
    
  private:
    vector<IntVec3> dir;
    SimInfo* info;
    IntVec3 replicateOpt;
};

class XYZVisitor : public BaseVisitor{
  public:
    XYZVisitor(SimInfo* info, bool printDipole = true);
    
    virtual void visit(Atom* atom);
    virtual void visit(DirectionalAtom* datom);
    virtual void visit(RigidBody* rb);

    virtual const string toString();
    
    void writeFrame(ostream& outStream);    
    void clear() {frame.clear();}
    
  protected:
    void internalVisit(StuntDouble* sd);
    bool isIgnore(StuntDouble* sd);

  private:  
    SimInfo* info;
    vector<string> frame;
    bool printDipole;
};


class PrepareVisitor : public BaseVisitor{
  public:
    PrepareVisitor() : BaseVisitor() {visitorName = "prepareVisitor";}

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

    virtual const string toString();

  protected:
    void internalVisit(Atom* atom);
    void internalVisit(RigidBody* rb);
};

class WaterTypeVisitor : public BaseVisitor{
  public:
    WaterTypeVisitor() ;
    virtual void visit(Atom* atom) {}
    virtual void visit(DirectionalAtom* datom) {}
    virtual void visit(RigidBody* rb);

    virtual const string toString();
    
  private:
    void replaceType(string& atomType);
      
    set<string> waterTypeList;
};
#endif //_OTHERVISITOR_H_
Revision:	1126
Committed:	Tue Apr 20 05:39:38 2004 UTC (21 years, 6 months ago) by tim
File size:	3231 byte(s)
Log Message:	DUMP2XYZ 0.99 version
#	Content
1	#ifndef _OTHERVISITOR_H_
2	#define _OTHERVISITOR_H_
3	#include <set>
4	#include <string>
5	#include <vector>
6
7	#include "BaseVisitor.hpp"
8	#include "StuntDouble.hpp"
9	using namespace std;
10
11	//IgnoreVisitor will turn on the ignoring flag of the stuntdouble
12	class IgnoreVisitor : public BaseVisitor{
13	public:
14	IgnoreVisitor() : BaseVisitor() {visitorName = "IgnoreVisitor";}
15
16	virtual void visit(Atom* atom);
17	virtual void visit(DirectionalAtom* datom);
18	virtual void visit(RigidBody* rb);
19
20	virtual const string toString();
21
22	void addIgnoreType(const string& type) {itList.insert(type);}
23
24	protected:
25	bool isIgnoreType(const string& name);
26	void internalVisit(StuntDouble* sd);
27	set<string> itList; //ignore type list;
28	};
29
30
31	class WrappingVisitor : public BaseVisitor{
32	public:
33	WrappingVisitor(SimInfo* info) : BaseVisitor() {
34	this->info = info;
35	visitorName = "WrappingVisitor";
36	}
37	virtual void visit(Atom* atom);
38	virtual void visit(DirectionalAtom* datom);
39	virtual void visit(RigidBody* rb);
40
41	virtual const string toString();
42
43	protected:
44	void internalVisit(StuntDouble* sd);
45	SimInfo* info;
46	};
47
48
49	class IntVec3 {
50	public:
51	IntVec3(){}
52	IntVec3(int i, int j, int k){
53	vec[0] = i;
54	vec[1] = j;
55	vec[2] = k;
56	}
57
58	int vec[3];
59	int& operator[](int index) {return vec[index];}
60	};
61
62	class ReplicateVisitor : public BaseVisitor{
63	public:
64	ReplicateVisitor(SimInfo* info, IntVec3 opt);
65	virtual void visit(Atom* atom);
66	virtual void visit(DirectionalAtom* datom);
67	virtual void visit(RigidBody* rb);
68
69	virtual const string toString();
70	protected:
71	void internalVisit(StuntDouble* sd);
72	void replicate(vector<AtomInfo>& infoList, AtomData data, double boxM[3][3]);
73
74	private:
75	vector<IntVec3> dir;
76	SimInfo* info;
77	IntVec3 replicateOpt;
78	};
79
80	class XYZVisitor : public BaseVisitor{
81	public:
82	XYZVisitor(SimInfo* info, bool printDipole = true);
83
84	virtual void visit(Atom* atom);
85	virtual void visit(DirectionalAtom* datom);
86	virtual void visit(RigidBody* rb);
87
88	virtual const string toString();
89
90	void writeFrame(ostream& outStream);
91	void clear() {frame.clear();}
92
93	protected:
94	void internalVisit(StuntDouble* sd);
95	bool isIgnore(StuntDouble* sd);
96
97	private:
98	SimInfo* info;
99	vector<string> frame;
100	bool printDipole;
101	};
102
103
104	class PrepareVisitor : public BaseVisitor{
105	public:
106	PrepareVisitor() : BaseVisitor() {visitorName = "prepareVisitor";}
107
108	virtual void visit(Atom* atom) {internalVisit(atom);}
109	virtual void visit(DirectionalAtom* datom) {internalVisit((Atom*)datom);}
110	virtual void visit(RigidBody* rb) {internalVisit(rb);}
111
112	virtual const string toString();
113
114	protected:
115	void internalVisit(Atom* atom);
116	void internalVisit(RigidBody* rb);
117	};
118
119	class WaterTypeVisitor : public BaseVisitor{
120	public:
121	WaterTypeVisitor() ;
122	virtual void visit(Atom* atom) {}
123	virtual void visit(DirectionalAtom* datom) {}
124	virtual void visit(RigidBody* rb);
125
126	virtual const string toString();
127
128	private:
129	void replaceType(string& atomType);
130
131	set<string> waterTypeList;
132	};
133	#endif //_OTHERVISITOR_H_