]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDtrack.cxx
Return the most probable momentum if the magnetic field is too weak and the momentum...
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
index 2aaf276e475bb7c6b0f5dc1182c30b4dd278dfe0..dbb51329bda836129ae2d483983e5f4a6897fb74 100644 (file)
 //      e-mail: Jouri.Belikov@cern.ch
 //-----------------------------------------------------------------
 
-#include "TMath.h"
+#include <TMath.h>
 
 #include "AliESDVertex.h"
 #include "AliESDtrack.h"
 #include "AliKalmanTrack.h"
-#include "AliTrackPointArray.h"
 #include "AliLog.h"
+#include "AliTrackPointArray.h"
 
 ClassImp(AliESDtrack)
 
@@ -76,7 +76,6 @@ AliESDtrack::AliESDtrack() :
   fITSsignal(0),
   fITSLabel(0),
   fITSFakeRatio(0),
-  fITStrack(0),
   fTPCchi2(0),
   fTPCncls(0),
   fTPCnclsF(0),
@@ -92,27 +91,29 @@ AliESDtrack::AliESDtrack() :
   fTRDLabel(0),
   fTRDQuality(0),
   fTRDBudget(0),
-  fTRDtrack(0),
   fTOFchi2(0),
   fTOFindex(0),
   fTOFCalChannel(-1),
   fTOFsignal(-1),
   fTOFsignalToT(0),
-  //  fPHOSsignal(-1),
   fRICHchi2(1e10),
   fRICHncls(0),
   fRICHindex(0),
   fRICHsignal(-1),
