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