| 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"
35 33
36 #include <tools/impi> 34 #include <tools/impi>
37 35
38 using namespace G4Analysis; <<
39 <<
40 //____________________________________________ <<
41 G4RootMpiAnalysisManager* G4RootMpiAnalysisMan <<
42 { <<
43 return fgInstance; <<
44 } <<
45 <<
46 //____________________________________________ 36 //_____________________________________________________________________________
47 G4RootMpiAnalysisManager::G4RootMpiAnalysisMan << 37 G4RootMpiAnalysisManager::G4RootMpiAnalysisManager(G4bool isMaster)
>> 38 : G4RootAnalysisManager(isMaster)
48 { 39 {
49 fgInstance = this; <<
50 <<
51 // Reset the ntuple file manager 40 // Reset the ntuple file manager
52 fNtupleFileManager.reset(); 41 fNtupleFileManager.reset();
53 42
54 // Ntuple file manager 43 // Ntuple file manager
55 fNtupleFileManager = std::make_shared<G4Root 44 fNtupleFileManager = std::make_shared<G4RootMpiNtupleFileManager>(fState);
56 SetNtupleFileManager(fNtupleFileManager); <<
57 fNtupleFileManager->SetFileManager(fFileMana 45 fNtupleFileManager->SetFileManager(fFileManager);
58 fNtupleFileManager->SetBookingManager(fNtupl 46 fNtupleFileManager->SetBookingManager(fNtupleBookingManager);
59 } 47 }
60 48
61 //____________________________________________ 49 //_____________________________________________________________________________
62 G4RootMpiAnalysisManager::~G4RootMpiAnalysisMa 50 G4RootMpiAnalysisManager::~G4RootMpiAnalysisManager()
63 { << 51 {}
64 fgInstance = 0; <<
65 } <<
66 52
67 // 53 //
68 // public methods 54 // public methods
69 // 55 //
70 56
71 //____________________________________________ 57 //_____________________________________________________________________________
72 void G4RootMpiAnalysisManager::SetMpiNtupleMer << 58 void G4RootMpiAnalysisManager::SetMpiNtupleMerging(tools::impi* impi,
73 << 59 G4int mpiRank, G4int mpiSize,
>> 60 G4int nofNtupleFiles)
74 { 61 {
75 // G4cout << "SetMpiNtupleMerging: " 62 // G4cout << "SetMpiNtupleMerging: "
76 // << impi << ", " 63 // << impi << ", "
77 // << mpiRank << "," 64 // << mpiRank << ","
78 // << mpiSize << "," 65 // << mpiSize << ","
79 // << nofNtupleFiles << G4endl; 66 // << nofNtupleFiles << G4endl;
80 67
81 std::static_pointer_cast<G4RootMpiNtupleFile 68 std::static_pointer_cast<G4RootMpiNtupleFileManager>(fNtupleFileManager)
82 ->SetMpiNtupleMerging(impi, mpiRank, mpiSi 69 ->SetMpiNtupleMerging(impi, mpiRank, mpiSize, nofNtupleFiles);
83 } 70 }
84 71
85 // << 72 //
86 // protected methods 73 // protected methods
87 // 74 //
88 75
89 //____________________________________________ 76 //_____________________________________________________________________________
90 G4bool G4RootMpiAnalysisManager::OpenFileImpl( 77 G4bool G4RootMpiAnalysisManager::OpenFileImpl(const G4String& fileName)
91 { 78 {
92 if (fNtupleFileManager->GetMergeMode() == G4 << 79 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kNone ) {
93 return G4ToolsAnalysisManager::OpenFileImp << 80 return G4RootAnalysisManager::OpenFileImpl(fileName);
94 } 81 }
95 82
96 // Create ntuple manager(s) 83 // Create ntuple manager(s)
97 // and set it to base class which takes then 84 // and set it to base class which takes then their ownership
98 if (!fVNtupleManager) { << 85 if ( ! fVNtupleManager ) {
99 SetNtupleManager(fNtupleFileManager->Creat 86 SetNtupleManager(fNtupleFileManager->CreateNtupleManager());
100 } 87 }
101 88
102 auto result = true; << 89 auto finalResult = true;
103 90
104 // Open file 91 // Open file
105 // In difference from base class a file is o 92 // In difference from base class a file is open also on slave ranks
106 result &= fFileManager->OpenFile(fileName); << 93 auto result = fFileManager->OpenFile(fileName);
>> 94 finalResult = finalResult && result;
107 95
108 // Open ntuple file(s) and create ntuples fr 96 // Open ntuple file(s) and create ntuples from bookings
109 result &= fNtupleFileManager->ActionAtOpenFi << 97 result = fNtupleFileManager->ActionAtOpenFile(fFileManager->GetFullFileName());
>> 98 finalResult = finalResult && result;
110 99
111 return result; << 100 return finalResult;
112 } << 101 }
113 102
114 //____________________________________________ 103 //_____________________________________________________________________________
115 G4bool G4RootMpiAnalysisManager::WriteImpl() << 104 G4bool G4RootMpiAnalysisManager::WriteImpl()
116 { 105 {
117 auto result = true; << 106
>> 107 auto finalResult = true;
118 108
119 // Call base class method 109 // Call base class method
120 result &= G4ToolsAnalysisManager::WriteImpl( << 110 auto result = G4RootAnalysisManager::WriteImpl();
>> 111 finalResult = finalResult && result;
121 112
122 // Write file also on Slave 113 // Write file also on Slave
123 // (skipped in base class) 114 // (skipped in base class)
124 if (fNtupleFileManager->GetMergeMode() == G4 << 115 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kSlave ) {
125 // write all open files 116 // write all open files
126 result &= fFileManager->WriteFiles(); << 117 result = fFileManager->WriteFiles();
>> 118 finalResult = finalResult && result;
127 } 119 }
128 120
129 Message(kVL2, "write", "slave files", "", re << 121 #ifdef G4VERBOSE
>> 122 if ( fState.GetVerboseL2() ) {
>> 123 fState.GetVerboseL2()->Message("write", "slave files", "", finalResult);
>> 124 }
>> 125 #endif
130 126
131 return result; << 127 return finalResult;
132 } 128 }
133 129
134 //____________________________________________ 130 //_____________________________________________________________________________
135 G4bool G4RootMpiAnalysisManager::CloseFileImpl 131 G4bool G4RootMpiAnalysisManager::CloseFileImpl(G4bool reset)
136 { 132 {
137 // Cannot call base class function, as we ne << 133 auto finalResult = true;
138 // on slave; an option in the base class can <<
139 134
140 Message(kVL4, "close", "files"); << 135 // Call base class method
141 << 136 auto result = G4RootAnalysisManager::CloseFileImpl(reset);
142 auto result = true; << 137 finalResult = finalResult && result;
143 if (fVNtupleFileManager) { <<
144 result &= fVNtupleFileManager->ActionAtClo <<
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 138
157 // delete empty files << 139 // Close file also on Slave
158 if (!fVFileManager->DeleteEmptyFiles()) { << 140 // (skipped in base class)
159 Warn("Deleting empty files failed", fkClas << 141 if ( fNtupleFileManager->GetMergeMode() == G4NtupleMergeMode::kSlave ) {
160 result = false; << 142 // close all open files
>> 143 result = fFileManager->CloseFiles();
>> 144 finalResult = finalResult && result;
161 } 145 }
162 146
163 // reset histograms << 147 #ifdef G4VERBOSE
164 if (reset) { << 148 if ( fState.GetVerboseL2() ) {
165 if (!Reset()) { << 149 fState.GetVerboseL2()->Message("close", "slave files", "", finalResult);
166 Warn("Resetting data failed", fkClass, " <<
167 result = false; <<
168 } <<
169 } 150 }
>> 151 #endif
170 152
171 Message(kVL3, "close", "files", "", result); << 153 return finalResult;
172 G4cout << "### Done G4RootMpiAnalysisManager <<
173 154
174 return result; <<
175 } 155 }
176 156