-  fRICHtheta(0),
-  fRICHphi(0),
-  fRICHdx(0),
-  fRICHdy(0),
-  fPoints(0)
+  fRICHtheta(-1),
+  fRICHphi(-1),
+  fRICHdx(-1),
+  fRICHdy(-1),
+  fRICHmipX(-1),
+  fRICHmipY(-1),
+  fPoints(0),
+  fFriendTrack(new AliESDfriendTrack())
 {
   //
   // The default ESD constructor 
   //
-  for (Int_t i=0; i<AliPID::kSPECIES; i++) {
+  Int_t i, j;
+  for (i=0; i<AliPID::kSPECIES; i++) {
     fTrackTime[i]=0.;
     fR[i]=1.;
     fITSr[i]=1.;
@@ -122,29 +123,17 @@ AliESDtrack::AliESDtrack() :
     fRICHr[i]=1.;
   }
   
-  //  for (Int_t i=0; i<AliPID::kSPECIESN; i++) {
-  //  fPHOSr[i]  = 1.;
-  // } 
-  //  fPHOSpos[0]=fPHOSpos[1]=fPHOSpos[2]=0.;
-
-  Int_t i;
-  for (i=0; i<12; i++)  fITSchi2MIP[i]=1e10;
-  for (i=0; i<12; i++)  { fITSindex[i]=-1; }
-  for (i=0; i<180; i++) { fTPCindex[i]=0; }
   for (i=0; i<3; i++)   { fKinkIndexes[i]=0;}
   for (i=0; i<3; i++)   { fV0Indexes[i]=-1;}
-  for (i=0; i<180; i++) { fTRDindex[i]=0; }
-  for (i=0;i<kNPlane;i++) {fTRDsignals[i]=0.; fTRDTimBin[i]=-1;}
+  for (i=0;i<kNPlane;i++) {
+    for (j=0;j<kNSlice;j++) {
+      fTRDsignals[i][j]=0.; 
+    }
+    fTRDTimBin[i]=-1;
+  }
   for (i=0;i<4;i++) {fTPCPoints[i]=-1;}
   for (i=0;i<3;i++) {fTOFLabel[i]=-1;}
   for (i=0;i<10;i++) {fTOFInfo[i]=-1;}
-  fTPCLabel = 0;
-  fTRDLabel = 0;
-  fTRDQuality =0;
-  fTRDBudget =0;
-  fITSLabel = 0;
-  fITStrack = 0;
-  fTRDtrack = 0;  
 }
 
 //_______________________________________________________________________
@@ -166,7 +155,6 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fITSsignal(track.fITSsignal),
   fITSLabel(track.fITSLabel),
   fITSFakeRatio(track.fITSFakeRatio),
-  fITStrack(0),    //coping separatelly - in user code
   fTPCchi2(track.fTPCchi2),
   fTPCncls(track.fTPCncls),
   fTPCnclsF(track.fTPCnclsF),
@@ -182,13 +170,11 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fTRDLabel(track.fTRDLabel),
   fTRDQuality(track.fTRDQuality),
   fTRDBudget(track.fTRDBudget),
-  fTRDtrack(0),
   fTOFchi2(track.fTOFchi2),
   fTOFindex(track.fTOFindex),
   fTOFCalChannel(track.fTOFCalChannel),
   fTOFsignal(track.fTOFsignal),
   fTOFsignalToT(track.fTOFsignalToT),
-  //fPHOSsignal(track.fPHOSsignal),
   fRICHchi2(track.fRICHchi2),
   fRICHncls(track.fRICHncls),
   fRICHindex(track.fRICHindex),
@@ -196,7 +182,11 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fRICHtheta(track.fRICHtheta),
   fRICHphi(track.fRICHphi),
   fRICHdx(track.fRICHdx),
-  fRICHdy(track.fRICHdy)
+  fRICHdy(track.fRICHdy),
+  fRICHmipX(track.fRICHmipX),
+  fRICHmipY(track.fRICHmipY),
+  fPoints(0),
+  fFriendTrack(new AliESDfriendTrack(*(track.fFriendTrack)))
 {
   //
   //copy constructor
@@ -204,40 +194,32 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTrackTime[i]=track.fTrackTime[i];
   for (Int_t i=0;i<AliPID::kSPECIES;i++)  fR[i]=track.fR[i];
   //
-  for (Int_t i=0;i<12;i++) fITSchi2MIP[i]=track.fITSchi2MIP[i];
-  for (Int_t i=0;i<12;i++) fITSindex[i]=track.fITSindex[i];    
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fITSr[i]=track.fITSr[i]; 
   //
-  for (Int_t i=0;i<180;i++) fTPCindex[i]=track.fTPCindex[i];  
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTPCr[i]=track.fTPCr[i]; 
   for (Int_t i=0;i<4;i++) {fTPCPoints[i]=track.fTPCPoints[i];}
   for (Int_t i=0; i<3;i++)   { fKinkIndexes[i]=track.fKinkIndexes[i];}
   for (Int_t i=0; i<3;i++)   { fV0Indexes[i]=track.fV0Indexes[i];}
   //
-  for (Int_t i=0;i<180;i++) fTRDindex[i]=track.fTRDindex[i];   
   for (Int_t i=0;i<kNPlane;i++) {
-      fTRDsignals[i]=track.fTRDsignals[i]; 
-      fTRDTimBin[i]=track.fTRDTimBin[i];
+    for (Int_t j=0;j<kNSlice;j++) {
+      fTRDsignals[i][j]=track.fTRDsignals[i][j]; 
+    }
+    fTRDTimBin[i]=track.fTRDTimBin[i];
   }
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i]=track.fTRDr[i]; 
-  //
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTOFr[i]=track.fTOFr[i];
   for (Int_t i=0;i<3;i++) fTOFLabel[i]=track.fTOFLabel[i];
   for (Int_t i=0;i<10;i++) fTOFInfo[i]=track.fTOFInfo[i];
-  //
-  //  for (Int_t i=0;i<3;i++) fPHOSpos[i]=track.fPHOSpos[i]; 
-  //for (Int_t i=0;i<AliPID::kSPECIESN;i++) fPHOSr[i]=track.fPHOSr[i]; 
-  //
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fRICHr[i]=track.fRICHr[i];
 
   if (track.fCp) fCp=new AliExternalTrackParam(*track.fCp);
   if (track.fIp) fIp=new AliExternalTrackParam(*track.fIp);
   if (track.fOp) fOp=new AliExternalTrackParam(*track.fOp);
-  fPoints =0;
-  if (track.fPoints){
-    fPoints  = new AliTrackPointArray(*(track.fPoints));
-  }
+
+  if (track.fPoints) fPoints=new AliTrackPointArray(*(track.fPoints));
 }
