New version of the V0 finder (M.Ivanov)
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
index 3637d9e..de33b51 100644 (file)
@@ -30,47 +30,54 @@ ClassImp(AliESDtrack)
 
 //_______________________________________________________________________
 AliESDtrack::AliESDtrack() : 
-fFlags(0),
-fLabel(0),
-fID(0),
-fTrackLength(0),
-fD(0),
-fZ(0),
-fStopVertex(0),
-fRalpha(0),
-fRx(0),
-fCalpha(0),
-fCx(0),
-fCchi2(1e10),
-fIalpha(0),
-fIx(0),
-fTalpha(0),
-fTx(0),
-fITSchi2(0),
-fITSncls(0),
-fITSsignal(0),
-fTPCchi2(0),
-fTPCncls(0),
-fTPCClusterMap(159),//number of padrows
-fTPCsignal(0),
-fTRDchi2(0),
-fTRDncls(0),
-fTRDncls0(0),
-fTRDsignal(0),
-fTRDQuality(0),
-fTOFchi2(0),
-fTOFindex(0),
-fTOFsignal(-1),
-fPHOSsignal(-1),
-fEMCALsignal(-1),
-fRICHchi2(1e10),
-fRICHncls(0),
-fRICHindex(0),
-fRICHsignal(-1),
-fRICHtheta(0),
-fRICHphi(0),
-fRICHdx(0),
-fRICHdy(0)
+  TObject(),
+  fFlags(0),
+  fLabel(0),
+  fID(0),
+  fTrackLength(0),
+  fD(0),
+  fZ(0),
+  fStopVertex(0),
+  fRalpha(0),
+  fRx(0),
+  fCalpha(0),
+  fCx(0),
+  fCchi2(1e10),
+  fIalpha(0),
+  fIx(0),
+  fTalpha(0),
+  fTx(0),
+  fITSchi2(0),
+  fITSncls(0),
+  fITSsignal(0),
+  fITSLabel(0),
+  fITSFakeRatio(0),
+  fITStrack(0),
+  fTPCchi2(0),
+  fTPCncls(0),
+  fTPCClusterMap(159),//number of padrows
+  fTPCsignal(0),
+  fTPCLabel(0),
+  fTRDchi2(0),
+  fTRDncls(0),
+  fTRDncls0(0),
+  fTRDsignal(0),
+  fTRDLabel(0),
+  fTRDQuality(0),
+  fTRDtrack(0),
+  fTOFchi2(0),
+  fTOFindex(0),
+  fTOFsignal(-1),
+  fPHOSsignal(-1),
+  fEMCALsignal(-1),
+  fRICHchi2(1e10),
+  fRICHncls(0),
+  fRICHindex(0),
+  fRICHsignal(-1),
+  fRICHtheta(0),
+  fRICHphi(0),
+  fRICHdx(0),
+  fRICHdy(0)
 {
   //
   // The default ESD constructor 
@@ -118,100 +125,102 @@ fRICHdy(0)
 }
 
 //_______________________________________________________________________
