Updates: finer mass bin, default cuts per system (Rossella, Marcel, Norman)
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliRDHFCutsLctopKpi.h
1 #ifndef ALIRDHFCUTSLCTOPKPI_H
2 #define ALIRDHFCUTSLCTOPKPI_H
3 /* Copyright(c) 1998-2010, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */ 
7
8 //***********************************************************
9 // Class AliRDHFCutsLctopKpi
10 // class for cuts on AOD reconstructed Lc->pKpi
11 // Author: A.Dainese, andrea.dainese@pd.infn.it
12 //***********************************************************
13
14 #include "AliRDHFCuts.h"
15 #include "AliAODPidHF.h"
16 #include "AliAODRecoDecayHF3Prong.h"
17
18 class AliRDHFCutsLctopKpi : public AliRDHFCuts 
19 {
20  public:
21
22  enum EPIDStrategy {
23   kNSigma,
24   kNSigmaPbPb,
25   kCombined,
26   kCombinedSoft,
27   kNSigmaStrong,
28   kCombinedpPb,
29   kCombinedpPb2
30  };
31  enum ECutsStrategy {
32   kStandard,
33   kKF
34  };
35
36
37   AliRDHFCutsLctopKpi(const char* name="CutsLctopKpi");
38   
39   virtual ~AliRDHFCutsLctopKpi();
40
41   AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi& source);
42   AliRDHFCutsLctopKpi& operator=(const AliRDHFCutsLctopKpi& source); 
43  
44   using AliRDHFCuts::GetCutVarsForOpt;
45   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters){
46     return GetCutVarsForOpt(d,vars,nvars,pdgdaughters,0x0);
47   }
48   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters,AliAODEvent *aod);
49
50   void SetPidpion(AliAODPidHF* pidPion) { 
51       if(fPidObjpion) delete fPidObjpion;
52       fPidObjpion=new AliAODPidHF(*pidPion);
53       }
54   void SetPidprot(AliAODPidHF* pidProt) {
55       if(fPidObjprot) delete fPidObjprot;
56       fPidObjprot=new AliAODPidHF(*pidProt);
57   }
58
59   virtual void SetStandardCutsPP2010();
60   virtual void SetStandardCutsPbPb2010();
61   virtual void SetStandardCutsPbPb2011();
62   virtual void SetStandardCutsPPb2013(); 
63
64
65   AliAODPidHF* GetPidpion() const {return fPidObjpion;}
66   AliAODPidHF* GetPidprot() const {return fPidObjprot;}
67   void SetPIDStrategy(EPIDStrategy pidStrategy) {
68    fPIDStrategy=pidStrategy;
69   }
70   EPIDStrategy GetPIDStrategy() const {
71    return fPIDStrategy;
72   }
73   void SetCutsStrategy(ECutsStrategy cutsStrategy) {
74    fCutsStrategy=cutsStrategy;
75   }
76   ECutsStrategy GetCutsStrategy() const {
77    return fCutsStrategy;
78   }
79   void SetPIDThreshold(AliPID::EParticleType species,Double_t threshold) {
80    fPIDThreshold[static_cast<Int_t>(species)]=threshold;
81   }
82  Double_t GetPIDThreshold(AliPID::EParticleType species) const {
83   return fPIDThreshold[static_cast<Int_t>(species)];
84  }
85  Bool_t GetUseSpecialCut(){return fUseSpecialCut;}
86  void SetUseSpecialCut(Bool_t useSpecialCut=kTRUE){fUseSpecialCut=useSpecialCut;}
87
88   using AliRDHFCuts::IsSelected;
89   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel)
90                            {return IsSelected(obj,selectionLevel,0);}
91   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod);
92   using AliRDHFCuts::IsSelectedPID;
93   virtual Int_t IsSelectedPID(AliAODRecoDecayHF* obj);
94   Int_t IsSelectedCombinedPID(AliAODRecoDecayHF* obj);
95   Int_t IsSelectedCombinedPIDSoft(AliAODRecoDecayHF* obj);
96   Int_t IsSelectedCombinedPIDpPb(AliAODRecoDecayHF* obj);
97   Int_t IsSelectedCombinedPIDpPb2(AliAODRecoDecayHF* obj);
98   Int_t IsSelectedPIDStrong(AliAODRecoDecayHF* obj);
99   Int_t IsSelectedNSigmaPbPb(AliAODRecoDecayHF* obj);
100   Int_t CombinePIDCuts (Int_t returnvalue, Int_t returnvaluePID) const;
101
102   virtual Bool_t IsInFiducialAcceptance(Double_t pt,Double_t y) const;
103   
104   Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
105   Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(11,iPtBin)] : 1.e6);}
106
107   void SetUseImpParProdCorrCut(Bool_t use){
108     fUseImpParProdCorrCut=use;
109   }
110   Bool_t GetUseImpParProdCorrCut() const {
111     return fUseImpParProdCorrCut;
112   }
113
114   AliKFParticle* ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field,Bool_t constraint) const;
115  protected:
116   AliAODPidHF *fPidObjprot;
117   AliAODPidHF *fPidObjpion;
118   Bool_t fUseImpParProdCorrCut; //switch for cut on d0p*d0K vs. d0K*d0pi 
119
120 private:
121   EPIDStrategy fPIDStrategy;                // PIS strategy (nsigma, combined)
122   Double_t fPIDThreshold[AliPID::kSPECIES]; // PID threshold for each species
123   ECutsStrategy fCutsStrategy;              // cut strategy (standard or KF)
124   Bool_t fUseSpecialCut;
125
126   ClassDef(AliRDHFCutsLctopKpi,8);  // class for cuts on AOD reconstructed Lc->pKpi
127 };
128
129 #endif
130
131