Geant4 Cross Reference |
1 // 1 2 // ******************************************* 3 // * License and Disclaimer 4 // * 5 // * The Geant4 software is copyright of th 6 // * the Geant4 Collaboration. It is provided 7 // * conditions of the Geant4 Software License 8 // * LICENSE and available at http://cern.ch/ 9 // * include a list of copyright holders. 10 // * 11 // * Neither the authors of this software syst 12 // * institutes,nor the agencies providing fin 13 // * work make any representation or warran 14 // * regarding this software system or assum 15 // * use. Please see the license in the file 16 // * for the full disclaimer and the limitatio 17 // * 18 // * This code implementation is the result 19 // * technical work of the GEANT4 collaboratio 20 // * By using, copying, modifying or distri 21 // * any work based on the software) you ag 22 // * use in resulting scientific publicati 23 // * acceptance of all terms of the Geant4 Sof 24 // ******************************************* 25 // 26 27 // Author: Ivana Hrivnacova, 10/09/2014 (ivan 28 29 #include "G4RootRFileManager.hh" 30 #include "G4RootRFileDef.hh" 31 #include "G4RootHnRFileManager.hh" 32 #include "G4AnalysisManagerState.hh" 33 #include "G4AnalysisUtilities.hh" 34 35 #include "tools/rroot/file" 36 #include "toolx/zlib" 37 38 using namespace G4Analysis; 39 using namespace tools; 40 41 //____________________________________________ 42 G4RootRFileManager::G4RootRFileManager(const G 43 : G4VRFileManager(state) 44 { 45 // Create helpers defined in the base class 46 fH1RFileManager = std::make_shared<G4RootHnR 47 fH2RFileManager = std::make_shared<G4RootHnR 48 fH3RFileManager = std::make_shared<G4RootHnR 49 fP1RFileManager = std::make_shared<G4RootHnR 50 fP2RFileManager = std::make_shared<G4RootHnR 51 } 52 53 //____________________________________________ 54 G4RootRFileManager::~G4RootRFileManager() 55 { 56 // Delete all open file and their directorie 57 for ( auto& mapElement : fRFiles ) { 58 auto rfileTuple = mapElement.second; 59 delete std::get<1>(*rfileTuple); // histo 60 delete std::get<2>(*rfileTuple);; // ntupl 61 delete std::get<0>(*rfileTuple); // rfile 62 delete rfileTuple; 63 } 64 } 65 66 // 67 // public methods 68 // 69 70 //____________________________________________ 71 G4bool G4RootRFileManager::OpenRFile(const G4S 72 G4bool is 73 { 74 // Get full file name 75 G4String name = GetFullFileName(fileName, is 76 77 Message(kVL4, "open", "read analysis file", 78 79 // create new file 80 auto newFile = new tools::rroot::file(G4cout 81 newFile->add_unziper('Z',toolx::decompress_b 82 83 if ( ! newFile->is_open() ) { 84 Warn("Cannot open file " + name, fkClass, 85 delete newFile; 86 return false; 87 } 88 89 auto newFileTuple = new G4RootRFile(newFile, 90 91 // add file in a map and delete the previous 92 auto it = fRFiles.find(name); 93 if ( it != fRFiles.end() ) { 94 delete it->second; 95 it->second = newFileTuple; 96 } 97 else { 98 fRFiles[name] = newFileTuple; 99 } 100 101 Message(kVL1, "open", "read analysis file", 102 103 return true; 104 } 105 106 //____________________________________________ 107 G4RootRFile* G4RootRFileManager::GetRFile(cons 108 G4bo 109 { 110 // Get full file name 111 G4String name = GetFullFileName(fileName, is 112 113 auto it = fRFiles.find(name); 114 if (it != fRFiles.end()) { 115 return it->second; 116 } 117 return nullptr; 118 } 119