From 30399c338e5fc5b87c9185cf4ed06a18a063add6 Mon Sep 17 00:00:00 2001
From: fprino <fprino@f7af4fe6-9843-0410-8265-dc069ae4e863>
Date: Fri, 18 Jan 2013 14:11:18 +0000
Subject: [PATCH] Disable centrality flattening in the MC

---
 PWGHF/vertexingHF/AliRDHFCuts.cxx | 8 +++++++-
 PWGHF/vertexingHF/AliRDHFCuts.h   | 4 +++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/PWGHF/vertexingHF/AliRDHFCuts.cxx b/PWGHF/vertexingHF/AliRDHFCuts.cxx
index 734f35a1ac1..f3b2cf9d0f3 100644
--- a/PWGHF/vertexingHF/AliRDHFCuts.cxx
+++ b/PWGHF/vertexingHF/AliRDHFCuts.cxx
@@ -99,6 +99,7 @@ fRemoveTrackletOutliers(kFALSE),
 fCutOnzVertexSPD(0),
 fKinkReject(kFALSE),
   fUseTrackSelectionWithFilterBits(kTRUE),
+  fUseCentrFlatteningInMC(kFALSE),
   fHistCentrDistr(0x0)
 {
   //
@@ -156,6 +157,7 @@ AliRDHFCuts::AliRDHFCuts(const AliRDHFCuts &source) :
   fCutOnzVertexSPD(source.fCutOnzVertexSPD),
   fKinkReject(source.fKinkReject),
   fUseTrackSelectionWithFilterBits(source.fUseTrackSelectionWithFilterBits),
+  fUseCentrFlatteningInMC(source.fUseCentrFlatteningInMC),
   fHistCentrDistr(0x0)
 {
   //
@@ -228,6 +230,7 @@ AliRDHFCuts &AliRDHFCuts::operator=(const AliRDHFCuts &source)
   fKinkReject=source.fKinkReject;
   fUseTrackSelectionWithFilterBits=source.fUseTrackSelectionWithFilterBits;
   if(fHistCentrDistr) delete fHistCentrDistr;
+  fUseCentrFlatteningInMC=source.fUseCentrFlatteningInMC;
   if(source.fHistCentrDistr)fHistCentrDistr=(TH1F*)(source.fHistCentrDistr->Clone());
 
   if(source.GetTrackCuts()) {delete fTrackCuts; fTrackCuts=new AliESDtrackCuts(*(source.GetTrackCuts()));}
@@ -548,7 +551,10 @@ Bool_t AliRDHFCuts::IsEventSelected(AliVEvent *event) {
   // centrality selection
   if (fUseCentrality!=kCentOff) {  
     Int_t rejection=IsEventSelectedInCentrality(event);    
-    if(rejection>1){      
+    Bool_t okCent=kFALSE;
+    if(rejection==0) okCent=kTRUE;
+    if(isMC && rejection==4 && !fUseCentrFlatteningInMC) okCent=kTRUE;
+    if(!okCent){      
       if(accept) fWhyRejection=rejection;      
       if(fWhyRejection==4)fEvRejectionBits+=1<<kCentralityFlattening;
       else fEvRejectionBits+=1<<kOutsideCentrality;
diff --git a/PWGHF/vertexingHF/AliRDHFCuts.h b/PWGHF/vertexingHF/AliRDHFCuts.h
index 2f81ac03a04..44ea3ff78bb 100644
--- a/PWGHF/vertexingHF/AliRDHFCuts.h
+++ b/PWGHF/vertexingHF/AliRDHFCuts.h
@@ -185,6 +185,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   Double_t GetMinPtCandidate() const {return fMinPtCand;}
   Double_t GetMaxPtCandidate() const {return fMaxPtCand;}
   TH1F *GetHistoForCentralityFlattening(){return fHistCentrDistr;}
+  void SetUseCentralityFlatteningInMC(Bool_t opt){fUseCentrFlatteningInMC=opt;}
   Bool_t IsSelected(TObject *obj) {return IsSelected(obj,AliRDHFCuts::kAll);}
   Bool_t IsSelected(TList *list) {if(!list) return kTRUE; return kFALSE;}
   Int_t  IsEventSelectedInCentrality(AliVEvent *event);
@@ -323,9 +324,10 @@ class AliRDHFCuts : public AliAnalysisCuts
   Int_t fCutOnzVertexSPD; // cut on zSPD vertex to remove outliers in centrality vs. tracklets (0=no cut, 1= cut at 12 cm, 2= cut on difference to z of vtx tracks
   Bool_t fKinkReject; // flag to reject kink daughters
   Bool_t fUseTrackSelectionWithFilterBits; // flag to enable/disable the check on filter bits
+  Bool_t fUseCentrFlatteningInMC; // flag for enabling/diabling centrality flattening in MC
   TH1F *fHistCentrDistr;   // histogram with reference centrality distribution for centrality distribution flattening
 
-  ClassDef(AliRDHFCuts,29);  // base class for cuts on AOD reconstructed heavy-flavour decays
+  ClassDef(AliRDHFCuts,30);  // base class for cuts on AOD reconstructed heavy-flavour decays
 };
 
 #endif
-- 
2.39.3