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