Message commented out
[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 "AliTPCTrackHitsV2.h"
57 #include "TClonesArray.h"    
58 #include "AliTPC.h"
59
60
61
62 ClassImp(AliTPCTrackHitsV2) 
63 ClassImp(AliTrackHitsParamV2)  
64
65   //
66 Int_t AliTrackHitsParamV2::fgCounter1 =0;
67 Int_t AliTrackHitsParamV2::fgCounter2 =0;
68 //
69 Int_t AliTPCTrackHitsV2::fgCounter1 =0;
70 Int_t AliTPCTrackHitsV2::fgCounter2 =0;
71 //
72 const Double_t AliTPCTrackHitsV2::fgkPrecision=1e-6;  //precision 
73 const Double_t AliTPCTrackHitsV2::fgkPrecision2=1e-20;  //precision
74
75
76
77
78 struct  AliTPCTempHitInfoV2 {
79   friend class AliTPCTrackHitsV2;
80 protected:
81   enum    { kStackSize = 10000};
82   AliTPCTempHitInfoV2();   
83   AliTPCTempHitInfoV2(const AliTPCTempHitInfoV2 &hit)
84     {hit.Copy(*this);}
85   AliTPCTempHitInfoV2& operator = (const AliTPCTempHitInfoV2 &hit)
86      {hit.Copy(*this); return (*this);}
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);
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
106   Double_t fOldR;     //previos r
107   Double_t fPositionStack[3*kStackSize];  //position stack 
108   UInt_t   fQStack[kStackSize];           //Q stack
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
114 private:
115   void Copy(AliTPCTempHitInfoV2 &) const
116   {printf("Not Implemented\n"); exit(1);}
117 };
118
119
120 AliTPCTempHitInfoV2::AliTPCTempHitInfoV2()
121 {
122   //
123   // Standard constructor
124   // set to default value
125   //
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
135 void 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
140   //
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
156 void 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
204 void   AliTPCTempHitInfoV2::UpdateParam(Double_t maxdelta)
205 {
206   //
207   // recalc parameters not fixing origin point
208   //
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 }
229
230 void   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 {
235   //
236   // fit of second order
237   //
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
259 void   AliTPCTempHitInfoV2::Fit(AliTrackHitsParamV2 * param)
260 {
261   //
262   // fit fixing first and the last point 
263   // result stored in new param
264   //
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
280 AliTrackHitsParamV2::AliTrackHitsParamV2()
281 {
282   //
283   // default constructor
284   //
285   fgCounter1++;
286   fgCounter2++;
287   fHitDistance=0;
288   fCharge=0;
289   fNHits=0;
290 }
291
292 AliTrackHitsParamV2::~AliTrackHitsParamV2()
293 {
294   //
295   // Standard destructor
296   //
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
309 AliTPCTrackHitsV2::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;
330   fHit =0;
331   fgCounter1++;
332   fgCounter2++;
333
334
335
336 AliTPCTrackHitsV2::~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   }
354   if (fHit){
355     delete fHit;
356     fHit=0;
357   }
358   fgCounter1--;
359 }
360
361 void AliTPCTrackHitsV2::Clear()
362 {
363   //
364   // clear object  
365   //
366   fSize = 0;
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; 
376     delete fHit;
377     fHit =0;
378     fTempInfo =0;
379   } 
380   if (fVolumes){
381     delete [] fVolumes;
382     fVolumes=0;
383     fNVolumes=0;
384   }
385 }
386
387
388 void AliTPCTrackHitsV2::AddHitKartez(Int_t volumeID, Int_t trackID, Double_t x, 
389               Double_t y, Double_t z,Int_t q)
390 {
391   //
392   // add hit to the container - it add hit at the end - input in global coordinata
393   //
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
401 void AliTPCTrackHitsV2::AddHit(Int_t volumeID, Int_t trackID, 
402                              Double_t r, Double_t z, Double_t fi, Int_t q)
403 {
404   //
405   // Adding one hit
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   }
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   //
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) ) ||  
474        (fTempInfo->fStackIndex+4>fTempInfo->kStackSize) || 
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
498 Bool_t AliTPCTrackHitsV2::FlushHitStack(Bool_t force)
499 {
500   //
501   // write fHitsPosAndQ information from the stack to te arrays
502   //
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
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
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
604 void AliTPCTrackHitsV2::FlushHitStack2(Int_t index1, Int_t index2)
605 {
606   //
607   // second iteration flush stack
608   // call only for hits where first iteration were not succesfully interpolated
609   //
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
625 void AliTPCTrackHitsV2::AddVolume(Int_t volume)
626 {
627   //
628   //add volumes to tthe list of volumes
629   //
630   Int_t * volumes = new Int_t[fNVolumes+1];
631   if (fVolumes) memcpy(volumes,fVolumes,(fNVolumes)*sizeof(Int_t));
632   volumes[fNVolumes]=volume;
633   fNVolumes++;
634   if (fVolumes) delete []fVolumes;
635   fVolumes = volumes;  
636 }
637
638
639 Bool_t AliTPCTrackHitsV2::First()
640 {
641   //
642   //set Current hit for the first hit
643   //
644
645   if (fArray->GetSize()<=0) {
646     fCurrentHit->fStatus = kFALSE;
647     return kFALSE;
648   }
649
650   AliTrackHitsParamV2 *param = (AliTrackHitsParamV2 *)fArray->At(0);
651   if (!fHit) fHit = new AliTPChit;
652   if (!(param) ) {
653     fCurrentHit->fStatus = kFALSE;
654     return kFALSE;
655   }
656   //
657   fCurrentHit->fParamIndex = 0;
658   fCurrentHit->fStackIndex = 0;
659   //
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   */
675   fCurrentHit->fR = param->fR;
676   
677   return fCurrentHit->fStatus = kTRUE;
678 }
679
680 Bool_t AliTPCTrackHitsV2::Next()
681 {
682   //
683   // Hit iterator  
684   //
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);
720   /*
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);
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
735   return kTRUE;
736 }
737   
738 AliHit * AliTPCTrackHitsV2::GetHit() const
739 {
740   //
741   // Return one hit
742   //
743    return (fCurrentHit->fStatus)? fHit:0;
744   //return &fCurrentHit->fHit;
745
746
747  
748 AliTrackHitsParamV2 * AliTPCTrackHitsV2::GetParam()
749 {
750   //
751   // Return current parameters
752   //
753   return (fCurrentHit->fStatus)? 
754     (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex):0;
755 }
756