| 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 // Structure containing the information related to Root pntuple.
28 //
29 // Author: Ivana Hrivnacova, 04/10/2016 (ivana@ipno.in2p3.fr)
30
31 #ifndef G4RootPNtupleDescription_h
32 #define G4RootPNtupleDescription_h 1
33
34 #include "G4TNtupleDescription.hh"
35 #include "G4RootFileDef.hh"
36 #include "globals.hh"
37
38 #include "tools/ntuple_booking"
39 #include "tools/wroot/mt_ntuple_row_wise"
40 #include "tools/wroot/mt_ntuple_column_wise"
41
42 #include <fstream>
43
44 namespace tools {
45 namespace wroot {
46 class branch;
47 class ntuple;
48 }
49 }
50
51 using RootNtupleDescription = G4TNtupleDescription<tools::wroot::ntuple, G4RootFile>;
52
53 class G4RootPNtupleDescription
54 {
55 public:
56 G4RootPNtupleDescription(G4NtupleBooking* g4NtupleBooking)
57 : fDescription(g4NtupleBooking) {}
58
59 ~G4RootPNtupleDescription()
60 {
61 if ( fDescription.GetIsNtupleOwner() ) delete fNtuple;
62 }
63
64 // Set methods
65 void SetNtuple(tools::wroot::imt_ntuple* intuple);
66 void SetBasePNtuple(tools::wroot::base_pntuple* basePNtuple);
67 void Reset();
68
69 // Get methods
70 RootNtupleDescription& GetDescription();
71 tools::wroot::imt_ntuple* GetNtuple() const;
72 tools::wroot::base_pntuple* GetBasePNtuple() const;
73 std::vector<tools::wroot::branch*>& GetMainBranches();
74
75 private:
76 RootNtupleDescription fDescription;
77 tools::wroot::imt_ntuple* fNtuple { nullptr };
78 tools::wroot::base_pntuple* fBasePNtuple { nullptr };
79 std::vector<tools::wroot::branch*> fMainBranches;
80 };
81
82 // inline function
83
84 inline void G4RootPNtupleDescription::SetNtuple(
85 tools::wroot::imt_ntuple* intuple)
86 { fNtuple = intuple; }
87
88 inline void G4RootPNtupleDescription::SetBasePNtuple(
89 tools::wroot::base_pntuple* basePNtuple)
90 { fBasePNtuple = basePNtuple; }
91
92 inline void G4RootPNtupleDescription::Reset()
93 {
94 if ( fDescription.GetIsNtupleOwner() ) delete fNtuple;
95 fNtuple = nullptr;
96 }
97
98 inline RootNtupleDescription&
99 G4RootPNtupleDescription::GetDescription()
100 { return fDescription; }
101
102 inline tools::wroot::imt_ntuple*
103 G4RootPNtupleDescription::GetNtuple() const
104 { return fNtuple; }
105
106 inline tools::wroot::base_pntuple*
107 G4RootPNtupleDescription::GetBasePNtuple() const
108 { return fBasePNtuple; }
109
110 inline std::vector<tools::wroot::branch*>&
111 G4RootPNtupleDescription::GetMainBranches()
112 { return fMainBranches; }
113
114 #endif
115