-
-AliESDtrack::AliESDtrack(const AliESDtrack& track):TObject(track){
+AliESDtrack::AliESDtrack(const AliESDtrack& track):
+  TObject(track),
+  fFlags(track.fFlags),
+  fLabel(track.fLabel),
+  fID(track.fID),
+  fTrackLength(track.fTrackLength),
+  fD(track.fD),
+  fZ(track.fZ),
+  fStopVertex(track.fStopVertex),
+  fRalpha(track.fRalpha),
+  fRx(track.fRx),
+  fCalpha(track.fCalpha),
+  fCx(track.fCx),
+  fCchi2(track.fCchi2),
+  fIalpha(track.fIalpha),
+  fIx(track.fIx),
+  fTalpha(track.fTalpha),
+  fTx(track.fTx),
+  fITSchi2(track.fITSchi2),
+  fITSncls(track.fITSncls),
+  fITSsignal(track.fITSsignal),
+  fITSLabel(track.fITSLabel),
+  fITSFakeRatio(track.fITSFakeRatio),
+  fITStrack(0),    //coping separatelly - in user code
+  fTPCchi2(track.fTPCchi2),
+  fTPCncls(track.fTPCncls),
+  fTPCClusterMap(track.fTPCClusterMap),
+  fTPCsignal(track.fTPCsignal),
+  fTPCLabel(track.fTPCLabel),
+  fTRDchi2(track.fTRDchi2),
+  fTRDncls(track.fTRDncls),
+  fTRDncls0(track.fTRDncls0),
+  fTRDsignal(track.fTRDsignal),
+  fTRDLabel(track.fTRDLabel),
+  fTRDQuality(track.fTRDQuality),
+  fTRDtrack(0),
+  fTOFchi2(track.fTOFchi2),
+  fTOFindex(track.fTOFindex),
+  fTOFsignal(track.fTOFsignal),
+  fPHOSsignal(track.fPHOSsignal),
+  fEMCALsignal(track.fEMCALsignal),
+  fRICHchi2(track.fRICHchi2),
+  fRICHncls(track.fRICHncls),
+  fRICHindex(track.fRICHindex),
+  fRICHsignal(track.fRICHsignal),
+  fRICHtheta(track.fRICHtheta),
+  fRICHphi(track.fRICHphi),
+  fRICHdx(track.fRICHdx),
+  fRICHdy(track.fRICHdy)
+{
   //
   //copy constructor
   //
-  fFlags = track.fFlags;
-  fLabel =track.fLabel;
-  fID = track.fID;
-  fTrackLength =track.fTrackLength;
-  fD=track.fD; fZ=track.fZ;
   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];
-  fStopVertex =track.fStopVertex;
   //
-  fRalpha =track.fRalpha;
-  fRx =track.fRx;
   for (Int_t i=0;i<5;i++) fRp[i] =track.fRp[i];
   for (Int_t i=0;i<15;i++) fRc[i] =track.fRc[i];
   //
-  fCalpha =track.fCalpha;
-  fCx =track.fCx;
   for (Int_t i=0;i<5;i++) fCp[i] =track.fCp[i];
   for (Int_t i=0;i<15;i++)  fCc[i] =track.fCc[i];
-  fCchi2 =track.fCchi2;
   //
-  fIalpha =track.fIalpha;
-  fIx =track.fIx;
   for (Int_t i=0;i<5;i++) fIp[i] =track.fIp[i];
   for (Int_t i=0;i<15;i++)  fIc[i] =track.fIc[i];
   //
-  fTalpha =track.fTalpha;
-  fTx =track.fTx;
   for (Int_t i=0;i<5;i++) fTp[i] =track.fTp[i];
   for (Int_t i=0;i<15;i++)  fTc[i] =track.fTc[i];
   //
-  fITSchi2 =track.fITSchi2;
   for (Int_t i=0;i<12;i++) fITSchi2MIP[i] =track.fITSchi2MIP[i];
-  fITSncls =track.fITSncls;       
   for (Int_t i=0;i<6;i++) fITSindex[i]=track.fITSindex[i];    
-  fITSsignal =track.fITSsignal;     
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fITSr[i]=track.fITSr[i]; 
-  fITSLabel =track.fITSLabel;       
-  fITSFakeRatio =track.fITSFakeRatio;   
-  fITStrack =0;  //coping separatelly - in user code
   //
-  fTPCchi2 =track.fTPCchi2;       
-  fTPCncls =track.fTPCncls;       
   for (Int_t i=0;i<180;i++) fTPCindex[i]=track.fTPCindex[i];  
-  fTPCClusterMap=track.fTPCClusterMap;  
-  fTPCsignal=track.fTPCsignal;      
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTPCr[i]=track.fTPCr[i]; 
-  fTPCLabel=track.fTPCLabel;       
   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];}
   //
