First implementation of combined Bayesian PID (Pietro, ChiaraZ)
[u/mrichter/AliRoot.git] / PWG3 / 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   AliRDHFCutsLctopKpi(const char* name="CutsLctopKpi");
23   
24   virtual ~AliRDHFCutsLctopKpi();
25
26   AliRDHFCutsLctopKpi(const AliRDHFCutsLctopKpi& source);
27   AliRDHFCutsLctopKpi& operator=(const AliRDHFCutsLctopKpi& source); 
28  
29   using AliRDHFCuts::GetCutVarsForOpt;
30   virtual void GetCutVarsForOpt(AliAODRecoDecayHF *d,Float_t *vars,Int_t nvars,Int_t *pdgdaughters);
31
32   void SetPidpion(AliAODPidHF* pidPion) { 
33       if(fPidObjpion) delete fPidObjpion;
34       fPidObjpion=new AliAODPidHF(*pidPion);
35       }
36   void SetPidprot(AliAODPidHF* pidProt) {
37       if(fPidObjprot) delete fPidObjprot;
38       fPidObjprot=new AliAODPidHF(*pidProt);
39   }
40
41   virtual void SetStandardCutsPP2010();
42   virtual void SetStandardCutsPbPb2010();
43
44   void SetRecoKF() {fRecoKF=kTRUE;}
45   Bool_t GetRecoKF() {return fRecoKF;}
46
47   AliAODPidHF* GetPidpion() const {return fPidObjpion;}
48   AliAODPidHF* GetPidprot() const {return fPidObjprot;}
49
50
51   using AliRDHFCuts::IsSelected;
52   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel)
53                            {return IsSelected(obj,selectionLevel,0);}
54   virtual Int_t IsSelected(TObject* obj,Int_t selectionLevel,AliAODEvent *aod);
55   using AliRDHFCuts::IsSelectedPID;
56   virtual Int_t IsSelectedPID(AliAODRecoDecayHF* obj);
57   Int_t IsSelectedCombinedPID(AliAODRecoDecayHF* obj);
58   Int_t CombinePIDCuts (Int_t returnvalue, Int_t returnvaluePID) const;
59   
60   Float_t GetMassCut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(0,iPtBin)] : 1.e6);}
61   Float_t GetDCACut(Int_t iPtBin=0) const { return (GetCuts() ? fCutsRD[GetGlobalIndex(11,iPtBin)] : 1.e6);}
62
63   Bool_t ReconstructKF(AliAODRecoDecayHF3Prong *d,Int_t *pdgs,Double_t field) const;
64  protected:
65   AliAODPidHF *fPidObjprot;
66   AliAODPidHF *fPidObjpion;
67   Bool_t fRecoKF;
68
69   ClassDef(AliRDHFCutsLctopKpi,3);  // class for cuts on AOD reconstructed Lc->pKpi
70 };
71
72 #endif
73