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 |
|
|
}; |