Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // >> 26 // $Id: G4CsvNtupleManager.cc 70604 2013-06-03 11:27:06Z ihrivnac $ 26 27 27 // Author: Ivana Hrivnacova, 18/06/2013 (ivan 28 // Author: Ivana Hrivnacova, 18/06/2013 (ivana@ipno.in2p3.fr) 28 29 29 #include "G4CsvNtupleManager.hh" 30 #include "G4CsvNtupleManager.hh" 30 #include "G4CsvFileManager.hh" 31 #include "G4CsvFileManager.hh" 31 #include "G4AnalysisManagerState.hh" 32 #include "G4AnalysisManagerState.hh" 32 #include "G4AnalysisUtilities.hh" 33 #include "G4AnalysisUtilities.hh" 33 34 >> 35 #include "G4Threading.hh" >> 36 >> 37 #include <iostream> >> 38 34 using namespace G4Analysis; 39 using namespace G4Analysis; 35 40 36 // << 41 // 37 // utility methods 42 // utility methods 38 // 43 // 39 44 40 //____________________________________________ 45 //_____________________________________________________________________________ 41 G4CsvNtupleManager::G4CsvNtupleManager(const G 46 G4CsvNtupleManager::G4CsvNtupleManager(const G4AnalysisManagerState& state) 42 : G4TNtupleManager<tools::wcsv::ntuple, std:: << 47 : G4TNtupleManager<tools::wcsv::ntuple>(state), >> 48 fFileManager(nullptr), >> 49 fIsCommentedHeader(true), >> 50 fIsHippoHeader(false) 43 {} 51 {} 44 52 45 // << 53 //_____________________________________________________________________________ >> 54 G4CsvNtupleManager::~G4CsvNtupleManager() >> 55 {} >> 56 >> 57 // 46 // private methods 58 // private methods 47 // 59 // 48 60 49 //____________________________________________ 61 //_____________________________________________________________________________ >> 62 void G4CsvNtupleManager::CreateTNtuple( >> 63 G4TNtupleDescription<tools::wcsv::ntuple>* ntupleDescription, >> 64 const G4String& /*name*/, const G4String& title) >> 65 { >> 66 // Create ntuple if the file is open (what means here that >> 67 // a filename was already set) >> 68 if ( fFileManager->GetFileName().size() ) { >> 69 if ( fFileManager->CreateNtupleFile(ntupleDescription) ) { >> 70 ntupleDescription->fNtuple >> 71 = new tools::wcsv::ntuple(*(ntupleDescription->fFile)); >> 72 // ntuple object is deleted when closing a file >> 73 (ntupleDescription->fNtuple)->set_title(title); >> 74 fNtupleVector.push_back(ntupleDescription->fNtuple); >> 75 } >> 76 } >> 77 } >> 78 >> 79 //_____________________________________________________________________________ 50 void G4CsvNtupleManager::CreateTNtupleFromBook 80 void G4CsvNtupleManager::CreateTNtupleFromBooking( 51 CsvNtupleDescription* ntupleDescription) << 81 G4TNtupleDescription<tools::wcsv::ntuple>* ntupleDescription) 52 { 82 { 53 // create a file for this ntuple << 83 // create a file for this ntuple 54 if ( ! fFileManager->CreateNtupleFile(ntuple << 84 if ( ! fFileManager->CreateNtupleFile(ntupleDescription) ) return; 55 85 56 // create ntuple << 86 // create ntuple 57 ntupleDescription->SetNtuple( << 87 ntupleDescription->fNtuple 58 new tools::wcsv::ntuple( << 88 = new tools::wcsv::ntuple( 59 *(ntupleDescription->GetFile()), G4c << 89 *(ntupleDescription->fFile), G4cerr, ntupleDescription->fNtupleBooking); >> 90 fNtupleVector.push_back(ntupleDescription->fNtuple); 60 } 91 } 61 92 62 //____________________________________________ 93 //_____________________________________________________________________________ 63 void G4CsvNtupleManager::FinishTNtuple( 94 void G4CsvNtupleManager::FinishTNtuple( 64 CsvNtupleDescription* ntupleDescription, << 95 G4TNtupleDescription<tools::wcsv::ntuple>* ntupleDescription) 65 G4bool /*fromBooking*/) << 66 { 96 { 67 << 97 if ( ! ntupleDescription->fNtuple ) return; 68 // Do nothing if the base file name was not << 69 if (fFileManager->GetFileName().size() == 0u << 70 << 71 // Create ntuple from booking << 72 if (ntupleDescription->GetNtuple() == nullpt << 73 CreateTNtupleFromBooking(ntupleDescription << 74 } << 75 << 76 // Return if creating ntuple failed << 77 if (ntupleDescription->GetNtuple() == nullpt << 78 Warn("Creating ntuple has failed.", fkClas << 79 return; << 80 } << 81 98 82 // Write header if ntuple already exists 99 // Write header if ntuple already exists 83 if ( ! WriteHeader(ntupleDescription->GetNtu << 100 if ( ! WriteHeader(ntupleDescription->fNtuple) ) { 84 Warn("Writing ntuple header has failed.", << 101 G4ExceptionDescription description; >> 102 description << " " >> 103 << "Writing ntuple header has failed. "; >> 104 G4Exception("G4CsvNtupleManager::FinishTNtuple()", >> 105 "Analysis_W022", JustWarning, description); 85 } 106 } 86 } 107 } 87 108 88 //____________________________________________ 109 //_____________________________________________________________________________ 89 G4bool G4CsvNtupleManager::WriteHeader(tools:: 110 G4bool G4CsvNtupleManager::WriteHeader(tools::wcsv::ntuple* ntuple) const 90 { 111 { 91 // Write header if ntuple already exists and i 112 // Write header if ntuple already exists and if this option is activated. 92 // When both Hippo and Commented headers are s 113 // When both Hippo and Commented headers are selected, only Commented 93 // header, which reading is supported. 114 // header, which reading is supported. 94 // Return false only if an error occurred. << 115 // Return false only if an error occurred. 95 116 96 if ( fIsCommentedHeader ) { 117 if ( fIsCommentedHeader ) { 97 return ntuple->write_commented_header(G4co 118 return ntuple->write_commented_header(G4cout); 98 } 119 } 99 << 120 100 // write hippo header (if activated and if n 121 // write hippo header (if activated and if not commented header) 101 if ( fIsHippoHeader ) { 122 if ( fIsHippoHeader ) { 102 ntuple->write_hippo_header(); 123 ntuple->write_hippo_header(); 103 return true; 124 return true; 104 } 125 } 105 << 126 106 return true; 127 return true; 107 } 128 } 108 129