| 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, 10/09/2014 (ivana@ipno.in2p3.fr)
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 G4AnalysisManagerState& state)
43 : G4VRFileManager(state)
44 {
45 // Create helpers defined in the base class
46 fH1RFileManager = std::make_shared<G4RootHnRFileManager<histo::h1d>>(this);
47 fH2RFileManager = std::make_shared<G4RootHnRFileManager<histo::h2d>>(this);
48 fH3RFileManager = std::make_shared<G4RootHnRFileManager<histo::h3d>>(this);
49 fP1RFileManager = std::make_shared<G4RootHnRFileManager<histo::p1d>>(this);
50 fP2RFileManager = std::make_shared<G4RootHnRFileManager<histo::p2d>>(this);
51 }
52
53 //_____________________________________________________________________________
54 G4RootRFileManager::~G4RootRFileManager()
55 {
56 // Delete all open file and their directories
57 for ( auto& mapElement : fRFiles ) {
58 auto rfileTuple = mapElement.second;
59 delete std::get<1>(*rfileTuple); // histo directory
60 delete std::get<2>(*rfileTuple);; // ntuple directory
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 G4String& fileName,
72 G4bool isPerThread)
73 {
74 // Get full file name
75 G4String name = GetFullFileName(fileName, isPerThread);
76
77 Message(kVL4, "open", "read analysis file", name);
78
79 // create new file
80 auto newFile = new tools::rroot::file(G4cout, name);
81 newFile->add_unziper('Z',toolx::decompress_buffer);
82
83 if ( ! newFile->is_open() ) {
84 Warn("Cannot open file " + name, fkClass, "OpenRFile");
85 delete newFile;
86 return false;
87 }
88
89 auto newFileTuple = new G4RootRFile(newFile, nullptr, nullptr);
90
91 // add file in a map and delete the previous file if it exists
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", name);
102
103 return true;
104 }
105
106 //_____________________________________________________________________________
107 G4RootRFile* G4RootRFileManager::GetRFile(const G4String& fileName,
108 G4bool isPerThread) const
109 {
110 // Get full file name
111 G4String name = GetFullFileName(fileName, isPerThread);
112
113 auto it = fRFiles.find(name);
114 if (it != fRFiles.end()) {
115 return it->second;
116 }
117 return nullptr;
118 }
119