OOPSE/libmdtools/Thermo.cpp

#include <cmath>
#include <iostream>
using namespace std;

#ifdef IS_MPI
#include <mpi.h>
#endif //is_mpi

#include "Thermo.hpp"
#include "SRI.hpp"
#include "Integrator.hpp"
#include "simError.h"

#ifdef IS_MPI
#define __C
#include "mpiSimulation.hpp"
#endif // is_mpi


#define BASE_SEED 123456789

Thermo::Thermo( SimInfo* the_entry_plug ) { 
  entry_plug = the_entry_plug;
  int baseSeed = BASE_SEED;
  
  gaussStream = new gaussianSPRNG( baseSeed );
}

Thermo::~Thermo(){
  delete gaussStream;
}

double Thermo::getKinetic(){

  const double e_convert = 4.184E-4; // convert kcal/mol -> (amu A^2)/fs^2
  double vx2, vy2, vz2;
  double kinetic, v_sqr;
  int kl;
  double jx2, jy2, jz2; // the square of the angular momentums

  DirectionalAtom *dAtom;

  int n_atoms;
  double kinetic_global;
  Atom** atoms;

  
  n_atoms = entry_plug->n_atoms;
  atoms = entry_plug->atoms;

  kinetic = 0.0;
  kinetic_global = 0.0;
  for( kl=0; kl < n_atoms; kl++ ){

    vx2 = atoms[kl]->get_vx() * atoms[kl]->get_vx();
    vy2 = atoms[kl]->get_vy() * atoms[kl]->get_vy();
    vz2 = atoms[kl]->get_vz() * atoms[kl]->get_vz();

    v_sqr = vx2 + vy2 + vz2;
    kinetic += atoms[kl]->getMass() * v_sqr;

    if( atoms[kl]->isDirectional() ){
            
      dAtom = (DirectionalAtom *)atoms[kl];
      
      jx2 = dAtom->getJx() * dAtom->getJx();    
      jy2 = dAtom->getJy() * dAtom->getJy();
      jz2 = dAtom->getJz() * dAtom->getJz();
      
      kinetic += (jx2 / dAtom->getIxx()) + (jy2 / dAtom->getIyy()) 
        + (jz2 / dAtom->getIzz());
    }
  }
#ifdef IS_MPI
  MPI_Allreduce(&kinetic,&kinetic_global,1,MPI_DOUBLE,
                MPI_SUM, MPI_COMM_WORLD);
  kinetic = kinetic_global;
#endif //is_mpi

  kinetic = kinetic * 0.5 / e_convert;

  return kinetic;
}

double Thermo::getPotential(){
  
  double potential_local;
  double potential;
  int el, nSRI;
  Molecule* molecules;

  molecules = entry_plug->molecules;
  nSRI = entry_plug->n_SRI;

  potential_local = 0.0;
  potential = 0.0;
  potential_local += entry_plug->lrPot;

  for( el=0; el<entry_plug->n_mol; el++ ){    
    potential_local += molecules[el].getPotential();
  }

  // Get total potential for entire system from MPI.
#ifdef IS_MPI
  MPI_Allreduce(&potential_local,&potential,1,MPI_DOUBLE,
                MPI_SUM, MPI_COMM_WORLD);
#else
  potential = potential_local; 
#endif // is_mpi

#ifdef IS_MPI
  /*
  std::cerr << "node " << worldRank << ": after pot = " << potential << "\n";
  */
#endif

  return potential;
}

double Thermo::getTotalE(){

  double total;

  total = this->getKinetic() + this->getPotential();
  return total;
}

double Thermo::getTemperature(){

  const double kb = 1.9872179E-3; // boltzman's constant in kcal/(mol K)
  double temperature;
  
  temperature = ( 2.0 * this->getKinetic() ) / ((double)entry_plug->ndf * kb );
  return temperature;
}

double Thermo::getEnthalpy() {

  const double e_convert = 4.184E-4; // convert kcal/mol -> (amu A^2)/fs^2
  double u, p, v;
  double press[9];

  u = this->getTotalE();

  this->getPressureTensor(press);
  p = (press[0] + press[4] + press[8]) / 3.0;

  v = this->getVolume();

  return (u + (p*v)/e_convert);
}

