| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer 3 // * License and Disclaimer *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. 9 // * include a list of copyright holders. *
10 // * 10 // * *
11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. *
17 // * 17 // * *
18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. *
24 // ******************************************* 24 // ********************************************************************
25 // 25 //
26 26
27 // Author: Ivana Hrivnacova, 21/11/2018 (ivana 27 // Author: Ivana Hrivnacova, 21/11/2018 (ivana@ipno.in2p3.fr)
28 28
29 #include "G4RootMpiAnalysisManager.hh" 29 #include "G4RootMpiAnalysisManager.hh"
30 <<
31 #include "G4AnalysisUtilities.hh" <<
32 #include "G4RootMpiNtupleFileManager.hh" 30 #include "G4RootMpiNtupleFileManager.hh"
33 #include "G4RootMpiNtupleManager.hh" 31 #include "G4RootMpiNtupleManager.hh"
34 #include "G4RootMpiPNtupleManager.hh" 32 #include "G4RootMpiPNtupleManager.hh"
>> 33 #include "G4AnalysisUtilities.hh"
35 34
36 #include <tools/impi> 35 #include <tools/impi>
37 36
38 using namespace G4Analysis; 37 using namespace G4Analysis;
39 38
40 //____________________________________________ 39 //_____________________________________________________________________________
41 G4RootMpiAnalysisManager* G4RootMpiAnalysisMan 40 G4RootMpiAnalysisManager* G4RootMpiAnalysisManager::Instance()
42 { 41 {
43 return fgInstance; 42 return fgInstance;
44 } 43 }
45 44
46 //____________________________________________ 45 //_____________________________________________________________________________
47 G4RootMpiAnalysisManager::G4RootMpiAnalysisMan << 46 G4RootMpiAnalysisManager::G4RootMpiAnalysisManager(G4bool /*isMaster*/)
>> 47 : G4RootAnalysisManager()
48 { 48 {
49 fgInstance = this; 49 fgInstance = this;
50 50
51 // Reset the ntuple file manager 51 // Reset the ntuple file manager
52 fNtupleFileManager.reset(); 52 fNtupleFileManager.reset();
53 53
54 // Ntuple file manager 54 // Ntuple file manager
55 fNtupleFileManager = std::make_shared<G4Root 55 fNtupleFileManager = std::make_shared<G4RootMpiNtupleFileManager>(fState);
56 SetNtupleFileManager(fNtupleFileManager); <<
57 fNtupleFileManager->SetFileManager(fFileMana 56 fNtupleFileManager->SetFileManager(fFileManager);
58 fNtupleFileManager->SetBookingManager(fNtupl 57 fNtupleFileManager->SetBookingManager(fNtupleBookingManager);
59 } 58 }
60 59
61 //____________________________________________ 60 //_____________________________________________________________________________
62 G4RootMpiAnalysisManager::~G4RootMpiAnalysisMa 61 G4RootMpiAnalysisManager::~G4RootMpiAnalysisManager()
63 { 62 {
64 fgInstance = 0; 63 fgInstance = 0;
65 } 64 }
66 65
67 // 66 //
68 // public methods 67 // public methods
69 // 68 //
70 69
71 //____________________________________________ 70 //_____________________________________________________________________________
72 void G4RootMpiAnalysisManager::SetMpiNtupleMer << 71 void G4RootMpiAnalysisManager::SetMpiNtupleMerging(tools::impi* impi,
73 << 72 G4int mpiRank, G4int mpiSize,
>> 73 G4int nofNtupleFiles)
74 { 74 {
75 // G4cout << "SetMpiNtupleMerging: " 75 // G4cout << "SetMpiNtupleMerging: "
76 // << impi << ", " 76 // << impi << ", "
77 // << mpiRank << "," 77 // << mpiRank << ","
78 // << mpiSize << "," 78 // << mpiSize << ","
79 // << nofNtupleFiles << G4endl; 79 // << nofNtupleFiles << G4endl;
80 80
81 std::static_pointer_cast<G4RootMpiNtupleFile 81 std::static_pointer_cast<G4RootMpiNtupleFileManager>(fNtupleFileManager)
82 ->SetMpiNtupleMerging(impi, mpiRank, mpiSi 82 ->SetMpiNtupleMerging(impi, mpiRank, mpiSize, nofNtupleFiles);
83 } 83 }
84 84
85 // << 85 //
86 // protected methods 86 // protected methods
87 // 87 //
88 88
89 //____________________________________________ 89 //_____________________________________________________________________________
90 G4bool G4RootMpiAnalysisManager::OpenFileImpl( 90 G4bool G4RootMpiAnalysisManager::OpenFileImpl(const G4String& fileName)
91 { 91 {
92 if (fNtupleFileManager->GetMergeMode() == G4 << 92 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kNone ) {
93 return G4ToolsAnalysisManager::OpenFileImp << 93 return G4RootAnalysisManager::OpenFileImpl(fileName);
94 } 94 }
95 95
96 // Create ntuple manager(s) 96 // Create ntuple manager(s)
97 // and set it to base class which takes then 97 // and set it to base class which takes then their ownership
98 if (!fVNtupleManager) { << 98 if ( ! fVNtupleManager ) {
99 SetNtupleManager(fNtupleFileManager->Creat 99 SetNtupleManager(fNtupleFileManager->CreateNtupleManager());
100 } 100 }
101 101
102 auto result = true; << 102 auto finalResult = true;
103 103
104 // Open file 104 // Open file
105 // In difference from base class a file is o 105 // In difference from base class a file is open also on slave ranks
106 result &= fFileManager->OpenFile(fileName); << 106 auto result = fFileManager->OpenFile(fileName);
>> 107 finalResult = finalResult && result;
107 108
108 // Open ntuple file(s) and create ntuples fr 109 // Open ntuple file(s) and create ntuples from bookings
109 result &= fNtupleFileManager->ActionAtOpenFi << 110 result = fNtupleFileManager->ActionAtOpenFile(fFileManager->GetFullFileName());
>> 111 finalResult = finalResult && result;
110 112
111 return result; << 113 return finalResult;
112 } << 114 }
113 115
114 //____________________________________________ 116 //_____________________________________________________________________________
115 G4bool G4RootMpiAnalysisManager::WriteImpl() << 117 G4bool G4RootMpiAnalysisManager::WriteImpl()
116 { 118 {
117 auto result = true; << 119
>> 120 auto finalResult = true;
118 121
119 // Call base class method 122 // Call base class method
120 result &= G4ToolsAnalysisManager::WriteImpl( << 123 auto result = G4RootAnalysisManager::WriteImpl();
>> 124 finalResult = finalResult && result;
121 125
122 // Write file also on Slave 126 // Write file also on Slave
123 // (skipped in base class) 127 // (skipped in base class)
124 if (fNtupleFileManager->GetMergeMode() == G4 << 128 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kSlave ) {
125 // write all open files 129 // write all open files
126 result &= fFileManager->WriteFiles(); << 130 result = fFileManager->WriteFiles();
>> 131 finalResult = finalResult && result;
127 } 132 }
128 133
129 Message(kVL2, "write", "slave files", "", re << 134 Message(kVL2, "write", "slave files", "", finalResult);
130 135
131 return result; << 136 return finalResult;
132 } 137 }
133 138
134 //____________________________________________ 139 //_____________________________________________________________________________
135 G4bool G4RootMpiAnalysisManager::CloseFileImpl 140 G4bool G4RootMpiAnalysisManager::CloseFileImpl(G4bool reset)
136 { 141 {
137 // Cannot call base class function, as we ne << 142 auto finalResult = true;
138 // on slave; an option in the base class can <<
139 <<
140 Message(kVL4, "close", "files"); <<
141 143
142 auto result = true; << 144 // Call base class method
143 if (fVNtupleFileManager) { << 145 auto result = G4RootAnalysisManager::CloseFileImpl(reset);
144 result &= fVNtupleFileManager->ActionAtClo << 146 finalResult = finalResult && result;
145 } <<
146 <<
147 // close file also on Slave <<
148 // - the conditoon used in the base class is <<
149 // if ( (fVNtupleFileManager == nullptr) || <<
150 // (fVNtupleFileManager->GetMergeMode() <<
151 if (!fVFileManager->CloseFiles()) { <<
152 Warn("Closing files failed", fkClass, "Clo <<
153 result = false; <<
154 } <<
155 // } <<
156 147
157 // delete empty files << 148 // Close file also on Slave
158 if (!fVFileManager->DeleteEmptyFiles()) { << 149 // (skipped in base class)
159 Warn("Deleting empty files failed", fkClas << 150 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kSlave ) {
160 result = false; << 151 // close all open files
>> 152 result = fFileManager->CloseFiles();
>> 153 finalResult = finalResult && result;
161 } 154 }
162 155
163 // reset histograms << 156 Message(kVL2, "close", "slave files", "", finalResult);
164 if (reset) { <<
165 if (!Reset()) { <<
166 Warn("Resetting data failed", fkClass, " <<
167 result = false; <<
168 } <<
169 } <<
170 157
171 Message(kVL3, "close", "files", "", result); << 158 return finalResult;
172 G4cout << "### Done G4RootMpiAnalysisManager <<
173 159
174 return result; <<
175 } 160 }
176 161