ViewVC Help

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

root/OpenMD/trunk/src/io/ZConsWriter.cpp

Comparing trunk/src/io/ZConsWriter.cpp (file contents):
Revision 2 by gezelter, Fri Sep 24 04:16:43 2004 UTC vs.
Revision 2071 by gezelter, Sat Mar 7 21:41:51 2015 UTC

#	Line 1 | Line 1
1	+	/*
2	+	* Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3	+	*
4	+	* The University of Notre Dame grants you ("Licensee") a
5	+	* non-exclusive, royalty free, license to use, modify and
6	+	* redistribute this software in source and binary code form, provided
7	+	* that the following conditions are met:
8	+	*
9	+	* 1. Redistributions of source code must retain the above copyright
10	+	*    notice, this list of conditions and the following disclaimer.
11	+	*
12	+	* 2. Redistributions in binary form must reproduce the above copyright
13	+	*    notice, this list of conditions and the following disclaimer in the
14	+	*    documentation and/or other materials provided with the
15	+	*    distribution.
16	+	*
17	+	* This software is provided "AS IS," without a warranty of any
18	+	* kind. All express or implied conditions, representations and
19	+	* warranties, including any implied warranty of merchantability,
20	+	* fitness for a particular purpose or non-infringement, are hereby
21	+	* excluded.  The University of Notre Dame and its licensors shall not
22	+	* be liable for any damages suffered by licensee as a result of
23	+	* using, modifying or distributing the software or its
24	+	* derivatives. In no event will the University of Notre Dame or its
25	+	* licensors be liable for any lost revenue, profit or data, or for
26	+	* direct, indirect, special, consequential, incidental or punitive
27	+	* damages, however caused and regardless of the theory of liability,
28	+	* arising out of the use of or inability to use software, even if the
29	+	* University of Notre Dame has been advised of the possibility of
30	+	* such damages.
31	+	*
32	+	* SUPPORT OPEN SCIENCE!  If you use OpenMD or its source code in your
33	+	* research, please cite the appropriate papers when you publish your
34	+	* work.  Good starting points are:
35	+	*
36	+	* [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
37	+	* [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
38	+	* [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).
39	+	* [4]  Kuang & Gezelter,  J. Chem. Phys. 133, 164101 (2010).
40	+	* [5]  Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41	+	*/
42	+
43	+	#ifdef IS_MPI
44	+	#include <mpi.h>
45	+	#endif
46	+
47		#include <algorithm>
48		#include <iostream>
49		#include <vector>
4	–	//#include <pair>
5	–	#include "ZConsWriter.hpp"
6	–	#include "simError.h"
50
51	<	using namespace std;
51	>	#include "io/ZConsWriter.hpp"
52	>	#include "utils/simError.h"
53
54	<	ZConsWriter::ZConsWriter(const char* filename, vector<ZConsParaItem>* thePara)
55	<	{
56	<	//use master - slave mode, only master node writes to disk
54	>	namespace OpenMD {
55	>	ZConsWriter::ZConsWriter(SimInfo* info, const std::string& filename) :
56	>	info_(info) {
57	>	// use master - slave mode, only master node writes to disk
58		#ifdef IS_MPI
59	<	if(worldRank == 0){
59	>	if(worldRank == 0){
60		#endif
61	+	output_.open(filename.c_str());
62
63	<	output.open(filename);
64	<
65	<	if(!output){
66	<	sprintf( painCave.errMsg,
67	<	"Could not open %s for z constrain output \n",
68	<	filename);
69	<	painCave.isFatal = 1;
24	<	simError();
25	<	}
26	<	output << "#number of z constrain molecules" << endl;
27	<	output << "#global Index of molecule\tzPos" << endl;
28	<	output << "#every frame will contain below data" <<endl;
29	<	output << "#time(fs)" << endl;
30	<	output << "#number of fixed z-constrain molecules" << endl;
31	<	output << "#global Index of molecule\tzconstrain force\tcurrentZPos" << endl;
63	>	if(!output_){
64	>	sprintf( painCave.errMsg,
65	>	"Could not open %s for z constrain output_ \n",
66	>	filename.c_str());
67	>	painCave.isFatal = 1;
68	>	simError();
69	>	}
70
71	<	parameters = thePara;
72	<	writeZPos();
73	<
71	>	output_ << "//time(fs)" << std::endl;
72	>	output_ << "//number of fixed z-constrain molecules" << std::endl;
73	>	output_ << "//global Index of molecule\tzconstrain force\tcurrentZPos"
74	>	<< std::endl;
75		#ifdef IS_MPI
76	+	}
77	+	#endif
78		}
79	<	#endif
80	<
81	<	}
82	<
42	<	ZConsWriter::~ZConsWriter()
43	<	{
44	<
79	>
80	>	ZConsWriter::~ZConsWriter()
81	>	{
82	>
83		#ifdef IS_MPI
84	<	if(worldRank == 0 ){
84	>	if(worldRank == 0 ){
85		#endif
86	<	output.close();
86	>	output_.close();
87		#ifdef IS_MPI
88	<	}
88	>	}
89		#endif
90	<	}
90	>	}
91
92	<	/**
55	<	*
56	<	*/
57	<	void ZConsWriter::writeFZ(double time, int num, int* index, double* fz, double* curZPos, double* zpos){
58	<
92	>	void ZConsWriter::writeFZ(const std::list<ZconstraintMol>& fixedZmols){
93		#ifndef IS_MPI
94	<	output << time << endl;
95	<	output << num << endl;
96	<
63	<	for(int i = 0; i < num; i++)
64	<	output << index[i] <<"\t" << fz[i] << "\t" << curZPos[i] << "\t" << zpos[i] <<endl;
94	>	output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime()
95	>	<< std::endl;
96	>	output_ << fixedZmols.size() << std::endl;
97
98	+	std::list<ZconstraintMol>::const_iterator i;
99	+	for ( i = fixedZmols.begin(); i != fixedZmols.end(); ++i) {
100	+	output_ << i->mol->getGlobalIndex() <<"\t" << i->fz << "\t" << i->zpos
101	+	<< "\t" << i->param.zTargetPos <<std::endl;
102	+	}
103		#else
67	–	int totalNum;
68	–	MPI_Allreduce(&num, &totalNum, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD);
69	–
70	–	if(worldRank == 0){
71	–	output << time << endl;
72	–	output << totalNum << endl;
73	–	}
74	–
75	–	int whichNode;
76	–	enum CommType { RequesPosAndForce, EndOfRequest} status;
77	–	double pos;
78	–	double force;
79	–	double zconsPos;
80	–	int localIndex;
81	–	MPI_Status ierr;
82	–	int tag = 0;
83	–
84	–	if(worldRank == 0){
104
105	<	int globalIndexOfCurMol;
106	<	int *MolToProcMap;
107	<	MolToProcMap = mpiSim->getMolToProcMap();
105	>	const int masterNode = 0;
106	>	int nproc;
107	>	int myNode;
108	>	MPI_Comm_size( MPI_COMM_WORLD, &nproc);
109	>	MPI_Comm_rank( MPI_COMM_WORLD, &myNode);
110
111	<	for(int i = 0; i < (int)(parameters->size()); i++){
111	>	std::vector<int> tmpNFixedZmols(nproc, 0);
112	>	std::vector<int> nFixedZmolsInProc(nproc, 0);
113	>	tmpNFixedZmols[myNode] = fixedZmols.size();
114	>
115	>	//do MPI_ALLREDUCE to exchange the total number of atoms,
116	>	//rigidbodies and cutoff groups
117	>	MPI_Allreduce(&tmpNFixedZmols[0], &nFixedZmolsInProc[0],
118	>	nproc, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
119	>
120	>	MPI_Status* ierr = NULL;
121	>	int zmolIndex;
122	>	RealType data[3];
123	>
124	>	if (myNode == masterNode) {
125
126	<	globalIndexOfCurMol = (*parameters)[i].zconsIndex;
127	<	whichNode = MolToProcMap[globalIndexOfCurMol];
128	<
129	<	if(whichNode == 0){
130	<
131	<	for(int j = 0; j < num; j++)
132	<	if(index[j] == globalIndexOfCurMol){
133	<	localIndex = j;
134	<	break;
135	<	}
136	<
137	<	force = fz[localIndex];
138	<	pos = curZPos[localIndex];
139	<
126	>	std::vector<ZconsData> zconsData;
127	>	ZconsData tmpData;
128	>	for(int i =0 ; i < nproc; ++i) {
129	>	if (i == masterNode) {
130	>	std::list<ZconstraintMol>::const_iterator j;
131	>	for ( j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
132	>	tmpData.zmolIndex = j->mol->getGlobalIndex() ;
133	>	tmpData.zforce= j->fz;
134	>	tmpData.zpos = j->zpos;
135	>	tmpData.zconsPos = j->param.zTargetPos;
136	>	zconsData.push_back(tmpData);
137	>	}
138	>
139	>	} else {
140	>	for(int k =0 ; k < nFixedZmolsInProc[i]; ++k) {
141	>	MPI_Recv(&zmolIndex, 1, MPI_INT, i, 0, MPI_COMM_WORLD, ierr);
142	>	MPI_Recv(data, 3, MPI_REALTYPE, i, 0, MPI_COMM_WORLD, ierr);
143	>	tmpData.zmolIndex = zmolIndex;
144	>	tmpData.zforce= data[0];
145	>	tmpData.zpos = data[1];
146	>	tmpData.zconsPos = data[2];
147	>	zconsData.push_back(tmpData);
148	>	}
149	>	}
150		}
107	–	else{
108	–	status = RequesPosAndForce;
109	–	MPI_Send(&status, 1, MPI_INT, whichNode, tag, MPI_COMM_WORLD);
110	–	MPI_Send(&globalIndexOfCurMol, 1, MPI_INT, whichNode, tag, MPI_COMM_WORLD);
111	–	MPI_Recv(&force, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr);
112	–	MPI_Recv(&pos, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr);
113	–	MPI_Recv(&zconsPos, 1, MPI_DOUBLE, whichNode, tag, MPI_COMM_WORLD, &ierr);
114	–	}
151
152	<	output << globalIndexOfCurMol << "\t" << force << "\t" << pos << "\t"<<  zconsPos << endl;
153	<
154	<	} //End of Request Loop
119	<
120	<	//Send ending request message to slave nodes
121	<	status = EndOfRequest;
122	<	for(int i =1; i < mpiSim->getNProcessors(); i++)
123	<	MPI_Send(&status, 1, MPI_INT, i, tag, MPI_COMM_WORLD);
124	<
125	<	}
126	<	else{
127	<
128	<	int whichMol;
129	<	bool done = false;
152	>	output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime()
153	>	<< std::endl;
154	>	output_ << zconsData.size() << std::endl;
155
156	<	while (!done){
157	<
158	<	MPI_Recv(&status, 1, MPI_INT, 0, tag, MPI_COMM_WORLD, &ierr);
159	<
135	<	switch (status){
136	<
137	<	case RequesPosAndForce :
138	<
139	<	MPI_Recv(&whichMol, 1, MPI_INT, 0, tag, MPI_COMM_WORLD,&ierr);
140	<
141	<	for(int i = 0; i < num; i++)
142	<	if(index[i] == whichMol){
143	<	localIndex = i;
144	<	break;
145	<	}
146	<
147	<	MPI_Send(&fz[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD);
148	<	MPI_Send(&curZPos[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD);
149	<	MPI_Send(&zpos[localIndex], 1, MPI_DOUBLE, 0, tag, MPI_COMM_WORLD);
150	<	break;
151	<
152	<	case EndOfRequest :
153	<
154	<	done = true;
155	<	break;
156	>	std::vector<ZconsData>::iterator l;
157	>	for (l = zconsData.begin(); l != zconsData.end(); ++l) {
158	>	output_ << l->zmolIndex << "\t" << l->zforce << "\t" << l->zpos
159	>	<< "\t" <<  l->zconsPos << std::endl;
160		}
161
162	+	} else {
163	+
164	+	std::list<ZconstraintMol>::const_iterator j;
165	+	for (j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
166	+	zmolIndex = j->mol->getGlobalIndex();
167	+	data[0] = j->fz;
168	+	data[1] = j->zpos;
169	+	data[2] = j->param.zTargetPos;
170	+	MPI_Send(&zmolIndex, 1, MPI_INT, masterNode, 0, MPI_COMM_WORLD);
171	+	MPI_Send(data, 3, MPI_REALTYPE, masterNode, 0, MPI_COMM_WORLD);
172	+	}
173		}
159	–
160	–	}
161	–
174		#endif
163	–
164	–	}
165	–
166	–	/*
167	–	*
168	–	*/
169	–	void ZConsWriter::writeZPos(){
170	–
171	–	#ifdef IS_MPI
172	–	if(worldRank == 0){
173	–	#endif
174	–
175	–	output << parameters->size() << endl;
176	–
177	–	for(int i =0 ; i < (int)(parameters->size()); i++)
178	–	output << (*parameters)[i].zconsIndex << "\t" <<  (*parameters)[i].zPos << endl;
179	–
180	–	#ifdef IS_MPI
175		}
182	–	#endif
176		}

Comparing trunk/src/io/ZConsWriter.cpp (property svn:keywords):
Revision 2 by gezelter, Fri Sep 24 04:16:43 2004 UTC vs.
Revision 2071 by gezelter, Sat Mar 7 21:41:51 2015 UTC

#	Line 0 | Line 1
1	+	Author Id Revision Date

Diff Legend

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