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