]> git.uio.no Git - u/mrichter/AliRoot.git/blob - TPC/AliTPCTrackHitsV2.cxx
Density effect added (Yu.Belikov)
[u/mrichter/AliRoot.git] / TPC / AliTPCTrackHitsV2.cxx
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
16 /* $Id$ */
17
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 ///////////////////////////////////////////////////////////////////////////////
54 //
55
56 #include <stdlib.h>
57 //#include "TVector3.h"
58 #include "AliTPCTrackHitsV2.h"
59
60 #include "TClonesArray.h"    
61 #include "AliTPC.h"
62
63 #include <Riostream.h>
64
65
66
67 ClassImp(AliTPCTrackHitsV2) 
68 ClassImp(AliTrackHitsParamV2)  
69
70   //
71 Int_t AliTrackHitsParamV2::fgCounter1 =0;
72 Int_t AliTrackHitsParamV2::fgCounter2 =0;
73 //
74 Int_t AliTPCTrackHitsV2::fgCounter1 =0;
75 Int_t AliTPCTrackHitsV2::fgCounter2 =0;
76 //
77 const Double_t AliTPCTrackHitsV2::fgkPrecision=1e-6;  //precision 
78 const Double_t AliTPCTrackHitsV2::fgkPrecision2=1e-20;  //precision
79
80
81
82
83 struct  AliTPCTempHitInfoV2 {
84   enum    { kStackSize = 10000};
85   AliTPCTempHitInfoV2();   
86   AliTPCTempHitInfoV2(const AliTPCTempHitInfoV2 &hit)
87     {hit.Copy(*this);}
88   AliTPCTempHitInfoV2& operator = (const AliTPCTempHitInfoV2 &hit)
89      {hit.Copy(*this); return (*this);}
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);
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
109   Double_t fOldR;     //previos r
110   Double_t fPositionStack[3*kStackSize];  //position stack 
111   UInt_t   fQStack[kStackSize];           //Q stack
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
117 private:
118   void Copy(AliTPCTempHitInfoV2 &) const
119   {printf("Not Implemented\n"); exit(1);}
120 };
121
122
123 AliTPCTempHitInfoV2::AliTPCTempHitInfoV2()
124 {
125   //
126   // Standard constructor
127   // set to default value
128   //
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
138 void 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
143   //
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
159 void 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
207 void   AliTPCTempHitInfoV2::UpdateParam(Double_t maxdelta)
208 {
209   //
210   // recalc parameters not fixing origin point
211   //
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 }
232
233 void   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 {
238   //
239   // fit of second order
240   //
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
262 void   AliTPCTempHitInfoV2::Fit(AliTrackHitsParamV2 * param)
263 {
264   //
265   // fit fixing first and the last point 
266   // result stored in new param
267   //
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
283 AliTrackHitsParamV2::AliTrackHitsParamV2()
284 {
285   //
286   // default constructor
287   //
288   fgCounter1++;
289   fgCounter2++;
290   fHitDistance=0;
291   fCharge=0;
292   fNHits=0;
293 }
294
295 AliTrackHitsParamV2::~AliTrackHitsParamV2()
296 {
297   //
298   // Standard destructor
299   //
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
312 AliTPCTrackHitsV2::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;
333   fHit =0;
334   fgCounter1++;
335   fgCounter2++;
336
337
338
339 AliTPCTrackHitsV2::~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   }
357   if (fHit){
358     delete fHit;
359     fHit=0;
360   }
361   fgCounter1--;
362 }
363
364 void AliTPCTrackHitsV2::Clear()
365 {
366   //
367   // clear object  
368   //
369   fSize = 0;
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; 
379     delete fHit;
380     fHit =0;
381     fTempInfo =0;
382   } 
383   if (fVolumes){
384     delete [] fVolumes;
385     fVolumes=0;
386     fNVolumes=0;
387   }
388 }
389
390
391 void AliTPCTrackHitsV2::AddHitKartez(Int_t volumeID, Int_t trackID, Double_t x, 
392               Double_t y, Double_t z,Int_t q)
393 {
394   //
395   // add hit to the container - it add hit at the end - input in global coordinata
396   //
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
404 void AliTPCTrackHitsV2::AddHit(Int_t volumeID, Int_t trackID, 
405                              Double_t r, Double_t z, Double_t fi, Int_t q)
406 {
407   //
408   // Adding one hit
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   }
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   //
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) ) ||  
477        (fTempInfo->fStackIndex+4>fTempInfo->kStackSize) || 
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
501 Bool_t AliTPCTrackHitsV2::FlushHitStack(Bool_t force)
502 {
503   //
504   // write fHitsPosAndQ information from the stack to te arrays
505   //
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
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
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
607 void AliTPCTrackHitsV2::FlushHitStack2(Int_t index1, Int_t index2)
608 {
609   //
610   // second iteration flush stack
611   // call only for hits where first iteration were not succesfully interpolated
612   //
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
628 void AliTPCTrackHitsV2::AddVolume(Int_t volume)
629 {
630   //
631   //add volumes to tthe list of volumes
632   //
633   Int_t * volumes = new Int_t[fNVolumes+1];
634   if (fVolumes) memcpy(volumes,fVolumes,(fNVolumes)*sizeof(Int_t));
635   volumes[fNVolumes]=volume;
636   fNVolumes++;
637   if (fVolumes) delete []fVolumes;
638   fVolumes = volumes;  
639 }
640
641
642 Bool_t AliTPCTrackHitsV2::First()
643 {
644   //
645   //set Current hit for the first hit
646   //
647
648   if (fArray->GetSize()<=0) {
649     fCurrentHit->fStatus = kFALSE;
650     return kFALSE;
651   }
652
653   AliTrackHitsParamV2 *param = (AliTrackHitsParamV2 *)fArray->At(0);
654   if (!fHit) fHit = new AliTPChit;
655   if (!(param) ) {
656     fCurrentHit->fStatus = kFALSE;
657     return kFALSE;
658   }
659   //
660   fCurrentHit->fParamIndex = 0;
661   fCurrentHit->fStackIndex = 0;
662   //
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   */
678   fCurrentHit->fR = param->fR;
679   
680   return fCurrentHit->fStatus = kTRUE;
681 }
682
683 Bool_t AliTPCTrackHitsV2::Next()
684 {
685   //
686   // Hit iterator  
687   //
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);
723   /*
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);
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
738   return kTRUE;
739 }
740   
741 AliHit * AliTPCTrackHitsV2::GetHit() const
742 {
743   //
744   // Return one hit
745   //
746    return (fCurrentHit->fStatus)? fHit:0;
747   //return &fCurrentHit->fHit;
748
749
750  
751 AliTrackHitsParamV2 * AliTPCTrackHitsV2::GetParam()
752 {
753   //
754   // Return current parameters
755   //
756   return (fCurrentHit->fStatus)? 
757     (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex):0;
758 }
759