| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 /// \file eventgenerator/HepMC/MCTruth/src/MCTruthTrackingAction.cc
27 /// \brief Implementation of the MCTruthTrackingAction class
28 //
29 //
30 //
31 //
32 // --------------------------------------------------------------
33 // GEANT 4 - MCTruthTrackingAction class
34 // --------------------------------------------------------------
35 //
36 // Author: Witold POKORSKI (Witold.Pokorski@cern.ch)
37 //
38 // --------------------------------------------------------------
39 //
40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
41
42 #include "MCTruthTrackingAction.hh"
43
44 #include "MCTruthTrackInformation.hh"
45
46 #include "G4Track.hh"
47 #include "G4TrackVector.hh"
48 #include "G4TrackingManager.hh"
49
50 #include <iostream>
51
52 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
53
54 MCTruthTrackingAction::MCTruthTrackingAction() : G4UserTrackingAction() {}
55
56 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
57
58 MCTruthTrackingAction::~MCTruthTrackingAction() {}
59
60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
61
62 void MCTruthTrackingAction::PreUserTrackingAction(const G4Track* track)
63 {
64 fmom = G4LorentzVector(track->GetMomentum(), track->GetTotalEnergy());
65
66 if (!track->GetUserInformation()) {
67 G4VUserTrackInformation* mcinf = new MCTruthTrackInformation;
68 fpTrackingManager->SetUserTrackInformation(mcinf);
69 }
70 }
71
72 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
73
74 void MCTruthTrackingAction::PostUserTrackingAction(const G4Track* track)
75 {
76 G4LorentzVector prodpos(track->GetGlobalTime() - track->GetLocalTime(),
77 track->GetVertexPosition());
78 G4LorentzVector endpos(track->GetGlobalTime(), track->GetPosition());
79
80 // here (?) make all different checks to decide whether to store the particle
81 //
82 if (TrackToBeStored(track)) {
83 MCTruthTrackInformation* mcinf = (MCTruthTrackInformation*)track->GetUserInformation();
84
85 MCTruthManager::GetInstance()->AddParticle(
86 fmom, prodpos, endpos, track->GetDefinition()->GetPDGEncoding(), track->GetTrackID(),
87 track->GetParentID(), mcinf->GetDirectParent());
88 }
89 else {
90 // If track is not to be stored, propagate it's parent ID (stored)
91 // to its secondaries
92 //
93 G4TrackVector* childrens = fpTrackingManager->GimmeSecondaries();
94
95 for (unsigned int index = 0; index < childrens->size(); ++index) {
96 G4Track* tr = (*childrens)[index];
97 tr->SetParentID(track->GetParentID());
98
99 // set the flag saying that the direct mother is not stored
100 //
101 MCTruthTrackInformation* mcinf = (MCTruthTrackInformation*)tr->GetUserInformation();
102 if (!mcinf) tr->SetUserInformation(mcinf = new MCTruthTrackInformation);
103
104 mcinf->SetDirectParent(false);
105 }
106 }
107 }
108
109 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
110
111 G4bool MCTruthTrackingAction::TrackToBeStored(const G4Track* track)
112 {
113 MCTruthConfig* config = MCTruthManager::GetInstance()->GetConfig();
114
115 // check energy
116 if (fmom.e() > config->GetMinE()) return true;
117
118 // particle type
119 std::vector<G4int> types = config->GetParticleTypes();
120
121 if (std::find(types.begin(), types.end(), track->GetDefinition()->GetPDGEncoding())
122 != types.end())
123 return true;
124
125 // creator process
126
127 // etc...
128
129 return false;
130 }
131
132 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....
133