| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 /// \file eventgenerator/HepMC/MCTruth/src/MCT
27 /// \brief Implementation of the MCTruthTracki
28 //
29 //
30 //
31 //
32 // -------------------------------------------
33 // GEANT 4 - MCTruthTrackingAction class
34 // -------------------------------------------
35 //
36 // Author: Witold POKORSKI (Witold.Pokorski@ce
37 //
38 // -------------------------------------------
39 //
40 //....oooOO0OOooo........oooOO0OOooo........oo
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........oo
53
54 MCTruthTrackingAction::MCTruthTrackingAction()
55
56 //....oooOO0OOooo........oooOO0OOooo........oo
57
58 MCTruthTrackingAction::~MCTruthTrackingAction(
59
60 //....oooOO0OOooo........oooOO0OOooo........oo
61
62 void MCTruthTrackingAction::PreUserTrackingAct
63 {
64 fmom = G4LorentzVector(track->GetMomentum(),
65
66 if (!track->GetUserInformation()) {
67 G4VUserTrackInformation* mcinf = new MCTru
68 fpTrackingManager->SetUserTrackInformation
69 }
70 }
71
72 //....oooOO0OOooo........oooOO0OOooo........oo
73
74 void MCTruthTrackingAction::PostUserTrackingAc
75 {
76 G4LorentzVector prodpos(track->GetGlobalTime
77 track->GetVertexPosi
78 G4LorentzVector endpos(track->GetGlobalTime(
79
80 // here (?) make all different checks to dec
81 //
82 if (TrackToBeStored(track)) {
83 MCTruthTrackInformation* mcinf = (MCTruthT
84
85 MCTruthManager::GetInstance()->AddParticle
86 fmom, prodpos, endpos, track->GetDefinit
87 track->GetParentID(), mcinf->GetDirectPa
88 }
89 else {
90 // If track is not to be stored, propagate
91 // to its secondaries
92 //
93 G4TrackVector* childrens = fpTrackingManag
94
95 for (unsigned int index = 0; index < child
96 G4Track* tr = (*childrens)[index];
97 tr->SetParentID(track->GetParentID());
98
99 // set the flag saying that the direct m
100 //
101 MCTruthTrackInformation* mcinf = (MCTrut
102 if (!mcinf) tr->SetUserInformation(mcinf
103
104 mcinf->SetDirectParent(false);
105 }
106 }
107 }
108
109 //....oooOO0OOooo........oooOO0OOooo........oo
110
111 G4bool MCTruthTrackingAction::TrackToBeStored(
112 {
113 MCTruthConfig* config = MCTruthManager::GetI
114
115 // check energy
116 if (fmom.e() > config->GetMinE()) return tru
117
118 // particle type
119 std::vector<G4int> types = config->GetPartic
120
121 if (std::find(types.begin(), types.end(), tr
122 != types.end())
123 return true;
124
125 // creator process
126
127 // etc...
128
129 return false;
130 }
131
132 //....oooOO0OOooo........oooOO0OOooo........oo
133