-  fTRDchi2=track.fTRDchi2;        
-  fTRDncls=track.fTRDncls;       
-  fTRDncls0=track.fTRDncls0;       
   for (Int_t i=0;i<130;i++) fTRDindex[i]=track.fTRDindex[i];   
-  fTRDsignal=track.fTRDsignal;      
   for (Int_t i=0;i<kNPlane;i++) {
       fTRDsignals[i]=track.fTRDsignals[i]; 
       fTRDTimBin[i]=track.fTRDTimBin[i];
   }
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i]=track.fTRDr[i]; 
-  fTRDLabel=track.fTRDLabel;       
-  fTRDtrack=0; 
-  fTRDQuality  = track.fTRDQuality;
   //
-  fTOFchi2=track.fTOFchi2;        
-  fTOFindex=track.fTOFindex;       
-  fTOFsignal=track.fTOFsignal;      
   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]; 
-  fPHOSsignal=track.fPHOSsignal; 
   for (Int_t i=0;i<AliPID::kSPECIESN;i++) fPHOSr[i]=track.fPHOSr[i]; 
   //
   for (Int_t i=0;i<3;i++) fEMCALpos[i]=track.fEMCALpos[i]; 
-  fEMCALsignal=track.fEMCALsignal; 
   for (Int_t i=0;i<AliPID::kSPECIESN;i++) fEMCALr[i]=track.fEMCALr[i]; 
   //
