Geant4  10.03.p02
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4RootFileManager.hh
Go to the documentation of this file.
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 // $Id: G4RootFileManager.hh 70604 2013-06-03 11:27:06Z ihrivnac $
27 
28 // The manager for Root output file operations.
29 
30 // Author: Ivana Hrivnacova, 18/06/2013 (ivana@ipno.in2p3.fr)
31 
32 #ifndef G4RootFileManager_h
33 #define G4RootFileManager_h 1
34 
35 #include "G4VFileManager.hh"
36 #include "globals.hh"
37 
38 #include <vector>
39 #include <memory>
40 
41 namespace tools {
42 namespace wroot {
43 class file;
44 class directory;
45 }
46 }
47 
49 {
50  public:
51  explicit G4RootFileManager(const G4AnalysisManagerState& state);
52  virtual ~G4RootFileManager();
53 
54  // Methods to manipulate file
55  virtual G4bool OpenFile(const G4String& fileName) final;
56  virtual G4bool WriteFile() final;
57  virtual G4bool CloseFile() final;
58 
61 
62  // Set methods
63  void SetNofNtupleFiles(G4int nofFiles);
64  void SetBasketSize(unsigned int basketSize);
65 
66  // Get methods
67  tools::wroot::directory* GetHistoDirectory() const;
68  tools::wroot::directory* GetNtupleDirectory() const;
69  std::shared_ptr<tools::wroot::file> GetNtupleFile(G4int index) const;
70  tools::wroot::directory* GetMainNtupleDirectory(G4int index) const;
71  unsigned int GetBasketSize() const;
72 
73  private:
74  // methods
75  G4bool OpenNtupleFiles();
76  G4bool WriteFile(std::shared_ptr<tools::wroot::file> rfile,
77  const G4String& fileName);
78  G4bool CloseFile(std::shared_ptr<tools::wroot::file> rfile,
79  const G4String& fileName);
80 
81  // data members
82  std::shared_ptr<tools::wroot::file> fFile;
83  tools::wroot::directory* fHistoDirectory;
84  tools::wroot::directory* fNtupleDirectory;
85  G4int fNofNtupleFiles;
86  std::vector<std::shared_ptr<tools::wroot::file> > fNtupleFiles;
87  std::vector<tools::wroot::directory*> fMainNtupleDirectories;
88  unsigned int fBasketSize;
89 };
90 
91 // inline functions
92 
94 { fNofNtupleFiles = nofFiles; }
95 
96 inline void G4RootFileManager::SetBasketSize(unsigned int basketSize)
97 { fBasketSize = basketSize; }
98 
99 inline tools::wroot::directory* G4RootFileManager::GetHistoDirectory() const
100 { return fHistoDirectory; }
101 
102 inline tools::wroot::directory* G4RootFileManager::GetNtupleDirectory() const
103 { return fNtupleDirectory; }
104 
105 inline unsigned int G4RootFileManager::GetBasketSize() const
106 { return fBasketSize; }
107 
108 #endif
109 
unsigned int GetBasketSize() const
virtual G4bool CloseFile() final
virtual G4bool WriteFile() final
int G4int
Definition: G4Types.hh:78
std::shared_ptr< tools::wroot::file > GetNtupleFile(G4int index) const
G4RootFileManager(const G4AnalysisManagerState &state)
virtual G4bool OpenFile(const G4String &fileName) final
virtual ~G4RootFileManager()
bool G4bool
Definition: G4Types.hh:79
tools::wroot::directory * GetNtupleDirectory() const
void SetBasketSize(unsigned int basketSize)
tools::wroot::directory * GetHistoDirectory() const
tools::wroot::directory * GetMainNtupleDirectory(G4int index) const
void SetNofNtupleFiles(G4int nofFiles)