ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/branches/new-templateless/OOPSE/libmdtools/Integrator.hpp
Revision: 850
Committed: Mon Nov 3 22:07:17 2003 UTC (20 years, 9 months ago) by mmeineke
File size: 12639 byte(s)
Log Message:
begun work on removing templates and most of standard template library from OOPSE.

File Contents

# Content
1 #ifndef _INTEGRATOR_H_
2 #define _INTEGRATOR_H_
3
4 #include "Atom.hpp"
5 #include "Molecule.hpp"
6 #include "SRI.hpp"
7 #include "AbstractClasses.hpp"
8 #include "SimInfo.hpp"
9 #include "ForceFields.hpp"
10 #include "Thermo.hpp"
11 #include "ReadWrite.hpp"
12 // #include "ZConsWriter.hpp"
13
14 using namespace std;
15 const double kB = 8.31451e-7;// boltzmann constant amu*Ang^2*fs^-2/K
16 const double eConvert = 4.184e-4; // converts kcal/mol -> amu*A^2/fs^2
17 const double p_convert = 1.63882576e8; //converts amu*fs^-2*Ang^-1 -> atm
18 const int maxIteration = 300;
19 const double tol = 1.0e-6;
20
21
22 class Integrator : public BaseIntegrator{
23
24 public:
25 Integrator( SimInfo *theInfo, ForceFields* the_ff );
26 virtual ~Integrator();
27 void integrate( void );
28 virtual double getConservedQuantity(void);
29 virtual string getAdditionalParameters(void);
30
31 protected:
32
33 virtual void integrateStep( int calcPot, int calcStress );
34 virtual void preMove( void );
35 virtual void moveA( void );
36 virtual void moveB( void );
37 virtual void constrainA( void );
38 virtual void constrainB( void );
39 virtual int readyCheck( void ) { return 1; }
40
41 virtual void resetIntegrator( void ) { }
42
43 virtual void calcForce( int calcPot, int calcStress );
44 virtual void thermalize();
45
46 virtual void rotationPropagation( DirectionalAtom* dAtom, double ji[3] );
47
48 void checkConstraints( void );
49 void rotate( int axes1, int axes2, double angle, double j[3],
50 double A[3][3] );
51
52 ForceFields* myFF;
53
54 SimInfo *info; // all the info we'll ever need
55 int nAtoms; /* the number of atoms */
56 int oldAtoms;
57 Atom **atoms; /* array of atom pointers */
58 Molecule* molecules;
59 int nMols;
60
61 int isConstrained; // boolean to know whether the systems contains
62 // constraints.
63 int nConstrained; // counter for number of constraints
64 int *constrainedA; // the i of a constraint pair
65 int *constrainedB; // the j of a constraint pair
66 double *constrainedDsqr; // the square of the constraint distance
67
68 int* moving; // tells whether we are moving atom i
69 int* moved; // tells whether we have moved atom i
70 double* oldPos; // pre constrained positions
71
72 short isFirst; /*boolean for the first time integrate is called */
73
74 double dt;
75 double dt2;
76
77 Thermo *tStats;
78 StatWriter* statOut;
79 DumpWriter* dumpOut;
80
81 };
82
83 class NVE : public Integrator {
84
85 public:
86 NVE ( SimInfo *theInfo, ForceFields* the_ff ):
87 Integrator( theInfo, the_ff ){}
88 virtual ~NVE(){}
89 };
90
91
92 class NVT : public Integrator {
93
94 public:
95
96 NVT ( SimInfo *theInfo, ForceFields* the_ff);
97 virtual ~NVT();
98
99 void setTauThermostat(double tt) {tauThermostat = tt; have_tau_thermostat=1;}
100 void setTargetTemp(double tt) {targetTemp = tt; have_target_temp = 1;}
101 void setChiTolerance(double tol) {chiTolerance = tol;}
102 virtual double getConservedQuantity(void);
103 virtual char* getAdditionalParameters(void);
104
105 protected:
106
107 virtual void moveA( void );
108 virtual void moveB( void );
109
110 virtual int readyCheck();
111
112 virtual void resetIntegrator( void );
113
114 // chi is a propagated degree of freedom.
115
116 double chi;
117
118 //integral of chi(t)dt
119 double integralOfChidt;
120
121 // targetTemp must be set. tauThermostat must also be set;
122
123 double targetTemp;
124 double tauThermostat;
125
126 short int have_tau_thermostat, have_target_temp;
127
128 double *oldVel;
129 double *oldJi;
130
131 double chiTolerance;
132 short int have_chi_tolerance;
133
134 char addParamBuffer[1000];
135
136 };
137
138
139
140 class NPT : public Integrator{
141
142 public:
143
144 NPT ( SimInfo *theInfo, ForceFields* the_ff);
145 virtual ~NPT();
146
147 virtual void integrateStep( int calcPot, int calcStress ){
148 calcStress = 1;
149 Integrator::integrateStep( calcPot, calcStress );
150 }
151
152 virtual double getConservedQuantity(void) = 0;
153 virtual string getAdditionalParameters(void) = 0;
154
155 double myTauThermo( void ) { return tauThermostat; }
156 double myTauBaro( void ) { return tauBarostat; }
157
158 void setTauThermostat(double tt) {tauThermostat = tt; have_tau_thermostat=1;}
159 void setTauBarostat(double tb) {tauBarostat = tb; have_tau_barostat=1;}
160 void setTargetTemp(double tt) {targetTemp = tt; have_target_temp = 1;}
161 void setTargetPressure(double tp) {targetPressure = tp; have_target_pressure = 1;}
162 void setChiTolerance(double tol) {chiTolerance = tol; have_chi_tolerance = 1;}
163 void setPosIterTolerance(double tol) {posIterTolerance = tol; have_pos_iter_tolerance = 1;}
164 void setEtaTolerance(double tol) {etaTolerance = tol; have_eta_tolerance = 1;}
165
166 protected:
167
168 virtual void moveA( void );
169 virtual void moveB( void );
170
171 virtual int readyCheck();
172
173 virtual void resetIntegrator( void );
174
175 virtual void getVelScaleA( double sc[3], double vel[3] ) = 0;
176 virtual void getVelScaleB( double sc[3], int index ) = 0;
177 virtual void getPosScale(double pos[3], double COM[3],
178 int index, double sc[3]) = 0;
179
180 virtual bool chiConverged( void );
181 virtual bool etaConverged( void ) = 0;
182
183 virtual void evolveChiA( void );
184 virtual void evolveEtaA( void ) = 0;
185 virtual void evolveChiB( void );
186 virtual void evolveEtaB( void ) = 0;
187
188 virtual void scaleSimBox( void ) = 0;
189
190 void accIntegralOfChidt(void) { integralOfChidt += dt * chi;}
191
192 // chi and eta are the propagated degrees of freedom
193
194 double oldChi;
195 double prevChi;
196 double chi;
197 double NkBT;
198 double fkBT;
199
200 double tt2, tb2;
201 double instaTemp, instaPress, instaVol;
202 double press[3][3];
203
204 int Nparticles;
205
206 double integralOfChidt;
207
208 // targetTemp, targetPressure, and tauBarostat must be set.
209 // One of qmass or tauThermostat must be set;
210
211 double targetTemp;
212 double targetPressure;
213 double tauThermostat;
214 double tauBarostat;
215
216 short int have_tau_thermostat, have_tau_barostat, have_target_temp;
217 short int have_target_pressure;
218
219 double *oldPos;
220 double *oldVel;
221 double *oldJi;
222
223 double chiTolerance;
224 short int have_chi_tolerance;
225 double posIterTolerance;
226 short int have_pos_iter_tolerance;
227 double etaTolerance;
228 short int have_eta_tolerance;
229
230 char addParamBuffer[1000];
231
232 };
233
234 class NPTi : public NPT{
235
236 public:
237 NPTi( SimInfo *theInfo, ForceFields* the_ff);
238 ~NPTi();
239
240 virtual double getConservedQuantity(void);
241 virtual void resetIntegrator(void);
242 virtual string getAdditionalParameters(void);
243 protected:
244
245
246
247 virtual void evolveEtaA(void);
248 virtual void evolveEtaB(void);
249
250 virtual bool etaConverged( void );
251
252 virtual void scaleSimBox( void );
253
254 virtual void getVelScaleA( double sc[3], double vel[3] );
255 virtual void getVelScaleB( double sc[3], int index );
256 virtual void getPosScale(double pos[3], double COM[3],
257 int index, double sc[3]);
258
259 double eta, oldEta, prevEta;
260 };
261
262 class NPTf : public NPT{
263
264 public:
265
266 NPTf ( SimInfo *theInfo, ForceFields* the_ff);
267 virtual ~NPTf();
268
269 virtual double getConservedQuantity(void);
270 virtual string getAdditionalParameters(void);
271 virtual void resetIntegrator(void);
272
273 protected:
274
275 virtual void evolveEtaA(void);
276 virtual void evolveEtaB(void);
277
278 virtual bool etaConverged( void );
279
280 virtual void scaleSimBox( void );
281
282 virtual void getVelScaleA( double sc[3], double vel[3] );
283 virtual void getVelScaleB( double sc[3], int index );
284 virtual void getPosScale(double pos[3], double COM[3],
285 int index, double sc[3]);
286
287 double eta[3][3];
288 double oldEta[3][3];
289 double prevEta[3][3];
290 };
291
292 class NPTxyz : public NPT{
293
294 public:
295
296 NPTxyz ( SimInfo *theInfo, ForceFields* the_ff);
297 virtual ~NPTxyz();
298
299 virtual double getConservedQuantity(void);
300 virtual string getAdditionalParameters(void);
301 virtual void resetIntegrator(void);
302
303 protected:
304
305 virtual void evolveEtaA(void);
306 virtual void evolveEtaB(void);
307
308 virtual bool etaConverged( void );
309
310 virtual void scaleSimBox( void );
311
312 virtual void getVelScaleA( double sc[3], double vel[3] );
313 virtual void getVelScaleB( double sc[3], int index );
314 virtual void getPosScale(double pos[3], double COM[3],
315 int index, double sc[3]);
316
317 double eta[3][3];
318 double oldEta[3][3];
319 double prevEta[3][3];
320 };
321
322
323 // template<typename T> class ZConstraint : public T {
324
325 // public:
326 // class ForceSubtractionPolicy{
327 // public:
328 // ForceSubtractionPolicy(ZConstraint<T>* integrator) {zconsIntegrator = integrator;}
329
330 // virtual void update() = 0;
331 // virtual double getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce) = 0;
332 // virtual double getZFOfMovingMols(Atom* atom, double totalForce) = 0;
333 // virtual double getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce) = 0;
334 // virtual double getHFOfUnconsMols(Atom* atom, double totalForce) = 0;
335
336 // protected:
337 // ZConstraint<T>* zconsIntegrator;
338 // };
339
340 // class PolicyByNumber : public ForceSubtractionPolicy{
341
342 // public:
343 // PolicyByNumber(ZConstraint<T>* integrator) :ForceSubtractionPolicy(integrator) {}
344 // virtual void update();
345 // virtual double getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce) ;
346 // virtual double getZFOfMovingMols(Atom* atom, double totalForce) ;
347 // virtual double getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce);
348 // virtual double getHFOfUnconsMols(Atom* atom, double totalForce);
349
350 // private:
351 // int totNumOfMovingAtoms;
352 // };
353
354 // class PolicyByMass : public ForceSubtractionPolicy{
355
356 // public:
357 // PolicyByMass(ZConstraint<T>* integrator) :ForceSubtractionPolicy(integrator) {}
358
359 // virtual void update();
360 // virtual double getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce) ;
361 // virtual double getZFOfMovingMols(Atom* atom, double totalForce) ;
362 // virtual double getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce);
363 // virtual double getHFOfUnconsMols(Atom* atom, double totalForce);
364
365 // private:
366 // double totMassOfMovingAtoms;
367 // };
368
369 // public:
370
371 // ZConstraint( SimInfo *theInfo, ForceFields* the_ff);
372 // ~ZConstraint();
373
374 // void setZConsTime(double time) {this->zconsTime = time;}
375 // void getZConsTime() {return zconsTime;}
376
377 // void setIndexOfAllZConsMols(vector<int> index) {indexOfAllZConsMols = index;}
378 // void getIndexOfAllZConsMols() {return indexOfAllZConsMols;}
379
380 // void setZConsOutput(const char * fileName) {zconsOutput = fileName;}
381 // string getZConsOutput() {return zconsOutput;}
382
383 // virtual void integrate();
384
385
386 // #ifdef IS_MPI
387 // virtual void update(); //which is called to indicate the molecules' migration
388 // #endif
389
390 // enum ZConsState {zcsMoving, zcsFixed};
391
392 // vector<Molecule*> zconsMols; //z-constraint molecules array
393 // vector<ZConsState> states; //state of z-constraint molecules
394
395
396
397 // int totNumOfUnconsAtoms; //total number of uncontraint atoms
398 // double totalMassOfUncons; //total mas of unconstraint molecules
399
400
401 // protected:
402
403
404
405 // virtual void calcForce( int calcPot, int calcStress );
406 // virtual void thermalize(void);
407
408 // void zeroOutVel();
409 // void doZconstraintForce();
410 // void doHarmonic();
411 // bool checkZConsState();
412
413 // bool haveFixedZMols();
414 // bool haveMovingZMols();
415
416 // double calcZSys();
417
418 // int isZConstraintMol(Molecule* mol);
419
420
421 // double zconsTime; //sample time
422 // double zconsTol; //tolerance of z-contratint
423 // double zForceConst; //base force constant term
424 // //which is estimate by OOPSE
425
426
427 // vector<double> massOfZConsMols; //mass of z-constraint molecule
428 // vector<double> kz; //force constant array
429
430 // vector<double> zPos; //
431
432
433 // vector<Molecule*> unconsMols; //unconstraint molecules array
434 // vector<double> massOfUnconsMols; //mass array of unconstraint molecules
435
436
437 // vector<ZConsParaItem>* parameters; //
438
439 // vector<int> indexOfAllZConsMols; //index of All Z-Constraint Molecuels
440
441 // int* indexOfZConsMols; //index of local Z-Constraint Molecules
442 // double* fz;
443 // double* curZPos;
444
445
446
447 // int whichDirection; //constraint direction
448
449 // private:
450
451 // string zconsOutput; //filename of zconstraint output
452 // ZConsWriter* fzOut; //z-constraint writer
453
454 // double curZconsTime;
455
456 // double calcMovingMolsCOMVel();
457 // double calcSysCOMVel();
458 // double calcTotalForce();
459
460 // ForceSubtractionPolicy* forcePolicy; //force subtraction policy
461 // friend class ForceSubtractionPolicy;
462
463 // };
464
465 #endif