]> git.uio.no Git - u/mrichter/AliRoot.git/blame - TPC/AliTPCTrackHitsV2.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Print removed
[u/mrichter/AliRoot.git] / TPC / AliTPCTrackHitsV2.cxx
CommitLineData
f641f6bd 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
88cb7938 16/* $Id$ */
19364939 17
f641f6bd 18///////////////////////////////////////////////////////////////////////////////
19// //
20// Time Projection Chamber track hits object //
21//
22// Origin: Marian Ivanov , GSI Darmstadt
23//
24// AliTPCTrackHitsV2
25// Container for Track Hits - based on standard TClonesArray -
26// fArray of AliTPCTrackHitsParamV2
27// In AliTPCTrackHitsParamV2 - parameterization of the track segment is stored
28// for each of the track segment - relative position ( distance between hits) and
29// charge of the hits is stored - comparing to classical TClonesArray of AliTPChit -
30// comperssion factor of 5-7 (depending on the required precision) -
31// In future release AliTPCTrackHitsV2 - will replace old AliTPCTrackHits - which were not
32// based on standard ROOT containers
33// Basic function:
34// // during building Container
35// AddHitKartez(Int_t volumeID, Int_t trackID, Double_t x, Double_t y, Double_t z,Int_t q)
36// void SetHitPrecision(Double_t prec) {fPrecision=prec;}
37// void SetStepPrecision(Double_t prec) {fStep=prec;}
38// Bool_t FlushHitStack(Bool_t force=kTRUE);
39// //at the end necessary to have Container in consistent state
40//
41// // looping over Container
42// Bool_t First(), Bool_t Next() - iterators - return status of the operation
43// AliTPChit * GetHit(); - return current hit
44
45
46//Begin_Html
47/*
48<img src="gif/AliTPCTrackHitsV2.gif">
49*/
50//End_Html
51// //
52// //
53///////////////////////////////////////////////////////////////////////////////
018a927a 54//
f641f6bd 55
f641f6bd 56#include "AliTPCTrackHitsV2.h"
f641f6bd 57#include "TClonesArray.h"
58#include "AliTPC.h"
59
f641f6bd 60
61
62ClassImp(AliTPCTrackHitsV2)
63ClassImp(AliTrackHitsParamV2)
64
65 //
66Int_t AliTrackHitsParamV2::fgCounter1 =0;
67Int_t AliTrackHitsParamV2::fgCounter2 =0;
68//
69Int_t AliTPCTrackHitsV2::fgCounter1 =0;
70Int_t AliTPCTrackHitsV2::fgCounter2 =0;
71//
72const Double_t AliTPCTrackHitsV2::fgkPrecision=1e-6; //precision
73const Double_t AliTPCTrackHitsV2::fgkPrecision2=1e-20; //precision
74
75
76
77
f641f6bd 78struct AliTPCTempHitInfoV2 {
f43e383b 79 friend class AliTPCTrackHitsV2;
80protected:
53110b9d 81 enum { kStackSize = 10000};
f641f6bd 82 AliTPCTempHitInfoV2();
53110b9d 83 AliTPCTempHitInfoV2(const AliTPCTempHitInfoV2 &hit)
84 {hit.Copy(*this);}
85 AliTPCTempHitInfoV2& operator = (const AliTPCTempHitInfoV2 &hit)
86 {hit.Copy(*this); return (*this);}
f641f6bd 87 void NewParam(Double_t r, Double_t z, Double_t fi, Int_t q);
88 void SetHit(Double_t r, Double_t z, Double_t fi, Int_t q);
89 Double_t * GetPosition(Int_t index){return &fPositionStack[index*3];}
90 void UpdateParam(Double_t maxdelta); //recal
91 void Fit2(Double_t fSumY, Double_t fSumYX, Double_t fSumYX2,
92 Double_t fSumX, Double_t fSumX2, Double_t fSumX3,
93 Double_t fSumX4, Int_t n,
94 Double_t &a, Double_t &b, Double_t &c);
95 void Fit(AliTrackHitsParamV2 * param);
53110b9d 96 Double_t fSumDr; // Sum of Dr
97 Double_t fSumDr2; // Square of sum of Dr
98 Double_t fSumDr3; // Cube of sum of Dr
99 Double_t fSumDr4; // Fourth power of sum of Dr
100 Double_t fSumDFi; // Sum of DFi
101 Double_t fSumDFiDr; // Sum of DFiDr
102 Double_t fSumDFiDr2;// Sum of square of DFiDr
103 Double_t fSumDZ; // Sum of DZ
104 Double_t fSumDZDr; // Sum of DZDr
105 Double_t fSumDZDr2; // Sum of square of DZDr
f641f6bd 106 Double_t fOldR; //previos r
53110b9d 107 Double_t fPositionStack[3*kStackSize]; //position stack
108 UInt_t fQStack[kStackSize]; //Q stack
f641f6bd 109 UInt_t fStackIndex; //current stack index
110 // UInt_t fInfoIndex; //current track info index
111 UInt_t fParamIndex; //current track parameters index
112 // AliTrackHitsInfo * fInfo; //current track info
113 AliTrackHitsParamV2 * fParam; //current track param
53110b9d 114private:
115 void Copy(AliTPCTempHitInfoV2 &) const
116 {printf("Not Implemented\n"); exit(1);}
f641f6bd 117};
118
119
120AliTPCTempHitInfoV2::AliTPCTempHitInfoV2()
121{
122 //
53110b9d 123 // Standard constructor
124 // set to default value
125 //
f641f6bd 126 fSumDr=fSumDr2=fSumDr3=fSumDr4=
127 fSumDFi=fSumDFiDr=fSumDFiDr2=
128 fSumDZ=fSumDZDr=fSumDZDr2=0;
129 fStackIndex = 0;
130 // fInfoIndex = 0;
131 fParamIndex = 0;
132}
133
134
135void AliTPCTempHitInfoV2::NewParam(Double_t r, Double_t z, Double_t fi, Int_t q)
136{
137 //
138 //reset stack and sum parameters
139 //store line initial point
53110b9d 140 //
f641f6bd 141 fSumDr=fSumDr2=fSumDr3=fSumDr4=
142 fSumDFi=fSumDFiDr=fSumDFiDr2=
143 fSumDZ=fSumDZDr=fSumDZDr2=0;
144 fStackIndex=0;
145 fParam->fR = r;
146 fOldR = r;
147 fParam->fZ = z;
148 fParam->fFi = fi;
149 fParam->fAn = 0.;
150 fParam->fAd = 0.;
151 fParam->fTheta =0.;
152 fParam->fThetaD =0.;
153 SetHit(r,z,fi,q);
154}
155
156void AliTPCTempHitInfoV2::SetHit(Double_t r, Double_t z, Double_t fi, Int_t q)
157{
158 //
159 //add hit to the stack
160 //recalculate new estimete of line parameters
161 Double_t *f = GetPosition(fStackIndex);
162 f[0] = r;
163 f[1] = z;
164 f[2] = fi;
165 fQStack[fStackIndex]=q;
166 if (fStackIndex==0) return;
167 Double_t dr = (r-fParam->fR);
168 if (TMath::Abs(dr)<AliTPCTrackHitsV2::fgkPrecision) dr =AliTPCTrackHitsV2::fgkPrecision;
169 Double_t dfi = fi-fParam->fFi;
170 Double_t dz = z -fParam->fZ;
171 Double_t dr2 =dr*dr;
172 Double_t dr3 =dr2*dr;
173 Double_t dr4 =dr3*dr;
174 fSumDr +=dr;
175 fSumDr2+=dr2;
176 fSumDr3+=dr3;
177 fSumDr4+=dr4;
178 fSumDFi +=dfi;
179 fSumDFiDr+=dfi*dr;
180 fSumDFiDr2+=dfi*dr2;
181 fSumDZ +=dz;
182 fSumDZDr+=dz*dr;
183 fSumDZDr2+=dz*dr2;
184
185 //update fit parameters
186 //
187 Double_t det = fSumDr2*fSumDr4-fSumDr3*fSumDr3;
188 if (TMath::Abs(det)<AliTPCTrackHitsV2::fgkPrecision2) return;
189 if ( ( fStackIndex>1 ) ){
190 fParam->fAn = (fSumDr4*fSumDFiDr-fSumDr3*fSumDFiDr2)/det;
191 fParam->fAd = (fSumDr2*fSumDFiDr2-fSumDr3*fSumDFiDr)/det;
192 }
193 else
194 fParam->fAn = fSumDFiDr/fSumDr2;
195 if ( ( fStackIndex>1 ) ){
196 fParam->fTheta = (fSumDr4*fSumDZDr-fSumDr3*fSumDZDr2)/det;
197 fParam->fThetaD= (fSumDr2*fSumDZDr2-fSumDr3*fSumDZDr)/det;
198 }
199 else
200 fParam->fTheta = fSumDZDr/fSumDr2;
201}
202
203
204void AliTPCTempHitInfoV2::UpdateParam(Double_t maxdelta)
205{
53110b9d 206 //
207 // recalc parameters not fixing origin point
208 //
f641f6bd 209 if (fStackIndex>5){
210 Double_t a,b,c;
211 a=b=c=0;
212 Fit2(fSumDFi, fSumDFiDr, fSumDFiDr2, fSumDr,fSumDr2,fSumDr3,fSumDr4,
213 fStackIndex, a,b,c);
214 if (TMath::Abs(a)<maxdelta){
215 fParam->fFi +=a/fParam->fR;
216 fParam->fAn = b;
217 fParam->fAd = c;
218 }
219 Fit2(fSumDZ, fSumDZDr, fSumDZDr2, fSumDr,fSumDr2,fSumDr3,fSumDr4,
220 fStackIndex, a,b,c) ;
221 if (TMath::Abs(a)<maxdelta){
222 fParam->fZ +=a;
223 fParam->fTheta = b;
224 fParam->fThetaD = c;
225 }
226 }
227
228}
53110b9d 229
f641f6bd 230void AliTPCTempHitInfoV2::Fit2(Double_t fSumY, Double_t fSumYX, Double_t fSumYX2,
231 Double_t fSumX, Double_t fSumX2, Double_t fSumX3,
232 Double_t fSumX4, Int_t n,
233 Double_t &a, Double_t &b, Double_t &c)
234{
53110b9d 235 //
236 // fit of second order
237 //
f641f6bd 238 Double_t det =
239 n* (fSumX2*fSumX4-fSumX3*fSumX3) -
240 fSumX* (fSumX*fSumX4-fSumX3*fSumX2)+
241 fSumX2* (fSumX*fSumX3-fSumX2*fSumX2);
242
243 if (TMath::Abs(det)> AliTPCTrackHitsV2::fgkPrecision) {
244 a =
245 (fSumY * (fSumX2*fSumX4-fSumX3*fSumX3)-
246 fSumX *(fSumYX*fSumX4-fSumYX2*fSumX3)+
247 fSumX2*(fSumYX*fSumX3-fSumYX2*fSumX2))/det;
248 b=
249 (n*(fSumYX*fSumX4-fSumX3*fSumYX2)-
250 fSumY*(fSumX*fSumX4-fSumX3*fSumX2)+
251 fSumX2*(fSumX*fSumYX2-fSumYX*fSumX2))/det;
252 c=
253 (n*(fSumX2*fSumYX2-fSumYX*fSumX3)-
254 fSumX*(fSumX*fSumYX2-fSumYX*fSumX2)+
255 fSumY*(fSumX*fSumX3-fSumX2*fSumX2))/det;
256 }
257}
258
259void AliTPCTempHitInfoV2::Fit(AliTrackHitsParamV2 * param)
260{
53110b9d 261 //
f641f6bd 262 // fit fixing first and the last point
53110b9d 263 // result stored in new param
264 //
f641f6bd 265 Double_t dx2 = (GetPosition(fStackIndex))[0]-fParam->fR;
266 Double_t det = fSumDr4+dx2*fSumDr2-2*dx2*fSumDr3;
267 if ( (TMath::Abs(det)> AliTPCTrackHitsV2::fgkPrecision) &&
268 ((TMath::Abs(dx2)> AliTPCTrackHitsV2::fgkPrecision))){
269 Double_t dfi2 = (GetPosition(fStackIndex))[1]-fParam->fFi;
270 param->fAd = (fSumDFiDr2+dfi2*fSumDr-dx2*fSumDFiDr-dfi2*fSumDr3/dx2)/det;
271 param->fAn = (dfi2-param->fAd*dx2*dx2)/dx2;
272
273 Double_t dz2 = (GetPosition(fStackIndex))[1]-fParam->fZ;
274 param->fTheta = (fSumDZDr2+dz2*fSumDr-dx2*fSumDZDr-dz2*fSumDr3/dx2)/det;
275 param->fTheta = (dz2-param->fAd*dx2*dx2)/dx2;
276 }
277
278}
279
280AliTrackHitsParamV2::AliTrackHitsParamV2()
281{
53110b9d 282 //
283 // default constructor
284 //
f641f6bd 285 fgCounter1++;
286 fgCounter2++;
287 fHitDistance=0;
288 fCharge=0;
289 fNHits=0;
290}
53110b9d 291
f641f6bd 292AliTrackHitsParamV2::~AliTrackHitsParamV2()
293{
53110b9d 294 //
295 // Standard destructor
296 //
f641f6bd 297 fgCounter1--;
298 if (fHitDistance) {
299 delete[]fHitDistance;
300 fHitDistance=0;
301 }
302 if (fCharge){
303 delete[]fCharge;
304 fCharge =0;
305 }
306}
307
308
309AliTPCTrackHitsV2::AliTPCTrackHitsV2()
310{
311 //
312 //default constructor
313 //
314 const Float_t kHitPrecision=0.002; //default precision for hit position in cm
315 const Float_t kStep =0.003; //30 mum step
316 const UShort_t kMaxDistance =100; //maximum distance 100
317
318 fPrecision=kHitPrecision; //precision in cm
319 fStep = kStep; //step size
320 fMaxDistance = kMaxDistance; //maximum distance
321 fTempInfo =0;
322 fSize=0;
323 //fTrackHitsInfo = new AliObjectArray("AliTrackHitsInfo");
324 //fTrackHitsParam = new AliObjectArray("AliTrackHitsParamV2");
325 //fHitsPosAndQ = new TArrayOfArrayVStack("AliHitInfo");
326 fArray = new TClonesArray("AliTrackHitsParamV2");
327 fCurrentHit = new AliTPCCurrentHitV2;
328 fVolumes =0;
329 fNVolumes =0;
fce0283f 330 fHit =0;
f641f6bd 331 fgCounter1++;
332 fgCounter2++;
333
334}
335
336AliTPCTrackHitsV2::~AliTPCTrackHitsV2()
337{
338 //
339 //default destructor
340 //
341 // if (fTrackHitsInfo) delete fTrackHitsInfo;
342 if (fArray) {
343 delete fArray;
344 fArray =0;
345 }
346 //if (fHitsPosAndQ) delete fHitsPosAndQ;
347 if (fCurrentHit) delete fCurrentHit;
348 if (fTempInfo) delete fTempInfo;
349 if (fVolumes) {
350 delete [] fVolumes;
351 fVolumes =0;
352 fNVolumes=0;
353 }
fce0283f 354 if (fHit){
355 delete fHit;
356 fHit=0;
357 }
f641f6bd 358 fgCounter1--;
359}
360
361void AliTPCTrackHitsV2::Clear()
362{
363 //
53110b9d 364 // clear object
365 //
6e493dba 366 fSize = 0;
f641f6bd 367 if (fArray){
368 for (Int_t i=0;i<fArray->GetEntriesFast();i++){
369 AliTrackHitsParamV2 * par = (AliTrackHitsParamV2 *)fArray->UncheckedAt(i);
370 par->~AliTrackHitsParamV2(); // delete object
371 }
372 fArray->Clear();
373 }
374 if (fTempInfo){
375 delete fTempInfo;
fce0283f 376 delete fHit;
377 fHit =0;
f641f6bd 378 fTempInfo =0;
379 }
380 if (fVolumes){
381 delete [] fVolumes;
382 fVolumes=0;
383 fNVolumes=0;
384 }
385}
386
387
388void AliTPCTrackHitsV2::AddHitKartez(Int_t volumeID, Int_t trackID, Double_t x,
389 Double_t y, Double_t z,Int_t q)
390{
391 //
53110b9d 392 // add hit to the container - it add hit at the end - input in global coordinata
393 //
f641f6bd 394 Double_t r = TMath::Sqrt(x*x+y*y);
395 Double_t fi = TMath::ACos(x/r);
396 if (y<0) fi*=-1.;
397 AddHit(volumeID,trackID,r,z,fi,q);
398}
399
400
401void AliTPCTrackHitsV2::AddHit(Int_t volumeID, Int_t trackID,
402 Double_t r, Double_t z, Double_t fi, Int_t q)
403{
53110b9d 404 //
405 // Adding one hit
f641f6bd 406 //
407 fSize++;
408 Bool_t diff=kFALSE;
409 if (!fTempInfo) { //initialisation of track - initialisation of parameters
410 fTempInfo = new AliTPCTempHitInfoV2;
411 fTempInfo->fParam = new((*fArray)[0]) AliTrackHitsParamV2;
412 fTempInfo->fParam->fVolumeID = volumeID;
413 fTempInfo->fParam->fTrackID = trackID;
414 AddVolume(volumeID);
415 //
416 fTempInfo->fParamIndex = 0;
417 fTempInfo->NewParam(r,z,fi,q);
418 return;
419 }
420
421 // if new volume or new trackID
422 if ( (volumeID!=fTempInfo->fParam->fVolumeID) ||
423 (trackID!=fTempInfo->fParam->fTrackID)){
424 if (volumeID!=fTempInfo->fParam->fVolumeID) AddVolume(volumeID);
425 diff=kTRUE;
426 FlushHitStack(kTRUE);
427
428 fTempInfo->fParamIndex++;
429 fTempInfo->fParam = new((*fArray)[fTempInfo->fParamIndex]) AliTrackHitsParamV2;
430 fTempInfo->fParam->fVolumeID = volumeID;
431 fTempInfo->fParam->fTrackID = trackID;
432 fTempInfo->NewParam(r,z,fi,q);
433 return;
434 }
435
436 //calculate current fit precission to next point
437 AliTrackHitsParamV2 &param = *(fTempInfo->fParam);
438 Double_t dd=0;
439 Double_t dl=0;
440 Double_t ratio=0;
441 Double_t dr,dz,dfi,ddz,ddfi;
442 Double_t drhit,ddl;
443 dr=dz=dfi=ddz=ddfi=0;
444 drhit = r-fTempInfo->fOldR;
445 {
446 //Double_t dfi2 = param.fAn+2*param.fAd*(r-param.fR);
447 Double_t dfi2 = param.fAn;
448 dfi2*=dfi2*fTempInfo->fOldR*fTempInfo->fOldR;
449 //Double_t ddz2 = param.fTheta+2*param.fThetaD*(r-param.fR);
450 Double_t ddz2 = param.fTheta;
451 ddz2*=ddz2;
452 ratio = TMath::Sqrt(1.+ dfi2+ ddz2);
453 }
c4a237ae 454 //
455 // dl = fStep * Short_t(TMath::Nint(drhit*ratio/fStep)); // MI change - range check
456 dl = drhit*ratio/fStep;
457 if (TMath::Abs(dl)>32765) dl =0;
458 dl = fStep * Short_t(TMath::Nint(dl));
459 //
f641f6bd 460 ddl = dl - drhit*ratio;
461 fTempInfo->fOldR += dl/ratio;
462
463 if (fTempInfo->fStackIndex>2){
464 dr = r-param.fR;
465 dz = z-param.fZ;
466 dfi = fi-param.fFi;
467 ddz = dr*param.fTheta+dr*dr*param.fThetaD-dz;
468 ddfi= dr*param.fAn+dr*dr*param.fAd-dfi;
469 dd = TMath::Sqrt(ddz*ddz+r*r*ddfi*ddfi+ddl*ddl);
470 //
471 }
472 //safety factor 1.25
473 if ( ( (dd*1.25>fPrecision) ) ||
53110b9d 474 (fTempInfo->fStackIndex+4>fTempInfo->kStackSize) ||
f641f6bd 475 (TMath::Abs(dl/fStep)>fMaxDistance) )
476 diff=kTRUE;
477 else{ // if precision OK
478 fTempInfo->fStackIndex++;
479 fTempInfo->SetHit(r,z,fi,q);
480 return;
481 }
482
483
484 //if parameter changed
485 if (FlushHitStack(kFALSE)){ //if full buffer flushed
486 fTempInfo->fParamIndex++;
487 fTempInfo->fParam = new((*fArray)[fTempInfo->fParamIndex]) AliTrackHitsParamV2;
488 fTempInfo->fParam->fVolumeID = volumeID;
489 fTempInfo->fParam->fTrackID = trackID;
490 fTempInfo->NewParam(r,z,fi,q);
491 }
492 else{
493 fTempInfo->fStackIndex++;
494 fTempInfo->SetHit(r,z,fi,q);
495 }
496}
497
498Bool_t AliTPCTrackHitsV2::FlushHitStack(Bool_t force)
499{
500 //
53110b9d 501 // write fHitsPosAndQ information from the stack to te arrays
502 //
f641f6bd 503 if (!fTempInfo) return kFALSE;
504
505 AliTrackHitsParamV2 & param = *(fTempInfo->fParam);
506 //recalculate track parameter not fixing first point
507 fTempInfo->UpdateParam(fStep/4.);
508 //fTempInfo->Fit(fTempInfo->fParam); //- fixing the first and the last point
509
510 Double_t oldr = param.fR;
511 UInt_t i;
512 Double_t dd;
513 param.fNHits = fTempInfo->fStackIndex+1;
514 if (param.fHitDistance) delete []param.fHitDistance;
515 if (param.fCharge) delete []param.fCharge;
516 param.fHitDistance = new Short_t[param.fNHits];
517 param.fCharge = new Short_t[param.fNHits];
518
519
520 for (i=0; i <= fTempInfo->fStackIndex; i++){
521 Double_t * position = fTempInfo->GetPosition(i);
522 Double_t dr = position[0]-oldr;
523 Double_t ratio;
524 {
525 //Double_t dfi2 = param.fAn+2*param.fAd*(position[0]-param.fR);
526 Double_t dfi2 = param.fAn;
527 dfi2*=dfi2*oldr*oldr;
528 //Double_t ddz2 = param.fTheta+2*param.fThetaD*(position[0]-param.fR);
529 Double_t ddz2 = param.fTheta;
530 ddz2*=ddz2;
531 ratio = TMath::Sqrt(1.+ dfi2+ ddz2);
532 }
533
595c3700 534 // Double_t dl = fStep*(Short_t)TMath::Nint(dr*ratio/fStep); //MI change
535 Double_t dl = dr*ratio/fStep;
536 if (TMath::Abs(dl)>32765) dl =0;
537 dl = fStep * Short_t(TMath::Nint(dl));
538
f641f6bd 539 dr = dl/ratio;
540 oldr+=dr;
541 //calculate precission
542 AliTrackHitsParamV2 &param = *(fTempInfo->fParam);
543 //real deltas
544 Double_t dr1= position[0]-param.fR;
545 Double_t dz = position[1]-param.fZ;
546 Double_t dfi = position[2]-param.fFi;
547 //extrapolated deltas
548 Double_t dr2 = oldr-param.fR;
549 Double_t ddr = dr2-dr1;
550 Double_t ddz = dr2*param.fTheta+dr2*dr2*param.fThetaD-dz;
551 Double_t ddfi= dr2*param.fAn+dr2*dr2*param.fAd-dfi;
552 dd = TMath::Sqrt(ddz*ddz+oldr*oldr*ddfi*ddfi+ddr*ddr);
553
554
555 if ( (dd>fPrecision) ){
556 //if ( (dd<0) ){
557 if (i==0){
558 param.fAn = 0;
559 param.fAd = 0;
560 param.fTheta =0;
561 param.fThetaD =0;
562 Double_t ddz = dr2*param.fTheta+dr2*dr2*param.fThetaD-dz;
563 Double_t ddfi= dr2*param.fAn+dr2*dr2*param.fAd-dfi;
564 dl = 0;
565 dd = TMath::Sqrt(ddz*ddz+oldr*oldr*ddfi*ddfi+ddr*ddr);
566 }
567 else
568 break;
569 }
570
571 param.fHitDistance[i]= Short_t(TMath::Nint(dl/fStep));
572 param.fCharge[i]= Short_t(fTempInfo->fQStack[i]);
573 }
574
575 if (i<=fTempInfo->fStackIndex){ //if previous iteration not succesfull
576 Short_t * charge = new Short_t[i];
577 Short_t * hitDistance= new Short_t[i];
578 memcpy(charge, param.fCharge,sizeof(Short_t)*i);
579 memcpy(hitDistance, param.fHitDistance,sizeof(Short_t)*i);
580 delete [] param.fCharge;
581 delete [] param.fHitDistance;
582 param.fNHits= i;
583 param.fCharge = charge;
584 param.fHitDistance = hitDistance;
585 //
586 Int_t volumeID = fTempInfo->fParam->fVolumeID;
587 Int_t trackID =fTempInfo->fParam->fTrackID;
588 fTempInfo->fParamIndex++;
589 fTempInfo->fParam = new((*fArray)[fTempInfo->fParamIndex]) AliTrackHitsParamV2;
590 Double_t * p = fTempInfo->GetPosition(i);
591 UInt_t index2 = fTempInfo->fStackIndex;
592 fTempInfo->NewParam(p[0],p[1],p[2],fTempInfo->fQStack[i]);
593 fTempInfo->fParam->fVolumeID= volumeID;
594 fTempInfo->fParam->fTrackID= trackID;
595 if (i+1<=index2) FlushHitStack2(i+1,index2);
596
597 if (force) return FlushHitStack(kTRUE);
598 return kFALSE;
599 }
600 return kTRUE;
601}
602
603
604void AliTPCTrackHitsV2::FlushHitStack2(Int_t index1, Int_t index2)
605{
606 //
607 // second iteration flush stack
53110b9d 608 // call only for hits where first iteration were not succesfully interpolated
609 //
f641f6bd 610 Double_t * positionstack = new Double_t[3*(index2-index1+1)];
611 UInt_t * qstack = new UInt_t[index2-index1+1];
612 memcpy(positionstack, &fTempInfo->fPositionStack[3*index1],
613 (3*(index2-index1+1))*sizeof(Double_t));
614 memcpy(qstack, &fTempInfo->fQStack[index1],(index2-index1+1)*sizeof(UInt_t));
615 Double_t *p = positionstack;
616 for (Int_t j=0; j<=index2-index1;j++){
617 fTempInfo->fStackIndex++;
618 fTempInfo->SetHit(p[3*j+0],p[3*j+1],p[3*j+2],qstack[j]);
619 }
620 delete []positionstack;
621 delete []qstack;
622}
623
624
625void AliTPCTrackHitsV2::AddVolume(Int_t volume)
626{
627 //
628 //add volumes to tthe list of volumes
53110b9d 629 //
f641f6bd 630 Int_t * volumes = new Int_t[fNVolumes+1];
f55b9162 631 if (fVolumes) memcpy(volumes,fVolumes,(fNVolumes)*sizeof(Int_t));
f641f6bd 632 volumes[fNVolumes]=volume;
633 fNVolumes++;
634 if (fVolumes) delete []fVolumes;
635 fVolumes = volumes;
636}
637
638
f641f6bd 639Bool_t AliTPCTrackHitsV2::First()
640{
641 //
642 //set Current hit for the first hit
643 //
6e493dba 644
645 if (fArray->GetSize()<=0) {
646 fCurrentHit->fStatus = kFALSE;
647 return kFALSE;
648 }
649
f641f6bd 650 AliTrackHitsParamV2 *param = (AliTrackHitsParamV2 *)fArray->At(0);
fce0283f 651 if (!fHit) fHit = new AliTPChit;
f641f6bd 652 if (!(param) ) {
653 fCurrentHit->fStatus = kFALSE;
654 return kFALSE;
655 }
656 //
657 fCurrentHit->fParamIndex = 0;
658 fCurrentHit->fStackIndex = 0;
659 //
fce0283f 660 //
661 ((AliTPChit*)fHit)->fSector = param->fVolumeID;
662 ((AliTPChit*)fHit)->SetTrack(param->fTrackID);
663 ((AliTPChit*)fHit)->SetX(param->fR*TMath::Cos(param->fFi));
664 ((AliTPChit*)fHit)->SetY(param->fR*TMath::Sin(param->fFi));
665 ((AliTPChit*)fHit)->SetZ(param->fZ);
666 ((AliTPChit*)fHit)->fQ = param->fCharge[0];
667 /*
668 fCurrentHit->fHit.fSector = param->fVolumeID;
669 fCurrentHit->fHit.SetTrack(param->fTrackID);
670 fCurrentHit->fHit.SetX(param->fR*TMath::Cos(param->fFi));
671 fCurrentHit->fHit.SetY(param->fR*TMath::Sin(param->fFi));
672 fCurrentHit->fHit.SetZ(param->fZ);
673 fCurrentHit->fHit.fQ = param->fCharge[0];
674 */
f641f6bd 675 fCurrentHit->fR = param->fR;
676
677 return fCurrentHit->fStatus = kTRUE;
678}
679
680Bool_t AliTPCTrackHitsV2::Next()
681{
682 //
53110b9d 683 // Hit iterator
684 //
f641f6bd 685 if (!(fCurrentHit->fStatus))
686 return kFALSE;
687
688 fCurrentHit->fStackIndex++;
689
690 AliTrackHitsParamV2 *param = (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex);
691 if (fCurrentHit->fStackIndex>=param->fNHits){
692 fCurrentHit->fParamIndex++;
693 if (fCurrentHit->fParamIndex>=fArray->GetEntriesFast()){
694 fCurrentHit->fStatus=kFALSE;
695 return kFALSE;
696 }
697 param = (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex);
698 fCurrentHit->fStackIndex=0;
699 fCurrentHit->fR = param->fR;
700 }
701
702
703
704 Double_t ratio;
705 {
706 // Double_t dfi2 = param->fAn+2*param->fAd*(fCurrentHit->fR-param->fR);
707 Double_t dfi2 = param->fAn;
708 dfi2*=dfi2*fCurrentHit->fR*fCurrentHit->fR;
709 // Double_t ddz2 = param->fTheta+2*param->fThetaD*(fCurrentHit->fR-param->fR);
710 Double_t ddz2 = param->fTheta;
711 ddz2*=ddz2;
712 ratio = TMath::Sqrt(1.+ dfi2+ ddz2);
713 }
714
715 fCurrentHit->fR += fStep*param->fHitDistance[fCurrentHit->fStackIndex]/ratio;
716
717 Double_t dR = fCurrentHit->fR - param->fR;
718 Double_t fi = param->fFi + (param->fAn*dR+param->fAd*dR*dR);
719 Double_t z = param->fZ + (param->fTheta*dR+param->fThetaD*dR*dR);
fce0283f 720 /*
f641f6bd 721 fCurrentHit->fHit.fQ = param->fCharge[fCurrentHit->fStackIndex];
722 fCurrentHit->fHit.SetX(fCurrentHit->fR*TMath::Cos(fi));
723 fCurrentHit->fHit.SetY(fCurrentHit->fR*TMath::Sin(fi));
724 fCurrentHit->fHit.SetZ(z);
725 fCurrentHit->fHit.fSector = param->fVolumeID;
726 fCurrentHit->fHit.SetTrack(param->fTrackID);
fce0283f 727 */
728 ((AliTPChit*)fHit)->fQ = param->fCharge[fCurrentHit->fStackIndex];
729 ((AliTPChit*)fHit)->SetX(fCurrentHit->fR*TMath::Cos(fi));
730 ((AliTPChit*)fHit)->SetY(fCurrentHit->fR*TMath::Sin(fi));
731 ((AliTPChit*)fHit)->SetZ(z);
732 ((AliTPChit*)fHit)->fSector = param->fVolumeID;
733 ((AliTPChit*)fHit)->SetTrack(param->fTrackID);
734
f641f6bd 735 return kTRUE;
736}
737
53110b9d 738AliHit * AliTPCTrackHitsV2::GetHit() const
f641f6bd 739{
53110b9d 740 //
741 // Return one hit
f641f6bd 742 //
fce0283f 743 return (fCurrentHit->fStatus)? fHit:0;
f641f6bd 744 //return &fCurrentHit->fHit;
745
746}
747
748AliTrackHitsParamV2 * AliTPCTrackHitsV2::GetParam()
749{
53110b9d 750 //
751 // Return current parameters
752 //
753 return (fCurrentHit->fStatus)?
754 (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex):0;
f641f6bd 755}
53110b9d 756