| 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 <<
27 /// \brief Implementation of the MCTruthTracki <<
28 // <<
29 // 26 //
>> 27 // $Id: MCTruthTrackingAction.cc,v 1.1 2006/11/22 14:51:30 gcosmo Exp $
>> 28 // GEANT4 tag $Name: geant4-09-02 $
30 // 29 //
31 // 30 //
32 // ------------------------------------------- 31 // --------------------------------------------------------------
33 // GEANT 4 - MCTruthTrackingAction class 32 // GEANT 4 - MCTruthTrackingAction class
34 // ------------------------------------------- 33 // --------------------------------------------------------------
35 // 34 //
36 // Author: Witold POKORSKI (Witold.Pokorski@ce 35 // Author: Witold POKORSKI (Witold.Pokorski@cern.ch)
37 // 36 //
38 // ------------------------------------------- 37 // --------------------------------------------------------------
39 // <<
40 //....oooOO0OOooo........oooOO0OOooo........oo <<
41 <<
42 #include "MCTruthTrackingAction.hh" <<
43 38
44 #include "MCTruthTrackInformation.hh" << 39 #include <iostream>
45 40
46 #include "G4Track.hh" 41 #include "G4Track.hh"
47 #include "G4TrackVector.hh" 42 #include "G4TrackVector.hh"
48 #include "G4TrackingManager.hh" 43 #include "G4TrackingManager.hh"
>> 44 #include "MCTruthTrackingAction.hh"
>> 45 #include "MCTruthTrackInformation.hh"
49 46
50 #include <iostream> << 47 MCTruthTrackingAction::MCTruthTrackingAction()
51 << 48 {}
52 //....oooOO0OOooo........oooOO0OOooo........oo <<
53 <<
54 MCTruthTrackingAction::MCTruthTrackingAction() <<
55 <<
56 //....oooOO0OOooo........oooOO0OOooo........oo <<
57 <<
58 MCTruthTrackingAction::~MCTruthTrackingAction( <<
59 49
60 //....oooOO0OOooo........oooOO0OOooo........oo << 50 MCTruthTrackingAction::~MCTruthTrackingAction()
>> 51 {}
61 52
62 void MCTruthTrackingAction::PreUserTrackingAct 53 void MCTruthTrackingAction::PreUserTrackingAction(const G4Track* track)
63 { 54 {
64 fmom = G4LorentzVector(track->GetMomentum(), 55 fmom = G4LorentzVector(track->GetMomentum(), track->GetTotalEnergy());
65 << 56
66 if (!track->GetUserInformation()) { << 57 if(!track->GetUserInformation())
>> 58 {
67 G4VUserTrackInformation* mcinf = new MCTru 59 G4VUserTrackInformation* mcinf = new MCTruthTrackInformation;
68 fpTrackingManager->SetUserTrackInformation 60 fpTrackingManager->SetUserTrackInformation(mcinf);
69 } 61 }
70 } 62 }
71 63
72 //....oooOO0OOooo........oooOO0OOooo........oo <<
73 <<
74 void MCTruthTrackingAction::PostUserTrackingAc 64 void MCTruthTrackingAction::PostUserTrackingAction(const G4Track* track)
75 { 65 {
>> 66
76 G4LorentzVector prodpos(track->GetGlobalTime 67 G4LorentzVector prodpos(track->GetGlobalTime() - track->GetLocalTime(),
77 track->GetVertexPosi << 68 track->GetVertexPosition());
78 G4LorentzVector endpos(track->GetGlobalTime( 69 G4LorentzVector endpos(track->GetGlobalTime(), track->GetPosition());
79 70
80 // here (?) make all different checks to dec 71 // here (?) make all different checks to decide whether to store the particle
81 // 72 //
82 if (TrackToBeStored(track)) { << 73 if (trackToBeStored(track))
83 MCTruthTrackInformation* mcinf = (MCTruthT << 74 {
84 << 75 MCTruthTrackInformation* mcinf =
85 MCTruthManager::GetInstance()->AddParticle << 76 (MCTruthTrackInformation*) track->GetUserInformation();
86 fmom, prodpos, endpos, track->GetDefinit << 77
87 track->GetParentID(), mcinf->GetDirectPa << 78 MCTruthManager::GetInstance()->
>> 79 AddParticle(fmom, prodpos, endpos,
>> 80 track->GetDefinition()->GetPDGEncoding(),
>> 81 track->GetTrackID(),
>> 82 track->GetParentID(), mcinf->GetDirectParent());
88 } 83 }
89 else { << 84 else
>> 85 {
90 // If track is not to be stored, propagate 86 // If track is not to be stored, propagate it's parent ID (stored)
91 // to its secondaries 87 // to its secondaries
92 // 88 //
93 G4TrackVector* childrens = fpTrackingManag << 89 G4TrackVector* childrens = fpTrackingManager->GimmeSecondaries() ;
94 90
95 for (unsigned int index = 0; index < child << 91 for( unsigned int index = 0 ; index < childrens->size() ; ++index )
96 G4Track* tr = (*childrens)[index]; << 92 {
97 tr->SetParentID(track->GetParentID()); << 93 G4Track* tr = (*childrens)[index] ;
>> 94 tr->SetParentID( track->GetParentID() );
98 95
99 // set the flag saying that the direct m 96 // set the flag saying that the direct mother is not stored
100 // 97 //
101 MCTruthTrackInformation* mcinf = (MCTrut << 98 MCTruthTrackInformation* mcinf =
102 if (!mcinf) tr->SetUserInformation(mcinf << 99 (MCTruthTrackInformation*) tr->GetUserInformation();
>> 100 if(!mcinf)
>> 101 tr->SetUserInformation(mcinf = new MCTruthTrackInformation);
103 102
104 mcinf->SetDirectParent(false); << 103 mcinf->SetDirectParent(false);
105 } << 104 }
106 } 105 }
107 } 106 }
108 107
109 //....oooOO0OOooo........oooOO0OOooo........oo << 108 G4bool MCTruthTrackingAction::trackToBeStored(const G4Track* track)
110 <<
111 G4bool MCTruthTrackingAction::TrackToBeStored( <<
112 { 109 {
113 MCTruthConfig* config = MCTruthManager::GetI 110 MCTruthConfig* config = MCTruthManager::GetInstance()->GetConfig();
114 << 111
115 // check energy 112 // check energy
116 if (fmom.e() > config->GetMinE()) return tru 113 if (fmom.e() > config->GetMinE()) return true;
117 << 114
118 // particle type 115 // particle type
119 std::vector<G4int> types = config->GetPartic 116 std::vector<G4int> types = config->GetParticleTypes();
120 << 117
121 if (std::find(types.begin(), types.end(), tr << 118 if(std::find( types.begin(), types.end(),
122 != types.end()) << 119 track->GetDefinition()->GetPDGEncoding())
123 return true; << 120 != types.end()) return true;
124 121
125 // creator process 122 // creator process
126 123
127 // etc... 124 // etc...
128 << 125
129 return false; 126 return false;
130 } 127 }
131 <<
132 //....oooOO0OOooo........oooOO0OOooo........oo <<
133 128