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