]>
Commit | Line | Data |
---|---|---|
72c0a987 | 1 | //-*- Mode: C++ -*- |
2 | // $Id$ | |
3 | ||
4 | //* This file is property of and copyright by the ALICE Project * | |
5 | //* ALICE Experiment at CERN, All rights reserved. * | |
6 | //* See cxx source for full Copyright notice * | |
7 | ||
8 | /// @file AliAnalysisTaskDxHFECorrelation.h | |
9 | /// @author Sedat Altinpinar, Hege Erdal, Matthias Richter | |
10 | /// @date 2012-03-19 | |
11 | /// @brief AnalysisTask D0 - HFE correlation | |
12 | /// | |
13 | ||
14 | #ifndef ALIANALYSISTASKDXHFECORRELATION_H | |
15 | #define ALIANALYSISTASKDXHFECORRELATION_H | |
16 | ||
17 | #include "AliAnalysisTaskSE.h" | |
18 | #include "TString.h" | |
9535cec9 | 19 | |
20 | class AliPID; | |
21 | class AliPIDResponse; | |
72c0a987 | 22 | class TList; |
23 | class AliDxHFEParticleSelection; | |
9535cec9 | 24 | class AliDxHFEParticleSelectionD0; |
25 | class AliDxHFEParticleSelectionEl; | |
72c0a987 | 26 | class AliDxHFECorrelation; |
9535cec9 | 27 | class AliAnalysisCuts; |
28 | class AliHFEpid; | |
29 | class AliHFEcuts; | |
dfe96b90 | 30 | class AliHFAssociatedTrackCuts; |
72c0a987 | 31 | |
32 | /** | |
33 | * @class AliAnalysisTaskDxHFECorrelation | |
34 | * Task for D0-HFE correleations | |
35 | */ | |
36 | class AliAnalysisTaskDxHFECorrelation : public AliAnalysisTaskSE { | |
37 | public: | |
38 | /// constructor | |
39 | AliAnalysisTaskDxHFECorrelation(const char* opt=""); | |
40 | /// destructor | |
41 | virtual ~AliAnalysisTaskDxHFECorrelation(); | |
42 | ||
43 | /// inherited from AliAnalysisTask: connect tree branches at input slots | |
44 | virtual void ConnectInputData(Option_t *option="") { | |
45 | return AliAnalysisTaskSE::ConnectInputData(option); | |
46 | } | |
47 | ||
48 | /// inherited from AliAnalysisTaskSE: create output objects | |
49 | virtual void UserCreateOutputObjects(); | |
50 | /// inherited from AliAnalysisTaskSE: event processing | |
51 | virtual void UserExec(Option_t*); | |
52 | /// inherited from AliAnalysisTask: called in SlaveTerminate phase for each task | |
53 | virtual void FinishTaskOutput(); | |
54 | /// inherited from AliAnalysisTask: final step | |
55 | virtual void Terminate(Option_t*); | |
56 | ||
57 | /// set options | |
58 | void SetOption(const char* opt) { fOption = opt; } | |
9535cec9 | 59 | void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;} |
60 | virtual void SetUseMC(Bool_t useMC){fUseMC=useMC;} | |
61 | virtual void SetCutsD0(AliAnalysisCuts* cuts){fCutsD0=cuts;} | |
62 | virtual void SetCutsHFE(AliHFEcuts* cuts){fCutsHFE=cuts;} | |
dfe96b90 | 63 | void SetCuts(AliAnalysisCuts* cuts){fCuts=cuts;} |
64 | void SetUseEventMixing(Bool_t useMixing) {fUseEventMixing=useMixing;} | |
65 | void SetSystem(Bool_t system){fSystem=system;} | |
9535cec9 | 66 | |
72c0a987 | 67 | /// overloaded from TObject: get option |
68 | virtual Option_t* GetOption() const { return fOption;} | |
9535cec9 | 69 | Int_t GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;} |
70 | Bool_t GetUseMC() const {return fUseMC;} | |
72c0a987 | 71 | |
72 | protected: | |
73 | ||
74 | private: | |
75 | /// copy constructor prohibited: might change | |
76 | AliAnalysisTaskDxHFECorrelation(const AliAnalysisTaskDxHFECorrelation&); | |
77 | /// assignment operator prohibited: might change | |
78 | AliAnalysisTaskDxHFECorrelation& operator=(const AliAnalysisTaskDxHFECorrelation&); | |
79 | ||
80 | int DefineSlots(); | |
81 | ||
82 | TList* fOutput; //! list send on output slot 1 | |
9535cec9 | 83 | TString fOption; // option string |
84 | AliDxHFECorrelation* fCorrelation; // correlation worker class | |
85 | AliDxHFEParticleSelection* fD0s; // selection of D0s | |
86 | AliDxHFEParticleSelection* fElectrons; // selection of electrons | |
87 | AliAnalysisCuts *fCutsD0; // Cuts D0 | |
88 | AliHFEcuts *fCutsHFE; // Cuts HFE | |
dfe96b90 | 89 | AliAnalysisCuts *fCuts; // Cuts which holds info for AliHFCorrelator |
9535cec9 | 90 | AliHFEpid *fPID; // dummy |
91 | Int_t fFillOnlyD0D0bar; // flag to set what to fill (0 = both, 1 = D0 only, 2 = D0bar only) | |
92 | Bool_t fUseMC; // use MC info | |
dfe96b90 | 93 | Bool_t fUseEventMixing; // Run Event Mixing analysis |
94 | Int_t fSystem; // Which system pp/PbPb | |
95 | TObjArray *fSelectedD0s; // Array for selected D0s | |
96 | TObjArray *fSelectedElectrons; // Array for selected Electrons | |
9535cec9 | 97 | |
72c0a987 | 98 | |
dfe96b90 | 99 | ClassDef(AliAnalysisTaskDxHFECorrelation, 3); |
72c0a987 | 100 | }; |
101 | ||
102 | #endif |