New common interface to the CORRFW for all charm analyses (Davide, ChiaraZ)
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliCFTaskVertexingHF.h
CommitLineData
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
30class TH1I;
31class TParticle ;
32class TFile ;
33class TClonesArray ;
34class AliCFManager;
35class AliAODRecoDecay;
36class AliAODRecoDecayHF2Prong;
37class AliAODMCParticle;
38class THnSparse;
39class AliRDHFCutsD0toKpi;
40class AliCFVertexingHF2Prong;
41
42class AliCFTaskVertexingHF: public AliAnalysisTaskSE {
43public:
44
45 enum {
46 kStepGeneratedLimAcc = 0,
47 kStepGenerated = 1,
48 kStepAcceptance = 2,
49 kStepVertex = 3,
50 kStepRefit = 4,
51 kStepReconstructed = 5,
52 kStepRecoAcceptance = 6,
53 kStepRecoITSClusters = 7,
54 kStepRecoPPR = 8,
55 kStepRecoPID = 9
56 };
57
58 AliCFTaskVertexingHF();
59 AliCFTaskVertexingHF(const Char_t* name, AliRDHFCutsD0toKpi* cuts);
60 AliCFTaskVertexingHF& operator= (const AliCFTaskVertexingHF& c);
61 AliCFTaskVertexingHF(const AliCFTaskVertexingHF& c);
62 virtual ~AliCFTaskVertexingHF();
63
64 // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
65 void UserCreateOutputObjects();
66 void UserExec(Option_t *option);
67 void Terminate(Option_t *);
68
69 // UNFOLDING
70 void SetCorrelationMatrix(THnSparse* h) {fCorrelation=h;}
71 void SetAcceptanceUnf(Bool_t AcceptanceUnf) {fAcceptanceUnf = AcceptanceUnf;}
72 Bool_t GetAcceptanceUnf() const {return fAcceptanceUnf;}
73
74
75 // CORRECTION FRAMEWORK RELATED FUNCTIONS
76 void SetCFManager(AliCFManager* io) {fCFManager = io;} // global correction manager
77 AliCFManager * GetCFManager() {return fCFManager;} // get corr manager
78
79 void SetPDG(Int_t code) {fPDG = code; } // defines the PDG code of searched HF
80
81 // Setters for teh config macro
82 void SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
83 Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
84
85
86 //void SetDselection(UShort_t originDselection);
87
88protected:
89 Int_t fPDG; // PDG code of searched V0's
90
91
92 AliCFManager *fCFManager; // pointer to the CF manager
93 TH1I *fHistEventsProcessed; //! simple histo for monitoring the number of events processed
94 THnSparse* fCorrelation; // response matrix for unfolding
95 Int_t fCountMC; // MC particle found
96 Int_t fCountAcc; // MC particle found that satisfy acceptance cuts
97 Int_t fCountVertex; // Reco particle found that satisfy vertex constrained
98 Int_t fCountRefit; // Reco particle found that satisfy kTPCrefit and kITSrefit
99 Int_t fCountReco; // Reco particle found that satisfy cuts
100 Int_t fCountRecoAcc; // Reco particle found that satisfy cuts in requested acceptance
101 Int_t fCountRecoITSClusters; // Reco particle found that satisfy cuts in n. of ITS clusters
102 Int_t fCountRecoPPR; // Reco particle found that satisfy cuts in PPR
103 Int_t fCountRecoPID; //Reco PID step
104 Int_t fEvents; // n. of events
105 Bool_t fFillFromGenerated; // flag to indicate whether data container should be filled with generated values also for reconstructed particles
106 UShort_t fOriginDselection; // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
107 Bool_t fAcceptanceUnf; // flag for unfolding before or after cuts.
108 AliRDHFCutsD0toKpi* fCuts; // cuts
109
110
111 ClassDef(AliCFTaskVertexingHF,1); // class for HF corrections as a function of many variables
112};
113
114#endif