]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDpid.cxx
Fix in composition of QAChecked output image file (Melinda S.)
[u/mrichter/AliRoot.git] / STEER / AliESDpid.cxx
index a2aa64d993accc2500320f357fac36bc539be6e0..589d963f145a61d8ceb75ad0a139740f2813b27b 100644 (file)
@@ -98,7 +98,7 @@ void AliESDpid::MakeTPCPID(AliESDtrack *track) const
     }
 
     if (mismatch)
-       for (Int_t j=0; j<AliPID::kSPECIES; j++) p[j]=1/AliPID::kSPECIES;
+       for (Int_t j=0; j<AliPID::kSPECIES; j++) p[j]=1./AliPID::kSPECIES;
 
     track->SetTPCpid(p);
 
@@ -232,12 +232,12 @@ void AliESDpid::MakeTOFPID(AliESDtrack *track, Float_t /*timeZeroTOF*/) const
   }
 
   if (mismatch)
-    for (Int_t j=0; j<AliPID::kSPECIES; j++) p[j]=1/AliPID::kSPECIES;
+    for (Int_t j=0; j<AliPID::kSPECIES; j++) p[j]=1./AliPID::kSPECIES;
 
   track->SetTOFpid(p);
 
   if (heavy) track->ResetStatus(AliESDtrack::kTOFpid);    
-  if (!CheckTOFMatching(track)) track->SetStatus(AliESDtrack::kTOFmismatch);    
+  if (!CheckTOFMatching(track)) track->SetStatus(AliESDtrack::kTOFmismatch);
   
 }
 //_________________________________________________________________________
@@ -246,16 +246,8 @@ void AliESDpid::MakeTRDPID(AliESDtrack *track) const
   //
   // Method to recalculate the TRD PID probabilities
   //
-  if((track->GetStatus()&AliESDtrack::kTRDout)==0) return;
-  Double_t prob[AliPID::kSPECIES]; Float_t mom[6];
-  Double_t dedx[48];  // Allocate space for the maximum number of TRD slices
-  for(Int_t ilayer = 0; ilayer < 6; ilayer++){
-    mom[ilayer] = track->GetTRDmomentum(ilayer);
-    for(Int_t islice = 0; islice < track->GetNumberOfTRDslices(); islice++){
-      dedx[ilayer*track->GetNumberOfTRDslices()+islice] = track->GetTRDslice(ilayer, islice);
-    }
-  }
-  fTRDResponse.GetResponse(track->GetNumberOfTRDslices(), dedx, mom, prob);
+  Double_t prob[AliPID::kSPECIES];
+  ComputeTRDProbability(track, AliPID::kSPECIES, prob);
   track->SetTRDpid(prob);
 }
 //_________________________________________________________________________
@@ -302,16 +294,19 @@ void AliESDpid::CombinePID(AliESDtrack *track) const
 }
 //_________________________________________________________________________
 Bool_t AliESDpid::CheckTOFMatching(AliESDtrack *track) const{
+  //
+  // Check pid matching of TOF with TPC as reference
+  //
     Bool_t status = kFALSE;
     
     Double_t exptimes[5];
     track->GetIntegratedTimes(exptimes);
     
-    Float_t dedx = track->GetTPCsignal();
-    Float_t time = track->GetTOFsignal();
-    
     Float_t p = track->P();
     
+    Float_t dedx = track->GetTPCsignal();
+    Float_t time = track->GetTOFsignal() - fTOFResponse.GetStartTime(p);
+    
     Double_t ptpc[3];
     track->GetInnerPxPyPz(ptpc);
     Float_t momtpc=TMath::Sqrt(ptpc[0]*ptpc[0] + ptpc[1]*ptpc[1] + ptpc[2]*ptpc[2]);
@@ -324,109 +319,112 @@ Bool_t AliESDpid::CheckTOFMatching(AliESDtrack *track) const{
            Float_t dedxExp = fTPCResponse.GetExpectedSignal(momtpc,type);
            Float_t resolutionTPC = fTPCResponse.GetExpectedSigma(momtpc,track->GetTPCsignalN(),type);
            
-           if(TMath::Abs(dedx - dedxExp) < fRange * resolutionTPC){
+           if(TMath::Abs(dedx - dedxExp) < fRangeTOFMismatch * resolutionTPC){
                status = kTRUE;
            }
        }
     }
     
     // for nuclei
-    Float_t dedxExpPr = fTPCResponse.GetExpectedSignal(momtpc,AliPID::kProton);
-    Float_t resolutionTPCpr =  fTPCResponse.GetExpectedSigma(momtpc,track->GetTPCsignalN(),AliPID::kProton);
-    if(!status && (exptimes[4] < time && dedx > dedxExpPr + resolutionTPCpr*fRange)) status = kTRUE;
+    Float_t resolutionTOFpr = fTOFResponse.GetExpectedSigma(p, exptimes[4], AliPID::ParticleMass(4));
+    if(!status && (exptimes[4] + fRange*resolutionTOFpr < time)) status = kTRUE;
     
     
     return status;
 }
 //_________________________________________________________________________
