Files
geant4/examples/extended/polarisation/Pol01/src/RunAction.cc
T
2025-12-05 08:54:02 +01:00

371 lines
13 KiB
C++

//
// ********************************************************************
// * License and Disclaimer *
// * *
// * The Geant4 software is copyright of the Copyright Holders of *
// * the Geant4 Collaboration. It is provided under the terms and *
// * conditions of the Geant4 Software License, included in the file *
// * LICENSE and available at http://cern.ch/geant4/license . These *
// * include a list of copyright holders. *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. Please see the license in the file LICENSE and URL above *
// * for the full disclaimer and the limitation of liability. *
// * *
// * This code implementation is the result of the scientific and *
// * technical work of the GEANT4 collaboration. *
// * By using, copying, modifying or distributing the software (or *
// * any work based on the software) you agree to acknowledge its *
// * use in resulting scientific publications, and indicate your *
// * acceptance of all terms of the Geant4 Software license. *
// ********************************************************************
//
/// \file RunAction.cc
/// \brief Implementation of the RunAction class
#include "RunAction.hh"
#include "DetectorConstruction.hh"
#include "PrimaryGeneratorAction.hh"
#include "ProcessesCount.hh"
#include "G4AccumulableManager.hh"
#include "G4Electron.hh"
#include "G4EmCalculator.hh"
#include "G4Gamma.hh"
#include "G4ParticleDefinition.hh"
#include "G4PhysicalConstants.hh"
#include "G4Positron.hh"
#include "G4Run.hh"
#include "G4RunManager.hh"
#include "G4SystemOfUnits.hh"
#include "G4UnitsTable.hh"
#include <iomanip>
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
RunAction::RunAction(DetectorConstruction* det, PrimaryGeneratorAction* prim)
: fDetector(det), fPrimary(prim), fAnalysisManager(0), fTotalEventCount(0)
{
fGamma = G4Gamma::Gamma();
fElectron = G4Electron::Electron();
fPositron = G4Positron::Positron();
auto accumulableManager = G4AccumulableManager::Instance();
auto fPhotonStats = new ParticleStatistics("PhotonStats");
auto fElectronStats = new ParticleStatistics("ElectronStats");
auto fPositronStats = new ParticleStatistics("PositronStats");
auto fProcCounter = new ProcessesCount("ProcCounter");
accumulableManager->Register(fPhotonStats);
accumulableManager->Register(fElectronStats);
accumulableManager->Register(fPositronStats);
accumulableManager->Register(fProcCounter);
BookHisto();
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
RunAction::~RunAction() {}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::BeginOfRunAction(const G4Run* aRun)
{
G4cout << "### Run " << aRun->GetRunID() << " start." << G4endl;
auto accumulableManager = G4AccumulableManager::Instance();
accumulableManager->Reset();
// save Rndm status
// G4RunManager::GetRunManager()->SetRandomNumberStore(false);
// CLHEP::HepRandom::showEngineStatus();
fTotalEventCount = 0;
// Open file histogram file
fAnalysisManager->OpenFile();
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::FillData(const G4ParticleDefinition* particle, G4double kinEnergy,
G4double costheta, G4double phi, G4double longitudinalPolarization)
{
auto accManager = G4AccumulableManager::Instance();
G4int id = -1;
if (particle == fGamma) {
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PhotonStats"))
->FillData(kinEnergy, costheta, longitudinalPolarization);
if (fAnalysisManager) {
id = 1;
}
}
else if (particle == fElectron) {
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("ElectronStats"))
->FillData(kinEnergy, costheta, longitudinalPolarization);
if (fAnalysisManager) {
id = 5;
}
}
else if (particle == fPositron) {
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PositronStats"))
->FillData(kinEnergy, costheta, longitudinalPolarization);
if (fAnalysisManager) {
id = 9;
}
}
if (id > 0) {
fAnalysisManager->FillH1(id, kinEnergy, 1.0);
fAnalysisManager->FillH1(id + 1, costheta, 1.0);
fAnalysisManager->FillH1(id + 2, phi, 1.0);
fAnalysisManager->FillH1(id + 3, longitudinalPolarization, 1.0);
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::BookHisto()
{
// Always creating analysis manager
fAnalysisManager = G4AnalysisManager::Instance();
fAnalysisManager->SetDefaultFileType("root");
fAnalysisManager->SetActivation(true);
fAnalysisManager->SetVerboseLevel(1);
// Open file histogram file
fAnalysisManager->SetFileName("pol01");
fAnalysisManager->SetFirstHistoId(1);
// Creating an 1-dimensional histograms in the root directory of the tree
const G4String id[] = {"h1", "h2", "h3", "h4", "h5", "h6", "h7", "h8", "h9", "h10", "h11", "h12"};
const G4String title[] = {
"Gamma Energy distribution", // 1
"Gamma Cos(Theta) distribution", // 2
"Gamma Phi angular distribution", // 3
"Gamma longitudinal Polarization", // 4
"Electron Energy distribution", // 5
"Electron Cos(Theta) distribution", // 6
"Electron Phi angular distribution", // 7
"Electron longitudinal Polarization", // 8
"Positron Energy distribution", // 9
"Positron Cos(Theta) distribution", // 10
"Positron Phi angular distribution", // 11
"Positron longitudinal Polarization" // 12
};
G4double vmin, vmax;
G4int nbins = 120;
for (int i = 0; i < 12; ++i) {
G4int j = i - i / 4 * 4;
if (0 == j) {
vmin = 0.;
vmax = 12. * MeV;
}
else if (1 == j) {
vmin = -1.;
vmax = 1.;
}
else if (2 == j) {
vmin = 0.;
vmax = pi;
}
else {
vmin = -1.5;
vmax = 1.5;
}
G4int ih = fAnalysisManager->CreateH1(id[i], title[i], nbins, vmin, vmax);
fAnalysisManager->SetH1Activation(ih, false);
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::SaveHisto(G4int nevents)
{
if (fAnalysisManager) {
if (IsMaster()) {
G4double norm = 1.0 / G4double(nevents);
for (int i = 0; i < 12; ++i) {
fAnalysisManager->ScaleH1(i, norm);
}
}
fAnalysisManager->Write();
fAnalysisManager->CloseFile();
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::CountProcesses(G4String& procName)
{
auto accManager = G4AccumulableManager::Instance();
dynamic_cast<ProcessesCount*>(accManager->GetAccumulable("ProcCounter"))->Count(procName);
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::EndOfRunAction(const G4Run* aRun)
{
// Total number of events in run (all threads)
G4int NbOfEvents = aRun->GetNumberOfEventToBeProcessed();
// G4int NbOfEvents = aRun->GetNumberOfEvent();
if (NbOfEvents == 0) return;
G4int prec = G4cout.precision(5);
G4Material* material = fDetector->GetMaterial();
G4double density = material->GetDensity();
if (fPrimary != nullptr) {
G4ParticleDefinition* particle = fPrimary->GetParticleGun()->GetParticleDefinition();
G4String Particle = particle->GetParticleName();
G4double energy = fPrimary->GetParticleGun()->GetParticleEnergy();
G4cout << "\n The run consists of " << fTotalEventCount << " " << Particle << " of "
<< G4BestUnit(energy, "Energy") << " through "
<< G4BestUnit(fDetector->GetBoxSizeZ(), "Length") << " of " << material->GetName()
<< " (density: " << G4BestUnit(density, "Volumic Mass") << ")" << G4endl;
}
// cross check from G4EmCalculator
// G4cout << "\n Verification from G4EmCalculator. \n";
// G4EmCalculator emCal;
auto accManager = G4AccumulableManager::Instance();
accManager->Merge();
if (IsMaster()) {
// frequency of processes
G4cout << "\n Process calls frequency --->\n";
dynamic_cast<ProcessesCount*>(accManager->GetAccumulable("ProcCounter"))->Print();
G4cout << " Gamma: \n";
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PhotonStats"))
->PrintResults(NbOfEvents);
G4cout << " Electron: \n";
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("ElectronStats"))
->PrintResults(NbOfEvents);
G4cout << " Positron: \n";
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PositronStats"))
->PrintResults(NbOfEvents);
G4cout << G4endl;
}
// restore default format
G4cout.precision(prec);
// write out histograms
SaveHisto(NbOfEvents);
if (IsMaster()) {
// show Rndm status
CLHEP::HepRandom::showEngineStatus();
}
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::EventFinished()
{
auto accManager = G4AccumulableManager::Instance();
++fTotalEventCount;
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PhotonStats"))->EventFinished();
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("ElectronStats"))->EventFinished();
dynamic_cast<ParticleStatistics*>(accManager->GetAccumulable("PositronStats"))->EventFinished();
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
RunAction::ParticleStatistics::ParticleStatistics(const G4String& name)
: G4VAccumulable(name),
fCurrentNumber(0),
fTotalNumber(0),
fTotalNumber2(0),
fSumEnergy(0),
fSumEnergy2(0),
fSumPolarization(0),
fSumPolarization2(0),
fSumCosTheta(0),
fSumCosTheta2(0)
{}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
RunAction::ParticleStatistics::~ParticleStatistics() {}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::ParticleStatistics::EventFinished()
{
fTotalNumber += fCurrentNumber;
fTotalNumber2 += fCurrentNumber * fCurrentNumber;
fCurrentNumber = 0;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::ParticleStatistics::FillData(G4double kinEnergy, G4double costheta,
G4double longitudinalPolarization)
{
++fCurrentNumber;
fSumEnergy += kinEnergy;
fSumEnergy2 += kinEnergy * kinEnergy;
fSumPolarization += longitudinalPolarization;
fSumPolarization2 += longitudinalPolarization * longitudinalPolarization;
fSumCosTheta += costheta;
fSumCosTheta2 += costheta * costheta;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::ParticleStatistics::PrintResults(G4int totalNumberOfEvents)
{
G4cout << "Mean Number per Event :" << G4double(fTotalNumber) / G4double(totalNumberOfEvents)
<< "\n";
if (fTotalNumber == 0) fTotalNumber = 1;
G4double energyMean = fSumEnergy / fTotalNumber;
G4double energyRms = std::sqrt(fSumEnergy2 / fTotalNumber - energyMean * energyMean);
G4cout << "Mean Energy :" << G4BestUnit(energyMean, "Energy") << " +- "
<< G4BestUnit(energyRms, "Energy") << "\n";
G4double polarizationMean = fSumPolarization / fTotalNumber;
G4double polarizationRms =
std::sqrt(fSumPolarization2 / fTotalNumber - polarizationMean * polarizationMean);
G4cout << "Mean Polarization :" << polarizationMean << " +- " << polarizationRms << "\n";
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::ParticleStatistics::Reset()
{
fCurrentNumber = 0;
fTotalNumber = fTotalNumber2 = 0;
fSumEnergy = fSumEnergy2 = 0;
fSumPolarization = fSumPolarization2 = 0;
fSumCosTheta = fSumCosTheta2 = 0;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
void RunAction::ParticleStatistics::Merge(const G4VAccumulable& other)
{
auto rstat = dynamic_cast<const RunAction::ParticleStatistics&>(other);
fCurrentNumber += rstat.fCurrentNumber;
fTotalNumber += rstat.fTotalNumber;
fTotalNumber2 += rstat.fTotalNumber2;
fSumEnergy += rstat.fSumEnergy;
fSumEnergy2 += rstat.fSumEnergy2;
fSumPolarization += rstat.fSumPolarization;
fSumPolarization2 += rstat.fSumPolarization2;
fSumCosTheta += rstat.fSumCosTheta;
fSumCosTheta2 += rstat.fSumCosTheta2;
}
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......