Set the TPC BB properly in case of MC

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliRDHFCuts.cxx

diff --git a/PWG3/vertexingHF/AliRDHFCuts.cxx b/PWG3/vertexingHF/AliRDHFCuts.cxx
index 3dcf22ffe83a72a139586ab481d2283a8f7b2252..1a0866e40d58edd640e25e9e8353a50884e4eddb 100644 (file)
--- a/PWG3/vertexingHF/AliRDHFCuts.cxx
+++ b/PWG3/vertexingHF/AliRDHFCuts.cxx
@@ -37,6 +37,7 @@
 #include "AliAODRecoDecayHF.h"
 #include "AliAnalysisVertexingHF.h"
 #include "AliAODMCHeader.h"
+#include "AliAODMCParticle.h"
 #include "AliRDHFCuts.h"
 
 ClassImp(AliRDHFCuts)
@@ -229,10 +230,15 @@ Bool_t AliRDHFCuts::IsEventSelected(AliVEvent *event) {
 
   fWhyRejection=0;
 
+  // check if it's MC
+  Bool_t isMC=kFALSE;
+  TClonesArray *mcArray = (TClonesArray*)((AliAODEvent*)event)->GetList()->FindObject(AliAODMCParticle::StdBranchName());
+  if(mcArray) isMC=kTRUE;
+
   // trigger class
   TString firedTriggerClasses=((AliAODEvent*)event)->GetFiredTriggerClasses();
-  // don't do for PbPb 2010 data
-  if(event->GetRunNumber()<136851 || event->GetRunNumber()>139517) {
+  // don't do for MC and for PbPb 2010 data
+  if(!isMC && (event->GetRunNumber()<136851 || event->GetRunNumber()>139517)) {
     if(!firedTriggerClasses.Contains(fTriggerClass.Data())) {
       fWhyRejection=5;
       return kFALSE;
@@ -267,15 +273,21 @@ Bool_t AliRDHFCuts::IsEventSelected(AliVEvent *event) {
     return kFALSE;
   } 
 
-  // switch to settings for 1-pad cls in TPC
+  // settings for the TPC dE/dx BB parameterization
   if(fPidHF) {
+    // pp, from LHC10d onwards
     if((event->GetRunNumber()>121693 && event->GetRunNumber()<136851) ||
        event->GetRunNumber()>139517)
       fPidHF->SetOnePad(kTRUE);
+    // PbPb LHC10h
     if(event->GetRunNumber()>=136851 && event->GetRunNumber()<=139517) 
       fPidHF->SetPbPb(kTRUE);
+    // MC
+    if(isMC) fPidHF->SetMC(kTRUE);
   }
 
+
+
   if(fOptPileup==kRejectPileupEvent){
     Int_t cutc=(Int_t)fMinContrPileup;
     Double_t cutz=(Double_t)fMinDzPileup;