Changes to compute efficiencies for particle or antiparticle separately (ChiaraZ...
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFVertexingHF.h
1 #ifndef ALICFVERTEXINGHF_H
2 #define ALICFVERTEXINGHF_H
3
4
5 /**************************************************************************
6  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
7  *                                                                        *
8  * Author: The ALICE Off-line Project.                                    *
9  * Contributors are mentioned in the code where appropriate.              *
10  *                                                                        *
11  * Permission to use, copy, modify and distribute this software and its   *
12  * documentation strictly for non-commercial purposes is hereby granted   *
13  * without fee, provided that the above copyright notice appears in all   *
14  * copies and that both the copyright notice and this permission notice   *
15  * appear in the supporting documentation. The authors make no claims     *
16  * about the suitability of this software for any purpose. It is          *
17  * provided "as is" without express or implied warranty.                  *
18  **************************************************************************/
19
20 //-----------------------------------------------------------------------
21 // Class for HF corrections as a function of many variables and step 
22 // Author : C. Zampolli, CERN
23 // D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it
24 // Base class for HF Unfolding - agrelli@uu.nl
25 //-----------------------------------------------------------------------
26
27 #include "AliCFContainer.h"
28 #include "AliAODRecoDecayHF.h"
29
30 class TH1I;
31 class TParticle ;
32 class TFile ;
33 class TClonesArray ;
34 class AliAODMCParticle;
35 class AliAODMCHeader;
36 class AliAODEvent;
37 class THnSparse;
38 class TClonesArray;
39 class AliESDtrackCuts;
40
41
42
43 class AliCFVertexingHF : public TObject {
44         public:
45
46         enum DecayChannel{kD0toKpi = 2, kDStartoKpipi = 21, kDplustoKpipi = 31, kLctopKpi = 32, kDstoKKpi = 33, kD0toKpipipi = 4};
47
48         AliCFVertexingHF() ;
49         AliCFVertexingHF(TClonesArray *mcArray, UShort_t originDselection);
50         AliCFVertexingHF(const AliCFVertexingHF& c);
51         AliCFVertexingHF& operator= (const AliCFVertexingHF& c);
52
53         virtual ~AliCFVertexingHF();
54                 
55         virtual Bool_t GetGeneratedValuesFromMCParticle(Double_t* /*vectorMC*/) {return kFALSE;} 
56         virtual Bool_t GetRecoValuesFromCandidate(Double_t* /*vectorReco*/) const {return kFALSE;}
57         virtual Bool_t CheckMCChannelDecay() const {return kFALSE;}
58         virtual Bool_t SetRecoCandidateParam(AliAODRecoDecayHF* /*recoCand*/) {return kFALSE;}
59
60         virtual void SetDecay3Prong(Int_t /*decay*/){};
61         
62         void   SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
63         Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
64                 
65         void  SetNVar(Int_t nVar) {fNVar = nVar;}  
66
67         void  SetRecoPrimVertex (Double_t zPrimVertex) {fzPrimVertex = zPrimVertex;}
68         void  SetMCPrimaryVertex (Double_t zMCVertex){fzMCVertex = zMCVertex;}
69         void  SetMCLabel (Int_t mcLabel) {fmcLabel = mcLabel;}
70         Int_t GetMCLabel () const {return  fmcLabel;}
71                 
72         void   SetMCCandidateParam(Int_t label);
73
74         Int_t  MCcquarkCounting(AliAODMCParticle* mcPart) const; 
75         Bool_t CheckMCPartFamily(AliAODMCParticle */*mcPart*/, TClonesArray */*mcArray*/) const;
76         //      Int_t  CheckOrigin(AliAODMCParticle* mcPart) const;
77         Int_t  CheckOrigin() const;
78         Bool_t CheckMCDaughters() const;
79         Bool_t FillMCContainer(Double_t *containerInputMC); 
80         Bool_t FillRecoContainer(Double_t *containerInput); 
81         Bool_t MCAcceptanceStep() const;
82         Bool_t MCRefitStep(AliAODEvent *aodEvent, AliESDtrackCuts *trackCuts) const;
83         Bool_t RecoStep();
84
85         Double_t GetEtaProng(Int_t iProng) const;
86         Double_t GetPtProng(Int_t iProng) const;
87
88         Double_t GetPtCand() const {return fRecoCandidate->Pt();}
89         Double_t GetYCand() const {return fRecoCandidate->Y();}
90
91         Bool_t RecoAcceptStep(AliESDtrackCuts *trackCuts) const;
92         
93         Bool_t FillUnfoldingMatrix(Double_t *fill) const;
94         
95         void SetNProngs(Int_t nProngs){fProngs = nProngs;}
96         void SetDselection(UShort_t originDselection); 
97         UShort_t GetDselection() {return fOriginDselection;}; 
98         Int_t CheckReflexion(Char_t isSign);
99         Bool_t SetLabelArray();
100
101         protected:
102         
103         TClonesArray      *fmcArray;               //mcArray candidate
104         AliAODRecoDecayHF *fRecoCandidate;         // Reconstructed HF candidate 
105         AliAODMCParticle  *fmcPartCandidate;
106         Int_t fNDaughters;
107         Int_t fNVar;                // get Number of variables for the container from the channel decay
108         Double_t fzPrimVertex;       //Reco z primary vertex    
109         Double_t fzMCVertex;         //MC z primary vertex
110         
111         Bool_t fFillFromGenerated;   //  flag to indicate whether data container should be filled  
112         UShort_t fOriginDselection;      // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
113         
114         Bool_t fKeepDfromB;           //flag for the feed down from b quark decay.                      
115         Bool_t fKeepDfromBOnly;       // flag to keep only the charm particles that comes from beauty decays
116         Int_t fmcLabel;              // results of the MatchToMC()
117         Int_t fProngs;               // n. of prongs    
118         Int_t* fLabelArray;          //[fProngs]  array of labels
119
120         ClassDef(AliCFVertexingHF, 2);
121         
122 };
123
124 #endif