| Geant4 Cross Reference |
1 // Copyright (C) 2010, Guy Barrand. All rights
2 // See the file tools.license for terms.
3
4 #ifndef toolx_mpi_hmpi
5 #define toolx_mpi_hmpi
6
7 // code to send, receive histos through MPI.
8
9 #include <tools/histo/hmpi>
10 #include <tools/histo/hd2mpi>
11
12 #include <tools/histo/h1d>
13 #include <tools/histo/h2d>
14 #include <tools/histo/h3d>
15 #include <tools/histo/p1d>
16 #include <tools/histo/p2d>
17
18 #include "wrmpi"
19
20 namespace toolx {
21 namespace mpi {
22
23 class hmpi : public virtual tools::histo::hmpi
24 typedef tools::histo::hmpi parent;
25 protected:
26 typedef unsigned int num_t;
27 static const std::string& s_class() {
28 static const std::string s_v("toolx::mpi::
29 return s_v;
30 }
31 public:
32 virtual bool pack(const tools::histo::h1d& a
33 if(!m_wrmpi.spack(a_h.s_cls())) return fal
34 if(!histo_data_duiuid_pack(m_wrmpi,a_h.dac
35 return true;
36 }
37 virtual bool pack(const tools::histo::h2d& a
38 if(!m_wrmpi.spack(a_h.s_cls())) return fal
39 if(!histo_data_duiuid_pack(m_wrmpi,a_h.dac
40 return true;
41 }
42 virtual bool pack(const tools::histo::h3d& a
43 if(!m_wrmpi.spack(a_h.s_cls())) return fal
44 if(!histo_data_duiuid_pack(m_wrmpi,a_h.dac
45 return true;
46 }
47 virtual bool pack(const tools::histo::p1d& a
48 if(!m_wrmpi.spack(a_h.s_cls())) return fal
49 if(!profile_data_duiuidd_pack(m_wrmpi,a_h.
50 return true;
51 }
52 virtual bool pack(const tools::histo::p2d& a
53 if(!m_wrmpi.spack(a_h.s_cls())) return fal
54 if(!profile_data_duiuidd_pack(m_wrmpi,a_h.
55 return true;
56 }
57 public:
58 virtual bool beg_send(unsigned int a_nhist)
59 m_wrmpi.pack_reset();
60 return m_wrmpi.pack(a_nhist);
61 }
62 virtual bool send(int a_dest) {
63 if(::MPI_Send(m_wrmpi.buffer(),m_wrmpi.ipo
64 m_out << "toolx::mpi::hmpi::send : rank
65 return false;
66 }
67 m_wrmpi.pack_reset();
68 return true;
69 }
70 public:
71 virtual bool wait_histos(int a_src,std::vect
72 a_hists.clear();
73
74 typedef std::pair<std::string,void*> class
75
76 MPI_Status status;
77 if(::MPI_Probe(a_src,m_tag,m_comm,&status)
78 m_out << "toolx::mpi::hmpi::wait_histos
79 return false;
80 }
81
82 int buffer_size = 0;
83 if(::MPI_Get_count(&status,MPI_CHAR,&buffe
84 m_out << "toolx::mpi::hmpi::wait_histos
85 return false;
86 }
87
88 if(!buffer_size) {
89 m_out << "exlb::mpi::wait_histos : MPI_G
90 return false;
91 }
92
93 if(m_verbose) m_out << "rank " << m_rank <
94
95 char* buffer = new char[buffer_size];
96 if(!buffer) {
97 m_out << "toolx::mpi::hmpi::wait_histos
98 return false;
99 }
100
101 if(::MPI_Recv(buffer,buffer_size,MPI_CHAR,
102 m_out << "toolx::mpi::hmpi::wait_histos
103 delete [] buffer;
104 return false;
105 }
106
107 if(m_verbose) m_out << "toolx::mpi::hmpi::
108
109 wrmpi _mpi(m_out,m_comm,buffer_size,buffer
110
111 num_t nhist;
112 if(!_mpi.unpack(nhist)) return false;
113
114 if(m_verbose)
115 m_out << "toolx::mpi::hmpi::wait_histos
116
117 {for(num_t ihist=0;ihist<nhist;ihist++) {
118
119 std::string scls;
120 if(!_mpi.sunpack(scls)) return false;
121
122 if(scls==tools::histo::h1d::s_class()) {
123 tools::histo::histo_data<double,unsign
124 if(!histo_data_duiuid_unpack(_mpi,hdat
125 tools::histo::h1d* h = new tools::hist
126 h->copy_from_data(hdata);
127 if(m_verbose) {
128 m_out << "toolx::mpi::hmpi::wait_his
129 << " : got a " << scls
130 << ", title " << h->title()
131 << ", mean_x " << h->mean() <<
132 << std::endl;
133 }
134 a_hists.push_back(class_pointer(h->s_c
135
136 } else if(scls==tools::histo::h2d::s_cla
137 tools::histo::histo_data<double,unsign
138 if(!histo_data_duiuid_unpack(_mpi,hdat
139 tools::histo::h2d* h = new tools::hist
140 h->copy_from_data(hdata);
141 if(m_verbose) {
142 m_out << "toolx::mpi::hmpi::wait_his
143 << " : got a " << scls
144 << ", title " << h->title()
145 << ", mean_x " << h->mean_x()
146 << ", mean_y " << h->mean_y()
147 << std::endl;
148 }
149 a_hists.push_back(class_pointer(h->s_c
150
151 } else if(scls==tools::histo::h3d::s_cla
152 tools::histo::histo_data<double,unsign
153 if(!histo_data_duiuid_unpack(_mpi,hdat
154 tools::histo::h3d* h = new tools::hist
155 h->copy_from_data(hdata);
156 if(m_verbose) {
157 m_out << "toolx::mpi::hmpi::wait_his
158 << " : got a " << scls
159 << ", title " << h->title()
160 << ", mean_x " << h->mean_x()
161 << ", mean_y " << h->mean_y()
162 << ", mean_z " << h->mean_z()
163 << std::endl;
164 }
165 a_hists.push_back(class_pointer(h->s_c
166
167 } else if(scls==tools::histo::p1d::s_cla
168 tools::histo::profile_data<double,unsi
169 if(!profile_data_duiuidd_unpack(_mpi,p
170 tools::histo::p1d* h = new tools::hist
171 h->copy_from_data(pdata);
172 if(m_verbose) {
173 m_out << "toolx::mpi::hmpi::wait_his
174 << " : got a " << scls
175 << ", title " << h->title()
176 << ", mean_x " << h->mean() <<
177 << std::endl;
178 }
179 a_hists.push_back(class_pointer(h->s_c
180
181 } else if(scls==tools::histo::p2d::s_cla
182 tools::histo::profile_data<double,unsi
183 if(!profile_data_duiuidd_unpack(_mpi,p
184 tools::histo::p2d* h = new tools::hist
185 h->copy_from_data(pdata);
186 if(m_verbose) {
187 m_out << "toolx::mpi::hmpi::wait_his
188 << " : got a " << scls
189 << ", title " << h->title()
190 << ", mean_x " << h->mean_x()
191 << ", mean_y " << h->mean_y()
192 << std::endl;
193 }
194 a_hists.push_back(class_pointer(h->s_c
195
196 } else {
197 m_out << "toolx::mpi::hmpi::wait_histo
198 << " : got not treated class " <
199 << std::endl;
200 }
201
202 }} //ihist
203
204 return true;
205 }
206
207 public:
208 virtual int rank() const { return m_rank;}
209 virtual bool comm_rank(int& a_rank) const {
210 if(::MPI_Comm_rank(m_comm,&a_rank)!=MPI_SU
211 return true;
212 }
213 virtual bool comm_size(int& a_size) const {
214 if(::MPI_Comm_size(m_comm,&a_size)!=MPI_SU
215 return true;
216 }
217 public:
218 hmpi(std::ostream& a_out,int a_rank,int a_ta
219 :m_out(a_out)
220 ,m_rank(a_rank)
221 ,m_tag(a_tag)
222 ,m_comm(a_comm)
223 ,m_verbose(a_verbose)
224 ,m_wrmpi(a_out,a_comm)
225 {
226 #ifdef TOOLS_MEM
227 tools::mem::increment(s_class().c_str());
228 #endif
229 }
230 virtual ~hmpi(){
231 #ifdef TOOLS_MEM
232 tools::mem::decrement(s_class().c_str());
233 #endif
234 }
235 protected:
236 hmpi(const hmpi& a_from)
237 :parent(a_from)
238 ,m_out(a_from.m_out)
239 ,m_rank(a_from.m_rank)
240 ,m_tag(a_from.m_tag)
241 ,m_comm(a_from.m_comm)
242 ,m_verbose(a_from.m_verbose)
243 ,m_wrmpi(a_from.m_out,a_from.m_comm)
244 {
245 #ifdef TOOLS_MEM
246 tools::mem::increment(s_class().c_str());
247 #endif
248 }
249 hmpi& operator=(const hmpi& a_from){
250 m_rank = a_from.m_rank;
251 m_tag = a_from.m_tag;
252 m_verbose = a_from.m_verbose;
253 return *this;
254 }
255 protected:
256 std::ostream& m_out;
257 int m_rank;
258 int m_tag;
259 const MPI_Comm& m_comm;
260 bool m_verbose;
261 wrmpi m_wrmpi;
262 };
263
264 }}
265
266 #endif