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: G4BaseFileManager.cc 70604 2013-06-03 11:27:06Z ihrivnac $ 26 27 27 // Author: Ivana Hrivnacova, 10/09/2014 (ivan 28 // Author: Ivana Hrivnacova, 10/09/2014 (ivana@ipno.in2p3.fr) 28 29 29 #include "G4BaseFileManager.hh" 30 #include "G4BaseFileManager.hh" 30 #include "G4AnalysisUtilities.hh" << 31 31 32 #include "G4Threading.hh" 32 #include "G4Threading.hh" 33 33 34 //____________________________________________ 34 //_____________________________________________________________________________ 35 G4BaseFileManager::G4BaseFileManager(const G4A 35 G4BaseFileManager::G4BaseFileManager(const G4AnalysisManagerState& state) 36 : fState(state) << 36 : fState(state), 37 {} << 37 fFileName("") >> 38 { >> 39 } 38 40 39 // << 41 //_____________________________________________________________________________ 40 // public methods << 42 G4BaseFileManager::~G4BaseFileManager() >> 43 { >> 44 } >> 45 >> 46 // >> 47 // private methods 41 // 48 // 42 49 43 //____________________________________________ 50 //_____________________________________________________________________________ 44 void G4BaseFileManager::AddFileName(const G4St << 51 G4String G4BaseFileManager::TakeOffExtension(G4String& name) const 45 { 52 { 46 // G4cout << "registering " << fileName << " << 53 G4String extension; 47 << 54 if ( name.find(".") != std::string::npos ) { 48 // Do nothing in file name is already presen << 55 extension = name.substr(name.find(".")); 49 for ( const auto& name : fFileNames ) { << 56 name = name.substr(0, name.find(".")); 50 if ( name == fileName ) return; << 57 } >> 58 else { >> 59 extension = "."; >> 60 extension.append(GetFileType()); 51 } 61 } >> 62 return extension; >> 63 } 52 64 53 fFileNames.push_back(fileName); << 65 // 54 } << 66 // public methods >> 67 // 55 68 56 //____________________________________________ 69 //_____________________________________________________________________________ 57 G4String G4BaseFileManager::GetFileType() cons << 70 G4String G4BaseFileManager::GetFullFileName(const G4String& baseFileName, 58 { << 71 G4bool isPerThread) const 59 return fState.GetFileType(); << 72 { 60 } << 73 G4String name(baseFileName); >> 74 if ( name == "" ) name = fFileName; >> 75 >> 76 // Take out file extension >> 77 G4String extension = TakeOffExtension(name); >> 78 >> 79 // Add thread Id to a file name if MT processing >> 80 if ( isPerThread && ! fState.GetIsMaster() ) { >> 81 std::ostringstream os; >> 82 os << G4Threading::G4GetThreadId(); >> 83 name.append("_t"); >> 84 name.append(os.str()); >> 85 } >> 86 >> 87 // Add (back if it was present) file extension >> 88 name.append(extension); >> 89 >> 90 return name; >> 91 } 61 92 62 //____________________________________________ 93 //_____________________________________________________________________________ 63 G4String G4BaseFileManager::GetFullFileName(co << 94 G4String G4BaseFileManager::GetNtupleFileName(const G4String& ntupleName) const 64 G4 << 95 { 65 { << 96 G4String name(fFileName); 66 G4String fileName(baseFileName); << 67 if ( fileName == "" ) fileName = fFileName; << 68 97 69 // Take out file extension 98 // Take out file extension 70 auto name = G4Analysis::GetBaseName(fileName << 99 auto extension = TakeOffExtension(name); >> 100 >> 101 // Add ntupleName >> 102 name.append("_nt_"); >> 103 name.append(ntupleName); 71 104 72 // Add thread Id to a file name if MT proces 105 // Add thread Id to a file name if MT processing 73 if ( isPerThread && ! fState.GetIsMaster() ) << 106 if ( ! fState.GetIsMaster() ) { 74 std::ostringstream os; 107 std::ostringstream os; 75 os << G4Threading::G4GetThreadId(); 108 os << G4Threading::G4GetThreadId(); 76 name.append("_t"); 109 name.append("_t"); 77 name.append(os.str()); 110 name.append(os.str()); 78 } << 111 } 79 112 80 // Add (back if it was present or is defined << 113 // Add (back if it was present) file extension 81 auto extension = G4Analysis::GetExtension(fi << 114 name.append(extension); 82 if (extension.size() != 0u) { << 115 83 name.append("."); << 116 return name; 84 name.append(extension); << 117 } 85 } << 86 118 >> 119 //_____________________________________________________________________________ >> 120 G4String G4BaseFileManager::GetNtupleFileName(G4int ntupleFileNumber) const >> 121 { >> 122 G4String name(fFileName); >> 123 >> 124 // Take out file extension >> 125 auto extension = TakeOffExtension(name); >> 126 >> 127 // Add _M followed by ntupleFileNumber >> 128 std::ostringstream os; >> 129 os << ntupleFileNumber; >> 130 name.append("_m"); >> 131 name.append(os.str()); >> 132 >> 133 // Add (back if it was present) file extension >> 134 name.append(extension); >> 135 87 return name; 136 return name; 88 } << 137 } 89 138 90 //____________________________________________ 139 //_____________________________________________________________________________ 91 G4String G4BaseFileManager::GetHnFileName(cons 140 G4String G4BaseFileManager::GetHnFileName(const G4String& hnType, 92 cons 141 const G4String& hnName) const 93 { 142 { 94 return G4Analysis::GetHnFileName(fFileName, << 143 G4String name(fFileName); 95 } << 96 144 97 //____________________________________________ << 145 // Take out file extension 98 G4String G4BaseFileManager::GetHnFileName(con << 146 auto extension = TakeOffExtension(name); 99 G4i << 147 100 { << 148 // Add _hnType_hnName 101 // Do nothing if cycle is supported by the o << 149 name.append("_"); 102 if (HasCycles()) return fileName; << 150 name.append(hnType); >> 151 name.append("_"); >> 152 name.append(hnName); >> 153 >> 154 // Add (back if it was present) file extension >> 155 name.append(extension); 103 156 104 return G4Analysis::GetHnFileName(fileName, G << 157 return name; 105 } 158 } 106 159 107 //____________________________________________ 160 //_____________________________________________________________________________ 108 G4String G4BaseFileManager::GetNtupleFileName( << 161 G4String G4BaseFileManager::GetPlotFileName() const 109 << 110 { 162 { 111 // Do not pass cycle if supported by the out << 163 G4String name(fFileName); 112 auto cycleToPass = (HasCycles()) ? 0 : cycle << 113 164 114 return G4Analysis::GetNtupleFileName(fFileNa << 165 // Take out file extension 115 } << 166 auto extension = TakeOffExtension(name); 116 167 117 //____________________________________________ << 168 // Add .ps extension 118 G4String G4BaseFileManager::GetNtupleFileName( << 169 name.append(".ps"); 119 << 120 { << 121 // Do not pass cycle if supported by the out << 122 auto cycleToPass = (HasCycles()) ? 0 : cycle << 123 170 124 return G4Analysis::GetNtupleFileName(fFileNa << 171 return name; 125 } 172 } 126 173 127 //____________________________________________ 174 //_____________________________________________________________________________ 128 G4String G4BaseFileManager::GetPlotFileName() << 175 G4String G4BaseFileManager::GetFileType() const 129 { 176 { 130 return G4Analysis::GetPlotFileName(fFileName << 177 G4String fileType = fState.GetType(); 131 } << 178 fileType.toLower(); >> 179 return fileType; >> 180 } 132 181 133 182 134 183