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: G4VFileManager.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 "G4VFileManager.hh" 30 #include "G4VFileManager.hh" 30 #include "G4AnalysisManagerState.hh" 31 #include "G4AnalysisManagerState.hh" 31 #include "G4AnalysisUtilities.hh" << 32 32 33 using namespace G4Analysis; << 33 #include "G4Threading.hh" 34 34 35 //____________________________________________ 35 //_____________________________________________________________________________ 36 G4VFileManager::G4VFileManager(const G4Analysi 36 G4VFileManager::G4VFileManager(const G4AnalysisManagerState& state) 37 : G4BaseFileManager(state) << 37 : fState(state), 38 {} << 38 fFileName(""), >> 39 fHistoDirectoryName(""), >> 40 fNtupleDirectoryName(""), >> 41 fLockFileName(false), >> 42 fLockHistoDirectoryName(false), >> 43 fLockNtupleDirectoryName(false) >> 44 { >> 45 } 39 46 40 // << 47 //_____________________________________________________________________________ >> 48 G4VFileManager::~G4VFileManager() >> 49 { >> 50 } >> 51 >> 52 // 41 // public methods 53 // public methods 42 // 54 // 43 55 44 //____________________________________________ 56 //_____________________________________________________________________________ 45 G4bool G4VFileManager::SetFileName(const G4Str << 57 G4bool G4VFileManager::SetFileName(const G4String& fileName) 46 { 58 { 47 // Check extension << 59 if ( fLockFileName ) { 48 auto name = fileName; << 60 G4ExceptionDescription description; 49 auto extension = G4Analysis::GetExtension(fi << 61 description 50 if ((extension.size() != 0u) && (GetFileType << 62 << "Cannot set File name as its value was already used."; 51 // replace extension << 63 G4Exception("G4VFileManager::SetFileName()", 52 name = G4Analysis::GetBaseName(fileName) + << 64 "Analysis_W009", JustWarning, description); 53 Warn(fileName + " file extension is not va << 65 return false; 54 name + " will be used.", fkClass, "Se << 66 } 55 } << 56 67 57 return G4BaseFileManager::SetFileName(name); << 68 fFileName = fileName; 58 } << 69 return true; >> 70 } 59 71 60 //____________________________________________ 72 //_____________________________________________________________________________ 61 G4bool G4VFileManager::SetHistoDirectoryName(c << 73 G4bool G4VFileManager::SetHistoDirectoryName(const G4String& dirName) 62 { 74 { 63 if ( fLockDirectoryNames ) { << 75 if ( fLockHistoDirectoryName ) { 64 Warn("Cannot set Histo directory name as i << 76 G4ExceptionDescription description; 65 fkClass, "SetHistoDirectoryName"); << 77 description >> 78 << "Cannot set Histo directory name as its value was already used."; >> 79 G4Exception("G4VFileManager::SetHistoDirectoryName()", >> 80 "Analysis_W009", JustWarning, description); 66 return false; 81 return false; 67 } << 82 } 68 83 69 fHistoDirectoryName = dirName; 84 fHistoDirectoryName = dirName; 70 return true; 85 return true; 71 } << 86 } 72 87 73 //____________________________________________ 88 //_____________________________________________________________________________ 74 G4bool G4VFileManager::SetNtupleDirectoryName( << 89 G4bool G4VFileManager::SetNtupleDirectoryName(const G4String& dirName) 75 { 90 { 76 if ( fLockDirectoryNames ) { << 91 if ( fLockNtupleDirectoryName ) { 77 Warn("Cannot set Ntuple directory name as << 92 G4ExceptionDescription description; 78 fkClass, "SetNtupleDirectoryName"); << 93 description >> 94 << "Cannot set Ntuple directory name as its value was already used."; >> 95 G4Exception("G4VFileManager::SetNtupleDirectoryName()", >> 96 "Analysis_W010", JustWarning, description); 79 return false; 97 return false; 80 } << 98 } 81 99 82 fNtupleDirectoryName = dirName; 100 fNtupleDirectoryName = dirName; 83 return true; 101 return true; 84 } << 102 } >> 103 >> 104 //_____________________________________________________________________________ >> 105 G4String G4VFileManager::GetFullFileName() const >> 106 { >> 107 G4String name(fFileName); >> 108 // Add thread Id to a file name if MT processing >> 109 if ( ! fState.GetIsMaster() ) { >> 110 std::ostringstream os; >> 111 os << G4Threading::G4GetThreadId(); >> 112 name.append("_t"); >> 113 name.append(os.str()); >> 114 } >> 115 // Add file extension .root if no extension is given >> 116 if ( name.find(".") == std::string::npos ) { >> 117 name.append("."); >> 118 name.append(GetFileType()); >> 119 } >> 120 >> 121 return name; >> 122 } >> 123 >> 124 //_____________________________________________________________________________ >> 125 G4String G4VFileManager::GetNtupleFileName(const G4String& ntupleName) const >> 126 { >> 127 G4String name(fFileName); >> 128 // Add ntupleName >> 129 name.append("_"); >> 130 name.append(ntupleName); >> 131 // Add thread Id to a file name if MT processing >> 132 if ( ! fState.GetIsMaster() ) { >> 133 std::ostringstream os; >> 134 os << G4Threading::G4GetThreadId(); >> 135 name.append("_t"); >> 136 name.append(os.str()); >> 137 } >> 138 // Add file extension .xml if no extension is given >> 139 if ( name.find(".") == std::string::npos ) { >> 140 name.append("."); >> 141 name.append(GetFileType()); >> 142 } >> 143 return name; >> 144 } >> 145 >> 146 //_____________________________________________________________________________ >> 147 G4String G4VFileManager::GetFileType() const >> 148 { >> 149 G4String fileType = fState.GetType(); >> 150 fileType.toLower(); >> 151 return fileType; >> 152 } >> 153 >> 154 85 155