Geant4 Cross Reference |
1 // 2 // ******************************************************************** 3 // * License and Disclaimer * 4 // * * 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. * 10 // * * 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitation of liability. * 17 // * * 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************************************** 25 // 26 // ------------------------------------------------------------------- 27 // ------------------------------------------------------------------- 28 29 #include "RunAction.hh" 30 #include "G4Run.hh" 31 #include "TrackingAction.hh" 32 #include "G4ParticleDefinition.hh" 33 #include "G4RunManager.hh" 34 #include "G4AnalysisManager.hh" 35 #include "G4Threading.hh" 36 37 void PrintNParticles(std::map<const G4ParticleDefinition*, int>& container); 38 39 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 40 41 RunAction::RunAction() 42 { 43 fFileName = "microelectronics"; 44 fpTrackingAction = 0; 45 fInitialized = 0; 46 fDebug = false; 47 } 48 49 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 50 51 RunAction::~RunAction() 52 {} 53 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 55 56 void RunAction::BeginOfRunAction(const G4Run* run) 57 { 58 // In this example, we considered that the same class was 59 // used for both master and worker threads. 60 // However, in case the run action is long, 61 // for better code review, this practice is not recommanded. 62 // 63 64 if(isMaster) // WARNING : in sequential mode, isMaster == true 65 BeginMaster(run); 66 else 67 BeginWorker(run); 68 } 69 70 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 71 72 void RunAction::EndOfRunAction(const G4Run* run) 73 { 74 if(isMaster) 75 EndMaster(run); 76 else 77 EndWorker(run); 78 } 79 80 81 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 82 83 void RunAction::BeginMaster(const G4Run* run) 84 { 85 bool sequential = (G4RunManager::GetRunManager()->GetRunManagerType() == G4RunManager::sequentialRM); 86 87 if(fDebug) 88 { 89 G4cout << "°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°" << G4endl; 90 if(!sequential) 91 G4cout << "°°°°°°°°°°°°°°°° RunAction::BeginMaster" << G4endl; 92 PrintRunInfo(run); 93 G4cout << "°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°" << G4endl; 94 } 95 96 if(sequential) 97 { 98 if(!fInitialized) 99 InitializeWorker(run); 100 // Note: fpTrackingAction could be used as a flag for initialization instead 101 102 CreateHistogram(); 103 } 104 } 105 106 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 107 108 void RunAction::BeginWorker(const G4Run* run) 109 { 110 if(fDebug) 111 { 112 G4cout << "°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°" << G4endl; 113 G4cout << "°°°°°°°°°°°°°°°° RunAction::BeginWorker" << G4endl; 114 PrintRunInfo(run); 115 G4cout << "°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°°" << G4endl; 116 } 117 if(!fInitialized) 118 InitializeWorker(run); 119 120 CreateHistogram(); 121 } 122 123 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 124 125 void RunAction::EndMaster(const G4Run* run) 126 { 127 bool sequential = (G4RunManager::GetRunManager()->GetRunManagerType() 128 == G4RunManager::sequentialRM); 129 if(sequential) 130 EndWorker(run); 131 } 132 133 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 134 135 void RunAction::EndWorker(const G4Run* run) 136 { 137 if(fDebug) 138 { 139 PrintRunInfo(run); 140 } 141 142 G4int nofEvents = run->GetNumberOfEvent(); 143 if ( nofEvents == 0 ) 144 { 145 if(fDebug) 146 { 147 G4cout << "°°°°°°°°°°°°°°°° NO EVENTS TREATED IN THIS RUN ==> LEAVING RunAction::EndOfRunAction "<< G4endl; 148 } 149 return; 150 } 151 152 /////////////// 153 // Write Histo 154 // 155 WriteHistogram(); 156 157 /////////////// 158 // Complete cleanup 159 // 160 G4AnalysisManager* analysisManager = G4AnalysisManager::Instance(); 161 analysisManager->Clear(); 162 163 /////////////// 164 // Printouts 165 // 166 std::map<const G4ParticleDefinition*, int>& 167 particlesCreatedInWorld = fpTrackingAction->GetNParticlesCreatedInWorld(); 168 169 G4cout << "Number and type of particles created outside region \"Target\" :" << G4endl; 170 171 PrintNParticles(particlesCreatedInWorld); 172 173 G4cout << "_______________________" << G4endl; 174 175 std::map<const G4ParticleDefinition*, int>& 176 particlesCreatedInTarget = fpTrackingAction->GetNParticlesCreatedInTarget(); 177 178 G4cout << "Number and type of particles created in region \"Target\" :" << G4endl; 179 180 PrintNParticles(particlesCreatedInTarget); 181 182 } 183 184 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 185 186 void RunAction::InitializeWorker(const G4Run*) 187 { 188 if (fpTrackingAction == 0) 189 { 190 fpTrackingAction = (TrackingAction*) G4RunManager::GetRunManager()->GetUserTrackingAction(); 191 192 if(fpTrackingAction == 0 && isMaster == false) 193 { 194 G4ExceptionDescription exDescrption ; 195 exDescrption << "fpTrackingAction is a null pointer. Has it been correctly initialized ?"; 196 G4Exception("RunAction::BeginOfRunAction","RunAction001",FatalException, exDescrption); 197 } 198 } 199 200 fInitialized = true; 201 } 202 203 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 204 205 void RunAction::CreateHistogram() 206 { 207 // Book histograms, ntuple 208 209 // Create analysis manager 210 // The choice of analysis technology is done via selection of a namespace 211 // in Analysis.hh 212 213 G4cout << "##### Create analysis manager " << " " << this << G4endl; 214 G4AnalysisManager* analysisManager = G4AnalysisManager::Instance(); 215 analysisManager->SetDefaultFileType("root"); 216 217 G4cout << "Using " << analysisManager->GetType() << " analysis manager" << G4endl; 218 219 // Create directories 220 221 //analysisManager->SetHistoDirectoryName("histograms"); 222 //analysisManager->SetNtupleDirectoryName("ntuple"); 223 analysisManager->SetVerboseLevel(1); 224 225 // Open an output file 226 227 analysisManager->OpenFile(fFileName); 228 229 // Creating ntuple 230 231 analysisManager->CreateNtuple("microelectronics", "physics"); 232 analysisManager->CreateNtupleDColumn("flagParticle"); 233 analysisManager->CreateNtupleDColumn("flagProcess"); 234 analysisManager->CreateNtupleDColumn("x"); 235 analysisManager->CreateNtupleDColumn("y"); 236 analysisManager->CreateNtupleDColumn("z"); 237 analysisManager->CreateNtupleDColumn("totalEnergyDeposit"); 238 analysisManager->CreateNtupleDColumn("stepLength"); 239 analysisManager->CreateNtupleDColumn("kineticEnergyDifference"); 240 analysisManager->FinishNtuple(); 241 } 242 243 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 244 245 void RunAction::WriteHistogram() 246 { 247 // print histogram statistics 248 // 249 G4AnalysisManager* analysisManager = G4AnalysisManager::Instance(); 250 251 // save histograms 252 // 253 analysisManager->Write(); 254 analysisManager->CloseFile(); 255 } 256 257 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 258 259 void RunAction::PrintRunInfo(const G4Run* run) 260 { 261 G4cout << "°°°°°°°°°°°°°°°° Run is = " << run->GetRunID() << G4endl; 262 G4cout << "°°°°°°°°°°°°°°°° Run type is = " << G4RunManager::GetRunManager()->GetRunManagerType() << G4endl; 263 G4cout << "°°°°°°°°°°°°°°°° Event processed = " << run->GetNumberOfEventToBeProcessed() << G4endl; 264 G4cout << "°°°°°°°°°°°°°°°° N° Event = " << run->GetNumberOfEvent() << G4endl; 265 } 266 267 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 268 void PrintNParticles(std::map<const G4ParticleDefinition*, int>& container) 269 { 270 std::map<const G4ParticleDefinition*, int>::iterator it; 271 for(it = container.begin() ; 272 it != container.end(); it ++) 273 { 274 G4cout << "N " << it->first->GetParticleName() << " : " << it->second << G4endl; 275 } 276 } 277 278 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 279