-void AliESDpid::SetTOFResponse(AliESDEvent *event,EStartTimeType_t option){
+void AliESDpid::SetTOFResponse(AliVEvent *vevent,EStartTimeType_t option){
   //
   // Set TOF response function
   // Input option for event_time used
   //
 
+    AliESDEvent *event=(AliESDEvent*)vevent;
+  
     Float_t t0spread = 0.; //event->GetEventTimeSpread();
     if(t0spread < 10) t0spread = 80;
 
     // T0 from TOF algorithm
-    //Float_t t0Gen,t0ResGen;
-    //Int_t nt0;
-    //Float_t t0ESD[fTOFResponse.GetNmomBins()],t0resESD[fTOFResponse.GetNmomBins()];
-    //Int_t it0ESD[fTOFResponse.GetNmomBins()];
 
     Bool_t flagT0TOF=kFALSE;
     Bool_t flagT0T0=kFALSE;
     Float_t *startTime = new Float_t[fTOFResponse.GetNmomBins()];
     Float_t *startTimeRes = new Float_t[fTOFResponse.GetNmomBins()];
+    Int_t *startTimeMask = new Int_t[fTOFResponse.GetNmomBins()];
 
+    // T0-TOF arrays
     Float_t *estimatedT0event = new Float_t[fTOFResponse.GetNmomBins()];
     Float_t *estimatedT0resolution = new Float_t[fTOFResponse.GetNmomBins()];
     for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
       estimatedT0event[i]=0.0;
       estimatedT0resolution[i]=0.0;
+      startTimeMask[i] = 0;
     }
 
-    if(event->GetT0TOF()){
+    Float_t resT0A=75,resT0C=65,resT0AC=55;
+    if(event->GetT0TOF()){ // check if T0 detector information is available
        flagT0T0=kTRUE;
     }
 
 
     AliTOFHeader *tofHeader =(AliTOFHeader*)event->GetTOFHeader();
 
-    if(tofHeader){
+    if(tofHeader){ // read global info and T0-TOF info from ESD
       fTOFResponse.SetTimeResolution(tofHeader->GetTOFResolution());
-      t0spread = tofHeader->GetT0spread();
+      t0spread = tofHeader->GetT0spread(); // read t0 sprad
       if(t0spread < 10) t0spread = 80;
 
       flagT0TOF=kTRUE;
-      for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-       startTime[i]=tofHeader->GetDefaultEventTimeVal();//t0Gen;
-       startTimeRes[i]=tofHeader->GetDefaultEventTimeRes();//t0ResGen;
+      for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){ // read T0-TOF default value
+       startTime[i]=tofHeader->GetDefaultEventTimeVal();
+       startTimeRes[i]=tofHeader->GetDefaultEventTimeRes();
+       if(startTimeRes[i] < 1.e-5) startTimeRes[i] = t0spread;
       }
 
       TArrayI *ibin=tofHeader->GetNvalues();
       TArrayF *t0Bin=tofHeader->GetEventTimeValues();
       TArrayF *t0ResBin=tofHeader->GetEventTimeRes();
-      for(Int_t j=0;j < tofHeader->GetNbins();j++){
+      for(Int_t j=0;j < tofHeader->GetNbins();j++){ // fill T0-TOF in p-bins
        Int_t icurrent = (Int_t)ibin->GetAt(j);
-       startTime[icurrent]=t0Bin->GetAt(j);//t0ESD[j];
-       startTimeRes[icurrent]=t0ResBin->GetAt(j);//t0resESD[j];
-      }
+       startTime[icurrent]=t0Bin->GetAt(j);
+       startTimeRes[icurrent]=t0ResBin->GetAt(j);
+       if(startTimeRes[icurrent] < 1.e-5) startTimeRes[icurrent] = t0spread;
+     }
     }
 
+    // for cut of 3 sigma on t0 spread
+    Float_t t0cut = 3 * t0spread;
+    if(t0cut < 500) t0cut = 500;
+
     if(option == kFILL_T0){ // T0-FILL is used
        for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
          estimatedT0event[i]=0.0;
          estimatedT0resolution[i]=t0spread;
-         //fT0event[i] = 0.0;
-         //fT0resolution[i] = t0spread;
        }
        fTOFResponse.SetT0event(estimatedT0event);
        fTOFResponse.SetT0resolution(estimatedT0resolution);
     }
+
     if(option == kTOF_T0){ // T0-TOF is used when available (T0-FILL otherwise) from ESD
        if(flagT0TOF){
-           for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-             estimatedT0event[i]=0.0;
-             estimatedT0resolution[i]=t0spread;
-             //fT0event[i] = startTime[i];
-             //fT0resolution[i] = startTimeRes[i];
-           }
            fTOFResponse.SetT0event(startTime);
            fTOFResponse.SetT0resolution(startTimeRes);
+           for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
+             if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
+             fTOFResponse.SetT0binMask(i,startTimeMask[i]);
+           }
        }
        else{
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
              estimatedT0event[i]=0.0;
              estimatedT0resolution[i]=t0spread;
-             //fT0event[i] = 0.0;
-             //fT0resolution[i] = t0spread;
+             fTOFResponse.SetT0binMask(i,startTimeMask[i]);
            }
            fTOFResponse.SetT0event(estimatedT0event);
            fTOFResponse.SetT0resolution(estimatedT0resolution);
        }
     }
