]>
Commit | Line | Data |
---|---|---|
809a4336 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
50685501 | 15 | // |
16 | // Task for Heavy Flavour Electron Analysis | |
17 | // Fills a single-inclusive electron pt-spectrum | |
18 | // For further information see implementation file | |
19 | // | |
c2690925 | 20 | #ifndef ALIANALYSISTASKHFE_H |
21 | #define ALIANALYSISTASKHFE_H | |
22 | ||
d2af20c5 | 23 | #ifndef ALIANALYSISTASKSE_H |
24 | #include "AliAnalysisTaskSE.h" | |
dbe3abbe | 25 | #endif |
26 | ||
c2690925 | 27 | #ifndef ROOT_TString |
28 | #include <TString.h> | |
29 | #endif | |
30 | ||
3a72645a | 31 | class AliHFEcontainer; |
32 | class AliHFEcollection; | |
809a4336 | 33 | class AliHFEcuts; |
3a72645a | 34 | class AliHFEelecbackground; |
259c3296 | 35 | class AliHFEmcQA; |
3a72645a | 36 | class AliHFEpid; |
37 | class AliHFEpidQAmanager; | |
259c3296 | 38 | class AliHFEsecVtx; |
3a72645a | 39 | class AliHFEsignalCuts; |
40 | class AliHFEvarManager; | |
41 | class AliHFEtaggedTrackAnalysis; | |
809a4336 | 42 | class AliCFManager; |
9bcfd1ab | 43 | class AliVEvent; |
809a4336 | 44 | class AliMCEvent; |
722347d8 | 45 | class AliVParticle; |
69ac0e6f | 46 | class AliTriggerAnalysis; |
809a4336 | 47 | class TH1I; |
48 | class TList; | |
49 | ||
d2af20c5 | 50 | class AliAnalysisTaskHFE : public AliAnalysisTaskSE{ |
dbe3abbe | 51 | public: |
9bcfd1ab | 52 | enum{ |
53 | kPIDqa = 0, | |
54 | kMCqa =1 | |
55 | }; | |
56 | enum{ | |
57 | kPriVtx = 0, | |
58 | kSecVtx = 1, | |
59 | kIsElecBackGround = 2, | |
3a72645a | 60 | kPostProcess = 3, |
61 | kDEstep = 4, | |
62 | kTaggedTrackAnalysis = 5 | |
9bcfd1ab | 63 | }; |
faee3b18 | 64 | enum CreationProcess_t{ |
65 | kSignalCharm = 0, | |
66 | kSignalBeauty = 1, | |
67 | kGammaConv = 2, | |
68 | kOther = 3 | |
69 | }; | |
dbe3abbe | 70 | AliAnalysisTaskHFE(); |
0792aa82 | 71 | AliAnalysisTaskHFE(const char * name); |
dbe3abbe | 72 | AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref); |
73 | AliAnalysisTaskHFE& operator=(const AliAnalysisTaskHFE &ref); | |
faee3b18 | 74 | virtual void Copy(TObject &o) const; |
75d81601 | 75 | virtual ~AliAnalysisTaskHFE(); |
809a4336 | 76 | |
d2af20c5 | 77 | virtual void UserCreateOutputObjects(); |
78 | virtual void UserExec(Option_t *); | |
dbe3abbe | 79 | virtual void Terminate(Option_t *); |
809a4336 | 80 | |
faee3b18 | 81 | virtual Bool_t IsEventInBinZero(); |
82 | ||
75d81601 | 83 | Bool_t IsQAOn(Int_t qaLevel) const { return TESTBIT(fQAlevel, qaLevel); }; |
9bcfd1ab | 84 | Bool_t IsAODanalysis() const { return TestBit(kAODanalysis); }; |
85 | Bool_t IsESDanalysis() const { return !TestBit(kAODanalysis); }; | |
722347d8 | 86 | Bool_t HasMCData() const { return TestBit(kHasMCdata); } |
c2690925 | 87 | Bool_t IsPbPb() const { return TestBit(kBeamType); } |
9bcfd1ab | 88 | Bool_t GetPlugin(Int_t plug) const { return TESTBIT(fPlugins, plug); }; |
3a72645a | 89 | |
90 | // Get Components for configuration | |
91 | AliHFEvarManager *GetVarManager() const { return fVarManager; } | |
bf892a6a | 92 | AliHFEpidQAmanager *GetPIDQAManager() const { return fPIDqa; } |
3a72645a | 93 | AliHFEpid *GetPID() const { return fPID; } |
94 | ||
722347d8 | 95 | void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; }; |
3a72645a | 96 | void SetTaggedTrackCuts(AliHFEcuts * const cuts) { fTaggedTrackCuts = cuts; } |
6555e2ad | 97 | void SetCleanTaggedTrack(Bool_t clean) { fCleanTaggedTrack = clean; }; |
e3ae862b | 98 | void SetVariablesTRDTaggedTrack(Bool_t variablesTRD) { fVariablesTRDTaggedTrack = variablesTRD; }; |
e3fc062d | 99 | void SetHFECutsPreselect(AliHFEcuts * const cuts) { fCutspreselect = cuts; }; |
70da6c5a | 100 | void SetHFEElecBackGround(AliHFEelecbackground * const elecBackGround) { fElecBackGround = elecBackGround; }; |
75d81601 | 101 | void SetQAOn(Int_t qaLevel) { SETBIT(fQAlevel, qaLevel); }; |
9bcfd1ab | 102 | void SwitchOnPlugin(Int_t plug); |
722347d8 | 103 | void SetHasMCData(Bool_t hasMC = kTRUE) { SetBit(kHasMCdata, hasMC); }; |
3a72645a | 104 | void SetFillSignalOnly(Bool_t signalOnly) { fFillSignalOnly = signalOnly; } |
105 | void SetRemovePileUp(Bool_t removePileUp) { fRemovePileUp = removePileUp; } | |
e3fc062d | 106 | void SetPIDPreselect(AliHFEpid * const cuts) { fPIDpreselect = cuts; }; |
9bcfd1ab | 107 | void SetAODAnalysis() { SetBit(kAODanalysis, kTRUE); }; |
108 | void SetESDAnalysis() { SetBit(kAODanalysis, kFALSE); }; | |
c2690925 | 109 | void SetPbPbAnalysis(Bool_t isPbPb = kFALSE) { SetBit(kBeamType, isPbPb); }; |
110 | void SetBackGroundFactorsFunction(TF1 * const backGroundFactorsFunction, Int_t centralitybin=0) | |
111 | { fBackGroundFactorArray[centralitybin]=backGroundFactorsFunction; | |
112 | fBackGroundFactorApply=kTRUE;}; | |
50685501 | 113 | void PrintStatus() const; |
c2690925 | 114 | Bool_t ReadCentrality(); |
3a72645a | 115 | void RejectionPileUpVertexRangeEventCut(); |
c2690925 | 116 | void SelectSpecialTrigger(const Char_t *trgclust){ fHasSpecialTriggerSelection = kTRUE; fSpecialTrigger = trgclust; } |
dbe3abbe | 117 | |
dbe3abbe | 118 | private: |
50685501 | 119 | enum{ |
9bcfd1ab | 120 | kHasMCdata = BIT(19), |
c2690925 | 121 | kAODanalysis = BIT(20), |
122 | kBeamType = BIT(21) | |
50685501 | 123 | }; |
78ea5ef4 | 124 | |
faee3b18 | 125 | Bool_t FillProductionVertex(const AliVParticle * const track) const; |
809a4336 | 126 | void MakeParticleContainer(); |
70da6c5a | 127 | void MakeEventContainer(); |
3a72645a | 128 | void InitPIDperformanceQA(); |
6555e2ad | 129 | void InitContaminationQA(); |
9bcfd1ab | 130 | void ProcessMC(); |
131 | void ProcessESD(); | |
132 | void ProcessAOD(); | |
e3fc062d | 133 | Bool_t PreSelectTrack(AliESDtrack *track) const; |
9bcfd1ab | 134 | Bool_t ProcessMCtrack(AliVParticle *track); |
3a72645a | 135 | Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track); |
75d81601 | 136 | ULong_t fQAlevel; // QA level |
9bcfd1ab | 137 | UShort_t fPlugins; // Enabled Plugins |
3a72645a | 138 | Bool_t fFillSignalOnly; // Fill container only with MC Signal Tracks |
c2690925 | 139 | Bool_t fBackGroundFactorApply; // Apply Background Function Subtraction |
3a72645a | 140 | Bool_t fRemovePileUp; // Remove Pile Up |
141 | Bool_t fIdentifiedAsPileUp; // Identified as pile-up | |
142 | Bool_t fIdentifiedAsOutInz; // Out Of Range in z | |
143 | Bool_t fPassTheEventCut; // Pass The Event Cut | |
c2690925 | 144 | Bool_t fHasSpecialTriggerSelection; // Select special triggered events |
145 | TString fSpecialTrigger; // Special trigger selection | |
3a72645a | 146 | Float_t fCentralityF; // Centrality |
c2690925 | 147 | Float_t fContributors; // Contributors |
148 | Double_t fWeightBackGround; // weight background function | |
149 | Double_t fVz; // z position of the primary vertex | |
150 | TF1 *fBackGroundFactorArray[12]; // Array of BackGround factors for each centrality bin, bin0 = min bias | |
3a72645a | 151 | AliHFEcontainer *fContainer; //! The HFE container |
152 | AliHFEvarManager *fVarManager; // The var manager as the backbone of the analysis | |
153 | AliHFEsignalCuts *fSignalCuts; //! MC true signal (electron coming from certain source) | |
dbe3abbe | 154 | AliCFManager *fCFM; //! Correction Framework Manager |
69ac0e6f | 155 | AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis for Normalisation |
3a72645a | 156 | AliHFEpid *fPID; // PID |
bf892a6a | 157 | AliHFEpidQAmanager *fPIDqa; // PID QA |
e3fc062d | 158 | AliHFEpid *fPIDpreselect; // PID oject for pre-selected tracks (without QA) |
722347d8 | 159 | AliHFEcuts *fCuts; // Cut Collection |
3a72645a | 160 | AliHFEcuts *fTaggedTrackCuts; // Cut Collection for V0 tagged tracks |
6555e2ad | 161 | Bool_t fCleanTaggedTrack; // Loose cleaning of the V0 tagged tracks electron |
e3ae862b | 162 | Bool_t fVariablesTRDTaggedTrack; // Take the variables at the TRD for the V0 tagged tracks electron |
e3fc062d | 163 | AliHFEcuts *fCutspreselect; // Cut Collection for pre-selected tracks |
259c3296 | 164 | AliHFEsecVtx *fSecVtx; //! Secondary Vertex Analysis |
9bcfd1ab | 165 | AliHFEelecbackground *fElecBackGround;//! Background analysis |
dbe3abbe | 166 | AliHFEmcQA *fMCQA; //! MC QA |
3a72645a | 167 | AliHFEtaggedTrackAnalysis *fTaggedTrackAnalysis; //!Analyse V0-tagged tracks |
168 | ||
169 | //-----------QA and output--------------- | |
dbe3abbe | 170 | TList *fQA; //! QA histos for the cuts |
259c3296 | 171 | TList *fOutput; //! Container for Task Output |
172 | TList *fHistMCQA; //! Output container for MC QA histograms | |
173 | TList *fHistSECVTX; //! Output container for sec. vertexing results | |
9bcfd1ab | 174 | TList *fHistELECBACKGROUND; //! Output container for electron background analysis |
69ac0e6f | 175 | AliHFEcollection *fQACollection; //! Tasks own QA collection |
3a72645a | 176 | //--------------------------------------- |
809a4336 | 177 | |
faee3b18 | 178 | ClassDef(AliAnalysisTaskHFE, 2) // The electron Analysis Task |
809a4336 | 179 | }; |
180 | #endif | |
dbe3abbe | 181 |