| 1 |
tim |
66 |
#include <iostream> |
| 2 |
|
|
#include <vector> |
| 3 |
|
|
|
| 4 |
|
|
using namespace std; |
| 5 |
|
|
|
| 6 |
|
|
class TFrame |
| 7 |
|
|
{ |
| 8 |
|
|
protected: |
| 9 |
|
|
int _numAtom; |
| 10 |
|
|
int _numPseudo; |
| 11 |
|
|
|
| 12 |
|
|
vector<float> _coor; |
| 13 |
|
|
vector<float> _velo; |
| 14 |
|
|
vector<float> _force; |
| 15 |
|
|
map<string, float> _energy; |
| 16 |
|
|
map<string, > _extra; |
| 17 |
|
|
|
| 18 |
|
|
public: |
| 19 |
|
|
float _boxLen[3]; |
| 20 |
|
|
float _alpha; |
| 21 |
|
|
float _beta; |
| 22 |
|
|
float _gamma; |
| 23 |
|
|
|
| 24 |
|
|
int GetNumAtom() { return _numAtom;} |
| 25 |
|
|
float * GetNumPseudo() { return _numPseudo;} |
| 26 |
|
|
float * GetEnergy(string energyName); |
| 27 |
|
|
vector<float> *GetCoor(); |
| 28 |
|
|
vector<float> *GetVelo(); |
| 29 |
|
|
vector<float> *GetForce(); |
| 30 |
|
|
map<string, float> *GetEnergy(); |
| 31 |
|
|
|
| 32 |
|
|
void SetNumAtom(int numAtom) { return _numAtom = numAtom;} |
| 33 |
|
|
|
| 34 |
|
|
void Reserve(int numAtom); |
| 35 |
|
|
TFrame(TFrame &frame); |
| 36 |
|
|
~TFrame(); |
| 37 |
|
|
|
| 38 |
|
|
}; |