Hadr03/src/SteppingAction.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 SteppingAction.cc
0027 /// \brief Implementation of the SteppingAction class
0028 //
0029 //
0030 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0031 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0032
0033 #include "SteppingAction.hh"
0034
0035 #include "HistoManager.hh"
0036 #include "Run.hh"
0037
0038 #include "G4HadronicProcess.hh"
0039 #include "G4ParticleTypes.hh"
0040 #include "G4RunManager.hh"
0041
0042 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0043
0044 // SteppingAction::SteppingAction()
0045 //: G4UserSteppingAction()
0046 //{ }
0047
0048 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0049
0050 void SteppingAction::UserSteppingAction(const G4Step* aStep)
0051 {
0052   // check trackID and stepNumber
0053   G4int trackID = aStep->GetTrack()->GetTrackID();
0054   G4int stepNb = aStep->GetTrack()->GetCurrentStepNumber();
0055   if (trackID * stepNb != 1) return;
0056   // ok, we are at first interaction of the primary particle
0057
0058   Run* run = static_cast<Run*>(G4RunManager::GetRunManager()->GetNonConstCurrentRun());
0059
0060   // count processes
0061   //
0062   const G4StepPoint* endPoint = aStep->GetPostStepPoint();
0063   G4VProcess* process = const_cast<G4VProcess*>(endPoint->GetProcessDefinedStep());
0064   run->CountProcesses(process);
0065
0066   // check that an real interaction occured (eg. not a transportation)
0067   G4StepStatus stepStatus = endPoint->GetStepStatus();
0068   G4bool transmit = (stepStatus == fGeomBoundary || stepStatus == fWorldBoundary);
0069   if (transmit) return;
0070
0071   // real processes : sum track length
0072   //
0073   G4double stepLength = aStep->GetStepLength();
0074   run->SumTrack(stepLength);
0075
0076   // energy-momentum balance initialisation
0077   //
0078   const G4StepPoint* prePoint = aStep->GetPreStepPoint();
0079   G4double Q = -prePoint->GetKineticEnergy();
0080   G4ThreeVector Pbalance = -prePoint->GetMomentum();
0081
0082   // initialisation of the nuclear channel identification
0083   //
0084   G4ParticleDefinition* particle = aStep->GetTrack()->GetDefinition();
0085   G4String partName = particle->GetParticleName();
0086   G4String nuclearChannel = partName;
0087   G4HadronicProcess* hproc = dynamic_cast<G4HadronicProcess*>(process);
0088   const G4Isotope* target = NULL;
0089   if (hproc) target = hproc->GetTargetIsotope();
0090   G4String targetName = "XXXX";
0091   if (target) targetName = target->GetName();
0092   nuclearChannel += " + " + targetName + " --> ";
0093   if (targetName == "XXXX") run->SetTargetXXX(true);
0094
0095   // scattered primary particle (if any)
0096   //
0097   G4AnalysisManager* analysis = G4AnalysisManager::Instance();
0098   G4int ih = 1;
0099   if (aStep->GetTrack()->GetTrackStatus() == fAlive) {
0100     G4double energy = endPoint->GetKineticEnergy();
0101     analysis->FillH1(ih, energy);
0102     //
0103     G4ThreeVector momentum = endPoint->GetMomentum();
0104     Q += energy;
0105     Pbalance += momentum;
0106     //
0107     nuclearChannel += partName + " + ";
0108   }
0109
0110   // secondaries
0111   //
0112   const std::vector<const G4Track*>* secondary = aStep->GetSecondaryInCurrentStep();
0113   for (size_t lp = 0; lp < (*secondary).size(); lp++) {
0114     particle = (*secondary)[lp]->GetDefinition();
0115     G4String name = particle->GetParticleName();
0116     G4String type = particle->GetParticleType();
0117     G4double energy = (*secondary)[lp]->GetKineticEnergy();
0118     run->ParticleCount(name, energy);
0119     // energy spectrum
0120     ih = 0;
0121     if (particle == G4Gamma::Gamma())
0122       ih = 2;
0123     else if (particle == G4Electron::Electron())
0124       ih = 3;
0125     else if (particle == G4Neutron::Neutron())
0126       ih = 4;
0127     else if (particle == G4Proton::Proton())
0128       ih = 5;
0129     else if (particle == G4Deuteron::Deuteron())
0130       ih = 6;
0131     else if (particle == G4Alpha::Alpha())
0132       ih = 7;
0133     else if (type == "nucleus")
0134       ih = 8;
0135     else if (type == "meson")
0136       ih = 9;
0137     else if (type == "baryon")
0138       ih = 10;
0139     if (ih > 0) analysis->FillH1(ih, energy);
0140     // atomic mass
0141     if (type == "nucleus") {
0142       G4int A = particle->GetAtomicMass();
0143       analysis->FillH1(13, A);
0144     }
0145     // energy-momentum balance
0146     G4ThreeVector momentum = (*secondary)[lp]->GetMomentum();
0147     Q += energy;
0148     Pbalance += momentum;
0149     // count e- from internal conversion together with gamma
0150     if (particle == G4Electron::Electron()) particle = G4Gamma::Gamma();
0151     // particle flag
0152     fParticleFlag[particle]++;
0153   }
0154
0155   // energy-momentum balance
0156   G4double Pbal = Pbalance.mag();
0157   run->Balance(Pbal);
0158   ih = 11;
0159   analysis->FillH1(ih, Q);
0160   ih = 12;
0161   analysis->FillH1(ih, Pbal);
0162
0163   // nuclear channel
0164   const G4int kMax = 16;
0165   const G4String conver[] = {"0",  "",    "2 ",  "3 ",  "4 ",  "5 ",  "6 ",  "7 ", "8 ",
0166                              "9 ", "10 ", "11 ", "12 ", "13 ", "14 ", "15 ", "16 "};
0167   std::map<G4ParticleDefinition*, G4int>::iterator ip;
0168   for (ip = fParticleFlag.begin(); ip != fParticleFlag.end(); ip++) {
0169     particle = ip->first;
0170     G4String name = particle->GetParticleName();
0171     G4int nb = ip->second;
0172     if (nb > kMax) nb = kMax;
0173     G4String Nb = conver[nb];
0174     if (particle == G4Gamma::Gamma()) {
0175       run->CountGamma(nb);
0176       Nb = "N ";
0177       name = "gamma or e-";
0178     }
0179     if (ip != fParticleFlag.begin()) nuclearChannel += " + ";
0180     nuclearChannel += Nb + name;
0181   }
0182
0183   /// G4cout << "\n nuclear channel: " << nuclearChannel << G4endl;
0184   run->CountNuclearChannel(nuclearChannel, Q);
0185
0186   fParticleFlag.clear();
0187
0188   // kill event after first interaction
0189   //
0190   G4RunManager::GetRunManager()->AbortEvent();
0191 }
0192
0193 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......