ViewVC Help

View File | Revision Log | Show Annotations | View Changeset | Root Listing

root/group/trunk/OOPSE/libmdtools/ZConstraint.cpp

Comparing trunk/OOPSE/libmdtools/ZConstraint.cpp (file contents):
Revision 702 by tim, Wed Aug 20 14:50:32 2003 UTC vs.
Revision 1091 by tim, Tue Mar 16 19:22:56 2004 UTC

#	Line 1 | Line 1
1	<	#include "Integrator.hpp"
2	<	#include "simError.h"
3	<	#include <cmath>
4	<	template<typename T> ZConstraint<T>::ZConstraint(SimInfo* theInfo, ForceFields* the_ff)
5	<	: T(theInfo, the_ff), fz(NULL), curZPos(NULL),
6	<	indexOfZConsMols(NULL), forcePolicy(NULL), curZconsTime(0)
7	<	{
8	<
9	<	//get properties from SimInfo
10	<	GenericData* data;
11	<	ZConsParaData* zConsParaData;
12	<	DoubleData* sampleTime;
13	<	DoubleData* tolerance;
14	<	StringData* policy;
15	<	StringData* filename;
16	<	double COM[3];
17	<
18	<	//by default, the direction of constraint is z
19	<	// 0 --> x
20	<	// 1 --> y
21	<	// 2 --> z
22	<	whichDirection = 2;
23	<
24	<	//estimate the force constant of harmonical potential
25	<	double Kb = 1.986E-3 ; //in kcal/K
26	<
27	<	double halfOfLargestBox = max(info->boxL[0], max(info->boxL[1], info->boxL[2])) /2;
28	<	zForceConst = Kb * info->target_temp /(halfOfLargestBox * halfOfLargestBox);
29	<
30	<	//creat force substraction policy
31	<	data = info->getProperty(ZCONSFORCEPOLICY_ID);
32	<	if(!data){
33	<	sprintf( painCave.errMsg,
34	<	"ZConstraint Warning: User does not set force substraction policy, "
35	<	"average force substraction policy is used\n");
36	<	painCave.isFatal = 0;
37	<	simError();
38	<
39	<	forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
40	<	}
41	<	else{
42	<	policy = dynamic_cast<StringData*>(data);
43	<
44	<	if(!policy){
45	<	sprintf( painCave.errMsg,
46	<	"ZConstraint Error: Convertion from GenericData to StringData failure, "
47	<	"average force substraction policy is used\n");
48	<	painCave.isFatal = 0;
49	<	simError();
50	<
51	<	forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
52	<	}
53	<	else{
54	<	if(policy->getData() == "BYNUMBER")
55	<	forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this);
56	<	else if(policy->getData() == "BYMASS")
57	<	forcePolicy = (ForceSubstractionPolicy*) new PolicyByMass(this);
58	<	else{
59	<	sprintf( painCave.errMsg,
60	<	"ZConstraint Warning: unknown force substraction policy, "
61	<	"average force substraction policy is used\n");
62	<	painCave.isFatal = 0;
63	<	simError();
64	<	}
65	<	}
66	<	}
67	<
68	<
69	<	//retrieve sample time of z-contraint
70	<	data = info->getProperty(ZCONSTIME_ID);
71	<
72	<	if(!data) {
73	<
74	<	sprintf( painCave.errMsg,
75	<	"ZConstraint error: If you use an ZConstraint\n"
76	<	" , you must set sample time.\n");
77	<	painCave.isFatal = 1;
78	<	simError();
79	<	}
80	<	else{
81	<
82	<	sampleTime = dynamic_cast<DoubleData*>(data);
83	<
84	<	if(!sampleTime){
85	<
86	<	sprintf( painCave.errMsg,
87	<	"ZConstraint error: Can not get property from SimInfo\n");
88	<	painCave.isFatal = 1;
89	<	simError();
90	<
91	<	}
92	<	else{
93	<	this->zconsTime = sampleTime->getData();
94	<	}
95	<
96	<	}
97	<
98	<	//retrieve output filename of z force
99	<	data = info->getProperty(ZCONSFILENAME_ID);
100	<	if(!data) {
101	<
102	<
103	<	sprintf( painCave.errMsg,
104	<	"ZConstraint error: If you use an ZConstraint\n"
105	<	" , you must set output filename of z-force.\n");
106	<	painCave.isFatal = 1;
107	<	simError();
108	<
109	<	}
110	<	else{
111	<
112	<	filename = dynamic_cast<StringData*>(data);
113	<
114	<	if(!filename){
115	<
116	<	sprintf( painCave.errMsg,
117	<	"ZConstraint error: Can not get property from SimInfo\n");
118	<	painCave.isFatal = 1;
119	<	simError();
120	<
121	<	}
122	<	else{
123	<	this->zconsOutput = filename->getData();
124	<	}
125	<
126	<	}
127	<
128	<	//retrieve tolerance for z-constraint molecuels
129	<	data = info->getProperty(ZCONSTOL_ID);
130	<
131	<	if(!data) {
132	<
133	<	sprintf( painCave.errMsg,
134	<	"ZConstraint error: can not get tolerance \n");
135	<	painCave.isFatal = 1;
136	<	simError();
137	<	}
138	<	else{
139	<
140	<	tolerance = dynamic_cast<DoubleData*>(data);
141	<
142	<	if(!tolerance){
143	<
144	<	sprintf( painCave.errMsg,
145	<	"ZConstraint error: Can not get property from SimInfo\n");
146	<	painCave.isFatal = 1;
147	<	simError();
148	<
149	<	}
150	<	else{
151	<	this->zconsTol = tolerance->getData();
152	<	}
153	<
154	<	}
155	<
156	<	//retrieve index of z-constraint molecules
157	<	data = info->getProperty(ZCONSPARADATA_ID);
158	<	if(!data) {
159	<
160	<	sprintf( painCave.errMsg,
161	<	"ZConstraint error: If you use an ZConstraint\n"
162	<	" , you must set index of z-constraint molecules.\n");
163	<	painCave.isFatal = 1;
164	<	simError();
165	<	}
166	<	else{
167	<
168	<	zConsParaData = dynamic_cast<ZConsParaData*>(data);
169	<
170	<	if(!zConsParaData){
171	<
172	<	sprintf( painCave.errMsg,
173	<	"ZConstraint error: Can not get parameters of zconstraint method from SimInfo\n");
174	<	painCave.isFatal = 1;
175	<	simError();
176	<
177	<	}
178	<	else{
179	<
180	<	parameters = zConsParaData->getData();
181	<
182	<	//check the range of zconsIndex
183	<	//and the minimum value of index is the first one (we already sorted the data)
184	<	//the maximum value of index is the last one
185	<
186	<	int maxIndex;
187	<	int minIndex;
188	<	int totalNumMol;
189	<
190	<	minIndex = (*parameters)[0].zconsIndex;
191	<	if(minIndex < 0){
192	<	sprintf( painCave.errMsg,
193	<	"ZConstraint error: index is out of range\n");
194	<	painCave.isFatal = 1;
195	<	simError();
196	<	}
197	<
198	<	maxIndex = (*parameters)[parameters->size() - 1].zconsIndex;
199	<
200	<	#ifndef IS_MPI
201	<	totalNumMol = nMols;
202	<	#else
203	<	totalNumMol = mpiSim->getTotNmol();
204	<	#endif
205	<
206	<	if(maxIndex > totalNumMol - 1){
207	<	sprintf( painCave.errMsg,
208	<	"ZConstraint error: index is out of range\n");
209	<	painCave.isFatal = 1;
210	<	simError();
211	<	}
212	<
213	<	//if user does not specify the zpos for the zconstraint molecule
214	<	//its initial z coordinate  will be used as default
215	<	for(int i = 0; i < parameters->size(); i++){
216	<
217	<	if(!(*parameters)[i].havingZPos){
218	<	#ifndef IS_MPI
219	<	for(int j = 0; j < nMols; j++){
220	<	if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
221	<	molecules[j].getCOM(COM);
222	<	break;
223	<	}
224	<	}
225	<	#else
226	<	//query which processor current zconstraint molecule belongs to
227	<	int *MolToProcMap;
228	<	int whichNode;
229	<	double initZPos;
230	<	MolToProcMap = mpiSim->getMolToProcMap();
231	<	whichNode = MolToProcMap[(*parameters)[i].zconsIndex];
232	<
233	<	//broadcast the zpos of current z-contraint molecule
234	<	//the node which contain this
235	<
236	<	if (worldRank == whichNode ){
237	<
238	<	for(int j = 0; j < nMols; j++)
239	<	if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
240	<	molecules[j].getCOM(COM);
241	<	break;
242	<	}
243	<
244	<	}
245	<
246	<	MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE_PRECISION, whichNode, MPI_COMM_WORLD);
247	<	#endif
248	<
249	<	(*parameters)[i].zPos = COM[whichDirection];
250	<
251	<	sprintf( painCave.errMsg,
252	<	"ZConstraint warningr: Does not specify zpos for z-constraint molecule "
253	<	"initial z coornidate will be used \n");
254	<	painCave.isFatal = 0;
255	<	simError();
256	<
257	<	}
258	<	}
259	<
260	<	}//end if (!zConsParaData)
261	<	}//end  if (!data)
262	<
263	<	//
264	<	#ifdef IS_MPI
265	<	update();
266	<	#else
267	<	int searchResult;
268	<
269	<	for(int i = 0; i < nMols; i++){
270	<
271	<	searchResult = isZConstraintMol(&molecules[i]);
272	<
273	<	if(searchResult > -1){
274	<
275	<	zconsMols.push_back(&molecules[i]);
276	<	massOfZConsMols.push_back(molecules[i].getTotalMass());
277	<
278	<	zPos.push_back((*parameters)[searchResult].zPos);
279	<	cout << "index: "<< (*parameters)[searchResult].zconsIndex
280	<	<<"\tzPos = " << (*parameters)[searchResult].zPos << endl;
281	<	kz.push_back((*parameters)[searchResult]. kRatio * zForceConst);
282	<
283	<	molecules[i].getCOM(COM);
284	<	}
285	<	else
286	<	{
287	<
288	<	unconsMols.push_back(&molecules[i]);
289	<	massOfUnconsMols.push_back(molecules[i].getTotalMass());
290	<
291	<	}
292	<	}
293	<
294	<	fz = new double[zconsMols.size()];
295	<	curZPos = new double[zconsMols.size()];
296	<	indexOfZConsMols = new int [zconsMols.size()];
297	<
298	<	if(!fz || !curZPos || !indexOfZConsMols){
299	<	sprintf( painCave.errMsg,
300	<	"Memory allocation failure in class Zconstraint\n");
301	<	painCave.isFatal = 1;
302	<	simError();
303	<	}
304	<
305	<	//determine the states of z-constraint molecules
306	<	for(int i = 0; i < zconsMols.size(); i++){
307	<	indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex();
308	<
309	<	zconsMols[i]->getCOM(COM);
310	<	if (fabs(zPos[i] - COM[whichDirection]) < zconsTol)
311	<	states.push_back(zcsFixed);
312	<	else
313	<	states.push_back(zcsMoving);
314	<	}
315	<
316	<	#endif
317	<
318	<	//get total masss of unconstraint molecules
319	<	double totalMassOfUncons_local;
320	<	totalMassOfUncons_local = 0;
321	<
322	<	for(int i = 0; i < unconsMols.size(); i++)
323	<	totalMassOfUncons_local += unconsMols[i]->getTotalMass();
324	<
325	<	#ifndef IS_MPI
326	<	totalMassOfUncons = totalMassOfUncons_local;
327	<	#else
328	<	MPI_Allreduce(&totalMassOfUncons_local, &totalMassOfUncons, 1,
329	<	MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
330	<	#endif
331	<
332	<
333	<	//get total number of unconstrained atoms
334	<	int nUnconsAtoms_local;
335	<	nUnconsAtoms_local = 0;
336	<	for(int i = 0; i < unconsMols.size(); i++)
337	<	nUnconsAtoms_local += unconsMols[i]->getNAtoms();
338	<
339	<	#ifndef IS_MPI
340	<	totNumOfUnconsAtoms = nUnconsAtoms_local;
341	<	#else
342	<	MPI_Allreduce(&nUnconsAtoms_local, &totNumOfUnconsAtoms, 1,
343	<	MPI_INT,MPI_SUM, MPI_COMM_WORLD);
344	<	#endif
345	<
346	<	// creat zconsWriter
347	<	fzOut = new ZConsWriter(zconsOutput.c_str(), parameters);
348	<
349	<	if(!fzOut){
350	<	sprintf( painCave.errMsg,
351	<	"Memory allocation failure in class Zconstraint\n");
352	<	painCave.isFatal = 1;
353	<	simError();
354	<	}
355	<
356	<	forcePolicy->update();
357	<	}
358	<
359	<	template<typename T> ZConstraint<T>::~ZConstraint()
360	<	{
361	<	if(fz)
362	<	delete[] fz;
363	<
364	<	if(curZPos)
365	<	delete[] curZPos;
366	<
367	<	if(indexOfZConsMols)
368	<	delete[] indexOfZConsMols;
369	<
370	<	if(fzOut)
371	<	delete fzOut;
372	<
373	<	if(forcePolicy)
374	<	delete forcePolicy;
375	<	}
376	<
377	<
378	<	/**
379	<	*
380	<	*/
381	<
382	<	#ifdef IS_MPI
383	<	template<typename T> void ZConstraint<T>::update()
384	<	{
385	<	double COM[3];
386	<	int index;
387	<
388	<	zconsMols.clear();
389	<	massOfZConsMols.clear();
390	<	zPos.clear();
391	<	kz.clear();
392	<
393	<	unconsMols.clear();
394	<	massOfUnconsMols.clear();
395	<
396	<
397	<	//creat zconsMol and unconsMol lists
398	<	for(int i = 0; i < nMols; i++){
399	<
400	<	index = isZConstraintMol(&molecules[i]);
401	<
402	<	if(index > -1){
403	<
404	<	zconsMols.push_back(&molecules[i]);
405	<	zPos.push_back((*parameters)[index].zPos);
406	<	kz.push_back((*parameters)[index].kRatio * zForceConst);
407	<
408	<	massOfZConsMols.push_back(molecules[i].getTotalMass());
409	<
410	<	molecules[i].getCOM(COM);
411	<	}
412	<	else
413	<	{
414	<
415	<	unconsMols.push_back(&molecules[i]);
416	<	massOfUnconsMols.push_back(molecules[i].getTotalMass());
417	<
418	<	}
419	<	}
420	<
421	<	//determine the states of z-constraint molecules
422	<	for(int i = 0; i < zconsMols.size(); i++){
423	<	zconsMols[i]->getCOM(COM);
424	<	if (fabs(zPos[i] - COM[whichDirection]) < zconsTol)
425	<	states.push_back(zcsFixed);
426	<	else
427	<	states.push_back(zcsMoving);
428	<	}
429	<
430	<
431	<	//The reason to declare fz and indexOfZconsMols as pointer to array is
432	<	// that we want to make the MPI communication simple
433	<	if(fz)
434	<	delete[] fz;
435	<
436	<	if(curZPos)
437	<	delete[] curZPos;
438	<
439	<	if(indexOfZConsMols)
440	<	delete[] indexOfZConsMols;
441	<
442	<	if (zconsMols.size() > 0){
443	<	fz = new double[zconsMols.size()];
444	<	curZPos = new double[zconsMols.size()];
445	<	indexOfZConsMols =  new int[zconsMols.size()];
446	<
447	<	if(!fz || !curZPos || !indexOfZConsMols){
448	<	sprintf( painCave.errMsg,
449	<	"Memory allocation failure in class Zconstraint\n");
450	<	painCave.isFatal = 1;
451	<	simError();
452	<	}
453	<
454	<	for(int i = 0; i < zconsMols.size(); i++){
455	<	indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex();
456	<	}
457	<
458	<	}
459	<	else{
460	<	fz = NULL;
461	<	curZPos = NULL;
462	<	indexOfZConsMols = NULL;
463	<	}
464	<
465	<	//
466	<	forcePolicy->update();
467	<
468	<	}
469	<
470	<	#endif
471	<
472	<	/**
473	<	*  Function Name: isZConstraintMol
474	<	*  Parameter
475	<	*    Molecule* mol
476	<	*  Return value:
477	<	*    -1, if the molecule is not z-constraint molecule,
478	<	*    other non-negative values, its index in indexOfAllZConsMols vector
479	<	*/
480	<
481	<	template<typename T> int ZConstraint<T>::isZConstraintMol(Molecule* mol)
482	<	{
483	<	int index;
484	<	int low;
485	<	int high;
486	<	int mid;
487	<
488	<	index = mol->getGlobalIndex();
489	<
490	<	low = 0;
491	<	high = parameters->size() - 1;
492	<
493	<	//Binary Search (we have sorted the array)
494	<	while(low <= high){
495	<	mid = (low + high) /2;
496	<	if ((*parameters)[mid].zconsIndex == index)
497	<	return mid;
498	<	else if ((*parameters)[mid].zconsIndex > index )
499	<	high = mid -1;
500	<	else
501	<	low = mid + 1;
502	<	}
503	<
504	<	return -1;
505	<	}
506	<
507	<	template<typename T> void ZConstraint<T>::integrate(){
508	<
509	<	//zero out the velocities of center of mass of unconstrained molecules
510	<	//and the velocities of center of mass of every single z-constrained molecueles
511	<	zeroOutVel();
512	<
513	<	T::integrate();
514	<
515	<	}
516	<
517	<
518	<	/**
519	<	*
520	<	*
521	<	*
522	<	*
523	<	*/
524	<	template<typename T> void ZConstraint<T>::calcForce(int calcPot, int calcStress){
525	<	double zsys;
526	<	double COM[3];
527	<	double force[3];
528	<	double zSysCOMVel;
529	<
530	<	T::calcForce(calcPot, calcStress);
531	<
532	<	if (checkZConsState()){
533	<	zeroOutVel();
534	<	forcePolicy->update();
535	<	}
536	<
537	<	zsys = calcZSys();
538	<	zSysCOMVel = calcSysCOMVel();
539	<	#ifdef IS_MPI
540	<	if(worldRank == 0){
541	<	#endif
542	<	cout << "---------------------------------------------------------------------" <<endl;
543	<	cout << "current time: " << info->getTime() << endl;
544	<	cout << "center of mass at z: " << zsys << endl;
545	<	cout << "before calcForce, the COMVel of system is " << zSysCOMVel <<endl;
546	<
547	<	#ifdef IS_MPI
548	<	}
549	<	#endif
550	<
551	<	//do zconstraint force;
552	<	if (haveFixedZMols())
553	<	this->doZconstraintForce();
554	<
555	<	//use harmonical poteintial to move the molecules to the specified positions
556	<	if (haveMovingZMols())
557	<	this->doHarmonic();
558	<
559	<	//write out forces and current positions of z-constraint molecules
560	<	if(info->getTime() >= curZconsTime){
561	<	for(int i = 0; i < zconsMols.size(); i++){
562	<	zconsMols[i]->getCOM(COM);
563	<	curZPos[i] = COM[whichDirection];
564	<
565	<	//if the z-constraint molecule is still moving, just record its force
566	<	if(states[i] == zcsMoving){
567	<	fz[i] = 0;
568	<	Atom** movingZAtoms;
569	<	movingZAtoms = zconsMols[i]->getMyAtoms();
570	<	for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
571	<	movingZAtoms[j]->getFrc(force);
572	<	fz[i] += force[whichDirection];
573	<	}
574	<	}
575	<	}
576	<	fzOut->writeFZ(info->getTime(), zconsMols.size(), indexOfZConsMols, fz, curZPos);
577	<	curZconsTime += zconsTime;
578	<	}
579	<
580	<	zSysCOMVel = calcSysCOMVel();
581	<	#ifdef IS_MPI
582	<	if(worldRank == 0){
583	<	#endif
584	<	cout << "after calcForce, the COMVel of system is " << zSysCOMVel <<endl;
585	<	#ifdef IS_MPI
586	<	}
587	<	#endif
588	<
589	<	}
590	<
591	<
592	<	/**
593	<	*
594	<	*/
595	<
596	<	template<typename T> double ZConstraint<T>::calcZSys()
597	<	{
598	<	//calculate reference z coordinate for z-constraint molecules
599	<	double totalMass_local;
600	<	double totalMass;
601	<	double totalMZ_local;
602	<	double totalMZ;
603	<	double massOfCurMol;
604	<	double COM[3];
605	<
606	<	totalMass_local = 0;
607	<	totalMZ_local = 0;
608	<
609	<	for(int i = 0; i < nMols; i++){
610	<	massOfCurMol = molecules[i].getTotalMass();
611	<	molecules[i].getCOM(COM);
612	<
613	<	totalMass_local += massOfCurMol;
614	<	totalMZ_local += massOfCurMol * COM[whichDirection];
615	<
616	<	}
617	<
618	<
619	<	#ifdef IS_MPI
620	<	MPI_Allreduce(&totalMass_local, &totalMass, 1,
621	<	MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
622	<	MPI_Allreduce(&totalMZ_local, &totalMZ, 1,
623	<	MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
624	<	#else
625	<	totalMass = totalMass_local;
626	<	totalMZ = totalMZ_local;
627	<	#endif
628	<
629	<	double zsys;
630	<	zsys = totalMZ / totalMass;
631	<
632	<	return zsys;
633	<	}
634	<
635	<	/**
636	<	*
637	<	*/
638	<	template<typename T> void ZConstraint<T>::thermalize( void ){
639	<
640	<	T::thermalize();
641	<	zeroOutVel();
642	<	}
643	<
644	<	/**
645	<	*
646	<	*/
647	<
648	<	template<typename T> void ZConstraint<T>::zeroOutVel(){
649	<
650	<	Atom** fixedZAtoms;
651	<	double COMvel[3];
652	<	double vel[3];
653	<	double zSysCOMVel;
654	<
655	<	//zero out the velocities of center of mass of fixed z-constrained molecules
656	<
657	<	for(int i = 0; i < zconsMols.size(); i++){
658	<
659	<	if (states[i] == zcsFixed){
660	<
661	<	zconsMols[i]->getCOMvel(COMvel);
662	<	//cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
663	<
664	<	fixedZAtoms = zconsMols[i]->getMyAtoms();
665	<
666	<	for(int j =0; j < zconsMols[i]->getNAtoms(); j++){
667	<	fixedZAtoms[j]->getVel(vel);
668	<	vel[whichDirection] -= COMvel[whichDirection];
669	<	fixedZAtoms[j]->setVel(vel);
670	<	}
671	<
672	<	zconsMols[i]->getCOMvel(COMvel);
673	<	//cout << "after resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
674	<	}
675	<
676	<	}
677	<
678	<	//cout << "before resetting the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl;
679	<
680	<	zSysCOMVel = calcSysCOMVel();
681	<	#ifdef IS_MPI
682	<	if(worldRank == 0){
683	<	#endif
684	<	cout << "before resetting the COMVel of sytem is " << zSysCOMVel << endl;
685	<	#ifdef IS_MPI
686	<	}
687	<	#endif
688	<
689	<	// calculate the vz of center of mass of unconstrained molecules and moving z-constrained molecules
690	<	double MVzOfMovingMols_local;
691	<	double MVzOfMovingMols;
692	<	double totalMassOfMovingZMols_local;
693	<	double totalMassOfMovingZMols;
694	<
695	<	MVzOfMovingMols_local = 0;
696	<	totalMassOfMovingZMols_local = 0;
697	<
698	<	for(int i =0; i < unconsMols.size(); i++){
699	<	unconsMols[i]->getCOMvel(COMvel);
700	<	MVzOfMovingMols_local += massOfUnconsMols[i] * COMvel[whichDirection];
701	<	}
702	<
703	<	for(int i = 0; i < zconsMols.size(); i++){
704	<	if (states[i] == zcsMoving){
705	<	zconsMols[i]->getCOMvel(COMvel);
706	<	MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];
707	<	totalMassOfMovingZMols_local += massOfZConsMols[i];
708	<	}
709	<
710	<	}
711	<
712	<	#ifndef IS_MPI
713	<	MVzOfMovingMols = MVzOfMovingMols_local;
714	<	totalMassOfMovingZMols = totalMassOfMovingZMols_local;
715	<	#else
716	<	MPI_Allreduce(&MVzOfMovingMols_local, &MVzOfMovingMols, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
717	<	MPI_Allreduce(&totalMassOfMovingZMols_local, &totalMassOfMovingZMols, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
718	<	#endif
719	<
720	<	double vzOfMovingMols;
721	<	vzOfMovingMols = MVzOfMovingMols / (totalMassOfUncons + totalMassOfMovingZMols);
722	<
723	<	//modify the velocites of unconstrained molecules
724	<	Atom** unconsAtoms;
725	<	for(int i = 0; i < unconsMols.size(); i++){
726	<
727	<	unconsAtoms = unconsMols[i]->getMyAtoms();
728	<	for(int j = 0; j < unconsMols[i]->getNAtoms();j++){
729	<	unconsAtoms[j]->getVel(vel);
730	<	vel[whichDirection] -= vzOfMovingMols;
731	<	unconsAtoms[j]->setVel(vel);
732	<	}
733	<
734	<	}
735	<
736	<	//modify the velocities of moving z-constrained molecuels
737	<	Atom** movingZAtoms;
738	<	for(int i = 0; i < zconsMols.size(); i++){
739	<
740	<	if (states[i] ==zcsMoving){
741	<
742	<	movingZAtoms = zconsMols[i]->getMyAtoms();
743	<	for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
744	<	movingZAtoms[j]->getVel(vel);
745	<	vel[whichDirection] -= vzOfMovingMols;
746	<	movingZAtoms[j]->setVel(vel);
747	<	}
748	<
749	<	}
750	<
751	<	}
752	<
753	<
754	<	zSysCOMVel = calcSysCOMVel();
755	<	#ifdef IS_MPI
756	<	if(worldRank == 0){
757	<	#endif
758	<	cout << "after resetting the COMVel of moving molecules is " << zSysCOMVel << endl;
759	<	#ifdef IS_MPI
760	<	}
761	<	#endif
762	<
763	<	}
764	<
765	<	/**
766	<	*
767	<	*/
768	<
769	<	template<typename T> void ZConstraint<T>::doZconstraintForce(){
770	<
771	<	Atom** zconsAtoms;
772	<	double totalFZ;
773	<	double totalFZ_local;
774	<	double COMvel[3];
775	<	double COM[3];
776	<	double force[3];
777	<
778	<
779	<
780	<	//constrain the molecules which do not reach the specified positions
781	<
782	<	//Zero Out the force of z-contrained molecules
783	<	totalFZ_local = 0;
784	<
785	<	//calculate the total z-contrained force of fixed z-contrained molecules
786	<
787	<	for(int i = 0; i < zconsMols.size(); i++){
788	<
789	<	if (states[i] == zcsFixed){
790	<
791	<	zconsMols[i]->getCOM(COM);
792	<	zconsAtoms = zconsMols[i]->getMyAtoms();
793	<
794	<	fz[i] = 0;
795	<	for(int j =0; j < zconsMols[i]->getNAtoms(); j++) {
796	<	zconsAtoms[j]->getFrc(force);
797	<	fz[i] += force[whichDirection];
798	<	}
799	<	totalFZ_local += fz[i];
800	<
801	<	cout << "Fixed Molecule\tindex: " << indexOfZConsMols[i]
802	<	<<"\tcurrent zpos: " << COM[whichDirection]
803	<	<< "\tcurrent fz: " <<fz[i] << endl;
804	<
805	<	}
806	<
807	<	}
808	<
809	<	//calculate total z-constraint force
810	<	#ifdef IS_MPI
811	<	MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
812	<	#else
813	<	totalFZ = totalFZ_local;
814	<	#endif
815	<
816	<
817	<	// apply negative to fixed z-constrained molecues;
818	<	force[0]= 0;
819	<	force[1]= 0;
820	<	force[2]= 0;
821	<
822	<	for(int i = 0; i < zconsMols.size(); i++){
823	<
824	<	if (states[i] == zcsFixed){
825	<
826	<	int nAtomOfCurZConsMol = zconsMols[i]->getNAtoms();
827	<	zconsAtoms = zconsMols[i]->getMyAtoms();
828	<
829	<	for(int j =0; j < nAtomOfCurZConsMol; j++) {
830	<	force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol;
831	<	//force[whichDirection] = - forcePolicy->getZFOfFixedZMols(zconsMols[i], zconsAtoms[j], fz[i]);
832	<	zconsAtoms[j]->addFrc(force);
833	<	}
834	<
835	<	}
836	<
837	<	}
838	<
839	<	//cout << "after zero out z-constraint force on fixed z-constraint molecuels "
840	<	//       << "total force is " << calcTotalForce() << endl;
841	<
842	<	//calculate the number of atoms of moving z-constrained molecules
843	<	int nMovingZAtoms_local;
844	<	int nMovingZAtoms;
845	<
846	<	nMovingZAtoms_local = 0;
847	<	for(int i = 0; i < zconsMols.size(); i++)
848	<	if(states[i] == zcsMoving)
849	<	nMovingZAtoms_local += zconsMols[i]->getNAtoms();
850	<
851	<	#ifdef IS_MPI
852	<	MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1,
853	<	MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
854	<	#else
855	<	nMovingZAtoms = nMovingZAtoms_local;
856	<	#endif
857	<
858	<	force[0]= 0;
859	<	force[1]= 0;
860	<	force[2]= 0;
861	<
862	<	//modify the forces of unconstrained molecules
863	<	for(int i = 0; i < unconsMols.size(); i++){
864	<
865	<	Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
866	<
867	<	for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){
868	<	force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
869	<	//force[whichDirection] = forcePolicy->getZFOfMovingMols(unconsAtoms[j],totalFZ);
870	<	unconsAtoms[j]->addFrc(force);
871	<	}
872	<
873	<	}
874	<
875	<	//modify the forces of moving z-constrained molecules
876	<	for(int i = 0; i < zconsMols.size(); i++) {
877	<	if (states[i] == zcsMoving){
878	<
879	<	Atom** movingZAtoms = zconsMols[i]->getMyAtoms();
880	<
881	<	for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
882	<	force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
883	<	//force[whichDirection] = forcePolicy->getZFOfMovingMols(movingZAtoms[j],totalFZ);
884	<	movingZAtoms[j]->addFrc(force);
885	<	}
886	<	}
887	<	}
888	<
889	<	//cout << "after substracting z-constraint force from moving molecuels "
890	<	//      << "total force is " << calcTotalForce()  << endl;
891	<
892	<	}
893	<
894	<	/**
895	<	*
896	<	*
897	<	*/
898	<
899	<	template<typename T> void ZConstraint<T>::doHarmonic(){
900	<	double force[3];
901	<	double harmonicU;
902	<	double harmonicF;
903	<	double COM[3];
904	<	double diff;
905	<	double totalFZ_local;
906	<	double totalFZ;
907	<
908	<	force[0] = 0;
909	<	force[1] = 0;
910	<	force[2] = 0;
911	<
912	<	totalFZ_local = 0;
913	<
914	<	for(int i = 0; i < zconsMols.size(); i++) {
915	<
916	<	if (states[i] == zcsMoving){
917	<	zconsMols[i]->getCOM(COM);
918	<	cout << "Moving Molecule\tindex: " << indexOfZConsMols[i] <<"\tcurrent zpos: " << COM[whichDirection] << endl;
919	<
920	<	diff = COM[whichDirection] -zPos[i];
921	<
922	<	harmonicU = 0.5 * kz[i] * diff * diff;
923	<	info->lrPot += harmonicU;
924	<
925	<	harmonicF =  - kz[i] * diff;
926	<	totalFZ_local += harmonicF;
927	<
928	<	//adjust force
929	<
930	<	Atom** movingZAtoms = zconsMols[i]->getMyAtoms();
931	<
932	<	for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){
933	<	force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms();
934	<	//force[whichDirection] = forcePolicy->getHFOfFixedZMols(zconsMols[i], movingZAtoms[j], harmonicF);
935	<	movingZAtoms[j]->addFrc(force);
936	<	}
937	<	}
938	<
939	<	}
940	<
941	<	#ifndef IS_MPI
942	<	totalFZ = totalFZ_local;
943	<	#else
944	<	MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
945	<	#endif
946	<
947	<	force[0]= 0;
948	<	force[1]= 0;
949	<	force[2]= 0;
950	<
951	<	//modify the forces of unconstrained molecules
952	<	for(int i = 0; i < unconsMols.size(); i++){
953	<
954	<	Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
955	<
956	<	for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){
957	<	force[whichDirection] = - totalFZ /totNumOfUnconsAtoms;
958	<	//force[whichDirection] = - forcePolicy->getHFOfUnconsMols(unconsAtoms[j], totalFZ);
959	<	unconsAtoms[j]->addFrc(force);
960	<	}
961	<	}
962	<
963	<	}
964	<
965	<	/**
966	<	*
967	<	*/
968	<
969	<	template<typename T> bool ZConstraint<T>::checkZConsState(){
970	<	double COM[3];
971	<	double diff;
972	<
973	<	int changed_local;
974	<	int changed;
975	<
976	<	changed_local = 0;
977	<
978	<	for(int i =0; i < zconsMols.size(); i++){
979	<
980	<	zconsMols[i]->getCOM(COM);
981	<	diff = fabs(COM[whichDirection] - zPos[i]);
982	<	if (  diff <= zconsTol && states[i] == zcsMoving){
983	<	states[i] = zcsFixed;
984	<	changed_local = 1;
985	<	}
986	<	else if ( diff > zconsTol && states[i] == zcsFixed){
987	<	states[i] = zcsMoving;
988	<	changed_local = 1;
989	<	}
990	<
991	<	}
992	<
993	<	#ifndef IS_MPI
994	<	changed =changed_local;
995	<	#else
996	<	MPI_Allreduce(&changed_local, &changed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
997	<	#endif
998	<
999	<	return changed > 0 ? true : false;
1000	<	}
1001	<
1002	<	template<typename T> bool ZConstraint<T>::haveFixedZMols(){
1003	<
1004	<	int havingFixed_local;
1005	<	int havingFixed;
1006	<
1007	<	havingFixed_local = 0;
1008	<
1009	<	for(int i = 0; i < zconsMols.size(); i++)
1010	<	if (states[i] == zcsFixed){
1011	<	havingFixed_local = 1;
1012	<	break;
1013	<	}
1014	<
1015	<	#ifndef IS_MPI
1016	<	havingFixed = havingFixed_local;
1017	<	#else
1018	<	MPI_Allreduce(&havingFixed_local, &havingFixed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
1019	<	#endif
1020	<
1021	<	return havingFixed > 0 ? true : false;
1022	<	}
1023	<
1024	<
1025	<	/**
1026	<	*
1027	<	*/
1028	<	template<typename T> bool ZConstraint<T>::haveMovingZMols(){
1029	<
1030	<	int havingMoving_local;
1031	<	int havingMoving;
1032	<
1033	<	havingMoving_local = 0;
1034	<
1035	<	for(int i = 0; i < zconsMols.size(); i++)
1036	<	if (states[i] == zcsMoving){
1037	<	havingMoving_local = 1;
1038	<	break;
1039	<	}
1040	<
1041	<	#ifndef IS_MPI
1042	<	havingMoving = havingMoving_local;
1043	<	#else
1044	<	MPI_Allreduce(&havingMoving_local, &havingMoving, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
1045	<	#endif
1046	<
1047	<	return havingMoving > 0 ? true : false;
1048	<
1049	<	}
1050	<
1051	<	/**
1052	<	*
1053	<	*/
1054	<
1055	<	template<typename T> double ZConstraint<T>::calcMovingMolsCOMVel()
1056	<	{
1057	<	double MVzOfMovingMols_local;
1058	<	double MVzOfMovingMols;
1059	<	double totalMassOfMovingZMols_local;
1060	<	double totalMassOfMovingZMols;
1061	<	double COMvel[3];
1062	<
1063	<	MVzOfMovingMols_local = 0;
1064	<	totalMassOfMovingZMols_local = 0;
1065	<
1066	<	for(int i =0; i < unconsMols.size(); i++){
1067	<	unconsMols[i]->getCOMvel(COMvel);
1068	<	MVzOfMovingMols_local += massOfUnconsMols[i] * COMvel[whichDirection];
1069	<	}
1070	<
1071	<	for(int i = 0; i < zconsMols.size(); i++){
1072	<
1073	<	if (states[i] == zcsMoving){
1074	<	zconsMols[i]->getCOMvel(COMvel);
1075	<	MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];
1076	<	totalMassOfMovingZMols_local += massOfZConsMols[i];
1077	<	}
1078	<
1079	<	}
1080	<
1081	<	#ifndef IS_MPI
1082	<	MVzOfMovingMols = MVzOfMovingMols_local;
1083	<	totalMassOfMovingZMols = totalMassOfMovingZMols_local;
1084	<	#else
1085	<	MPI_Allreduce(&MVzOfMovingMols_local, &MVzOfMovingMols, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1086	<	MPI_Allreduce(&totalMassOfMovingZMols_local, &totalMassOfMovingZMols, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1087	<	#endif
1088	<
1089	<	double vzOfMovingMols;
1090	<	vzOfMovingMols = MVzOfMovingMols / (totalMassOfUncons + totalMassOfMovingZMols);
1091	<
1092	<	return vzOfMovingMols;
1093	<	}
1094	<
1095	<	/**
1096	<	*
1097	<	*/
1098	<
1099	<	template<typename T> double ZConstraint<T>::calcSysCOMVel()
1100	<	{
1101	<	double COMvel[3];
1102	<	double tempMVz_local;
1103	<	double tempMVz;
1104	<	double massOfZCons_local;
1105	<	double massOfZCons;
1106	<
1107	<
1108	<	tempMVz_local = 0;
1109	<
1110	<	for(int i =0 ; i < nMols; i++){
1111	<	molecules[i].getCOMvel(COMvel);
1112	<	tempMVz_local += molecules[i].getTotalMass()*COMvel[whichDirection];
1113	<	}
1114	<
1115	<	massOfZCons_local = 0;
1116	<
1117	<	for(int i = 0; i < massOfZConsMols.size(); i++){
1118	<	massOfZCons_local += massOfZConsMols[i];
1119	<	}
1120	<	#ifndef IS_MPI
1121	<	massOfZCons = massOfZCons_local;
1122	<	tempMVz = tempMVz_local;
1123	<	#else
1124	<	MPI_Allreduce(&massOfZCons_local, &massOfZCons, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1125	<	MPI_Allreduce(&tempMVz_local, &tempMVz, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1126	<	#endif
1127	<
1128	<	return tempMVz /(totalMassOfUncons + massOfZCons);
1129	<	}
1130	<
1131	<	/**
1132	<	*
1133	<	*/
1134	<
1135	<	template<typename T> double ZConstraint<T>::calcTotalForce(){
1136	<
1137	<	double force[3];
1138	<	double totalForce_local;
1139	<	double totalForce;
1140	<
1141	<	totalForce_local = 0;
1142	<
1143	<	for(int i = 0; i < nAtoms; i++){
1144	<	atoms[i]->getFrc(force);
1145	<	totalForce_local += force[whichDirection];
1146	<	}
1147	<
1148	<	#ifndef IS_MPI
1149	<	totalForce = totalForce_local;
1150	<	#else
1151	<	MPI_Allreduce(&totalForce_local, &totalForce, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1152	<	#endif
1153	<
1154	<	return totalForce;
1155	<
1156	<	}
1157	<
1158	<	/**
1159	<	*
1160	<	*/
1161	<
1162	<	template<typename T> void ZConstraint<T>::PolicyByNumber::update(){
1163	<	//calculate the number of atoms of moving z-constrained molecules
1164	<	int nMovingZAtoms_local;
1165	<	int nMovingZAtoms;
1166	<
1167	<	nMovingZAtoms_local = 0;
1168	<	for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++)
1169	<	if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving))
1170	<	nMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getNAtoms();
1171	<
1172	<	#ifdef IS_MPI
1173	<	MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
1174	<	#else
1175	<	nMovingZAtoms = nMovingZAtoms_local;
1176	<	#endif
1177	<	totNumOfMovingAtoms = nMovingZAtoms + zconsIntegrator->totNumOfUnconsAtoms;
1178	<	}
1179	<
1180	<	template<typename T>double ZConstraint<T>::PolicyByNumber::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
1181	<	return totalForce / mol->getNAtoms();
1182	<	}
1183	<
1184	<	template<typename T> double ZConstraint<T>::PolicyByNumber::getZFOfMovingMols(Atom* atom, double totalForce){
1185	<	return totalForce / totNumOfMovingAtoms;
1186	<	}
1187	<
1188	<	template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
1189	<	return totalForce / mol->getNAtoms();
1190	<	}
1191	<
1192	<	template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfUnconsMols(Atom* atom, double totalForce){
1193	<	return totalForce / zconsIntegrator->totNumOfUnconsAtoms;
1194	<	}
1195	<
1196	<	/**
1197	<	*
1198	<	*/
1199	<
1200	<	template<typename T> void ZConstraint<T>::PolicyByMass::update(){
1201	<	//calculate the number of atoms of moving z-constrained molecules
1202	<	double massOfMovingZAtoms_local;
1203	<	double massOfMovingZAtoms;
1204	<
1205	<	massOfMovingZAtoms_local = 0;
1206	<	for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++)
1207	<	if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving))
1208	<	massOfMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getTotalMass();
1209	<
1210	<	#ifdef IS_MPI
1211	<	MPI_Allreduce(&massOfMovingZAtoms_local, &massOfMovingZAtoms, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);
1212	<	#else
1213	<	massOfMovingZAtoms = massOfMovingZAtoms_local;
1214	<	#endif
1215	<	totMassOfMovingAtoms = massOfMovingZAtoms_local + zconsIntegrator->totalMassOfUncons;
1216	<	}
1217	<
1218	<	template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
1219	<	return totalForce * atom->getMass() / mol->getTotalMass();
1220	<	}
1221	<
1222	<	template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfMovingMols( Atom* atom, double totalForce){
1223	<	return totalForce * atom->getMass() / totMassOfMovingAtoms;
1224	<	}
1225	<
1226	<	template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){
1227	<	return totalForce * atom->getMass() / mol->getTotalMass();
1228	<	}
1229	<
1230	<	template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfUnconsMols(Atom* atom, double totalForce){
1231	<	return totalForce * atom->getMass() / zconsIntegrator->totalMassOfUncons;
1232	<	}
1233	<
1	>	#include "Integrator.hpp"
2	>	#include "simError.h"
3	>	#include <math.h>
4	>
5	>	const double INFINITE_TIME = 10e30;
6	>	template<typename T> ZConstraint<T>::ZConstraint(SimInfo* theInfo,
7	>	ForceFields* the_ff): T(theInfo, the_ff),
8	>	indexOfZConsMols(NULL),
9	>	fz(NULL),
10	>	curZPos(NULL),
11	>	fzOut(NULL),
12	>	curZconsTime(0),
13	>	forcePolicy(NULL),
14	>	hasZConsGap(false){
15	>	//get properties from SimInfo
16	>	GenericData* data;
17	>	ZConsParaData* zConsParaData;
18	>	DoubleData* sampleTime;
19	>	DoubleData* tolerance;
20	>	DoubleData* gap;
21	>	DoubleData* fixtime;
22	>	StringData* policy;
23	>	StringData* filename;
24	>	double COM[3];
25	>
26	>	//by default, the direction of constraint is z
27	>	// 0 --> x
28	>	// 1 --> y
29	>	// 2 --> z
30	>	whichDirection = 2;
31	>
32	>	//estimate the force constant of harmonical potential
33	>	double Kb = 1.986E-3 ; //in kcal/K
34	>
35	>	double halfOfLargestBox = max(info->boxL[0], max(info->boxL[1], info->boxL[2])) /
36	>	2;
37	>	zForceConst = Kb * info->target_temp / (halfOfLargestBox * halfOfLargestBox);
38	>
39	>	//creat force Subtraction policy
40	>	data = info->getProperty(ZCONSFORCEPOLICY_ID);
41	>	if (!data){
42	>	sprintf(painCave.errMsg,
43	>	"ZConstraint Warning: User does not set force Subtraction policy, "
44	>	"PolicyByMass is used\n");
45	>	painCave.isFatal = 0;
46	>	simError();
47	>
48	>	forcePolicy = (ForceSubtractionPolicy *) new PolicyByMass(this);
49	>	}
50	>	else{
51	>	policy = dynamic_cast<StringData*>(data);
52	>
53	>	if (!policy){
54	>	sprintf(painCave.errMsg,
55	>	"ZConstraint Error: Convertion from GenericData to StringData failure, "
56	>	"PolicyByMass is used\n");
57	>	painCave.isFatal = 0;
58	>	simError();
59	>
60	>	forcePolicy = (ForceSubtractionPolicy *) new PolicyByMass(this);
61	>	}
62	>	else{
63	>	if (policy->getData() == "BYNUMBER")
64	>	forcePolicy = (ForceSubtractionPolicy *) new PolicyByNumber(this);
65	>	else if (policy->getData() == "BYMASS")
66	>	forcePolicy = (ForceSubtractionPolicy *) new PolicyByMass(this);
67	>	else{
68	>	sprintf(painCave.errMsg,
69	>	"ZConstraint Warning: unknown force Subtraction policy, "
70	>	"PolicyByMass is used\n");
71	>	painCave.isFatal = 0;
72	>	simError();
73	>	forcePolicy = (ForceSubtractionPolicy *) new PolicyByMass(this);
74	>	}
75	>	}
76	>	}
77	>
78	>
79	>	//retrieve sample time of z-contraint
80	>	data = info->getProperty(ZCONSTIME_ID);
81	>
82	>	if (!data){
83	>	sprintf(painCave.errMsg,
84	>	"ZConstraint error: If you use an ZConstraint\n"
85	>	" , you must set sample time.\n");
86	>	painCave.isFatal = 1;
87	>	simError();
88	>	}
89	>	else{
90	>	sampleTime = dynamic_cast<DoubleData*>(data);
91	>
92	>	if (!sampleTime){
93	>	sprintf(painCave.errMsg,
94	>	"ZConstraint error: Can not get property from SimInfo\n");
95	>	painCave.isFatal = 1;
96	>	simError();
97	>	}
98	>	else{
99	>	this->zconsTime = sampleTime->getData();
100	>	}
101	>	}
102	>
103	>	//retrieve output filename of z force
104	>	data = info->getProperty(ZCONSFILENAME_ID);
105	>	if (!data){
106	>	sprintf(painCave.errMsg,
107	>	"ZConstraint error: If you use an ZConstraint\n"
108	>	" , you must set output filename of z-force.\n");
109	>	painCave.isFatal = 1;
110	>	simError();
111	>	}
112	>	else{
113	>	filename = dynamic_cast<StringData*>(data);
114	>
115	>	if (!filename){
116	>	sprintf(painCave.errMsg,
117	>	"ZConstraint error: Can not get property from SimInfo\n");
118	>	painCave.isFatal = 1;
119	>	simError();
120	>	}
121	>	else{
122	>	this->zconsOutput = filename->getData();
123	>	}
124	>	}
125	>
126	>	//retrieve tolerance for z-constraint molecuels
127	>	data = info->getProperty(ZCONSTOL_ID);
128	>
129	>	if (!data){
130	>	sprintf(painCave.errMsg, "ZConstraint error: can not get tolerance \n");
131	>	painCave.isFatal = 1;
132	>	simError();
133	>	}
134	>	else{
135	>	tolerance = dynamic_cast<DoubleData*>(data);
136	>
137	>	if (!tolerance){
138	>	sprintf(painCave.errMsg,
139	>	"ZConstraint error: Can not get property from SimInfo\n");
140	>	painCave.isFatal = 1;
141	>	simError();
142	>	}
143	>	else{
144	>	this->zconsTol = tolerance->getData();
145	>	}
146	>	}
147	>
148	>	//quick hack here
149	>	data = info->getProperty(ZCONSGAP_ID);
150	>
151	>	if (data){
152	>	gap = dynamic_cast<DoubleData*>(data);
153	>	}
154	>
155	>	if (!gap){
156	>	sprintf(painCave.errMsg,
157	>	"ZConstraint error: Can not get property from SimInfo\n");
158	>	painCave.isFatal = 1;
159	>	simError();
160	>	}
161	>	else{
162	>	this->hasZConsGap = true;
163	>	this->zconsGap = gap->getData();
164	>	}
165	>
166	>	data = info->getProperty(ZCONSFIXTIME_ID);
167	>
168	>	if (data){
169	>	fixtime = dynamic_cast<DoubleData*>(data);
170	>	}
171	>
172	>	if (!fixtime){
173	>	sprintf(painCave.errMsg,
174	>	"ZConstraint error: Can not get property from SimInfo\n");
175	>	painCave.isFatal = 1;
176	>	simError();
177	>	}
178	>	else{
179	>	this->zconsFixTime = fixtime->getData();
180	>	}
181	>
182	>
183	>
184	>	//retrieve index of z-constraint molecules
185	>	data = info->getProperty(ZCONSPARADATA_ID);
186	>	if (!data){
187	>	sprintf(painCave.errMsg,
188	>	"ZConstraint error: If you use an ZConstraint\n"
189	>	" , you must set index of z-constraint molecules.\n");
190	>	painCave.isFatal = 1;
191	>	simError();
192	>	}
193	>	else{
194	>	zConsParaData = dynamic_cast<ZConsParaData*>(data);
195	>
196	>	if (!zConsParaData){
197	>	sprintf(painCave.errMsg,
198	>	"ZConstraint error: Can not get parameters of zconstraint method from SimInfo\n");
199	>	painCave.isFatal = 1;
200	>	simError();
201	>	}
202	>	else{
203	>	parameters = zConsParaData->getData();
204	>
205	>	//check the range of zconsIndex
206	>	//and the minimum value of index is the first one (we already sorted the data)
207	>	//the maximum value of index is the last one
208	>
209	>	int maxIndex;
210	>	int minIndex;
211	>	int totalNumMol;
212	>
213	>	minIndex = (*parameters)[0].zconsIndex;
214	>	if (minIndex < 0){
215	>	sprintf(painCave.errMsg, "ZConstraint error: index is out of range\n");
216	>	painCave.isFatal = 1;
217	>	simError();
218	>	}
219	>
220	>	maxIndex = (*parameters)[parameters->size() - 1].zconsIndex;
221	>
222	>	#ifndef IS_MPI
223	>	totalNumMol = nMols;
224	>	#else
225	>	totalNumMol = mpiSim->getTotNmol();
226	>	#endif
227	>
228	>	if (maxIndex > totalNumMol - 1){
229	>	sprintf(painCave.errMsg, "ZConstraint error: index is out of range\n");
230	>	painCave.isFatal = 1;
231	>	simError();
232	>	}
233	>
234	>	//if user does not specify the zpos for the zconstraint molecule
235	>	//its initial z coordinate  will be used as default
236	>	for (int i = 0; i < (int) (parameters->size()); i++){
237	>	if (!(*parameters)[i].havingZPos){
238	>	#ifndef IS_MPI
239	>	for (int j = 0; j < nMols; j++){
240	>	if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
241	>	molecules[j].getCOM(COM);
242	>	break;
243	>	}
244	>	}
245	>	#else
246	>	//query which processor current zconstraint molecule belongs to
247	>	int* MolToProcMap;
248	>	int whichNode;
249	>
250	>	MolToProcMap = mpiSim->getMolToProcMap();
251	>	whichNode = MolToProcMap[(*parameters)[i].zconsIndex];
252	>
253	>	//broadcast the zpos of current z-contraint molecule
254	>	//the node which contain this
255	>
256	>	if (worldRank == whichNode){
257	>	for (int j = 0; j < nMols; j++)
258	>	if (molecules[j].getGlobalIndex() == (*parameters)[i].zconsIndex){
259	>	molecules[j].getCOM(COM);
260	>	break;
261	>	}
262	>	}
263	>
264	>	MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE, whichNode,
265	>	MPI_COMM_WORLD);
266	>	#endif
267	>
268	>	(*parameters)[i].zPos = COM[whichDirection];
269	>
270	>	sprintf(painCave.errMsg,
271	>	"ZConstraint warning: Does not specify zpos for z-constraint molecule "
272	>	"initial z coornidate will be used \n");
273	>	painCave.isFatal = 0;
274	>	simError();
275	>	}
276	>	}
277	>	}//end if (!zConsParaData)
278	>
279	>	}//end  if (!data)
280	>
281	>	//
282	>	#ifdef IS_MPI
283	>	update();
284	>	#else
285	>	int searchResult;
286	>
287	>	for (int i = 0; i < nMols; i++){
288	>	searchResult = isZConstraintMol(&molecules[i]);
289	>
290	>	if (searchResult > -1){
291	>	zconsMols.push_back(&molecules[i]);
292	>	massOfZConsMols.push_back(molecules[i].getTotalMass());
293	>
294	>	zPos.push_back((*parameters)[searchResult].zPos);
295	>	//       cout << "index: "<< (*parameters)[searchResult].zconsIndex
296	>	//              <<"\tzPos = " << (*parameters)[searchResult].zPos << endl;
297	>
298	>	kz.push_back((*parameters)[searchResult]. kRatio * zForceConst);
299	>	molecules[i].getCOM(COM);
300	>	}
301	>	else{
302	>	unconsMols.push_back(&molecules[i]);
303	>	massOfUnconsMols.push_back(molecules[i].getTotalMass());
304	>	}
305	>	}
306	>
307	>	fz = new double[zconsMols.size()];
308	>	curZPos = new double[zconsMols.size()];
309	>	indexOfZConsMols = new int [zconsMols.size()];
310	>
311	>	if (!fz || !curZPos || !indexOfZConsMols){
312	>	sprintf(painCave.errMsg, "Memory allocation failure in class Zconstraint\n");
313	>	painCave.isFatal = 1;
314	>	simError();
315	>	}
316	>
317	>	//determine the states of z-constraint molecules
318	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
319	>	indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex();
320	>
321	>	zconsMols[i]->getCOM(COM);
322	>	if (fabs(zPos[i] - COM[whichDirection]) < zconsTol){
323	>	states.push_back(zcsFixed);
324	>
325	>	if (hasZConsGap)
326	>	endFixTime.push_back(info->getTime() + zconsFixTime);
327	>	}
328	>	else{
329	>	states.push_back(zcsMoving);
330	>
331	>	if (hasZConsGap)
332	>	endFixTime.push_back(INFINITE_TIME);
333	>	}
334	>	}
335	>
336	>	#endif
337	>
338	>	//get total masss of unconstraint molecules
339	>	double totalMassOfUncons_local;
340	>	totalMassOfUncons_local = 0;
341	>
342	>	for (int i = 0; i < (int) (unconsMols.size()); i++)
343	>	totalMassOfUncons_local += unconsMols[i]->getTotalMass();
344	>
345	>	#ifndef IS_MPI
346	>	totalMassOfUncons = totalMassOfUncons_local;
347	>	#else
348	>	MPI_Allreduce(&totalMassOfUncons_local, &totalMassOfUncons, 1, MPI_DOUBLE,
349	>	MPI_SUM, MPI_COMM_WORLD);
350	>	#endif
351	>
352	>	//get total number of unconstrained atoms
353	>	int nUnconsAtoms_local;
354	>	nUnconsAtoms_local = 0;
355	>	for (int i = 0; i < (int) (unconsMols.size()); i++)
356	>	nUnconsAtoms_local += unconsMols[i]->getNAtoms();
357	>
358	>	#ifndef IS_MPI
359	>	totNumOfUnconsAtoms = nUnconsAtoms_local;
360	>	#else
361	>	MPI_Allreduce(&nUnconsAtoms_local, &totNumOfUnconsAtoms, 1, MPI_INT, MPI_SUM,
362	>	MPI_COMM_WORLD);
363	>	#endif
364	>
365	>	forcePolicy->update();
366	>	}
367	>
368	>	template<typename T> ZConstraint<T>::~ZConstraint(){
369	>	if (fz){
370	>	delete[] fz;
371	>	}
372	>
373	>	if (curZPos){
374	>	delete[] curZPos;
375	>	}
376	>
377	>	if (indexOfZConsMols){
378	>	delete[] indexOfZConsMols;
379	>	}
380	>
381	>	if (fzOut){
382	>	delete fzOut;
383	>	}
384	>
385	>	if (forcePolicy){
386	>	delete forcePolicy;
387	>	}
388	>	}
389	>
390	>
391	>	/**
392	>	*
393	>	*/
394	>
395	>	#ifdef IS_MPI
396	>	template<typename T> void ZConstraint<T>::update(){
397	>	double COM[3];
398	>	int index;
399	>
400	>	zconsMols.clear();
401	>	massOfZConsMols.clear();
402	>	zPos.clear();
403	>	kz.clear();
404	>
405	>	unconsMols.clear();
406	>	massOfUnconsMols.clear();
407	>
408	>
409	>	//creat zconsMol and unconsMol lists
410	>	for (int i = 0; i < nMols; i++){
411	>	index = isZConstraintMol(&molecules[i]);
412	>
413	>	if (index > -1){
414	>	zconsMols.push_back(&molecules[i]);
415	>	zPos.push_back((*parameters)[index].zPos);
416	>	kz.push_back((*parameters)[index].kRatio * zForceConst);
417	>	massOfZConsMols.push_back(molecules[i].getTotalMass());
418	>
419	>	molecules[i].getCOM(COM);
420	>	}
421	>	else{
422	>	unconsMols.push_back(&molecules[i]);
423	>	massOfUnconsMols.push_back(molecules[i].getTotalMass());
424	>	}
425	>	}
426	>
427	>
428	>	//The reason to declare fz and indexOfZconsMols as pointer to array is
429	>	// that we want to make the MPI communication simple
430	>	if (fz){
431	>	delete[] fz;
432	>	}
433	>
434	>	if (curZPos){
435	>	delete[] curZPos;
436	>	}
437	>
438	>	if (indexOfZConsMols){
439	>	delete[] indexOfZConsMols;
440	>	}
441	>
442	>	if (zconsMols.size() > 0){
443	>	fz = new double[zconsMols.size()];
444	>	curZPos = new double[zconsMols.size()];
445	>	indexOfZConsMols = new int[zconsMols.size()];
446	>
447	>	if (!fz || !curZPos || !indexOfZConsMols){
448	>	sprintf(painCave.errMsg,
449	>	"Memory allocation failure in class Zconstraint\n");
450	>	painCave.isFatal = 1;
451	>	simError();
452	>	}
453	>
454	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
455	>	indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex();
456	>	}
457	>	}
458	>	else{
459	>	fz = NULL;
460	>	curZPos = NULL;
461	>	indexOfZConsMols = NULL;
462	>	}
463	>
464	>	//determine the states of z-constraint molecules
465	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
466	>	zconsMols[i]->getCOM(COM);
467	>	if (fabs(zPos[i] - COM[whichDirection]) < zconsTol){
468	>	states.push_back(zcsFixed);
469	>
470	>	if (hasZConsGap)
471	>	endFixTime.push_back(info->getTime() + zconsFixTime);
472	>	}
473	>	else{
474	>	states.push_back(zcsMoving);
475	>
476	>	if (hasZConsGap)
477	>	endFixTime.push_back(INFINITE_TIME);
478	>	}
479	>	}
480	>	//
481	>	forcePolicy->update();
482	>	}
483	>
484	>	#endif
485	>
486	>	/**
487	>	*  Function Name: isZConstraintMol
488	>	*  Parameter
489	>	*    Molecule* mol
490	>	*  Return value:
491	>	*    -1, if the molecule is not z-constraint molecule,
492	>	*    other non-negative values, its index in indexOfAllZConsMols vector
493	>	*/
494	>
495	>	template<typename T> int ZConstraint<T>::isZConstraintMol(Molecule* mol){
496	>	int index;
497	>	int low;
498	>	int high;
499	>	int mid;
500	>
501	>	index = mol->getGlobalIndex();
502	>
503	>	low = 0;
504	>	high = parameters->size() - 1;
505	>
506	>	//Binary Search (we have sorted the array)
507	>	while (low <= high){
508	>	mid = (low + high) / 2;
509	>	if ((*parameters)[mid].zconsIndex == index)
510	>	return mid;
511	>	else if ((*parameters)[mid].zconsIndex > index)
512	>	high = mid - 1;
513	>	else
514	>	low = mid + 1;
515	>	}
516	>
517	>	return -1;
518	>	}
519	>
520	>	template<typename T> void ZConstraint<T>::integrate(){
521	>	// creat zconsWriter
522	>	fzOut = new ZConsWriter(zconsOutput.c_str(), parameters);
523	>
524	>	if (!fzOut){
525	>	sprintf(painCave.errMsg, "Memory allocation failure in class Zconstraint\n");
526	>	painCave.isFatal = 1;
527	>	simError();
528	>	}
529	>
530	>	//zero out the velocities of center of mass of unconstrained molecules
531	>	//and the velocities of center of mass of every single z-constrained molecueles
532	>	zeroOutVel();
533	>
534	>	curZconsTime = zconsTime + info->getTime();
535	>
536	>	T::integrate();
537	>	}
538	>
539	>
540	>	/**
541	>	*
542	>	*
543	>	*
544	>	*
545	>	*/
546	>	template<typename T> void ZConstraint<T>::calcForce(int calcPot, int calcStress){
547	>	double zsys;
548	>	double COM[3];
549	>	double force[3];
550	>	double zSysCOMVel;
551	>
552	>	T::calcForce(calcPot, calcStress);
553	>
554	>
555	>	if (hasZConsGap){
556	>	updateZPos();
557	>	}
558	>
559	>	if (checkZConsState()){
560	>	zeroOutVel();
561	>	forcePolicy->update();
562	>	}
563	>
564	>	zsys = calcZSys();
565	>	zSysCOMVel = calcSysCOMVel();
566	>	#ifdef IS_MPI
567	>	if (worldRank == 0){
568	>	#endif
569	>	//cout << "---------------------------------------------------------------------" <<endl;
570	>	//cout << "current time: " << info->getTime() << endl;
571	>	//cout << "center of mass at z: " << zsys << endl;
572	>	//cout << "before calcForce, the COMVel of system is " << zSysCOMVel <<endl;
573	>
574	>	#ifdef IS_MPI
575	>	}
576	>	#endif
577	>
578	>	//do zconstraint force;
579	>	if (haveFixedZMols()){
580	>	this->doZconstraintForce();
581	>	}
582	>
583	>	//use harmonical poteintial to move the molecules to the specified positions
584	>	if (haveMovingZMols()){
585	>	this->doHarmonic();
586	>	}
587	>
588	>	//write out forces and current positions of z-constraint molecules
589	>	if (info->getTime() >= curZconsTime){
590	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
591	>	zconsMols[i]->getCOM(COM);
592	>	curZPos[i] = COM[whichDirection];
593	>
594	>	//if the z-constraint molecule is still moving, just record its force
595	>	if (states[i] == zcsMoving){
596	>	fz[i] = 0;
597	>	Atom** movingZAtoms;
598	>	movingZAtoms = zconsMols[i]->getMyAtoms();
599	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
600	>	movingZAtoms[j]->getFrc(force);
601	>	fz[i] += force[whichDirection];
602	>	}
603	>	}
604	>	}
605	>	fzOut->writeFZ(info->getTime(), zconsMols.size(), indexOfZConsMols, fz,
606	>	curZPos, &zPos[0]);
607	>	curZconsTime += zconsTime;
608	>	}
609	>
610	>	zSysCOMVel = calcSysCOMVel();
611	>	#ifdef IS_MPI
612	>	if (worldRank == 0){
613	>	#endif
614	>	//cout << "after calcForce, the COMVel of system is " << zSysCOMVel <<endl;
615	>	#ifdef IS_MPI
616	>	}
617	>	#endif
618	>	}
619	>
620	>
621	>	/**
622	>	*
623	>	*/
624	>
625	>	template<typename T> double ZConstraint<T>::calcZSys(){
626	>	//calculate reference z coordinate for z-constraint molecules
627	>	double totalMass_local;
628	>	double totalMass;
629	>	double totalMZ_local;
630	>	double totalMZ;
631	>	double massOfCurMol;
632	>	double COM[3];
633	>
634	>	totalMass_local = 0;
635	>	totalMZ_local = 0;
636	>
637	>	for (int i = 0; i < nMols; i++){
638	>	massOfCurMol = molecules[i].getTotalMass();
639	>	molecules[i].getCOM(COM);
640	>
641	>	totalMass_local += massOfCurMol;
642	>	totalMZ_local += massOfCurMol * COM[whichDirection];
643	>	}
644	>
645	>
646	>	#ifdef IS_MPI
647	>	MPI_Allreduce(&totalMass_local, &totalMass, 1, MPI_DOUBLE, MPI_SUM,
648	>	MPI_COMM_WORLD);
649	>	MPI_Allreduce(&totalMZ_local, &totalMZ, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
650	>	#else
651	>	totalMass = totalMass_local;
652	>	totalMZ = totalMZ_local;
653	>	#endif
654	>
655	>	double zsys;
656	>	zsys = totalMZ / totalMass;
657	>
658	>	return zsys;
659	>	}
660	>
661	>	/**
662	>	*
663	>	*/
664	>	template<typename T> void ZConstraint<T>::thermalize(void){
665	>	T::thermalize();
666	>	zeroOutVel();
667	>	}
668	>
669	>	/**
670	>	*
671	>	*/
672	>
673	>	template<typename T> void ZConstraint<T>::zeroOutVel(){
674	>	Atom** fixedZAtoms;
675	>	double COMvel[3];
676	>	double vel[3];
677	>	double zSysCOMVel;
678	>
679	>	//zero out the velocities of center of mass of fixed z-constrained molecules
680	>
681	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
682	>	if (states[i] == zcsFixed){
683	>	zconsMols[i]->getCOMvel(COMvel);
684	>	//cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
685	>
686	>	fixedZAtoms = zconsMols[i]->getMyAtoms();
687	>
688	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
689	>	fixedZAtoms[j]->getVel(vel);
690	>	vel[whichDirection] -= COMvel[whichDirection];
691	>	fixedZAtoms[j]->setVel(vel);
692	>	}
693	>
694	>	zconsMols[i]->getCOMvel(COMvel);
695	>	//cout << "after resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl;
696	>	}
697	>	}
698	>
699	>	//cout << "before resetting the COMVel of moving molecules is " << calcMovingMolsCOMVel() <<endl;
700	>
701	>	zSysCOMVel = calcSysCOMVel();
702	>	#ifdef IS_MPI
703	>	if (worldRank == 0){
704	>	#endif
705	>	//cout << "before resetting the COMVel of sytem is " << zSysCOMVel << endl;
706	>	#ifdef IS_MPI
707	>	}
708	>	#endif
709	>
710	>	// calculate the vz of center of mass of unconstrained molecules and moving z-constrained molecules
711	>	double MVzOfMovingMols_local;
712	>	double MVzOfMovingMols;
713	>	double totalMassOfMovingZMols_local;
714	>	double totalMassOfMovingZMols;
715	>
716	>	MVzOfMovingMols_local = 0;
717	>	totalMassOfMovingZMols_local = 0;
718	>
719	>	for (int i = 0; i < (int) (unconsMols.size()); i++){
720	>	unconsMols[i]->getCOMvel(COMvel);
721	>	MVzOfMovingMols_local += massOfUnconsMols[i] * COMvel[whichDirection];
722	>	}
723	>
724	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
725	>	if (states[i] == zcsMoving){
726	>	zconsMols[i]->getCOMvel(COMvel);
727	>	MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];
728	>	totalMassOfMovingZMols_local += massOfZConsMols[i];
729	>	}
730	>	}
731	>
732	>	#ifndef IS_MPI
733	>	MVzOfMovingMols = MVzOfMovingMols_local;
734	>	totalMassOfMovingZMols = totalMassOfMovingZMols_local;
735	>	#else
736	>	MPI_Allreduce(&MVzOfMovingMols_local, &MVzOfMovingMols, 1, MPI_DOUBLE,
737	>	MPI_SUM, MPI_COMM_WORLD);
738	>	MPI_Allreduce(&totalMassOfMovingZMols_local, &totalMassOfMovingZMols, 1,
739	>	MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
740	>	#endif
741	>
742	>	double vzOfMovingMols;
743	>	vzOfMovingMols = MVzOfMovingMols /
744	>	(totalMassOfUncons + totalMassOfMovingZMols);
745	>
746	>	//modify the velocites of unconstrained molecules
747	>	Atom** unconsAtoms;
748	>	for (int i = 0; i < (int) (unconsMols.size()); i++){
749	>	unconsAtoms = unconsMols[i]->getMyAtoms();
750	>	for (int j = 0; j < unconsMols[i]->getNAtoms(); j++){
751	>	unconsAtoms[j]->getVel(vel);
752	>	vel[whichDirection] -= vzOfMovingMols;
753	>	unconsAtoms[j]->setVel(vel);
754	>	}
755	>	}
756	>
757	>	//modify the velocities of moving z-constrained molecuels
758	>	Atom** movingZAtoms;
759	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
760	>	if (states[i] == zcsMoving){
761	>	movingZAtoms = zconsMols[i]->getMyAtoms();
762	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
763	>	movingZAtoms[j]->getVel(vel);
764	>	vel[whichDirection] -= vzOfMovingMols;
765	>	movingZAtoms[j]->setVel(vel);
766	>	}
767	>	}
768	>	}
769	>
770	>
771	>	zSysCOMVel = calcSysCOMVel();
772	>	#ifdef IS_MPI
773	>	if (worldRank == 0){
774	>	#endif
775	>	//cout << "after resetting the COMVel of moving molecules is " << zSysCOMVel << endl;
776	>	#ifdef IS_MPI
777	>	}
778	>	#endif
779	>	}
780	>
781	>	/**
782	>	*
783	>	*/
784	>
785	>	template<typename T> void ZConstraint<T>::doZconstraintForce(){
786	>	Atom** zconsAtoms;
787	>	double totalFZ;
788	>	double totalFZ_local;
789	>	double COM[3];
790	>	double force[3];
791	>
792	>	//constrain the molecules which do not reach the specified positions
793	>
794	>	//Zero Out the force of z-contrained molecules
795	>	totalFZ_local = 0;
796	>
797	>	//calculate the total z-contrained force of fixed z-contrained molecules
798	>
799	>	//cout << "before zero out z-constraint force on fixed z-constraint molecuels "
800	>	//       << "total force is " << calcTotalForce() << endl;
801	>
802	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
803	>	if (states[i] == zcsFixed){
804	>	zconsMols[i]->getCOM(COM);
805	>	zconsAtoms = zconsMols[i]->getMyAtoms();
806	>
807	>	fz[i] = 0;
808	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
809	>	zconsAtoms[j]->getFrc(force);
810	>	fz[i] += force[whichDirection];
811	>	}
812	>	totalFZ_local += fz[i];
813	>
814	>	//cout << "Fixed Molecule\tindex: " << indexOfZConsMols[i]
815	>	//      <<"\tcurrent zpos: " << COM[whichDirection]
816	>	//      << "\tcurrent fz: " <<fz[i] << endl;
817	>	}
818	>	}
819	>
820	>	//calculate total z-constraint force
821	>	#ifdef IS_MPI
822	>	MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
823	>	#else
824	>	totalFZ = totalFZ_local;
825	>	#endif
826	>
827	>
828	>	// apply negative to fixed z-constrained molecues;
829	>	force[0] = 0;
830	>	force[1] = 0;
831	>	force[2] = 0;
832	>
833	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
834	>	if (states[i] == zcsFixed){
835	>	int nAtomOfCurZConsMol = zconsMols[i]->getNAtoms();
836	>	zconsAtoms = zconsMols[i]->getMyAtoms();
837	>
838	>	for (int j = 0; j < nAtomOfCurZConsMol; j++){
839	>	//force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol;
840	>	force[whichDirection] = -forcePolicy->getZFOfFixedZMols(zconsMols[i],
841	>	zconsAtoms[j],
842	>	fz[i]);
843	>	zconsAtoms[j]->addFrc(force);
844	>	}
845	>	}
846	>	}
847	>
848	>	//cout << "after zero out z-constraint force on fixed z-constraint molecuels "
849	>	//      << "total force is " << calcTotalForce() << endl;
850	>
851	>
852	>	force[0] = 0;
853	>	force[1] = 0;
854	>	force[2] = 0;
855	>
856	>	//modify the forces of unconstrained molecules
857	>	for (int i = 0; i < (int) (unconsMols.size()); i++){
858	>	Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
859	>
860	>	for (int j = 0; j < unconsMols[i]->getNAtoms(); j++){
861	>	//force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
862	>	force[whichDirection] = forcePolicy->getZFOfMovingMols(unconsAtoms[j],
863	>	totalFZ);
864	>	unconsAtoms[j]->addFrc(force);
865	>	}
866	>	}
867	>
868	>	//modify the forces of moving z-constrained molecules
869	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
870	>	if (states[i] == zcsMoving){
871	>	Atom** movingZAtoms = zconsMols[i]->getMyAtoms();
872	>
873	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
874	>	//force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms);
875	>	force[whichDirection] = forcePolicy->getZFOfMovingMols(movingZAtoms[j],
876	>	totalFZ);
877	>	movingZAtoms[j]->addFrc(force);
878	>	}
879	>	}
880	>	}
881	>	//  cout << "after substracting z-constraint force from moving molecuels "
882	>	//        << "total force is " << calcTotalForce()  << endl;
883	>	}
884	>
885	>	/**
886	>	*
887	>	*
888	>	*/
889	>
890	>	template<typename T> void ZConstraint<T>::doHarmonic(){
891	>	double force[3];
892	>	double harmonicU;
893	>	double harmonicF;
894	>	double COM[3];
895	>	double diff;
896	>	double totalFZ_local;
897	>	double totalFZ;
898	>
899	>	force[0] = 0;
900	>	force[1] = 0;
901	>	force[2] = 0;
902	>
903	>	totalFZ_local = 0;
904	>
905	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
906	>	if (states[i] == zcsMoving){
907	>	zconsMols[i]->getCOM(COM);
908	>	//       cout << "Moving Molecule\tindex: " << indexOfZConsMols[i]
909	>	//     << "\tcurrent zpos: " << COM[whichDirection] << endl;
910	>
911	>	diff = COM[whichDirection] - zPos[i];
912	>
913	>	harmonicU = 0.5 * kz[i] * diff * diff;
914	>	info->lrPot += harmonicU;
915	>
916	>	harmonicF = -kz[i] * diff;
917	>	totalFZ_local += harmonicF;
918	>
919	>	//adjust force
920	>
921	>	Atom** movingZAtoms = zconsMols[i]->getMyAtoms();
922	>
923	>	for (int j = 0; j < zconsMols[i]->getNAtoms(); j++){
924	>	//force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms();
925	>	force[whichDirection] = forcePolicy->getHFOfFixedZMols(zconsMols[i],
926	>	movingZAtoms[j],
927	>	harmonicF);
928	>	movingZAtoms[j]->addFrc(force);
929	>	}
930	>	}
931	>	}
932	>
933	>	#ifndef IS_MPI
934	>	totalFZ = totalFZ_local;
935	>	#else
936	>	MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
937	>	#endif
938	>
939	>	//cout << "before substracting harmonic force from moving molecuels "
940	>	//      << "total force is " << calcTotalForce()  << endl;
941	>
942	>	force[0] = 0;
943	>	force[1] = 0;
944	>	force[2] = 0;
945	>
946	>	//modify the forces of unconstrained molecules
947	>	for (int i = 0; i < (int) (unconsMols.size()); i++){
948	>	Atom** unconsAtoms = unconsMols[i]->getMyAtoms();
949	>
950	>	for (int j = 0; j < unconsMols[i]->getNAtoms(); j++){
951	>	//force[whichDirection] = - totalFZ /totNumOfUnconsAtoms;
952	>	force[whichDirection] = -forcePolicy->getHFOfUnconsMols(unconsAtoms[j],
953	>	totalFZ);
954	>	unconsAtoms[j]->addFrc(force);
955	>	}
956	>	}
957	>
958	>	//cout << "after substracting harmonic force from moving molecuels "
959	>	//      << "total force is " << calcTotalForce()  << endl;
960	>	}
961	>
962	>	/**
963	>	*
964	>	*/
965	>
966	>	template<typename T> bool ZConstraint<T>::checkZConsState(){
967	>	double COM[3];
968	>	double diff;
969	>
970	>	int changed_local;
971	>	int changed;
972	>
973	>	changed_local = 0;
974	>
975	>	for (int i = 0; i < (int) (zconsMols.size()); i++){
976	>	zconsMols[i]->getCOM(COM);
977	>	diff = fabs(COM[whichDirection] - zPos[i]);
978	>	if (diff <= zconsTol && states[i] == zcsMoving){
979	>	states[i] = zcsFixed;
980	>	changed_local = 1;
981	>
982	>	if (hasZConsGap)
983	>	endFixTime[i] = info->getTime() + zconsFixTime;
984	>	}
985	>	else if (diff > zconsTol && states[i] == zcsFixed){
986	>	states[i] = zcsMoving;
987	>	changed_local = 1;
988	>
989	>	if (hasZConsGap)
990	>	endFixTime[i] = INFINITE_TIME;
991	>	}
992	>	}
993	>
994	>	#ifndef IS_MPI
995	>	changed = changed_local;
996	>	#else
997	>	MPI_Allreduce(&changed_local, &changed, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
998	>	#endif
999	>
1000	>	return (changed > 0);
1001	>	}
1002	>
1003	>	template<typename T> bool ZConstraint<T>::haveFixedZMols(){
1004	>	int havingFixed_local;
1005	>	int havingFixed;
1006	>
1007	>	havingFixed_local = 0;
1008	>
1009	>	for (int i = 0; i < (int) (zconsMols.size()); i++)
1010	>	if (states[i] == zcsFixed){
1011	>	havingFixed_local = 1;
1012	>	break;
1013	>	}
1014	>
1015	>	#ifndef IS_MPI
1016	>	havingFixed = havingFixed_local;
1017	>	#else
1018	>	MPI_Allreduce(&havingFixed_local, &havingFixed, 1, MPI_INT, MPI_SUM,
1019	>	MPI_COMM_WORLD);
1020	>	#endif
1021	>
1022	>	return (havingFixed > 0);
1023	>	}
1024	>
1025	>
1026	>	/**
1027	>	*
1028	>	*/
1029	>	template<typename T> bool ZConstraint<T>::haveMovingZMols(){
1030	>	int havingMoving_local;
1031	>	int havingMoving;
1032	>
1033	>	havingMoving_local = 0;
1034	>
1035	>	for (int i = 0; i < (int) (zconsMols.size()); i++)
1036	>	if (states[i] == zcsMoving){
1037	>	havingMoving_local = 1;
1038	>	break;
1039	>	}
1040	>
1041	>	#ifndef IS_MPI
1042	>	havingMoving = havingMoving_local;
1043	>	#else
1044	>	MPI_Allreduce(&havingMoving_local, &havingMoving, 1, MPI_INT, MPI_SUM,
1045	>	MPI_COMM_WORLD);
1046	>	#endif
1047	>
1048	>	return (havingMoving > 0);
1049	>	}
1050	>
1051	>	/**
1052	>	*
1053	>	*/
1054	>
1055	>	template<typename T> double ZConstraint<T>::calcMovingMolsCOMVel(){
1056	>	double MVzOfMovingMols_local;
1057	>	double MVzOfMovingMols;
1058	>	double totalMassOfMovingZMols_local;
1059	>	double totalMassOfMovingZMols;
1060	>	double COMvel[3];
1061	>
1062	>	MVzOfMovingMols_local = 0;
1063	>	totalMassOfMovingZMols_local = 0;
1064	>
1065	>	for (int i = 0; i < unconsMols.size(); i++){
1066	>	unconsMols[i]->getCOMvel(COMvel);
1067	>	MVzOfMovingMols_local += massOfUnconsMols[i] * COMvel[whichDirection];
1068	>	}
1069	>
1070	>	for (int i = 0; i < zconsMols.size(); i++){
1071	>	if (states[i] == zcsMoving){
1072	>	zconsMols[i]->getCOMvel(COMvel);
1073	>	MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];
1074	>	totalMassOfMovingZMols_local += massOfZConsMols[i];
1075	>	}
1076	>	}
1077	>
1078	>	#ifndef IS_MPI
1079	>	MVzOfMovingMols = MVzOfMovingMols_local;
1080	>	totalMassOfMovingZMols = totalMassOfMovingZMols_local;
1081	>	#else
1082	>	MPI_Allreduce(&MVzOfMovingMols_local, &MVzOfMovingMols, 1, MPI_DOUBLE,
1083	>	MPI_SUM, MPI_COMM_WORLD);
1084	>	MPI_Allreduce(&totalMassOfMovingZMols_local, &totalMassOfMovingZMols, 1,
1085	>	MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
1086	>	#endif
1087	>
1088	>	double vzOfMovingMols;
1089	>	vzOfMovingMols = MVzOfMovingMols /
1090	>	(totalMassOfUncons + totalMassOfMovingZMols);
1091	>
1092	>	return vzOfMovingMols;
1093	>	}
1094	>
1095	>	/**
1096	>	*
1097	>	*/
1098	>
1099	>	template<typename T> double ZConstraint<T>::calcSysCOMVel(){
1100	>	double COMvel[3];
1101	>	double tempMVz_local;
1102	>	double tempMVz;
1103	>	double massOfZCons_local;
1104	>	double massOfZCons;
1105	>
1106	>
1107	>	tempMVz_local = 0;
1108	>
1109	>	for (int i = 0 ; i < nMols; i++){
1110	>	molecules[i].getCOMvel(COMvel);
1111	>	tempMVz_local += molecules[i].getTotalMass() * COMvel[whichDirection];
1112	>	}
1113	>
1114	>	massOfZCons_local = 0;
1115	>
1116	>	for (int i = 0; i < (int) (massOfZConsMols.size()); i++){
1117	>	massOfZCons_local += massOfZConsMols[i];
1118	>	}
1119	>	#ifndef IS_MPI
1120	>	massOfZCons = massOfZCons_local;
1121	>	tempMVz = tempMVz_local;
1122	>	#else
1123	>	MPI_Allreduce(&massOfZCons_local, &massOfZCons, 1, MPI_DOUBLE, MPI_SUM,
1124	>	MPI_COMM_WORLD);
1125	>	MPI_Allreduce(&tempMVz_local, &tempMVz, 1, MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);
1126	>	#endif
1127	>
1128	>	return tempMVz / (totalMassOfUncons + massOfZCons);
1129	>	}
1130	>
1131	>	/**
1132	>	*
1133	>	*/
1134	>
1135	>	template<typename T> double ZConstraint<T>::calcTotalForce(){
1136	>	double force[3];
1137	>	double totalForce_local;
1138	>	double totalForce;
1139	>
1140	>	totalForce_local = 0;
1141	>
1142	>	for (int i = 0; i < nAtoms; i++){
1143	>	atoms[i]->getFrc(force);
1144	>	totalForce_local += force[whichDirection];
1145	>	}
1146	>
1147	>	#ifndef IS_MPI
1148	>	totalForce = totalForce_local;
1149	>	#else
1150	>	MPI_Allreduce(&totalForce_local, &totalForce, 1, MPI_DOUBLE, MPI_SUM,
1151	>	MPI_COMM_WORLD);
1152	>	#endif
1153	>
1154	>	return totalForce;
1155	>	}
1156	>
1157	>	/**
1158	>	*
1159	>	*/
1160	>
1161	>	template<typename T> void ZConstraint<T>::PolicyByNumber::update(){
1162	>	//calculate the number of atoms of moving z-constrained molecules
1163	>	int nMovingZAtoms_local;
1164	>	int nMovingZAtoms;
1165	>
1166	>	nMovingZAtoms_local = 0;
1167	>	for (int i = 0; i < (int) ((zconsIntegrator->zconsMols).size()); i++)
1168	>	if ((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving)){
1169	>	nMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getNAtoms();
1170	>	}
1171	>
1172	>	#ifdef IS_MPI
1173	>	MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1, MPI_INT, MPI_SUM,
1174	>	MPI_COMM_WORLD);
1175	>	#else
1176	>	nMovingZAtoms = nMovingZAtoms_local;
1177	>	#endif
1178	>	totNumOfMovingAtoms = nMovingZAtoms + zconsIntegrator->totNumOfUnconsAtoms;
1179	>	}
1180	>
1181	>	template<typename T> double ZConstraint<T>::PolicyByNumber::getZFOfFixedZMols(Molecule* mol,
1182	>	Atom* atom,
1183	>	double totalForce){
1184	>	return totalForce / mol->getNAtoms();
1185	>	}
1186	>
1187	>	template<typename T> double ZConstraint<T>::PolicyByNumber::getZFOfMovingMols(Atom* atom,
1188	>	double totalForce){
1189	>	return totalForce / totNumOfMovingAtoms;
1190	>	}
1191	>
1192	>	template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfFixedZMols(Molecule* mol,
1193	>	Atom* atom,
1194	>	double totalForce){
1195	>	return totalForce / mol->getNAtoms();
1196	>	}
1197	>
1198	>	template<typename T> double ZConstraint<T>::PolicyByNumber::getHFOfUnconsMols(Atom* atom,
1199	>	double totalForce){
1200	>	return totalForce / zconsIntegrator->totNumOfUnconsAtoms;
1201	>	}
1202	>
1203	>	/**
1204	>	*
1205	>	*/
1206	>
1207	>	template<typename T> void ZConstraint<T>::PolicyByMass::update(){
1208	>	//calculate the number of atoms of moving z-constrained molecules
1209	>	double massOfMovingZAtoms_local;
1210	>	double massOfMovingZAtoms;
1211	>
1212	>	massOfMovingZAtoms_local = 0;
1213	>	for (int i = 0; i < (int) ((zconsIntegrator->zconsMols).size()); i++)
1214	>	if ((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving)){
1215	>	massOfMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getTotalMass();
1216	>	}
1217	>
1218	>	#ifdef IS_MPI
1219	>	MPI_Allreduce(&massOfMovingZAtoms_local, &massOfMovingZAtoms, 1, MPI_DOUBLE,
1220	>	MPI_SUM, MPI_COMM_WORLD);
1221	>	#else
1222	>	massOfMovingZAtoms = massOfMovingZAtoms_local;
1223	>	#endif
1224	>	totMassOfMovingAtoms = massOfMovingZAtoms +
1225	>	zconsIntegrator->totalMassOfUncons;
1226	>	}
1227	>
1228	>	template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfFixedZMols(Molecule* mol,
1229	>	Atom* atom,
1230	>	double totalForce){
1231	>	return totalForce * atom->getMass() / mol->getTotalMass();
1232	>	}
1233	>
1234	>	template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfMovingMols(Atom* atom,
1235	>	double totalForce){
1236	>	return totalForce * atom->getMass() / totMassOfMovingAtoms;
1237	>	}
1238	>
1239	>	template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfFixedZMols(Molecule* mol,
1240	>	Atom* atom,
1241	>	double totalForce){
1242	>	return totalForce * atom->getMass() / mol->getTotalMass();
1243	>	}
1244	>
1245	>	template<typename T> double ZConstraint<T>::PolicyByMass::getHFOfUnconsMols(Atom* atom,
1246	>	double totalForce){
1247	>	return totalForce * atom->getMass() / zconsIntegrator->totalMassOfUncons;
1248	>	}
1249	>
1250	>	template<typename T> void ZConstraint<T>::updateZPos(){
1251	>	double curTime;
1252	>
1253	>	curTime = info->getTime();
1254	>
1255	>	for (size_t i = 0; i < zconsMols.size(); i++){
1256	>	if (states[i] == zcsFixed && curTime >= endFixTime[i]){
1257	>	zPos[i] += zconsGap;
1258	>	}
1259	>	}
1260	>	}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines