src/io/ZConsWriter.cpp

#include <algorithm>
#include <iostream>
#include <vector>


#include "io/ZConsWriter.hpp"
#include "utils/simError.h"


namespace oopse {
ZConsWriter::ZConsWriter(SimInfo* info, const std::string& filename) : info_(info) {
  //use master - slave mode, only master node writes to disk
#ifdef IS_MPI
    if(worldRank == 0){
#endif

    output_.open(filename.c_str());

    if(!output_){
         sprintf( painCave.errMsg,
                  "Could not open %s for z constrain output_ \n", filename.c_str());
         painCave.isFatal = 1;
         simError();
    }

    output_ << "//time(fs)" << std::endl;
    output_ << "//number of fixed z-constrain molecules" << std::endl;
    output_ << "//global Index of molecule\tzconstrain force\tcurrentZPos" << std::endl;

#ifdef IS_MPI
    }
#endif  

}

ZConsWriter::~ZConsWriter()
{

#ifdef IS_MPI
  if(worldRank == 0 ){
#endif  
  output_.close();  
#ifdef IS_MPI  
  }
#endif
}

void ZConsWriter::writeFZ(const std::list<ZconstraintMol>& fixedZmols){
#ifndef IS_MPI
    output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime() << std::endl;
    output_ << fixedZmols.size() << std::endl;

    std::list<ZconstraintMol>::const_iterator i;
    for ( i = fixedZmols.begin(); i != fixedZmols.end(); ++i) {
        output_ << i->mol->getGlobalIndex() <<"\t" << i->fz << "\t" << i->zpos << "\t" << i->param.zTargetPos <<std::endl;
    }
#else
    int nproc;
    MPI_Comm_size(MPI_COMM_WORLD, &nproc);
    const int masterNode = 0;
    int myNode = worldRank;
    std::vector<int> tmpNFixedZmols(nproc, 0);
    std::vector<int> nFixedZmolsInProc(nproc, 0);
    tmpNFixedZmols[myNode] = fixedZmols.size();
    
    //do MPI_ALLREDUCE to exchange the total number of atoms, rigidbodies and cutoff groups
    MPI_Allreduce(&tmpNFixedZmols[0], &nFixedZmolsInProc[0], nproc, MPI_INT,
                  MPI_SUM, MPI_COMM_WORLD);

    MPI_Status ierr;
    int zmolIndex;
    double data[3];
    
    if (masterNode == 0) {

        std::vector<ZconsData> zconsData;
        ZconsData tmpData;       
        for(int i =0 ; i < nproc; ++i) {
            if (i == masterNode) {
                std::list<ZconstraintMol>::const_iterator j;
                for ( j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
                    tmpData.zmolIndex = j->mol->getGlobalIndex() ;
                    tmpData.zforce= j->fz;
                    tmpData.zpos = j->zpos;
                    tmpData.zconsPos = j->param.zTargetPos;
                    zconsData.push_back(tmpData);
                }                

            } else {
                for(int k =0 ; k < nFixedZmolsInProc[i]; ++k) {
                    MPI_Recv(&zmolIndex, 1, MPI_INT, i, 0, MPI_COMM_WORLD,&ierr);
                    MPI_Recv(data, 3, MPI_DOUBLE, i, 0, MPI_COMM_WORLD,&ierr);
                    tmpData.zmolIndex = zmolIndex;
                    tmpData.zforce= data[0];
                    tmpData.zpos = data[1];
                    tmpData.zconsPos = data[2];
                    zconsData.push_back(tmpData);                                        
                }
            }
            
        }


        output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime() << std::endl;
        output_ << zconsData.size() << std::endl;

        std::vector<ZconsData>::iterator l;
        for (l = zconsData.begin(); l != zconsData.end(); ++l) {
            output_ << l->zmolIndex << "\t" << l->zforce << "\t" << l->zpos << "\t" <<  l->zconsPos << std::endl;
        }
        
    } else {

        std::list<ZconstraintMol>::const_iterator j;
        for (j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
            zmolIndex = j->mol->getGlobalIndex();            
            data[0] = j->fz;
            data[1] = j->zpos;
            data[2] = j->param.zTargetPos;
            MPI_Send(&zmolIndex, 1, MPI_INT, masterNode, 0, MPI_COMM_WORLD);
            MPI_Send(data, 3, MPI_DOUBLE, masterNode, 0, MPI_COMM_WORLD);
            
        }
    }
#endif
}

}
Revision:	1918
Committed:	Tue Jan 11 15:59:58 2005 UTC (20 years, 4 months ago) by tim
File size:	4022 byte(s)
Log Message:	* empty log message *
#	User	Rev	Content
1	gezelter	1490	#include <algorithm>
2			#include <iostream>
3			#include <vector>
4	tim	1883
5
6	tim	1492	#include "io/ZConsWriter.hpp"
7			#include "utils/simError.h"
8	gezelter	1490
9
10	tim	1911	namespace oopse {
11			ZConsWriter::ZConsWriter(SimInfo* info, const std::string& filename) : info_(info) {
12	gezelter	1490	//use master - slave mode, only master node writes to disk
13			#ifdef IS_MPI
14	tim	1911	if(worldRank == 0){
15	gezelter	1490	#endif
16
17	tim	1911	output_.open(filename.c_str());
18	gezelter	1490
19	tim	1911	if(!output_){
20			sprintf( painCave.errMsg,
21			"Could not open %s for z constrain output_ \n", filename.c_str());
22			painCave.isFatal = 1;
23			simError();
24			}
25	gezelter	1490
26	tim	1912	output_ << "//time(fs)" << std::endl;
27			output_ << "//number of fixed z-constrain molecules" << std::endl;
28			output_ << "//global Index of molecule\tzconstrain force\tcurrentZPos" << std::endl;
29	tim	1911
30	gezelter	1490	#ifdef IS_MPI
31	tim	1911	}
32	gezelter	1490	#endif
33
34			}
35
36			ZConsWriter::~ZConsWriter()
37			{
38
39			#ifdef IS_MPI
40			if(worldRank == 0 ){
41			#endif
42	tim	1911	output_.close();
43	gezelter	1490	#ifdef IS_MPI
44			}
45			#endif
46			}
47
48	tim	1911	void ZConsWriter::writeFZ(const std::list<ZconstraintMol>& fixedZmols){
49	gezelter	1490	#ifndef IS_MPI
50	tim	1911	output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime() << std::endl;
51			output_ << fixedZmols.size() << std::endl;
52	gezelter	1490
53	tim	1911	std::list<ZconstraintMol>::const_iterator i;
54			for ( i = fixedZmols.begin(); i != fixedZmols.end(); ++i) {
55			output_ << i->mol->getGlobalIndex() <<"\t" << i->fz << "\t" << i->zpos << "\t" << i->param.zTargetPos <<std::endl;
56			}
57	gezelter	1490	#else
58	tim	1911	int nproc;
59			MPI_Comm_size(MPI_COMM_WORLD, &nproc);
60			const int masterNode = 0;
61			int myNode = worldRank;
62			std::vector<int> tmpNFixedZmols(nproc, 0);
63			std::vector<int> nFixedZmolsInProc(nproc, 0);
64			tmpNFixedZmols[myNode] = fixedZmols.size();
65	gezelter	1490
66	tim	1911	//do MPI_ALLREDUCE to exchange the total number of atoms, rigidbodies and cutoff groups
67			MPI_Allreduce(&tmpNFixedZmols[0], &nFixedZmolsInProc[0], nproc, MPI_INT,
68			MPI_SUM, MPI_COMM_WORLD);
69
70			MPI_Status ierr;
71			int zmolIndex;
72			double data[3];
73	gezelter	1490
74	tim	1911	if (masterNode == 0) {
75
76			std::vector<ZconsData> zconsData;
77			ZconsData tmpData;
78			for(int i =0 ; i < nproc; ++i) {
79			if (i == masterNode) {
80			std::list<ZconstraintMol>::const_iterator j;
81	tim	1918	for ( j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
82	tim	1911	tmpData.zmolIndex = j->mol->getGlobalIndex() ;
83			tmpData.zforce= j->fz;
84			tmpData.zpos = j->zpos;
85			tmpData.zconsPos = j->param.zTargetPos;
86			zconsData.push_back(tmpData);
87			}
88
89			} else {
90			for(int k =0 ; k < nFixedZmolsInProc[i]; ++k) {
91	tim	1918	MPI_Recv(&zmolIndex, 1, MPI_INT, i, 0, MPI_COMM_WORLD,&ierr);
92			MPI_Recv(data, 3, MPI_DOUBLE, i, 0, MPI_COMM_WORLD,&ierr);
93	tim	1911	tmpData.zmolIndex = zmolIndex;
94			tmpData.zforce= data[0];
95			tmpData.zpos = data[1];
96			tmpData.zconsPos = data[2];
97			zconsData.push_back(tmpData);
98			}
99			}
100
101	gezelter	1490	}
102
103
104	tim	1911	output_ << info_->getSnapshotManager()->getCurrentSnapshot()->getTime() << std::endl;
105			output_ << zconsData.size() << std::endl;
106	tim	1883
107	tim	1911	std::vector<ZconsData>::iterator l;
108			for (l = zconsData.begin(); l != zconsData.end(); ++l) {
109			output_ << l->zmolIndex << "\t" << l->zforce << "\t" << l->zpos << "\t" << l->zconsPos << std::endl;
110			}
111
112			} else {
113	tim	1883
114	tim	1911	std::list<ZconstraintMol>::const_iterator j;
115			for (j = fixedZmols.begin(); j != fixedZmols.end(); ++j) {
116			zmolIndex = j->mol->getGlobalIndex();
117			data[0] = j->fz;
118			data[1] = j->zpos;
119			data[2] = j->param.zTargetPos;
120	tim	1918	MPI_Send(&zmolIndex, 1, MPI_INT, masterNode, 0, MPI_COMM_WORLD);
121			MPI_Send(data, 3, MPI_DOUBLE, masterNode, 0, MPI_COMM_WORLD);
122	tim	1911
123			}
124	gezelter	1490	}
125			#endif
126			}
127
128			}