]>
Commit | Line | Data |
---|---|---|
379592af | 1 | #ifndef ALICFTASKVERTEXINGHF_H |
2 | #define ALICFTASKVERTEXINGHF_H | |
3 | /************************************************************************** | |
4 | * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. * | |
5 | * * | |
6 | * Author: The ALICE Off-line Project. * | |
7 | * Contributors are mentioned in the code where appropriate. * | |
8 | * * | |
9 | * Permission to use, copy, modify and distribute this software and its * | |
10 | * documentation strictly for non-commercial purposes is hereby granted * | |
11 | * without fee, provided that the above copyright notice appears in all * | |
12 | * copies and that both the copyright notice and this permission notice * | |
13 | * appear in the supporting documentation. The authors make no claims * | |
14 | * about the suitability of this software for any purpose. It is * | |
15 | * provided "as is" without express or implied warranty. * | |
16 | **************************************************************************/ | |
17 | ||
18 | //----------------------------------------------------------------------- | |
19 | // Class for HF corrections as a function of many variables and step | |
20 | // Author : C. Zampolli, CERN | |
21 | // D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it | |
22 | // Base class for HF Unfolding - agrelli@uu.nl | |
23 | //----------------------------------------------------------------------- | |
24 | ||
25 | ||
26 | #include "AliAnalysisTaskSE.h" | |
27 | #include "AliCFVertexingHF2Prong.h" | |
28 | #include "AliCFVertexingHF.h" | |
29 | ||
30 | class TH1I; | |
31 | class TParticle ; | |
32 | class TFile ; | |
33 | class TClonesArray ; | |
34 | class AliCFManager; | |
35 | class AliAODRecoDecay; | |
36 | class AliAODRecoDecayHF2Prong; | |
37 | class AliAODMCParticle; | |
38 | class THnSparse; | |
f2703bd2 | 39 | class AliRDHFCuts; |
379592af | 40 | class AliCFVertexingHF2Prong; |
6e2e4f50 | 41 | class AliCFVertexingHF3Prong; |
379592af | 42 | |
43 | class AliCFTaskVertexingHF: public AliAnalysisTaskSE { | |
44 | public: | |
45 | ||
46 | enum { | |
47 | kStepGeneratedLimAcc = 0, | |
48 | kStepGenerated = 1, | |
49 | kStepAcceptance = 2, | |
50 | kStepVertex = 3, | |
51 | kStepRefit = 4, | |
52 | kStepReconstructed = 5, | |
53 | kStepRecoAcceptance = 6, | |
54 | kStepRecoITSClusters = 7, | |
55 | kStepRecoPPR = 8, | |
56 | kStepRecoPID = 9 | |
57 | }; | |
58 | ||
59 | AliCFTaskVertexingHF(); | |
f2703bd2 | 60 | AliCFTaskVertexingHF(const Char_t* name, AliRDHFCuts* cuts); |
379592af | 61 | AliCFTaskVertexingHF& operator= (const AliCFTaskVertexingHF& c); |
62 | AliCFTaskVertexingHF(const AliCFTaskVertexingHF& c); | |
63 | virtual ~AliCFTaskVertexingHF(); | |
64 | ||
65 | // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects | |
66 | void UserCreateOutputObjects(); | |
67 | void UserExec(Option_t *option); | |
f2703bd2 | 68 | void Init(); |
69 | void LocalInit() {Init();} | |
379592af | 70 | void Terminate(Option_t *); |
71 | ||
72 | // UNFOLDING | |
73 | void SetCorrelationMatrix(THnSparse* h) {fCorrelation=h;} | |
74 | void SetAcceptanceUnf(Bool_t AcceptanceUnf) {fAcceptanceUnf = AcceptanceUnf;} | |
75 | Bool_t GetAcceptanceUnf() const {return fAcceptanceUnf;} | |
76 | ||
77 | ||
78 | // CORRECTION FRAMEWORK RELATED FUNCTIONS | |
79 | void SetCFManager(AliCFManager* io) {fCFManager = io;} // global correction manager | |
80 | AliCFManager * GetCFManager() {return fCFManager;} // get corr manager | |
81 | ||
f2703bd2 | 82 | // Setters (and getters) for the config macro |
83 | void SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;} | |
84 | Bool_t GetFillFromGenerated() const {return fFillFromGenerated;} | |
85 | void SetDecayChannel (Int_t decayChannel) {fDecayChannel = decayChannel;} | |
86 | void SetUseWeight(Bool_t useWeight){fUseWeight=useWeight;} | |
87 | Bool_t GetUseWeight() const {return fUseWeight;} | |
88 | Double_t GetWeight(Float_t pt); | |
89 | Double_t dNdptFit(Float_t pt, Double_t* par); | |
90 | ||
6e2e4f50 | 91 | void SetDselection(UShort_t originDselection) {fOriginDselection=originDselection;} |
379592af | 92 | |
93 | protected: | |
379592af | 94 | AliCFManager *fCFManager; // pointer to the CF manager |
95 | TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed | |
96 | THnSparse* fCorrelation; // response matrix for unfolding | |
97 | Int_t fCountMC; // MC particle found | |
98 | Int_t fCountAcc; // MC particle found that satisfy acceptance cuts | |
99 | Int_t fCountVertex; // Reco particle found that satisfy vertex constrained | |
100 | Int_t fCountRefit; // Reco particle found that satisfy kTPCrefit and kITSrefit | |
101 | Int_t fCountReco; // Reco particle found that satisfy cuts | |
102 | Int_t fCountRecoAcc; // Reco particle found that satisfy cuts in requested acceptance | |
103 | Int_t fCountRecoITSClusters; // Reco particle found that satisfy cuts in n. of ITS clusters | |
104 | Int_t fCountRecoPPR; // Reco particle found that satisfy cuts in PPR | |
105 | Int_t fCountRecoPID; //Reco PID step | |
106 | Int_t fEvents; // n. of events | |
f2703bd2 | 107 | Int_t fDecayChannel; // decay channel to configure the task |
379592af | 108 | Bool_t fFillFromGenerated; // flag to indicate whether data container should be filled with generated values also for reconstructed particles |
109 | UShort_t fOriginDselection; // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty | |
110 | Bool_t fAcceptanceUnf; // flag for unfolding before or after cuts. | |
f2703bd2 | 111 | AliRDHFCuts* fCuts; // cuts |
112 | Bool_t fUseWeight; //flag to decide whether to use weights != 1 when filling the container or not | |
113 | Double_t fWeight; //weight used to fill the container | |
114 | Int_t fNvar; // number of variables for the container | |
31da6b05 | 115 | TString fPartName; // D meson name |
116 | TString fDauNames; // daughter in fin state | |
117 | ClassDef(AliCFTaskVertexingHF,3); // class for HF corrections as a function of many variables | |
379592af | 118 | }; |
119 | ||
120 | #endif |