Usage of multi-vertexer to tag pileup events in D meson analysis (Zaida)

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliRDHFCuts.h

diff --git a/PWGHF/vertexingHF/AliRDHFCuts.h b/PWGHF/vertexingHF/AliRDHFCuts.h
index 6aa9fcb8f74df6197d1969d4986dc79ad3a6a7ae..12d1ddab18ead164bcb22a5cd28646fef2d85de5 100644 (file)
--- a/PWGHF/vertexingHF/AliRDHFCuts.h
+++ b/PWGHF/vertexingHF/AliRDHFCuts.h
@@ -22,16 +22,18 @@
 class AliAODTrack;
 class AliAODRecoDecayHF;
 class AliESDVertex;
+class TF1;
+class TFormula;
 
 class AliRDHFCuts : public AliAnalysisCuts 
 {
  public:
 
-  enum ECentrality {kCentOff,kCentV0M,kCentTRK,kCentTKL,kCentCL1,kCentInvalid};
+  enum ECentrality {kCentOff,kCentV0M,kCentTRK,kCentTKL,kCentCL1,kCentZNA,kCentZPA,kCentV0A,kCentInvalid};
   enum ESelLevel {kAll,kTracks,kPID,kCandidate};
-  enum EPileup {kNoPileupSelection,kRejectPileupEvent,kRejectTracksFromPileupVertex};
+  enum EPileup {kNoPileupSelection,kRejectPileupEvent,kRejectTracksFromPileupVertex,kRejectMVPileupEvent};
   enum ESele {kD0toKpiCuts,kD0toKpiPID,kD0fromDstarCuts,kD0fromDstarPID,kDplusCuts,kDplusPID,kDsCuts,kDsPID,kLcCuts,kLcPID,kDstarCuts,kDstarPID};
-  enum ERejBits {kNotSelTrigger,kNoVertex,kTooFewVtxContrib,kZVtxOutFid,kPileupSPD,kOutsideCentrality,kPhysicsSelection,kBadSPDVertex,kZVtxSPDOutFid,kCentralityFlattening,kBadTrackV0Correl};
+  enum ERejBits {kNotSelTrigger,kNoVertex,kTooFewVtxContrib,kZVtxOutFid,kPileup,kOutsideCentrality,kPhysicsSelection,kBadSPDVertex,kZVtxSPDOutFid,kCentralityFlattening,kBadTrackV0Correl};
   enum EV0sel  {kAllV0s = 0, kOnlyOfflineV0s = 1, kOnlyOnTheFlyV0s = 2};
 
   AliRDHFCuts(const Char_t* name="RDHFCuts", const Char_t* title="");
@@ -59,7 +61,11 @@ class AliRDHFCuts : public AliAnalysisCuts
   void SetMinSPDMultiplicity(Int_t mult=0) {fMinSPDMultiplicity=mult;}  
 
   void SetTriggerMask(ULong64_t mask=0) {fTriggerMask=mask;}
-  void SetUseAnyTrigger(){fTriggerMask=AliVEvent::kAny;}
+  void SetUseOnlyOneTrigger(Bool_t onlyOne) {fUseOnlyOneTrigger=onlyOne;}
+  ULong64_t GetTriggerMask() {return fTriggerMask;}
+  Bool_t GetUseOnlyOneTrigger() {return fUseOnlyOneTrigger;}
+
+  void SetUseAnyTrigger() {fTriggerMask=AliVEvent::kAny;}
   void EnableMBTrigger(){
     fTriggerMask|=AliVEvent::kMB;
     fUseOnlyOneTrigger=kFALSE;
@@ -108,6 +114,73 @@ class AliRDHFCuts : public AliAnalysisCuts
     fTriggerMask=(AliVEvent::kEMCEJE|AliVEvent::kEMCEGA);
     fUseOnlyOneTrigger=kTRUE;
   }
+  //
+  //  Setters (helpers) for pp 2012 data
+  void SetUseInt1TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kMB;
+    fTriggerClass[0]="CINT1";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseInt7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kINT7;
+    fTriggerClass[0]="CINT7";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseInt8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kINT8;
+    fTriggerClass[0]="CINT8";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCAL7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMC7;
+    fTriggerClass[0]="CEMC7";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCAL8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMC8;
+    fTriggerClass[0]="CEMC8";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCALJET7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMCEJE;
+    fTriggerClass[0]="CEMC7EJE";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCALJET8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMCEJE;
+    fTriggerClass[0]="CEMC8EJE";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCALGA7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMCEGA;
+    fTriggerClass[0]="CEMC7EGA";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseEMCALGA8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kEMCEGA;
+    fTriggerClass[0]="CEMC8EGA";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseSPI7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kSPI7;
+    fTriggerClass[0]="CSPI7";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseSPI8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kSPI;
+    fTriggerClass[0]="CSPI8";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseHighMult7TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kHighMult;
+    fTriggerClass[0]="CSHM7";
+    fUseOnlyOneTrigger=kTRUE;
+  }
+  void SetUseHighMult8TriggerExclusivelyPP2012(){
+    fTriggerMask=AliVEvent::kHighMult;
+    fTriggerClass[0]="CSHM8";
+    fUseOnlyOneTrigger=kTRUE;
+  }
 
   void SetMaxDifferenceTRKV0Centraltity(Double_t maxd=5.) {fMaxDiffTRKV0Centr=maxd;}
   void SetNotUseCutOnTRKVsV0Centraltity() {fMaxDiffTRKV0Centr=-1.;}
