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