| 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/particleGun/src/Track 26 /// \file eventgenerator/particleGun/src/TrackingAction.cc
27 /// \brief Implementation of the TrackingActio 27 /// \brief Implementation of the TrackingAction class
28 // 28 //
29 // 29 //
30 // << 30 //
31 //....oooOO0OOooo........oooOO0OOooo........oo 31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
32 //....oooOO0OOooo........oooOO0OOooo........oo 32 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
33 33
34 #include "TrackingAction.hh" 34 #include "TrackingAction.hh"
35 <<
36 #include "HistoManager.hh" <<
37 #include "PrimaryGeneratorAction.hh" 35 #include "PrimaryGeneratorAction.hh"
38 #include "PrimaryGeneratorAction2.hh" 36 #include "PrimaryGeneratorAction2.hh"
39 #include "PrimaryGeneratorAction3.hh" 37 #include "PrimaryGeneratorAction3.hh"
40 #include "PrimaryGeneratorAction4.hh" 38 #include "PrimaryGeneratorAction4.hh"
41 #include "RunAction.hh" 39 #include "RunAction.hh"
>> 40 #include "HistoManager.hh"
42 41
43 #include "G4PhysicalConstants.hh" <<
44 #include "G4Track.hh" 42 #include "G4Track.hh"
>> 43 #include "G4PhysicalConstants.hh"
45 44
46 //....oooOO0OOooo........oooOO0OOooo........oo 45 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
47 46
48 TrackingAction::TrackingAction(PrimaryGenerato << 47 TrackingAction::TrackingAction(PrimaryGeneratorAction* prim)
49 { << 48 :G4UserTrackingAction(), fPrimary(prim)
50 // parameters for generator action #3 << 49 {
>> 50 // parameters for generator action #3
51 fNewUz = fPrimary->GetAction3()->GetNewUz(); 51 fNewUz = fPrimary->GetAction3()->GetNewUz();
52 } 52 }
53 53
54 //....oooOO0OOooo........oooOO0OOooo........oo 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
55 55
56 void TrackingAction::PreUserTrackingAction(con 56 void TrackingAction::PreUserTrackingAction(const G4Track* track)
57 { << 57 {
58 G4int selectedGeneratorAction = fPrimary->Ge << 58 G4int selectedGeneratorAction = fPrimary->GetSelectedAction();
59 G4AnalysisManager* analysis = G4AnalysisMana << 59 G4AnalysisManager* analysis = G4AnalysisManager::Instance();
60 G4int id = 0; << 60 G4int id = 0;
61 << 61
62 if (selectedGeneratorAction == 2) { << 62 if(selectedGeneratorAction==2)
63 // energy spectrum << 63 {
64 // << 64 //energy spectrum
65 id = 1; << 65 //
66 analysis->FillH1(id, track->GetKineticEner << 66 id = 1;
67 } << 67 analysis->FillH1(id, track->GetKineticEnergy());
68 << 68 }
69 else if (selectedGeneratorAction == 0) { << 69
70 // particle direction: cos(alpha) << 70 else if(selectedGeneratorAction==0)
71 // << 71 {
72 id = 5; << 72 //particle direction: cos(alpha)
73 G4ThreeVector um = track->GetMomentumDirec << 73 //
74 G4double cosalpha = um.z(); << 74 id = 5;
75 analysis->FillH1(id, cosalpha); << 75 G4ThreeVector um = track->GetMomentumDirection();
76 << 76 G4double cosalpha = um.z();
77 // particle direction: psi << 77 analysis->FillH1(id, cosalpha);
78 // << 78
79 id = 6; << 79 //particle direction: psi
80 G4double psi = std::atan2(um.y(), um.x()); << 80 //
81 if (psi < 0.) psi += twopi; << 81 id = 6;
82 analysis->FillH1(id, psi); << 82 G4double psi = std::atan2(um.y(), um.x());
83 } << 83 if (psi < 0.) psi += twopi;
84 else if (selectedGeneratorAction == 3) { << 84 analysis->FillH1(id, psi);
85 // particle direction in local frame: cos( << 85 }
86 // << 86 else if(selectedGeneratorAction==3)
87 id = 5; << 87 {
88 G4ThreeVector um = track->GetMomentumDirec << 88 //particle direction in local frame: cos(alpha)
89 G4double cosalpha = fNewUz * um; << 89 //
90 analysis->FillH1(id, cosalpha); << 90 id = 5;
91 << 91 G4ThreeVector um = track->GetMomentumDirection();
92 // particle direction in local frame: psi << 92 G4double cosalpha = fNewUz*um;
93 // << 93 analysis->FillH1(id, cosalpha);
94 id = 6; << 94
95 // complete local frame << 95 //particle direction in local frame: psi
96 G4ThreeVector u1(1., 0., 0.); << 96 //
97 u1.rotateUz(fNewUz); << 97 id = 6;
98 G4ThreeVector u2(0., 1., 0.); << 98 // complete local frame
99 u2.rotateUz(fNewUz); << 99 G4ThreeVector u1(1.,0.,0.); u1.rotateUz(fNewUz);
100 // << 100 G4ThreeVector u2(0.,1.,0.); u2.rotateUz(fNewUz);
101 G4double psi = std::atan2(u2 * um, u1 * um << 101 //
102 if (psi < 0.) psi += twopi; << 102 G4double psi = std::atan2(u2*um, u1*um);
103 analysis->FillH1(id, psi); << 103 if (psi < 0.) psi += twopi;
104 } << 104 analysis->FillH1(id, psi);
105 << 105 }
106 else if (selectedGeneratorAction == 4) { << 106
107 G4ThreeVector vertex = track->GetVertexPos << 107 else if(selectedGeneratorAction==4)
108 G4double r = vertex.mag(); << 108 {
109 if (r <= 0.0) return; << 109 G4ThreeVector vertex = track->GetVertexPosition();
110 // local frame : new uz = ur << 110 G4double r = vertex.mag();
111 G4ThreeVector ur = vertex / r; << 111 if (r <= 0.0) return;
112 << 112 //local frame : new uz = ur
113 // vertex position. radial distribution dN << 113 G4ThreeVector ur = vertex/r;
114 // << 114
115 id = 2; << 115 //vertex position. radial distribution dN/dv = f(r)
116 G4double dr = analysis->GetH1Width(id); << 116 //
117 G4double dv = 2 * twopi * r * r * dr; << 117 id = 2;
118 if (dv > 0.) analysis->FillH1(id, r, 1. / << 118 G4double dr = analysis->GetH1Width(id);
119 << 119 G4double dv = 2*twopi*r*r*dr;
120 // vertex position: cos(theta) << 120 if (dv > 0.) analysis->FillH1(id, r, 1./dv);
121 // << 121
122 id = 3; << 122 //vertex position: cos(theta)
123 G4double costheta = ur.z(); << 123 //
124 analysis->FillH1(id, costheta); << 124 id = 3;
125 << 125 G4double costheta = ur.z();
126 // vertex position: phi << 126 analysis->FillH1(id, costheta);
127 // << 127
128 id = 4; << 128 //vertex position: phi
129 G4double phi = std::atan2(ur.y(), ur.x()); << 129 //
130 if (phi < 0.) phi += twopi; << 130 id = 4;
131 analysis->FillH1(id, phi); << 131 G4double phi = std::atan2(ur.y(), ur.x());
132 << 132 if (phi < 0.) phi += twopi;
133 // particle direction in local frame: cos( << 133 analysis->FillH1(id, phi);
134 // << 134
135 id = 5; << 135 //particle direction in local frame: cos(alpha)
136 G4ThreeVector um = track->GetMomentumDirec << 136 //
137 G4double cosalpha = ur * um; << 137 id = 5;
138 analysis->FillH1(id, cosalpha); << 138 G4ThreeVector um = track->GetMomentumDirection();
139 << 139 G4double cosalpha = ur*um;
140 // particle direction in local frame: psi << 140 analysis->FillH1(id, cosalpha);
141 // << 141
142 id = 6; << 142 //particle direction in local frame: psi
143 // complete local frame << 143 //
144 G4ThreeVector u1(1., 0., 0.); << 144 id = 6;
145 u1.rotateUz(ur); << 145 // complete local frame
146 G4ThreeVector u2(0., 1., 0.); << 146 G4ThreeVector u1(1.,0.,0.); u1.rotateUz(ur);
147 u2.rotateUz(ur); << 147 G4ThreeVector u2(0.,1.,0.); u2.rotateUz(ur);
148 // << 148 //
149 G4double psi = std::atan2(u2 * um, u1 * um << 149 G4double psi = std::atan2(u2*um, u1*um);
150 if (psi < 0.) psi += twopi; << 150 if (psi < 0.) psi += twopi;
151 analysis->FillH1(id, psi); << 151 analysis->FillH1(id, psi);
152 } << 152 }
153 } 153 }
154 154
155 //....oooOO0OOooo........oooOO0OOooo........oo 155 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
156 156
157 void TrackingAction::PostUserTrackingAction(co << 157 void TrackingAction::PostUserTrackingAction(const G4Track*)
>> 158 { }
158 159
159 //....oooOO0OOooo........oooOO0OOooo........oo 160 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
>> 161
160 162