MCTruth/src/MCTruthTrackingAction.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 eventgenerator/HepMC/MCTruth/src/MCTruthTrackingAction.cc
0027 /// \brief Implementation of the MCTruthTrackingAction class
0028 //
0029 //
0030 //
0031 //
0032 // --------------------------------------------------------------
0033 //      GEANT 4 - MCTruthTrackingAction class
0034 // --------------------------------------------------------------
0035 //
0036 // Author: Witold POKORSKI (Witold.Pokorski@cern.ch)
0037 //
0038 // --------------------------------------------------------------
0039 //
0040 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0041
0042 #include "MCTruthTrackingAction.hh"
0043
0044 #include "MCTruthTrackInformation.hh"
0045
0046 #include "G4Track.hh"
0047 #include "G4TrackVector.hh"
0048 #include "G4TrackingManager.hh"
0049
0050 #include <iostream>
0051
0052 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0053
0054 MCTruthTrackingAction::MCTruthTrackingAction() : G4UserTrackingAction() {}
0055
0056 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0057
0058 MCTruthTrackingAction::~MCTruthTrackingAction() {}
0059
0060 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0061
0062 void MCTruthTrackingAction::PreUserTrackingAction(const G4Track* track)
0063 {
0064   fmom = G4LorentzVector(track->GetMomentum(), track->GetTotalEnergy());
0065
0066   if (!track->GetUserInformation()) {
0067     G4VUserTrackInformation* mcinf = new MCTruthTrackInformation;
0068     fpTrackingManager->SetUserTrackInformation(mcinf);
0069   }
0070 }
0071
0072 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0073
0074 void MCTruthTrackingAction::PostUserTrackingAction(const G4Track* track)
0075 {
0076   G4LorentzVector prodpos(track->GetGlobalTime() - track->GetLocalTime(),
0077                           track->GetVertexPosition());
0078   G4LorentzVector endpos(track->GetGlobalTime(), track->GetPosition());
0079
0080   // here (?) make all different checks to decide whether to store the particle
0081   //
0082   if (TrackToBeStored(track)) {
0083     MCTruthTrackInformation* mcinf = (MCTruthTrackInformation*)track->GetUserInformation();
0084
0085     MCTruthManager::GetInstance()->AddParticle(
0086       fmom, prodpos, endpos, track->GetDefinition()->GetPDGEncoding(), track->GetTrackID(),
0087       track->GetParentID(), mcinf->GetDirectParent());
0088   }
0089   else {
0090     // If track is not to be stored, propagate it's parent ID (stored)
0091     // to its secondaries
0092     //
0093     G4TrackVector* childrens = fpTrackingManager->GimmeSecondaries();
0094
0095     for (unsigned int index = 0; index < childrens->size(); ++index) {
0096       G4Track* tr = (*childrens)[index];
0097       tr->SetParentID(track->GetParentID());
0098
0099       // set the flag saying that the direct mother is not stored
0100       //
0101       MCTruthTrackInformation* mcinf = (MCTruthTrackInformation*)tr->GetUserInformation();
0102       if (!mcinf) tr->SetUserInformation(mcinf = new MCTruthTrackInformation);
0103
0104       mcinf->SetDirectParent(false);
0105     }
0106   }
0107 }
0108
0109 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
0110
0111 G4bool MCTruthTrackingAction::TrackToBeStored(const G4Track* track)
0112 {
0113   MCTruthConfig* config = MCTruthManager::GetInstance()->GetConfig();
0114
0115   // check energy
0116   if (fmom.e() > config->GetMinE()) return true;
0117
0118   // particle type
0119   std::vector<G4int> types = config->GetParticleTypes();
0120
0121   if (std::find(types.begin(), types.end(), track->GetDefinition()->GetPDGEncoding())
0122       != types.end())
0123     return true;
0124
0125   // creator process
0126
0127   // etc...
0128
0129   return false;
0130 }
0131
0132 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....