]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGGA/GammaConv/AliAnalysisTaskGammaConvDalitzV1.h
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnalysisTaskGammaConvDalitzV1.h
CommitLineData
2bb2434e 1#ifndef ALIANALYSISTASKGAMMACONVDALITZV1_H
2#define ALIANALYSISTASKGAMMACONVDALITZV1_H
3
4/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
6
7// Analysis task for pi0->e+e-gamma (Dalitz decay)
8
9#include "AliAnalysisTaskSE.h"
10#include "AliV0ReaderV1.h"
11#include "AliKFConversionPhoton.h"
12#include "AliDalitzElectronSelector.h"
13#include "AliConversionMesonCuts.h"
14#include "AliGammaConversionAODBGHandler.h"
8a52eed5 15#include "TProfile2D.h"
2bb2434e 16
17class AliESDInputHandler;
18class AliMCEventHandler;
19class AliESDEvent;
20class AliESDtrack;
21class AliESDtrackCuts;
22class AliESDpidCuts;
23class AliV0Reader;
24class AliGammaConversionHistograms;
25class AliTriggerAnalysis;
26
27class AliAnalysisTaskGammaConvDalitzV1: public AliAnalysisTaskSE
28{
29 public:
30
31 AliAnalysisTaskGammaConvDalitzV1();
32 AliAnalysisTaskGammaConvDalitzV1( const char* name );
33 virtual ~AliAnalysisTaskGammaConvDalitzV1();
34
35 virtual void UserExec(Option_t *);
36 virtual void UserCreateOutputObjects();
ae947965 37 virtual Bool_t Notify();
2bb2434e 38 virtual void Terminate(const Option_t *);
39
40
41
42
ae947965 43 void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
44
45 void SetIsHeavyIon(Int_t flag){
46 if (flag == 1 || flag ==2 ){
47 fIsHeavyIon = 1;
48 } else {
49 fIsHeavyIon = 0;
50 }
51 }
52
53 void SetIsMC(Bool_t isMC){fIsMC=isMC;}
54 void SetConversionCutList(Int_t nCuts, TList *CutArray){
55 fnCuts= nCuts;
56 fCutGammaArray = CutArray;
57 }
58 void SetElectronCutList(TList *CutArray){
59 fCutElectronArray = CutArray;
60 }
61 void SetMesonCutList(TList *CutArray){
62 fCutMesonArray = CutArray;
63 }
fcc79cf5 64 void SetDoChicAnalysis(Bool_t flag){ fDoChicAnalysis = flag; }
65 void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; }
ae947965 66
2bb2434e 67
68 private:
69
2bb2434e 70 void InitBack();
71 void ProcessPhotonCandidates();
72 void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
73 void ProcessTrueMesonCandidates(AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
74 void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
75 void ProcessElectronCandidates();
76 void ProcessMCParticles();
77 void CountESDTracks();
78 void CalculatePi0DalitzCandidates();
79 void CalculateBackground();
80 void UpdateEventByEventData();
a280ac15 81 Double_t GetPsiPair( const AliESDtrack *trackPos, const AliESDtrack *trackNeg ) const;
ae947965 82 Bool_t IsDalitz(TParticle *fMCMother) const;
ccfa8c0d 83 Bool_t IsPi0DalitzDaughter( Int_t label ) const;
2bb2434e 84
85
86
87 AliV0ReaderV1 *fV0Reader;
88 AliDalitzElectronSelector* fElecSelector;
89 AliGammaConversionAODBGHandler **fBGHandler;
90 AliESDEvent *fESDEvent;
91 AliMCEvent *fMCEvent;
92 AliStack *fMCStack;
93 TList **fCutFolder;
94 TList **fESDList;
95 TList **fBackList;
fcc79cf5 96 TList **fMotherList;
2bb2434e 97 TList **fTrueList;
98 TList **fMCList;
99 TList *fOutputContainer;
100 TClonesArray *fReaderGammas;
101 vector<Int_t> fSelectorElectronIndex;
102 vector<Int_t> fSelectorPositronIndex;
103 TList *fGoodGammas;
104 TList *fGoodVirtualGammas;
105 TList *fGoodElectrons;
106 TList *fGoodPositrons;
107 TList *fCutGammaArray;
108 TList *fCutElectronArray;
109 TList *fCutMesonArray;
110 TList **fGammasPool;
111 AliConversionCuts *fConversionCuts;
112 TH1F **hESDConvGammaPt;
2279d237 113 TH1F **hESDConvGammaEta;
114 TH2F **hESDConvGammaZR;
2bb2434e 115 TH1F **hESDDalitzElectronPt;
116 TH1F **hESDDalitzPositronPt;
4803eb1f 117 TH1F **hESDDalitzElectronPhi;
118 TH1F **hESDDalitzPositronPhi;
119 TH1F **hESDDalitzElectronAfterPt;
120 TH1F **hESDDalitzPositronAfterPt;
2279d237 121 TH1F **hESDDalitzElectronAfterEta;
122 TH1F **hESDDalitzPositronAfterEta;
4803eb1f 123 TH1F **hESDDalitzElectronAfterPhi;
124 TH1F **hESDDalitzPositronAfterPhi;
125 TH2F **hESDDalitzElectronAfterNFindClsTPC;
126 TH2F **hESDDalitzPositronAfterNFindClsTPC;
127 TH2F **hESDDalitzPosEleAfterDCAxy;
128 TH2F **hESDDalitzPosEleAfterDCAz;
129 TH2F **hESDDalitzPosEleAfterTPCdEdx;
130 TH2F **hESDDalitzPosEleAfterTPCdEdxSignal;
131 TH1F **hESDMotherPhi;
2bb2434e 132 TH2F **hESDEposEnegPsiPairDPhi;
a280ac15 133 TH2F **hESDEposEnegInvMassPt;
0a2b2b4b 134 TH2F **hESDEposEnegLikeSignBackInvMassPt;
2bb2434e 135 TH2F **hESDMotherInvMassPt;
a280ac15 136 TH2F **hESDPi0MotherInvMassPt;
137 TH2F **hESDPi0MotherDiffInvMassPt;
ae947965 138 TH2F **hESDPi0MotherDiffLimInvMassPt;
2bb2434e 139 THnSparseF **sESDMotherInvMassPtZM;
140 TH2F **hESDMotherBackInvMassPt;
141 THnSparseF **sESDMotherBackInvMassPtZM;
ccfa8c0d 142 TH1F **hMCAllGammaPt;
ae947965 143 TH1F **hMCConvGammaPt;
144 TH1F **hMCConvGammaRSPt;
ccfa8c0d 145 TH1F **hMCAllPositronsPt;
146 TH1F **hMCAllElectronsPt;
147 TH1F **hMCPi0DalitzGammaPt;
148 TH1F **hMCPi0DalitzElectronPt;
149 TH1F **hMCPi0DalitzPositronPt;
2bb2434e 150 TH1F **hMCPi0Pt;
a280ac15 151 TH1F **hMCPi0GGPt;
2bb2434e 152 TH1F **hMCEtaPt;
a280ac15 153 TH1F **hMCEtaGGPt;
2bb2434e 154 TH1F **hMCPi0InAccPt;
155 TH1F **hMCEtaInAccPt;
0f8c33c1 156 TH1F **hMCChiCPt;
157 TH1F **hMCChiCInAccPt;
ccfa8c0d 158 TH2F **hESDEposEnegTruePi0DalitzInvMassPt;
a072aeaa 159 TH2F **hESDEposEnegTruePi0DalitzPsiPairDPhi;
ccfa8c0d 160 TH2F **hESDEposEnegTrueEtaDalitzInvMassPt;
a072aeaa 161 TH2F **hESDEposEnegTrueEtaDalitzPsiPairDPhi;
ccfa8c0d 162 TH2F **hESDEposEnegTruePhotonInvMassPt;
a072aeaa 163 TH2F **hESDEposEnegTruePhotonPsiPairDPhi;
0f8c33c1 164 TH2F **hESDEposEnegTrueJPsiInvMassPt;
165 TH2F **hESDTrueMotherChiCInvMassPt;
ae947965 166 TH2F **hESDTrueMotherChiCDiffInvMassPt;
2bb2434e 167 TH2F **hESDTrueMotherInvMassPt;
4803eb1f 168 TH2F **hESDTrueMotherDalitzInvMassPt;
2bb2434e 169 TH2F **hESDTrueMotherPi0GGInvMassPt;
86eaf9ae 170 TH2F **hESDTruePrimaryMotherPi0GGInvMassPt;
171 TH2F **hESDTrueSecondaryMotherPi0GGInvMassPt;
2bb2434e 172 TH2F **hESDTruePrimaryMotherInvMassMCPt;
86eaf9ae 173 TH2F **hESDTruePrimaryMotherInvMassPt;
174 TH2F **hESDTruePrimaryMotherW0WeightingInvMassPt;
2bb2434e 175 TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt;
176 TH2F **hESDTrueSecondaryMotherInvMassPt;
177 TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
178 TH2F **hESDTrueBckGGInvMassPt;
179 TH2F **hESDTrueBckContInvMassPt;
180 TH2F **hESDTrueMotherGGInvMassPt;
181 TH1F **hESDTrueConvGammaPt;
ccfa8c0d 182 TH1F **hESDTruePositronPt;
183 TH1F **hESDTrueElectronPt;
ae947965 184 TH1F **hESDTrueSecConvGammaPt;
185 TH1F **hESDTrueSecPositronPt;
186 TH1F **hESDTrueSecElectronPt;
ccfa8c0d 187 TH1F **hESDTruePi0DalitzConvGammaPt;
188 TH1F **hESDTruePi0DalitzPositronPt;
189 TH1F **hESDTruePi0DalitzElectronPt;
ae947965 190 TH1F **hESDTruePi0DalitzSecConvGammaPt;
191 TH1F **hESDTruePi0DalitzSecPositronPt;
192 TH1F **hESDTruePi0DalitzSecElectronPt;
ccfa8c0d 193 //if(fDoMesonAnalysis){
194
195
2bb2434e 196 TH1I **hNEvents;
197 TH1I **hNGoodESDTracks;
8a52eed5 198 TProfile **hEtaShift;
2bb2434e 199
200 TRandom3 fRandom;
201 Double_t *fUnsmearedPx;
202 Double_t *fUnsmearedPy;
203 Double_t *fUnsmearedPz;
204 Double_t *fUnsmearedE;
205 Int_t fnCuts;
206 Int_t fiCut;
207 Int_t fNumberOfESDTracks;
208 Bool_t fMoveParticleAccordingToVertex;
209 Bool_t fIsHeavyIon;
210 Bool_t fDoMesonAnalysis;
fcc79cf5 211 Bool_t fDoChicAnalysis;
212 Bool_t fDoMesonQA;
ae947965 213 Bool_t fIsFromMBHeader;
214 Bool_t fIsMC;
215
2bb2434e 216 private:
217 AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
218 AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
219
ae947965 220 ClassDef( AliAnalysisTaskGammaConvDalitzV1, 3 );
2bb2434e 221};
222
223#endif // ALIANALYSISTASKGAMMACONVDALITZV1_H
224