Update the mult corr histograms

[u/mrichter/AliRoot.git] / CORRFW / AliCFTrackQualityCuts.cxx

diff --git a/CORRFW/AliCFTrackQualityCuts.cxx b/CORRFW/AliCFTrackQualityCuts.cxx
index 2d94f1fc356eb7a97b6720894287d5f8222ea94c..fd6324bbac555c0f7f524780a06e1af67fcd9169 100644 (file)
--- a/CORRFW/AliCFTrackQualityCuts.cxx
+++ b/CORRFW/AliCFTrackQualityCuts.cxx
@@ -438,6 +438,11 @@ void AliCFTrackQualityCuts::SelectionBitMap(TObject* obj)
   Bool_t isESDTrack = strcmp(obj->ClassName(),"AliESDtrack") == 0 ? kTRUE : kFALSE ;
   Bool_t isAODTrack = strcmp(obj->ClassName(),"AliAODTrack") == 0 ? kTRUE : kFALSE ;
 
+  if (!(isESDTrack || isAODTrack)) {
+    AliError("object must be an ESDtrack or an AODtrack !");
+    return;
+  }
+
   AliESDtrack * esdTrack = 0x0 ;
   AliAODTrack * aodTrack = 0x0 ; 
   if (isESDTrack) esdTrack = dynamic_cast<AliESDtrack*>(obj);
@@ -502,7 +507,7 @@ void AliCFTrackQualityCuts::SelectionBitMap(TObject* obj)
   if (nTrackletsTRD >= fMinNTrackletTRD)
     fBitmap->SetBitNumber(iCutBit,kTRUE);
   iCutBit++;
-  if (esdTrack->GetTRDntrackletsPID() >= fMinNTrackletTRDpid)
+  if (!isESDTrack || esdTrack->GetTRDntrackletsPID() >= fMinNTrackletTRDpid)
     fBitmap->SetBitNumber(iCutBit,kTRUE);
   iCutBit++;
 
@@ -520,7 +525,7 @@ void AliCFTrackQualityCuts::SelectionBitMap(TObject* obj)
   if (chi2PerTrackletTRD <= fMaxChi2PerTrackletTRD)
     fBitmap->SetBitNumber(iCutBit,kTRUE);
   iCutBit++;
-  if (esdTrack->GetTPCsignalN() >= fMinNdEdxClusterTPC)
+  if (!isESDTrack || esdTrack->GetTPCsignalN() >= fMinNdEdxClusterTPC)
     fBitmap->SetBitNumber(iCutBit,kTRUE);
   iCutBit++;
 
@@ -960,9 +965,10 @@ void AliCFTrackQualityCuts::FillHistograms(TObject* obj, Bool_t b)
 //   if (b==0 || (b==1 && fTrackCuts->GetCutVariable(2)>0)) fhQA[kCutMinFoundClusterTPC][b]->Fill((float)fractionFoundClustersTPC);
   if (b==0 || (b==1 && nClustersTPC>0)) fhQA[kCutMinFoundClusterTPC][b]->Fill((float)fractionFoundClustersTPC);
   fhQA[kCutTrackletTRD][b]->Fill((float)nTrackletsTRD);
-  fhQA[kCutTrackletTRDpid][b]->Fill((float)esdTrack->GetTRDntrackletsPID());
-  fhQA[kCutdEdxClusterTPC][b]->Fill((float)esdTrack->GetTPCsignalN());
-
+  if (isESDTrack) {
+   fhQA[kCutTrackletTRDpid][b]->Fill((float)esdTrack->GetTRDntrackletsPID());
+   fhQA[kCutdEdxClusterTPC][b]->Fill((float)esdTrack->GetTPCsignalN());
+  }
 // // // include following lines when AliESDtrackCuts is updated
 //   fhQA[kCutCovElement11][b]->Fill(fTrackCuts->GetCutVariable(6));
 //   fhQA[kCutCovElement22][b]->Fill(fTrackCuts->GetCutVariable(7));