-    else if(option == kBest_T0){ // T0-T0 is used when available (T0-FILL otherwise) from ESD
+    else if(option == kBest_T0){ // T0-T0 or T0-TOF are used when available (T0-FILL otherwise) from ESD
        Float_t t0AC=-10000;
        Float_t t0A=-10000;
        Float_t t0C=-10000;
@@ -438,67 +436,67 @@ void AliESDpid::SetTOFResponse(AliESDEvent *event,EStartTimeType_t option){
 
        Float_t t0t0Best = 0;
        Float_t t0t0BestRes = 9999;
-       if(TMath::Abs(t0AC) < 500){
+       Int_t t0used=0;
+       if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
            t0t0Best = t0AC;
-           t0t0BestRes = 55;
-       }
-       else if(TMath::Abs(t0A) < 500){
-           t0t0Best = t0A;
-           t0t0BestRes = 75;
+           t0t0BestRes = resT0AC;
+           t0used=6;
        }
-       else if(TMath::Abs(t0C) < 500){
+       else if(TMath::Abs(t0C) < t0cut){
            t0t0Best = t0C;
-           t0t0BestRes = 65;
+           t0t0BestRes = resT0C;
+           t0used=4;
+       }
+       else if(TMath::Abs(t0A) < t0cut){
+           t0t0Best = t0A;
+           t0t0BestRes = resT0A;
+           t0used=2;
        }
 
        if(flagT0TOF){ // if T0-TOF info is available
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-               if(t0t0BestRes<300){
-                 if(startTimeRes[i]<t0spread){
+               if(t0t0BestRes < 999){
+                 if(startTimeRes[i] < t0spread){
                    Double_t wtot = 1./startTimeRes[i]/startTimeRes[i] + 1./t0t0BestRes/t0t0BestRes;
                    Double_t t0best = startTime[i]/startTimeRes[i]/startTimeRes[i] + t0t0Best/t0t0BestRes/t0t0BestRes;
                    estimatedT0event[i]=t0best / wtot;
                    estimatedT0resolution[i]=1./TMath::Sqrt(wtot);
-                   //fT0event[i]=t0best / wtot;
-                   //fT0resolution[i]=1./TMath::Sqrt(wtot);
+                   startTimeMask[i] = t0used+1;
                  }
                  else {
                    estimatedT0event[i]=t0t0Best;
                    estimatedT0resolution[i]=t0t0BestRes;
-                   //fT0event[i] = t0t0Best;
-                   //fT0resolution[i] = t0t0BestRes;
+                   startTimeMask[i] = t0used;
                  }
                }
                else{
                  estimatedT0event[i]=startTime[i];
                  estimatedT0resolution[i]=startTimeRes[i];
-                 //fT0event[i] = startTime[i];
-                 //fT0resolution[i] = startTimeRes[i];
+                 if(startTimeRes[i]<t0spread) startTimeMask[i]=1;
                }
+               fTOFResponse.SetT0binMask(i,startTimeMask[i]);
            }
            fTOFResponse.SetT0event(estimatedT0event);
            fTOFResponse.SetT0resolution(estimatedT0resolution);
        }
        else{ // if no T0-TOF info is available
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-               if(t0t0BestRes<300){
-                 estimatedT0event[i]=t0t0Best;
-                 estimatedT0resolution[i]=t0t0BestRes;
-                 //fT0event[i] = t0t0Best;
-                 //fT0resolution[i] = t0t0BestRes;
-               }
-               else{
-                 estimatedT0event[i]=0.0;
-                 estimatedT0resolution[i]=t0spread;
-                 //fT0event[i] = 0.0;
-                 //fT0resolution[i] = t0spread;
-               }
+             fTOFResponse.SetT0binMask(i,t0used);
+             if(t0t0BestRes < 999){
+               estimatedT0event[i]=t0t0Best;
+               estimatedT0resolution[i]=t0t0BestRes;
+             }
+             else{
+               estimatedT0event[i]=0.0;
+               estimatedT0resolution[i]=t0spread;
+             }
            }
            fTOFResponse.SetT0event(estimatedT0event);
            fTOFResponse.SetT0resolution(estimatedT0resolution);
        }
     }
-    else if(option == kT0_T0){ // the best of what available (T0-FILL otherwise) from ESD
+
+    else if(option == kT0_T0){ // T0-T0 is used when available (T0-FILL otherwise) from ESD
        Float_t t0AC=-10000;
        Float_t t0A=-10000;
        Float_t t0C=-10000;
@@ -508,38 +506,32 @@ void AliESDpid::SetTOFResponse(AliESDEvent *event,EStartTimeType_t option){
            t0C= event->GetT0TOF()[2];
        }
 
-//     printf("T0_T0 = %f %f %f\n",t0AC,t0A,t0C);
-
-       if(TMath::Abs(t0AC) < 500){
+       if(TMath::Abs(t0A) < t0cut && TMath::Abs(t0C) < t0cut && TMath::Abs(t0C-t0A) < 500){
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
              estimatedT0event[i]=t0AC;
-             estimatedT0resolution[i]=55;
-             //fT0event[i] = t0AC;
-             //fT0resolution[i] = 55;
+             estimatedT0resolution[i]=resT0AC;
+             fTOFResponse.SetT0binMask(i,6);
            }
        }
-       else if(TMath::Abs(t0A) < 500){
+       else if(TMath::Abs(t0C) < t0cut){
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-             estimatedT0event[i]=t0A;
-             estimatedT0resolution[i]=75;
-             //fT0event[i] = t0A;
-             //fT0resolution[i] = 75;
+             estimatedT0event[i]=t0C;
+             estimatedT0resolution[i]=resT0C;
+             fTOFResponse.SetT0binMask(i,4);
            }
        }
-       else if(TMath::Abs(t0C) < 500){
+       else if(TMath::Abs(t0A) < t0cut){
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
-             estimatedT0event[i]=t0C;
-             estimatedT0resolution[i]=65;
-             //fT0event[i] = t0C;
-             //fT0resolution[i] = 65;
+             estimatedT0event[i]=t0A;
+             estimatedT0resolution[i]=resT0A;
+             fTOFResponse.SetT0binMask(i,2);
            }
        }
        else{
            for(Int_t i=0;i<fTOFResponse.GetNmomBins();i++){
              estimatedT0event[i]=0.0;
              estimatedT0resolution[i]=t0spread;
-             //fT0event[i] = 0.0;
-             //fT0resolution[i] = t0spread;
+             fTOFResponse.SetT0binMask(i,0);
            }
        }
        fTOFResponse.SetT0event(estimatedT0event);
@@ -547,6 +539,7 @@ void AliESDpid::SetTOFResponse(AliESDEvent *event,EStartTimeType_t option){
     }
     delete [] startTime;
     delete [] startTimeRes;
+    delete [] startTimeMask;
     delete [] estimatedT0event;
     delete [] estimatedT0resolution;
 }