-  fRICHchi2=track.fRICHchi2;     
-  fRICHncls=track.fRICHncls;     
-  fRICHindex=track.fRICHindex;     
-  fRICHsignal=track.fRICHsignal;     
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fRICHr[i]=track.fRICHr[i];
-  fRICHtheta=track.fRICHtheta;     
-  fRICHphi=track.fRICHphi;      
-  fRICHdx=track.fRICHdx;     
-  fRICHdy=track.fRICHdy;      
 }
 //_______________________________________________________________________
 AliESDtrack::~AliESDtrack(){ 
@@ -224,6 +233,112 @@ AliESDtrack::~AliESDtrack(){
 }
 
 //_______________________________________________________________________
+void AliESDtrack::MakeMiniESDtrack(){
+  // Resets everything except
+  // fFlags: Reconstruction status flags 
+  // fLabel: Track label
+  // fID:  Unique ID of the track
+  // fD: Impact parameter in XY-plane
+  // fZ: Impact parameter in Z 
+  // fR[AliPID::kSPECIES]: combined "detector response probability"
+  // Running track parameters
+  // fRalpha: track rotation angle
+  // fRx: X-coordinate of the track reference plane 
+  // fRp[5]: external track parameters  
+  // fRc[15]: external cov. matrix of the track parameters
+  
+  fTrackLength = 0;
+  for (Int_t i=0;i<AliPID::kSPECIES;i++) fTrackTime[i] = 0;
+  fStopVertex = 0;
+
+  // Reset track parameters constrained to the primary vertex
+  fCalpha = 0;
+  fCx = 0;
+  for (Int_t i=0;i<5;i++) fCp[i] = 0;
+  for (Int_t i=0;i<15;i++)  fCc[i] = 0;
+  fCchi2 = 0;
+
+  // Reset track parameters at the inner wall of TPC
+  fIalpha = 0;
+  fIx = 0;
+  for (Int_t i=0;i<5;i++) fIp[i] = 0;
+  for (Int_t i=0;i<15;i++)  fIc[i] = 0;
+
+  // Reset track parameters at the inner wall of the TRD
+  fTalpha = 0;
+  fTx = 0;
+  for (Int_t i=0;i<5;i++) fTp[i] = 0;
+  for (Int_t i=0;i<15;i++)  fTc[i] = 0;
+
+  // 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<6;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;       
+  for (Int_t i=0;i<180;i++) fTPCindex[i] = 0;  
+  fTPCClusterMap = 0;  
+  fTPCsignal= 0;      
+  for (Int_t i=0;i<AliPID::kSPECIES;i++) fTPCr[i]=0; 
+  fTPCLabel=0;       
+  for (Int_t i=0;i<4;i++) fTPCPoints[i] = 0;
+  for (Int_t i=0; i<3;i++)   fKinkIndexes[i] = 0;
+  for (Int_t i=0; i<3;i++)   fV0Indexes[i] = 0;
+
+  // Reset TRD related track information
+  fTRDchi2 = 0;        
+  fTRDncls = 0;       
+  fTRDncls0 = 0;       
+  for (Int_t i=0;i<130;i++) fTRDindex[i] = 0;   
+  fTRDsignal = 0;      
+  for (Int_t i=0;i<kNPlane;i++) {
+      fTRDsignals[i] = 0; 
+      fTRDTimBin[i]  = 0;
+  }
+  for (Int_t i=0;i<AliPID::kSPECIES;i++) fTRDr[i] = 0; 
+  fTRDLabel = 0;       
+  fTRDtrack = 0; 
+  fTRDQuality  = 0;
+
+  // Reset TOF related track information
+  fTOFchi2 = 0;        
+  fTOFindex = 0;       
+  fTOFsignal = 0;      
+  for (Int_t i=0;i<AliPID::kSPECIES;i++) fTOFr[i] = 0;
+  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 EMCAL related track information
+  for (Int_t i=0;i<3;i++) fEMCALpos[i] = 0; 
+  fEMCALsignal = 0; 
+  for (Int_t i=0;i<AliPID::kSPECIESN;i++) fEMCALr[i] = 0;
+  // Reset RICH related track information
+  fRICHchi2 = 0;     
+  fRICHncls = 0;     
+  fRICHindex = 0;     
+  fRICHsignal = 0;     
+  for (Int_t i=0;i<AliPID::kSPECIES;i++) fRICHr[i] = 0;
+  fRICHtheta = 0;     
+  fRICHphi = 0;      
+  fRICHdx = 0;     
+  fRICHdy = 0;      
+
+} 
+//_______________________________________________________________________
 Double_t AliESDtrack::GetMass() const {
   // Returns the mass of the most probable particle type
   Float_t max=0.;
@@ -718,10 +833,42 @@ Int_t AliESDtrack::GetTPCclusters(Int_t *idx) const {
   return fTPCncls;
 }
 
+Float_t AliESDtrack::GetTPCdensity(Int_t row0, Int_t row1) const{
+  //
+  // GetDensity of the clusters on given region between row0 and row1
+  // Dead zone effect takin into acoount
+  //
+  Int_t good  = 0;
+  Int_t found = 0;
+  //  
+  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++;
+  }
+  Float_t density=0.5;
+  if (good>(row1-row0)*0.5) density = Float_t(found)/Float_t(good);
+  return density;
+}
 //_______________________________________________________________________
 void AliESDtrack::SetTPCpid(const Double_t *p) {  
   // Sets values for the probability of each particle type (in TPC)
-  for (Int_t i=0; i<AliPID::kSPECIES; i++) fTPCr[i]=p[i];
+  // normalize probabiluty to 1
+  // 
+  //
+//   Double_t sump=0;
+//   for (Int_t i=0; i<AliPID::kSPECIES; i++) {
+//     sump+=p[i];
+//   }
+//   for (Int_t i=0; i<AliPID::kSPECIES; i++) {
+//     if (sump>0){
+//       fTPCr[i]=p[i]/sump;
+//     }
+//     else{
+//       fTPCr[i]=p[i];
+//     }
+//   }
+  for (Int_t i=0; i<AliPID::kSPECIES; i++) fTPCr[i] = p[i];
   SetStatus(AliESDtrack::kTPCpid);
 }