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 | // |
---|
27 | // |
---|
28 | // ------------------------------------------------------------ |
---|
29 | // GEANT 4 class header file --- Copyright CERN 1998 |
---|
30 | // CERN Geneva Switzerland |
---|
31 | // |
---|
32 | // History: first implementation, based on object model of |
---|
33 | // 2nd December 1995, G.Cosmo |
---|
34 | // ------------ G4GHEKinematicsVector utility class ------ |
---|
35 | // by Larry Felawka (TRIUMF), March 1997 |
---|
36 | // E-mail: felawka@alph04.triumf.ca |
---|
37 | // ************************************************************ |
---|
38 | //----------------------------------------------------------------------------- |
---|
39 | |
---|
40 | // Store, Retrieve and manipulate particle data. |
---|
41 | // Based on "G4GHEVector" class of H. Fesefeldt. |
---|
42 | |
---|
43 | #ifndef G4GHEKinematicsVector_h |
---|
44 | #define G4GHEKinematicsVector_h 1 |
---|
45 | |
---|
46 | #include "G4ios.hh" |
---|
47 | |
---|
48 | #include <CLHEP/Units/PhysicalConstants.h> |
---|
49 | |
---|
50 | class G4GHEKinematicsVector |
---|
51 | { |
---|
52 | public: |
---|
53 | inline |
---|
54 | G4GHEKinematicsVector() |
---|
55 | { |
---|
56 | momentum.setX( 0.0 ); |
---|
57 | momentum.setY( 0.0 ); |
---|
58 | momentum.setZ( 0.0 ); |
---|
59 | energy = 0.0; |
---|
60 | kineticEnergy = 0.0; |
---|
61 | mass = 0.0; |
---|
62 | charge = 0.0; |
---|
63 | timeOfFlight = 0.0; |
---|
64 | side = 0; |
---|
65 | flag = false; |
---|
66 | code = 0; |
---|
67 | particleDef = NULL; |
---|
68 | } |
---|
69 | |
---|
70 | ~G4GHEKinematicsVector() {} |
---|
71 | |
---|
72 | inline |
---|
73 | G4GHEKinematicsVector( const G4GHEKinematicsVector & p ) |
---|
74 | { |
---|
75 | momentum.setX( p.momentum.x() ); |
---|
76 | momentum.setY( p.momentum.y() ); |
---|
77 | momentum.setZ( p.momentum.z() ); |
---|
78 | energy = p.energy; |
---|
79 | kineticEnergy = p.kineticEnergy; |
---|
80 | mass = p.mass; |
---|
81 | charge = p.charge; |
---|
82 | timeOfFlight = p.timeOfFlight; |
---|
83 | side = p.side; |
---|
84 | flag = p.flag; |
---|
85 | code = p.code; |
---|
86 | particleDef = p.particleDef; |
---|
87 | } |
---|
88 | |
---|
89 | inline |
---|
90 | G4GHEKinematicsVector & operator = ( const G4GHEKinematicsVector & p ) |
---|
91 | { |
---|
92 | momentum.setX( p.momentum.x() ); |
---|
93 | momentum.setY( p.momentum.y() ); |
---|
94 | momentum.setZ( p.momentum.z() ); |
---|
95 | energy = p.energy; |
---|
96 | kineticEnergy = p.kineticEnergy; |
---|
97 | mass = p.mass; |
---|
98 | charge = p.charge; |
---|
99 | timeOfFlight = p.timeOfFlight; |
---|
100 | side = p.side; |
---|
101 | flag = p.flag; |
---|
102 | code = p.code; |
---|
103 | particleDef = p.particleDef; |
---|
104 | return *this; |
---|
105 | } |
---|
106 | |
---|
107 | inline |
---|
108 | void SetMomentum( G4ParticleMomentum mom ) { momentum = mom; return; }; |
---|
109 | |
---|
110 | inline |
---|
111 | void SetMomentumAndUpdate( G4ParticleMomentum mom ) |
---|
112 | { |
---|
113 | momentum = mom; |
---|
114 | energy = std::sqrt(mass*mass + momentum.mag2()); |
---|
115 | kineticEnergy = std::max(0.,energy - mass); |
---|
116 | return; |
---|
117 | } |
---|
118 | |
---|
119 | inline const |
---|
120 | G4ParticleMomentum GetMomentum() const { return momentum; } |
---|
121 | |
---|
122 | inline |
---|
123 | void SetMomentum( G4double x, G4double y, G4double z) |
---|
124 | { |
---|
125 | momentum.setX( x ); |
---|
126 | momentum.setY( y ); |
---|
127 | momentum.setZ( z ); |
---|
128 | return; |
---|
129 | } |
---|
130 | |
---|
131 | inline |
---|
132 | void SetMomentumAndUpdate( G4double x, G4double y, G4double z ) |
---|
133 | { |
---|
134 | momentum.setX( x ); |
---|
135 | momentum.setY( y ); |
---|
136 | momentum.setZ( z ); |
---|
137 | energy = std::sqrt(mass*mass + momentum.mag2()); |
---|
138 | kineticEnergy = std::max(0.,energy-mass); |
---|
139 | return; |
---|
140 | } |
---|
141 | |
---|
142 | inline |
---|
143 | void SetMomentum( G4double x, G4double y ) |
---|
144 | { |
---|
145 | momentum.setX( x ); |
---|
146 | momentum.setY( y ); |
---|
147 | return; |
---|
148 | } |
---|
149 | |
---|
150 | inline |
---|
151 | void SetMomentumAndUpdate( G4double x, G4double y ) |
---|
152 | { |
---|
153 | momentum.setX( x ); |
---|
154 | momentum.setY( y ); |
---|
155 | energy = std::sqrt(mass*mass + momentum.mag2()); |
---|
156 | kineticEnergy = std::max(0.,energy-mass); |
---|
157 | return; |
---|
158 | } |
---|
159 | |
---|
160 | inline |
---|
161 | void SetMomentum( G4double z ) |
---|
162 | { |
---|
163 | momentum.setZ( z ); |
---|
164 | return; |
---|
165 | } |
---|
166 | |
---|
167 | inline |
---|
168 | void SetMomentumAndUpdate( G4double z ) |
---|
169 | { |
---|
170 | momentum.setZ( z ); |
---|
171 | energy = std::sqrt(mass*mass + momentum.mag2()); |
---|
172 | kineticEnergy = std::max(0.,energy-mass); |
---|
173 | return; |
---|
174 | } |
---|
175 | |
---|
176 | inline |
---|
177 | void SetEnergy( G4double e ) { energy = e; return; } |
---|
178 | |
---|
179 | inline |
---|
180 | void SetEnergyAndUpdate( G4double e ) |
---|
181 | { |
---|
182 | if (e <= mass) |
---|
183 | { |
---|
184 | energy = mass; |
---|
185 | kineticEnergy = 0.; |
---|
186 | momentum.setX( 0.); |
---|
187 | momentum.setY( 0.); |
---|
188 | momentum.setZ( 0.); |
---|
189 | } |
---|
190 | else |
---|
191 | { |
---|
192 | energy = e; |
---|
193 | kineticEnergy = energy - mass; |
---|
194 | G4double momold = momentum.mag(); |
---|
195 | G4double momnew = std::sqrt(energy*energy - mass*mass); |
---|
196 | if (momold == 0.) |
---|
197 | { |
---|
198 | G4double cost = 1.0- 2.0*G4UniformRand(); |
---|
199 | G4double sint = std::sqrt(1. - cost*cost); |
---|
200 | G4double phi = twopi* G4UniformRand(); |
---|
201 | momentum.setX( momnew * sint * std::cos(phi)); |
---|
202 | momentum.setY( momnew * sint * std::sin(phi)); |
---|
203 | momentum.setZ( momnew * cost); |
---|
204 | } |
---|
205 | else |
---|
206 | { |
---|
207 | momnew /= momold; |
---|
208 | momentum.setX(momentum.x()*momnew); |
---|
209 | momentum.setY(momentum.y()*momnew); |
---|
210 | momentum.setZ(momentum.z()*momnew); |
---|
211 | } |
---|
212 | } |
---|
213 | return; |
---|
214 | } |
---|
215 | |
---|
216 | inline |
---|
217 | void SetKineticEnergy( G4double ekin ) { kineticEnergy = ekin; return; } |
---|
218 | |
---|
219 | inline |
---|
220 | void SetKineticEnergyAndUpdate(G4double ekin) |
---|
221 | { |
---|
222 | if (ekin <= 0.) |
---|
223 | { |
---|
224 | energy = mass; |
---|
225 | kineticEnergy = 0.; |
---|
226 | momentum.setX( 0.); |
---|
227 | momentum.setY( 0.); |
---|
228 | momentum.setZ( 0.); |
---|
229 | } |
---|
230 | else |
---|
231 | { |
---|
232 | energy = ekin + mass; |
---|
233 | kineticEnergy = ekin; |
---|
234 | G4double momold = momentum.mag(); |
---|
235 | G4double momnew = std::sqrt(energy*energy - mass*mass); |
---|
236 | if (momold == 0.) |
---|
237 | { |
---|
238 | G4double cost = 1.0-2.0*G4UniformRand(); |
---|
239 | G4double sint = std::sqrt(1. - cost*cost); |
---|
240 | G4double phi = twopi* G4UniformRand(); |
---|
241 | momentum.setX( momnew * sint * std::cos(phi)); |
---|
242 | momentum.setY( momnew * sint * std::sin(phi)); |
---|
243 | momentum.setZ( momnew * cost); |
---|
244 | } |
---|
245 | else |
---|
246 | { |
---|
247 | momnew /= momold; |
---|
248 | momentum.setX(momentum.x()*momnew); |
---|
249 | momentum.setY(momentum.y()*momnew); |
---|
250 | momentum.setZ(momentum.z()*momnew); |
---|
251 | } |
---|
252 | } |
---|
253 | return; |
---|
254 | } |
---|
255 | |
---|
256 | inline |
---|
257 | G4double GetEnergy() {return energy;} |
---|
258 | |
---|
259 | inline |
---|
260 | G4double GetKineticEnergy() {return kineticEnergy;} |
---|
261 | |
---|
262 | inline |
---|
263 | void SetMass( G4double m ) { mass = m; return; } |
---|
264 | |
---|
265 | inline |
---|
266 | void SetMassAndUpdate( G4double m ) |
---|
267 | { |
---|
268 | kineticEnergy = std::max(0., energy - m); |
---|
269 | mass = m; |
---|
270 | energy = kineticEnergy + mass; |
---|
271 | G4double momnew = std::sqrt(std::max(0., energy*energy - mass*mass)); |
---|
272 | if ( momnew == 0.0) |
---|
273 | { |
---|
274 | momentum.setX( 0.0 ); |
---|
275 | momentum.setY( 0.0 ); |
---|
276 | momentum.setZ( 0.0 ); |
---|
277 | } |
---|
278 | else |
---|
279 | { |
---|
280 | G4double momold = momentum.mag(); |
---|
281 | if (momold == 0.) |
---|
282 | { |
---|
283 | G4double cost = 1.-2.*G4UniformRand(); |
---|
284 | G4double sint = std::sqrt(1.-cost*cost); |
---|
285 | G4double phi = twopi*G4UniformRand(); |
---|
286 | momentum.setX( momnew*sint*std::cos(phi)); |
---|
287 | momentum.setY( momnew*sint*std::sin(phi)); |
---|
288 | momentum.setZ( momnew*cost); |
---|
289 | } |
---|
290 | else |
---|
291 | { |
---|
292 | momnew /= momold; |
---|
293 | momentum.setX( momentum.x()*momnew ); |
---|
294 | momentum.setY( momentum.y()*momnew ); |
---|
295 | momentum.setZ( momentum.z()*momnew ); |
---|
296 | } |
---|
297 | } |
---|
298 | return; |
---|
299 | } |
---|
300 | |
---|
301 | inline |
---|
302 | G4double GetMass() { return mass; } |
---|
303 | |
---|
304 | inline |
---|
305 | void SetCharge( G4double c ) { charge = c; return; } |
---|
306 | |
---|
307 | inline |
---|
308 | G4double GetCharge() {return charge; } |
---|
309 | |
---|
310 | inline |
---|
311 | void SetTOF( G4double t ) { timeOfFlight = t; return; } |
---|
312 | |
---|
313 | inline |
---|
314 | G4double GetTOF() { return timeOfFlight; } |
---|
315 | |
---|
316 | inline |
---|
317 | void SetSide( G4int s ) { side = s; return; } |
---|
318 | |
---|
319 | inline |
---|
320 | G4int GetSide() { return side; } |
---|
321 | |
---|
322 | inline |
---|
323 | void setFlag( G4bool f ) { flag = f; return; } |
---|
324 | |
---|
325 | inline |
---|
326 | G4bool getFlag() { return flag; } |
---|
327 | |
---|
328 | inline |
---|
329 | void SetCode( G4int c ) { code = c; return; } |
---|
330 | |
---|
331 | inline |
---|
332 | void SetParticleDef( G4ParticleDefinition * c ) { particleDef = c; return; } |
---|
333 | |
---|
334 | inline |
---|
335 | G4int GetCode() { return code; } |
---|
336 | |
---|
337 | inline |
---|
338 | G4ParticleDefinition * GetParticleDef() { return particleDef; } |
---|
339 | |
---|
340 | inline |
---|
341 | void SetZero() |
---|
342 | { |
---|
343 | momentum.setX( 0.0 ); |
---|
344 | momentum.setY( 0.0 ); |
---|
345 | momentum.setZ( 0.0 ); |
---|
346 | energy = 0.0; |
---|
347 | kineticEnergy = 0.0; |
---|
348 | mass = 0.0; |
---|
349 | charge = 0.0; |
---|
350 | timeOfFlight = 0.0; |
---|
351 | side = 0; |
---|
352 | flag = false; |
---|
353 | code = 0; |
---|
354 | particleDef = NULL; |
---|
355 | } |
---|
356 | |
---|
357 | inline |
---|
358 | void Add( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2 ) |
---|
359 | { |
---|
360 | momentum = p1.momentum + p2.momentum; |
---|
361 | energy = p1.energy + p2.energy; |
---|
362 | G4double b = energy*energy - momentum.mag2(); |
---|
363 | if( b < 0 ) |
---|
364 | mass = -1. * std::sqrt( -b ); |
---|
365 | else |
---|
366 | mass = std::sqrt( b ); |
---|
367 | kineticEnergy = std::max(0.,energy - mass); |
---|
368 | charge = p1.charge + p2.charge; |
---|
369 | code = p1.code + p2.code; |
---|
370 | particleDef = p1.particleDef; |
---|
371 | } |
---|
372 | |
---|
373 | inline |
---|
374 | void Sub( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2 ) |
---|
375 | { |
---|
376 | momentum = p1.momentum - p2.momentum; |
---|
377 | energy = p1.energy - p2.energy; |
---|
378 | G4double b = energy*energy - momentum.mag2(); |
---|
379 | if( b < 0 ) |
---|
380 | mass = -1. * std::sqrt( -b ); |
---|
381 | else |
---|
382 | mass = std::sqrt( b ); |
---|
383 | kineticEnergy = std::max(0.,energy - mass); |
---|
384 | charge = p1.charge - p2.charge; |
---|
385 | code = p1.code - p2.code; |
---|
386 | particleDef = p1.particleDef; |
---|
387 | } |
---|
388 | |
---|
389 | inline |
---|
390 | void Lor( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2 ) |
---|
391 | { |
---|
392 | G4double a; |
---|
393 | a = ( p1.momentum.dot(p2.momentum)/(p2.energy+p2.mass) - p1.energy ) / p2.mass; |
---|
394 | momentum.setX( p1.momentum.x()+a*p2.momentum.x() ); |
---|
395 | momentum.setY( p1.momentum.y()+a*p2.momentum.y() ); |
---|
396 | momentum.setZ( p1.momentum.z()+a*p2.momentum.z() ); |
---|
397 | energy = std::sqrt( sqr(p1.mass) + momentum.mag2() ); |
---|
398 | mass = p1.mass; |
---|
399 | kineticEnergy = std::max(0.,energy - mass); |
---|
400 | timeOfFlight = p1.timeOfFlight; |
---|
401 | side = p1.side; |
---|
402 | flag = p1.flag; |
---|
403 | code = p1.code; |
---|
404 | particleDef = p1.particleDef; |
---|
405 | } |
---|
406 | |
---|
407 | inline |
---|
408 | G4double CosAng( const G4GHEKinematicsVector & p ) |
---|
409 | { |
---|
410 | G4double a = std::sqrt( momentum.mag2() * p.momentum.mag2() ); |
---|
411 | if( a != 0.0 ) |
---|
412 | { |
---|
413 | a = (momentum.x()*p.momentum.x() + |
---|
414 | momentum.y()*p.momentum.y() + |
---|
415 | momentum.z()*p.momentum.z()) / a; |
---|
416 | if( std::fabs(a) > 1.0 ) a<0.0 ? a=-1.0 : a=1.0; |
---|
417 | } |
---|
418 | return a; |
---|
419 | } |
---|
420 | inline |
---|
421 | G4double Ang(const G4GHEKinematicsVector & p ) |
---|
422 | { |
---|
423 | G4double a = std::sqrt( momentum.mag2() * p.momentum.mag2() ); |
---|
424 | if( a != 0.0 ) |
---|
425 | { |
---|
426 | a = (momentum.x()*p.momentum.x() + |
---|
427 | momentum.y()*p.momentum.y() + |
---|
428 | momentum.z()*p.momentum.z()) / a; |
---|
429 | if( std::fabs(a) > 1.0 ) a<0.0 ? a=-1.0 : a=1.0; |
---|
430 | } |
---|
431 | return std::acos(a); |
---|
432 | } |
---|
433 | |
---|
434 | inline |
---|
435 | G4double Dot4( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
436 | { |
---|
437 | return ( p1.energy * p2.energy |
---|
438 | - p1.momentum.x() * p2.momentum.x() |
---|
439 | - p1.momentum.y() * p2.momentum.y() |
---|
440 | - p1.momentum.z() * p2.momentum.z() ); |
---|
441 | } |
---|
442 | |
---|
443 | inline |
---|
444 | G4double Impu( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
445 | { |
---|
446 | return ( - sqr( p1.energy - p2.energy) |
---|
447 | + sqr(p1.momentum.x() - p2.momentum.x()) |
---|
448 | + sqr(p1.momentum.y() - p2.momentum.y()) |
---|
449 | + sqr(p1.momentum.z() - p2.momentum.z()) ); |
---|
450 | } |
---|
451 | |
---|
452 | inline |
---|
453 | void Add3( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
454 | { |
---|
455 | momentum.setX( p1.momentum.x() + p2.momentum.x()); |
---|
456 | momentum.setY( p1.momentum.y() + p2.momentum.y()); |
---|
457 | momentum.setZ( p1.momentum.z() + p2.momentum.z()); |
---|
458 | return; |
---|
459 | } |
---|
460 | |
---|
461 | inline |
---|
462 | void Sub3( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
463 | { |
---|
464 | momentum.setX( p1.momentum.x() - p2.momentum.x()); |
---|
465 | momentum.setY( p1.momentum.y() - p2.momentum.y()); |
---|
466 | momentum.setZ( p1.momentum.z() - p2.momentum.z()); |
---|
467 | return; |
---|
468 | } |
---|
469 | |
---|
470 | inline |
---|
471 | void Cross( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
472 | { |
---|
473 | G4double px, py, pz; |
---|
474 | px = p1.momentum.y() * p2.momentum.z() - p1.momentum.z() * p2.momentum.y(); |
---|
475 | py = p1.momentum.z() * p2.momentum.x() - p1.momentum.x() * p2.momentum.z(); |
---|
476 | pz = p1.momentum.x() * p2.momentum.y() - p1.momentum.y() * p2.momentum.x(); |
---|
477 | momentum.setX( px ); |
---|
478 | momentum.setY( py ); |
---|
479 | momentum.setZ( pz ); |
---|
480 | return; |
---|
481 | } |
---|
482 | |
---|
483 | inline |
---|
484 | G4double Dot( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
485 | { |
---|
486 | return ( p1.momentum.x() * p2.momentum.x() |
---|
487 | + p1.momentum.y() * p2.momentum.y() |
---|
488 | + p1.momentum.z() * p2.momentum.z() ); |
---|
489 | } |
---|
490 | |
---|
491 | inline |
---|
492 | void Smul( const G4GHEKinematicsVector & p, G4double h) |
---|
493 | { |
---|
494 | momentum.setX( h * p.momentum.x()); |
---|
495 | momentum.setY( h * p.momentum.y()); |
---|
496 | momentum.setZ( h * p.momentum.z()); |
---|
497 | return; |
---|
498 | } |
---|
499 | |
---|
500 | inline |
---|
501 | void SmulAndUpdate( const G4GHEKinematicsVector & p, G4double h) |
---|
502 | { |
---|
503 | momentum.setX( h * p.momentum.x()); |
---|
504 | momentum.setY( h * p.momentum.y()); |
---|
505 | momentum.setZ( h * p.momentum.z()); |
---|
506 | mass = p.mass; |
---|
507 | energy = std::sqrt(momentum.mag2() + mass*mass); |
---|
508 | kineticEnergy = energy - mass; |
---|
509 | charge = p.charge; |
---|
510 | timeOfFlight = p.timeOfFlight; |
---|
511 | side = p.side; |
---|
512 | flag = p.flag; |
---|
513 | code = p.code; |
---|
514 | particleDef = p.particleDef; |
---|
515 | return; |
---|
516 | } |
---|
517 | |
---|
518 | inline |
---|
519 | void Norz( const G4GHEKinematicsVector & p ) |
---|
520 | { |
---|
521 | G4double a = p.momentum.mag2(); |
---|
522 | if (a > 0.0) a = 1./std::sqrt(a); |
---|
523 | momentum.setX( a * p.momentum.x() ); |
---|
524 | momentum.setY( a * p.momentum.y() ); |
---|
525 | momentum.setZ( a * p.momentum.z() ); |
---|
526 | mass = p.mass; |
---|
527 | energy = std::sqrt(momentum.mag2() + mass*mass); |
---|
528 | kineticEnergy = energy - mass; |
---|
529 | charge = p.charge; |
---|
530 | timeOfFlight = p.timeOfFlight; |
---|
531 | side = p.side; |
---|
532 | flag = p.flag; |
---|
533 | code = p.code; |
---|
534 | particleDef = p.particleDef; |
---|
535 | return; |
---|
536 | } |
---|
537 | |
---|
538 | inline |
---|
539 | G4double Length() |
---|
540 | { |
---|
541 | return momentum.mag() ; |
---|
542 | } |
---|
543 | |
---|
544 | inline |
---|
545 | void Exch( G4GHEKinematicsVector & p1) |
---|
546 | { |
---|
547 | G4GHEKinematicsVector mx = *this; |
---|
548 | // mx.momentum.SetX( momentum.x()); |
---|
549 | // mx.momentum.SetY( momentum.y()); |
---|
550 | // mx.momentum.SetZ( momentum.z()); |
---|
551 | // mx.energy = energy; |
---|
552 | // mx.kineticEnergy = kineticEnergy; |
---|
553 | // mx.mass = mass; |
---|
554 | // mx.charge = charge; |
---|
555 | // mx.timeOfFlight = timeOfFlight; |
---|
556 | // mx.side = side; |
---|
557 | // mx.flag = flag; |
---|
558 | // mx.code = code; |
---|
559 | // momentum.setX( p1.momentum.x()); |
---|
560 | // momentum.setY( p1.momentum.y()); |
---|
561 | // momentum.setZ( p1.momentum.z()); |
---|
562 | // energy = p1.energy; |
---|
563 | // kineticEnergy = p1.kineticEnergy; |
---|
564 | // mass = p1.mass; |
---|
565 | // charge = p1.charge; |
---|
566 | // timeOfFlight = p1.timeOfFlight; |
---|
567 | // side = p1.side |
---|
568 | // flag = p1.flag; |
---|
569 | // code = p1.code; |
---|
570 | *this = p1; |
---|
571 | p1 = mx; |
---|
572 | return; |
---|
573 | } |
---|
574 | |
---|
575 | inline |
---|
576 | void Defs1( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2) |
---|
577 | { |
---|
578 | G4double pt2 = sqr(p1.momentum.x()) + sqr(p1.momentum.y()); |
---|
579 | if (pt2 > 0.0) |
---|
580 | { |
---|
581 | G4double ph, px, py, pz; |
---|
582 | G4double cost = p2.momentum.z()/p2.momentum.mag(); |
---|
583 | G4double sint = 0.5 * ( std::sqrt(std::fabs((1.-cost)*(1.+cost))) |
---|
584 | + std::sqrt(pt2)/p2.momentum.mag()); |
---|
585 | (p2.momentum.y() < 0.) ? ph = 1.5*pi : ph = halfpi; |
---|
586 | if( p2.momentum.x() != 0.0) |
---|
587 | ph = std::atan2(p2.momentum.y(),p2.momentum.x()); |
---|
588 | px = cost*std::cos(ph)*p1.momentum.x() - std::sin(ph)*p1.momentum.y() |
---|
589 | + sint*std::cos(ph)*p1.momentum.z(); |
---|
590 | py = cost*std::sin(ph)*p1.momentum.x() + std::cos(ph)*p1.momentum.y() |
---|
591 | + sint*std::sin(ph)*p1.momentum.z(); |
---|
592 | pz = - sint *p1.momentum.x() |
---|
593 | + cost *p1.momentum.z(); |
---|
594 | momentum.setX( px ); |
---|
595 | momentum.setY( py ); |
---|
596 | momentum.setZ( pz ); |
---|
597 | } |
---|
598 | else |
---|
599 | { |
---|
600 | momentum = p1.momentum; |
---|
601 | } |
---|
602 | } |
---|
603 | |
---|
604 | inline |
---|
605 | void Defs( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & p2, |
---|
606 | G4GHEKinematicsVector & my, G4GHEKinematicsVector & mz ) |
---|
607 | { |
---|
608 | my = p1; |
---|
609 | mz = p2; |
---|
610 | momentum.setX( my.momentum.y()*mz.momentum.z() |
---|
611 | - my.momentum.z()*mz.momentum.y()); |
---|
612 | momentum.setY( my.momentum.z()*mz.momentum.x() |
---|
613 | - my.momentum.x()*mz.momentum.z()); |
---|
614 | momentum.setZ( my.momentum.x()*mz.momentum.y() |
---|
615 | - my.momentum.y()*mz.momentum.x()); |
---|
616 | my.momentum.setX( mz.momentum.y()*momentum.z() |
---|
617 | - mz.momentum.z()*momentum.y()); |
---|
618 | my.momentum.setY( mz.momentum.z()*momentum.x() |
---|
619 | - mz.momentum.x()*momentum.z()); |
---|
620 | my.momentum.setZ( mz.momentum.x()*momentum.y() |
---|
621 | - mz.momentum.y()*momentum.x()); |
---|
622 | G4double pp; |
---|
623 | pp = momentum.mag(); |
---|
624 | if (pp > 0.) |
---|
625 | { |
---|
626 | pp = 1./pp; |
---|
627 | momentum.setX( momentum.x()*pp ); |
---|
628 | momentum.setY( momentum.y()*pp ); |
---|
629 | momentum.setZ( momentum.z()*pp ); |
---|
630 | } |
---|
631 | pp = my.momentum.mag(); |
---|
632 | if (pp > 0.) |
---|
633 | { |
---|
634 | pp = 1./pp; |
---|
635 | my.momentum.setX( my.momentum.x()*pp ); |
---|
636 | my.momentum.setY( my.momentum.y()*pp ); |
---|
637 | my.momentum.setZ( my.momentum.z()*pp ); |
---|
638 | } |
---|
639 | pp = mz.momentum.mag(); |
---|
640 | if (pp > 0.) |
---|
641 | { |
---|
642 | pp = 1./pp; |
---|
643 | mz.momentum.setX( mz.momentum.x()*pp ); |
---|
644 | mz.momentum.setY( mz.momentum.y()*pp ); |
---|
645 | mz.momentum.setZ( mz.momentum.z()*pp ); |
---|
646 | } |
---|
647 | return; |
---|
648 | } |
---|
649 | |
---|
650 | inline |
---|
651 | void Trac( const G4GHEKinematicsVector & p1, const G4GHEKinematicsVector & mx, |
---|
652 | const G4GHEKinematicsVector & my, const G4GHEKinematicsVector & mz) |
---|
653 | { |
---|
654 | double px, py, pz; |
---|
655 | px = mx.momentum.x()*p1.momentum.x() |
---|
656 | + mx.momentum.y()*p1.momentum.y() |
---|
657 | + mx.momentum.z()*p1.momentum.z(); |
---|
658 | py = my.momentum.x()*p1.momentum.x() |
---|
659 | + my.momentum.y()*p1.momentum.y() |
---|
660 | + my.momentum.z()*p1.momentum.z(); |
---|
661 | pz = mz.momentum.x()*p1.momentum.x() |
---|
662 | + mz.momentum.y()*p1.momentum.y() |
---|
663 | + mz.momentum.z()*p1.momentum.z(); |
---|
664 | momentum.setX( px ); |
---|
665 | momentum.setY( py ); |
---|
666 | momentum.setZ( pz ); |
---|
667 | return; |
---|
668 | } |
---|
669 | |
---|
670 | inline |
---|
671 | void Print( G4int L) |
---|
672 | { |
---|
673 | G4cout << "G4GHEKinematicsVector: " |
---|
674 | << L << " " << momentum.x() << " " << momentum.y() << " " << momentum.z() << " " |
---|
675 | << energy << " " << kineticEnergy << " " << mass << " " << charge << " " |
---|
676 | << timeOfFlight << " " << side << " " << flag << " " << code << particleDef << G4endl; |
---|
677 | return; |
---|
678 | } |
---|
679 | |
---|
680 | G4ParticleMomentum momentum; |
---|
681 | G4double energy; |
---|
682 | G4double kineticEnergy; |
---|
683 | G4double mass; |
---|
684 | G4double charge; |
---|
685 | G4double timeOfFlight; |
---|
686 | G4int side; |
---|
687 | G4bool flag; |
---|
688 | G4int code; |
---|
689 | G4ParticleDefinition * particleDef; |
---|
690 | }; |
---|
691 | |
---|
692 | #endif |
---|