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 // This example is provided by the Geant4-DNA collaboration
0027 // Any report or published results obtained using the Geant4-DNA software
0028 // shall cite the following Geant4-DNA collaboration publications:
0029 // Med. Phys. 45 (2018) e722-e739
0030 // Phys. Med. 31 (2015) 861-874
0031 // Med. Phys. 37 (2010) 4692-4708
0032 // Int. J. Model. Simul. Sci. Comput. 1 (2010) 157–178
0033 //
0034 // The Geant4-DNA web site is available at http://geant4-dna.org
0035 //
0036 /// \file RunAction.cc
0037 /// \brief Implementation of the RunAction class
0038
0039 #include "RunAction.hh"
0040
0041 #include "DetectorConstruction.hh"
0042 #include "PrimaryGeneratorAction.hh"
0043 #include "Run.hh"
0044
0045 #include "G4AnalysisManager.hh"
0046 #include "G4RunManager.hh"
0047
0048 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0049
0050 RunAction::RunAction() : G4UserRunAction(), fpDetector(0), fpRun(0)
0051 {
0052     fpDetector = dynamic_cast<const DetectorConstruction*>(
0053     G4RunManager::GetRunManager()->GetUserDetectorConstruction());
0054
0055     // Create analysis manager
0056     G4cout << "##### Create analysis manager "
0057            << "  " << this << G4endl;
0058     auto analysisManager = G4AnalysisManager::Instance();
0059
0060     analysisManager->SetDefaultFileType("root");
0061
0062     analysisManager->SetNtupleMerging(true);
0063
0064     analysisManager->SetActivation(true); // Set the possibility to activate or not
0065
0066     G4cout << "Using " << analysisManager->GetType() << " analysis manager" << G4endl;
0067
0068     // Create track information ntuple
0069     analysisManager->CreateNtuple("track", "dnaphysics");
0070     analysisManager->CreateNtupleDColumn("flagParticle");
0071     analysisManager->CreateNtupleDColumn("x");
0072     analysisManager->CreateNtupleDColumn("y");
0073     analysisManager->CreateNtupleDColumn("z");
0074     analysisManager->CreateNtupleDColumn("dirx");
0075     analysisManager->CreateNtupleDColumn("diry");
0076     analysisManager->CreateNtupleDColumn("dirz");
0077     analysisManager->CreateNtupleDColumn("kineticEnergy");
0078     analysisManager->CreateNtupleIColumn("trackID");
0079     analysisManager->CreateNtupleIColumn("parentID");
0080     analysisManager->FinishNtuple();
0081
0082     // Not activated by default
0083     analysisManager->SetNtupleActivation(false);
0084
0085 }
0086
0087 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0088
0089 RunAction::~RunAction()
0090 {}
0091
0092 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0093
0094 G4Run* RunAction::GenerateRun()
0095 {
0096   fpRun = new Run(fpDetector);
0097   return fpRun;
0098 }
0099
0100 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0101
0102 void RunAction::BeginOfRunAction(const G4Run*)
0103 {
0104   auto analysisManager = G4AnalysisManager::Instance();
0105
0106   // Open an output file
0107   G4String fileName = "spower";
0108   analysisManager->OpenFile(fileName);
0109
0110   //
0111   const PrimaryGeneratorAction* primary = dynamic_cast<const PrimaryGeneratorAction*>(
0112     G4RunManager::GetRunManager()->GetUserPrimaryGeneratorAction());
0113
0114   if (!primary) return;  //
0115
0116   // Keep run condition
0117   G4ParticleDefinition* particle = primary->GetParticleGun()->GetParticleDefinition();
0118   G4double energy = primary->GetParticleGun()->GetParticleEnergy();
0119   fpRun->SetPrimary(particle, energy);
0120 }
0121
0122 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0123
0124 void RunAction::EndOfRunAction(const G4Run*)
0125 {
0126   if (isMaster) fpRun->EndOfRun();
0127
0128   // Save histograms
0129   auto analysisManager = G4AnalysisManager::Instance();
0130   analysisManager->Write();
0131   analysisManager->CloseFile();
0132 }