| | 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" |
| | 15 | #include "TProfile2D.h" |
| | 16 | |
| | 17 | class AliESDInputHandler; |
| | 18 | class AliMCEventHandler; |
| | 19 | class AliESDEvent; |
| | 20 | class AliESDtrack; |
| | 21 | class AliESDtrackCuts; |
| | 22 | class AliESDpidCuts; |
| | 23 | class AliV0Reader; |
| | 24 | class AliGammaConversionHistograms; |
| | 25 | class AliTriggerAnalysis; |
| | 26 | |
| | 27 | class 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(); |
| | 37 | virtual Bool_t Notify(); |
| | 38 | virtual void Terminate(const Option_t *); |
| | 39 | |
| | 40 | |
| | 41 | |
| | 42 | |
| | 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 | } |
| | 64 | void SetDoChicAnalysis(Bool_t flag){ fDoChicAnalysis = flag; } |
| | 65 | void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; } |
| | 66 | void SetProductionVertextoVGamma(Bool_t flag) { fSetProductionVertextoVGamma = flag; } |
| | 67 | |
| | 68 | |
| | 69 | private: |
| | 70 | |
| | 71 | void InitBack(); |
| | 72 | void ProcessPhotonCandidates(); |
| | 73 | void ProcessTruePhotonCandidates(AliAODConversionPhoton*); |
| | 74 | void ProcessTrueMesonCandidates(AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate); |
| | 75 | void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex); |
| | 76 | void ProcessElectronCandidates(); |
| | 77 | void ProcessVirtualGammasCandidates(); |
| | 78 | void ProcessMCParticles(); |
| | 79 | void CountESDTracks(); |
| | 80 | void CalculatePi0DalitzCandidates(); |
| | 81 | void CalculateBackground(); |
| | 82 | void UpdateEventByEventData(); |
| | 83 | void FillElectronQAHistos(AliAODConversionPhoton *Vgamma) const; |
| | 84 | Double_t GetPsiPair( const AliESDtrack *trackPos, const AliESDtrack *trackNeg ) const; |
| | 85 | Bool_t IsDalitz(TParticle *fMCMother) const; |
| | 86 | Bool_t IsPi0DalitzDaughter( Int_t label ) const; |
| | 87 | |
| | 88 | |
| | 89 | |
| | 90 | AliV0ReaderV1 *fV0Reader; |
| | 91 | AliDalitzElectronSelector* fElecSelector; |
| | 92 | AliGammaConversionAODBGHandler **fBGHandler; |
| | 93 | AliESDEvent *fESDEvent; |
| | 94 | AliMCEvent *fMCEvent; |
| | 95 | AliStack *fMCStack; |
| | 96 | TList **fCutFolder; |
| | 97 | TList **fESDList; |
| | 98 | TList **fBackList; |
| | 99 | TList **fMotherList; |
| | 100 | TList **fTrueList; |
| | 101 | TList **fMCList; |
| | 102 | TList **fQAFolder; |
| | 103 | TList *fOutputContainer; |
| | 104 | TClonesArray *fReaderGammas; |
| | 105 | vector<Int_t> fSelectorElectronIndex; |
| | 106 | vector<Int_t> fSelectorPositronIndex; |
| | 107 | TList *fGoodGammas; |
| | 108 | TList *fGoodVirtualGammas; |
| | 109 | TList *fGoodElectrons; |
| | 110 | TList *fGoodPositrons; |
| | 111 | TList *fCutGammaArray; |
| | 112 | TList *fCutElectronArray; |
| | 113 | TList *fCutMesonArray; |
| | 114 | TList **fGammasPool; |
| | 115 | AliConversionCuts *fConversionCuts; |
| | 116 | TH1F **hESDConvGammaPt; |
| | 117 | TH1F **hESDConvGammaEta; |
| | 118 | TH2F **hESDConvGammaZR; |
| | 119 | TH1F **hESDDalitzElectronPt; |
| | 120 | TH1F **hESDDalitzPositronPt; |
| | 121 | TH1F **hESDDalitzElectronPhi; |
| | 122 | TH1F **hESDDalitzPositronPhi; |
| | 123 | TH1F **hESDDalitzElectronAfterPt; |
| | 124 | TH1F **hESDDalitzPositronAfterPt; |
| | 125 | TH1F **hESDDalitzElectronAfterEta; |
| | 126 | TH1F **hESDDalitzPositronAfterEta; |
| | 127 | TH1F **hESDDalitzElectronAfterPhi; |
| | 128 | TH1F **hESDDalitzPositronAfterPhi; |
| | 129 | TH1F **hESDDalitzElectronAfterNClsITS; |
| | 130 | TH1F **hESDDalitzPositronAfterNClsITS; |
| | 131 | TH2F **hESDDalitzElectronAfterNFindClsTPC; |
| | 132 | TH2F **hESDDalitzPositronAfterNFindClsTPC; |
| | 133 | TH2F **hESDDalitzElectronAfterNClsTPC; |
| | 134 | TH2F **hESDDalitzPositronAfterNClsTPC; |
| | 135 | TH2F **hESDDalitzElectronAfterNCrossedRowsTPC; |
| | 136 | TH2F **hESDDalitzPositronAfterNCrossedRowsTPC; |
| | 137 | TH2F **hESDDalitzPosEleAfterDCAxy; |
| | 138 | TH2F **hESDDalitzPosEleAfterDCAz; |
| | 139 | TH2F **hESDDalitzElectronAfterTPCdEdxVsP; |
| | 140 | TH2F **hESDDalitzPositronAfterTPCdEdxVsP; |
| | 141 | TH2F **hESDDalitzElectronAfterTPCdEdxSignalVsP; |
| | 142 | TH2F **hESDDalitzPositronAfterTPCdEdxSignalVsP; |
| | 143 | TH2F **hESDDalitzElectronAfterTPCdEdxVsEta; |
| | 144 | TH2F **hESDDalitzPositronAfterTPCdEdxVsEta; |
| | 145 | TH2F **hESDDalitzElectronAfterTPCdEdxVsPhi; |
| | 146 | TH2F **hESDDalitzPositronAfterTPCdEdxVsPhi; |
| | 147 | TH1F **hESDMotherPhi; |
| | 148 | TH2F **hESDEposEnegPsiPairDPhi; |
| | 149 | TH2F **hESDEposEnegInvMassPt; |
| | 150 | TH2F **hESDEposEnegAfterMassCutInvMassPi0Pt; |
| | 151 | TH2F **hESDEposEnegInvMassPi0Pt; |
| | 152 | TH2F **hESDEposEnegLikeSignBackInvMassPt; |
| | 153 | TH2F **hESDMotherInvMassPt; |
| | 154 | TH2F **hESDPi0MotherInvMassPt; |
| | 155 | TH2F **hESDPi0MotherDiffInvMassPt; |
| | 156 | TH2F **hESDPi0MotherDiffLimInvMassPt; |
| | 157 | THnSparseF **sESDMotherInvMassPtZM; |
| | 158 | TH2F **hESDMotherBackInvMassPt; |
| | 159 | THnSparseF **sESDMotherBackInvMassPtZM; |
| | 160 | TH1F **hMCAllGammaPt; |
| | 161 | TH1F **hMCConvGammaPt; |
| | 162 | TH1F **hMCConvGammaRSPt; |
| | 163 | TH1F **hMCAllPositronsPt; |
| | 164 | TH1F **hMCAllElectronsPt; |
| | 165 | TH1F **hMCConvGammaEta; |
| | 166 | TH1F **hMCAllPositronsEta; |
| | 167 | TH1F **hMCAllElectronsEta; |
| | 168 | TH1F **hMCPi0DalitzGammaPt; |
| | 169 | TH1F **hMCPi0DalitzElectronPt; |
| | 170 | TH1F **hMCPi0DalitzPositronPt; |
| | 171 | TH1F **hMCPi0Pt; |
| | 172 | TH1F **hMCPi0GGPt; |
| | 173 | TH1F **hMCEtaPt; |
| | 174 | TH1F **hMCEtaGGPt; |
| | 175 | TH1F **hMCPi0InAccPt; |
| | 176 | TH1F **hMCEtaInAccPt; |
| | 177 | TH1F **hMCChiCPt; |
| | 178 | TH1F **hMCChiCInAccPt; |
| | 179 | TH2F **hMCPi0EposEnegInvMassPt; |
| | 180 | TH2F **hMCEtaEposEnegInvMassPt; |
| | 181 | TH2F **hESDEposEnegTruePi0DalitzInvMassPt; |
| | 182 | TH1F **hESDEposEnegTruePrimPi0DalitzInvMass; |
| | 183 | TH2F **hESDEposEnegTruePi0DalitzPsiPairDPhi; |
| | 184 | TH2F **hESDEposEnegTrueEtaDalitzInvMassPt; |
| | 185 | TH1F **hESDEposEnegTruePrimEtaDalitzInvMass; |
| | 186 | TH2F **hESDEposEnegTrueEtaDalitzPsiPairDPhi; |
| | 187 | TH2F **hESDEposEnegTruePhotonInvMassPt; |
| | 188 | TH2F **hESDEposEnegTrueInvMassPt; |
| | 189 | TH2F **hESDEposEnegTruePhotonPsiPairDPhi; |
| | 190 | TH2F **hESDEposEnegTruePhotonPsiPairDPhiPtCut; |
| | 191 | TH2F **hESDEposEnegTrueJPsiInvMassPt; |
| | 192 | TH2F **hESDTrueMotherChiCInvMassPt; |
| | 193 | TH2F **hESDTrueMotherChiCDiffInvMassPt; |
| | 194 | TH2F **hESDTrueMotherInvMassPt; |
| | 195 | TH2F **hESDTrueMotherDalitzInvMassPt; |
| | 196 | TH2F **hESDTrueMotherPi0GGInvMassPt; |
| | 197 | TH2F **hESDTruePrimaryMotherPi0GGInvMassPt; |
| | 198 | TH2F **hESDTrueSecondaryMotherPi0GGInvMassPt; |
| | 199 | TH2F **hESDTruePrimaryMotherInvMassMCPt; |
| | 200 | TH2F **hESDTruePrimaryMotherInvMassPt; |
| | 201 | TH2F **hESDTruePrimaryMotherW0WeightingInvMassPt; |
| | 202 | TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt; |
| | 203 | TH2F **hESDTrueSecondaryMotherInvMassPt; |
| | 204 | TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt; |
| | 205 | TH2F **hESDTrueBckGGInvMassPt; |
| | 206 | TH2F **hESDTrueBckContInvMassPt; |
| | 207 | TH2F **hESDTrueMotherGGInvMassPt; |
| | 208 | TH1F **hESDTrueConvGammaPt; |
| | 209 | TH1F **hESDTruePositronPt; |
| | 210 | TH1F **hESDTrueElectronPt; |
| | 211 | TH1F **hESDTrueSecConvGammaPt; |
| | 212 | TH1F **hESDTrueSecPositronPt; |
| | 213 | TH1F **hESDTrueSecElectronPt; |
| | 214 | TH1F **hESDTruePi0DalitzConvGammaPt; |
| | 215 | TH1F **hESDTruePi0DalitzPositronPt; |
| | 216 | TH1F **hESDTruePi0DalitzElectronPt; |
| | 217 | TH1F **hESDTruePi0DalitzSecConvGammaPt; |
| | 218 | TH1F **hESDTruePi0DalitzSecPositronPt; |
| | 219 | TH1F **hESDTruePi0DalitzSecElectronPt; |
| | 220 | //if(fDoMesonAnalysis){ |
| | 221 | |
| | 222 | |
| | 223 | TH1I **hNEvents; |
| | 224 | TH1I **hNGoodESDTracks; |
| | 225 | TH2F **hNGoodESDTracksVsNGoodGammas; |
| | 226 | TH2F **hNGoodESDTracksVsNGoodVGammas; |
| | 227 | TH1I **hNV0Tracks; |
| | 228 | TProfile **hEtaShift; |
| | 229 | |
| | 230 | TRandom3 fRandom; |
| | 231 | Double_t *fUnsmearedPx; |
| | 232 | Double_t *fUnsmearedPy; |
| | 233 | Double_t *fUnsmearedPz; |
| | 234 | Double_t *fUnsmearedE; |
| | 235 | Double_t *fUnsmearedVPx; |
| | 236 | Double_t *fUnsmearedVPy; |
| | 237 | Double_t *fUnsmearedVPz; |
| | 238 | Double_t *fUnsmearedVE; |
| | 239 | |
| | 240 | |
| | 241 | Int_t fnCuts; |
| | 242 | Int_t fiCut; |
| | 243 | Int_t fNumberOfESDTracks; |
| | 244 | Int_t fNumberOfESDTrackskBoth; |
| | 245 | Int_t fNVirtualGammas; |
| | 246 | Bool_t fMoveParticleAccordingToVertex; |
| | 247 | Bool_t fIsHeavyIon; |
| | 248 | Bool_t fDoMesonAnalysis; |
| | 249 | Bool_t fDoChicAnalysis; |
| | 250 | Bool_t fDoMesonQA; |
| | 251 | Bool_t fSetProductionVertextoVGamma; |
| | 252 | Bool_t fIsFromMBHeader; |
| | 253 | Bool_t fIsMC; |
| | 254 | |
| | 255 | private: |
| | 256 | AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented |
| | 257 | AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented |
| | 258 | |
| | 259 | ClassDef( AliAnalysisTaskGammaConvDalitzV1, 3 ); |
| | 260 | }; |
| | 261 | |
| | 262 | #endif // ALIANALYSISTASKGAMMACONVDALITZV1_H |
| | 263 | |
git://git.uio.no / u / mrichter / AliRoot.git / blame_incremental