| 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, 20/07/2017 (ivana
28
29 #include "G4Hdf5NtupleManager.hh"
30 #include "G4Hdf5FileManager.hh"
31 #include "G4AnalysisManagerState.hh"
32 #include "G4AnalysisUtilities.hh"
33 #include "G4UnitsTable.hh"
34
35 #include "tools/ntuple_booking"
36
37 using namespace G4Analysis;
38
39 //____________________________________________
40 G4Hdf5NtupleManager::G4Hdf5NtupleManager(const
41 : G4TNtupleManager<toolx::hdf5::ntuple, G4Hdf
42 {}
43
44 //
45 // private methods
46 //
47 //____________________________________________
48 void G4Hdf5NtupleManager::CreateTNtuple(
49 Hdf5NtupleDescription* ntupleDescription, G4
50 {
51 // Ntuple will be created from ntupleDescripti
52 // return with or without warning otherwise
53
54 // Get ntuple file from ntuple description
55 auto ntupleFile = ntupleDescription->GetFile
56 if (! ntupleFile) {
57 ntupleFile = fFileManager->GetFile();
58 }
59
60 // Check file
61 if ( ! ntupleFile ) {
62 if (warn) {
63 Warn( "Cannot create ntuple. Ntuple file
64 fkClass, "CreateTNtuple");
65 }
66 return;
67 }
68
69 auto directory = std::get<2>(*ntupleFile);
70 auto basketSize = fFileManager->GetBasketSiz
71 auto compressionLevel = fFileManager->GetCom
72
73 // Update ntuple name if cycle >0
74 auto ntupleBooking = ntupleDescription->GetN
75 if (GetCycle() > 0) {
76 auto newNtupleName = ntupleDescription->Ge
77 newNtupleName.append("_v");
78 newNtupleName.append(std::to_string(GetCyc
79 ntupleBooking.set_name(newNtupleName);
80 }
81
82 // create ntuple
83 ntupleDescription->SetNtuple(
84 new toolx::hdf5::ntuple(
85 G4cout, directory, ntupleBooking, co
86 }
87
88 //____________________________________________
89 void G4Hdf5NtupleManager::CreateTNtupleFromBoo
90 Hdf5NtupleDescription* ntupleDescription)
91 {
92 // Create file if file name per object is se
93 if (ntupleDescription->GetFileName().size()
94 fFileManager->CreateNtupleFile(ntupleDescr
95 }
96
97 // Create ntuple from booking and print a wa
98 CreateTNtuple(ntupleDescription, true);
99 }
100
101 //____________________________________________
102 void G4Hdf5NtupleManager::FinishTNtuple(
103 Hdf5NtupleDescription* ntupleDescription,
104 G4bool /*fromBooking*/)
105 {
106 if (ntupleDescription->GetNtuple() == nullpt
107 // Create ntuple from booking if file is o
108 CreateTNtuple(ntupleDescription, false);
109 }
110
111 fFileManager->LockDirectoryNames();
112 }
113
114