spower/src/RunAction.cc

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 /// \file RunAction.cc
0027 /// \brief Implementation of the RunAction class
0028
0029 // This example is provided by the Geant4-DNA collaboration
0030 // Any report or published results obtained using the Geant4-DNA software
0031 // shall cite the following Geant4-DNA collaboration publications:
0032 // Med. Phys. 45 (2018) e722-e739
0033 // Phys. Med. 31 (2015) 861-874
0034 // Med. Phys. 37 (2010) 4692-4708
0035 // Int. J. Model. Simul. Sci. Comput. 1 (2010) 157–178
0036 //
0037 // The Geant4-DNA web site is available at http://geant4-dna.org
0038 //
0039
0040 #include "RunAction.hh"
0041
0042 #include "DetectorConstruction.hh"
0043 #include "PrimaryGeneratorAction.hh"
0044 #include "Run.hh"
0045
0046 #include "G4AnalysisManager.hh"
0047 #include "G4RunManager.hh"
0048
0049 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0050
0051 RunAction::RunAction() : G4UserRunAction(), fpDetector(0), fpRun(0)
0052 {
0053     fpDetector = dynamic_cast<const DetectorConstruction*>(
0054     G4RunManager::GetRunManager()->GetUserDetectorConstruction());
0055
0056     // Create analysis manager
0057     G4cout << "##### Create analysis manager "
0058            << "  " << this << G4endl;
0059     auto analysisManager = G4AnalysisManager::Instance();
0060
0061     analysisManager->SetDefaultFileType("root");
0062
0063     analysisManager->SetNtupleMerging(true);
0064
0065     analysisManager->SetActivation(true); // Set the possibility to activate or not
0066
0067     G4cout << "Using " << analysisManager->GetType() << " analysis manager" << G4endl;
0068
0069     // Create track information ntuple
0070     analysisManager->CreateNtuple("track", "dnaphysics");
0071     analysisManager->CreateNtupleDColumn("flagParticle");
0072     analysisManager->CreateNtupleDColumn("x");
0073     analysisManager->CreateNtupleDColumn("y");
0074     analysisManager->CreateNtupleDColumn("z");
0075     analysisManager->CreateNtupleDColumn("dirx");
0076     analysisManager->CreateNtupleDColumn("diry");
0077     analysisManager->CreateNtupleDColumn("dirz");
0078     analysisManager->CreateNtupleDColumn("kineticEnergy");
0079     analysisManager->CreateNtupleIColumn("trackID");
0080     analysisManager->CreateNtupleIColumn("parentID");
0081     analysisManager->FinishNtuple();
0082
0083     // Not activated by default
0084     analysisManager->SetNtupleActivation(false);
0085
0086 }
0087
0088 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0089
0090 RunAction::~RunAction()
0091 {}
0092
0093 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0094
0095 G4Run* RunAction::GenerateRun()
0096 {
0097   fpRun = new Run(fpDetector);
0098   return fpRun;
0099 }
0100
0101 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0102
0103 void RunAction::BeginOfRunAction(const G4Run*)
0104 {
0105   auto analysisManager = G4AnalysisManager::Instance();
0106
0107   // Open an output file
0108   G4String fileName = "spower";
0109   analysisManager->OpenFile(fileName);
0110
0111   //
0112   const PrimaryGeneratorAction* primary = dynamic_cast<const PrimaryGeneratorAction*>(
0113     G4RunManager::GetRunManager()->GetUserPrimaryGeneratorAction());
0114
0115   if (!primary) return;  //
0116
0117   // Keep run condition
0118   G4ParticleDefinition* particle = primary->GetParticleGun()->GetParticleDefinition();
0119   G4double energy = primary->GetParticleGun()->GetParticleEnergy();
0120   fpRun->SetPrimary(particle, energy);
0121 }
0122
0123 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0124
0125 void RunAction::EndOfRunAction(const G4Run*)
0126 {
0127   if (isMaster) fpRun->EndOfRun();
0128
0129   // Save histograms
0130   auto analysisManager = G4AnalysisManager::Instance();
0131   analysisManager->Write();
0132   analysisManager->CloseFile();
0133 }