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