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