+
 //_______________________________________________________________________
 AliESDtrack::~AliESDtrack(){ 
   //
@@ -247,8 +229,7 @@ AliESDtrack::~AliESDtrack(){
   delete fIp; 
   delete fOp;
   delete fCp; 
-  delete fITStrack;
-  delete fTRDtrack; 
+  delete fFriendTrack;
   delete fPoints;
 }
 
@@ -283,20 +264,16 @@ void AliESDtrack::MakeMiniESDtrack(){
 
   // Reset ITS track related information
   fITSchi2 = 0;
-  for (Int_t i=0;i<12;i++) fITSchi2MIP[i]=0;
   fITSncls = 0;       
-  for (Int_t i=0;i<12;i++) fITSindex[i]=0;    
   fITSsignal = 0;     
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fITSr[i]=0; 
   fITSLabel = 0;       
   fITSFakeRatio = 0;   
-  fITStrack =0;
 
   // Reset TPC related track information
   fTPCchi2 = 0;       
   fTPCncls = 0;       
   fTPCnclsF = 0;       
-  for (Int_t i=0;i<180;i++) fTPCindex[i] = 0;  
   fTPCClusterMap = 0;  
   fTPCsignal= 0;      
   fTPCsignalS= 0;      
@@ -311,15 +288,15 @@ void AliESDtrack::MakeMiniESDtrack(){
   fTRDchi2 = 0;        
   fTRDncls = 0;       
   fTRDncls0 = 0;       
-  for (Int_t i=0;i<180;i++) fTRDindex[i] = 0;   
   fTRDsignal = 0;      
   for (Int_t i=0;i<kNPlane;i++) {
-      fTRDsignals[i] = 0; 
-      fTRDTimBin[i]  = 0;
+    for (Int_t j=0;j<kNSlice;j++) {
+      fTRDsignals[i][j] = 0; 
+    }
+    fTRDTimBin[i]  = 0;
   }
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i] = 0; 
   fTRDLabel = 0;       
-  fTRDtrack = 0; 
   fTRDQuality  = 0;
   fTRDBudget  = 0;
 
@@ -333,11 +310,6 @@ void AliESDtrack::MakeMiniESDtrack(){
   for (Int_t i=0;i<3;i++) fTOFLabel[i] = 0;
   for (Int_t i=0;i<10;i++) fTOFInfo[i] = 0;
 
-  // Reset PHOS related track information
-  //for (Int_t i=0;i<3;i++) fPHOSpos[i] = 0; 
-  //fPHOSsignal = 0; 
-  //for (Int_t i=0;i<AliPID::kSPECIESN;i++) fPHOSr[i] = 0;
-  
   // Reset RICH related track information
   fRICHchi2 = 0;     
   fRICHncls = 0;     
@@ -349,7 +321,8 @@ void AliESDtrack::MakeMiniESDtrack(){
   fRICHdx = 0;     
   fRICHdy = 0;      
 
-  fPoints = 0;
+  delete fFriendTrack; fFriendTrack = 0;
+  delete fPoints; fPoints = 0;
 } 
 //_______________________________________________________________________
 Double_t AliESDtrack::GetMass() const {
@@ -380,6 +353,7 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
   //
   // This function updates track's running parameters 
   //
+  Int_t *index=0;
   Bool_t rc=kTRUE;
 
   SetStatus(flags);
@@ -396,9 +370,11 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
   switch (flags) {
     
   case kITSin: case kITSout: case kITSrefit:
+    index=fFriendTrack->GetITSindices(); 
+    for (Int_t i=0;i<AliESDfriendTrack::kMaxITScluster;i++) 
+         index[i]=t->GetClusterIndex(i);
     fITSncls=t->GetNumberOfClusters();
     fITSchi2=t->GetChi2();
-    for (Int_t i=0;i<12;i++) fITSindex[i]=t->GetClusterIndex(i);
     fITSsignal=t->GetPIDsignal();
     fITSLabel = t->GetLabel();
     fITSFakeRatio = t->GetFakeRatio();
@@ -409,6 +385,7 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
     if (!fIp) fIp=new AliExternalTrackParam(*t);
     else fIp->Set(*t);
   case kTPCout:
+    index=fFriendTrack->GetTPCindices(); 
     if (flags & kTPCout){
       if (!fOp) fOp=new AliExternalTrackParam(*t);
       else fOp->Set(*t);
@@ -420,14 +397,16 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
       //"jump to case label crosses initialization of `Int_t prevrow'"
        Int_t prevrow = -1;
        //       for (Int_t i=0;i<fTPCncls;i++) 
-       for (Int_t i=0;i<160;i++) 
+       for (Int_t i=0;i<AliESDfriendTrack::kMaxTPCcluster;i++) 
         {
-          fTPCindex[i]=t->GetClusterIndex(i);
+          index[i]=t->GetClusterIndex(i);
+          Int_t idx = index[i];
+
+         if (idx<0) continue; 
 
           // Piotr's Cluster Map for HBT  
           // ### please change accordingly if cluster array is changing 
           // to "New TPC Tracking" style (with gaps in array) 
-          Int_t idx = fTPCindex[i];
           Int_t sect = (idx&0xff000000)>>24;
           Int_t row = (idx&0x00ff0000)>>16;
           if (sect > 18) row +=63; //if it is outer sector, add number of inner sectors
@@ -469,10 +448,11 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
     break;
 
   case kTRDout: case kTRDin: case kTRDrefit:
+    index=fFriendTrack->GetTRDindices();
     fTRDLabel = t->GetLabel(); 
     fTRDncls=t->GetNumberOfClusters();
     fTRDchi2=t->GetChi2();
-    for (Int_t i=0;i<fTRDncls;i++) fTRDindex[i]=t->GetClusterIndex(i);
+    for (Int_t i=0;i<fTRDncls;i++) index[i]=t->GetClusterIndex(i);
     fTRDsignal=t->GetPIDsignal();
     break;
   case kTRDbackup:
@@ -656,9 +636,6 @@ void AliESDtrack::SetITSpid(const Double_t *p) {
   SetStatus(AliESDtrack::kITSpid);
 }
 
-void AliESDtrack::SetITSChi2MIP(const Float_t *chi2mip){
-  for (Int_t i=0; i<12; i++) fITSchi2MIP[i]=chi2mip[i];
-}
 //_______________________________________________________________________
 void AliESDtrack::GetITSpid(Double_t *p) const {
   // Gets the probability of each particle type (in ITS)
@@ -670,7 +647,10 @@ Int_t AliESDtrack::GetITSclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined ITS clusters 
   //---------------------------------------------------------------------
-  for (Int_t i=0; i<12; i++) idx[i]=fITSindex[i];
+  if (idx!=0) {
+     Int_t *index=fFriendTrack->GetITSindices();
+     for (Int_t i=0; i<AliESDfriendTrack::kMaxITScluster; i++) idx[i]=index[i];
+  }
   return fITSncls;
 }
 
@@ -679,8 +659,10 @@ Int_t AliESDtrack::GetTPCclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined ITS clusters 
   //---------------------------------------------------------------------
-  if (idx!=0)
-    for (Int_t i=0; i<180; i++) idx[i]=fTPCindex[i];  // MI I prefer some constant
+  if (idx!=0) {
+    Int_t *index=fFriendTrack->GetTPCindices();
+    for (Int_t i=0; i<AliESDfriendTrack::kMaxTPCcluster; i++) idx[i]=index[i];
+  }
   return fTPCncls;
 }
 
@@ -692,10 +674,11 @@ Float_t AliESDtrack::GetTPCdensity(Int_t row0, Int_t row1) const{
   Int_t good  = 0;
   Int_t found = 0;
   //  
+  Int_t *index=fFriendTrack->GetTPCindices();
   for (Int_t i=row0;i<=row1;i++){     
-    Int_t index = fTPCindex[i];
-    if (index!=-1)  good++;             // track outside of dead zone
-    if (index>0)    found++;
+    Int_t idx = index[i];
+    if (idx!=-1)  good++;             // track outside of dead zone
+    if (idx>0)    found++;
   }
   Float_t density=0.5;
   if (good>(row1-row0)*0.5) density = Float_t(found)/Float_t(good);
@@ -720,8 +703,10 @@ Int_t AliESDtrack::GetTRDclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined TRD clusters 
   //---------------------------------------------------------------------
-  if (idx!=0)
-    for (Int_t i=0; i<180; i++) idx[i]=fTRDindex[i];  // MI I prefer some constant
+  if (idx!=0) {
+     Int_t *index=fFriendTrack->GetTRDindices();
+     for (Int_t i=0; i<AliESDfriendTrack::kMaxTRDcluster; i++) idx[i]=index[i];
+  }
   return fTRDncls;
 }
 
@@ -845,9 +830,13 @@ Bool_t AliESDtrack::RelateToVertex
   if (d > maxd) return kFALSE; 
 
   //Propagate to the DCA
-  Double_t crv=0.299792458e-3*b*GetParameter()[4];
+  Double_t crv=kB2C*b*GetParameter()[4];
+  if (TMath::Abs(b) < kAlmost0Field) crv=0.;
+
   Double_t tgfv=-(crv*x - snp)/(crv*y + TMath::Sqrt(1.-snp*snp));
-  sn=tgfv/TMath::Sqrt(1.+ tgfv*tgfv); cs=TMath::Sqrt(1.- sn*sn);
+  sn=tgfv/TMath::Sqrt(1.+ tgfv*tgfv);
+  if (TMath::Abs(tgfv)>0.) cs = sn/tgfv;
+  else cs=1.;
 
   x = xv*cs + yv*sn;
   yv=-xv*sn + yv*cs; xv=x;
@@ -879,6 +868,15 @@ Bool_t AliESDtrack::RelateToVertex
   return kTRUE;
 }
 
+void AliESDtrack::SetTrackPointArray(AliTrackPointArray *points) { 
+    fPoints=points;
+    //fFriendTrack->SetTrackPointArray(points); 
+}
+const AliTrackPointArray *AliESDtrack::GetTrackPointArray() const { 
+    return fPoints;
+  //return fFriendTrack->GetTrackPointArray(); 
+}
+
 //_______________________________________________________________________
 void AliESDtrack::Print(Option_t *) const {
   // Prints info on the track
@@ -921,13 +919,4 @@ void AliESDtrack::Print(Option_t *) const {
       printf("%f, ", p[index]) ;
     printf("\n           signal = %f\n", GetRICHsignal()) ;
   }
-  // Since 9 March 2006 PHOS left ESDtrack for ESDCaloCluster
-  // and cannot participate in the global PID for the moment
-//   if( IsOn(kPHOSpid) ){
-//     printf("From PHOS: ") ; 
-//     GetPHOSpid(p) ; 
-//     for(index = 0 ; index < AliPID::kSPECIESN; index++) 
-//       printf("%f, ", p[index]) ;
-//     printf("\n           signal = %f\n", GetPHOSsignal()) ;
-//   }
 }