]>
Commit | Line | Data |
---|---|---|
f826d409 | 1 | //--------------------------------------------------------------------------------- |
2 | // Implementation of the AliKFParticleBase class | |
3 | // . | |
57ddc2cb | 4 | // @author S.Gorbunov, I.Kisel, I.Kulakov, M.Zyzak |
f826d409 | 5 | // @version 1.0 |
6 | // @since 13.05.07 | |
7 | // | |
8 | // Class to reconstruct and store the decayed particle parameters. | |
9 | // The method is described in CBM-SOFT note 2007-003, | |
10 | // ``Reconstruction of decayed particles based on the Kalman filter'', | |
11 | // http://www.gsi.de/documents/DOC-2007-May-14-1.pdf | |
12 | // | |
13 | // This class describes general mathematics which is used by AliKFParticle class | |
14 | // | |
15 | // -= Copyright © ALICE HLT Group =- | |
16 | //_________________________________________________________________________________ | |
17 | ||
18 | ||
19 | #include "AliKFParticleBase.h" | |
20 | #include "TMath.h" | |
21 | ||
57ddc2cb | 22 | #include <iostream> |
effa6338 | 23 | ClassImp(AliKFParticleBase) |
f826d409 | 24 | |
25 | ||
57ddc2cb | 26 | AliKFParticleBase::AliKFParticleBase() :fQ(0), fNDF(-3), fChi2(0), fSFromDecay(0), fAtProductionVertex(0), fIsLinearized(0), |
27 | fConstructMethod(2), SumDaughterMass(0), fMassHypo(-1) | |
f826d409 | 28 | { |
29 | //* Constructor | |
30 | ||
31 | Initialize(); | |
32 | } | |
e7b09c95 | 33 | |
34 | void AliKFParticleBase::Initialize( const Double_t Param[], const Double_t Cov[], Int_t Charge, Double_t Mass ) | |
35 | { | |
36 | // Constructor from "cartesian" track, particle mass hypothesis should be provided | |
37 | // | |
38 | // Param[6] = { X, Y, Z, Px, Py, Pz } - position and momentum | |
39 | // Cov [21] = lower-triangular part of the covariance matrix: | |
40 | // | |
41 | // ( 0 . . . . . ) | |
42 | // ( 1 2 . . . . ) | |
43 | // Cov. matrix = ( 3 4 5 . . . ) - numbering of covariance elements in Cov[] | |
44 | // ( 6 7 8 9 . . ) | |
45 | // ( 10 11 12 13 14 . ) | |
46 | // ( 15 16 17 18 19 20 ) | |
47 | ||
48 | ||
49 | for( Int_t i=0; i<6 ; i++ ) fP[i] = Param[i]; | |
50 | for( Int_t i=0; i<21; i++ ) fC[i] = Cov[i]; | |
51 | ||
52 | Double_t energy = TMath::Sqrt( Mass*Mass + fP[3]*fP[3] + fP[4]*fP[4] + fP[5]*fP[5]); | |
53 | fP[6] = energy; | |
54 | fP[7] = 0; | |
55 | fQ = Charge; | |
56 | fNDF = 0; | |
57 | fChi2 = 0; | |
58 | fAtProductionVertex = 0; | |
59 | fIsLinearized = 0; | |
60 | fSFromDecay = 0; | |
61 | ||
62 | Double_t energyInv = 1./energy; | |
63 | Double_t | |
64 | h0 = fP[3]*energyInv, | |
65 | h1 = fP[4]*energyInv, | |
66 | h2 = fP[5]*energyInv; | |
67 | ||
68 | fC[21] = h0*fC[ 6] + h1*fC[10] + h2*fC[15]; | |
69 | fC[22] = h0*fC[ 7] + h1*fC[11] + h2*fC[16]; | |
70 | fC[23] = h0*fC[ 8] + h1*fC[12] + h2*fC[17]; | |
71 | fC[24] = h0*fC[ 9] + h1*fC[13] + h2*fC[18]; | |
72 | fC[25] = h0*fC[13] + h1*fC[14] + h2*fC[19]; | |
73 | fC[26] = h0*fC[18] + h1*fC[19] + h2*fC[20]; | |
706952f5 | 74 | fC[27] = ( h0*h0*fC[ 9] + h1*h1*fC[14] + h2*h2*fC[20] |
75 | + 2*(h0*h1*fC[13] + h0*h2*fC[18] + h1*h2*fC[19] ) ); | |
e7b09c95 | 76 | for( Int_t i=28; i<36; i++ ) fC[i] = 0; |
77 | fC[35] = 1.; | |
57ddc2cb | 78 | |
79 | SumDaughterMass = Mass; | |
80 | fMassHypo = Mass; | |
e7b09c95 | 81 | } |
82 | ||
f826d409 | 83 | void AliKFParticleBase::Initialize() |
84 | { | |
85 | //* Initialise covariance matrix and set current parameters to 0.0 | |
86 | ||
87 | for( Int_t i=0; i<8; i++) fP[i] = 0; | |
88 | for(Int_t i=0;i<36;++i) fC[i]=0.; | |
89 | fC[0] = fC[2] = fC[5] = 100.; | |
90 | fC[35] = 1.; | |
91 | fNDF = -3; | |
92 | fChi2 = 0.; | |
93 | fQ = 0; | |
94 | fSFromDecay = 0; | |
95 | fAtProductionVertex = 0; | |
96 | fVtxGuess[0]=fVtxGuess[1]=fVtxGuess[2]=0.; | |
97 | fIsLinearized = 0; | |
57ddc2cb | 98 | SumDaughterMass = 0; |
99 | fMassHypo = -1; | |
f826d409 | 100 | } |
101 | ||
102 | void AliKFParticleBase::SetVtxGuess( Double_t x, Double_t y, Double_t z ) | |
103 | { | |
104 | //* Set decay vertex parameters for linearisation | |
105 | ||
106 | fVtxGuess[0] = x; | |
107 | fVtxGuess[1] = y; | |
108 | fVtxGuess[2] = z; | |
109 | fIsLinearized = 1; | |
110 | } | |
111 | ||
5fc72f28 | 112 | Int_t AliKFParticleBase::GetMomentum( Double_t &p, Double_t &error ) const |
f826d409 | 113 | { |
114 | //* Calculate particle momentum | |
115 | ||
116 | Double_t x = fP[3]; | |
117 | Double_t y = fP[4]; | |
118 | Double_t z = fP[5]; | |
55ac3e1e | 119 | |
f826d409 | 120 | Double_t x2 = x*x; |
121 | Double_t y2 = y*y; | |
122 | Double_t z2 = z*z; | |
123 | Double_t p2 = x2+y2+z2; | |
5fc72f28 | 124 | p = TMath::Sqrt(p2); |
55ac3e1e | 125 | |
5fc72f28 | 126 | error = (x2*fC[9]+y2*fC[14]+z2*fC[20] + 2*(x*y*fC[13]+x*z*fC[18]+y*z*fC[19]) ); |
55ac3e1e | 127 | if( error>1.e-16 && p>1.e-4 ){ |
5fc72f28 | 128 | error = TMath::Sqrt(error)/p; |
f826d409 | 129 | return 0; |
130 | } | |
55ac3e1e | 131 | error = 1.e8; |
f826d409 | 132 | return 1; |
133 | } | |
134 | ||
5fc72f28 | 135 | Int_t AliKFParticleBase::GetPt( Double_t &pt, Double_t &error ) const |
706952f5 | 136 | { |
137 | //* Calculate particle transverse momentum | |
138 | ||
139 | Double_t px = fP[3]; | |
140 | Double_t py = fP[4]; | |
141 | Double_t px2 = px*px; | |
142 | Double_t py2 = py*py; | |
143 | Double_t pt2 = px2+py2; | |
5fc72f28 | 144 | pt = TMath::Sqrt(pt2); |
145 | error = (px2*fC[9] + py2*fC[14] + 2*px*py*fC[13] ); | |
146 | if( error>0 && pt>1.e-4 ){ | |
147 | error = TMath::Sqrt(error)/pt; | |
706952f5 | 148 | return 0; |
149 | } | |
5fc72f28 | 150 | error = 1.e10; |
706952f5 | 151 | return 1; |
152 | } | |
153 | ||
5fc72f28 | 154 | Int_t AliKFParticleBase::GetEta( Double_t &eta, Double_t &error ) const |
706952f5 | 155 | { |
156 | //* Calculate particle pseudorapidity | |
157 | ||
158 | Double_t px = fP[3]; | |
159 | Double_t py = fP[4]; | |
160 | Double_t pz = fP[5]; | |
161 | Double_t pt2 = px*px + py*py; | |
162 | Double_t p2 = pt2 + pz*pz; | |
163 | Double_t p = TMath::Sqrt(p2); | |
164 | Double_t a = p + pz; | |
165 | Double_t b = p - pz; | |
5fc72f28 | 166 | eta = 1.e10; |
706952f5 | 167 | if( b > 1.e-8 ){ |
168 | Double_t c = a/b; | |
a65041d0 | 169 | if( c>1.e-8 ) eta = 0.5*TMath::Log(c); |
706952f5 | 170 | } |
171 | Double_t h3 = -px*pz; | |
a65041d0 | 172 | Double_t h4 = -py*pz; |
173 | Double_t pt4 = pt2*pt2; | |
174 | Double_t p2pt4 = p2*pt4; | |
175 | error = (h3*h3*fC[9] + h4*h4*fC[14] + pt4*fC[20] + 2*( h3*(h4*fC[13] + fC[18]*pt2) + pt2*h4*fC[19] ) ); | |
706952f5 | 176 | |
a65041d0 | 177 | if( error>0 && p2pt4>1.e-10 ){ |
178 | error = TMath::Sqrt(error/p2pt4); | |
706952f5 | 179 | return 0; |
180 | } | |
a65041d0 | 181 | |
5fc72f28 | 182 | error = 1.e10; |
706952f5 | 183 | return 1; |
184 | } | |
185 | ||
5fc72f28 | 186 | Int_t AliKFParticleBase::GetPhi( Double_t &phi, Double_t &error ) const |
706952f5 | 187 | { |
188 | //* Calculate particle polar angle | |
189 | ||
190 | Double_t px = fP[3]; | |
191 | Double_t py = fP[4]; | |
192 | Double_t px2 = px*px; | |
193 | Double_t py2 = py*py; | |
194 | Double_t pt2 = px2 + py2; | |
5fc72f28 | 195 | phi = TMath::ATan2(py,px); |
196 | error = (py2*fC[9] + px2*fC[14] - 2*px*py*fC[13] ); | |
197 | if( error>0 && pt2>1.e-4 ){ | |
198 | error = TMath::Sqrt(error)/pt2; | |
706952f5 | 199 | return 0; |
200 | } | |
5fc72f28 | 201 | error = 1.e10; |
706952f5 | 202 | return 1; |
203 | } | |
204 | ||
5fc72f28 | 205 | Int_t AliKFParticleBase::GetR( Double_t &r, Double_t &error ) const |
706952f5 | 206 | { |
207 | //* Calculate distance to the origin | |
208 | ||
209 | Double_t x = fP[0]; | |
210 | Double_t y = fP[1]; | |
211 | Double_t x2 = x*x; | |
212 | Double_t y2 = y*y; | |
5fc72f28 | 213 | r = TMath::Sqrt(x2 + y2); |
214 | error = (x2*fC[0] + y2*fC[2] - 2*x*y*fC[1] ); | |
215 | if( error>0 && r>1.e-4 ){ | |
216 | error = TMath::Sqrt(error)/r; | |
706952f5 | 217 | return 0; |
218 | } | |
5fc72f28 | 219 | error = 1.e10; |
706952f5 | 220 | return 1; |
221 | } | |
222 | ||
5fc72f28 | 223 | Int_t AliKFParticleBase::GetMass( Double_t &m, Double_t &error ) const |
f826d409 | 224 | { |
225 | //* Calculate particle mass | |
226 | ||
227 | // s = sigma^2 of m2/2 | |
228 | ||
229 | Double_t s = ( fP[3]*fP[3]*fC[9] + fP[4]*fP[4]*fC[14] + fP[5]*fP[5]*fC[20] | |
230 | + fP[6]*fP[6]*fC[27] | |
231 | +2*( + fP[3]*fP[4]*fC[13] + fP[5]*(fP[3]*fC[18] + fP[4]*fC[19]) | |
232 | - fP[6]*( fP[3]*fC[24] + fP[4]*fC[25] + fP[5]*fC[26] ) ) | |
233 | ); | |
57ddc2cb | 234 | // Double_t m2 = TMath::Abs(fP[6]*fP[6] - fP[3]*fP[3] - fP[4]*fP[4] - fP[5]*fP[5]); |
235 | // m = TMath::Sqrt(m2); | |
236 | // if( m>1.e-10 ){ | |
237 | // if( s>=0 ){ | |
238 | // error = TMath::Sqrt(s)/m; | |
239 | // return 0; | |
240 | // } | |
241 | // } | |
242 | // error = 1.e20; | |
243 | // return 1; | |
244 | Double_t m2 = (fP[6]*fP[6] - fP[3]*fP[3] - fP[4]*fP[4] - fP[5]*fP[5]); | |
245 | ||
246 | if(m2<0.) | |
247 | { | |
248 | error = 1.e20; | |
249 | m = -TMath::Sqrt(-m2); | |
250 | return 1; | |
251 | } | |
252 | ||
5fc72f28 | 253 | m = TMath::Sqrt(m2); |
57ddc2cb | 254 | if( m>1.e-6 ){ |
255 | if( s >= 0 ) { | |
5fc72f28 | 256 | error = TMath::Sqrt(s)/m; |
616ffc76 | 257 | return 0; |
258 | } | |
f826d409 | 259 | } |
57ddc2cb | 260 | else { |
261 | error = 1.e20; | |
262 | return 0; | |
263 | } | |
5fc72f28 | 264 | error = 1.e20; |
57ddc2cb | 265 | |
f826d409 | 266 | return 1; |
267 | } | |
268 | ||
269 | ||
5fc72f28 | 270 | Int_t AliKFParticleBase::GetDecayLength( Double_t &l, Double_t &error ) const |
f826d409 | 271 | { |
272 | //* Calculate particle decay length [cm] | |
273 | ||
274 | Double_t x = fP[3]; | |
275 | Double_t y = fP[4]; | |
276 | Double_t z = fP[5]; | |
277 | Double_t t = fP[7]; | |
278 | Double_t x2 = x*x; | |
279 | Double_t y2 = y*y; | |
280 | Double_t z2 = z*z; | |
281 | Double_t p2 = x2+y2+z2; | |
5fc72f28 | 282 | l = t*TMath::Sqrt(p2); |
f826d409 | 283 | if( p2>1.e-4){ |
5fc72f28 | 284 | error = p2*fC[35] + t*t/p2*(x2*fC[9]+y2*fC[14]+z2*fC[20] |
f826d409 | 285 | + 2*(x*y*fC[13]+x*z*fC[18]+y*z*fC[19]) ) |
286 | + 2*t*(x*fC[31]+y*fC[32]+z*fC[33]); | |
5fc72f28 | 287 | error = TMath::Sqrt(TMath::Abs(error)); |
f826d409 | 288 | return 0; |
289 | } | |
5fc72f28 | 290 | error = 1.e20; |
f826d409 | 291 | return 1; |
292 | } | |
293 | ||
446ce366 | 294 | Int_t AliKFParticleBase::GetDecayLengthXY( Double_t &l, Double_t &error ) const |
295 | { | |
296 | //* Calculate particle decay length in XY projection [cm] | |
297 | ||
298 | Double_t x = fP[3]; | |
299 | Double_t y = fP[4]; | |
300 | Double_t t = fP[7]; | |
301 | Double_t x2 = x*x; | |
302 | Double_t y2 = y*y; | |
303 | Double_t pt2 = x2+y2; | |
304 | l = t*TMath::Sqrt(pt2); | |
305 | if( pt2>1.e-4){ | |
306 | error = pt2*fC[35] + t*t/pt2*(x2*fC[9]+y2*fC[14] + 2*x*y*fC[13] ) | |
307 | + 2*t*(x*fC[31]+y*fC[32]); | |
308 | error = TMath::Sqrt(TMath::Abs(error)); | |
309 | return 0; | |
310 | } | |
311 | error = 1.e20; | |
312 | return 1; | |
313 | } | |
314 | ||
315 | ||
5fc72f28 | 316 | Int_t AliKFParticleBase::GetLifeTime( Double_t &tauC, Double_t &error ) const |
f826d409 | 317 | { |
318 | //* Calculate particle decay time [s] | |
319 | ||
320 | Double_t m, dm; | |
321 | GetMass( m, dm ); | |
322 | Double_t cTM = (-fP[3]*fC[31] - fP[4]*fC[32] - fP[5]*fC[33] + fP[6]*fC[34]); | |
5fc72f28 | 323 | tauC = fP[7]*m; |
324 | error = m*m*fC[35] + 2*fP[7]*cTM + fP[7]*fP[7]*dm*dm; | |
325 | if( error > 0 ){ | |
326 | error = TMath::Sqrt( error ); | |
f826d409 | 327 | return 0; |
328 | } | |
5fc72f28 | 329 | error = 1.e20; |
f826d409 | 330 | return 1; |
331 | } | |
332 | ||
333 | ||
334 | void AliKFParticleBase::operator +=( const AliKFParticleBase &Daughter ) | |
335 | { | |
336 | //* Add daughter via operator+= | |
337 | ||
338 | AddDaughter( Daughter ); | |
339 | } | |
340 | ||
706952f5 | 341 | Double_t AliKFParticleBase::GetSCorrection( const Double_t Part[], const Double_t XYZ[] ) |
342 | { | |
343 | //* Get big enough correction for S error to let the particle Part be fitted to XYZ point | |
344 | ||
345 | Double_t d[3] = { XYZ[0]-Part[0], XYZ[1]-Part[1], XYZ[2]-Part[2] }; | |
346 | Double_t p2 = Part[3]*Part[3]+Part[4]*Part[4]+Part[5]*Part[5]; | |
a65041d0 | 347 | Double_t sigmaS = (p2>1.e-4) ? ( 10.1+3.*TMath::Sqrt( d[0]*d[0]+d[1]*d[1]+d[2]*d[2]) )/TMath::Sqrt(p2) : 1.; |
706952f5 | 348 | return sigmaS; |
349 | } | |
616ffc76 | 350 | |
351 | void AliKFParticleBase::GetMeasurement( const Double_t XYZ[], Double_t m[], Double_t V[] ) const | |
352 | { | |
e7b09c95 | 353 | //* Get additional covariances V used during measurement |
354 | ||
616ffc76 | 355 | Double_t b[3]; |
356 | GetFieldValue( XYZ, b ); | |
357 | const Double_t kCLight = 0.000299792458; | |
358 | b[0]*=kCLight; b[1]*=kCLight; b[2]*=kCLight; | |
359 | ||
360 | Transport( GetDStoPoint(XYZ), m, V ); | |
361 | ||
706952f5 | 362 | Double_t sigmaS = GetSCorrection( m, XYZ ); |
616ffc76 | 363 | |
364 | Double_t h[6]; | |
365 | ||
366 | h[0] = m[3]*sigmaS; | |
367 | h[1] = m[4]*sigmaS; | |
368 | h[2] = m[5]*sigmaS; | |
369 | h[3] = ( h[1]*b[2]-h[2]*b[1] )*GetQ(); | |
370 | h[4] = ( h[2]*b[0]-h[0]*b[2] )*GetQ(); | |
371 | h[5] = ( h[0]*b[1]-h[1]*b[0] )*GetQ(); | |
372 | ||
616ffc76 | 373 | V[ 0]+= h[0]*h[0]; |
374 | V[ 1]+= h[1]*h[0]; | |
375 | V[ 2]+= h[1]*h[1]; | |
376 | V[ 3]+= h[2]*h[0]; | |
377 | V[ 4]+= h[2]*h[1]; | |
378 | V[ 5]+= h[2]*h[2]; | |
379 | ||
380 | V[ 6]+= h[3]*h[0]; | |
381 | V[ 7]+= h[3]*h[1]; | |
382 | V[ 8]+= h[3]*h[2]; | |
383 | V[ 9]+= h[3]*h[3]; | |
384 | ||
385 | V[10]+= h[4]*h[0]; | |
386 | V[11]+= h[4]*h[1]; | |
387 | V[12]+= h[4]*h[2]; | |
388 | V[13]+= h[4]*h[3]; | |
389 | V[14]+= h[4]*h[4]; | |
390 | ||
391 | V[15]+= h[5]*h[0]; | |
392 | V[16]+= h[5]*h[1]; | |
393 | V[17]+= h[5]*h[2]; | |
394 | V[18]+= h[5]*h[3]; | |
395 | V[19]+= h[5]*h[4]; | |
396 | V[20]+= h[5]*h[5]; | |
397 | } | |
398 | ||
f826d409 | 399 | void AliKFParticleBase::AddDaughter( const AliKFParticleBase &Daughter ) |
400 | { | |
616ffc76 | 401 | if( fNDF<-1 ){ // first daughter -> just copy |
402 | fNDF = -1; | |
403 | fQ = Daughter.GetQ(); | |
404 | for( Int_t i=0; i<7; i++) fP[i] = Daughter.fP[i]; | |
405 | for( Int_t i=0; i<28; i++) fC[i] = Daughter.fC[i]; | |
406 | fSFromDecay = 0; | |
57ddc2cb | 407 | fMassHypo = Daughter.fMassHypo; |
408 | SumDaughterMass = Daughter.SumDaughterMass; | |
616ffc76 | 409 | return; |
410 | } | |
411 | ||
57ddc2cb | 412 | if(fConstructMethod == 0) |
413 | AddDaughterWithEnergyFit(Daughter); | |
414 | else if(fConstructMethod == 1) | |
415 | AddDaughterWithEnergyCalc(Daughter); | |
416 | else if(fConstructMethod == 2) | |
417 | AddDaughterWithEnergyFitMC(Daughter); | |
418 | ||
419 | SumDaughterMass += Daughter.SumDaughterMass; | |
420 | fMassHypo = -1; | |
421 | } | |
422 | ||
423 | void AliKFParticleBase::AddDaughterWithEnergyFit( const AliKFParticleBase &Daughter ) | |
424 | { | |
425 | //* Energy considered as an independent veriable, fitted independently from momentum, without any constraints on mass | |
426 | ||
427 | //* Add daughter | |
428 | ||
f826d409 | 429 | TransportToDecayVertex(); |
430 | ||
431 | Double_t b[3]; | |
432 | Int_t maxIter = 1; | |
433 | ||
434 | if( !fIsLinearized ){ | |
435 | if( fNDF==-1 ){ | |
436 | Double_t ds, ds1; | |
616ffc76 | 437 | GetDStoParticle(Daughter, ds, ds1); |
f826d409 | 438 | TransportToDS( ds ); |
616ffc76 | 439 | Double_t m[8]; |
440 | Double_t mCd[36]; | |
441 | Daughter.Transport( ds1, m, mCd ); | |
442 | fVtxGuess[0] = .5*( fP[0] + m[0] ); | |
443 | fVtxGuess[1] = .5*( fP[1] + m[1] ); | |
444 | fVtxGuess[2] = .5*( fP[2] + m[2] ); | |
445 | } else { | |
446 | fVtxGuess[0] = fP[0]; | |
447 | fVtxGuess[1] = fP[1]; | |
448 | fVtxGuess[2] = fP[2]; | |
f826d409 | 449 | } |
f826d409 | 450 | maxIter = 3; |
451 | } | |
452 | ||
453 | for( Int_t iter=0; iter<maxIter; iter++ ){ | |
454 | ||
455 | { | |
456 | GetFieldValue( fVtxGuess, b ); | |
457 | const Double_t kCLight = 0.000299792458; | |
458 | b[0]*=kCLight; b[1]*=kCLight; b[2]*=kCLight; | |
459 | } | |
f826d409 | 460 | |
616ffc76 | 461 | Double_t *ffP = fP, *ffC = fC, tmpP[8], tmpC[36]; |
de0d0ceb | 462 | if( fNDF==-1 ){ |
616ffc76 | 463 | GetMeasurement( fVtxGuess, tmpP, tmpC ); |
464 | ffP = tmpP; | |
465 | ffC = tmpC; | |
f826d409 | 466 | } |
467 | ||
e7b09c95 | 468 | Double_t m[8], mV[36]; |
de0d0ceb | 469 | |
e7b09c95 | 470 | if( Daughter.fC[35]>0 ){ |
471 | Daughter.GetMeasurement( fVtxGuess, m, mV ); | |
472 | } else { | |
473 | for( Int_t i=0; i<8; i++ ) m[i] = Daughter.fP[i]; | |
474 | for( Int_t i=0; i<36; i++ ) mV[i] = Daughter.fC[i]; | |
475 | } | |
f826d409 | 476 | //* |
477 | ||
478 | Double_t mS[6]; | |
479 | { | |
616ffc76 | 480 | Double_t mSi[6] = { ffC[0]+mV[0], |
481 | ffC[1]+mV[1], ffC[2]+mV[2], | |
482 | ffC[3]+mV[3], ffC[4]+mV[4], ffC[5]+mV[5] }; | |
57ddc2cb | 483 | |
f826d409 | 484 | mS[0] = mSi[2]*mSi[5] - mSi[4]*mSi[4]; |
485 | mS[1] = mSi[3]*mSi[4] - mSi[1]*mSi[5]; | |
486 | mS[2] = mSi[0]*mSi[5] - mSi[3]*mSi[3]; | |
487 | mS[3] = mSi[1]*mSi[4] - mSi[2]*mSi[3]; | |
488 | mS[4] = mSi[1]*mSi[3] - mSi[0]*mSi[4]; | |
489 | mS[5] = mSi[0]*mSi[2] - mSi[1]*mSi[1]; | |
706952f5 | 490 | |
57ddc2cb | 491 | Double_t s = ( mSi[0]*mS[0] + mSi[1]*mS[1] + mSi[3]*mS[3] ); |
492 | s = ( TMath::Abs(s) > 1.E-20 ) ?1./s :0; | |
f826d409 | 493 | mS[0]*=s; |
494 | mS[1]*=s; | |
495 | mS[2]*=s; | |
496 | mS[3]*=s; | |
497 | mS[4]*=s; | |
498 | mS[5]*=s; | |
499 | } | |
f826d409 | 500 | //* Residual (measured - estimated) |
57ddc2cb | 501 | |
616ffc76 | 502 | Double_t zeta[3] = { m[0]-ffP[0], m[1]-ffP[1], m[2]-ffP[2] }; |
de0d0ceb | 503 | |
f826d409 | 504 | //* CHt = CH' - D' |
57ddc2cb | 505 | |
f826d409 | 506 | Double_t mCHt0[7], mCHt1[7], mCHt2[7]; |
57ddc2cb | 507 | |
616ffc76 | 508 | mCHt0[0]=ffC[ 0] ; mCHt1[0]=ffC[ 1] ; mCHt2[0]=ffC[ 3] ; |
509 | mCHt0[1]=ffC[ 1] ; mCHt1[1]=ffC[ 2] ; mCHt2[1]=ffC[ 4] ; | |
510 | mCHt0[2]=ffC[ 3] ; mCHt1[2]=ffC[ 4] ; mCHt2[2]=ffC[ 5] ; | |
511 | mCHt0[3]=ffC[ 6]-mV[ 6]; mCHt1[3]=ffC[ 7]-mV[ 7]; mCHt2[3]=ffC[ 8]-mV[ 8]; | |
512 | mCHt0[4]=ffC[10]-mV[10]; mCHt1[4]=ffC[11]-mV[11]; mCHt2[4]=ffC[12]-mV[12]; | |
513 | mCHt0[5]=ffC[15]-mV[15]; mCHt1[5]=ffC[16]-mV[16]; mCHt2[5]=ffC[17]-mV[17]; | |
514 | mCHt0[6]=ffC[21]-mV[21]; mCHt1[6]=ffC[22]-mV[22]; mCHt2[6]=ffC[23]-mV[23]; | |
f826d409 | 515 | |
516 | //* Kalman gain K = mCH'*S | |
517 | ||
518 | Double_t k0[7], k1[7], k2[7]; | |
519 | ||
520 | for(Int_t i=0;i<7;++i){ | |
521 | k0[i] = mCHt0[i]*mS[0] + mCHt1[i]*mS[1] + mCHt2[i]*mS[3]; | |
522 | k1[i] = mCHt0[i]*mS[1] + mCHt1[i]*mS[2] + mCHt2[i]*mS[4]; | |
523 | k2[i] = mCHt0[i]*mS[3] + mCHt1[i]*mS[4] + mCHt2[i]*mS[5]; | |
524 | } | |
525 | ||
4bbc290d | 526 | //* New estimation of the vertex position |
f826d409 | 527 | |
528 | if( iter<maxIter-1 ){ | |
529 | for(Int_t i=0; i<3; ++i) | |
616ffc76 | 530 | fVtxGuess[i]= ffP[i] + k0[i]*zeta[0]+k1[i]*zeta[1]+k2[i]*zeta[2]; |
f826d409 | 531 | continue; |
532 | } | |
533 | ||
534 | // last itearation -> update the particle | |
535 | ||
536 | //* Add the daughter momentum to the particle momentum | |
537 | ||
616ffc76 | 538 | ffP[ 3] += m[ 3]; |
539 | ffP[ 4] += m[ 4]; | |
540 | ffP[ 5] += m[ 5]; | |
541 | ffP[ 6] += m[ 6]; | |
f826d409 | 542 | |
616ffc76 | 543 | ffC[ 9] += mV[ 9]; |
544 | ffC[13] += mV[13]; | |
545 | ffC[14] += mV[14]; | |
546 | ffC[18] += mV[18]; | |
547 | ffC[19] += mV[19]; | |
548 | ffC[20] += mV[20]; | |
549 | ffC[24] += mV[24]; | |
550 | ffC[25] += mV[25]; | |
551 | ffC[26] += mV[26]; | |
552 | ffC[27] += mV[27]; | |
f826d409 | 553 | |
de0d0ceb | 554 | |
555 | //* New estimation of the vertex position r += K*zeta | |
f826d409 | 556 | |
557 | for(Int_t i=0;i<7;++i) | |
616ffc76 | 558 | fP[i] = ffP[i] + k0[i]*zeta[0] + k1[i]*zeta[1] + k2[i]*zeta[2]; |
f826d409 | 559 | |
560 | //* New covariance matrix C -= K*(mCH')' | |
de0d0ceb | 561 | |
f826d409 | 562 | for(Int_t i=0, k=0;i<7;++i){ |
de0d0ceb | 563 | for(Int_t j=0;j<=i;++j,++k){ |
616ffc76 | 564 | fC[k] = ffC[k] - (k0[i]*mCHt0[j] + k1[i]*mCHt1[j] + k2[i]*mCHt2[j] ); |
de0d0ceb | 565 | } |
f826d409 | 566 | } |
de0d0ceb | 567 | |
f826d409 | 568 | //* Calculate Chi^2 |
569 | ||
570 | fNDF += 2; | |
571 | fQ += Daughter.GetQ(); | |
572 | fSFromDecay = 0; | |
573 | fChi2 += (mS[0]*zeta[0] + mS[1]*zeta[1] + mS[3]*zeta[2])*zeta[0] | |
574 | + (mS[1]*zeta[0] + mS[2]*zeta[1] + mS[4]*zeta[2])*zeta[1] | |
706952f5 | 575 | + (mS[3]*zeta[0] + mS[4]*zeta[1] + mS[5]*zeta[2])*zeta[2]; |
de0d0ceb | 576 | |
f826d409 | 577 | } |
578 | } | |
579 | ||
57ddc2cb | 580 | void AliKFParticleBase::AddDaughterWithEnergyCalc( const AliKFParticleBase &Daughter ) |
581 | { | |
582 | //* Energy considered as a dependent variable, calculated from the momentum and mass hypothesis | |
583 | ||
584 | //* Add daughter | |
585 | ||
586 | TransportToDecayVertex(); | |
587 | ||
588 | Double_t b[3]; | |
589 | Int_t maxIter = 1; | |
590 | ||
591 | if( !fIsLinearized ){ | |
592 | if( fNDF==-1 ){ | |
593 | Double_t ds, ds1; | |
594 | GetDStoParticle(Daughter, ds, ds1); | |
595 | TransportToDS( ds ); | |
596 | Double_t m[8]; | |
597 | Double_t mCd[36]; | |
598 | Daughter.Transport( ds1, m, mCd ); | |
599 | fVtxGuess[0] = .5*( fP[0] + m[0] ); | |
600 | fVtxGuess[1] = .5*( fP[1] + m[1] ); | |
601 | fVtxGuess[2] = .5*( fP[2] + m[2] ); | |
602 | } else { | |
603 | fVtxGuess[0] = fP[0]; | |
604 | fVtxGuess[1] = fP[1]; | |
605 | fVtxGuess[2] = fP[2]; | |
606 | } | |
607 | maxIter = 3; | |
608 | } | |
609 | ||
610 | for( Int_t iter=0; iter<maxIter; iter++ ){ | |
611 | ||
612 | { | |
613 | GetFieldValue( fVtxGuess, b ); | |
614 | const Double_t kCLight = 0.000299792458; | |
615 | b[0]*=kCLight; b[1]*=kCLight; b[2]*=kCLight; | |
616 | } | |
617 | ||
618 | Double_t *ffP = fP, *ffC = fC, tmpP[8], tmpC[36]; | |
619 | if( fNDF==-1 ){ | |
620 | GetMeasurement( fVtxGuess, tmpP, tmpC ); | |
621 | ffP = tmpP; | |
622 | ffC = tmpC; | |
623 | } | |
624 | ||
625 | Double_t m[8], mV[36]; | |
626 | ||
627 | if( Daughter.fC[35]>0 ){ | |
628 | Daughter.GetMeasurement( fVtxGuess, m, mV ); | |
629 | } else { | |
630 | for( Int_t i=0; i<8; i++ ) m[i] = Daughter.fP[i]; | |
631 | for( Int_t i=0; i<36; i++ ) mV[i] = Daughter.fC[i]; | |
632 | } | |
633 | ||
5f653c6e | 634 | double massMf2 = m[6]*m[6] - (m[3]*m[3] + m[4]*m[4] + m[5]*m[5]); |
635 | double massRf2 = fP[6]*fP[6] - (fP[3]*fP[3] + fP[4]*fP[4] + fP[5]*fP[5]); | |
57ddc2cb | 636 | |
637 | //* | |
638 | ||
639 | Double_t mS[6]; | |
640 | { | |
641 | Double_t mSi[6] = { ffC[0]+mV[0], | |
642 | ffC[1]+mV[1], ffC[2]+mV[2], | |
643 | ffC[3]+mV[3], ffC[4]+mV[4], ffC[5]+mV[5] }; | |
644 | ||
645 | mS[0] = mSi[2]*mSi[5] - mSi[4]*mSi[4]; | |
646 | mS[1] = mSi[3]*mSi[4] - mSi[1]*mSi[5]; | |
647 | mS[2] = mSi[0]*mSi[5] - mSi[3]*mSi[3]; | |
648 | mS[3] = mSi[1]*mSi[4] - mSi[2]*mSi[3]; | |
649 | mS[4] = mSi[1]*mSi[3] - mSi[0]*mSi[4]; | |
650 | mS[5] = mSi[0]*mSi[2] - mSi[1]*mSi[1]; | |
651 | ||
652 | Double_t s = ( mSi[0]*mS[0] + mSi[1]*mS[1] + mSi[3]*mS[3] ); | |
653 | ||
654 | s = ( s > 1.E-20 ) ?1./s :0; | |
655 | mS[0]*=s; | |
656 | mS[1]*=s; | |
657 | mS[2]*=s; | |
658 | mS[3]*=s; | |
659 | mS[4]*=s; | |
660 | mS[5]*=s; | |
661 | } | |
662 | ||
663 | //* Residual (measured - estimated) | |
664 | ||
665 | Double_t zeta[3] = { m[0]-ffP[0], m[1]-ffP[1], m[2]-ffP[2] }; | |
666 | ||
667 | //* CHt = CH' - D' | |
668 | ||
669 | Double_t mCHt0[6], mCHt1[6], mCHt2[6]; | |
670 | ||
671 | mCHt0[0]=ffC[ 0] ; mCHt1[0]=ffC[ 1] ; mCHt2[0]=ffC[ 3] ; | |
672 | mCHt0[1]=ffC[ 1] ; mCHt1[1]=ffC[ 2] ; mCHt2[1]=ffC[ 4] ; | |
673 | mCHt0[2]=ffC[ 3] ; mCHt1[2]=ffC[ 4] ; mCHt2[2]=ffC[ 5] ; | |
674 | mCHt0[3]=ffC[ 6]-mV[ 6]; mCHt1[3]=ffC[ 7]-mV[ 7]; mCHt2[3]=ffC[ 8]-mV[ 8]; | |
675 | mCHt0[4]=ffC[10]-mV[10]; mCHt1[4]=ffC[11]-mV[11]; mCHt2[4]=ffC[12]-mV[12]; | |
676 | mCHt0[5]=ffC[15]-mV[15]; mCHt1[5]=ffC[16]-mV[16]; mCHt2[5]=ffC[17]-mV[17]; | |
677 | ||
678 | //* Kalman gain K = mCH'*S | |
679 | ||
680 | Double_t k0[6], k1[6], k2[6]; | |
681 | ||
682 | for(Int_t i=0;i<6;++i){ | |
683 | k0[i] = mCHt0[i]*mS[0] + mCHt1[i]*mS[1] + mCHt2[i]*mS[3]; | |
684 | k1[i] = mCHt0[i]*mS[1] + mCHt1[i]*mS[2] + mCHt2[i]*mS[4]; | |
685 | k2[i] = mCHt0[i]*mS[3] + mCHt1[i]*mS[4] + mCHt2[i]*mS[5]; | |
686 | } | |
687 | ||
688 | //* New estimation of the vertex position | |
689 | ||
690 | if( iter<maxIter-1 ){ | |
691 | for(Int_t i=0; i<3; ++i) | |
692 | fVtxGuess[i]= ffP[i] + k0[i]*zeta[0]+k1[i]*zeta[1]+k2[i]*zeta[2]; | |
693 | continue; | |
694 | } | |
695 | ||
5f653c6e | 696 | //* find mf and mVf - optimum value of the measurement and its covariance matrix |
57ddc2cb | 697 | //* mVHt = V*H' |
698 | Double_t mVHt0[6], mVHt1[6], mVHt2[6]; | |
699 | ||
700 | mVHt0[0]= mV[ 0] ; mVHt1[0]= mV[ 1] ; mVHt2[0]= mV[ 3] ; | |
701 | mVHt0[1]= mV[ 1] ; mVHt1[1]= mV[ 2] ; mVHt2[1]= mV[ 4] ; | |
702 | mVHt0[2]= mV[ 3] ; mVHt1[2]= mV[ 4] ; mVHt2[2]= mV[ 5] ; | |
703 | mVHt0[3]= mV[ 6] ; mVHt1[3]= mV[ 7] ; mVHt2[3]= mV[ 8] ; | |
704 | mVHt0[4]= mV[10] ; mVHt1[4]= mV[11] ; mVHt2[4]= mV[12] ; | |
705 | mVHt0[5]= mV[15] ; mVHt1[5]= mV[16] ; mVHt2[5]= mV[17] ; | |
706 | ||
707 | //* Kalman gain Km = mCH'*S | |
708 | ||
709 | Double_t km0[6], km1[6], km2[6]; | |
710 | ||
711 | for(Int_t i=0;i<6;++i){ | |
712 | km0[i] = mVHt0[i]*mS[0] + mVHt1[i]*mS[1] + mVHt2[i]*mS[3]; | |
713 | km1[i] = mVHt0[i]*mS[1] + mVHt1[i]*mS[2] + mVHt2[i]*mS[4]; | |
714 | km2[i] = mVHt0[i]*mS[3] + mVHt1[i]*mS[4] + mVHt2[i]*mS[5]; | |
715 | } | |
716 | ||
717 | Double_t mf[7] = { m[0], m[1], m[2], m[3], m[4], m[5], m[6] }; | |
718 | ||
719 | for(Int_t i=0;i<6;++i) | |
720 | mf[i] = mf[i] - km0[i]*zeta[0] - km1[i]*zeta[1] - km2[i]*zeta[2]; | |
721 | ||
5f653c6e | 722 | Double_t energyMf = TMath::Sqrt( massMf2 + (mf[3]*mf[3] + mf[4]*mf[4] + mf[5]*mf[5]) ); |
57ddc2cb | 723 | |
5f653c6e | 724 | Double_t mVf[28]; |
57ddc2cb | 725 | for(Int_t iC=0; iC<28; iC++) |
5f653c6e | 726 | mVf[iC] = mV[iC]; |
57ddc2cb | 727 | |
5f653c6e | 728 | //* hmf = d(energyMf)/d(mf) |
57ddc2cb | 729 | Double_t hmf[7]; |
5f653c6e | 730 | if( TMath::Abs(energyMf) < 1.e-10) hmf[3] = 0; else hmf[3] = mf[3]/energyMf; |
731 | if( TMath::Abs(energyMf) < 1.e-10) hmf[4] = 0; else hmf[4] = mf[4]/energyMf; | |
732 | if( TMath::Abs(energyMf) < 1.e-10) hmf[5] = 0; else hmf[5] = mf[5]/energyMf; | |
733 | // if( TMath::Abs(energyMf) < 1.e-10) hmf[6] = 0; else hmf[6] = mf[6]/energyMf; | |
57ddc2cb | 734 | hmf[6] = 0; |
735 | ||
736 | for(Int_t i=0, k=0;i<6;++i){ | |
737 | for(Int_t j=0;j<=i;++j,++k){ | |
5f653c6e | 738 | mVf[k] = mVf[k] - (km0[i]*mVHt0[j] + km1[i]*mVHt1[j] + km2[i]*mVHt2[j] ); |
57ddc2cb | 739 | } |
740 | } | |
5f653c6e | 741 | Double_t mVf24 = mVf[24], mVf25 = mVf[25], mVf26 = mVf[26]; |
742 | mVf[21] = mVf[6 ]*hmf[3] + mVf[10]*hmf[4] + mVf[15]*hmf[5] + mVf[21]*hmf[6]; | |
743 | mVf[22] = mVf[7 ]*hmf[3] + mVf[11]*hmf[4] + mVf[16]*hmf[5] + mVf[22]*hmf[6]; | |
744 | mVf[23] = mVf[8 ]*hmf[3] + mVf[12]*hmf[4] + mVf[17]*hmf[5] + mVf[23]*hmf[6]; | |
745 | mVf[24] = mVf[9 ]*hmf[3] + mVf[13]*hmf[4] + mVf[18]*hmf[5] + mVf[24]*hmf[6]; | |
746 | mVf[25] = mVf[13]*hmf[3] + mVf[14]*hmf[4] + mVf[19]*hmf[5] + mVf[25]*hmf[6]; | |
747 | mVf[26] = mVf[18]*hmf[3] + mVf[19]*hmf[4] + mVf[20]*hmf[5] + mVf[26]*hmf[6]; | |
748 | mVf[27] = mVf[24]*hmf[3] + mVf[25]*hmf[4] + mVf[26]*hmf[5] + (mVf24*hmf[3] + mVf25*hmf[4] + mVf26*hmf[5] + mVf[27]*hmf[6])*hmf[6]; //here mVf[] are already modified | |
57ddc2cb | 749 | |
5f653c6e | 750 | mf[6] = energyMf; |
57ddc2cb | 751 | |
5f653c6e | 752 | //* find rf and mCf - optimum value of the measurement and its covariance matrix |
57ddc2cb | 753 | |
754 | //* mCCHt = C*H' | |
755 | Double_t mCCHt0[6], mCCHt1[6], mCCHt2[6]; | |
756 | ||
757 | mCCHt0[0]=ffC[ 0]; mCCHt1[0]=ffC[ 1]; mCCHt2[0]=ffC[ 3]; | |
758 | mCCHt0[1]=ffC[ 1]; mCCHt1[1]=ffC[ 2]; mCCHt2[1]=ffC[ 4]; | |
759 | mCCHt0[2]=ffC[ 3]; mCCHt1[2]=ffC[ 4]; mCCHt2[2]=ffC[ 5]; | |
760 | mCCHt0[3]=ffC[ 6]; mCCHt1[3]=ffC[ 7]; mCCHt2[3]=ffC[ 8]; | |
761 | mCCHt0[4]=ffC[10]; mCCHt1[4]=ffC[11]; mCCHt2[4]=ffC[12]; | |
762 | mCCHt0[5]=ffC[15]; mCCHt1[5]=ffC[16]; mCCHt2[5]=ffC[17]; | |
763 | ||
764 | //* Kalman gain Krf = mCH'*S | |
765 | ||
766 | Double_t krf0[6], krf1[6], krf2[6]; | |
767 | ||
768 | for(Int_t i=0;i<6;++i){ | |
769 | krf0[i] = mCCHt0[i]*mS[0] + mCCHt1[i]*mS[1] + mCCHt2[i]*mS[3]; | |
770 | krf1[i] = mCCHt0[i]*mS[1] + mCCHt1[i]*mS[2] + mCCHt2[i]*mS[4]; | |
771 | krf2[i] = mCCHt0[i]*mS[3] + mCCHt1[i]*mS[4] + mCCHt2[i]*mS[5]; | |
772 | } | |
773 | Double_t rf[7] = { ffP[0], ffP[1], ffP[2], ffP[3], ffP[4], ffP[5], ffP[6] }; | |
774 | ||
775 | for(Int_t i=0;i<6;++i) | |
776 | rf[i] = rf[i] + krf0[i]*zeta[0] + krf1[i]*zeta[1] + krf2[i]*zeta[2]; | |
777 | ||
5f653c6e | 778 | Double_t energyRf = TMath::Sqrt( massRf2 + (rf[3]*rf[3] + rf[4]*rf[4] + rf[5]*rf[5]) ); |
57ddc2cb | 779 | |
5f653c6e | 780 | Double_t mCf[28]; |
57ddc2cb | 781 | for(Int_t iC=0; iC<28; iC++) |
5f653c6e | 782 | mCf[iC] = ffC[iC]; |
57ddc2cb | 783 | //* hrf = d(Erf)/d(rf) |
784 | Double_t hrf[7]; | |
5f653c6e | 785 | if( TMath::Abs(energyRf) < 1.e-10) hrf[3] = 0; else hrf[3] = rf[3]/energyRf; |
786 | if( TMath::Abs(energyRf) < 1.e-10) hrf[4] = 0; else hrf[4] = rf[4]/energyRf; | |
787 | if( TMath::Abs(energyRf) < 1.e-10) hrf[5] = 0; else hrf[5] = rf[5]/energyRf; | |
788 | // if( TMath::Abs(energyRf) < 1.e-10) hrf[6] = 0; else hrf[6] = rf[6]/energyRf; | |
57ddc2cb | 789 | hrf[6] = 0; |
790 | ||
791 | for(Int_t i=0, k=0;i<6;++i){ | |
792 | for(Int_t j=0;j<=i;++j,++k){ | |
5f653c6e | 793 | mCf[k] = mCf[k] - (krf0[i]*mCCHt0[j] + krf1[i]*mCCHt1[j] + krf2[i]*mCCHt2[j] ); |
57ddc2cb | 794 | } |
795 | } | |
5f653c6e | 796 | Double_t mCf24 = mCf[24], mCf25 = mCf[25], mCf26 = mCf[26]; |
797 | mCf[21] = mCf[6 ]*hrf[3] + mCf[10]*hrf[4] + mCf[15]*hrf[5] + mCf[21]*hrf[6]; | |
798 | mCf[22] = mCf[7 ]*hrf[3] + mCf[11]*hrf[4] + mCf[16]*hrf[5] + mCf[22]*hrf[6]; | |
799 | mCf[23] = mCf[8 ]*hrf[3] + mCf[12]*hrf[4] + mCf[17]*hrf[5] + mCf[23]*hrf[6]; | |
800 | mCf[24] = mCf[9 ]*hrf[3] + mCf[13]*hrf[4] + mCf[18]*hrf[5] + mCf[24]*hrf[6]; | |
801 | mCf[25] = mCf[13]*hrf[3] + mCf[14]*hrf[4] + mCf[19]*hrf[5] + mCf[25]*hrf[6]; | |
802 | mCf[26] = mCf[18]*hrf[3] + mCf[19]*hrf[4] + mCf[20]*hrf[5] + mCf[26]*hrf[6]; | |
803 | mCf[27] = mCf[24]*hrf[3] + mCf[25]*hrf[4] + mCf[26]*hrf[5] + (mCf24*hrf[3] + mCf25*hrf[4] + mCf26*hrf[5] + mCf[27]*hrf[6])*hrf[6]; //here mCf[] are already modified | |
57ddc2cb | 804 | |
805 | for(Int_t iC=21; iC<28; iC++) | |
806 | { | |
5f653c6e | 807 | ffC[iC] = mCf[iC]; |
808 | mV[iC] = mVf[iC]; | |
57ddc2cb | 809 | } |
810 | ||
5f653c6e | 811 | fP[6] = energyRf + energyMf; |
812 | rf[6] = energyRf; | |
57ddc2cb | 813 | |
814 | //Double_t Dvv[3][3]; do not need this | |
5f653c6e | 815 | Double_t mDvp[3][3]; |
026bab91 | 816 | // Double_t mDpv[3][3]; |
5f653c6e | 817 | Double_t mDpp[3][3]; |
818 | Double_t mDe[7]; | |
57ddc2cb | 819 | |
820 | for(int i=0; i<3; i++) | |
821 | { | |
822 | for(int j=0; j<3; j++) | |
823 | { | |
5f653c6e | 824 | mDvp[i][j] = km0[i+3]*mCCHt0[j] + km1[i+3]*mCCHt1[j] + km2[i+3]*mCCHt2[j]; |
026bab91 | 825 | // mDpv[i][j] = km0[i]*mCCHt0[j+3] + km1[i]*mCCHt1[j+3] + km2[i]*mCCHt2[j+3]; |
5f653c6e | 826 | mDpp[i][j] = km0[i+3]*mCCHt0[j+3] + km1[i+3]*mCCHt1[j+3] + km2[i+3]*mCCHt2[j+3]; |
57ddc2cb | 827 | } |
828 | } | |
829 | ||
5f653c6e | 830 | mDe[0] = hmf[3]*mDvp[0][0] + hmf[4]*mDvp[1][0] + hmf[5]*mDvp[2][0]; |
831 | mDe[1] = hmf[3]*mDvp[0][1] + hmf[4]*mDvp[1][1] + hmf[5]*mDvp[2][1]; | |
832 | mDe[2] = hmf[3]*mDvp[0][2] + hmf[4]*mDvp[1][2] + hmf[5]*mDvp[2][2]; | |
833 | mDe[3] = hmf[3]*mDpp[0][0] + hmf[4]*mDpp[1][0] + hmf[5]*mDpp[2][0]; | |
834 | mDe[4] = hmf[3]*mDpp[0][1] + hmf[4]*mDpp[1][1] + hmf[5]*mDpp[2][1]; | |
835 | mDe[5] = hmf[3]*mDpp[0][2] + hmf[4]*mDpp[1][2] + hmf[5]*mDpp[2][2]; | |
836 | mDe[6] = 2*(mDe[3]*hrf[3] + mDe[4]*hrf[4] + mDe[5]*hrf[5]); | |
57ddc2cb | 837 | |
838 | // last itearation -> update the particle | |
839 | ||
840 | //* Add the daughter momentum to the particle momentum | |
841 | ||
842 | ffP[ 3] += m[ 3]; | |
843 | ffP[ 4] += m[ 4]; | |
844 | ffP[ 5] += m[ 5]; | |
845 | ||
846 | ffC[ 9] += mV[ 9]; | |
847 | ffC[13] += mV[13]; | |
848 | ffC[14] += mV[14]; | |
849 | ffC[18] += mV[18]; | |
850 | ffC[19] += mV[19]; | |
851 | ffC[20] += mV[20]; | |
852 | ffC[24] += mV[24]; | |
853 | ffC[25] += mV[25]; | |
854 | ffC[26] += mV[26]; | |
855 | ffC[27] += mV[27]; | |
856 | ||
5f653c6e | 857 | ffC[21] += mDe[0]; |
858 | ffC[22] += mDe[1]; | |
859 | ffC[23] += mDe[2]; | |
860 | ffC[24] += mDe[3]; | |
861 | ffC[25] += mDe[4]; | |
862 | ffC[26] += mDe[5]; | |
863 | ffC[27] += mDe[6]; | |
57ddc2cb | 864 | |
865 | //* New estimation of the vertex position r += K*zeta | |
866 | ||
867 | for(Int_t i=0;i<6;++i) | |
868 | fP[i] = ffP[i] + k0[i]*zeta[0] + k1[i]*zeta[1] + k2[i]*zeta[2]; | |
869 | ||
870 | //* New covariance matrix C -= K*(mCH')' | |
871 | ||
872 | for(Int_t i=0, k=0;i<6;++i){ | |
873 | for(Int_t j=0;j<=i;++j,++k){ | |
874 | fC[k] = ffC[k] - (k0[i]*mCHt0[j] + k1[i]*mCHt1[j] + k2[i]*mCHt2[j] ); | |
875 | } | |
876 | } | |
877 | ||
878 | for(int i=21; i<28; i++) fC[i] = ffC[i]; | |
879 | ||
880 | //* Calculate Chi^2 | |
881 | ||
882 | fNDF += 2; | |
883 | fQ += Daughter.GetQ(); | |
884 | fSFromDecay = 0; | |
885 | fChi2 += (mS[0]*zeta[0] + mS[1]*zeta[1] + mS[3]*zeta[2])*zeta[0] | |
886 | + (mS[1]*zeta[0] + mS[2]*zeta[1] + mS[4]*zeta[2])*zeta[1] | |
887 | + (mS[3]*zeta[0] + mS[4]*zeta[1] + mS[5]*zeta[2])*zeta[2]; | |
888 | } | |
889 | } | |
890 | ||
891 | void AliKFParticleBase::AddDaughterWithEnergyFitMC( const AliKFParticleBase &Daughter ) | |
892 | { | |
893 | //* Energy considered as an independent variable, fitted independently from momentum, without any constraints on mass | |
894 | ||
895 | //* Add daughter | |
896 | ||
897 | TransportToDecayVertex(); | |
898 | ||
899 | Double_t b[3]; | |
900 | Int_t maxIter = 1; | |
901 | ||
902 | if( !fIsLinearized ){ | |
903 | if( fNDF==-1 ){ | |
904 | Double_t ds, ds1; | |
905 | GetDStoParticle(Daughter, ds, ds1); | |
906 | TransportToDS( ds ); | |
907 | Double_t m[8]; | |
908 | Double_t mCd[36]; | |
909 | Daughter.Transport( ds1, m, mCd ); | |
910 | fVtxGuess[0] = .5*( fP[0] + m[0] ); | |
911 | fVtxGuess[1] = .5*( fP[1] + m[1] ); | |
912 | fVtxGuess[2] = .5*( fP[2] + m[2] ); | |
913 | } else { | |
914 | fVtxGuess[0] = fP[0]; | |
915 | fVtxGuess[1] = fP[1]; | |
916 | fVtxGuess[2] = fP[2]; | |
917 | } | |
918 | maxIter = 3; | |
919 | } | |
920 | ||
921 | for( Int_t iter=0; iter<maxIter; iter++ ){ | |
922 | ||
923 | { | |
924 | GetFieldValue( fVtxGuess, b ); | |
925 | const Double_t kCLight = 0.000299792458; | |
926 | b[0]*=kCLight; b[1]*=kCLight; b[2]*=kCLight; | |
927 | } | |
928 | ||
929 | Double_t *ffP = fP, *ffC = fC, tmpP[8], tmpC[36]; | |
930 | if( fNDF==-1 ){ | |
931 | GetMeasurement( fVtxGuess, tmpP, tmpC ); | |
932 | ffP = tmpP; | |
933 | ffC = tmpC; | |
934 | } | |
935 | Double_t m[8], mV[36]; | |
936 | ||
937 | if( Daughter.fC[35]>0 ){ | |
938 | Daughter.GetMeasurement( fVtxGuess, m, mV ); | |
939 | } else { | |
940 | for( Int_t i=0; i<8; i++ ) m[i] = Daughter.fP[i]; | |
941 | for( Int_t i=0; i<36; i++ ) mV[i] = Daughter.fC[i]; | |
942 | } | |
943 | //* | |
944 | ||
945 | Double_t mS[6]; | |
946 | { | |
947 | Double_t mSi[6] = { ffC[0]+mV[0], | |
948 | ffC[1]+mV[1], ffC[2]+mV[2], | |
949 | ffC[3]+mV[3], ffC[4]+mV[4], ffC[5]+mV[5] }; | |
950 | ||
951 | mS[0] = mSi[2]*mSi[5] - mSi[4]*mSi[4]; | |
952 | mS[1] = mSi[3]*mSi[4] - mSi[1]*mSi[5]; | |
953 | mS[2] = mSi[0]*mSi[5] - mSi[3]*mSi[3]; | |
954 | mS[3] = mSi[1]*mSi[4] - mSi[2]*mSi[3]; | |
955 | mS[4] = mSi[1]*mSi[3] - mSi[0]*mSi[4]; | |
956 | mS[5] = mSi[0]*mSi[2] - mSi[1]*mSi[1]; | |
957 | ||
958 | Double_t s = ( mSi[0]*mS[0] + mSi[1]*mS[1] + mSi[3]*mS[3] ); | |
959 | ||
960 | s = ( s > 1.E-20 ) ?1./s :0; | |
961 | mS[0]*=s; | |
962 | mS[1]*=s; | |
963 | mS[2]*=s; | |
964 | mS[3]*=s; | |
965 | mS[4]*=s; | |
966 | mS[5]*=s; | |
967 | } | |
968 | //* Residual (measured - estimated) | |
969 | ||
970 | Double_t zeta[3] = { m[0]-ffP[0], m[1]-ffP[1], m[2]-ffP[2] }; | |
971 | ||
972 | ||
973 | //* CHt = CH' | |
974 | ||
975 | Double_t mCHt0[7], mCHt1[7], mCHt2[7]; | |
976 | ||
977 | mCHt0[0]=ffC[ 0] ; mCHt1[0]=ffC[ 1] ; mCHt2[0]=ffC[ 3] ; | |
978 | mCHt0[1]=ffC[ 1] ; mCHt1[1]=ffC[ 2] ; mCHt2[1]=ffC[ 4] ; | |
979 | mCHt0[2]=ffC[ 3] ; mCHt1[2]=ffC[ 4] ; mCHt2[2]=ffC[ 5] ; | |
980 | mCHt0[3]=ffC[ 6] ; mCHt1[3]=ffC[ 7] ; mCHt2[3]=ffC[ 8] ; | |
981 | mCHt0[4]=ffC[10] ; mCHt1[4]=ffC[11] ; mCHt2[4]=ffC[12] ; | |
982 | mCHt0[5]=ffC[15] ; mCHt1[5]=ffC[16] ; mCHt2[5]=ffC[17] ; | |
983 | mCHt0[6]=ffC[21] ; mCHt1[6]=ffC[22] ; mCHt2[6]=ffC[23] ; | |
984 | ||
985 | //* Kalman gain K = mCH'*S | |
986 | ||
987 | Double_t k0[7], k1[7], k2[7]; | |
988 | ||
989 | for(Int_t i=0;i<7;++i){ | |
990 | k0[i] = mCHt0[i]*mS[0] + mCHt1[i]*mS[1] + mCHt2[i]*mS[3]; | |
991 | k1[i] = mCHt0[i]*mS[1] + mCHt1[i]*mS[2] + mCHt2[i]*mS[4]; | |
992 | k2[i] = mCHt0[i]*mS[3] + mCHt1[i]*mS[4] + mCHt2[i]*mS[5]; | |
993 | } | |
994 | ||
995 | //* New estimation of the vertex position | |
996 | ||
997 | if( iter<maxIter-1 ){ | |
998 | for(Int_t i=0; i<3; ++i) | |
999 | fVtxGuess[i]= ffP[i] + k0[i]*zeta[0]+k1[i]*zeta[1]+k2[i]*zeta[2]; | |
1000 | continue; | |
1001 | } | |
1002 | ||
1003 | // last itearation -> update the particle | |
1004 | ||
1005 | //* VHt = VH' | |
1006 | ||
1007 | Double_t mVHt0[7], mVHt1[7], mVHt2[7]; | |
1008 | ||
1009 | mVHt0[0]=mV[ 0] ; mVHt1[0]=mV[ 1] ; mVHt2[0]=mV[ 3] ; | |
1010 | mVHt0[1]=mV[ 1] ; mVHt1[1]=mV[ 2] ; mVHt2[1]=mV[ 4] ; | |
1011 | mVHt0[2]=mV[ 3] ; mVHt1[2]=mV[ 4] ; mVHt2[2]=mV[ 5] ; | |
1012 | mVHt0[3]=mV[ 6] ; mVHt1[3]=mV[ 7] ; mVHt2[3]=mV[ 8] ; | |
1013 | mVHt0[4]=mV[10] ; mVHt1[4]=mV[11] ; mVHt2[4]=mV[12] ; | |
1014 | mVHt0[5]=mV[15] ; mVHt1[5]=mV[16] ; mVHt2[5]=mV[17] ; | |
1015 | mVHt0[6]=mV[21] ; mVHt1[6]=mV[22] ; mVHt2[6]=mV[23] ; | |
1016 | ||
1017 | //* Kalman gain Km = mCH'*S | |
1018 | ||
1019 | Double_t km0[7], km1[7], km2[7]; | |
1020 | ||
1021 | for(Int_t i=0;i<7;++i){ | |
1022 | km0[i] = mVHt0[i]*mS[0] + mVHt1[i]*mS[1] + mVHt2[i]*mS[3]; | |
1023 | km1[i] = mVHt0[i]*mS[1] + mVHt1[i]*mS[2] + mVHt2[i]*mS[4]; | |
1024 | km2[i] = mVHt0[i]*mS[3] + mVHt1[i]*mS[4] + mVHt2[i]*mS[5]; | |
1025 | } | |
1026 | ||
1027 | for(Int_t i=0;i<7;++i) | |
1028 | ffP[i] = ffP[i] + k0[i]*zeta[0] + k1[i]*zeta[1] + k2[i]*zeta[2]; | |
1029 | ||
1030 | for(Int_t i=0;i<7;++i) | |
1031 | m[i] = m[i] - km0[i]*zeta[0] - km1[i]*zeta[1] - km2[i]*zeta[2]; | |
1032 | ||
1033 | for(Int_t i=0, k=0;i<7;++i){ | |
1034 | for(Int_t j=0;j<=i;++j,++k){ | |
1035 | ffC[k] = ffC[k] - (k0[i]*mCHt0[j] + k1[i]*mCHt1[j] + k2[i]*mCHt2[j] ); | |
1036 | } | |
1037 | } | |
1038 | ||
1039 | for(Int_t i=0, k=0;i<7;++i){ | |
1040 | for(Int_t j=0;j<=i;++j,++k){ | |
1041 | mV[k] = mV[k] - (km0[i]*mVHt0[j] + km1[i]*mVHt1[j] + km2[i]*mVHt2[j] ); | |
1042 | } | |
1043 | } | |
1044 | ||
5f653c6e | 1045 | Double_t mDf[7][7]; |
57ddc2cb | 1046 | |
1047 | for(Int_t i=0;i<7;++i){ | |
1048 | for(Int_t j=0;j<7;++j){ | |
5f653c6e | 1049 | mDf[i][j] = (km0[i]*mCHt0[j] + km1[i]*mCHt1[j] + km2[i]*mCHt2[j] ); |
57ddc2cb | 1050 | } |
1051 | } | |
1052 | ||
5f653c6e | 1053 | Double_t mJ1[7][7], mJ2[7][7]; |
57ddc2cb | 1054 | for(Int_t iPar1=0; iPar1<7; iPar1++) |
1055 | { | |
1056 | for(Int_t iPar2=0; iPar2<7; iPar2++) | |
1057 | { | |
5f653c6e | 1058 | mJ1[iPar1][iPar2] = 0; |
1059 | mJ2[iPar1][iPar2] = 0; | |
57ddc2cb | 1060 | } |
1061 | } | |
1062 | ||
1063 | Double_t mMassParticle = ffP[6]*ffP[6] - (ffP[3]*ffP[3] + ffP[4]*ffP[4] + ffP[5]*ffP[5]); | |
1064 | Double_t mMassDaughter = m[6]*m[6] - (m[3]*m[3] + m[4]*m[4] + m[5]*m[5]); | |
1065 | if(mMassParticle > 0) mMassParticle = TMath::Sqrt(mMassParticle); | |
1066 | if(mMassDaughter > 0) mMassDaughter = TMath::Sqrt(mMassDaughter); | |
1067 | ||
1068 | if( fMassHypo > -0.5) | |
5f653c6e | 1069 | SetMassConstraint(ffP,ffC,mJ1,fMassHypo); |
57ddc2cb | 1070 | else if((mMassParticle < SumDaughterMass) || (ffP[6]<0) ) |
5f653c6e | 1071 | SetMassConstraint(ffP,ffC,mJ1,SumDaughterMass); |
57ddc2cb | 1072 | |
1073 | if(Daughter.fMassHypo > -0.5) | |
5f653c6e | 1074 | SetMassConstraint(m,mV,mJ2,Daughter.fMassHypo); |
57ddc2cb | 1075 | else if((mMassDaughter < Daughter.SumDaughterMass) || (m[6] < 0) ) |
5f653c6e | 1076 | SetMassConstraint(m,mV,mJ2,Daughter.SumDaughterMass); |
57ddc2cb | 1077 | |
5f653c6e | 1078 | Double_t mDJ[7][7]; |
57ddc2cb | 1079 | |
1080 | for(Int_t i=0; i<7; i++) { | |
1081 | for(Int_t j=0; j<7; j++) { | |
5f653c6e | 1082 | mDJ[i][j] = 0; |
57ddc2cb | 1083 | for(Int_t k=0; k<7; k++) { |
5f653c6e | 1084 | mDJ[i][j] += mDf[i][k]*mJ1[j][k]; |
57ddc2cb | 1085 | } |
1086 | } | |
1087 | } | |
1088 | ||
1089 | for(Int_t i=0; i<7; ++i){ | |
1090 | for(Int_t j=0; j<7; ++j){ | |
5f653c6e | 1091 | mDf[i][j]=0; |
57ddc2cb | 1092 | for(Int_t l=0; l<7; l++){ |
5f653c6e | 1093 | mDf[i][j] += mJ2[i][l]*mDJ[l][j]; |
57ddc2cb | 1094 | } |
1095 | } | |
1096 | } | |
1097 | ||
1098 | //* Add the daughter momentum to the particle momentum | |
1099 | ||
1100 | ffP[ 3] += m[ 3]; | |
1101 | ffP[ 4] += m[ 4]; | |
1102 | ffP[ 5] += m[ 5]; | |
1103 | ffP[ 6] += m[ 6]; | |
1104 | ||
1105 | ffC[ 9] += mV[ 9]; | |
1106 | ffC[13] += mV[13]; | |
1107 | ffC[14] += mV[14]; | |
1108 | ffC[18] += mV[18]; | |
1109 | ffC[19] += mV[19]; | |
1110 | ffC[20] += mV[20]; | |
1111 | ffC[24] += mV[24]; | |
1112 | ffC[25] += mV[25]; | |
1113 | ffC[26] += mV[26]; | |
1114 | ffC[27] += mV[27]; | |
1115 | ||
5f653c6e | 1116 | ffC[6 ] += mDf[3][0]; ffC[7 ] += mDf[3][1]; ffC[8 ] += mDf[3][2]; |
1117 | ffC[10] += mDf[4][0]; ffC[11] += mDf[4][1]; ffC[12] += mDf[4][2]; | |
1118 | ffC[15] += mDf[5][0]; ffC[16] += mDf[5][1]; ffC[17] += mDf[5][2]; | |
1119 | ffC[21] += mDf[6][0]; ffC[22] += mDf[6][1]; ffC[23] += mDf[6][2]; | |
57ddc2cb | 1120 | |
5f653c6e | 1121 | ffC[9 ] += mDf[3][3] + mDf[3][3]; |
1122 | ffC[13] += mDf[4][3] + mDf[3][4]; ffC[14] += mDf[4][4] + mDf[4][4]; | |
1123 | ffC[18] += mDf[5][3] + mDf[3][5]; ffC[19] += mDf[5][4] + mDf[4][5]; ffC[20] += mDf[5][5] + mDf[5][5]; | |
1124 | ffC[24] += mDf[6][3] + mDf[3][6]; ffC[25] += mDf[6][4] + mDf[4][6]; ffC[26] += mDf[6][5] + mDf[5][6]; ffC[27] += mDf[6][6] + mDf[6][6]; | |
57ddc2cb | 1125 | |
1126 | //* New estimation of the vertex position r += K*zeta | |
1127 | ||
1128 | for(Int_t i=0;i<7;++i) | |
1129 | fP[i] = ffP[i]; | |
1130 | ||
1131 | //* New covariance matrix C -= K*(mCH')' | |
1132 | ||
1133 | for(Int_t i=0, k=0;i<7;++i){ | |
1134 | for(Int_t j=0;j<=i;++j,++k){ | |
1135 | fC[k] = ffC[k]; | |
1136 | } | |
1137 | } | |
1138 | //* Calculate Chi^2 | |
1139 | ||
1140 | fNDF += 2; | |
1141 | fQ += Daughter.GetQ(); | |
1142 | fSFromDecay = 0; | |
1143 | fChi2 += (mS[0]*zeta[0] + mS[1]*zeta[1] + mS[3]*zeta[2])*zeta[0] | |
1144 | + (mS[1]*zeta[0] + mS[2]*zeta[1] + mS[4]*zeta[2])*zeta[1] | |
1145 | + (mS[3]*zeta[0] + mS[4]*zeta[1] + mS[5]*zeta[2])*zeta[2]; | |
1146 | } | |
1147 | } | |
1148 | ||
f826d409 | 1149 | void AliKFParticleBase::SetProductionVertex( const AliKFParticleBase &Vtx ) |
1150 | { | |
706952f5 | 1151 | //* Set production vertex for the particle, when the particle was not used in the vertex fit |
f826d409 | 1152 | |
1153 | const Double_t *m = Vtx.fP, *mV = Vtx.fC; | |
1154 | ||
e7b09c95 | 1155 | Bool_t noS = ( fC[35]<=0 ); // no decay length allowed |
1156 | ||
1157 | if( noS ){ | |
1158 | TransportToDecayVertex(); | |
1159 | fP[7] = 0; | |
55ac3e1e | 1160 | fC[28] = fC[29] = fC[30] = fC[31] = fC[32] = fC[33] = fC[34] = fC[35] = 0; |
e7b09c95 | 1161 | } else { |
55ac3e1e | 1162 | TransportToDS( GetDStoPoint( m ) ); |
e7b09c95 | 1163 | fP[7] = -fSFromDecay; |
55ac3e1e | 1164 | fC[28] = fC[29] = fC[30] = fC[31] = fC[32] = fC[33] = fC[34] = 0; |
0d76e8c6 | 1165 | fC[35] = 0.1; |
55ac3e1e | 1166 | |
e7b09c95 | 1167 | Convert(1); |
1168 | } | |
1169 | ||
f826d409 | 1170 | Double_t mAi[6]; |
706952f5 | 1171 | |
446ce366 | 1172 | InvertSym3( fC, mAi ); |
f826d409 | 1173 | |
1174 | Double_t mB[5][3]; | |
1175 | ||
1176 | mB[0][0] = fC[ 6]*mAi[0] + fC[ 7]*mAi[1] + fC[ 8]*mAi[3]; | |
1177 | mB[0][1] = fC[ 6]*mAi[1] + fC[ 7]*mAi[2] + fC[ 8]*mAi[4]; | |
1178 | mB[0][2] = fC[ 6]*mAi[3] + fC[ 7]*mAi[4] + fC[ 8]*mAi[5]; | |
1179 | ||
1180 | mB[1][0] = fC[10]*mAi[0] + fC[11]*mAi[1] + fC[12]*mAi[3]; | |
1181 | mB[1][1] = fC[10]*mAi[1] + fC[11]*mAi[2] + fC[12]*mAi[4]; | |
1182 | mB[1][2] = fC[10]*mAi[3] + fC[11]*mAi[4] + fC[12]*mAi[5]; | |
1183 | ||
1184 | mB[2][0] = fC[15]*mAi[0] + fC[16]*mAi[1] + fC[17]*mAi[3]; | |
1185 | mB[2][1] = fC[15]*mAi[1] + fC[16]*mAi[2] + fC[17]*mAi[4]; | |
1186 | mB[2][2] = fC[15]*mAi[3] + fC[16]*mAi[4] + fC[17]*mAi[5]; | |
1187 | ||
1188 | mB[3][0] = fC[21]*mAi[0] + fC[22]*mAi[1] + fC[23]*mAi[3]; | |
1189 | mB[3][1] = fC[21]*mAi[1] + fC[22]*mAi[2] + fC[23]*mAi[4]; | |
1190 | mB[3][2] = fC[21]*mAi[3] + fC[22]*mAi[4] + fC[23]*mAi[5]; | |
1191 | ||
1192 | mB[4][0] = fC[28]*mAi[0] + fC[29]*mAi[1] + fC[30]*mAi[3]; | |
1193 | mB[4][1] = fC[28]*mAi[1] + fC[29]*mAi[2] + fC[30]*mAi[4]; | |
1194 | mB[4][2] = fC[28]*mAi[3] + fC[29]*mAi[4] + fC[30]*mAi[5]; | |
1195 | ||
1196 | Double_t z[3] = { m[0]-fP[0], m[1]-fP[1], m[2]-fP[2] }; | |
1197 | ||
1198 | { | |
a65041d0 | 1199 | Double_t mAVi[6] = { fC[0]-mV[0], fC[1]-mV[1], fC[2]-mV[2], |
f826d409 | 1200 | fC[3]-mV[3], fC[4]-mV[4], fC[5]-mV[5] }; |
706952f5 | 1201 | |
a65041d0 | 1202 | if( !InvertSym3( mAVi, mAVi ) ){ |
706952f5 | 1203 | |
1204 | Double_t dChi2 = ( +(mAVi[0]*z[0] + mAVi[1]*z[1] + mAVi[3]*z[2])*z[0] | |
1205 | +(mAVi[1]*z[0] + mAVi[2]*z[1] + mAVi[4]*z[2])*z[1] | |
a65041d0 | 1206 | +(mAVi[3]*z[0] + mAVi[4]*z[1] + mAVi[5]*z[2])*z[2] ); |
706952f5 | 1207 | |
1208 | // Take Abs(dChi2) here. Negative value of 'det' or 'dChi2' shows that the particle | |
1209 | // was not used in the production vertex fit | |
1210 | ||
1211 | fChi2+= TMath::Abs( dChi2 ); | |
1212 | } | |
f826d409 | 1213 | fNDF += 2; |
f826d409 | 1214 | } |
1215 | ||
1216 | fP[0] = m[0]; | |
1217 | fP[1] = m[1]; | |
1218 | fP[2] = m[2]; | |
1219 | fP[3]+= mB[0][0]*z[0] + mB[0][1]*z[1] + mB[0][2]*z[2]; | |
1220 | fP[4]+= mB[1][0]*z[0] + mB[1][1]*z[1] + mB[1][2]*z[2]; | |
1221 | fP[5]+= mB[2][0]*z[0] + mB[2][1]*z[1] + mB[2][2]*z[2]; | |
1222 | fP[6]+= mB[3][0]*z[0] + mB[3][1]*z[1] + mB[3][2]*z[2]; | |
1223 | fP[7]+= mB[4][0]*z[0] + mB[4][1]*z[1] + mB[4][2]*z[2]; | |
1224 | ||
1225 | Double_t d0, d1, d2; | |
1226 | ||
1227 | fC[0] = mV[0]; | |
1228 | fC[1] = mV[1]; | |
1229 | fC[2] = mV[2]; | |
1230 | fC[3] = mV[3]; | |
1231 | fC[4] = mV[4]; | |
1232 | fC[5] = mV[5]; | |
1233 | ||
1234 | d0= mB[0][0]*mV[0] + mB[0][1]*mV[1] + mB[0][2]*mV[3] - fC[ 6]; | |
1235 | d1= mB[0][0]*mV[1] + mB[0][1]*mV[2] + mB[0][2]*mV[4] - fC[ 7]; | |
1236 | d2= mB[0][0]*mV[3] + mB[0][1]*mV[4] + mB[0][2]*mV[5] - fC[ 8]; | |
1237 | ||
1238 | fC[ 6]+= d0; | |
1239 | fC[ 7]+= d1; | |
1240 | fC[ 8]+= d2; | |
1241 | fC[ 9]+= d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
1242 | ||
1243 | d0= mB[1][0]*mV[0] + mB[1][1]*mV[1] + mB[1][2]*mV[3] - fC[10]; | |
1244 | d1= mB[1][0]*mV[1] + mB[1][1]*mV[2] + mB[1][2]*mV[4] - fC[11]; | |
1245 | d2= mB[1][0]*mV[3] + mB[1][1]*mV[4] + mB[1][2]*mV[5] - fC[12]; | |
1246 | ||
1247 | fC[10]+= d0; | |
1248 | fC[11]+= d1; | |
1249 | fC[12]+= d2; | |
1250 | fC[13]+= d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
1251 | fC[14]+= d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
1252 | ||
1253 | d0= mB[2][0]*mV[0] + mB[2][1]*mV[1] + mB[2][2]*mV[3] - fC[15]; | |
1254 | d1= mB[2][0]*mV[1] + mB[2][1]*mV[2] + mB[2][2]*mV[4] - fC[16]; | |
1255 | d2= mB[2][0]*mV[3] + mB[2][1]*mV[4] + mB[2][2]*mV[5] - fC[17]; | |
1256 | ||
1257 | fC[15]+= d0; | |
1258 | fC[16]+= d1; | |
1259 | fC[17]+= d2; | |
1260 | fC[18]+= d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
1261 | fC[19]+= d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
1262 | fC[20]+= d0*mB[2][0] + d1*mB[2][1] + d2*mB[2][2]; | |
1263 | ||
1264 | d0= mB[3][0]*mV[0] + mB[3][1]*mV[1] + mB[3][2]*mV[3] - fC[21]; | |
1265 | d1= mB[3][0]*mV[1] + mB[3][1]*mV[2] + mB[3][2]*mV[4] - fC[22]; | |
1266 | d2= mB[3][0]*mV[3] + mB[3][1]*mV[4] + mB[3][2]*mV[5] - fC[23]; | |
1267 | ||
1268 | fC[21]+= d0; | |
1269 | fC[22]+= d1; | |
1270 | fC[23]+= d2; | |
1271 | fC[24]+= d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
1272 | fC[25]+= d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
1273 | fC[26]+= d0*mB[2][0] + d1*mB[2][1] + d2*mB[2][2]; | |
1274 | fC[27]+= d0*mB[3][0] + d1*mB[3][1] + d2*mB[3][2]; | |
1275 | ||
1276 | d0= mB[4][0]*mV[0] + mB[4][1]*mV[1] + mB[4][2]*mV[3] - fC[28]; | |
1277 | d1= mB[4][0]*mV[1] + mB[4][1]*mV[2] + mB[4][2]*mV[4] - fC[29]; | |
1278 | d2= mB[4][0]*mV[3] + mB[4][1]*mV[4] + mB[4][2]*mV[5] - fC[30]; | |
1279 | ||
1280 | fC[28]+= d0; | |
1281 | fC[29]+= d1; | |
1282 | fC[30]+= d2; | |
1283 | fC[31]+= d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
1284 | fC[32]+= d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
1285 | fC[33]+= d0*mB[2][0] + d1*mB[2][1] + d2*mB[2][2]; | |
1286 | fC[34]+= d0*mB[3][0] + d1*mB[3][1] + d2*mB[3][2]; | |
1287 | fC[35]+= d0*mB[4][0] + d1*mB[4][1] + d2*mB[4][2]; | |
1288 | ||
e7b09c95 | 1289 | if( noS ){ |
1290 | fP[7] = 0; | |
55ac3e1e | 1291 | fC[28] = fC[29] = fC[30] = fC[31] = fC[32] = fC[33] = fC[34] = fC[35] = 0; |
e7b09c95 | 1292 | } else { |
1293 | TransportToDS( fP[7] ); | |
1294 | Convert(0); | |
1295 | } | |
1296 | ||
f826d409 | 1297 | fSFromDecay = 0; |
f826d409 | 1298 | } |
1299 | ||
5f653c6e | 1300 | void AliKFParticleBase::SetMassConstraint( Double_t *mP, Double_t *mC, Double_t mJ[7][7], Double_t mass ) |
57ddc2cb | 1301 | { |
5f653c6e | 1302 | //* Set nonlinear mass constraint (Mass) on the state vector mP with a covariance matrix mC. |
1303 | ||
1304 | const Double_t energy2 = mP[6]*mP[6], p2 = mP[3]*mP[3]+mP[4]*mP[4]+mP[5]*mP[5], mass2 = mass*mass; | |
57ddc2cb | 1305 | |
5f653c6e | 1306 | const Double_t a = energy2 - p2 + 2.*mass2; |
1307 | const Double_t b = -2.*(energy2 + p2); | |
1308 | const Double_t c = energy2 - p2 - mass2; | |
57ddc2cb | 1309 | |
5f653c6e | 1310 | Double_t lambda = 0; |
1311 | if(TMath::Abs(b) > 1.e-10) lambda = -c / b; | |
57ddc2cb | 1312 | |
5f653c6e | 1313 | Double_t d = 4.*energy2*p2 - mass2*(energy2-p2-2.*mass2); |
1314 | if(d>=0 && TMath::Abs(a) > 1.e-10) lambda = (energy2 + p2 - sqrt(d))/a; | |
57ddc2cb | 1315 | |
5f653c6e | 1316 | if(mP[6] < 0) //If energy < 0 we need a lambda < 0 |
1317 | lambda = -1000000.; //Empirical, a better solution should be found | |
57ddc2cb | 1318 | |
1319 | Int_t iIter=0; | |
1320 | for(iIter=0; iIter<100; iIter++) | |
1321 | { | |
5f653c6e | 1322 | Double_t lambda2 = lambda*lambda; |
1323 | Double_t lambda4 = lambda2*lambda2; | |
57ddc2cb | 1324 | |
5f653c6e | 1325 | Double_t lambda0 = lambda; |
f826d409 | 1326 | |
5f653c6e | 1327 | Double_t f = -mass2 * lambda4 + a*lambda2 + b*lambda + c; |
1328 | Double_t df = -4.*mass2 * lambda2*lambda + 2.*a*lambda + b; | |
121dc8bd | 1329 | if(TMath::Abs(df) < 1.e-10) break; |
5f653c6e | 1330 | lambda -= f/df; |
1331 | if(TMath::Abs(lambda0 - lambda) < 1.e-8) break; | |
57ddc2cb | 1332 | } |
1333 | ||
5f653c6e | 1334 | const Double_t lpi = 1./(1. + lambda); |
1335 | const Double_t lmi = 1./(1. - lambda); | |
1336 | const Double_t lp2i = lpi*lpi; | |
1337 | const Double_t lm2i = lmi*lmi; | |
57ddc2cb | 1338 | |
5f653c6e | 1339 | Double_t lambda2 = lambda*lambda; |
57ddc2cb | 1340 | |
5f653c6e | 1341 | Double_t dfl = -4.*mass2 * lambda2*lambda + 2.*a*lambda + b; |
a1b97102 | 1342 | Double_t dfx[7] = {0};//,0,0,0}; |
5f653c6e | 1343 | dfx[0] = -2.*(1. + lambda)*(1. + lambda)*mP[3]; |
1344 | dfx[1] = -2.*(1. + lambda)*(1. + lambda)*mP[4]; | |
1345 | dfx[2] = -2.*(1. + lambda)*(1. + lambda)*mP[5]; | |
1346 | dfx[3] = 2.*(1. - lambda)*(1. - lambda)*mP[6]; | |
c3af7f6d | 1347 | Double_t dlx[4] = {1,1,1,1}; |
121dc8bd | 1348 | if(TMath::Abs(dfl) > 1.e-10 ) |
57ddc2cb | 1349 | { |
c3af7f6d | 1350 | for(int i=0; i<4; i++) |
57ddc2cb | 1351 | dlx[i] = -dfx[i] / dfl; |
1352 | } | |
1353 | ||
5f653c6e | 1354 | Double_t dxx[4] = {mP[3]*lm2i, mP[4]*lm2i, mP[5]*lm2i, -mP[6]*lp2i}; |
57ddc2cb | 1355 | |
1356 | for(Int_t i=0; i<7; i++) | |
1357 | for(Int_t j=0; j<7; j++) | |
5f653c6e | 1358 | mJ[i][j]=0; |
1359 | mJ[0][0] = 1.; | |
1360 | mJ[1][1] = 1.; | |
1361 | mJ[2][2] = 1.; | |
57ddc2cb | 1362 | |
1363 | for(Int_t i=3; i<7; i++) | |
1364 | for(Int_t j=3; j<7; j++) | |
5f653c6e | 1365 | mJ[i][j] = dlx[j-3]*dxx[i-3]; |
57ddc2cb | 1366 | |
1367 | for(Int_t i=3; i<6; i++) | |
5f653c6e | 1368 | mJ[i][i] += lmi; |
1369 | mJ[6][6] += lpi; | |
57ddc2cb | 1370 | |
5f653c6e | 1371 | Double_t mCJ[7][7]; |
57ddc2cb | 1372 | |
1373 | for(Int_t i=0; i<7; i++) { | |
1374 | for(Int_t j=0; j<7; j++) { | |
5f653c6e | 1375 | mCJ[i][j] = 0; |
57ddc2cb | 1376 | for(Int_t k=0; k<7; k++) { |
5f653c6e | 1377 | mCJ[i][j] += mC[IJ(i,k)]*mJ[j][k]; |
57ddc2cb | 1378 | } |
1379 | } | |
1380 | } | |
1381 | ||
1382 | for(Int_t i=0; i<7; ++i){ | |
1383 | for(Int_t j=0; j<=i; ++j){ | |
1384 | mC[IJ(i,j)]=0; | |
1385 | for(Int_t l=0; l<7; l++){ | |
5f653c6e | 1386 | mC[IJ(i,j)] += mJ[i][l]*mCJ[l][j]; |
57ddc2cb | 1387 | } |
1388 | } | |
1389 | } | |
1390 | ||
5f653c6e | 1391 | mP[3] *= lmi; |
1392 | mP[4] *= lmi; | |
1393 | mP[5] *= lmi; | |
1394 | mP[6] *= lpi; | |
57ddc2cb | 1395 | } |
1396 | ||
5f653c6e | 1397 | void AliKFParticleBase::SetNonlinearMassConstraint( Double_t mass ) |
57ddc2cb | 1398 | { |
5f653c6e | 1399 | //* Set nonlinear mass constraint (mass) |
1400 | ||
1401 | Double_t mJ[7][7]; | |
1402 | SetMassConstraint( fP, fC, mJ, mass ); | |
1403 | fMassHypo = mass; | |
1404 | SumDaughterMass = mass; | |
57ddc2cb | 1405 | } |
f826d409 | 1406 | |
e7b09c95 | 1407 | void AliKFParticleBase::SetMassConstraint( Double_t Mass, Double_t SigmaMass ) |
f826d409 | 1408 | { |
706952f5 | 1409 | //* Set hard( SigmaMass=0 ) or soft (SigmaMass>0) mass constraint |
1410 | ||
57ddc2cb | 1411 | fMassHypo = Mass; |
1412 | SumDaughterMass = Mass; | |
1413 | ||
706952f5 | 1414 | Double_t m2 = Mass*Mass; // measurement, weighted by Mass |
1415 | Double_t s2 = m2*SigmaMass*SigmaMass; // sigma^2 | |
1416 | ||
1417 | Double_t p2 = fP[3]*fP[3] + fP[4]*fP[4] + fP[5]*fP[5]; | |
1418 | Double_t e0 = TMath::Sqrt(m2+p2); | |
f826d409 | 1419 | |
1420 | Double_t mH[8]; | |
1421 | mH[0] = mH[1] = mH[2] = 0.; | |
1422 | mH[3] = -2*fP[3]; | |
1423 | mH[4] = -2*fP[4]; | |
1424 | mH[5] = -2*fP[5]; | |
706952f5 | 1425 | mH[6] = 2*fP[6];//e0; |
f826d409 | 1426 | mH[7] = 0; |
f826d409 | 1427 | |
706952f5 | 1428 | Double_t zeta = e0*e0 - e0*fP[6]; |
1429 | zeta = m2 - (fP[6]*fP[6]-p2); | |
f826d409 | 1430 | |
706952f5 | 1431 | Double_t mCHt[8], s2_est=0; |
1432 | for( Int_t i=0; i<8; ++i ){ | |
f826d409 | 1433 | mCHt[i] = 0.0; |
1434 | for (Int_t j=0;j<8;++j) mCHt[i] += Cij(i,j)*mH[j]; | |
706952f5 | 1435 | s2_est += mH[i]*mCHt[i]; |
f826d409 | 1436 | } |
1437 | ||
706952f5 | 1438 | if( s2_est<1.e-20 ) return; // calculated mass error is already 0, |
1439 | // the particle can not be constrained on mass | |
1440 | ||
1441 | Double_t w2 = 1./( s2 + s2_est ); | |
1442 | fChi2 += zeta*zeta*w2; | |
f826d409 | 1443 | fNDF += 1; |
1444 | for( Int_t i=0, ii=0; i<8; ++i ){ | |
706952f5 | 1445 | Double_t ki = mCHt[i]*w2; |
f826d409 | 1446 | fP[i]+= ki*zeta; |
1447 | for(Int_t j=0;j<=i;++j) fC[ii++] -= ki*mCHt[j]; | |
1448 | } | |
1449 | } | |
1450 | ||
1451 | ||
e7b09c95 | 1452 | void AliKFParticleBase::SetNoDecayLength() |
1453 | { | |
1454 | //* Set no decay length for resonances | |
1455 | ||
1456 | TransportToDecayVertex(); | |
1457 | ||
1458 | Double_t h[8]; | |
1459 | h[0] = h[1] = h[2] = h[3] = h[4] = h[5] = h[6] = 0; | |
1460 | h[7] = 1; | |
1461 | ||
1462 | Double_t zeta = 0 - fP[7]; | |
1463 | for(Int_t i=0;i<8;++i) zeta -= h[i]*(fP[i]-fP[i]); | |
1464 | ||
1465 | Double_t s = fC[35]; | |
1466 | if( s>1.e-20 ){ | |
1467 | s = 1./s; | |
1468 | fChi2 += zeta*zeta*s; | |
1469 | fNDF += 1; | |
1470 | for( Int_t i=0, ii=0; i<7; ++i ){ | |
1471 | Double_t ki = fC[28+i]*s; | |
1472 | fP[i]+= ki*zeta; | |
1473 | for(Int_t j=0;j<=i;++j) fC[ii++] -= ki*fC[28+j]; | |
1474 | } | |
1475 | } | |
1476 | fP[7] = 0; | |
55ac3e1e | 1477 | fC[28] = fC[29] = fC[30] = fC[31] = fC[32] = fC[33] = fC[34] = fC[35] = 0; |
e7b09c95 | 1478 | } |
1479 | ||
1480 | ||
f826d409 | 1481 | void AliKFParticleBase::Construct( const AliKFParticleBase* vDaughters[], Int_t NDaughters, |
706952f5 | 1482 | const AliKFParticleBase *Parent, Double_t Mass, Bool_t IsConstrained ) |
f826d409 | 1483 | { |
1484 | //* Full reconstruction in one go | |
1485 | ||
1486 | Int_t maxIter = 1; | |
1487 | bool wasLinearized = fIsLinearized; | |
706952f5 | 1488 | if( !fIsLinearized || IsConstrained ){ |
616ffc76 | 1489 | //fVtxGuess[0] = fVtxGuess[1] = fVtxGuess[2] = 0; //!!!! |
1490 | fVtxGuess[0] = GetX(); | |
1491 | fVtxGuess[1] = GetY(); | |
1492 | fVtxGuess[2] = GetZ(); | |
f826d409 | 1493 | fIsLinearized = 1; |
1494 | maxIter = 3; | |
1495 | } | |
1496 | ||
706952f5 | 1497 | Double_t constraintC[6]; |
1498 | ||
1499 | if( IsConstrained ){ | |
1500 | for(Int_t i=0;i<6;++i) constraintC[i]=fC[i]; | |
1501 | } else { | |
1502 | for(Int_t i=0;i<6;++i) constraintC[i]=0.; | |
1503 | constraintC[0] = constraintC[2] = constraintC[5] = 100.; | |
1504 | } | |
1505 | ||
1506 | ||
f826d409 | 1507 | for( Int_t iter=0; iter<maxIter; iter++ ){ |
f826d409 | 1508 | fAtProductionVertex = 0; |
1509 | fSFromDecay = 0; | |
1510 | fP[0] = fVtxGuess[0]; | |
1511 | fP[1] = fVtxGuess[1]; | |
1512 | fP[2] = fVtxGuess[2]; | |
1513 | fP[3] = 0; | |
1514 | fP[4] = 0; | |
1515 | fP[5] = 0; | |
1516 | fP[6] = 0; | |
1517 | fP[7] = 0; | |
57ddc2cb | 1518 | SumDaughterMass = 0; |
1519 | ||
706952f5 | 1520 | for(Int_t i=0;i<6; ++i) fC[i]=constraintC[i]; |
1521 | for(Int_t i=6;i<36;++i) fC[i]=0.; | |
f826d409 | 1522 | fC[35] = 1.; |
57ddc2cb | 1523 | |
706952f5 | 1524 | fNDF = IsConstrained ?0 :-3; |
f826d409 | 1525 | fChi2 = 0.; |
1526 | fQ = 0; | |
1527 | ||
4bbc290d | 1528 | for( Int_t itr =0; itr<NDaughters; itr++ ){ |
1529 | AddDaughter( *vDaughters[itr] ); | |
1530 | } | |
f826d409 | 1531 | if( iter<maxIter-1){ |
1532 | for( Int_t i=0; i<3; i++ ) fVtxGuess[i] = fP[i]; | |
1533 | } | |
1534 | } | |
1535 | fIsLinearized = wasLinearized; | |
1536 | ||
1537 | if( Mass>=0 ) SetMassConstraint( Mass ); | |
1538 | if( Parent ) SetProductionVertex( *Parent ); | |
1539 | } | |
1540 | ||
1541 | ||
1542 | void AliKFParticleBase::Convert( bool ToProduction ) | |
1543 | { | |
1544 | //* Tricky function - convert the particle error along its trajectory to | |
1545 | //* the value which corresponds to its production/decay vertex | |
1546 | //* It is done by combination of the error of decay length with the position errors | |
1547 | ||
1548 | Double_t fld[3]; | |
1549 | { | |
1550 | GetFieldValue( fP, fld ); | |
1551 | const Double_t kCLight = fQ*0.000299792458; | |
1552 | fld[0]*=kCLight; fld[1]*=kCLight; fld[2]*=kCLight; | |
1553 | } | |
1554 | ||
1555 | Double_t h[6]; | |
1556 | ||
1557 | h[0] = fP[3]; | |
1558 | h[1] = fP[4]; | |
1559 | h[2] = fP[5]; | |
1560 | if( ToProduction ){ h[0]=-h[0]; h[1]=-h[1]; h[2]=-h[2]; } | |
1561 | h[3] = h[1]*fld[2]-h[2]*fld[1]; | |
1562 | h[4] = h[2]*fld[0]-h[0]*fld[2]; | |
1563 | h[5] = h[0]*fld[1]-h[1]*fld[0]; | |
1564 | ||
1565 | Double_t c; | |
1566 | ||
1567 | c = fC[28]+h[0]*fC[35]; | |
1568 | fC[ 0]+= h[0]*(c+fC[28]); | |
1569 | fC[28] = c; | |
1570 | ||
1571 | fC[ 1]+= h[1]*fC[28] + h[0]*fC[29]; | |
1572 | c = fC[29]+h[1]*fC[35]; | |
1573 | fC[ 2]+= h[1]*(c+fC[29]); | |
1574 | fC[29] = c; | |
1575 | ||
1576 | fC[ 3]+= h[2]*fC[28] + h[0]*fC[30]; | |
1577 | fC[ 4]+= h[2]*fC[29] + h[1]*fC[30]; | |
1578 | c = fC[30]+h[2]*fC[35]; | |
1579 | fC[ 5]+= h[2]*(c+fC[30]); | |
1580 | fC[30] = c; | |
1581 | ||
1582 | fC[ 6]+= h[3]*fC[28] + h[0]*fC[31]; | |
1583 | fC[ 7]+= h[3]*fC[29] + h[1]*fC[31]; | |
1584 | fC[ 8]+= h[3]*fC[30] + h[2]*fC[31]; | |
1585 | c = fC[31]+h[3]*fC[35]; | |
1586 | fC[ 9]+= h[3]*(c+fC[31]); | |
1587 | fC[31] = c; | |
1588 | ||
1589 | fC[10]+= h[4]*fC[28] + h[0]*fC[32]; | |
1590 | fC[11]+= h[4]*fC[29] + h[1]*fC[32]; | |
1591 | fC[12]+= h[4]*fC[30] + h[2]*fC[32]; | |
1592 | fC[13]+= h[4]*fC[31] + h[3]*fC[32]; | |
1593 | c = fC[32]+h[4]*fC[35]; | |
1594 | fC[14]+= h[4]*(c+fC[32]); | |
1595 | fC[32] = c; | |
1596 | ||
1597 | fC[15]+= h[5]*fC[28] + h[0]*fC[33]; | |
1598 | fC[16]+= h[5]*fC[29] + h[1]*fC[33]; | |
1599 | fC[17]+= h[5]*fC[30] + h[2]*fC[33]; | |
1600 | fC[18]+= h[5]*fC[31] + h[3]*fC[33]; | |
1601 | fC[19]+= h[5]*fC[32] + h[4]*fC[33]; | |
1602 | c = fC[33]+h[5]*fC[35]; | |
1603 | fC[20]+= h[5]*(c+fC[33]); | |
1604 | fC[33] = c; | |
1605 | ||
1606 | fC[21]+= h[0]*fC[34]; | |
1607 | fC[22]+= h[1]*fC[34]; | |
1608 | fC[23]+= h[2]*fC[34]; | |
1609 | fC[24]+= h[3]*fC[34]; | |
1610 | fC[25]+= h[4]*fC[34]; | |
1611 | fC[26]+= h[5]*fC[34]; | |
1612 | } | |
1613 | ||
1614 | ||
1615 | void AliKFParticleBase::TransportToDecayVertex() | |
1616 | { | |
1617 | //* Transport the particle to its decay vertex | |
1618 | ||
1619 | if( fSFromDecay != 0 ) TransportToDS( -fSFromDecay ); | |
1620 | if( fAtProductionVertex ) Convert(0); | |
1621 | fAtProductionVertex = 0; | |
1622 | } | |
1623 | ||
1624 | void AliKFParticleBase::TransportToProductionVertex() | |
1625 | { | |
1626 | //* Transport the particle to its production vertex | |
616ffc76 | 1627 | |
f826d409 | 1628 | if( fSFromDecay != -fP[7] ) TransportToDS( -fSFromDecay-fP[7] ); |
1629 | if( !fAtProductionVertex ) Convert( 1 ); | |
1630 | fAtProductionVertex = 1; | |
1631 | } | |
1632 | ||
1633 | ||
1634 | void AliKFParticleBase::TransportToDS( Double_t dS ) | |
1635 | { | |
1636 | //* Transport the particle on dS parameter (SignedPath/Momentum) | |
616ffc76 | 1637 | |
f826d409 | 1638 | Transport( dS, fP, fC ); |
1639 | fSFromDecay+= dS; | |
1640 | } | |
1641 | ||
1642 | ||
1643 | Double_t AliKFParticleBase::GetDStoPointLine( const Double_t xyz[] ) const | |
1644 | { | |
1645 | //* Get dS to a certain space point without field | |
1646 | ||
1647 | Double_t p2 = fP[3]*fP[3] + fP[4]*fP[4] + fP[5]*fP[5]; | |
1648 | if( p2<1.e-4 ) p2 = 1; | |
1649 | return ( fP[3]*(xyz[0]-fP[0]) + fP[4]*(xyz[1]-fP[1]) + fP[5]*(xyz[2]-fP[2]) )/p2; | |
1650 | } | |
1651 | ||
1652 | ||
1653 | Double_t AliKFParticleBase::GetDStoPointBz( Double_t B, const Double_t xyz[] ) | |
1654 | const | |
1655 | { | |
616ffc76 | 1656 | |
f826d409 | 1657 | //* Get dS to a certain space point for Bz field |
f826d409 | 1658 | const Double_t kCLight = 0.000299792458; |
1659 | Double_t bq = B*fQ*kCLight; | |
1660 | Double_t pt2 = fP[3]*fP[3] + fP[4]*fP[4]; | |
1661 | if( pt2<1.e-4 ) return 0; | |
1662 | Double_t dx = xyz[0] - fP[0]; | |
1663 | Double_t dy = xyz[1] - fP[1]; | |
1664 | Double_t a = dx*fP[3]+dy*fP[4]; | |
706952f5 | 1665 | Double_t dS; |
1666 | ||
1667 | if( TMath::Abs(bq)<1.e-8 ) dS = a/pt2; | |
1668 | else dS = TMath::ATan2( bq*a, pt2 + bq*(dy*fP[3] -dx*fP[4]) )/bq; | |
1669 | ||
1670 | if(0){ | |
1671 | ||
1672 | Double_t px = fP[3]; | |
1673 | Double_t py = fP[4]; | |
1674 | Double_t pz = fP[5]; | |
1675 | Double_t ss[2], g[2][5]; | |
1676 | ||
1677 | ss[0] = dS; | |
1678 | ss[1] = -dS; | |
1679 | for( Int_t i=0; i<2; i++){ | |
1680 | Double_t bs = bq*ss[i]; | |
1681 | Double_t c = TMath::Cos(bs), s = TMath::Sin(bs); | |
1682 | Double_t cB,sB; | |
1683 | if( TMath::Abs(bq)>1.e-8){ | |
1684 | cB= (1-c)/bq; | |
1685 | sB= s/bq; | |
1686 | }else{ | |
bfd20868 | 1687 | const Double_t kOvSqr6 = 1./TMath::Sqrt(6.); |
1688 | sB = (1.-bs*kOvSqr6)*(1.+bs*kOvSqr6)*ss[i]; | |
706952f5 | 1689 | cB = .5*sB*bs; |
1690 | } | |
1691 | g[i][0] = fP[0] + sB*px + cB*py; | |
1692 | g[i][1] = fP[1] - cB*px + sB*py; | |
1693 | g[i][2] = fP[2] + ss[i]*pz; | |
1694 | g[i][3] = + c*px + s*py; | |
1695 | g[i][4] = - s*px + c*py; | |
1696 | } | |
1697 | ||
1698 | Int_t i=0; | |
1699 | ||
1700 | Double_t dMin = 1.e10; | |
1701 | for( Int_t j=0; j<2; j++){ | |
1702 | Double_t xx = g[j][0]-xyz[0]; | |
1703 | Double_t yy = g[j][1]-xyz[1]; | |
1704 | Double_t zz = g[j][2]-xyz[2]; | |
1705 | Double_t d = xx*xx + yy*yy + zz*zz; | |
1706 | if( d<dMin ){ | |
1707 | dMin = d; | |
1708 | i = j; | |
1709 | } | |
1710 | } | |
1711 | ||
1712 | dS = ss[i]; | |
1713 | ||
1714 | Double_t x= g[i][0], y= g[i][1], z= g[i][2], ppx= g[i][3], ppy= g[i][4]; | |
5fc72f28 | 1715 | Double_t ddx = x-xyz[0]; |
1716 | Double_t ddy = y-xyz[1]; | |
1717 | Double_t ddz = z-xyz[2]; | |
1718 | Double_t c = ddx*ppx + ddy*ppy + ddz*pz ; | |
706952f5 | 1719 | Double_t pp2 = ppx*ppx + ppy*ppy + pz*pz; |
1720 | if( TMath::Abs(pp2)>1.e-8 ){ | |
1721 | dS+=c/pp2; | |
1722 | } | |
1723 | } | |
1724 | return dS; | |
f826d409 | 1725 | } |
1726 | ||
1727 | ||
1728 | void AliKFParticleBase::GetDStoParticleBz( Double_t B, const AliKFParticleBase &p, | |
616ffc76 | 1729 | Double_t &DS, Double_t &DS1 ) |
f826d409 | 1730 | const |
1731 | { | |
1732 | //* Get dS to another particle for Bz field | |
f826d409 | 1733 | Double_t px = fP[3]; |
1734 | Double_t py = fP[4]; | |
1735 | Double_t pz = fP[5]; | |
1736 | ||
1737 | Double_t px1 = p.fP[3]; | |
1738 | Double_t py1 = p.fP[4]; | |
1739 | Double_t pz1 = p.fP[5]; | |
1740 | ||
1741 | const Double_t kCLight = 0.000299792458; | |
1742 | ||
1743 | Double_t bq = B*fQ*kCLight; | |
1744 | Double_t bq1 = B*p.fQ*kCLight; | |
f826d409 | 1745 | Double_t s=0, ds=0, s1=0, ds1=0; |
1746 | ||
1747 | if( TMath::Abs(bq)>1.e-8 || TMath::Abs(bq1)>1.e-8 ){ | |
4bbc290d | 1748 | |
f826d409 | 1749 | Double_t dx = (p.fP[0] - fP[0]); |
1750 | Double_t dy = (p.fP[1] - fP[1]); | |
1751 | Double_t d2 = (dx*dx+dy*dy); | |
1752 | ||
1753 | Double_t p2 = (px *px + py *py); | |
1754 | Double_t p21 = (px1*px1 + py1*py1); | |
1d96fee9 | 1755 | |
1756 | if( TMath::Abs(p2) < 1.e-8 || TMath::Abs(p21) < 1.e-8 ) | |
1757 | { | |
1758 | DS=0.; | |
1759 | DS1=0.; | |
1760 | return; | |
1761 | } | |
1762 | ||
f826d409 | 1763 | Double_t a = (px*py1 - py*px1); |
1764 | Double_t b = (px*px1 + py*py1); | |
1765 | ||
1766 | Double_t ldx = bq*bq1*dx - bq1*py + bq*py1 ; | |
1767 | Double_t ldy = bq*bq1*dy + bq1*px - bq*px1 ; | |
1768 | Double_t l2 = ldx*ldx + ldy*ldy; | |
1769 | ||
1770 | Double_t cS = bq1*p2 + bq*bq1*(dy* px - dx* py) - bq*b; | |
1771 | Double_t cS1= bq*p21 - bq*bq1*(dy*px1 - dx*py1) - bq1*b; | |
1772 | ||
1773 | Double_t ca = bq*bq*bq1*d2 +2*( cS + bq*bq*(py1*dx-px1*dy)) ; | |
1774 | Double_t ca1 = bq*bq1*bq1*d2 +2*( cS1 - bq1*bq1*(py*dx-px*dy)) ; | |
1775 | ||
1776 | Double_t sa = 4*l2*p2 - ca*ca; | |
1777 | Double_t sa1 = 4*l2*p21 - ca1*ca1; | |
e7b09c95 | 1778 | |
f826d409 | 1779 | if(sa<0) sa=0; |
1780 | if(sa1<0)sa1=0; | |
1781 | ||
1782 | if( TMath::Abs(bq)>1.e-8){ | |
1783 | s = TMath::ATan2( bq*( bq1*(dx*px +dy*py) + a ) , cS )/bq; | |
1784 | ds = TMath::ATan2(TMath::Sqrt(sa),ca)/bq; | |
1785 | } else { | |
1786 | s = ( (dx*px + dy*py) + (py*px1-px*py1)/bq1)/p2; | |
1787 | ds = s*s - (d2-2*(px1*dy-py1*dx)/bq1)/p2; | |
1788 | if( ds<0 ) ds = 0; | |
1789 | ds = TMath::Sqrt(ds); | |
1790 | } | |
1791 | ||
1792 | if( TMath::Abs(bq1)>1.e-8){ | |
1793 | s1 = TMath::ATan2( -bq1*( bq*(dx*px1+dy*py1) + a), cS1 )/bq1; | |
1794 | ds1 = TMath::ATan2(TMath::Sqrt(sa1),ca1)/bq1; | |
1795 | } else { | |
1796 | s1 = (-(dx*px1 + dy*py1) + (py*px1-px*py1)/bq)/p21; | |
1797 | ds1 = s1*s1 - (d2+2*(px*dy-py*dx)/bq)/p21; | |
1798 | if( ds1<0 ) ds1 = 0; | |
1799 | ds1 = TMath::Sqrt(ds1); | |
1800 | } | |
1801 | } | |
1802 | ||
f826d409 | 1803 | Double_t ss[2], ss1[2], g[2][5],g1[2][5]; |
1804 | ||
1805 | ss[0] = s + ds; | |
1806 | ss[1] = s - ds; | |
1807 | ss1[0] = s1 + ds1; | |
1808 | ss1[1] = s1 - ds1; | |
1809 | for( Int_t i=0; i<2; i++){ | |
1810 | Double_t bs = bq*ss[i]; | |
5fc72f28 | 1811 | Double_t c = TMath::Cos(bs), sss = TMath::Sin(bs); |
f826d409 | 1812 | Double_t cB,sB; |
1813 | if( TMath::Abs(bq)>1.e-8){ | |
1814 | cB= (1-c)/bq; | |
5fc72f28 | 1815 | sB= sss/bq; |
f826d409 | 1816 | }else{ |
bfd20868 | 1817 | const Double_t kOvSqr6 = 1./TMath::Sqrt(6.); |
1818 | sB = (1.-bs*kOvSqr6)*(1.+bs*kOvSqr6)*ss[i]; | |
f826d409 | 1819 | cB = .5*sB*bs; |
1820 | } | |
1821 | g[i][0] = fP[0] + sB*px + cB*py; | |
1822 | g[i][1] = fP[1] - cB*px + sB*py; | |
1823 | g[i][2] = fP[2] + ss[i]*pz; | |
5fc72f28 | 1824 | g[i][3] = + c*px + sss*py; |
1825 | g[i][4] = - sss*px + c*py; | |
f826d409 | 1826 | |
1827 | bs = bq1*ss1[i]; | |
5fc72f28 | 1828 | c = TMath::Cos(bs); sss = TMath::Sin(bs); |
f826d409 | 1829 | if( TMath::Abs(bq1)>1.e-8){ |
1830 | cB= (1-c)/bq1; | |
5fc72f28 | 1831 | sB= sss/bq1; |
f826d409 | 1832 | }else{ |
bfd20868 | 1833 | const Double_t kOvSqr6 = 1./TMath::Sqrt(6.); |
1834 | sB = (1.-bs*kOvSqr6)*(1.+bs*kOvSqr6)*ss1[i]; | |
f826d409 | 1835 | cB = .5*sB*bs; |
1836 | } | |
1837 | ||
e7b09c95 | 1838 | g1[i][0] = p.fP[0] + sB*px1 + cB*py1; |
1839 | g1[i][1] = p.fP[1] - cB*px1 + sB*py1; | |
1840 | g1[i][2] = p.fP[2] + ss[i]*pz1; | |
5fc72f28 | 1841 | g1[i][3] = + c*px1 + sss*py1; |
1842 | g1[i][4] = - sss*px1 + c*py1; | |
f826d409 | 1843 | } |
1844 | ||
1845 | Int_t i=0, i1=0; | |
1846 | ||
1847 | Double_t dMin = 1.e10; | |
1848 | for( Int_t j=0; j<2; j++){ | |
1849 | for( Int_t j1=0; j1<2; j1++){ | |
1850 | Double_t xx = g[j][0]-g1[j1][0]; | |
1851 | Double_t yy = g[j][1]-g1[j1][1]; | |
1852 | Double_t zz = g[j][2]-g1[j1][2]; | |
1853 | Double_t d = xx*xx + yy*yy + zz*zz; | |
1854 | if( d<dMin ){ | |
1855 | dMin = d; | |
1856 | i = j; | |
1857 | i1 = j1; | |
1858 | } | |
1859 | } | |
1860 | } | |
1861 | ||
1862 | DS = ss[i]; | |
1863 | DS1 = ss1[i1]; | |
91468b27 | 1864 | if(0){ |
1865 | Double_t x= g[i][0], y= g[i][1], z= g[i][2], ppx= g[i][3], ppy= g[i][4]; | |
1866 | Double_t x1=g1[i1][0], y1= g1[i1][1], z1= g1[i1][2], ppx1= g1[i1][3], ppy1= g1[i1][4]; | |
1867 | Double_t dx = x1-x; | |
1868 | Double_t dy = y1-y; | |
1869 | Double_t dz = z1-z; | |
1870 | Double_t a = ppx*ppx1 + ppy*ppy1 + pz*pz1; | |
1871 | Double_t b = dx*ppx1 + dy*ppy1 + dz*pz1; | |
1872 | Double_t c = dx*ppx + dy*ppy + dz*pz ; | |
1873 | Double_t pp2 = ppx*ppx + ppy*ppy + pz*pz; | |
1874 | Double_t pp21= ppx1*ppx1 + ppy1*ppy1 + pz1*pz1; | |
706952f5 | 1875 | Double_t det = pp2*pp21 - a*a; |
91468b27 | 1876 | if( TMath::Abs(det)>1.e-8 ){ |
1877 | DS+=(a*b-pp21*c)/det; | |
1878 | DS1+=(a*c-pp2*b)/det; | |
1879 | } | |
f826d409 | 1880 | } |
1881 | } | |
1882 | ||
1883 | ||
1884 | ||
1885 | void AliKFParticleBase::TransportCBM( Double_t dS, | |
1886 | Double_t P[], Double_t C[] ) const | |
1887 | { | |
1888 | //* Transport the particle on dS, output to P[],C[], for CBM field | |
1889 | ||
1890 | if( fQ==0 ){ | |
1891 | TransportLine( dS, P, C ); | |
1892 | return; | |
1893 | } | |
1894 | ||
1895 | const Double_t kCLight = 0.000299792458; | |
1896 | ||
1897 | Double_t c = fQ*kCLight; | |
1898 | ||
1899 | // construct coefficients | |
1900 | ||
1901 | Double_t | |
1902 | px = fP[3], | |
1903 | py = fP[4], | |
1904 | pz = fP[5]; | |
1905 | ||
1906 | Double_t sx=0, sy=0, sz=0, syy=0, syz=0, syyy=0, ssx=0, ssy=0, ssz=0, ssyy=0, ssyz=0, ssyyy=0; | |
1907 | ||
1908 | { // get field integrals | |
1909 | ||
1910 | Double_t fld[3][3]; | |
1911 | Double_t p0[3], p1[3], p2[3]; | |
1912 | ||
1913 | // line track approximation | |
1914 | ||
1915 | p0[0] = fP[0]; | |
1916 | p0[1] = fP[1]; | |
1917 | p0[2] = fP[2]; | |
1918 | ||
1919 | p2[0] = fP[0] + px*dS; | |
1920 | p2[1] = fP[1] + py*dS; | |
1921 | p2[2] = fP[2] + pz*dS; | |
1922 | ||
1923 | p1[0] = 0.5*(p0[0]+p2[0]); | |
1924 | p1[1] = 0.5*(p0[1]+p2[1]); | |
1925 | p1[2] = 0.5*(p0[2]+p2[2]); | |
1926 | ||
1927 | // first order track approximation | |
1928 | { | |
1929 | GetFieldValue( p0, fld[0] ); | |
1930 | GetFieldValue( p1, fld[1] ); | |
1931 | GetFieldValue( p2, fld[2] ); | |
1932 | ||
1933 | Double_t ssy1 = ( 7*fld[0][1] + 6*fld[1][1]-fld[2][1] )*c*dS*dS/96.; | |
1934 | Double_t ssy2 = ( fld[0][1] + 2*fld[1][1] )*c*dS*dS/6.; | |
1935 | ||
1936 | p1[0] -= ssy1*pz; | |
1937 | p1[2] += ssy1*px; | |
1938 | p2[0] -= ssy2*pz; | |
1939 | p2[2] += ssy2*px; | |
1940 | } | |
1941 | ||
1942 | GetFieldValue( p0, fld[0] ); | |
1943 | GetFieldValue( p1, fld[1] ); | |
1944 | GetFieldValue( p2, fld[2] ); | |
1945 | ||
1946 | sx = c*( fld[0][0] + 4*fld[1][0] + fld[2][0] )*dS/6.; | |
1947 | sy = c*( fld[0][1] + 4*fld[1][1] + fld[2][1] )*dS/6.; | |
1948 | sz = c*( fld[0][2] + 4*fld[1][2] + fld[2][2] )*dS/6.; | |
1949 | ||
1950 | ssx = c*( fld[0][0] + 2*fld[1][0])*dS*dS/6.; | |
1951 | ssy = c*( fld[0][1] + 2*fld[1][1])*dS*dS/6.; | |
1952 | ssz = c*( fld[0][2] + 2*fld[1][2])*dS*dS/6.; | |
1953 | ||
1954 | Double_t c2[3][3] = { { 5, -4, -1},{ 44, 80, -4},{ 11, 44, 5} }; // /=360. | |
1955 | Double_t cc2[3][3] = { { 38, 8, -4},{ 148, 208, -20},{ 3, 36, 3} }; // /=2520. | |
1956 | for(Int_t n=0; n<3; n++) | |
1957 | for(Int_t m=0; m<3; m++) | |
1958 | { | |
1959 | syz += c2[n][m]*fld[n][1]*fld[m][2]; | |
1960 | ssyz += cc2[n][m]*fld[n][1]*fld[m][2]; | |
1961 | } | |
1962 | ||
1963 | syz *= c*c*dS*dS/360.; | |
1964 | ssyz *= c*c*dS*dS*dS/2520.; | |
1965 | ||
1966 | syy = c*( fld[0][1] + 4*fld[1][1] + fld[2][1] )*dS; | |
1967 | syyy = syy*syy*syy / 1296; | |
1968 | syy = syy*syy/72; | |
1969 | ||
1970 | ssyy = ( fld[0][1]*( 38*fld[0][1] + 156*fld[1][1] - fld[2][1] )+ | |
1971 | fld[1][1]*( 208*fld[1][1] +16*fld[2][1] )+ | |
1972 | fld[2][1]*( 3*fld[2][1] ) | |
1973 | )*dS*dS*dS*c*c/2520.; | |
1974 | ssyyy = | |
1975 | ( | |
1976 | fld[0][1]*( fld[0][1]*( 85*fld[0][1] + 526*fld[1][1] - 7*fld[2][1] )+ | |
1977 | fld[1][1]*( 1376*fld[1][1] +84*fld[2][1] )+ | |
1978 | fld[2][1]*( 19*fld[2][1] ) )+ | |
1979 | fld[1][1]*( fld[1][1]*( 1376*fld[1][1] +256*fld[2][1] )+ | |
1980 | fld[2][1]*( 62*fld[2][1] ) )+ | |
1981 | fld[2][1]*fld[2][1] *( 3*fld[2][1] ) | |
1982 | )*dS*dS*dS*dS*c*c*c/90720.; | |
1983 | ||
1984 | } | |
1985 | ||
1986 | Double_t mJ[8][8]; | |
1987 | for( Int_t i=0; i<8; i++ ) for( Int_t j=0; j<8; j++) mJ[i][j]=0; | |
1988 | ||
1989 | mJ[0][0]=1; mJ[0][1]=0; mJ[0][2]=0; mJ[0][3]=dS-ssyy; mJ[0][4]=ssx; mJ[0][5]=ssyyy-ssy; | |
1990 | mJ[1][0]=0; mJ[1][1]=1; mJ[1][2]=0; mJ[1][3]=-ssz; mJ[1][4]=dS; mJ[1][5]=ssx+ssyz; | |
1991 | mJ[2][0]=0; mJ[2][1]=0; mJ[2][2]=1; mJ[2][3]=ssy-ssyyy; mJ[2][4]=-ssx; mJ[2][5]=dS-ssyy; | |
1992 | ||
1993 | mJ[3][0]=0; mJ[3][1]=0; mJ[3][2]=0; mJ[3][3]=1-syy; mJ[3][4]=sx; mJ[3][5]=syyy-sy; | |
1994 | mJ[4][0]=0; mJ[4][1]=0; mJ[4][2]=0; mJ[4][3]=-sz; mJ[4][4]=1; mJ[4][5]=sx+syz; | |
1995 | mJ[5][0]=0; mJ[5][1]=0; mJ[5][2]=0; mJ[5][3]=sy-syyy; mJ[5][4]=-sx; mJ[5][5]=1-syy; | |
1996 | mJ[6][6] = mJ[7][7] = 1; | |
1997 | ||
1998 | P[0] = fP[0] + mJ[0][3]*px + mJ[0][4]*py + mJ[0][5]*pz; | |
1999 | P[1] = fP[1] + mJ[1][3]*px + mJ[1][4]*py + mJ[1][5]*pz; | |
2000 | P[2] = fP[2] + mJ[2][3]*px + mJ[2][4]*py + mJ[2][5]*pz; | |
2001 | P[3] = mJ[3][3]*px + mJ[3][4]*py + mJ[3][5]*pz; | |
2002 | P[4] = mJ[4][3]*px + mJ[4][4]*py + mJ[4][5]*pz; | |
2003 | P[5] = mJ[5][3]*px + mJ[5][4]*py + mJ[5][5]*pz; | |
2004 | P[6] = fP[6]; | |
2005 | P[7] = fP[7]; | |
2006 | ||
2007 | MultQSQt( mJ[0], fC, C); | |
2008 | ||
2009 | } | |
2010 | ||
2011 | ||
0d14829a | 2012 | void AliKFParticleBase::TransportBz( Double_t b, Double_t t, |
2013 | Double_t p[], Double_t e[] ) const | |
f826d409 | 2014 | { |
2015 | //* Transport the particle on dS, output to P[],C[], for Bz field | |
616ffc76 | 2016 | |
f826d409 | 2017 | const Double_t kCLight = 0.000299792458; |
5fc72f28 | 2018 | b = b*fQ*kCLight; |
0d14829a | 2019 | Double_t bs= b*t; |
f826d409 | 2020 | Double_t s = TMath::Sin(bs), c = TMath::Cos(bs); |
2021 | Double_t sB, cB; | |
616ffc76 | 2022 | if( TMath::Abs(bs)>1.e-10){ |
5fc72f28 | 2023 | sB= s/b; |
2024 | cB= (1-c)/b; | |
f826d409 | 2025 | }else{ |
bfd20868 | 2026 | const Double_t kOvSqr6 = 1./TMath::Sqrt(6.); |
0d14829a | 2027 | sB = (1.-bs*kOvSqr6)*(1.+bs*kOvSqr6)*t; |
f826d409 | 2028 | cB = .5*sB*bs; |
2029 | } | |
2030 | ||
2031 | Double_t px = fP[3]; | |
2032 | Double_t py = fP[4]; | |
2033 | Double_t pz = fP[5]; | |
2034 | ||
5fc72f28 | 2035 | p[0] = fP[0] + sB*px + cB*py; |
2036 | p[1] = fP[1] - cB*px + sB*py; | |
0d14829a | 2037 | p[2] = fP[2] + t*pz; |
5fc72f28 | 2038 | p[3] = c*px + s*py; |
2039 | p[4] = -s*px + c*py; | |
2040 | p[5] = fP[5]; | |
2041 | p[6] = fP[6]; | |
2042 | p[7] = fP[7]; | |
0d14829a | 2043 | |
2044 | /* | |
f826d409 | 2045 | Double_t mJ[8][8] = { {1,0,0, sB, cB, 0, 0, 0 }, |
616ffc76 | 2046 | {0,1,0, -cB, sB, 0, 0, 0 }, |
0d14829a | 2047 | {0,0,1, 0, 0, t, 0, 0 }, |
616ffc76 | 2048 | {0,0,0, c, s, 0, 0, 0 }, |
2049 | {0,0,0, -s, c, 0, 0, 0 }, | |
2050 | {0,0,0, 0, 0, 1, 0, 0 }, | |
2051 | {0,0,0, 0, 0, 0, 1, 0 }, | |
2052 | {0,0,0, 0, 0, 0, 0, 1 } }; | |
f826d409 | 2053 | Double_t mA[8][8]; |
2054 | for( Int_t k=0,i=0; i<8; i++) | |
2055 | for( Int_t j=0; j<=i; j++, k++ ) mA[i][j] = mA[j][i] = fC[k]; | |
2056 | ||
2057 | Double_t mJC[8][8]; | |
2058 | for( Int_t i=0; i<8; i++ ) | |
2059 | for( Int_t j=0; j<8; j++ ){ | |
2060 | mJC[i][j]=0; | |
2061 | for( Int_t k=0; k<8; k++ ) mJC[i][j]+=mJ[i][k]*mA[k][j]; | |
2062 | } | |
2063 | ||
2064 | for( Int_t k=0,i=0; i<8; i++) | |
2065 | for( Int_t j=0; j<=i; j++, k++ ){ | |
0d14829a | 2066 | e[k] = 0; |
2067 | for( Int_t l=0; l<8; l++ ) e[k]+=mJC[i][l]*mJ[j][l]; | |
f826d409 | 2068 | } |
2069 | ||
2070 | return; | |
f826d409 | 2071 | */ |
0d14829a | 2072 | |
2073 | Double_t | |
2074 | c6=fC[6], c7=fC[7], c8=fC[8], c17=fC[17], c18=fC[18], | |
2075 | c24 = fC[24], c31 = fC[31]; | |
2076 | ||
2077 | Double_t | |
2078 | cBC13 = cB*fC[13], | |
2079 | mJC13 = c7 - cB*fC[9] + sB*fC[13], | |
2080 | mJC14 = fC[11] - cBC13 + sB*fC[14], | |
2081 | mJC23 = c8 + t*c18, | |
2082 | mJC24 = fC[12] + t*fC[19], | |
2083 | mJC33 = c*fC[9] + s*fC[13], | |
2084 | mJC34 = c*fC[13] + s*fC[14], | |
2085 | mJC43 = -s*fC[9] + c*fC[13], | |
2086 | mJC44 = -s*fC[13] + c*fC[14]; | |
2087 | ||
2088 | ||
2089 | e[0]= fC[0] + 2*(sB*c6 + cB*fC[10]) + (sB*fC[9] + 2*cBC13)*sB + cB*cB*fC[14]; | |
2090 | e[1]= fC[1] - cB*c6 + sB*fC[10] + mJC13*sB + mJC14*cB; | |
2091 | e[2]= fC[2] - cB*c7 + sB*fC[11] - mJC13*cB + mJC14*sB; | |
2092 | e[3]= fC[3] + t*fC[15] + mJC23*sB + mJC24*cB; | |
2093 | e[4]= fC[4] + t*fC[16] - mJC23*cB + mJC24*sB; | |
2094 | ||
2095 | e[15]= fC[15] + c18*sB + fC[19]*cB; | |
2096 | e[16]= fC[16] - c18*cB + fC[19]*sB; | |
2097 | e[17]= c17 + fC[20]*t; | |
2098 | e[18]= c18*c + fC[19]*s; | |
2099 | e[19]= -c18*s + fC[19]*c; | |
2100 | ||
2101 | e[5]= fC[5] + (c17 + e[17] )*t; | |
2102 | ||
2103 | e[6]= c*c6 + s*fC[10] + mJC33*sB + mJC34*cB; | |
2104 | e[7]= c*c7 + s*fC[11] - mJC33*cB + mJC34*sB; | |
2105 | e[8]= c*c8 + s*fC[12] + e[18]*t; | |
2106 | e[9]= mJC33*c + mJC34*s; | |
2107 | e[10]= -s*c6 + c*fC[10] + mJC43*sB + mJC44*cB; | |
2108 | ||
2109 | ||
2110 | e[11]= -s*c7 + c*fC[11] - mJC43*cB + mJC44*sB; | |
2111 | e[12]= -s*c8 + c*fC[12] + e[19]*t; | |
2112 | e[13]= mJC43*c + mJC44*s; | |
2113 | e[14]= -mJC43*s + mJC44*c; | |
2114 | e[20]= fC[20]; | |
2115 | e[21]= fC[21] + fC[25]*cB + c24*sB; | |
2116 | e[22]= fC[22] - c24*cB + fC[25]*sB; | |
2117 | e[23]= fC[23] + fC[26]*t; | |
2118 | e[24]= c*c24 + s*fC[25]; | |
2119 | e[25]= c*fC[25] - c24*s; | |
2120 | e[26]= fC[26]; | |
2121 | e[27]= fC[27]; | |
2122 | e[28]= fC[28] + fC[32]*cB + c31*sB; | |
2123 | e[29]= fC[29] - c31*cB + fC[32]*sB; | |
2124 | e[30]= fC[30] + fC[33]*t; | |
2125 | e[31]= c*c31 + s*fC[32]; | |
2126 | e[32]= c*fC[32] - s*c31; | |
2127 | e[33]= fC[33]; | |
2128 | e[34]= fC[34]; | |
2129 | e[35]= fC[35]; | |
f826d409 | 2130 | } |
2131 | ||
2132 | ||
2133 | Double_t AliKFParticleBase::GetDistanceFromVertex( const AliKFParticleBase &Vtx ) const | |
2134 | { | |
2135 | //* Calculate distance from vertex [cm] | |
2136 | ||
2137 | return GetDistanceFromVertex( Vtx.fP ); | |
2138 | } | |
2139 | ||
2140 | Double_t AliKFParticleBase::GetDistanceFromVertex( const Double_t vtx[] ) const | |
2141 | { | |
2142 | //* Calculate distance from vertex [cm] | |
2143 | ||
2144 | Double_t mP[8], mC[36]; | |
2145 | Transport( GetDStoPoint(vtx), mP, mC ); | |
2146 | Double_t d[3]={ vtx[0]-mP[0], vtx[1]-mP[1], vtx[2]-mP[2]}; | |
2147 | return TMath::Sqrt( d[0]*d[0]+d[1]*d[1]+d[2]*d[2] ); | |
2148 | } | |
2149 | ||
616ffc76 | 2150 | Double_t AliKFParticleBase::GetDistanceFromParticle( const AliKFParticleBase &p ) |
f826d409 | 2151 | const |
2152 | { | |
616ffc76 | 2153 | //* Calculate distance to other particle [cm] |
f826d409 | 2154 | |
2155 | Double_t dS, dS1; | |
616ffc76 | 2156 | GetDStoParticle( p, dS, dS1 ); |
f826d409 | 2157 | Double_t mP[8], mC[36], mP1[8], mC1[36]; |
616ffc76 | 2158 | Transport( dS, mP, mC ); |
2159 | p.Transport( dS1, mP1, mC1 ); | |
f826d409 | 2160 | Double_t dx = mP[0]-mP1[0]; |
2161 | Double_t dy = mP[1]-mP1[1]; | |
2162 | Double_t dz = mP[2]-mP1[2]; | |
55ac3e1e | 2163 | dz = 0; |
f826d409 | 2164 | return TMath::Sqrt(dx*dx+dy*dy+dz*dz); |
2165 | } | |
2166 | ||
2167 | Double_t AliKFParticleBase::GetDeviationFromVertex( const AliKFParticleBase &Vtx ) const | |
2168 | { | |
2169 | //* Calculate sqrt(Chi2/ndf) deviation from vertex | |
2170 | ||
2171 | return GetDeviationFromVertex( Vtx.fP, Vtx.fC ); | |
2172 | } | |
2173 | ||
2174 | ||
2175 | Double_t AliKFParticleBase::GetDeviationFromVertex( const Double_t v[], const Double_t Cv[] ) const | |
2176 | { | |
2177 | //* Calculate sqrt(Chi2/ndf) deviation from vertex | |
2178 | //* v = [xyz], Cv=[Cxx,Cxy,Cyy,Cxz,Cyz,Czz]-covariance matrix | |
2179 | ||
2180 | Double_t mP[8]; | |
2181 | Double_t mC[36]; | |
2182 | ||
2183 | Transport( GetDStoPoint(v), mP, mC ); | |
2184 | ||
2185 | Double_t d[3]={ v[0]-mP[0], v[1]-mP[1], v[2]-mP[2]}; | |
2186 | ||
2187 | Double_t sigmaS = .1+10.*TMath::Sqrt( (d[0]*d[0]+d[1]*d[1]+d[2]*d[2])/ | |
2188 | (mP[3]*mP[3]+mP[4]*mP[4]+mP[5]*mP[5]) ); | |
2189 | ||
2190 | ||
2191 | Double_t h[3] = { mP[3]*sigmaS, mP[4]*sigmaS, mP[5]*sigmaS }; | |
2192 | ||
2193 | Double_t mSi[6] = | |
2194 | { mC[0] +h[0]*h[0], | |
2195 | mC[1] +h[1]*h[0], mC[2] +h[1]*h[1], | |
2196 | mC[3] +h[2]*h[0], mC[4] +h[2]*h[1], mC[5] +h[2]*h[2] }; | |
2197 | ||
2198 | if( Cv ){ | |
2199 | mSi[0]+=Cv[0]; | |
2200 | mSi[1]+=Cv[1]; | |
2201 | mSi[2]+=Cv[2]; | |
2202 | mSi[3]+=Cv[3]; | |
2203 | mSi[4]+=Cv[4]; | |
2204 | mSi[5]+=Cv[5]; | |
2205 | } | |
2206 | ||
2207 | Double_t mS[6]; | |
2208 | ||
2209 | mS[0] = mSi[2]*mSi[5] - mSi[4]*mSi[4]; | |
2210 | mS[1] = mSi[3]*mSi[4] - mSi[1]*mSi[5]; | |
2211 | mS[2] = mSi[0]*mSi[5] - mSi[3]*mSi[3]; | |
2212 | mS[3] = mSi[1]*mSi[4] - mSi[2]*mSi[3]; | |
2213 | mS[4] = mSi[1]*mSi[3] - mSi[0]*mSi[4]; | |
2214 | mS[5] = mSi[0]*mSi[2] - mSi[1]*mSi[1]; | |
2215 | ||
2216 | Double_t s = ( mSi[0]*mS[0] + mSi[1]*mS[1] + mSi[3]*mS[3] ); | |
2217 | s = ( s > 1.E-20 ) ?1./s :0; | |
2218 | ||
2219 | return TMath::Sqrt( TMath::Abs(s*( ( mS[0]*d[0] + mS[1]*d[1] + mS[3]*d[2])*d[0] | |
2220 | +(mS[1]*d[0] + mS[2]*d[1] + mS[4]*d[2])*d[1] | |
2221 | +(mS[3]*d[0] + mS[4]*d[1] + mS[5]*d[2])*d[2] ))/2); | |
2222 | } | |
2223 | ||
2224 | ||
616ffc76 | 2225 | Double_t AliKFParticleBase::GetDeviationFromParticle( const AliKFParticleBase &p ) |
f826d409 | 2226 | const |
2227 | { | |
616ffc76 | 2228 | //* Calculate sqrt(Chi2/ndf) deviation from other particle |
f826d409 | 2229 | |
2230 | Double_t dS, dS1; | |
616ffc76 | 2231 | GetDStoParticle( p, dS, dS1 ); |
f826d409 | 2232 | Double_t mP1[8], mC1[36]; |
616ffc76 | 2233 | p.Transport( dS1, mP1, mC1 ); |
f826d409 | 2234 | |
2235 | Double_t d[3]={ fP[0]-mP1[0], fP[1]-mP1[1], fP[2]-mP1[2]}; | |
2236 | ||
2237 | Double_t sigmaS = .1+10.*TMath::Sqrt( (d[0]*d[0]+d[1]*d[1]+d[2]*d[2])/ | |
616ffc76 | 2238 | (mP1[3]*mP1[3]+mP1[4]*mP1[4]+mP1[5]*mP1[5]) ); |
f826d409 | 2239 | |
2240 | Double_t h[3] = { mP1[3]*sigmaS, mP1[4]*sigmaS, mP1[5]*sigmaS }; | |
2241 | ||
2242 | mC1[0] +=h[0]*h[0]; | |
2243 | mC1[1] +=h[1]*h[0]; | |
2244 | mC1[2] +=h[1]*h[1]; | |
2245 | mC1[3] +=h[2]*h[0]; | |
2246 | mC1[4] +=h[2]*h[1]; | |
2247 | mC1[5] +=h[2]*h[2]; | |
2248 | ||
2249 | return GetDeviationFromVertex( mP1, mC1 )*TMath::Sqrt(2./1.); | |
2250 | } | |
2251 | ||
2252 | ||
2253 | ||
de0d0ceb | 2254 | void AliKFParticleBase::SubtractFromVertex( AliKFParticleBase &Vtx ) const |
24be0330 | 2255 | { |
f826d409 | 2256 | //* Subtract the particle from the vertex |
de0d0ceb | 2257 | |
f826d409 | 2258 | Double_t fld[3]; |
2259 | { | |
de0d0ceb | 2260 | GetFieldValue( Vtx.fP, fld ); |
f826d409 | 2261 | const Double_t kCLight = 0.000299792458; |
2262 | fld[0]*=kCLight; fld[1]*=kCLight; fld[2]*=kCLight; | |
2263 | } | |
2264 | ||
2265 | Double_t m[8]; | |
2266 | Double_t mCm[36]; | |
f826d409 | 2267 | |
de0d0ceb | 2268 | if( Vtx.fIsLinearized ){ |
2269 | GetMeasurement( Vtx.fVtxGuess, m, mCm ); | |
2270 | } else { | |
2271 | GetMeasurement( Vtx.fP, m, mCm ); | |
f826d409 | 2272 | } |
de0d0ceb | 2273 | |
f826d409 | 2274 | Double_t mV[6]; |
2275 | ||
de0d0ceb | 2276 | mV[ 0] = mCm[ 0]; |
2277 | mV[ 1] = mCm[ 1]; | |
2278 | mV[ 2] = mCm[ 2]; | |
2279 | mV[ 3] = mCm[ 3]; | |
2280 | mV[ 4] = mCm[ 4]; | |
2281 | mV[ 5] = mCm[ 5]; | |
f826d409 | 2282 | |
2283 | //* | |
2284 | ||
2285 | Double_t mS[6]; | |
2286 | { | |
de0d0ceb | 2287 | Double_t mSi[6] = { mV[0]-Vtx.fC[0], |
2288 | mV[1]-Vtx.fC[1], mV[2]-Vtx.fC[2], | |
2289 | mV[3]-Vtx.fC[3], mV[4]-Vtx.fC[4], mV[5]-Vtx.fC[5] }; | |
f826d409 | 2290 | |
2291 | mS[0] = mSi[2]*mSi[5] - mSi[4]*mSi[4]; | |
2292 | mS[1] = mSi[3]*mSi[4] - mSi[1]*mSi[5]; | |
2293 | mS[2] = mSi[0]*mSi[5] - mSi[3]*mSi[3]; | |
2294 | mS[3] = mSi[1]*mSi[4] - mSi[2]*mSi[3]; | |
2295 | mS[4] = mSi[1]*mSi[3] - mSi[0]*mSi[4]; | |
2296 | mS[5] = mSi[0]*mSi[2] - mSi[1]*mSi[1]; | |
2297 | ||
2298 | Double_t s = ( mSi[0]*mS[0] + mSi[1]*mS[1] + mSi[3]*mS[3] ); | |
2299 | s = ( s > 1.E-20 ) ?1./s :0; | |
2300 | mS[0]*=s; | |
2301 | mS[1]*=s; | |
2302 | mS[2]*=s; | |
2303 | mS[3]*=s; | |
2304 | mS[4]*=s; | |
2305 | mS[5]*=s; | |
2306 | } | |
2307 | ||
2308 | //* Residual (measured - estimated) | |
2309 | ||
de0d0ceb | 2310 | Double_t zeta[3] = { m[0]-Vtx.fP[0], m[1]-Vtx.fP[1], m[2]-Vtx.fP[2] }; |
f826d409 | 2311 | |
2312 | //* mCHt = mCH' - D' | |
2313 | ||
2314 | Double_t mCHt0[3], mCHt1[3], mCHt2[3]; | |
2315 | ||
de0d0ceb | 2316 | mCHt0[0]=Vtx.fC[ 0] ; mCHt1[0]=Vtx.fC[ 1] ; mCHt2[0]=Vtx.fC[ 3] ; |
2317 | mCHt0[1]=Vtx.fC[ 1] ; mCHt1[1]=Vtx.fC[ 2] ; mCHt2[1]=Vtx.fC[ 4] ; | |
2318 | mCHt0[2]=Vtx.fC[ 3] ; mCHt1[2]=Vtx.fC[ 4] ; mCHt2[2]=Vtx.fC[ 5] ; | |
f826d409 | 2319 | |
2320 | //* Kalman gain K = mCH'*S | |
2321 | ||
2322 | Double_t k0[3], k1[3], k2[3]; | |
2323 | ||
2324 | for(Int_t i=0;i<3;++i){ | |
2325 | k0[i] = mCHt0[i]*mS[0] + mCHt1[i]*mS[1] + mCHt2[i]*mS[3]; | |
2326 | k1[i] = mCHt0[i]*mS[1] + mCHt1[i]*mS[2] + mCHt2[i]*mS[4]; | |
2327 | k2[i] = mCHt0[i]*mS[3] + mCHt1[i]*mS[4] + mCHt2[i]*mS[5]; | |
2328 | } | |
2329 | ||
2330 | //* New estimation of the vertex position r += K*zeta | |
2331 | ||
de0d0ceb | 2332 | Double_t dChi2 = -(mS[0]*zeta[0] + mS[1]*zeta[1] + mS[3]*zeta[2])*zeta[0] |
2333 | + (mS[1]*zeta[0] + mS[2]*zeta[1] + mS[4]*zeta[2])*zeta[1] | |
2334 | + (mS[3]*zeta[0] + mS[4]*zeta[1] + mS[5]*zeta[2])*zeta[2]; | |
2335 | ||
2336 | if( Vtx.fChi2 - dChi2 < 0 ) return; | |
2337 | ||
f826d409 | 2338 | for(Int_t i=0;i<3;++i) |
de0d0ceb | 2339 | Vtx.fP[i] -= k0[i]*zeta[0] + k1[i]*zeta[1] + k2[i]*zeta[2]; |
f826d409 | 2340 | |
2341 | //* New covariance matrix C -= K*(mCH')' | |
2342 | ||
2343 | for(Int_t i=0, k=0;i<3;++i){ | |
2344 | for(Int_t j=0;j<=i;++j,++k) | |
de0d0ceb | 2345 | Vtx.fC[k] += k0[i]*mCHt0[j] + k1[i]*mCHt1[j] + k2[i]*mCHt2[j]; |
f826d409 | 2346 | } |
2347 | ||
2348 | //* Calculate Chi^2 | |
de0d0ceb | 2349 | |
2350 | Vtx.fNDF -= 2; | |
2351 | Vtx.fChi2 -= dChi2; | |
f826d409 | 2352 | } |
2353 | ||
2354 | ||
2355 | ||
2356 | void AliKFParticleBase::TransportLine( Double_t dS, | |
2357 | Double_t P[], Double_t C[] ) const | |
2358 | { | |
2359 | //* Transport the particle as a straight line | |
2360 | ||
2361 | P[0] = fP[0] + dS*fP[3]; | |
2362 | P[1] = fP[1] + dS*fP[4]; | |
2363 | P[2] = fP[2] + dS*fP[5]; | |
2364 | P[3] = fP[3]; | |
2365 | P[4] = fP[4]; | |
2366 | P[5] = fP[5]; | |
2367 | P[6] = fP[6]; | |
2368 | P[7] = fP[7]; | |
2369 | ||
2370 | Double_t c6 = fC[ 6] + dS*fC[ 9]; | |
2371 | Double_t c11 = fC[11] + dS*fC[14]; | |
2372 | Double_t c17 = fC[17] + dS*fC[20]; | |
2373 | Double_t sc13 = dS*fC[13]; | |
2374 | Double_t sc18 = dS*fC[18]; | |
2375 | Double_t sc19 = dS*fC[19]; | |
2376 | ||
2377 | C[ 0] = fC[ 0] + dS*( fC[ 6] + c6 ); | |
2378 | C[ 2] = fC[ 2] + dS*( fC[11] + c11 ); | |
2379 | C[ 5] = fC[ 5] + dS*( fC[17] + c17 ); | |
2380 | ||
2381 | C[ 7] = fC[ 7] + sc13; | |
2382 | C[ 8] = fC[ 8] + sc18; | |
2383 | C[ 9] = fC[ 9]; | |
2384 | ||
2385 | C[12] = fC[12] + sc19; | |
2386 | ||
2387 | C[ 1] = fC[ 1] + dS*( fC[10] + C[ 7] ); | |
2388 | C[ 3] = fC[ 3] + dS*( fC[15] + C[ 8] ); | |
2389 | C[ 4] = fC[ 4] + dS*( fC[16] + C[12] ); | |
2390 | C[ 6] = c6; | |
2391 | ||
2392 | C[10] = fC[10] + sc13; | |
2393 | C[11] = c11; | |
2394 | ||
2395 | C[13] = fC[13]; | |
2396 | C[14] = fC[14]; | |
2397 | C[15] = fC[15] + sc18; | |
2398 | C[16] = fC[16] + sc19; | |
2399 | C[17] = c17; | |
2400 | ||
2401 | C[18] = fC[18]; | |
2402 | C[19] = fC[19]; | |
2403 | C[20] = fC[20]; | |
2404 | C[21] = fC[21] + dS*fC[24]; | |
2405 | C[22] = fC[22] + dS*fC[25]; | |
2406 | C[23] = fC[23] + dS*fC[26]; | |
2407 | ||
2408 | C[24] = fC[24]; | |
2409 | C[25] = fC[25]; | |
2410 | C[26] = fC[26]; | |
2411 | C[27] = fC[27]; | |
2412 | C[28] = fC[28] + dS*fC[31]; | |
2413 | C[29] = fC[29] + dS*fC[32]; | |
2414 | C[30] = fC[30] + dS*fC[33]; | |
2415 | ||
2416 | C[31] = fC[31]; | |
2417 | C[32] = fC[32]; | |
2418 | C[33] = fC[33]; | |
2419 | C[34] = fC[34]; | |
2420 | C[35] = fC[35]; | |
2421 | } | |
2422 | ||
2423 | ||
a65041d0 | 2424 | void AliKFParticleBase::ConstructGammaBz( const AliKFParticleBase &daughter1, |
2425 | const AliKFParticleBase &daughter2, double Bz ) | |
2426 | { | |
2427 | //* Create gamma | |
2428 | ||
2429 | const AliKFParticleBase *daughters[2] = { &daughter1, &daughter2}; | |
2430 | ||
2431 | double v0[3]; | |
2432 | ||
2433 | if( !fIsLinearized ){ | |
2434 | Double_t ds, ds1; | |
2435 | Double_t m[8]; | |
2436 | Double_t mCd[36]; | |
2437 | daughter1.GetDStoParticle(daughter2, ds, ds1); | |
2438 | daughter1.Transport( ds, m, mCd ); | |
55ac3e1e | 2439 | fP[0] = m[0]; |
2440 | fP[1] = m[1]; | |
2441 | fP[2] = m[2]; | |
a65041d0 | 2442 | daughter2.Transport( ds1, m, mCd ); |
55ac3e1e | 2443 | fP[0] = .5*( fP[0] + m[0] ); |
2444 | fP[1] = .5*( fP[1] + m[1] ); | |
2445 | fP[2] = .5*( fP[2] + m[2] ); | |
a65041d0 | 2446 | } else { |
55ac3e1e | 2447 | fP[0] = fVtxGuess[0]; |
2448 | fP[1] = fVtxGuess[1]; | |
2449 | fP[2] = fVtxGuess[2]; | |
a65041d0 | 2450 | } |
2451 | ||
a65041d0 | 2452 | double daughterP[2][8], daughterC[2][36]; |
2453 | double vtxMom[2][3]; | |
2454 | ||
55ac3e1e | 2455 | int nIter = fIsLinearized ?1 :2; |
a65041d0 | 2456 | |
55ac3e1e | 2457 | for( int iter=0; iter<nIter; iter++){ |
a65041d0 | 2458 | |
55ac3e1e | 2459 | v0[0] = fP[0]; |
2460 | v0[1] = fP[1]; | |
2461 | v0[2] = fP[2]; | |
a65041d0 | 2462 | |
55ac3e1e | 2463 | fAtProductionVertex = 0; |
2464 | fSFromDecay = 0; | |
2465 | fP[0] = v0[0]; | |
2466 | fP[1] = v0[1]; | |
2467 | fP[2] = v0[2]; | |
2468 | fP[3] = 0; | |
2469 | fP[4] = 0; | |
2470 | fP[5] = 0; | |
2471 | fP[6] = 0; | |
2472 | fP[7] = 0; | |
2473 | ||
2474 | ||
2475 | // fit daughters to the vertex guess | |
a65041d0 | 2476 | |
55ac3e1e | 2477 | { |
2478 | for( int id=0; id<2; id++ ){ | |
2479 | ||
2480 | double *p = daughterP[id]; | |
2481 | double *mC = daughterC[id]; | |
2482 | ||
2483 | daughters[id]->GetMeasurement( v0, p, mC ); | |
2484 | ||
2485 | Double_t mAi[6]; | |
2486 | InvertSym3(mC, mAi ); | |
2487 | ||
2488 | Double_t mB[3][3]; | |
2489 | ||
2490 | mB[0][0] = mC[ 6]*mAi[0] + mC[ 7]*mAi[1] + mC[ 8]*mAi[3]; | |
2491 | mB[0][1] = mC[ 6]*mAi[1] + mC[ 7]*mAi[2] + mC[ 8]*mAi[4]; | |
2492 | mB[0][2] = mC[ 6]*mAi[3] + mC[ 7]*mAi[4] + mC[ 8]*mAi[5]; | |
2493 | ||
2494 | mB[1][0] = mC[10]*mAi[0] + mC[11]*mAi[1] + mC[12]*mAi[3]; | |
2495 | mB[1][1] = mC[10]*mAi[1] + mC[11]*mAi[2] + mC[12]*mAi[4]; | |
2496 | mB[1][2] = mC[10]*mAi[3] + mC[11]*mAi[4] + mC[12]*mAi[5]; | |
2497 | ||
2498 | mB[2][0] = mC[15]*mAi[0] + mC[16]*mAi[1] + mC[17]*mAi[3]; | |
2499 | mB[2][1] = mC[15]*mAi[1] + mC[16]*mAi[2] + mC[17]*mAi[4]; | |
2500 | mB[2][2] = mC[15]*mAi[3] + mC[16]*mAi[4] + mC[17]*mAi[5]; | |
2501 | ||
2502 | Double_t z[3] = { v0[0]-p[0], v0[1]-p[1], v0[2]-p[2] }; | |
2503 | ||
2504 | vtxMom[id][0] = p[3] + mB[0][0]*z[0] + mB[0][1]*z[1] + mB[0][2]*z[2]; | |
2505 | vtxMom[id][1] = p[4] + mB[1][0]*z[0] + mB[1][1]*z[1] + mB[1][2]*z[2]; | |
2506 | vtxMom[id][2] = p[5] + mB[2][0]*z[0] + mB[2][1]*z[1] + mB[2][2]*z[2]; | |
2507 | ||
2508 | daughters[id]->Transport( daughters[id]->GetDStoPoint(v0), p, mC ); | |
a65041d0 | 2509 | |
55ac3e1e | 2510 | } |
a65041d0 | 2511 | |
55ac3e1e | 2512 | } // fit daughters to guess |
2513 | ||
2514 | ||
2515 | // fit new vertex | |
2516 | { | |
a65041d0 | 2517 | |
55ac3e1e | 2518 | double mpx0 = vtxMom[0][0]+vtxMom[1][0]; |
2519 | double mpy0 = vtxMom[0][1]+vtxMom[1][1]; | |
2520 | double mpt0 = TMath::Sqrt(mpx0*mpx0 + mpy0*mpy0); | |
2521 | // double a0 = TMath::ATan2(mpy0,mpx0); | |
a65041d0 | 2522 | |
55ac3e1e | 2523 | double ca0 = mpx0/mpt0; |
2524 | double sa0 = mpy0/mpt0; | |
2525 | double r[3] = { v0[0], v0[1], v0[2] }; | |
c3af7f6d | 2526 | double mC[3][3] = {{1000., 0 , 0 }, |
2527 | {0, 1000., 0 }, | |
2528 | {0, 0, 1000. } }; | |
55ac3e1e | 2529 | double chi2=0; |
a65041d0 | 2530 | |
55ac3e1e | 2531 | for( int id=0; id<2; id++ ){ |
2532 | const Double_t kCLight = 0.000299792458; | |
2533 | Double_t q = Bz*daughters[id]->GetQ()*kCLight; | |
2534 | Double_t px0 = vtxMom[id][0]; | |
2535 | Double_t py0 = vtxMom[id][1]; | |
2536 | Double_t pz0 = vtxMom[id][2]; | |
2537 | Double_t pt0 = TMath::Sqrt(px0*px0+py0*py0); | |
2538 | Double_t mG[3][6], mB[3], mH[3][3]; | |
2539 | // r = {vx,vy,vz}; | |
2540 | // m = {x,y,z,Px,Py,Pz}; | |
2541 | // V = daughter.C | |
2542 | // G*m + B = H*r; | |
2543 | // q*x + Py - q*vx - sin(a)*Pt = 0 | |
2544 | // q*y - Px - q*vy + cos(a)*Pt = 0 | |
2545 | // (Px*cos(a) + Py*sin(a) ) (vz -z) - Pz( cos(a)*(vx-x) + sin(a)*(vy-y)) = 0 | |
2546 | ||
2547 | mG[0][0] = q; | |
2548 | mG[0][1] = 0; | |
2549 | mG[0][2] = 0; | |
2550 | mG[0][3] = -sa0*px0/pt0; | |
2551 | mG[0][4] = 1 -sa0*py0/pt0; | |
2552 | mG[0][5] = 0; | |
2553 | mH[0][0] = q; | |
2554 | mH[0][1] = 0; | |
2555 | mH[0][2] = 0; | |
2556 | mB[0] = py0 - sa0*pt0 - mG[0][3]*px0 - mG[0][4]*py0 ; | |
2557 | ||
2558 | // q*y - Px - q*vy + cos(a)*Pt = 0 | |
2559 | ||
2560 | mG[1][0] = 0; | |
2561 | mG[1][1] = q; | |
2562 | mG[1][2] = 0; | |
2563 | mG[1][3] = -1 + ca0*px0/pt0; | |
2564 | mG[1][4] = + ca0*py0/pt0; | |
2565 | mG[1][5] = 0; | |
2566 | mH[1][0] = 0; | |
2567 | mH[1][1] = q; | |
2568 | mH[1][2] = 0; | |
2569 | mB[1] = -px0 + ca0*pt0 - mG[1][3]*px0 - mG[1][4]*py0 ; | |
2570 | ||
2571 | // (Px*cos(a) + Py*sin(a) ) (z -vz) - Pz( cos(a)*(x-vx) + sin(a)*(y-vy)) = 0 | |
a65041d0 | 2572 | |
55ac3e1e | 2573 | mG[2][0] = -pz0*ca0; |
2574 | mG[2][1] = -pz0*sa0; | |
2575 | mG[2][2] = px0*ca0 + py0*sa0; | |
2576 | mG[2][3] = 0; | |
2577 | mG[2][4] = 0; | |
2578 | mG[2][5] = 0; | |
2579 | ||
2580 | mH[2][0] = mG[2][0]; | |
2581 | mH[2][1] = mG[2][1]; | |
2582 | mH[2][2] = mG[2][2]; | |
2583 | ||
2584 | mB[2] = 0; | |
2585 | ||
2586 | // fit the vertex | |
a65041d0 | 2587 | |
55ac3e1e | 2588 | // V = GVGt |
a65041d0 | 2589 | |
55ac3e1e | 2590 | double mGV[3][6]; |
2591 | double mV[6]; | |
2592 | double m[3]; | |
2593 | for( int i=0; i<3; i++ ){ | |
2594 | m[i] = mB[i]; | |
2595 | for( int k=0; k<6; k++ ) m[i]+=mG[i][k]*daughterP[id][k]; | |
a65041d0 | 2596 | } |
55ac3e1e | 2597 | for( int i=0; i<3; i++ ){ |
2598 | for( int j=0; j<6; j++ ){ | |
2599 | mGV[i][j] = 0; | |
2600 | for( int k=0; k<6; k++ ) mGV[i][j]+=mG[i][k]*daughterC[id][ IJ(k,j) ]; | |
2601 | } | |
a65041d0 | 2602 | } |
55ac3e1e | 2603 | for( int i=0, k=0; i<3; i++ ){ |
2604 | for( int j=0; j<=i; j++,k++ ){ | |
2605 | mV[k] = 0; | |
2606 | for( int l=0; l<6; l++ ) mV[k]+=mGV[i][l]*mG[j][l]; | |
2607 | } | |
2608 | } | |
2609 | ||
a65041d0 | 2610 | |
55ac3e1e | 2611 | //* CHt |
2612 | ||
2613 | Double_t mCHt[3][3]; | |
2614 | Double_t mHCHt[6]; | |
2615 | Double_t mHr[3]; | |
2616 | for( int i=0; i<3; i++ ){ | |
2617 | mHr[i] = 0; | |
2618 | for( int k=0; k<3; k++ ) mHr[i]+= mH[i][k]*r[k]; | |
2619 | } | |
a65041d0 | 2620 | |
55ac3e1e | 2621 | for( int i=0; i<3; i++ ){ |
2622 | for( int j=0; j<3; j++){ | |
2623 | mCHt[i][j] = 0; | |
2624 | for( int k=0; k<3; k++ ) mCHt[i][j]+= mC[i][k]*mH[j][k]; | |
2625 | } | |
a65041d0 | 2626 | } |
a65041d0 | 2627 | |
55ac3e1e | 2628 | for( int i=0, k=0; i<3; i++ ){ |
2629 | for( int j=0; j<=i; j++, k++ ){ | |
2630 | mHCHt[k] = 0; | |
2631 | for( int l=0; l<3; l++ ) mHCHt[k]+= mH[i][l]*mCHt[l][j]; | |
2632 | } | |
a65041d0 | 2633 | } |
a65041d0 | 2634 | |
55ac3e1e | 2635 | Double_t mS[6] = { mHCHt[0]+mV[0], |
2636 | mHCHt[1]+mV[1], mHCHt[2]+mV[2], | |
2637 | mHCHt[3]+mV[3], mHCHt[4]+mV[4], mHCHt[5]+mV[5] }; | |
a65041d0 | 2638 | |
2639 | ||
55ac3e1e | 2640 | InvertSym3(mS,mS); |
a65041d0 | 2641 | |
55ac3e1e | 2642 | //* Residual (measured - estimated) |
a65041d0 | 2643 | |
55ac3e1e | 2644 | Double_t zeta[3] = { m[0]-mHr[0], m[1]-mHr[1], m[2]-mHr[2] }; |
a65041d0 | 2645 | |
55ac3e1e | 2646 | //* Kalman gain K = mCH'*S |
a65041d0 | 2647 | |
55ac3e1e | 2648 | Double_t k[3][3]; |
a65041d0 | 2649 | |
55ac3e1e | 2650 | for(Int_t i=0;i<3;++i){ |
2651 | k[i][0] = mCHt[i][0]*mS[0] + mCHt[i][1]*mS[1] + mCHt[i][2]*mS[3]; | |
2652 | k[i][1] = mCHt[i][0]*mS[1] + mCHt[i][1]*mS[2] + mCHt[i][2]*mS[4]; | |
2653 | k[i][2] = mCHt[i][0]*mS[3] + mCHt[i][1]*mS[4] + mCHt[i][2]*mS[5]; | |
2654 | } | |
a65041d0 | 2655 | |
55ac3e1e | 2656 | //* New estimation of the vertex position r += K*zeta |
a65041d0 | 2657 | |
55ac3e1e | 2658 | for(Int_t i=0;i<3;++i) |
2659 | r[i] = r[i] + k[i][0]*zeta[0] + k[i][1]*zeta[1] + k[i][2]*zeta[2]; | |
a65041d0 | 2660 | |
55ac3e1e | 2661 | //* New covariance matrix C -= K*(mCH')' |
a65041d0 | 2662 | |
55ac3e1e | 2663 | for(Int_t i=0;i<3;++i){ |
2664 | for(Int_t j=0;j<=i;++j){ | |
2665 | mC[i][j] = mC[i][j] - (k[i][0]*mCHt[j][0] + k[i][1]*mCHt[j][1] + k[i][2]*mCHt[j][2]); | |
2666 | mC[j][i] = mC[i][j]; | |
2667 | } | |
a65041d0 | 2668 | } |
a65041d0 | 2669 | |
55ac3e1e | 2670 | //* Calculate Chi^2 |
a65041d0 | 2671 | |
55ac3e1e | 2672 | chi2 += ( ( mS[0]*zeta[0] + mS[1]*zeta[1] + mS[3]*zeta[2] )*zeta[0] |
2673 | +(mS[1]*zeta[0] + mS[2]*zeta[1] + mS[4]*zeta[2] )*zeta[1] | |
2674 | +(mS[3]*zeta[0] + mS[4]*zeta[1] + mS[5]*zeta[2] )*zeta[2] ); | |
2675 | } | |
a65041d0 | 2676 | |
55ac3e1e | 2677 | // store vertex |
a65041d0 | 2678 | |
55ac3e1e | 2679 | fNDF = 2; |
2680 | fChi2 = chi2; | |
2681 | for( int i=0; i<3; i++ ) fP[i] = r[i]; | |
2682 | for( int i=0,k=0; i<3; i++ ){ | |
2683 | for( int j=0; j<=i; j++,k++ ){ | |
2684 | fC[k] = mC[i][j]; | |
2685 | } | |
a65041d0 | 2686 | } |
2687 | } | |
55ac3e1e | 2688 | |
2689 | } // iterations | |
a65041d0 | 2690 | |
2691 | // now fit daughters to the vertex | |
2692 | ||
2693 | fQ = 0; | |
2694 | fSFromDecay = 0; | |
2695 | ||
2696 | for(Int_t i=3;i<8;++i) fP[i]=0.; | |
55ac3e1e | 2697 | for(Int_t i=6;i<35;++i) fC[i]=0.; |
2698 | fC[35] = 100.; | |
a65041d0 | 2699 | |
2700 | for( int id=0; id<2; id++ ){ | |
2701 | ||
2702 | double *p = daughterP[id]; | |
2703 | double *mC = daughterC[id]; | |
2704 | daughters[id]->GetMeasurement( v0, p, mC ); | |
2705 | ||
2706 | const Double_t *m = fP, *mV = fC; | |
2707 | ||
2708 | Double_t mAi[6]; | |
2709 | InvertSym3(mC, mAi ); | |
2710 | ||
2711 | Double_t mB[4][3]; | |
2712 | ||
2713 | mB[0][0] = mC[ 6]*mAi[0] + mC[ 7]*mAi[1] + mC[ 8]*mAi[3]; | |
2714 | mB[0][1] = mC[ 6]*mAi[1] + mC[ 7]*mAi[2] + mC[ 8]*mAi[4]; | |
2715 | mB[0][2] = mC[ 6]*mAi[3] + mC[ 7]*mAi[4] + mC[ 8]*mAi[5]; | |
2716 | ||
2717 | mB[1][0] = mC[10]*mAi[0] + mC[11]*mAi[1] + mC[12]*mAi[3]; | |
2718 | mB[1][1] = mC[10]*mAi[1] + mC[11]*mAi[2] + mC[12]*mAi[4]; | |
2719 | mB[1][2] = mC[10]*mAi[3] + mC[11]*mAi[4] + mC[12]*mAi[5]; | |
2720 | ||
2721 | mB[2][0] = mC[15]*mAi[0] + mC[16]*mAi[1] + mC[17]*mAi[3]; | |
2722 | mB[2][1] = mC[15]*mAi[1] + mC[16]*mAi[2] + mC[17]*mAi[4]; | |
2723 | mB[2][2] = mC[15]*mAi[3] + mC[16]*mAi[4] + mC[17]*mAi[5]; | |
2724 | ||
2725 | mB[3][0] = mC[21]*mAi[0] + mC[22]*mAi[1] + mC[23]*mAi[3]; | |
2726 | mB[3][1] = mC[21]*mAi[1] + mC[22]*mAi[2] + mC[23]*mAi[4]; | |
2727 | mB[3][2] = mC[21]*mAi[3] + mC[22]*mAi[4] + mC[23]*mAi[5]; | |
2728 | ||
2729 | ||
2730 | Double_t z[3] = { m[0]-p[0], m[1]-p[1], m[2]-p[2] }; | |
2731 | ||
6698c9c8 | 2732 | // { |
2733 | // Double_t mAV[6] = { mC[0]-mV[0], mC[1]-mV[1], mC[2]-mV[2], | |
2734 | // mC[3]-mV[3], mC[4]-mV[4], mC[5]-mV[5] }; | |
2735 | // | |
2736 | // Double_t mAVi[6]; | |
2737 | // if( !InvertSym3(mAV, mAVi) ){ | |
2738 | // Double_t dChi2 = ( +(mAVi[0]*z[0] + mAVi[1]*z[1] + mAVi[3]*z[2])*z[0] | |
2739 | // +(mAVi[1]*z[0] + mAVi[2]*z[1] + mAVi[4]*z[2])*z[1] | |
2740 | // +(mAVi[3]*z[0] + mAVi[4]*z[1] + mAVi[5]*z[2])*z[2] ); | |
2741 | // fChi2+= TMath::Abs( dChi2 ); | |
2742 | // } | |
2743 | // fNDF += 2; | |
2744 | // } | |
a65041d0 | 2745 | |
2746 | //* Add the daughter momentum to the particle momentum | |
2747 | ||
2748 | fP[3]+= p[3] + mB[0][0]*z[0] + mB[0][1]*z[1] + mB[0][2]*z[2]; | |
2749 | fP[4]+= p[4] + mB[1][0]*z[0] + mB[1][1]*z[1] + mB[1][2]*z[2]; | |
2750 | fP[5]+= p[5] + mB[2][0]*z[0] + mB[2][1]*z[1] + mB[2][2]*z[2]; | |
2751 | fP[6]+= p[6] + mB[3][0]*z[0] + mB[3][1]*z[1] + mB[3][2]*z[2]; | |
2752 | ||
2753 | Double_t d0, d1, d2; | |
2754 | ||
2755 | d0= mB[0][0]*mV[0] + mB[0][1]*mV[1] + mB[0][2]*mV[3] - mC[ 6]; | |
2756 | d1= mB[0][0]*mV[1] + mB[0][1]*mV[2] + mB[0][2]*mV[4] - mC[ 7]; | |
2757 | d2= mB[0][0]*mV[3] + mB[0][1]*mV[4] + mB[0][2]*mV[5] - mC[ 8]; | |
2758 | ||
2759 | //fC[6]+= mC[ 6] + d0; | |
2760 | //fC[7]+= mC[ 7] + d1; | |
2761 | //fC[8]+= mC[ 8] + d2; | |
2762 | fC[9]+= mC[ 9] + d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
2763 | ||
2764 | d0= mB[1][0]*mV[0] + mB[1][1]*mV[1] + mB[1][2]*mV[3] - mC[10]; | |
2765 | d1= mB[1][0]*mV[1] + mB[1][1]*mV[2] + mB[1][2]*mV[4] - mC[11]; | |
2766 | d2= mB[1][0]*mV[3] + mB[1][1]*mV[4] + mB[1][2]*mV[5] - mC[12]; | |
2767 | ||
2768 | //fC[10]+= mC[10]+ d0; | |
2769 | //fC[11]+= mC[11]+ d1; | |
2770 | //fC[12]+= mC[12]+ d2; | |
2771 | fC[13]+= mC[13]+ d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
2772 | fC[14]+= mC[14]+ d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
2773 | ||
2774 | d0= mB[2][0]*mV[0] + mB[2][1]*mV[1] + mB[2][2]*mV[3] - mC[15]; | |
2775 | d1= mB[2][0]*mV[1] + mB[2][1]*mV[2] + mB[2][2]*mV[4] - mC[16]; | |
2776 | d2= mB[2][0]*mV[3] + mB[2][1]*mV[4] + mB[2][2]*mV[5] - mC[17]; | |
2777 | ||
2778 | //fC[15]+= mC[15]+ d0; | |
2779 | //fC[16]+= mC[16]+ d1; | |
2780 | //fC[17]+= mC[17]+ d2; | |
2781 | fC[18]+= mC[18]+ d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
2782 | fC[19]+= mC[19]+ d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
2783 | fC[20]+= mC[20]+ d0*mB[2][0] + d1*mB[2][1] + d2*mB[2][2]; | |
2784 | ||
2785 | d0= mB[3][0]*mV[0] + mB[3][1]*mV[1] + mB[3][2]*mV[3] - mC[21]; | |
2786 | d1= mB[3][0]*mV[1] + mB[3][1]*mV[2] + mB[3][2]*mV[4] - mC[22]; | |
2787 | d2= mB[3][0]*mV[3] + mB[3][1]*mV[4] + mB[3][2]*mV[5] - mC[23]; | |
2788 | ||
2789 | //fC[21]+= mC[21] + d0; | |
2790 | //fC[22]+= mC[22] + d1; | |
2791 | //fC[23]+= mC[23] + d2; | |
2792 | fC[24]+= mC[24] + d0*mB[0][0] + d1*mB[0][1] + d2*mB[0][2]; | |
2793 | fC[25]+= mC[25] + d0*mB[1][0] + d1*mB[1][1] + d2*mB[1][2]; | |
2794 | fC[26]+= mC[26] + d0*mB[2][0] + d1*mB[2][1] + d2*mB[2][2]; | |
2795 | fC[27]+= mC[27] + d0*mB[3][0] + d1*mB[3][1] + d2*mB[3][2]; | |
2796 | } | |
2797 | ||
57ddc2cb | 2798 | // SetMassConstraint(0,0); |
2799 | SetNonlinearMassConstraint(0); | |
a65041d0 | 2800 | } |
2801 | ||
24be0330 | 2802 | void AliKFParticleBase::GetArmenterosPodolanski(AliKFParticleBase& positive, AliKFParticleBase& negative, Double_t QtAlfa[2] ) |
2803 | { | |
c3af7f6d | 2804 | // example: |
2805 | // AliKFParticle PosParticle(...) | |
2806 | // AliKFParticle NegParticle(...) | |
2807 | // Gamma.ConstructGamma(PosParticle, NegParticle); | |
2808 | // Double_t VertexGamma[3] = {Gamma.GetX(), Gamma.GetY(), Gamma.GetZ()}; | |
2809 | // PosParticle.TransportToPoint(VertexGamma); | |
2810 | // NegParticle.TransportToPoint(VertexGamma); | |
2811 | // Double_t armenterosQtAlfa[2] = {0.}; | |
2812 | // AliKFParticle::GetArmenterosPodolanski(PosParticle, NegParticle, armenterosQtAlfa ); | |
2813 | ||
24be0330 | 2814 | Double_t alpha = 0., qt = 0.; |
2815 | Double_t spx = positive.GetPx() + negative.GetPx(); | |
2816 | Double_t spy = positive.GetPy() + negative.GetPy(); | |
2817 | Double_t spz = positive.GetPz() + negative.GetPz(); | |
2818 | Double_t sp = sqrt(spx*spx + spy*spy + spz*spz); | |
2819 | if( sp == 0.0) return; | |
026bab91 | 2820 | Double_t pn, pln, plp; // ,pp; |
24be0330 | 2821 | |
2822 | pn = TMath::Sqrt(negative.GetPx()*negative.GetPx() + negative.GetPy()*negative.GetPy() + negative.GetPz()*negative.GetPz()); | |
026bab91 | 2823 | // pp = TMath::Sqrt(positive.GetPx()*positive.GetPx() + positive.GetPy()*positive.GetPy() + positive.GetPz()*positive.GetPz()); |
24be0330 | 2824 | pln = (negative.GetPx()*spx+negative.GetPy()*spy+negative.GetPz()*spz)/sp; |
2825 | plp = (positive.GetPx()*spx+positive.GetPy()*spy+positive.GetPz()*spz)/sp; | |
2826 | ||
2827 | if( pn == 0.0) return; | |
2828 | Double_t ptm = (1.-((pln/pn)*(pln/pn))); | |
2829 | qt= (ptm>=0.)? pn*sqrt(ptm) :0; | |
2830 | alpha = (plp-pln)/(plp+pln); | |
2831 | ||
2832 | QtAlfa[0] = qt; | |
2833 | QtAlfa[1] = alpha; | |
2834 | } | |
2835 | ||
2836 | void AliKFParticleBase::RotateXY(Double_t angle, Double_t Vtx[3]) | |
2837 | { | |
2838 | // Rotates the KFParticle object around OZ axis, OZ axis is set by the vertex position | |
2839 | // Double_t angle - angle of rotation in XY plane in [rad] | |
2840 | // Double_t Vtx[3] - position of the vertex in [cm] | |
24be0330 | 2841 | |
95142b87 | 2842 | // Before rotation the center of the coordinat system should be moved to the vertex position; move back after rotation |
2843 | X() = X() - Vtx[0]; | |
2844 | Y() = Y() - Vtx[1]; | |
2845 | Z() = Z() - Vtx[2]; | |
24be0330 | 2846 | |
2847 | // Rotate the kf particle | |
2848 | Double_t c = TMath::Cos(angle); | |
2849 | Double_t s = TMath::Sin(angle); | |
2850 | ||
2851 | Double_t mA[8][ 8]; | |
2852 | for( Int_t i=0; i<8; i++ ){ | |
2853 | for( Int_t j=0; j<8; j++){ | |
2854 | mA[i][j] = 0; | |
2855 | } | |
2856 | } | |
2857 | for( int i=0; i<8; i++ ){ | |
2858 | mA[i][i] = 1; | |
2859 | } | |
2860 | mA[0][0] = c; mA[0][1] = s; | |
2861 | mA[1][0] = -s; mA[1][1] = c; | |
2862 | mA[3][3] = c; mA[3][4] = s; | |
2863 | mA[4][3] = -s; mA[4][4] = c; | |
2864 | ||
2865 | Double_t mAC[8][8]; | |
2866 | Double_t mAp[8]; | |
2867 | ||
2868 | for( Int_t i=0; i<8; i++ ){ | |
2869 | mAp[i] = 0; | |
2870 | for( Int_t k=0; k<8; k++){ | |
2871 | mAp[i]+=mA[i][k] * fP[k]; | |
2872 | } | |
2873 | } | |
2874 | ||
2875 | for( Int_t i=0; i<8; i++){ | |
2876 | fP[i] = mAp[i]; | |
2877 | } | |
2878 | ||
2879 | for( Int_t i=0; i<8; i++ ){ | |
2880 | for( Int_t j=0; j<8; j++ ){ | |
2881 | mAC[i][j] = 0; | |
2882 | for( Int_t k=0; k<8; k++ ){ | |
2883 | mAC[i][j]+= mA[i][k] * GetCovariance(k,j); | |
2884 | } | |
2885 | } | |
2886 | } | |
2887 | ||
2888 | for( Int_t i=0; i<8; i++ ){ | |
2889 | for( Int_t j=0; j<=i; j++ ){ | |
2890 | Double_t xx = 0; | |
2891 | for( Int_t k=0; k<8; k++){ | |
2892 | xx+= mAC[i][k]*mA[j][k]; | |
2893 | } | |
2894 | Covariance(i,j) = xx; | |
2895 | } | |
2896 | } | |
2897 | ||
95142b87 | 2898 | X() = GetX() + Vtx[0]; |
2899 | Y() = GetY() + Vtx[1]; | |
2900 | Z() = GetZ() + Vtx[2]; | |
24be0330 | 2901 | } |
2902 | ||
a65041d0 | 2903 | Bool_t AliKFParticleBase::InvertSym3( const Double_t A[], Double_t Ai[] ) |
2904 | { | |
2905 | //* Invert symmetric matric stored in low-triagonal form | |
2906 | ||
2907 | bool ret = 0; | |
2908 | double a0 = A[0], a1 = A[1], a2 = A[2], a3 = A[3]; | |
2909 | ||
2910 | Ai[0] = a2*A[5] - A[4]*A[4]; | |
2911 | Ai[1] = a3*A[4] - a1*A[5]; | |
2912 | Ai[3] = a1*A[4] - a2*a3; | |
2913 | Double_t det = (a0*Ai[0] + a1*Ai[1] + a3*Ai[3]); | |
57ddc2cb | 2914 | if( TMath::Abs(det)>1.e-20 ) det = 1./det; |
a65041d0 | 2915 | else{ |
2916 | det = 0; | |
2917 | ret = 1; | |
2918 | } | |
2919 | Ai[0] *= det; | |
2920 | Ai[1] *= det; | |
2921 | Ai[3] *= det; | |
2922 | Ai[2] = ( a0*A[5] - a3*a3 )*det; | |
2923 | Ai[4] = ( a1*a3 - a0*A[4] )*det; | |
2924 | Ai[5] = ( a0*a2 - a1*a1 )*det; | |
2925 | return ret; | |
2926 | } | |
2927 | ||
f826d409 | 2928 | void AliKFParticleBase::MultQSQt( const Double_t Q[], const Double_t S[], Double_t SOut[] ) |
2929 | { | |
2930 | //* Matrix multiplication Q*S*Q^T, Q - square matrix, S - symmetric | |
2931 | ||
2932 | const Int_t kN= 8; | |
2933 | Double_t mA[kN*kN]; | |
2934 | ||
2935 | for( Int_t i=0, ij=0; i<kN; i++ ){ | |
2936 | for( Int_t j=0; j<kN; j++, ++ij ){ | |
2937 | mA[ij] = 0 ; | |
2938 | for( Int_t k=0; k<kN; ++k ) mA[ij]+= S[( k<=i ) ? i*(i+1)/2+k :k*(k+1)/2+i] * Q[ j*kN+k]; | |
2939 | } | |
2940 | } | |
2941 | ||
2942 | for( Int_t i=0; i<kN; i++ ){ | |
2943 | for( Int_t j=0; j<=i; j++ ){ | |
2944 | Int_t ij = ( j<=i ) ? i*(i+1)/2+j :j*(j+1)/2+i; | |
2945 | SOut[ij] = 0 ; | |
2946 | for( Int_t k=0; k<kN; k++ ) SOut[ij] += Q[ i*kN+k ] * mA[ k*kN+j ]; | |
2947 | } | |
2948 | } | |
2949 | } | |
2950 | ||
2951 | ||
2952 | // 72-charachters line to define the printer border | |
2953 | //3456789012345678901234567890123456789012345678901234567890123456789012 | |
446ce366 | 2954 |