double Thermo::getVolume() {
  double theBox[3];

  entry_plug->getBox(theBox);
  return (theBox[0] * theBox[1] * theBox[2]);
}

double Thermo::getPressure() {
  // returns the pressure in units of atm
  // Relies on the calculation of the full molecular pressure tensor
  
  const double p_convert = 1.63882576e8;
  double press[9];
  double pressure;

  this->getPressureTensor(press);

  pressure = p_convert * (press[0] + press[4] + press[8]) / 3.0;

  return pressure;
}


void Thermo::getPressureTensor(double press[9]){
  // returns pressure tensor in units amu*fs^-2*Ang^-1
  // routine derived via viral theorem description in:
  // Paci, E. and Marchi, M. J.Phys.Chem. 1996, 100, 4314-4322

  const double e_convert = 4.184e-4;

  double molmass, volume;
  double vcom[3];
  double p_local[9], p_global[9];
  double theBox[3];
  double* tau;
  int i, nMols;
  Molecule* molecules;

  nMols = entry_plug->n_mol;
  molecules = entry_plug->molecules;
  tau = entry_plug->tau;

  // use velocities of molecular centers of mass and molecular masses:
  for (i=0; i < 9; i++) {    
    p_local[i] = 0.0;
    p_global[i] = 0.0;
  }

  for (i=0; i < nMols; i++) {
    molmass = molecules[i].getCOMvel(vcom);

    p_local[0] += molmass * (vcom[0] * vcom[0]); 
    p_local[1] += molmass * (vcom[0] * vcom[1]); 
    p_local[2] += molmass * (vcom[0] * vcom[2]); 
    p_local[3] += molmass * (vcom[1] * vcom[0]); 
    p_local[4] += molmass * (vcom[1] * vcom[1]); 
    p_local[5] += molmass * (vcom[1] * vcom[2]); 
    p_local[6] += molmass * (vcom[2] * vcom[0]); 
    p_local[7] += molmass * (vcom[2] * vcom[1]); 
    p_local[8] += molmass * (vcom[2] * vcom[2]); 
  }

  // Get total for entire system from MPI.

#ifdef IS_MPI
  MPI_Allreduce(p_local,p_global,9,MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
#else
  for (i=0; i<9; i++) {
    p_global[i] = p_local[i]; 
  }
#endif // is_mpi

  entry_plug->getBox(theBox);

  volume = theBox[0] * theBox[1] * theBox[2];

  for(i=0; i<9; i++) {
    press[i] = (p_global[i] - tau[i]*e_convert) / volume;
  }
}

void Thermo::velocitize() {
  
  double x,y;
  double vx, vy, vz;
  double jx, jy, jz;
  int i, vr, vd; // velocity randomizer loop counters
  double vdrift[3];
  double vbar;
  const double kb = 8.31451e-7; // kb in amu, angstroms, fs, etc.
  double av2;
  double kebar;
  int n_atoms;
  Atom** atoms;
  DirectionalAtom* dAtom;
  double temperature;
  int n_oriented;
  int n_constraints;

  atoms         = entry_plug->atoms;
  n_atoms       = entry_plug->n_atoms;
  temperature   = entry_plug->target_temp;
  n_oriented    = entry_plug->n_oriented;
  n_constraints = entry_plug->n_constraints;
  
  kebar = kb * temperature * (double)entry_plug->ndf / 
    ( 2.0 * (double)entry_plug->ndfRaw );
  
  for(vr = 0; vr < n_atoms; vr++){
    
    // uses equipartition theory to solve for vbar in angstrom/fs

    av2 = 2.0 * kebar / atoms[vr]->getMass();
    vbar = sqrt( av2 );
 
//     vbar = sqrt( 8.31451e-7 * temperature / atoms[vr]->getMass() );
    
    // picks random velocities from a gaussian distribution
    // centered on vbar

    vx = vbar * gaussStream->getGaussian();
    vy = vbar * gaussStream->getGaussian();
    vz = vbar * gaussStream->getGaussian();

    atoms[vr]->set_vx( vx ); 
    atoms[vr]->set_vy( vy );
    atoms[vr]->set_vz( vz );
  }

  // Get the Center of Mass drift velocity.

  getCOMVel(vdrift);
  
  //  Corrects for the center of mass drift.
  // sums all the momentum and divides by total mass.

  for(vd = 0; vd < n_atoms; vd++){
    
    vx = atoms[vd]->get_vx();
    vy = atoms[vd]->get_vy();
    vz = atoms[vd]->get_vz();
        
    vx -= vdrift[0];
    vy -= vdrift[1];
    vz -= vdrift[2];
    
    atoms[vd]->set_vx(vx);
    atoms[vd]->set_vy(vy);
    atoms[vd]->set_vz(vz);
  }
  if( n_oriented ){
  
    for( i=0; i<n_atoms; i++ ){
      
      if( atoms[i]->isDirectional() ){
        
        dAtom = (DirectionalAtom *)atoms[i];

        vbar = sqrt( 2.0 * kebar * dAtom->getIxx() );
        jx = vbar * gaussStream->getGaussian();

        vbar = sqrt( 2.0 * kebar * dAtom->getIyy() );
        jy = vbar * gaussStream->getGaussian();
        
        vbar = sqrt( 2.0 * kebar * dAtom->getIzz() );
        jz = vbar * gaussStream->getGaussian();
        
        dAtom->setJx( jx );
        dAtom->setJy( jy );
        dAtom->setJz( jz );
      }
    }   
  }
}

void Thermo::getCOMVel(double vdrift[3]){

  double mtot, mtot_local;
  double vdrift_local[3];
  int vd, n_atoms;
  Atom** atoms;

  // We are very careless here with the distinction between n_atoms and n_local
  // We should really fix this before someone pokes an eye out.

  n_atoms = entry_plug->n_atoms;  
  atoms   = entry_plug->atoms;

  mtot_local = 0.0;
  vdrift_local[0] = 0.0;
  vdrift_local[1] = 0.0;
  vdrift_local[2] = 0.0;
  
  for(vd = 0; vd < n_atoms; vd++){
    
    vdrift_local[0] += atoms[vd]->get_vx() * atoms[vd]->getMass();
    vdrift_local[1] += atoms[vd]->get_vy() * atoms[vd]->getMass();
    vdrift_local[2] += atoms[vd]->get_vz() * atoms[vd]->getMass();
    
    mtot_local += atoms[vd]->getMass();
  }

#ifdef IS_MPI
  MPI_Allreduce(&mtot_local,&mtot,1,MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
  MPI_Allreduce(vdrift_local,vdrift,3,MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
#else
  mtot = mtot_local;
  for(vd = 0; vd < 3; vd++) {
    vdrift[vd] = vdrift_local[vd];
  }
#endif
    
  for (vd = 0; vd < 3; vd++) {
    vdrift[vd] = vdrift[vd] / mtot;
  }
  
}

Revision:	484
Committed:	Wed Apr 9 13:59:35 2003 UTC (21 years, 3 months ago) by gezelter
File size:	8487 byte(s)
Log Message:	Added volume and enthalpy to status file
#	User	Rev	Content
1	mmeineke	377	#include <cmath>
2			#include <iostream>
3			using namespace std;
4
5			#ifdef IS_MPI
6			#include <mpi.h>
7			#endif //is_mpi
8
9			#include "Thermo.hpp"
10			#include "SRI.hpp"
11			#include "Integrator.hpp"
12	chuckv	438	#include "simError.h"
13	mmeineke	402
14			#ifdef IS_MPI
15	chuckv	401	#define __C
16	mmeineke	402	#include "mpiSimulation.hpp"
17			#endif // is_mpi
18	mmeineke	377
19	mmeineke	402
20	mmeineke	377	#define BASE_SEED 123456789
21
22			Thermo::Thermo( SimInfo* the_entry_plug ) {
23			entry_plug = the_entry_plug;
24			int baseSeed = BASE_SEED;
25
26			gaussStream = new gaussianSPRNG( baseSeed );
27			}
28
29			Thermo::~Thermo(){
30			delete gaussStream;
31			}
32
33			double Thermo::getKinetic(){
34
35			const double e_convert = 4.184E-4; // convert kcal/mol -> (amu A^2)/fs^2
36			double vx2, vy2, vz2;
37			double kinetic, v_sqr;
38			int kl;
39			double jx2, jy2, jz2; // the square of the angular momentums
40
41			DirectionalAtom *dAtom;
42
43			int n_atoms;
44			double kinetic_global;
45			Atom** atoms;
46
47
48			n_atoms = entry_plug->n_atoms;
49			atoms = entry_plug->atoms;
50
51			kinetic = 0.0;
52			kinetic_global = 0.0;
53			for( kl=0; kl < n_atoms; kl++ ){
54
55			vx2 = atoms[kl]->get_vx() * atoms[kl]->get_vx();
56			vy2 = atoms[kl]->get_vy() * atoms[kl]->get_vy();
57			vz2 = atoms[kl]->get_vz() * atoms[kl]->get_vz();
58
59			v_sqr = vx2 + vy2 + vz2;
60			kinetic += atoms[kl]->getMass() * v_sqr;
61
62			if( atoms[kl]->isDirectional() ){
63
64			dAtom = (DirectionalAtom *)atoms[kl];
65
66			jx2 = dAtom->getJx() * dAtom->getJx();
67			jy2 = dAtom->getJy() * dAtom->getJy();
68			jz2 = dAtom->getJz() * dAtom->getJz();
69
70			kinetic += (jx2 / dAtom->getIxx()) + (jy2 / dAtom->getIyy())
71			+ (jz2 / dAtom->getIzz());
72			}
73			}
74			#ifdef IS_MPI
75	mmeineke	447	MPI_Allreduce(&kinetic,&kinetic_global,1,MPI_DOUBLE,
76			MPI_SUM, MPI_COMM_WORLD);
77	mmeineke	377	kinetic = kinetic_global;
78			#endif //is_mpi
79
80			kinetic = kinetic * 0.5 / e_convert;
81
82			return kinetic;
83			}
84
85			double Thermo::getPotential(){
86
87	chuckv	401	double potential_local;
88	mmeineke	377	double potential;
89			int el, nSRI;
90	mmeineke	428	Molecule* molecules;
91	mmeineke	377
92	mmeineke	428	molecules = entry_plug->molecules;
93	mmeineke	377	nSRI = entry_plug->n_SRI;
94
95	chuckv	401	potential_local = 0.0;
96	chuckv	438	potential = 0.0;
97	chuckv	401	potential_local += entry_plug->lrPot;
98	mmeineke	377
99	mmeineke	423	for( el=0; el<entry_plug->n_mol; el++ ){
100	mmeineke	428	potential_local += molecules[el].getPotential();
101	mmeineke	377	}
102
103			// Get total potential for entire system from MPI.
104			#ifdef IS_MPI
105	mmeineke	447	MPI_Allreduce(&potential_local,&potential,1,MPI_DOUBLE,
106			MPI_SUM, MPI_COMM_WORLD);
107	chuckv	401	#else
108			potential = potential_local;
109	mmeineke	377	#endif // is_mpi
110
111	chuckv	438	#ifdef IS_MPI
112			/*
113			std::cerr << "node " << worldRank << ": after pot = " << potential << "\n";
114			*/
115			#endif
116
117	mmeineke	377	return potential;
118			}
119
120			double Thermo::getTotalE(){
121
122			double total;
123
124			total = this->getKinetic() + this->getPotential();
125			return total;
126			}
127
128	gezelter	454	double Thermo::getTemperature(){
129
130			const double kb = 1.9872179E-3; // boltzman's constant in kcal/(mol K)
131			double temperature;
132
133	gezelter	458	temperature = ( 2.0 * this->getKinetic() ) / ((double)entry_plug->ndf * kb );
134	mmeineke	377	return temperature;
135			}
136
137	gezelter	484	double Thermo::getEnthalpy() {
138
139			const double e_convert = 4.184E-4; // convert kcal/mol -> (amu A^2)/fs^2
140			double u, p, v;
141			double press[9];
142
143			u = this->getTotalE();
144
145			this->getPressureTensor(press);
146			p = (press[0] + press[4] + press[8]) / 3.0;
147
148			v = this->getVolume();
149
150			return (u + (p*v)/e_convert);
151			}
152
153			double Thermo::getVolume() {
154			double theBox[3];
155
156			entry_plug->getBox(theBox);
157			return (theBox[0] * theBox[1] * theBox[2]);
158			}
159
160	gezelter	483	double Thermo::getPressure() {
161			// returns the pressure in units of atm
162			// Relies on the calculation of the full molecular pressure tensor
163
164			const double p_convert = 1.63882576e8;
165			double press[9];
166			double pressure;
167
168			this->getPressureTensor(press);
169
170			pressure = p_convert * (press[0] + press[4] + press[8]) / 3.0;
171
172			return pressure;
173			}
174
175
176			void Thermo::getPressureTensor(double press[9]){
177			// returns pressure tensor in units amufs^-2Ang^-1
178	gezelter	445	// routine derived via viral theorem description in:
179			// Paci, E. and Marchi, M. J.Phys.Chem. 1996, 100, 4314-4322
180	mmeineke	377
181	gezelter	477	const double e_convert = 4.184e-4;
182	gezelter	483
183			double molmass, volume;
184	gezelter	468	double vcom[3];
185	gezelter	483	double p_local[9], p_global[9];
186	gezelter	468	double theBox[3];
187			double* tau;
188			int i, nMols;
189			Molecule* molecules;
190
191			nMols = entry_plug->n_mol;
192			molecules = entry_plug->molecules;
193			tau = entry_plug->tau;
194
195			// use velocities of molecular centers of mass and molecular masses:
196	gezelter	483	for (i=0; i < 9; i++) {
197			p_local[i] = 0.0;
198			p_global[i] = 0.0;
199			}
200	gezelter	475
201	gezelter	468	for (i=0; i < nMols; i++) {
202	gezelter	475	molmass = molecules[i].getCOMvel(vcom);
203	gezelter	483
204			p_local[0] += molmass * (vcom[0] * vcom[0]);
205			p_local[1] += molmass * (vcom[0] * vcom[1]);
206			p_local[2] += molmass * (vcom[0] * vcom[2]);
207			p_local[3] += molmass * (vcom[1] * vcom[0]);
208			p_local[4] += molmass * (vcom[1] * vcom[1]);
209			p_local[5] += molmass * (vcom[1] * vcom[2]);
210			p_local[6] += molmass * (vcom[2] * vcom[0]);
211			p_local[7] += molmass * (vcom[2] * vcom[1]);
212			p_local[8] += molmass * (vcom[2] * vcom[2]);
213	gezelter	468	}
214
215			// Get total for entire system from MPI.
216	chuckv	479
217	gezelter	468	#ifdef IS_MPI
218	gezelter	483	MPI_Allreduce(p_local,p_global,9,MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
219	gezelter	468	#else
220	gezelter	483	for (i=0; i<9; i++) {
221			p_global[i] = p_local[i];
222			}
223	gezelter	468	#endif // is_mpi
224
225			entry_plug->getBox(theBox);
226
227	gezelter	483	volume = theBox[0] * theBox[1] * theBox[2];
228	gezelter	477
229	gezelter	483	for(i=0; i<9; i++) {
230			press[i] = (p_global[i] - tau[i]*e_convert) / volume;
231			}
232	mmeineke	377	}
233
234			void Thermo::velocitize() {
235
236			double x,y;
237			double vx, vy, vz;
238			double jx, jy, jz;
239			int i, vr, vd; // velocity randomizer loop counters
240	chuckv	403	double vdrift[3];
241	mmeineke	377	double vbar;
242			const double kb = 8.31451e-7; // kb in amu, angstroms, fs, etc.
243			double av2;
244			double kebar;
245			int n_atoms;
246			Atom** atoms;
247			DirectionalAtom* dAtom;
248			double temperature;
249			int n_oriented;
250			int n_constraints;
251
252			atoms = entry_plug->atoms;
253			n_atoms = entry_plug->n_atoms;
254			temperature = entry_plug->target_temp;
255			n_oriented = entry_plug->n_oriented;
256			n_constraints = entry_plug->n_constraints;
257
258	gezelter	458	kebar = kb * temperature * (double)entry_plug->ndf /
259			( 2.0 * (double)entry_plug->ndfRaw );
260	chuckv	403
261	mmeineke	377	for(vr = 0; vr < n_atoms; vr++){
262
263			// uses equipartition theory to solve for vbar in angstrom/fs
264
265			av2 = 2.0 * kebar / atoms[vr]->getMass();
266			vbar = sqrt( av2 );
267	gezelter	444
268	mmeineke	377	// vbar = sqrt( 8.31451e-7 * temperature / atoms[vr]->getMass() );
269
270			// picks random velocities from a gaussian distribution
271			// centered on vbar
272
273			vx = vbar * gaussStream->getGaussian();
274			vy = vbar * gaussStream->getGaussian();
275			vz = vbar * gaussStream->getGaussian();
276
277			atoms[vr]->set_vx( vx );
278			atoms[vr]->set_vy( vy );
279			atoms[vr]->set_vz( vz );
280			}
281	chuckv	401
282			// Get the Center of Mass drift velocity.
283
284	chuckv	403	getCOMVel(vdrift);
285	mmeineke	377
286			// Corrects for the center of mass drift.
287			// sums all the momentum and divides by total mass.
288
289			for(vd = 0; vd < n_atoms; vd++){
290
291			vx = atoms[vd]->get_vx();
292			vy = atoms[vd]->get_vy();
293			vz = atoms[vd]->get_vz();
294	chuckv	401
295	mmeineke	377	vx -= vdrift[0];
296			vy -= vdrift[1];
297			vz -= vdrift[2];
298
299			atoms[vd]->set_vx(vx);
300			atoms[vd]->set_vy(vy);
301			atoms[vd]->set_vz(vz);
302			}
303			if( n_oriented ){
304
305			for( i=0; i<n_atoms; i++ ){
306
307			if( atoms[i]->isDirectional() ){
308
309			dAtom = (DirectionalAtom *)atoms[i];
310
311			vbar = sqrt( 2.0 * kebar * dAtom->getIxx() );
312			jx = vbar * gaussStream->getGaussian();
313
314			vbar = sqrt( 2.0 * kebar * dAtom->getIyy() );
315			jy = vbar * gaussStream->getGaussian();
316	gezelter	454
317	mmeineke	377	vbar = sqrt( 2.0 * kebar * dAtom->getIzz() );
318			jz = vbar * gaussStream->getGaussian();
319
320			dAtom->setJx( jx );
321			dAtom->setJy( jy );
322			dAtom->setJz( jz );
323			}
324			}
325			}
326			}
327	chuckv	401
328	chuckv	403	void Thermo::getCOMVel(double vdrift[3]){
329	chuckv	401
330			double mtot, mtot_local;
331			double vdrift_local[3];
332			int vd, n_atoms;
333			Atom** atoms;
334
335			// We are very careless here with the distinction between n_atoms and n_local
336			// We should really fix this before someone pokes an eye out.
337
338			n_atoms = entry_plug->n_atoms;
339			atoms = entry_plug->atoms;
340
341			mtot_local = 0.0;
342			vdrift_local[0] = 0.0;
343			vdrift_local[1] = 0.0;
344			vdrift_local[2] = 0.0;
345
346			for(vd = 0; vd < n_atoms; vd++){
347
348			vdrift_local[0] += atoms[vd]->get_vx() * atoms[vd]->getMass();
349			vdrift_local[1] += atoms[vd]->get_vy() * atoms[vd]->getMass();
350			vdrift_local[2] += atoms[vd]->get_vz() * atoms[vd]->getMass();
351
352			mtot_local += atoms[vd]->getMass();
353			}
354
355			#ifdef IS_MPI
356	mmeineke	447	MPI_Allreduce(&mtot_local,&mtot,1,MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
357			MPI_Allreduce(vdrift_local,vdrift,3,MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
358	chuckv	401	#else
359			mtot = mtot_local;
360			for(vd = 0; vd < 3; vd++) {
361			vdrift[vd] = vdrift_local[vd];
362			}
363			#endif
364
365			for (vd = 0; vd < 3; vd++) {
366			vdrift[vd] = vdrift[vd] / mtot;
367			}
368
369			}
370