| 12 |
|
#define ZCONSFORCEPOLICY_ID "ZCONSFORCEPOLICY" |
| 13 |
|
#define ZCONSGAP_ID "ZCONSGAP" |
| 14 |
|
#define ZCONSFIXTIME_ID "ZCONSFIXTIME" |
| 15 |
+ |
#define ZCONSUSINGSMD_ID "ZCONSUSINGSMD" |
| 16 |
|
|
| 17 |
|
#define CHIVALUE_ID "CHIVALUE" |
| 18 |
|
#define INTEGRALOFCHIDT_ID "INTEGRALOFCHIDT" |
| 38 |
|
/** Something we can improve it here is to use template |
| 39 |
|
** |
| 40 |
|
*/ |
| 41 |
+ |
|
| 42 |
+ |
class IntData : public GenericData{ |
| 43 |
+ |
|
| 44 |
+ |
public: |
| 45 |
+ |
|
| 46 |
+ |
double getData() { return data; } |
| 47 |
+ |
void setData(int rhs) { data = rhs; } |
| 48 |
+ |
|
| 49 |
+ |
protected: |
| 50 |
+ |
int data; |
| 51 |
+ |
}; |
| 52 |
+ |
|
| 53 |
|
class DoubleData : public GenericData{ |
| 54 |
|
|
| 55 |
|
public: |
| 75 |
|
bool havingZPos; |
| 76 |
|
double zPos; |
| 77 |
|
double kRatio; |
| 78 |
+ |
bool havingCantVel; |
| 79 |
+ |
double cantVel; |
| 80 |
|
}; |
| 81 |
|
|
| 82 |
|
class ZConsParaData : public GenericData{ |
| 113 |
|
protected: |
| 114 |
|
vector<double> data; |
| 115 |
|
}; |
| 116 |
+ |
|
| 117 |
+ |
struct AtomInfo : public GenericData{ |
| 118 |
+ |
public: |
| 119 |
+ |
string AtomType; |
| 120 |
+ |
double pos[3]; |
| 121 |
+ |
double dipole[3]; |
| 122 |
+ |
}; |
| 123 |
+ |
|
| 124 |
+ |
class AtomData : public GenericData{ |
| 125 |
+ |
public: |
| 126 |
+ |
~AtomData(); |
| 127 |
+ |
void addAtomInfo(AtomInfo* info) {data.push_back(info);} |
| 128 |
+ |
void clearAllAtomInfo(); |
| 129 |
+ |
AtomInfo* beginAtomInfo(vector<AtomInfo*>::iterator& i){ |
| 130 |
+ |
i = data.begin(); |
| 131 |
+ |
return i != data.end()? *i : NULL; |
| 132 |
+ |
} |
| 133 |
+ |
AtomInfo* nextAtomInfo(vector<AtomInfo*>::iterator& i){ |
| 134 |
+ |
i++; |
| 135 |
+ |
return i != data.end()? *i: NULL; |
| 136 |
+ |
} |
| 137 |
+ |
vector<AtomInfo*> getData() {return data;} |
| 138 |
+ |
int getSize() {return data.size();} |
| 139 |
+ |
protected: |
| 140 |
+ |
vector<AtomInfo*> data; |
| 141 |
+ |
}; |
| 142 |
+ |
|
| 143 |
+ |
|
| 144 |
+ |
|
| 145 |
|
#endif |