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 26 27 // Author: Ivana Hrivnacova, 18/06/2013 (ivan 27 // Author: Ivana Hrivnacova, 18/06/2013 (ivana@ipno.in2p3.fr) 28 28 29 #include "G4CsvNtupleManager.hh" 29 #include "G4CsvNtupleManager.hh" 30 #include "G4CsvFileManager.hh" 30 #include "G4CsvFileManager.hh" 31 #include "G4AnalysisManagerState.hh" 31 #include "G4AnalysisManagerState.hh" 32 #include "G4AnalysisUtilities.hh" 32 #include "G4AnalysisUtilities.hh" 33 33 34 using namespace G4Analysis; 34 using namespace G4Analysis; 35 35 36 // 36 // 37 // utility methods 37 // utility methods 38 // 38 // 39 39 40 //____________________________________________ 40 //_____________________________________________________________________________ 41 G4CsvNtupleManager::G4CsvNtupleManager(const G 41 G4CsvNtupleManager::G4CsvNtupleManager(const G4AnalysisManagerState& state) 42 : G4TNtupleManager<tools::wcsv::ntuple, std:: 42 : G4TNtupleManager<tools::wcsv::ntuple, std::ofstream>(state) 43 {} 43 {} 44 44 45 // 45 // 46 // private methods 46 // private methods 47 // 47 // 48 48 49 //____________________________________________ 49 //_____________________________________________________________________________ 50 void G4CsvNtupleManager::CreateTNtupleFromBook 50 void G4CsvNtupleManager::CreateTNtupleFromBooking( 51 CsvNtupleDescription* ntupleDescription) 51 CsvNtupleDescription* ntupleDescription) 52 { 52 { 53 // create a file for this ntuple 53 // create a file for this ntuple 54 if ( ! fFileManager->CreateNtupleFile(ntuple 54 if ( ! fFileManager->CreateNtupleFile(ntupleDescription) ) return; 55 55 56 // create ntuple 56 // create ntuple 57 ntupleDescription->SetNtuple( 57 ntupleDescription->SetNtuple( 58 new tools::wcsv::ntuple( 58 new tools::wcsv::ntuple( 59 *(ntupleDescription->GetFile()), G4c 59 *(ntupleDescription->GetFile()), G4cerr, ntupleDescription->GetNtupleBooking())); 60 } 60 } 61 61 62 //____________________________________________ 62 //_____________________________________________________________________________ 63 void G4CsvNtupleManager::FinishTNtuple( 63 void G4CsvNtupleManager::FinishTNtuple( 64 CsvNtupleDescription* ntupleDescription, 64 CsvNtupleDescription* ntupleDescription, 65 G4bool /*fromBooking*/) 65 G4bool /*fromBooking*/) 66 { 66 { 67 67 68 // Do nothing if the base file name was not 68 // Do nothing if the base file name was not yet defined 69 if (fFileManager->GetFileName().size() == 0u 69 if (fFileManager->GetFileName().size() == 0u) return; 70 70 71 // Create ntuple from booking 71 // Create ntuple from booking 72 if (ntupleDescription->GetNtuple() == nullpt 72 if (ntupleDescription->GetNtuple() == nullptr) { 73 CreateTNtupleFromBooking(ntupleDescription 73 CreateTNtupleFromBooking(ntupleDescription); 74 } 74 } 75 75 76 // Return if creating ntuple failed 76 // Return if creating ntuple failed 77 if (ntupleDescription->GetNtuple() == nullpt 77 if (ntupleDescription->GetNtuple() == nullptr) { 78 Warn("Creating ntuple has failed.", fkClas 78 Warn("Creating ntuple has failed.", fkClass, "FinishTNtuple"); 79 return; 79 return; 80 } 80 } 81 81 82 // Write header if ntuple already exists 82 // Write header if ntuple already exists 83 if ( ! WriteHeader(ntupleDescription->GetNtu 83 if ( ! WriteHeader(ntupleDescription->GetNtuple()) ) { 84 Warn("Writing ntuple header has failed.", 84 Warn("Writing ntuple header has failed.", fkClass, "FinishTNtuple"); 85 } 85 } 86 } 86 } 87 87 88 //____________________________________________ 88 //_____________________________________________________________________________ 89 G4bool G4CsvNtupleManager::WriteHeader(tools:: 89 G4bool G4CsvNtupleManager::WriteHeader(tools::wcsv::ntuple* ntuple) const 90 { 90 { 91 // Write header if ntuple already exists and i 91 // Write header if ntuple already exists and if this option is activated. 92 // When both Hippo and Commented headers are s 92 // When both Hippo and Commented headers are selected, only Commented 93 // header, which reading is supported. 93 // header, which reading is supported. 94 // Return false only if an error occurred. 94 // Return false only if an error occurred. 95 95 96 if ( fIsCommentedHeader ) { 96 if ( fIsCommentedHeader ) { 97 return ntuple->write_commented_header(G4co 97 return ntuple->write_commented_header(G4cout); 98 } 98 } 99 99 100 // write hippo header (if activated and if n 100 // write hippo header (if activated and if not commented header) 101 if ( fIsHippoHeader ) { 101 if ( fIsHippoHeader ) { 102 ntuple->write_hippo_header(); 102 ntuple->write_hippo_header(); 103 return true; 103 return true; 104 } 104 } 105 105 106 return true; 106 return true; 107 } 107 } 108 108