X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PWGHF%2FvertexingHF%2FAliRDHFCuts.h;h=cfb9b6383d67cacb26125385da4c4019cca5eaef;hb=a39671d19ab8e136a8d0652622fb65c9875de369;hp=a0bbdca9357128a08bab2b2ddf58cb854ed8cce8;hpb=fdd5a897a061e4623b3d14592c8646a273d70fa8;p=u%2Fmrichter%2FAliRoot.git diff --git a/PWGHF/vertexingHF/AliRDHFCuts.h b/PWGHF/vertexingHF/AliRDHFCuts.h index a0bbdca9357..cfb9b6383d6 100644 --- a/PWGHF/vertexingHF/AliRDHFCuts.h +++ b/PWGHF/vertexingHF/AliRDHFCuts.h @@ -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 @@ -20,16 +22,20 @@ 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}; + 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=""); virtual ~AliRDHFCuts(); @@ -45,13 +51,21 @@ class AliRDHFCuts : public AliAnalysisCuts void SetMinCentrality(Float_t minCentrality=0.) {fMinCentrality=minCentrality;} void SetMaxCentrality(Float_t maxCentrality=100.) {fMaxCentrality=maxCentrality;} void SetMinVtxType(Int_t type=3) {fMinVtxType=type;} + void SetUseEventsWithOnlySPDVertex(Bool_t flag=kTRUE){ + if(flag) fMinVtxType=1; + else fMinVtxType=3; + } void SetMinVtxContr(Int_t contr=1) {fMinVtxContr=contr;} void SetMaxVtxRdChi2(Float_t chi2=1e6) {fMaxVtxRedChi2=chi2;} void SetMaxVtxZ(Float_t z=1e6) {fMaxVtxZ=z;} 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; @@ -100,9 +114,84 @@ class AliRDHFCuts : public AliAnalysisCuts fTriggerMask=(AliVEvent::kEMCEJE|AliVEvent::kEMCEGA); fUseOnlyOneTrigger=kTRUE; } + // + // Setters (helpers) for pp 2012 data + void SetUseInt1TriggerPP2012(){ + fTriggerMask=AliVEvent::kMB; + fTriggerClass[0]="CINT1"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseInt7TriggerPP2012(){ + fTriggerMask=AliVEvent::kINT7; + fTriggerClass[0]="CINT7"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseInt8TriggerPP2012(){ + fTriggerMask=AliVEvent::kINT8; + fTriggerClass[0]="CINT8"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCAL7TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMC7; + fTriggerClass[0]="CEMC7"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCAL8TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMC8; + fTriggerClass[0]="CEMC8"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCALJET7TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMCEJE; + fTriggerClass[0]="CEMC7EJE"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCALJET8TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMCEJE; + fTriggerClass[0]="CEMC8EJE"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCALGA7TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMCEGA; + fTriggerClass[0]="CEMC7EGA"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseEMCALGA8TriggerPP2012(){ + fTriggerMask=AliVEvent::kEMCEGA; + fTriggerClass[0]="CEMC8EGA"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseSPI7TriggerPP2012(){ + fTriggerMask=AliVEvent::kSPI7; + fTriggerClass[0]="CSPI7"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseSPI8TriggerPP2012(){ + fTriggerMask=AliVEvent::kSPI; + fTriggerClass[0]="CSPI8"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseHighMult7TriggerPP2012(){ + fTriggerMask=AliVEvent::kHighMult; + fTriggerClass[0]="CSHM7"; + fUseOnlyOneTrigger=kFALSE; + } + void SetUseHighMult8TriggerPP2012(){ + fTriggerMask=AliVEvent::kHighMult; + fTriggerClass[0]="CSHM8"; + fUseOnlyOneTrigger=kFALSE; + } + + 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; + else AliError("Wrong option for cut on zVertexSPD"); + } void SetTriggerClass(TString trclass0, TString trclass1="") {fTriggerClass[0]=trclass0; fTriggerClass[1]=trclass1;} void ApplySPDDeadPbPb2011(){fApplySPDDeadPbPb2011=kTRUE;} + void ApplySPDMisalignedCutPP2012(){fApplySPDMisalignedPP2012=kTRUE;} void SetVarsForOpt(Int_t nVars,Bool_t *forOpt); void SetGlobalIndex(){fGlobalIndex=fnVars*fnPtBins;} void SetGlobalIndex(Int_t nVars,Int_t nptBins){fnVars=nVars; fnPtBins=nptBins; SetGlobalIndex();} @@ -115,24 +204,29 @@ class AliRDHFCuts : public AliAnalysisCuts void SetUsePID(Bool_t flag=kTRUE) {fUsePID=flag; return;} void SetUseAOD049(Bool_t flag=kTRUE) {fUseAOD049=flag; return;} void SetKinkRejection(Bool_t flag=kTRUE) {fKinkReject=flag; return;} + void SetUseTrackSelectionWithFilterBits(Bool_t flag=kTRUE){ + fUseTrackSelectionWithFilterBits=flag; return;} void SetUseCentrality(Int_t flag=1); // see enum below void SetPidHF(AliAODPidHF* pidObj) { if(fPidHF) delete fPidHF; fPidHF=new AliAODPidHF(*pidObj); } void SetRemoveDaughtersFromPrim(Bool_t removeDaughtersPrim) {fRemoveDaughtersFromPrimary=removeDaughtersPrim;} - 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; } + void SetHistoForCentralityFlattening(TH1F *h,Double_t minCentr,Double_t maxCentr,Double_t centrRef=0.,Int_t switchTRand=0); void ConfigurePileupCuts(Int_t minContrib=3, Float_t minDz=0.6){ 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;} @@ -150,6 +244,7 @@ class AliRDHFCuts : public AliAnalysisCuts Bool_t *GetIsUpperCut() const {return fIsUpperCut;} AliESDtrackCuts *GetTrackCuts() const {return fTrackCuts;} virtual AliESDtrackCuts *GetTrackCutsSoftPi() const {return 0;} + virtual AliESDtrackCuts *GetTrackCutsV0daughters() const {return 0;} virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters) = 0; virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters,AliAODEvent * /*aod*/) {return GetCutVarsForOpt(d,vars,nvars,pdgdaughters);} @@ -158,6 +253,11 @@ class AliRDHFCuts : public AliAnalysisCuts Bool_t GetIsUsePID() const {return fUsePID;} Bool_t GetUseAOD049() const {return fUseAOD049;} Bool_t GetUseKinkRejection() const {return fKinkReject;} + Bool_t GetUseEventsWithOnlySPDVertex() const{ + if(fMinVtxType==1 || fMinVtxType==2) return kTRUE; + return kFALSE; + } + Bool_t GetUseTrackSelectionWithFilterBits() const{return fUseTrackSelectionWithFilterBits;} Bool_t GetIsPrimaryWithoutDaughters() const {return fRemoveDaughtersFromPrimary;} Bool_t GetOptPileUp() const {return fOptPileup;} Int_t GetUseCentrality() const {return fUseCentrality;} @@ -165,19 +265,26 @@ class AliRDHFCuts : public AliAnalysisCuts Float_t GetMaxCentrality() const {return fMaxCentrality;} 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); + Bool_t IsEventSelectedForCentrFlattening(Float_t centvalue); Bool_t IsEventSelected(AliVEvent *event); Bool_t AreDaughtersSelected(AliAODRecoDecayHF *rd) const; Bool_t IsDaughterSelected(AliAODTrack *track,const AliESDVertex *primary,AliESDtrackCuts *cuts) const; virtual Int_t IsSelectedPID(AliAODRecoDecayHF * /*rd*/) {return 1;} + void SetupPID(AliVEvent *event); virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel) = 0; 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;} @@ -197,12 +304,18 @@ class AliRDHFCuts : public AliAnalysisCuts Bool_t IsEventRejectedDueToZVertexOutsideFiducialRegion() const { return fEvRejectionBits&(1<