Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/analysis/hdf5/src/G4Hdf5NtupleManager.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  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, 20/07/2017 (ivana@ipno.in2p3.fr)
 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 G4AnalysisManagerState& state)
 41  : G4TNtupleManager<toolx::hdf5::ntuple, G4Hdf5File>(state)
 42 {}
 43 
 44 //
 45 // private methods
 46 //
 47 //_____________________________________________________________________________
 48 void G4Hdf5NtupleManager::CreateTNtuple(
 49   Hdf5NtupleDescription* ntupleDescription, G4bool warn)
 50 {
 51 // Ntuple will be created from ntupleDescription if file is open,
 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 does not exist.",
 64         fkClass, "CreateTNtuple");
 65     }
 66     return;
 67   }
 68 
 69   auto directory = std::get<2>(*ntupleFile);
 70   auto basketSize = fFileManager->GetBasketSize();
 71   auto compressionLevel = fFileManager->GetCompressionLevel();
 72 
 73   // Update ntuple name if cycle >0
 74   auto ntupleBooking = ntupleDescription->GetNtupleBooking();
 75   if (GetCycle() > 0) {
 76     auto newNtupleName = ntupleDescription->GetNtupleBooking().name();
 77     newNtupleName.append("_v");
 78     newNtupleName.append(std::to_string(GetCycle()));
 79     ntupleBooking.set_name(newNtupleName);
 80   }
 81 
 82   // create ntuple
 83   ntupleDescription->SetNtuple(
 84     new toolx::hdf5::ntuple(
 85           G4cout, directory, ntupleBooking, compressionLevel, basketSize));
 86 }
 87 
 88 //_____________________________________________________________________________
 89 void G4Hdf5NtupleManager::CreateTNtupleFromBooking(
 90   Hdf5NtupleDescription* ntupleDescription)
 91 {
 92   // Create file if file name per object is set
 93   if (ntupleDescription->GetFileName().size() != 0u) {
 94     fFileManager->CreateNtupleFile(ntupleDescription);
 95   }
 96 
 97   // Create ntuple from booking and print a warning if file is not open
 98   CreateTNtuple(ntupleDescription, true);
 99 }
100 
101 //_____________________________________________________________________________
102 void G4Hdf5NtupleManager::FinishTNtuple(
103   Hdf5NtupleDescription* ntupleDescription,
104   G4bool /*fromBooking*/)
105 {
106   if (ntupleDescription->GetNtuple() == nullptr) {
107     // Create ntuple from booking if file is open, do nothing otherwise
108     CreateTNtuple(ntupleDescription, false);
109   }
110 
111   fFileManager->LockDirectoryNames();
112 }
113 
114