1 #ifndef ALICFVERTEXINGHF_H
2 #define ALICFVERTEXINGHF_H
5 /**************************************************************************
6 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
8 * Author: The ALICE Off-line Project. *
9 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
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 //-----------------------------------------------------------------------
27 #include "AliCFContainer.h"
28 #include "AliAODRecoDecayHF.h"
34 class AliAODMCParticle;
39 class AliESDtrackCuts;
43 class AliCFVertexingHF : public TObject {
46 enum DecayChannel{kD0toKpi = 2, kDStartoKpipi = 21, kDplustoKpipi = 31, kLctopKpi = 32, kDstoKKpi = 33, kD0toKpipipi = 4};
49 AliCFVertexingHF(TClonesArray *mcArray, UShort_t originDselection);
50 AliCFVertexingHF(const AliCFVertexingHF& c);
51 AliCFVertexingHF& operator= (const AliCFVertexingHF& c);
53 virtual ~AliCFVertexingHF();
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;}
60 virtual void SetDecay3Prong(Int_t /*decay*/){};
62 void SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
63 Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
65 void SetNVar(Int_t nVar) {fNVar = nVar;}
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;}
72 void SetMCCandidateParam(Int_t label);
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;
85 virtual Double_t GetEtaProng(Int_t iProng) const;
86 virtual Double_t GetPtProng(Int_t iProng) const;
88 Double_t GetPtCand() const {return fRecoCandidate->Pt();}
89 Double_t GetYCand() const {return fRecoCandidate->Y();}
91 Bool_t RecoAcceptStep(AliESDtrackCuts **trackCuts) const;
93 Bool_t FillUnfoldingMatrix(Double_t fill[4]) const;
95 void SetNProngs(Int_t nProngs){fProngs = nProngs;}
96 Int_t GetNProngs() const {return fProngs;}
97 void SetDselection(UShort_t originDselection);
98 UShort_t GetDselection() {return fOriginDselection;};
99 Int_t CheckReflexion(Char_t isSign);
100 Bool_t SetLabelArray();
102 void SetCentralityValue(Float_t centValue) {fCentValue = centValue;}
104 virtual void SetPtAccCut(Float_t* ptAccCut);
105 virtual void SetEtaAccCut(Float_t* etaAccCut);
106 virtual void SetAccCut(Float_t* ptAccCut, Float_t* etaAccCut);
107 virtual void SetAccCut();
109 void SetFakeSelection(Int_t fakeSel){fFakeSelection = fakeSel;}
110 Int_t GetFakeSelection(){return fFakeSelection;}
112 void SetRejectCandidateIfNotFromQuark(Bool_t opt){fRejectIfNoQuark=opt;}
116 TClonesArray *fmcArray; //mcArray candidate
117 AliAODRecoDecayHF *fRecoCandidate; // Reconstructed HF candidate
118 AliAODMCParticle *fmcPartCandidate;
120 Int_t fNVar; // get Number of variables for the container from the channel decay
121 Double_t fzPrimVertex; //Reco z primary vertex
122 Double_t fzMCVertex; //MC z primary vertex
124 Bool_t fFillFromGenerated; // flag to indicate whether data container should be filled
125 UShort_t fOriginDselection; // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
127 Bool_t fKeepDfromB; //flag for the feed down from b quark decay.
128 Bool_t fKeepDfromBOnly; // flag to keep only the charm particles that comes from beauty decays
129 Int_t fmcLabel; // results of the MatchToMC()
130 Int_t fProngs; // n. of prongs
131 Int_t* fLabelArray; //[fProngs] array of labels
133 Float_t fCentValue; // centrality value
134 Float_t* fPtAccCut; //[fProngs] array of pt cut values for the Acceptance (MC+Rec) steps
135 Float_t* fEtaAccCut; //[fProngs] array of eta (absolute value) cut values for the Acceptance (MC+Rec) steps
136 Int_t fFakeSelection; // fakes selection: 0 --> all, 1 --> non-fake, 2 --> fake
137 Float_t fFake; // variable to indicate whether the D0 was a fake or not: 0 --> fake, 1 --> MC, 2 --> non-fake
138 Bool_t fRejectIfNoQuark; // flag to remove events not geenrated with PYTHIA
140 ClassDef(AliCFVertexingHF, 6);