ITS/AliITStrackerV2.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 //               Implementation of the ITS tracker class
  18 //
  19 //          Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
  20 //-------------------------------------------------------------------------
  21 #include <TFile.h>
  22 #include <TTree.h>
  23 #include <TRandom.h>
  24 #include <iostream.h>
  25
  26 #include "AliITSgeom.h"
  27 #include "AliITSRecPoint.h"
  28 #include "AliTPCtrack.h"
  29 #include "AliITSclusterV2.h"
  30 #include "AliITStrackerV2.h"
  31
  32 //#define DEBUG
  33
  34 #ifdef DEBUG
  35 Int_t LAB=7;
  36 #endif
  37
  38 AliITStrackerV2::AliITSlayer AliITStrackerV2::fLayers[kMaxLayer]; // ITS layers
  39
  40 AliITStrackerV2::
  41 AliITStrackerV2(const AliITSgeom *geom, Int_t eventn) throw (const Char_t *) :
  42 AliTracker() {
  43   //--------------------------------------------------------------------
  44   //This is the AliITStracker constructor
  45   //It also reads clusters from a root file
  46   //--------------------------------------------------------------------
  47   fEventN=eventn;  //MI change add event number  - used to generate identifier
  48
  49   AliITSgeom *g=(AliITSgeom*)geom;
  50
  51   Float_t x,y,z;
  52   Int_t i;
  53   for (i=1; i<kMaxLayer+1; i++) {
  54     Int_t nlad=g->GetNladders(i);
  55     Int_t ndet=g->GetNdetectors(i);
  56
  57     g->GetTrans(i,1,1,x,y,z);
  58     Double_t r=TMath::Sqrt(x*x + y*y);
  59     Double_t poff=TMath::ATan2(y,x);
  60     Double_t zoff=z;
  61
  62     g->GetTrans(i,1,2,x,y,z);
  63     r += TMath::Sqrt(x*x + y*y);
  64     g->GetTrans(i,2,1,x,y,z);
  65     r += TMath::Sqrt(x*x + y*y);
  66     g->GetTrans(i,2,2,x,y,z);
  67     r += TMath::Sqrt(x*x + y*y);
  68     r*=0.25;
  69
  70     new (fLayers+i-1) AliITSlayer(r,poff,zoff,nlad,ndet);
  71
  72     for (Int_t j=1; j<nlad+1; j++) {
  73       for (Int_t k=1; k<ndet+1; k++) { //Fill this layer with detectors
  74         Float_t x,y,zshift; g->GetTrans(i,j,k,x,y,zshift);
  75         Double_t rot[9]; g->GetRotMatrix(i,j,k,rot);
  76
  77         Double_t r =-x*rot[1] + y*rot[0];         if (i==1) r=-r;
  78         Double_t phi=TMath::ATan2(rot[1],rot[0]); if (i==1) phi-=3.1415927;
  79         phi+=0.5*TMath::Pi();
  80         AliITSdetector &det=fLayers[i-1].GetDetector((j-1)*ndet + k-1);
  81
  82 if (phi<0) phi += 2*TMath::Pi();
  83
  84         new(&det) AliITSdetector(r,phi);
  85       }
  86     }
  87
  88   }
  89
  90   try {
  91      //Read clusters
  92      //MI change
  93      char   cname[100];
  94      sprintf(cname,"TreeC_ITS_%d",eventn);
  95      TTree *cTree=(TTree*)gDirectory->Get(cname);
  96
  97      if (!cTree) throw
  98         ("AliITStrackerV2::AliITStrackerV2 can't get cTree !\n");
  99
 100      TBranch *branch=cTree->GetBranch("Clusters");
 101      if (!branch) throw
 102         ("AliITStrackerV2::AliITStrackerV2 can't get Clusters branch !\n");
 103
 104      TClonesArray dummy("AliITSclusterV2",10000), *clusters=&dummy;
 105      branch->SetAddress(&clusters);
 106
 107      Int_t nentr=(Int_t)cTree->GetEntries();
 108      for (i=0; i<nentr; i++) {
 109        if (!cTree->GetEvent(i)) continue;
 110        Int_t lay,lad,det; g->GetModuleId(i,lay,lad,det);
 111        Int_t ncl=clusters->GetEntriesFast();
 112        while (ncl--) {
 113          AliITSclusterV2 *c=(AliITSclusterV2*)clusters->UncheckedAt(ncl);
 114
 115 #ifdef DEBUG
 116 if (c->GetLabel(2)!=LAB)
 117 if (c->GetLabel(1)!=LAB)
 118 if (c->GetLabel(0)!=LAB) continue;
 119 cout<<lay-1<<' '<<lad-1<<' '<<det-1<<' '<<c->GetY()<<' '<<c->GetZ()<<endl;
 120 #endif
 121
 122          fLayers[lay-1].InsertCluster(new AliITSclusterV2(*c));
 123        }
 124        clusters->Delete();
 125      }
 126      delete cTree; //Thanks to Mariana Bondila
 127   }
 128   catch (const Char_t *msg) {
 129     cerr<<msg<<endl;
 130     throw;
 131   }
 132
 133   fI=kMaxLayer;
 134
 135   fPass=0;
 136   fConstraint[0]=1; fConstraint[1]=0;
 137 }
 138
 139 #ifdef DEBUG
 140 static Int_t lbl;
 141 #endif
 142
 143 Int_t AliITStrackerV2::Clusters2Tracks(const TFile *inp, TFile *out) {
 144   //--------------------------------------------------------------------
 145   //This functions reconstructs ITS tracks
 146   //--------------------------------------------------------------------
 147   TFile *in=(TFile*)inp;
 148   TDirectory *savedir=gDirectory;
 149
 150   if (!in->IsOpen()) {
 151      cerr<<"AliITStrackerV2::Clusters2Tracks(): ";
 152      cerr<<"file with TPC tracks is not open !\n";
 153      return 1;
 154   }
 155
 156   if (!out->IsOpen()) {
 157      cerr<<"AliITStrackerV2::Clusters2Tracks(): ";
 158      cerr<<"file for ITS tracks is not open !\n";
 159      return 2;
 160   }
 161
 162   in->cd();
 163
 164   char   tname[100];
 165   Int_t nentr=0; TObjArray itsTracks(10000);
 166
 167   {/* Read TPC tracks */
 168     sprintf(tname,"TreeT_TPC_%d",fEventN);
 169     TTree *tpcTree=(TTree*)in->Get(tname);
 170     if (!tpcTree) {
 171        cerr<<"AliITStrackerV2::Clusters2Tracks(): "
 172              "can't get a tree with TPC tracks !\n";
 173        return 3;
 174     }
 175     AliTPCtrack *itrack=new AliTPCtrack;
 176     tpcTree->SetBranchAddress("tracks",&itrack);
 177     nentr=(Int_t)tpcTree->GetEntries();
 178     for (Int_t i=0; i<nentr; i++) {
 179        tpcTree->GetEvent(i);
 180        itsTracks.AddLast(new AliITStrackV2(*itrack));
 181     }
 182     delete tpcTree; //Thanks to Mariana Bondila
 183     delete itrack;
 184   }
 185   itsTracks.Sort();
 186
 187   out->cd();
 188
 189   sprintf(tname,"TreeT_ITS_%d",fEventN);
 190   TTree itsTree(tname,"Tree with ITS tracks");
 191   AliITStrackV2 *otrack=&fBestTrack;
 192   itsTree.Branch("tracks","AliITStrackV2",&otrack,32000,0);
 193
 194   for (fPass=0; fPass<2; fPass++) {
 195      Int_t &constraint=fConstraint[fPass]; if (constraint<0) continue;
 196      for (Int_t i=0; i<nentr; i++) {
 197        if (i%10==0) cerr<<nentr-i<<" \r";
 198        AliITStrackV2 *t=(AliITStrackV2*)itsTracks.UncheckedAt(i);
 199        if (t==0) continue;           //this track has been already tracked
 200        Int_t tpcLabel=t->GetLabel(); //save the TPC track label
 201 #ifdef DEBUG
 202 lbl=tpcLabel;
 203 if (TMath::Abs(tpcLabel)!=LAB) continue;
 204 cout<<tpcLabel<<" *****************\n";
 205 #endif
 206        try {
 207          ResetTrackToFollow(*t);
 208        } catch (const Char_t *msg) {
 209          cerr<<msg<<endl;
 210          continue;
 211        }
 212        ResetBestTrack();
 213
 214        Double_t r2=fTrackToFollow.GetX()*fTrackToFollow.GetX();
 215        Double_t x0=0.082/21.82*2.33*(45*45+40*40)/r2+2.000/41.28*0.03*75*75/r2;
 216        if (constraint) fTrackToFollow.Improve(x0,GetY(),GetZ());
 217
 218        Double_t xk=80.;
 219        fTrackToFollow.PropagateTo(xk,0.,0.); //Ne if it's still there
 220
 221        xk-=0.005;
 222        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71); //Tedlar
 223        xk-=0.02;
 224        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);   //Kevlar
 225           xk-=2.0;
 226           fTrackToFollow.PropagateTo(xk, 2.0/41.28*0.029, 2.0*0.029);//Nomex
 227        xk-=0.02;
 228        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);   //Kevlar
 229        xk-=0.005;
 230        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71); //Tedlar
 231
 232        xk=61.;
 233        fTrackToFollow.PropagateTo(xk,0.,0.); //C02
 234
 235        xk -=0.005;
 236        fTrackToFollow.PropagateTo(xk, 0.005/24.01*2.7, 0.005*2.7);    //Al
 237        xk -=0.005;
 238        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71);  //Tedlar
 239        xk -=0.02;
 240        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);    //Kevlar
 241           xk -=0.5;
 242           fTrackToFollow.PropagateTo(xk, 0.5/41.28*.029, 0.5*0.029);  //Nomex
 243        xk -=0.02;
 244        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);    //Kevlar
 245        xk -=0.005;
 246        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71);  //Tedlar
 247        xk -=0.005;
 248        fTrackToFollow.PropagateTo(xk, 0.005/24.01*2.7, 0.005*2.7);    //Al
 249
 250
 251        for (FollowProlongation(); fI<kMaxLayer; fI++) {
 252           while (TakeNextProlongation()) FollowProlongation();
 253        }
 254
 255 #ifdef DEBUG
 256 cout<<fBestTrack.GetNumberOfClusters()<<" number of clusters\n\n";
 257 #endif
 258
 259        if (fBestTrack.GetNumberOfClusters() >= kMaxLayer-kLayersToSkip) {
 260           fBestTrack.SetLabel(tpcLabel);
 261           fBestTrack.CookdEdx();
 262           CookLabel(&fBestTrack,0.); //For comparison only
 263           itsTree.Fill();
 264           UseClusters(&fBestTrack);
 265           delete itsTracks.RemoveAt(i);
 266        }
 267
 268      }
 269   }
 270
 271   itsTree.Write();
 272
 273   itsTracks.Delete();
 274   savedir->cd();
 275   cerr<<"Number of TPC tracks: "<<nentr<<endl;
 276   cerr<<"Number of prolonged tracks: "<<itsTree.GetEntries()<<endl;
 277
 278   return 0;
 279 }
 280
 281 Int_t AliITStrackerV2::PropagateBack(const TFile *inp, TFile *out) {
 282   //--------------------------------------------------------------------
 283   //This functions propagates reconstructed ITS tracks back
 284   //--------------------------------------------------------------------
 285   TFile *in=(TFile*)inp;
 286   TDirectory *savedir=gDirectory;
 287
 288   if (!in->IsOpen()) {
 289      cerr<<"AliITStrackerV2::PropagateBack(): ";
 290      cerr<<"file with ITS tracks is not open !\n";
 291      return 1;
 292   }
 293
 294   if (!out->IsOpen()) {
 295      cerr<<"AliITStrackerV2::PropagateBack(): ";
 296      cerr<<"file for back propagated ITS tracks is not open !\n";
 297      return 2;
 298   }
 299
 300   in->cd();
 301   TTree *itsTree=(TTree*)in->Get("TreeT_ITS_0");
 302   if (!itsTree) {
 303      cerr<<"AliITStrackerV2::PropagateBack() ";
 304      cerr<<"can't get a tree with ITS tracks !\n";
 305      return 3;
 306   }
 307   AliITStrackV2 *itrack=new AliITStrackV2;
 308   itsTree->SetBranchAddress("tracks",&itrack);
 309
 310   out->cd();
 311   TTree backTree("TreeT_ITSb_0","Tree with back propagated ITS tracks");
 312   AliTPCtrack *otrack=0;
 313   backTree.Branch("tracks","AliTPCtrack",&otrack,32000,0);
 314
 315   Int_t ntrk=0;
 316
 317   Int_t nentr=(Int_t)itsTree->GetEntries();
 318   for (Int_t i=0; i<nentr; i++) {
 319     itsTree->GetEvent(i);
 320     ResetTrackToFollow(*itrack);
 321     fTrackToFollow.ResetCovariance(); fTrackToFollow.ResetClusters();
 322     Int_t itsLabel=fTrackToFollow.GetLabel(); //save the ITS track label
 323
 324 #ifdef DEBUG
 325 if (TMath::Abs(itsLabel)!=LAB) continue;
 326 cout<<itsLabel<<" *****************\n";
 327 #endif
 328
 329     try {
 330        Int_t nc=itrack->GetNumberOfClusters();
 331 #ifdef DEBUG
 332 for (Int_t k=0; k<nc; k++) {
 333     Int_t index=itrack->GetClusterIndex(k);
 334     AliITSclusterV2 *c=(AliITSclusterV2*)GetCluster(index);
 335     cout<<c->GetLabel(0)<<' '<<c->GetLabel(1)<<' '<<c->GetLabel(2)<<endl;
 336 }
 337 #endif
 338        Int_t idx=0, l=0;
 339        const  AliITSclusterV2 *c=0;
 340        if (nc--) {
 341           idx=itrack->GetClusterIndex(nc); l=(idx&0xf0000000)>>28;
 342           c=(AliITSclusterV2*)GetCluster(idx);
 343        }
 344        for (fI=0; fI<kMaxLayer; fI++) {
 345          AliITSlayer &layer=fLayers[fI];
 346          Double_t r=layer.GetR();
 347          if (fI==2 || fI==4) {
 348             Double_t rs=0.5*(fLayers[fI-1].GetR() + r);
 349             Double_t ds=0.034; if (fI==4) ds=0.039;
 350             Double_t dx0r=ds/21.82*2.33, dr=ds*2.33;
 351             if (!fTrackToFollow.PropagateTo(rs,1*dx0r,dr)) throw "";
 352          }
 353
 354          Double_t x,y,z;
 355          if (!fTrackToFollow.GetGlobalXYZat(r,x,y,z))
 356             throw "AliITStrackerV2::PropagateBack: failed to estimate track !";
 357          Double_t phi=TMath::ATan2(y,x);
 358          Double_t d=layer.GetThickness(phi,z);
 359          Int_t idet=layer.FindDetectorIndex(phi,z);
 360          if (idet<0)
 361          throw "AliITStrackerV2::PropagateBack: failed to find a detector !\n";
 362          const AliITSdetector &det=layer.GetDetector(idet);
 363          r=det.GetR(); phi=det.GetPhi();
 364          if (!fTrackToFollow.Propagate(phi,r,d/21.82*2.33,d*2.33)) throw "";
 365
 366          const AliITSclusterV2 *cl=0;
 367          Int_t index=0;
 368          Double_t maxchi2=kMaxChi2;
 369
 370          if (l==fI) {
 371            if (idet != c->GetDetectorIndex()) {
 372              idet=c->GetDetectorIndex();
 373              const AliITSdetector &det=layer.GetDetector(idet);
 374              r=det.GetR(); phi=det.GetPhi();
 375              if (!fTrackToFollow.Propagate(phi,r,0.,0.)) throw "";
 376            }
 377            Double_t chi2=fTrackToFollow.GetPredictedChi2(c);
 378            if (chi2<kMaxChi2) {
 379               cl=c; maxchi2=chi2; index=idx;
 380            }
 381            if (nc--) {
 382               idx=itrack->GetClusterIndex(nc); l=(idx&0xf0000000)>>28;
 383               c=(AliITSclusterV2*)GetCluster(idx);
 384            }
 385          }
 386
 387          if (fTrackToFollow.GetNumberOfClusters()>2) {
 388            Double_t dz=3*TMath::Sqrt(fTrackToFollow.GetSigmaZ2()+kSigmaZ2[fI]);
 389            Double_t dy=3*TMath::Sqrt(fTrackToFollow.GetSigmaY2()+kSigmaY2[fI]);
 390            Double_t zmin=fTrackToFollow.GetZ() - dz;
 391            Double_t zmax=fTrackToFollow.GetZ() + dz;
 392            Double_t ymin=fTrackToFollow.GetY() + phi*r - dy;
 393            Double_t ymax=fTrackToFollow.GetY() + phi*r + dy;
 394            layer.SelectClusters(zmin,zmax,ymin,ymax);
 395
 396            const AliITSclusterV2 *cc=0; Int_t ci;
 397            while ((cc=layer.GetNextCluster(ci))!=0) {
 398               if (idet != cc->GetDetectorIndex()) continue;
 399               Double_t chi2=fTrackToFollow.GetPredictedChi2(cc);
 400               if (chi2<maxchi2) {
 401                  cl=cc; index=(fI<<28)+ci; maxchi2=chi2;
 402               }
 403            }
 404          }
 405
 406          if (cl) {
 407             if (!fTrackToFollow.Update(cl,maxchi2,index))
 408               cerr<<"AliITStrackerV2::PropagateBack: filtering failed !\n";
 409               continue;
 410          }
 411        }
 412
 413        Double_t xk=61.;
 414        fTrackToFollow.PropagateTo(xk,0.,0.); //Air
 415
 416        xk +=0.005;
 417        fTrackToFollow.PropagateTo(xk, 0.005/24.01*2.7, 0.005*2.7);    //Al
 418        xk +=0.005;
 419        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71);  //Tedlar
 420        xk +=0.02;
 421        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);    //Kevlar
 422           xk +=0.5;
 423           fTrackToFollow.PropagateTo(xk, 0.5/41.28*.029, 0.5*0.029);  //Nomex
 424        xk +=0.02;
 425        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);    //Kevlar
 426        xk +=0.005;
 427        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71);  //Tedlar
 428        xk +=0.005;
 429        fTrackToFollow.PropagateTo(xk, 0.005/24.01*2.7, 0.005*2.7);    //Al
 430
 431        xk=80.;
 432        fTrackToFollow.PropagateTo(xk,0.,0.); //CO2
 433
 434        xk+=0.005;
 435        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71); //Tedlar
 436        xk+=0.02;
 437        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);   //Kevlar
 438           xk+=2.0;
 439           fTrackToFollow.PropagateTo(xk, 2.0/41.28*0.029, 2.0*0.029);//Nomex
 440        xk+=0.02;
 441        fTrackToFollow.PropagateTo(xk, 0.02/44.86*1.45, 0.02*1.45);   //Kevlar
 442        xk+=0.005;
 443        fTrackToFollow.PropagateTo(xk, 0.005/44.77*1.71, 0.005*1.71); //Tedlar
 444
 445        fTrackToFollow.SetLabel(itsLabel);
 446        otrack=new AliTPCtrack(fTrackToFollow,fTrackToFollow.GetAlpha());
 447        backTree.Fill(); delete otrack;
 448        UseClusters(&fTrackToFollow);
 449        cerr << ++ntrk << "                \r";
 450     }
 451     catch (const Char_t *msg) {
 452        cerr<<msg<<endl;
 453     }
 454   }
 455
 456   backTree.Write();
 457   savedir->cd();
 458   cerr<<"Number of ITS tracks: "<<nentr<<endl;
 459   cerr<<"Number of back propagated ITS tracks: "<<ntrk<<endl;
 460
 461   delete itrack;
 462
 463   delete itsTree; //Thanks to Mariana Bondila
 464
 465   return 0;
 466 }
 467
 468 AliCluster *AliITStrackerV2::GetCluster(Int_t index) const {
 469   //--------------------------------------------------------------------
 470   //       Return pointer to a given cluster
 471   //--------------------------------------------------------------------
 472   Int_t l=(index & 0xf0000000) >> 28;
 473   Int_t c=(index & 0x0fffffff) >> 00;
 474   return fLayers[l].GetCluster(c);
 475 }
 476
 477
 478 void AliITStrackerV2::FollowProlongation() {
 479   //--------------------------------------------------------------------
 480   //This function finds a track prolongation
 481   //--------------------------------------------------------------------
 482   Int_t tryAgain=kLayersToSkip;
 483
 484   while (fI) {
 485     fI--;
 486 #ifdef DEBUG
 487 cout<<fI<<' ';
 488 #endif
 489     AliITSlayer &layer=fLayers[fI];
 490     AliITStrackV2 &track=fTracks[fI];
 491
 492     Double_t r=layer.GetR();
 493     if (fI==3 || fI==1) {
 494       Double_t rs=0.5*(fLayers[fI+1].GetR() + r);
 495       Double_t ds=0.034; if (fI==3) ds=0.039;
 496       Double_t dx0r=ds/21.82*2.33, dr=ds*2.33;
 497       fTrackToFollow.Propagate(fTrackToFollow.GetAlpha(),rs,dx0r,dr);
 498     }
 499
 500     //find intersection
 501     Double_t x,y,z;
 502     if (!fTrackToFollow.GetGlobalXYZat(r,x,y,z)) {
 503       //cerr<<"AliITStrackerV2::FollowProlongation: "
 504       //"failed to estimate track !\n";
 505      break;
 506     }
 507     Double_t phi=TMath::ATan2(y,x);
 508     Double_t d=layer.GetThickness(phi,z);
 509     Int_t idet=layer.FindDetectorIndex(phi,z);
 510     if (idet<0) {
 511       //cerr<<"AliITStrackerV2::FollowProlongation: "
 512       //"failed to find a detector !\n";
 513       break;
 514     }
 515
 516     //propagate to the intersection
 517     const AliITSdetector &det=layer.GetDetector(idet);
 518     phi=det.GetPhi();
 519     if (!fTrackToFollow.Propagate(phi,det.GetR(),d/21.82*2.33,d*2.33)) {
 520       //cerr<<"AliITStrackerV2::FollowProlongation: propagation failed !\n";
 521       break;
 522     }
 523     if (TMath::Abs(fTrackToFollow.GetZ()-GetZ()) > r+1) break;
 524     fTrackToFollow.SetDetectorIndex(idet);
 525
 526     //Select possible prolongations and store the current track estimation
 527     track.~AliITStrackV2(); new(&track) AliITStrackV2(fTrackToFollow);
 528     Double_t dz=3*TMath::Sqrt(track.GetSigmaZ2() + kSigmaZ2[fI]);
 529     if (dz < 0.5*TMath::Abs(track.GetTgl())) dz=0.5*TMath::Abs(track.GetTgl());
 530     if (dz > kMaxRoad) {
 531       //cerr<<"AliITStrackerV2::FollowProlongation: too broad road in Z !\n";
 532       break;
 533     }
 534     Double_t dy=4*TMath::Sqrt(track.GetSigmaY2() + kSigmaY2[fI]);
 535     if (dy < 0.5*TMath::Abs(track.GetSnp())) dy=0.5*TMath::Abs(track.GetSnp());
 536     if (dy > kMaxRoad) {
 537       //cerr<<"AliITStrackerV2::FollowProlongation: too broad road in Y !\n";
 538       break;
 539     }
 540     Double_t zmin=track.GetZ() - dz;
 541     Double_t zmax=track.GetZ() + dz;
 542     Double_t ymin=track.GetY() + r*phi - dy;
 543     Double_t ymax=track.GetY() + r*phi + dy;
 544     layer.SelectClusters(zmin,zmax,ymin,ymax);
 545
 546     //take another prolongation
 547     if (!TakeNextProlongation()) if (!tryAgain--) break;
 548     tryAgain=kLayersToSkip;
 549
 550   }
 551
 552   //deal with the best track
 553   Int_t ncl=fTrackToFollow.GetNumberOfClusters();
 554   Int_t nclb=fBestTrack.GetNumberOfClusters();
 555   if (ncl)
 556   if (ncl >= nclb) {
 557      Double_t chi2=fTrackToFollow.GetChi2();
 558      if (chi2/ncl < kChi2PerCluster) {
 559         if (ncl > nclb || chi2 < fBestTrack.GetChi2()) {
 560            ResetBestTrack();
 561         }
 562      }
 563   }
 564
 565   if (fI) fI++;
 566 }
 567
 568
 569 Int_t AliITStrackerV2::TakeNextProlongation() {
 570   //--------------------------------------------------------------------
 571   //This function takes another track prolongation
 572   //--------------------------------------------------------------------
 573   //Double_t m[20];
 574   Double_t d=GetEffectiveThickness(0,0); //Think of this !!!!
 575
 576   AliITSlayer &layer=fLayers[fI];
 577   AliITStrackV2 &t=fTracks[fI];
 578   Int_t &constraint=fConstraint[fPass];
 579
 580   Double_t dz=4*TMath::Sqrt(t.GetSigmaZ2() + kSigmaZ2[fI]);
 581   Double_t dy=4*TMath::Sqrt(t.GetSigmaY2() + kSigmaY2[fI]);
 582
 583   const AliITSclusterV2 *c=0; Int_t ci=-1;
 584   Double_t chi2=12345.;
 585   while ((c=layer.GetNextCluster(ci))!=0) {
 586
 587 #ifdef DEBUG
 588 //if (fI==0)
 589 //if (c->GetLabel(0)!=TMath::Abs(lbl)) continue;
 590 #endif
 591
 592     Int_t idet=c->GetDetectorIndex();
 593
 594     if (t.GetDetectorIndex()!=idet) {
 595        const AliITSdetector &det=layer.GetDetector(idet);
 596        if (!t.Propagate(det.GetPhi(),det.GetR(),0.,0.)) {
 597          //cerr<<"AliITStrackerV2::TakeNextProlongation: "
 598          //"propagation failed !\n";
 599          continue;
 600        }
 601        t.SetDetectorIndex(idet);
 602
 603 #ifdef DEBUG
 604 cout<<fI<<" change detector !\n";
 605 #endif
 606
 607     }
 608
 609     if (TMath::Abs(t.GetZ() - c->GetZ()) > dz) continue;
 610     if (TMath::Abs(t.GetY() - c->GetY()) > dy) continue;
 611
 612     //m[0]=fYV; m[1]=fZV;
 613     //chi2=t.GetPredictedChi2(c,m,d/21.82*2.33);
 614     chi2=t.GetPredictedChi2(c);
 615
 616     if (chi2<kMaxChi2) break;
 617   }
 618
 619 #ifdef DEBUG
 620 cout<<fI<<" chi2="<<chi2<<' '<<t.GetY()<<' '<<t.GetZ()<<' '<<dy<<' '<<dz<<endl;
 621 #endif
 622
 623   if (chi2>=kMaxChi2) return 0;
 624   if (!c) return 0;
 625
 626   ResetTrackToFollow(t);
 627
 628   //if (!fTrackToFollow.Update(m,chi2,(fI<<28)+ci)) {
 629   if (!fTrackToFollow.Update(c,chi2,(fI<<28)+ci)) {
 630      //cerr<<"AliITStrackerV2::TakeNextProlongation: filtering failed !\n";
 631      return 0;
 632   }
 633   // b.b. change for the dEdX
 634   fTrackToFollow.SetSampledEdx(c->GetQ(),
 635                                fTrackToFollow.GetNumberOfClusters()-1);
 636
 637   if (constraint) fTrackToFollow.Improve(d/21.82*2.33,GetY(),GetZ());
 638
 639 #ifdef DEBUG
 640 cout<<"accepted lab="<<c->GetLabel(0)<<' '
 641     <<fTrackToFollow.GetNumberOfClusters()<<' '
 642     <<fTrackToFollow.GetY()<<' '<<fTrackToFollow.GetZ()<<endl<<endl;
 643 #endif
 644
 645   return 1;
 646 }
 647
 648
 649
 650
 651 AliITStrackerV2::AliITSlayer::AliITSlayer() {
 652   //--------------------------------------------------------------------
 653   //default AliITSlayer constructor
 654   //--------------------------------------------------------------------
 655   fN=0;
 656   fDetectors=0;
 657 }
 658
 659 AliITStrackerV2::AliITSlayer::
 660 AliITSlayer(Double_t r,Double_t p,Double_t z,Int_t nl,Int_t nd) {
 661   //--------------------------------------------------------------------
 662   //main AliITSlayer constructor
 663   //--------------------------------------------------------------------
 664   fR=r; fPhiOffset=p; fZOffset=z;
 665   fNladders=nl; fNdetectors=nd;
 666   fDetectors=new AliITSdetector[fNladders*fNdetectors];
 667
 668   fN=0;
 669   fI=0;
 670 }
 671
 672 AliITStrackerV2::AliITSlayer::~AliITSlayer() {
 673   //--------------------------------------------------------------------
 674   // AliITSlayer destructor
 675   //--------------------------------------------------------------------
 676   delete[] fDetectors;
 677   for (Int_t i=0; i<fN; i++) delete fClusters[i];
 678 }
 679
 680 Int_t AliITStrackerV2::AliITSlayer::InsertCluster(AliITSclusterV2 *c) {
 681   //--------------------------------------------------------------------
 682   //This function adds a cluster to this layer
 683   //--------------------------------------------------------------------
 684   if (fN==kMaxClusterPerLayer) {
 685  cerr<<"AliITStrackerV2::AliITSlayer::InsertCluster(): Too many clusters !\n";
 686    return 1;
 687   }
 688
 689   if (fN==0) {fClusters[fN++]=c; return 0;}
 690   Int_t i=FindClusterIndex(c->GetZ());
 691   memmove(fClusters+i+1 ,fClusters+i,(fN-i)*sizeof(AliITSclusterV2*));
 692   fClusters[i]=c; fN++;
 693
 694   return 0;
 695 }
 696
 697 Int_t AliITStrackerV2::AliITSlayer::FindClusterIndex(Double_t z) const {
 698   //--------------------------------------------------------------------
 699   // This function returns the index of the nearest cluster
 700   //--------------------------------------------------------------------
 701   if (fN==0) return 0;
 702   if (z <= fClusters[0]->GetZ()) return 0;
 703   if (z > fClusters[fN-1]->GetZ()) return fN;
 704   Int_t b=0, e=fN-1, m=(b+e)/2;
 705   for (; b<e; m=(b+e)/2) {
 706     if (z > fClusters[m]->GetZ()) b=m+1;
 707     else e=m;
 708   }
 709   return m;
 710 }
 711
 712 void AliITStrackerV2::AliITSlayer::
 713 SelectClusters(Double_t zmin,Double_t zmax,Double_t ymin, Double_t ymax) {
 714   //--------------------------------------------------------------------
 715   // This function sets the "window"
 716   //--------------------------------------------------------------------
 717   fI=FindClusterIndex(zmin); fZmax=zmax;
 718   Double_t circle=2*TMath::Pi()*fR;
 719   if (ymax>circle) { ymax-=circle; ymin-=circle; }
 720   fYmin=ymin; fYmax=ymax;
 721 }
 722
 723 const AliITSclusterV2 *AliITStrackerV2::AliITSlayer::GetNextCluster(Int_t &ci){
 724   //--------------------------------------------------------------------
 725   // This function returns clusters within the "window"
 726   //--------------------------------------------------------------------
 727   const AliITSclusterV2 *cluster=0;
 728   for (Int_t i=fI; i<fN; i++) {
 729     const AliITSclusterV2 *c=fClusters[i];
 730     if (c->GetZ() > fZmax) break;
 731     if (c->IsUsed()) continue;
 732     const AliITSdetector &det=GetDetector(c->GetDetectorIndex());
 733     Double_t y=fR*det.GetPhi() + c->GetY();
 734
 735     if (y>2.*fR*TMath::Pi()) y -= 2*fR*TMath::Pi();
 736     if (y>1.*fR*TMath::Pi() && fYmax<y) y -= 2*fR*TMath::Pi();
 737
 738     if (y<fYmin) continue;
 739     if (y>fYmax) continue;
 740     cluster=c; ci=i;
 741     fI=i+1;
 742     break;
 743   }
 744
 745   return cluster;
 746 }
 747
 748 Int_t AliITStrackerV2::AliITSlayer::
 749 FindDetectorIndex(Double_t phi, Double_t z) const {
 750   //--------------------------------------------------------------------
 751   //This function finds the detector crossed by the track
 752   //--------------------------------------------------------------------
 753   Double_t dphi=phi-fPhiOffset;
 754   if      (dphi <  0) dphi += 2*TMath::Pi();
 755   else if (dphi >= 2*TMath::Pi()) dphi -= 2*TMath::Pi();
 756   Int_t np=Int_t(dphi*fNladders*0.5/TMath::Pi()+0.5);
 757
 758   Double_t dz=fZOffset-z;
 759   Int_t nz=Int_t(dz*(fNdetectors-1)*0.5/fZOffset+0.5);
 760
 761   if (np>=fNladders) np-=fNladders;
 762   if (np<0)          np+=fNladders;
 763
 764 #ifdef DEBUG
 765 cout<<np<<' '<<nz<<endl;
 766 #endif
 767
 768   return np*fNdetectors + nz;
 769 }
 770
 771 Double_t
 772 AliITStrackerV2::AliITSlayer::GetThickness(Double_t phi, Double_t z) const {
 773   //--------------------------------------------------------------------
 774   //This function returns the thickness of this layer
 775   //--------------------------------------------------------------------
 776   //-pi<phi<+pi
 777   if (3 <fR&&fR<8 ) return 2.5*0.096;
 778   if (13<fR&&fR<26) return 1.1*0.088;
 779   if (37<fR&&fR<41) return 1.1*0.085;
 780   return 1.1*0.081;
 781 }
 782
 783
 784 Double_t AliITStrackerV2::GetEffectiveThickness(Double_t phi,Double_t z) const
 785 {
 786   //--------------------------------------------------------------------
 787   //Returns the thickness between the current layer and the vertex
 788   //--------------------------------------------------------------------
 789   //-pi<phi<+pi
 790   Double_t d=0.1;
 791
 792   Double_t xn=fLayers[fI].GetR();
 793   for (Int_t i=0; i<fI; i++) {
 794     Double_t xi=fLayers[i].GetR();
 795     d+=fLayers[i].GetThickness(phi,z)*xi*xi;
 796   }
 797
 798   if (fI>1) {
 799     Double_t xi=0.5*(fLayers[1].GetR()+fLayers[2].GetR());
 800     d+=0.034*xi*xi;
 801   }
 802
 803   if (fI>3) {
 804     Double_t xi=0.5*(fLayers[3].GetR()+fLayers[4].GetR());
 805     d+=0.039*xi*xi;
 806   }
 807   return d/(xn*xn);
 808 }
 809
 810
 811
 812 Int_t AliITStrackerV2::AliITSlayer::InRoad() const {
 813   //--------------------------------------------------------------------
 814   // This function returns number of clusters within the "window"
 815   //--------------------------------------------------------------------
 816   Int_t ncl=0;
 817   for (Int_t i=fI; i<fN; i++) {
 818     const AliITSclusterV2 *c=fClusters[i];
 819     if (c->GetZ() > fZmax) break;
 820     //if (c->IsUsed()) continue;
 821     const AliITSdetector &det=GetDetector(c->GetDetectorIndex());
 822     Double_t y=fR*det.GetPhi() + c->GetY();
 823
 824     if (y>2.*fR*TMath::Pi()) y -= 2*fR*TMath::Pi();
 825     if (y>1.*fR*TMath::Pi() && fYmax<y) y -= 2*fR*TMath::Pi();
 826
 827     if (y<fYmin) continue;
 828     if (y>fYmax) continue;
 829     ncl++;
 830   }
 831   return ncl;
 832 }
 833
 834
 835