]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGHF/vertexingHF/AliRDHFCuts.h
Usage of multi-vertexer to tag pileup events in D meson analysis (Zaida)
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliRDHFCuts.h
index 2f81ac03a0476d3a88d3736faab0edaf022c8d95..12d1ddab18ead164bcb22a5cd28646fef2d85de5 100644 (file)
@@ -3,6 +3,8 @@
 /* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
+/* $Id$ */
+
 //***********************************************************
 // Class AliRDHFCuts
 // base class for cuts on AOD reconstructed heavy-flavour decays
 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};
+  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="");
@@ -57,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;
@@ -106,6 +114,76 @@ 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.;}
   void SetRemoveTrackletOutliers(Bool_t opt) {fRemoveTrackletOutliers=opt;}
   void SetCutOnzVertexSPD(Int_t opt) {
     if(opt>=0 && opt<=2) fCutOnzVertexSPD=opt;
@@ -136,6 +214,7 @@ class AliRDHFCuts : public AliAnalysisCuts
   void SetRecomputePrimaryVertex(Bool_t opt) {fRecomputePrimVertex=opt;}
   void SetMinPtCandidate(Double_t ptCand=-1.) {fMinPtCand=ptCand; return;}
   void SetMaxPtCandidate(Double_t ptCand=1000.) {fMaxPtCand=ptCand; return;}
+  void SetMaxRapidityCandidate(Double_t ycand) {fMaxRapidityCand=ycand; return;}
   void SetOptPileup(Int_t opt=0){
     // see enum below
     fOptPileup=opt;
@@ -145,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;}
@@ -185,6 +266,10 @@ 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;}
+  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);
@@ -199,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;}
@@ -219,12 +304,18 @@ 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);
   }
+  Bool_t IsEventRejectedDueToCentralityFlattening() const {
+    return fEvRejectionBits&(1<<kCentralityFlattening);
+  }
+  Bool_t IsEventRejectedDueToTRKV0CentralityCorrel() const {
+    return fEvRejectionBits&(1<<kBadTrackV0Correl);
+  }
   Bool_t IsEventRejectedDuePhysicsSelection() const {
     return fEvRejectionBits&(1<<kPhysicsSelection);
   }
@@ -260,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:
 
@@ -315,17 +409,25 @@ class AliRDHFCuts : public AliAnalysisCuts
   Int_t  fIsSelectedPID;  // outcome of PID selection
   Double_t fMinPtCand; // minimum pt of the candidate
   Double_t fMaxPtCand; // minimum pt of the candidate
+  Double_t fMaxRapidityCand; // max rapidity of candidate (if !=-999 overrides IsInFiducialAcceptance)
   Bool_t  fKeepSignalMC; // IsSelected returns always kTRUE for MC signal
   Bool_t fIsCandTrackSPDFirst; // flag to select the track kFirst criteria for pt < ptlimit
   Double_t fMaxPtCandTrackSPDFirst; // maximum pt of the candidate for which to check if the daughters fulfill kFirst criteria
   Bool_t fApplySPDDeadPbPb2011;  // flag to apply SPD dead module map of PbPb2011
+  Double_t fMaxDiffTRKV0Centr;   // Max. difference between TRK and V0 centrality (remove TPC pileup for PbPb 2011)
   Bool_t fRemoveTrackletOutliers; // flag to apply cut on tracklets vs. centrality for 2011 data
   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
+  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,29);  // base class for cuts on AOD reconstructed heavy-flavour decays
+  ClassDef(AliRDHFCuts,34);  // base class for cuts on AOD reconstructed heavy-flavour decays
 };
 
 #endif