@@ -151,7 +224,9 @@ class AliRDHFCuts : public AliAnalysisCuts
     fMinContrPileup=minContrib;
     fMinDzPileup=minDz;
   }
-
+  void SetMinCrossedRowsTPCPtDep(const char *rows="");
+  void SetMinRatioClsOverCrossRowsTPC(Float_t ratio=0.) {fCutRatioClsOverCrossRowsTPC = ratio;}
+  void SetMinRatioSignalNOverCrossRowsTPC(Float_t ratio=0.) {fCutRatioSignalNOverCrossRowsTPC = ratio;}
 
   AliAODPidHF* GetPidHF() const {return fPidHF;}
   Float_t *GetPtBinLimits() const {return fPtBinLimits;}
@@ -192,6 +267,9 @@ class AliRDHFCuts : public AliAnalysisCuts
   Double_t GetMaxPtCandidate() const {return fMaxPtCand;}
   TH1F *GetHistoForCentralityFlattening(){return fHistCentrDistr;}
   void SetUseCentralityFlatteningInMC(Bool_t opt){fUseCentrFlatteningInMC=opt;}
+  const char* GetMinCrossedRowsTPCPtDep() const {return fCutMinCrossedRowsTPCPtDep;}
+  Float_t GetMinRatioClsOverCrossRowsTPC() const {return fCutRatioClsOverCrossRowsTPC;}
+  Float_t GetMinRatioSignalNOverCrossRowsTPC() const {return fCutRatioSignalNOverCrossRowsTPC;}
   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);
@@ -206,7 +284,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent* /*aod*/)
                 {return IsSelected(obj,selectionLevel);}
   Int_t PtBin(Double_t pt) const;
-  void PrintAll()const;
+  virtual void PrintAll()const;
   void PrintTrigger() const;
 
   virtual Bool_t IsInFiducialAcceptance(Double_t /*pt*/,Double_t /*y*/) const {return kTRUE;}
@@ -226,8 +304,8 @@ class AliRDHFCuts : public AliAnalysisCuts
   Bool_t IsEventRejectedDueToZVertexOutsideFiducialRegion() const {
     return fEvRejectionBits&(1<<kZVtxOutFid);
   }
-  Bool_t IsEventRejectedDueToPileupSPD() const {
-    return fEvRejectionBits&(1<<kPileupSPD);
+  Bool_t IsEventRejectedDueToPileup() const {
+    return fEvRejectionBits&(1<<kPileup);
   }
   Bool_t IsEventRejectedDueToCentrality() const {
     return fEvRejectionBits&(1<<kOutsideCentrality);
@@ -273,6 +351,9 @@ class AliRDHFCuts : public AliAnalysisCuts
   Bool_t IsSelectCandTrackSPDFirst() const { return fIsCandTrackSPDFirst; }
   Double_t IsMaxCandTrackSPDFirst() const { return fMaxPtCandTrackSPDFirst; }
 
+  Bool_t CheckPtDepCrossedRows(TString rows,Bool_t print=kFALSE) const;
+
+
 
  protected:
 
@@ -340,8 +421,13 @@ class AliRDHFCuts : public AliAnalysisCuts
   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
+  Float_t fCutRatioClsOverCrossRowsTPC; // min. value ratio NTPCClusters/NTPCCrossedRows, cut if !=0
+  Float_t fCutRatioSignalNOverCrossRowsTPC;   // min. value ratio TPCPointsUsedForPID/NTPCCrossedRows, cut if !=0 
+  TString fCutMinCrossedRowsTPCPtDep; // pT-dep cut in TPC minimum n crossed rows
+  TFormula *f1CutMinNCrossedRowsTPCPtDep; // pT-dep cut in TPC minimum n crossed rows
+ 
 
-  ClassDef(AliRDHFCuts,32);  // base class for cuts on AOD reconstructed heavy-flavour decays
+  ClassDef(AliRDHFCuts,34);  // base class for cuts on AOD reconstructed heavy-flavour decays
 };
 
 #endif