From c852fdae59519e47f28190c825c63d919fc26ccc Mon Sep 17 00:00:00 2001 From: ssakai Date: Thu, 13 Jun 2013 13:39:51 +0000 Subject: [PATCH] new EMCal tasks for PA analysis --- PWGHF/CMakelibPWGHFhfe.pkg | 1 + PWGHF/PWGHFhfeLinkDef.h | 2 + PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.cxx | 2450 +++++++++++++++++ PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h | 334 +++ PWGHF/hfe/macros/AddTaskEMCalHFEpA.C | 59 + .../hfe/macros/AddTaskEMCalHFEpACorrelation.C | 62 + .../hfe/macros/configs/pPb/ConfigEMCalHFEpA.C | 134 + .../configs/pPb/ConfigEMCalHFEpACorrelation.C | 152 + 8 files changed, 3194 insertions(+) create mode 100644 PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.cxx create mode 100644 PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h create mode 100644 PWGHF/hfe/macros/AddTaskEMCalHFEpA.C create mode 100644 PWGHF/hfe/macros/AddTaskEMCalHFEpACorrelation.C create mode 100644 PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpA.C create mode 100644 PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpACorrelation.C diff --git a/PWGHF/CMakelibPWGHFhfe.pkg b/PWGHF/CMakelibPWGHFhfe.pkg index cda62029550..34a7ca2b6fe 100644 --- a/PWGHF/CMakelibPWGHFhfe.pkg +++ b/PWGHF/CMakelibPWGHFhfe.pkg @@ -76,6 +76,7 @@ set (SRCS hfe/AliAnalysisTaskElecHadronCorrel.cxx hfe/AliAnalysisTaskFlowTPCEMCalEP.cxx hfe/AliAnalysisTaskHFECal.cxx + hfe/AliAnalysisTaskEMCalHFEpA.cxx hfe/AliHFEdebugTreeTask.cxx hfe/AliHFEVZEROEventPlane.cxx hfe/AliAnalysisTaskFlowTPCTOFEPSP.cxx diff --git a/PWGHF/PWGHFhfeLinkDef.h b/PWGHF/PWGHFhfeLinkDef.h index 274f6c16412..2000a747c50 100644 --- a/PWGHF/PWGHFhfeLinkDef.h +++ b/PWGHF/PWGHFhfeLinkDef.h @@ -64,6 +64,8 @@ #pragma link C++ class AliehDPhiBasicParticle+; #pragma link C++ class AliAnalysisTaskFlowTPCEMCalEP+; #pragma link C++ class AliAnalysisTaskHFECal+; +#pragma link C++ class AliAnalysisTaskEMCalHFEpA+; +#pragma link C++ class AliEHCParticle+; #pragma link C++ class AliHFEdebugTreeTask+; diff --git a/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.cxx b/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.cxx new file mode 100644 index 00000000000..2334db8cc89 --- /dev/null +++ b/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.cxx @@ -0,0 +1,2450 @@ +/************************************************************************** + * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * + * * + * Author: The ALICE Off-line Project. * + * Contributors are mentioned in the code where appropriate. * + * * + * Permission to use, copy, modify and distribute this software and its * + * documentation strictly for non-commercial purposes is hereby granted * + * without fee, provided that the above copyright notice appears in all * + * copies and that both the copyright notice and this permission notice * + * appear in the supporting documentation. The authors make no claims * + * about the suitability of this software for any purpose. It is * + * provided "as is" without express or implied warranty. * + **************************************************************************/ + +//////////////////////////////////////////////////////////////////////// +// // +// Task for Heavy-flavour electron analysis in pPb collisions // +// (+ Electron-Hadron Jetlike Azimuthal Correlation) // +// // +// v1.0 // +// // +// Authors // +// Elienos Pereira de Oliveira Filho (epereira@cern.ch) // +// Cristiane Jahnke (cristiane.jahnke@cern.ch) // +// // +//////////////////////////////////////////////////////////////////////// + +#include "TChain.h" +#include "TTree.h" +#include "TNtuple.h" +#include "TH1F.h" +#include "TH2F.h" +#include "TCanvas.h" +#include "AliAnalysisTask.h" +#include "AliAnalysisManager.h" +#include "AliESDEvent.h" +#include "AliAODEvent.h" +#include "AliVEvent.h" +#include "AliESDInputHandler.h" +#include "AliESDtrackCuts.h" +#include "AliESDCaloCluster.h" +#include "AliESDCaloCells.h" +#include "AliEMCALTrack.h" +#include "AliExternalTrackParam.h" +#include "AliPhysicsSelection.h" +#include "TGeoGlobalMagField.h" +#include "AliMagF.h" +#include "AliLog.h" +#include "AliStack.h" +#include "AliCentrality.h" +#include "AliAODMCParticle.h" +#include "AliAODMCHeader.h" +#include "AliPID.h" +#include "AliPIDResponse.h" +#include "AliHFEcontainer.h" +#include "AliHFEcuts.h" +#include "AliHFEpid.h" +#include "AliHFEpidBase.h" +#include "AliHFEpidQAmanager.h" +#include "AliHFEtools.h" +#include "AliCFContainer.h" +#include "AliCFManager.h" +#include "AliSelectNonHFE.h" +#include "AliHFEpidTPC.h" +#include "AliAnalysisTaskEMCalHFEpA.h" +#include "TMath.h" +#include "THnSparse.h" +#include "TLorentzVector.h" +#include "TString.h" +#include "TFile.h" +#include "AliESDHandler.h" +#include "AliMCEventHandler.h" +#include "AliMCEvent.h" +#include "AliStack.h" +#include "TParticle.h" +#include "AliLog.h" +#include "AliAnalysisTaskSE.h" +#include "TRefArray.h" +#include "TVector.h" +#include "stdio.h" +#include "TGeoManager.h" +#include "iostream" +#include "fstream" +#include "AliKFParticle.h" +#include "AliKFVertex.h" +#include "AliVParticle.h" +#include "AliVTrack.h" +#include "AliEventPoolManager.h" +#include "TObjArray.h" +//______________________________________________________________________ + +//______________________________________________________________________ +ClassImp(AliAnalysisTaskEMCalHFEpA) + +//______________________________________________________________________ +AliAnalysisTaskEMCalHFEpA::AliAnalysisTaskEMCalHFEpA(const char *name) + : AliAnalysisTaskSE(name) + ,fCorrelationFlag(0) + ,fIsMC(0) + ,fUseEMCal(kFALSE) + ,fEMCEG1(kFALSE) + ,fEMCEG2(kFALSE) + ,fIsHFE1(kFALSE) + ,fIsHFE2(kFALSE) + ,fIsNonHFE(kFALSE) + ,fIsFromD(kFALSE) + ,fIsFromB(kFALSE) + ,fIsFromPi0(kFALSE) + ,fIsFromEta(kFALSE) + ,fIsFromGamma(kFALSE) + ,fESD(0) + ,fAOD(0) + ,fVevent(0) + ,fPartnerCuts(new AliESDtrackCuts()) + ,fOutputList(0) + ,fPidResponse(0) + ,fNonHFE(new AliSelectNonHFE()) + ,fIsAOD(kFALSE) + ,fCentrality(0) + ,fCentralityMin(0) + ,fCentralityMax(100) + ,fHasCentralitySelection(kFALSE) + ,fCentralityHist(0) + ,fCentralityHistPass(0) + ,fZvtx(0) + ,fEstimator(0) + ,fClus(0) + ,fNevent(0) + ,fPtElec_Inc(0) + ,fPtElec_ULS(0) + ,fPtElec_LS(0) + ,fpid(0) + ,fEoverP_pt(0) + ,fEoverP_tpc(0) + ,fTPC_pt(0) + ,fTPC_p(0) + ,fTPCnsigma_pt(0) + ,fTPCnsigma_p(0) + ,fTPCnsigma_pt_2D(0) + ,fTPCnsigma_eta(0) + ,fTPCnsigma_phi(0) + ,fECluster(0) + ,fEtaPhi(0) + ,fVtxZ(0) + ,fNTracks(0) + ,fNClusters(0) + ,fTPCNcls_EoverP(0) + ,fEta(0) + ,fPhi(0) + ,fR(0) + ,fR_EoverP(0) + ,fNcells(0) + ,fNcells_EoverP(0) + ,fNcells_electrons(0) + ,fNcells_hadrons(0) + ,fECluster_ptbins(0) + ,fEoverP_ptbins(0) + ,fEoverP_wSSCut(0) + ,fM02_EoverP(0) + ,fM20_EoverP(0) + ,fTPCnsigma_eta_electrons(0) + ,fTPCnsigma_eta_hadrons(0) + ,fEoverP_pt_pions(0) + ,ftpc_p_EoverPcut(0) + ,fnsigma_p_EoverPcut(0) + ,fEoverP_pt_pions2(0) + ,fNcells_pt(0) + ,fEoverP_pt_hadrons(0) + ,fCEtaPhi_Inc(0) + ,fCEtaPhi_ULS(0) + ,fCEtaPhi_LS(0) + ,fCEtaPhi_ULS_NoP(0) + ,fCEtaPhi_LS_NoP(0) + ,fCEtaPhi_ULS_Weight(0) + ,fCEtaPhi_LS_Weight(0) + ,fCEtaPhi_ULS_NoP_Weight(0) + ,fCEtaPhi_LS_NoP_Weight(0) + ,fInvMass(0) + ,fInvMassBack(0) + ,fDCA(0) + ,fDCABack(0) + ,fOpAngle(0) + ,fOpAngleBack(0) + ,fMassCut(0.1) + ,fEtaCutMin(-0.9) + ,fEtaCutMax(0.9) + ,fEoverPCutMin(0.8) + ,fEoverPCutMax(1.2) + ,fAngleCut(999) + ,fChi2Cut(3.5) + ,fDCAcut(999) + ,fMassCutFlag(kTRUE) + ,fAngleCutFlag(kFALSE) + ,fChi2CutFlag(kFALSE) + ,fDCAcutFlag(kFALSE) + ,fPtBackgroundBeforeReco(0) + ,fPtBackgroundAfterReco(0) + ,fPtMCparticleAll(0) + ,fPtMCparticleReco(0) + ,fPtMCparticleAllHfe1(0) + ,fPtMCparticleRecoHfe1(0) + ,fPtMCparticleAllHfe2(0) + ,fPtMCparticleRecoHfe2(0) + ,fPtMCelectronAfterAll(0) + ,fPtMCpi0(0) + ,fPtMC_EMCal_All(0) + ,fPtMC_EMCal_Selected(0) + ,fPtMC_TPC_All(0) + ,fPtMC_TPC_Selected(0) + ,fPtMCWithLabel(0) + ,fPtMCWithoutLabel(0) + ,fPtIsPhysicaPrimary(0) + ,fCuts(0) + ,fCFM(0) + ,fPID(new AliHFEpid("hfePid")) + ,fPIDqa(0) + ,fMCstack(0) + ,fRejectKinkMother(kFALSE) + ,fMCtrack(0) + ,fMCtrackMother(0) + ,fMCtrackGMother(0) + ,fMCtrackGGMother(0) + ,fMCtrackGGGMother(0) + ,fMCarray(0) + ,fMCheader(0) + ,fMCparticle(0) + ,fMCparticleMother(0) + ,fMCparticleGMother(0) + ,fMCparticleGGMother(0) + ,fMCparticleGGGMother(0) + ,fEventHandler(0) + ,fMCevent(0) + ,fPoolMgr(0) + ,fPool(0) + ,fTracksClone(0) + ,fTracks(0) + ,fCEtaPhi_Inc_EM(0) + ,fCEtaPhi_ULS_EM(0) + ,fCEtaPhi_LS_EM(0) + ,fCEtaPhi_ULS_Weight_EM(0) + ,fCEtaPhi_LS_Weight_EM(0) + ,fPoolNevents(0) + ,fEventMixingFlag(0) + ,fCEtaPhi_Inc_DiHadron(0) + ,fPtTrigger_Inc(0) +{ + //Named constructor + // Define input and output slots here + // Input slot #0 works with a TChain + DefineInput(0, TChain::Class()); + // Output slot #0 id reserved by the base class for AOD + // Output slot #1 writes into a TH1 container + // DefineOutput(1, TH1I::Class()); + DefineOutput(1, TList::Class()); + // DefineOutput(3, TTree::Class()); +} + +//________________________________________________________________________ +AliAnalysisTaskEMCalHFEpA::AliAnalysisTaskEMCalHFEpA() + : AliAnalysisTaskSE("DefaultAnalysis_AliAnalysisTaskEMCalHFEpA") + ,fCorrelationFlag(0) + ,fIsMC(0) + ,fUseEMCal(kFALSE) + ,fEMCEG1(kFALSE) + ,fEMCEG2(kFALSE) + ,fIsHFE1(kFALSE) + ,fIsHFE2(kFALSE) + ,fIsNonHFE(kFALSE) + ,fIsFromD(kFALSE) + ,fIsFromB(kFALSE) + ,fIsFromPi0(kFALSE) + ,fIsFromEta(kFALSE) + ,fIsFromGamma(kFALSE) + ,fESD(0) + ,fAOD(0) + ,fVevent(0) + ,fPartnerCuts(new AliESDtrackCuts()) + ,fOutputList(0) + ,fPidResponse(0) + ,fNonHFE(new AliSelectNonHFE()) + ,fIsAOD(kFALSE) + ,fCentrality(0) + ,fCentralityMin(0) + ,fCentralityMax(100) + ,fHasCentralitySelection(kFALSE) + ,fCentralityHist(0) + ,fCentralityHistPass(0) + ,fZvtx(0) + ,fEstimator(0) + ,fClus(0) + ,fNevent(0) + ,fPtElec_Inc(0) + ,fPtElec_ULS(0) + ,fPtElec_LS(0) + ,fpid(0) + ,fEoverP_pt(0) + ,fEoverP_tpc(0) + ,fTPC_pt(0) + ,fTPC_p(0) + ,fTPCnsigma_pt(0) + ,fTPCnsigma_p(0) + ,fTPCnsigma_pt_2D(0) + ,fTPCnsigma_eta(0) + ,fTPCnsigma_phi(0) + ,fECluster(0) + ,fEtaPhi(0) + ,fVtxZ(0) + ,fNTracks(0) + ,fNClusters(0) + ,fTPCNcls_EoverP(0) + ,fEta(0) + ,fPhi(0) + ,fR(0) + ,fR_EoverP(0) + ,fNcells(0) + ,fNcells_EoverP(0) + ,fNcells_electrons(0) + ,fNcells_hadrons(0) + ,fECluster_ptbins(0) + ,fEoverP_ptbins(0) + ,fEoverP_wSSCut(0) + ,fM02_EoverP(0) + ,fM20_EoverP(0) + ,fTPCnsigma_eta_electrons(0) + ,fTPCnsigma_eta_hadrons(0) + ,fEoverP_pt_pions(0) + ,ftpc_p_EoverPcut(0) + ,fnsigma_p_EoverPcut(0) + ,fEoverP_pt_pions2(0) + ,fNcells_pt(0) + ,fEoverP_pt_hadrons(0) + ,fCEtaPhi_Inc(0) + ,fCEtaPhi_ULS(0) + ,fCEtaPhi_LS(0) + ,fCEtaPhi_ULS_NoP(0) + ,fCEtaPhi_LS_NoP(0) + ,fCEtaPhi_ULS_Weight(0) + ,fCEtaPhi_LS_Weight(0) + ,fCEtaPhi_ULS_NoP_Weight(0) + ,fCEtaPhi_LS_NoP_Weight(0) + ,fInvMass(0) + ,fInvMassBack(0) + ,fDCA(0) + ,fDCABack(0) + ,fOpAngle(0) + ,fOpAngleBack(0) + ,fMassCut(0.1) + ,fEtaCutMin(-0.9) + ,fEtaCutMax(0.9) + ,fEoverPCutMin(0.8) + ,fEoverPCutMax(1.2) + ,fAngleCut(999) + ,fChi2Cut(3.5) + ,fDCAcut(999) + ,fMassCutFlag(kTRUE) + ,fAngleCutFlag(kFALSE) + ,fChi2CutFlag(kFALSE) + ,fDCAcutFlag(kFALSE) + ,fPtBackgroundBeforeReco(0) + ,fPtBackgroundAfterReco(0) + ,fPtMCparticleAll(0) + ,fPtMCparticleReco(0) + ,fPtMCparticleAllHfe1(0) + ,fPtMCparticleRecoHfe1(0) + ,fPtMCparticleAllHfe2(0) + ,fPtMCparticleRecoHfe2(0) + ,fPtMCelectronAfterAll(0) + ,fPtMCpi0(0) + ,fPtMC_EMCal_All(0) + ,fPtMC_EMCal_Selected(0) + ,fPtMC_TPC_All(0) + ,fPtMC_TPC_Selected(0) + ,fPtMCWithLabel(0) + ,fPtMCWithoutLabel(0) + ,fPtIsPhysicaPrimary(0) + ,fCuts(0) + ,fCFM(0) + ,fPID(new AliHFEpid("hfePid")) + ,fPIDqa(0) + ,fMCstack(0) + ,fRejectKinkMother(kFALSE) + ,fMCtrack(0) + ,fMCtrackMother(0) + ,fMCtrackGMother(0) + ,fMCtrackGGMother(0) + ,fMCtrackGGGMother(0) + ,fMCarray(0) + ,fMCheader(0) + ,fMCparticle(0) + ,fMCparticleMother(0) + ,fMCparticleGMother(0) + ,fMCparticleGGMother(0) + ,fMCparticleGGGMother(0) + ,fEventHandler(0) + ,fMCevent(0) + ,fPoolMgr(0) + ,fPool(0) + ,fTracksClone(0) + ,fTracks(0) + ,fCEtaPhi_Inc_EM(0) + ,fCEtaPhi_ULS_EM(0) + ,fCEtaPhi_LS_EM(0) + ,fCEtaPhi_ULS_Weight_EM(0) + ,fCEtaPhi_LS_Weight_EM(0) + ,fPoolNevents(0) + ,fEventMixingFlag(0) + ,fCEtaPhi_Inc_DiHadron(0) + ,fPtTrigger_Inc(0) +{ + // Constructor + // Define input and output slots here + // Input slot #0 works with a TChain + DefineInput(0, TChain::Class()); + // Output slot #0 id reserved by the base class for AOD + // Output slot #1 writes into a TH1 container + // DefineOutput(1, TH1I::Class()); + DefineOutput(1, TList::Class()); + //DefineOutput(3, TTree::Class()); +} + +//______________________________________________________________________ +AliAnalysisTaskEMCalHFEpA::~AliAnalysisTaskEMCalHFEpA() +{ + //Destructor + delete fOutputList; + delete fPID; + delete fCFM; + delete fPIDqa; +} + +//______________________________________________________________________ +//Create Output Objects +//Here we can define the histograms and others output files +//Called once +void AliAnalysisTaskEMCalHFEpA::UserCreateOutputObjects() +{ +//______________________________________________________________________ +//Initialize PID + if(!fPID->GetNumberOfPIDdetectors()) + { + fPID->AddDetector("TPC", 0); + } + + fPID->SortDetectors(); + + fPIDqa = new AliHFEpidQAmanager(); + fPIDqa->Initialize(fPID); +//______________________________________________________________________ + +//______________________________________________________________________ +//Initialize correction Framework and Cuts + fCFM = new AliCFManager; + const Int_t kNcutSteps = AliHFEcuts::kNcutStepsMCTrack + AliHFEcuts::kNcutStepsRecTrack + AliHFEcuts::kNcutStepsDETrack; + fCFM->SetNStepParticle(kNcutSteps); + for(Int_t istep = 0; istep < kNcutSteps; istep++) fCFM->SetParticleCutsList(istep, NULL); + + if(!fCuts) + { + AliWarning("Cuts not available. Default cuts will be used"); + fCuts = new AliHFEcuts; + fCuts->CreateStandardCuts(); + } + + fCuts->Initialize(fCFM); +//______________________________________________________________________ + +///______________________________________________________________________ +///Output Tlist +//Create TList + fOutputList = new TList(); + fOutputList->SetOwner(); + +//PIDqa + fOutputList->Add(fPIDqa->MakeList("PIDQA")); + +//Store the number of events + //Define the histo + fNevent = new TH1F("fNevent","Number of Events",5,-0.5,4.5); + //And then, add to the output list + fOutputList->Add(fNevent); + + fpid = new TH1F("fpid","PID flag",5,0,5); + fOutputList->Add(fpid); + +//pt Distribution + fPtElec_Inc = new TH1F("fPtElec_Inc","Inclusive Electrons; p_{t} (GeV/c); Count",200,0,20); + fPtElec_ULS = new TH1F("fPtElec_ULS","Inclusive Electrons; p_{t} (GeV/c); Count",200,0,20); + fPtElec_LS = new TH1F("fPtElec_LS","Inclusive Electrons; p_{t} (GeV/c); Count",200,0,20); + fPtTrigger_Inc = new TH1F("fPtTrigger_Inc","pT dist for Hadron Contamination; p_{t} (GeV/c); Count",200,0,20); + fTPCnsigma_pt_2D = new TH2F("fTPCnsigma_pt_2D",";pt (GeV/c);TPC Electron N#sigma",1000,0.3,30,1000,-15,10); + + + fOutputList->Add(fPtElec_Inc); + fOutputList->Add(fPtElec_ULS); + fOutputList->Add(fPtElec_LS); + fOutputList->Add(fPtTrigger_Inc); + fOutputList->Add(fTPCnsigma_pt_2D); + + //General Histograms + + //Steps + //Step 1: Before Track cuts + //Step 2: Before PID + //Step 3: After PID + + fEoverP_pt = new TH2F *[3]; + fTPC_p = new TH2F *[3]; + fTPCnsigma_p = new TH2F *[3]; + + fECluster= new TH1F *[3]; + fEtaPhi= new TH2F *[3]; + fVtxZ= new TH1F *[3]; + fNTracks= new TH1F *[3]; + fNClusters= new TH1F *[3]; + fTPCNcls_EoverP= new TH2F *[3]; + + for(Int_t i = 0; i < 3; i++) + { + fEoverP_pt[i] = new TH2F(Form("fEoverP_pt%d",i),";p_{t} (GeV/c);E / p ",1000,0,30,500,0,2); + fTPC_p[i] = new TH2F(Form("fTPC_p%d",i),";pt (GeV/c);TPC dE/dx (a. u.)",1000,0.3,15,1000,-20,200); + fTPCnsigma_p[i] = new TH2F(Form("fTPCnsigma_p%d",i),";p (GeV/c);TPC Electron N#sigma",1000,0.3,15,1000,-15,10); + + + fECluster[i]= new TH1F(Form("fECluster%d",i), ";ECluster",2000, 0,100); + fEtaPhi[i]= new TH2F(Form("fEtaPhi%d",i),"#eta x #phi Clusters;#phi;#eta",200,0.,TMath::Pi(),50,-1.,1.); + fVtxZ[i]= new TH1F(Form("fVtxZ%d",i),"VtxZ",1000, -50,50); + fNTracks[i]= new TH1F(Form("fNTracks%d",i),"NTracks",1000, 0,1000); + fNClusters[i]= new TH1F(Form("fNClusters%d",i),"fNClusters0",200, 0,100); + fTPCNcls_EoverP[i]= new TH2F(Form("fTPCNcls_EoverP%d",i),"TPCNcls_EoverP",1000,0,200,200,0,2); + + + fOutputList->Add(fEoverP_pt[i]); + fOutputList->Add(fTPC_p[i]); + fOutputList->Add(fTPCnsigma_p[i]); + + + fOutputList->Add(fECluster[i]); + fOutputList->Add(fEtaPhi[i]); + fOutputList->Add(fVtxZ[i]); + fOutputList->Add(fNTracks[i]); + fOutputList->Add(fNClusters[i]); + fOutputList->Add(fTPCNcls_EoverP[i]); + } + + //pt bin + Int_t fPtBin[6] = {2,4,6,8,10,15}; + + fEoverP_tpc = new TH2F *[5]; + fTPC_pt = new TH1F *[5]; + fTPCnsigma_pt = new TH1F *[5]; + + fEta=new TH1F *[5]; + fPhi=new TH1F *[5]; + fR=new TH1F *[5]; + fR_EoverP=new TH2F *[5]; + fNcells=new TH1F *[5]; + fNcells_EoverP=new TH2F *[5]; + fM02_EoverP= new TH2F *[5]; + fM20_EoverP= new TH2F *[5]; + fEoverP_ptbins=new TH1F *[5]; + fECluster_ptbins=new TH1F *[5]; + fEoverP_wSSCut=new TH1F *[5]; + fNcells_electrons=new TH1F *[5]; + fNcells_hadrons=new TH1F *[5]; + fTPCnsigma_eta_electrons=new TH2F *[5]; + fTPCnsigma_eta_hadrons=new TH2F *[5]; + + if(fCorrelationFlag) + { + fCEtaPhi_Inc = new TH2F *[5]; + fCEtaPhi_Inc_DiHadron = new TH2F *[5]; + + fCEtaPhi_ULS = new TH2F *[5]; + fCEtaPhi_LS = new TH2F *[5]; + fCEtaPhi_ULS_NoP = new TH2F *[5]; + fCEtaPhi_LS_NoP = new TH2F *[5]; + + fCEtaPhi_ULS_Weight = new TH2F *[5]; + fCEtaPhi_LS_Weight = new TH2F *[5]; + fCEtaPhi_ULS_NoP_Weight = new TH2F *[5]; + fCEtaPhi_LS_NoP_Weight = new TH2F *[5]; + + fCEtaPhi_Inc_EM = new TH2F *[5]; + + fCEtaPhi_ULS_EM = new TH2F *[5]; + fCEtaPhi_LS_EM = new TH2F *[5]; + + fCEtaPhi_ULS_Weight_EM = new TH2F *[5]; + fCEtaPhi_LS_Weight_EM = new TH2F *[5]; + + fInvMass = new TH1F("fInvMass","",200,0,0.3); + fInvMassBack = new TH1F("fInvMassBack","",200,0,0.3); + fDCA = new TH1F("fDCA","",200,0,1); + fDCABack = new TH1F("fDCABack","",200,0,1); + fOpAngle = new TH1F("fOpAngle","",200,0,0.5); + fOpAngleBack = new TH1F("fOpAngleBack","",200,0,0.5); + + fOutputList->Add(fInvMass); + fOutputList->Add(fInvMassBack); + fOutputList->Add(fDCA); + fOutputList->Add(fDCABack); + fOutputList->Add(fOpAngle); + fOutputList->Add(fOpAngleBack); + } + + for(Int_t i = 0; i < 5; i++) + { + fEoverP_tpc[i] = new TH2F(Form("fEoverP_tpc%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),1000,-15,15,100,0,2); + fTPC_pt[i] = new TH1F(Form("fTPC_pt%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;Count",fPtBin[i],fPtBin[i+1]),200,20,200); + fTPCnsigma_pt[i] = new TH1F(Form("fTPCnsigma_pt%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;Count",fPtBin[i],fPtBin[i+1]),200,-15,10); + + fEta[i]=new TH1F(Form("fEta%d",i), Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; dEta )",fPtBin[i],fPtBin[i+1]),100, -0.1,0.1); + fPhi[i]=new TH1F(Form("fPhi%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; dPhi )",fPtBin[i],fPtBin[i+1]), 100, -0.1,0.1); + fR[i]=new TH1F(Form("fR%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; dR )",fPtBin[i],fPtBin[i+1]), 100, -0.1,0.1); + fR_EoverP[i]=new TH2F(Form("fR_EoverP%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),100, 0,0.1,1000,0,10); + fNcells[i]=new TH1F(Form("fNcells%d",i), Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),100, 0, 30); + fNcells_electrons[i]=new TH1F(Form("fNcells_electrons%d",i), Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma )",fPtBin[i],fPtBin[i+1]),100, 0, 30); + fNcells_hadrons[i]=new TH1F(Form("fNcells_hadrons%d",i), Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; Ncells; E / p )",fPtBin[i],fPtBin[i+1]),100, 0, 30); + fNcells_EoverP[i]=new TH2F(Form("fNcells_EoverP%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; Ncells; E / p )",fPtBin[i],fPtBin[i+1]),1000, 0,20,100,0,30); + fM02_EoverP[i]= new TH2F(Form("fM02_EoverP%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; M02; E / p )",fPtBin[i],fPtBin[i+1]),1000,0,1,100,0,2); + fM20_EoverP[i]= new TH2F(Form("fM20_EoverP%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma; M20; E / p )",fPtBin[i],fPtBin[i+1]),1000,0,1,100,0,2); + fEoverP_ptbins[i] = new TH1F(Form("fEoverP_ptbins%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),500,0,2); + fECluster_ptbins[i]= new TH1F(Form("fECluster_ptbins%d",i), Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),2000, 0,100); + fEoverP_wSSCut[i]=new TH1F(Form("fEoverP_wSSCut%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;E / p )",fPtBin[i],fPtBin[i+1]),500,0,2); + fTPCnsigma_eta_electrons[i]=new TH2F(Form("fTPCnsigma_eta_electrons%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;Eta )",fPtBin[i],fPtBin[i+1]),1000,-15,15,100,-1,1); + fTPCnsigma_eta_hadrons[i]=new TH2F(Form("fTPCnsigma_eta_hadrons%d",i),Form("%d < p_{t} < %d GeV/c;TPC Electron N#sigma;Eta )",fPtBin[i],fPtBin[i+1]),1000,-15,15,100,-1,1); + + fOutputList->Add(fEoverP_tpc[i]); + fOutputList->Add(fTPC_pt[i]); + fOutputList->Add(fTPCnsigma_pt[i]); + + fOutputList->Add(fEta[i]); + fOutputList->Add(fPhi[i]); + fOutputList->Add(fR[i]); + fOutputList->Add(fR_EoverP[i]); + fOutputList->Add(fNcells[i]); + fOutputList->Add(fNcells_electrons[i]); + fOutputList->Add(fNcells_hadrons[i]); + fOutputList->Add(fNcells_EoverP[i]); + fOutputList->Add(fECluster_ptbins[i]); + fOutputList->Add(fEoverP_ptbins[i]); + fOutputList->Add(fEoverP_wSSCut[i]); + fOutputList->Add(fM02_EoverP[i]); + fOutputList->Add(fM20_EoverP[i]); + fOutputList->Add(fTPCnsigma_eta_electrons[i]); + fOutputList->Add(fTPCnsigma_eta_hadrons[i]); + + + if(fCorrelationFlag) + { + fCEtaPhi_Inc[i] = new TH2F(Form("fCEtaPhi_Inc%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_Inc_DiHadron[i] = new TH2F(Form("fCEtaPhi_Inc_DiHadron%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fCEtaPhi_ULS[i] = new TH2F(Form("fCEtaPhi_ULS%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS[i] = new TH2F(Form("fCEtaPhi_LS%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_ULS_NoP[i] = new TH2F(Form("fCEtaPhi_ULS_NoP%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS_NoP[i] = new TH2F(Form("fCEtaPhi_LS_NoP%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fCEtaPhi_ULS_Weight[i] = new TH2F(Form("fCEtaPhi_ULS_Weight%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS_Weight[i] = new TH2F(Form("fCEtaPhi_LS_Weight%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_ULS_NoP_Weight[i] = new TH2F(Form("fCEtaPhi_ULS_NoP_Weight%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS_NoP_Weight[i] = new TH2F(Form("fCEtaPhi_LS_NoP_Weight%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fOutputList->Add(fCEtaPhi_Inc[i]); + fOutputList->Add(fCEtaPhi_Inc_DiHadron[i]); + + fOutputList->Add(fCEtaPhi_ULS[i]); + fOutputList->Add(fCEtaPhi_LS[i]); + fOutputList->Add(fCEtaPhi_ULS_NoP[i]); + fOutputList->Add(fCEtaPhi_LS_NoP[i]); + + fOutputList->Add(fCEtaPhi_ULS_Weight[i]); + fOutputList->Add(fCEtaPhi_LS_Weight[i]); + fOutputList->Add(fCEtaPhi_ULS_NoP_Weight[i]); + fOutputList->Add(fCEtaPhi_LS_NoP_Weight[i]); + + if(fEventMixingFlag) + { + fCEtaPhi_Inc_EM[i] = new TH2F(Form("fCEtaPhi_Inc_EM%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fCEtaPhi_ULS_EM[i] = new TH2F(Form("fCEtaPhi_ULS_EM%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS_EM[i] = new TH2F(Form("fCEtaPhi_LS_EM%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fCEtaPhi_ULS_Weight_EM[i] = new TH2F(Form("fCEtaPhi_ULS_Weight_EM%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + fCEtaPhi_LS_Weight_EM[i] = new TH2F(Form("fCEtaPhi_LS_Weight_EM%d",i),Form("%d < p_{t} < %d GeV/c;#DeltaPhi (rad);#Delta#eta",fPtBin[i],fPtBin[i+1]),200,-0.5*TMath::Pi(),1.5*TMath::Pi(),200,-2,2); + + fOutputList->Add(fCEtaPhi_Inc_EM[i]); + + fOutputList->Add(fCEtaPhi_ULS_EM[i]); + fOutputList->Add(fCEtaPhi_LS_EM[i]); + + fOutputList->Add(fCEtaPhi_ULS_Weight_EM[i]); + fOutputList->Add(fCEtaPhi_LS_Weight_EM[i]); + } + } + } + + //pt integrated + fTPCnsigma_eta = new TH2F("fTPCnsigma_eta",";Pseudorapidity #eta; TPC signal - _{elec} (#sigma)",200,-0.9,0.9,200,-15,15); + fTPCnsigma_phi = new TH2F("fTPCnsigma_phi",";Azimuthal Angle #phi; TPC signal - _{elec} (#sigma)",200,0,2*TMath::Pi(),200,-15,15); + + + fNcells_pt=new TH2F("fNcells_pt","fNcells_pt",1000, 0,20,100,0,30); + fEoverP_pt_pions= new TH2F("fEoverP_pt_pions","fEoverP_pt_pions",1000,0,30,500,0,2); + + ftpc_p_EoverPcut= new TH2F("ftpc_p_EoverPcut","ftpc_p_EoverPcut",1000,0,30,200,20,200); + fnsigma_p_EoverPcut= new TH2F("fnsigma_p_EoverPcut","fnsigma_p_EoverPcut",1000,0,30,500,-15,15); + + fEoverP_pt_pions2= new TH2F("fEoverP_pt_pions2","fEoverP_pt_pions2",1000,0,30,500,0,2); + fEoverP_pt_hadrons= new TH2F("fEoverP_pt_hadrons","fEoverP_pt_hadrons",1000,0,30,500,0,2); + + + fOutputList->Add(fTPCnsigma_eta); + fOutputList->Add(fTPCnsigma_phi); + + fOutputList->Add(fNcells_pt); + fOutputList->Add(fEoverP_pt_pions); + + fOutputList->Add(ftpc_p_EoverPcut); + fOutputList->Add(fnsigma_p_EoverPcut); + + fOutputList->Add(fEoverP_pt_pions2); + fOutputList->Add(fEoverP_pt_hadrons); + + //__________________________________________________________________ + //Efficiency studies + if(fIsMC) + { + fPtBackgroundBeforeReco = new TH1F("fPtBackgroundBeforeReco",";p_{t} (GeV/c);Count",30,0,15); + fPtBackgroundAfterReco = new TH1F("fPtBackgroundAfterReco",";p_{t} (GeV/c);Count",30,0,15); + fPtMCparticleAll = new TH1F("fPtMCparticleAll",";p_{t} (GeV/c);Count",200,0,40); + fPtMCparticleReco = new TH1F("fPtMCparticleReco",";p_{t} (GeV/c);Count",200,0,40); + fPtMCparticleAllHfe1 = new TH1F("fPtMCparticleAllHfe1",";p_{t} (GeV/c);Count",200,0,40); + fPtMCparticleRecoHfe1 = new TH1F("fPtMCparticleRecoHfe1",";p_{t} (GeV/c);Count",200,0,40); + fPtMCparticleAllHfe2 = new TH1F("fPtMCparticleAllHfe2",";p_{t} (GeV/c);Count",200,0,40); + fPtMCparticleRecoHfe2 = new TH1F("fPtMCparticleRecoHfe2",";p_{t} (GeV/c);Count",200,0,40); + fPtMCelectronAfterAll = new TH1F("fPtMCelectronAfterAll",";p_{t} (GeV/c);Count",200,0,40); + fPtMCpi0 = new TH1F("fPtMCpi0",";p_{t} (GeV/c);Count",200,0,40); + fPtMC_EMCal_All= new TH1F("fPtMC_EMCal_All",";p_{t} (GeV/c);Count",200,0,40); + fPtMC_EMCal_Selected= new TH1F("fPtMC_EMCal_Selected",";p_{t} (GeV/c);Count",200,0,40); + fPtMC_TPC_All= new TH1F("fPtMC_TPC_All",";p_{t} (GeV/c);Count",200,0,40); + fPtMC_TPC_Selected = new TH1F("fPtMC_TPC_Selected",";p_{t} (GeV/c);Count",200,0,40); + fPtMCWithLabel = new TH1F("fPtMCWithLabel",";p_{t} (GeV/c);Count",200,0,40); + fPtMCWithoutLabel = new TH1F("fPtMCWithoutLabel",";p_{t} (GeV/c);Count",200,0,40); + fPtIsPhysicaPrimary = new TH1F("fPtIsPhysicaPrimary",";p_{t} (GeV/c);Count",200,0,40); + + fOutputList->Add(fPtBackgroundBeforeReco); + fOutputList->Add(fPtBackgroundAfterReco); + fOutputList->Add(fPtMCparticleAll); + fOutputList->Add(fPtMCparticleReco); + fOutputList->Add(fPtMCparticleAllHfe1); + fOutputList->Add(fPtMCparticleRecoHfe1); + fOutputList->Add(fPtMCparticleAllHfe2); + fOutputList->Add(fPtMCparticleRecoHfe2); + fOutputList->Add(fPtMCelectronAfterAll); + fOutputList->Add(fPtMCpi0); + fOutputList->Add(fPtMC_EMCal_All); + fOutputList->Add(fPtMC_EMCal_Selected); + fOutputList->Add(fPtMC_TPC_All); + fOutputList->Add(fPtMC_TPC_Selected); + fOutputList->Add(fPtMCWithLabel); + fOutputList->Add(fPtMCWithoutLabel); + fOutputList->Add(fPtIsPhysicaPrimary); + } + + fCentralityHist = new TH1F("fCentralityHist",";Centrality (%); Count",1000000,0,100); + fCentralityHistPass = new TH1F("fCentralityHistPass",";Centrality (%); Count",1000000,0,100); + fOutputList->Add(fCentralityHist); + fOutputList->Add(fCentralityHistPass); + + //______________________________________________________________________ + //Mixed event analysis + if(fEventMixingFlag) + { + fPoolNevents = new TH1F("fPoolNevents","Event Mixing Statistics; Number of events; Count",1000,0,1000); + fOutputList->Add(fPoolNevents); + + Int_t trackDepth = 2000; // number of objects (tracks) kept per event buffer bin. Once the number of stored objects (tracks) is above that limit, the oldest ones are removed. + Int_t poolsize = 1000; // Maximum number of events, ignored in the present implemented of AliEventPoolManager + + Int_t nCentralityBins = 15; + Double_t centralityBins[] = { 0, 1, 2, 3, 4, 5, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100.1 }; + + Int_t nZvtxBins = 9; + Double_t vertexBins[] = {-10, -7, -5, -3, -1, 1, 3, 5, 7, 10}; + + fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, (Double_t*) centralityBins, nZvtxBins, (Double_t*) vertexBins); + } + //______________________________________________________________________ + + PostData(1, fOutputList); + +///______________________________________________________________________ +} + +//______________________________________________________________________ +//Main loop +//Called for each event +void AliAnalysisTaskEMCalHFEpA::UserExec(Option_t *) +{ +//Check Event + fESD = dynamic_cast(InputEvent()); + fAOD = dynamic_cast(InputEvent()); + + if(!(fESD || fAOD)) + { + printf("ERROR: fESD & fAOD not available\n"); + return; + } + + fVevent = dynamic_cast(InputEvent()); + + if(!fVevent) + { + printf("ERROR: fVEvent not available\n"); + return; + } + +//Check Cuts + if(!fCuts) + { + AliError("HFE cuts not available"); + return; + } +//Check PID + if(!fPID->IsInitialized()) + { + // Initialize PID with the given run number + AliWarning("PID not initialised, get from Run no"); + + if(fIsAOD) + { + fPID->InitializePID(fAOD->GetRunNumber()); + } + else + { + fPID->InitializePID(fESD->GetRunNumber()); + } + } + +//PID response + fPidResponse = fInputHandler->GetPIDResponse(); + + +//Check PID response + if(!fPidResponse) + { + AliDebug(1, "Using default PID Response"); + fPidResponse = AliHFEtools::GetDefaultPID(kFALSE, fInputEvent->IsA() == AliAODEvent::Class()); + } + + fPID->SetPIDResponse(fPidResponse); + + fCFM->SetRecEventInfo(fVevent); + + Double_t *fListOfmotherkink = 0; + Int_t fNumberOfVertices = 0; + Int_t fNumberOfMotherkink = 0; + +//______________________________________________________________________ +//Vertex Selection + if(fIsAOD) + { + const AliAODVertex* trkVtx = fAOD->GetPrimaryVertex(); + if(!trkVtx || trkVtx->GetNContributors()<=0) return; + TString vtxTtl = trkVtx->GetTitle(); + if(!vtxTtl.Contains("VertexerTracks")) return; + Float_t zvtx = trkVtx->GetZ(); + fZvtx = zvtx; + const AliAODVertex* spdVtx = fAOD->GetPrimaryVertexSPD(); + if(spdVtx->GetNContributors()<=0) return; + TString vtxTyp = spdVtx->GetTitle(); + Double_t cov[6]={0}; + spdVtx->GetCovarianceMatrix(cov); + Double_t zRes = TMath::Sqrt(cov[5]); + if(vtxTyp.Contains("vertexer:Z") && (zRes>0.25)) return; + if(TMath::Abs(spdVtx->GetZ() - trkVtx->GetZ())>0.5) return; + if(TMath::Abs(zvtx) > 10) return; + + //Look for kink mother for AOD + + fNumberOfVertices = 0; + fNumberOfMotherkink = 0; + + if(fIsAOD) + { + fNumberOfVertices = fAOD->GetNumberOfVertices(); + + fListOfmotherkink = new Double_t[fNumberOfVertices]; + + for(Int_t ivertex=0; ivertex < fNumberOfVertices; ivertex++) + { + AliAODVertex *aodvertex = fAOD->GetVertex(ivertex); + if(!aodvertex) continue; + if(aodvertex->GetType()==AliAODVertex::kKink) + { + AliAODTrack *mother1 = (AliAODTrack *) aodvertex->GetParent(); + if(!mother1) continue; + Int_t idmother = mother1->GetID(); + fListOfmotherkink[fNumberOfMotherkink] = idmother; + fNumberOfMotherkink++; + } + } + } + } + else + { + /// (has to be checked) + + const AliESDVertex *trkVtx = fESD->GetPrimaryVertex(); + Float_t zvtx = trkVtx->GetZ(); + fZvtx = zvtx; + if(TMath::Abs(zvtx) > 10) return; + + //Double_t NTracks = fESD->GetNumberOfTracks(); + + /*const AliVVertex* trkVtx = fVevent->GetPrimaryVertex(); + if(!trkVtx || trkVtx->GetNContributors()<=0) return; + TString vtxTtl = trkVtx->GetTitle(); + if(!vtxTtl.Contains("VertexerTracks")) return; + Float_t zvtx = trkVtx->GetZ(); + const AliVVertex* spdVtx = fVevent->GetPrimaryVertexSPD(); + if(spdVtx->GetNContributors()<=0) return; + TString vtxTyp = spdVtx->GetTitle(); + Double_t cov[6]={0}; + spdVtx->GetCovarianceMatrix(cov); + Double_t zRes = TMath::Sqrt(cov[5]); + if(vtxTyp.Contains("vertexer:Z") && (zRes>0.25)) return; + if(TMath::Abs(spdVtx->GetZ() - trkVtx->GetZ())>0.5) return; + if(TMath::Abs(zvtx) > 10) return;*/ + } + +//______________________________________________________________________ + +//Only events with at least 2 tracks are accepted + Int_t fNOtrks = fVevent->GetNumberOfTracks(); + if(fNOtrks<2) return; + +//______________________________________________________________________ +//Centrality Selection + if(fHasCentralitySelection) + { + Float_t centrality = -1; + + if(fIsAOD) + { + fCentrality = fAOD->GetHeader()->GetCentralityP(); + } + else + { + fCentrality = fESD->GetCentrality(); + } + + if(fEstimator==1) centrality = fCentrality->GetCentralityPercentile("ZDC"); + else centrality = fCentrality->GetCentralityPercentile("V0A"); + + //cout << "Centrality = " << centrality << " %" << endl; + + fCentralityHist->Fill(centrality); + + if(centralityfCentralityMax) return; + + fCentralityHistPass->Fill(centrality); + } +//______________________________________________________________________ + +//______________________________________________________________________ + + if(fIsMC) + { + if(fIsAOD) + { + fMCarray = dynamic_cast(fAOD->FindListObject(AliAODMCParticle::StdBranchName())); + + if(!fMCarray) + { + AliError("Array of MC particles not found"); + return; + } + + fMCheader = dynamic_cast(fAOD->GetList()->FindObject(AliAODMCHeader::StdBranchName())); + + if(!fMCheader) + { + AliError("Could not find MC Header in AOD"); + return; + } + + for(Int_t iMC = 0; iMC < fMCarray->GetEntries(); iMC++) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(iMC); + + Int_t pdg = fMCparticle->GetPdgCode(); + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax && fMCparticle->Charge()!=0) + { + if( TMath::Abs(pdg) == 211 || TMath::Abs(pdg) == 2212 || TMath::Abs(pdg) == 321 || TMath::Abs(pdg) == 11 || TMath::Abs(pdg) == 13 ) + { + if(fMCparticle->IsPhysicalPrimary()) + { + fPtMCparticleAll->Fill(fMCparticle->Pt()); + + Bool_t MotherFound = FindMother(iMC); + if(MotherFound) + { + if(fIsHFE1) fPtMCparticleAllHfe1->Fill(fMCparticle->Pt()); //denominator for total efficiency + if(fIsHFE2) fPtMCparticleAllHfe2->Fill(fMCparticle->Pt()); + } + } + } + } + if(TMath::Abs(pdg)==111) fPtMCpi0->Fill(fMCparticle->Pt()); + } + } + else + { + fEventHandler = dynamic_cast (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()); + if (!fEventHandler) { + Printf("ERROR: Could not retrieve MC event handler"); + return; + } + + fMCevent = fEventHandler->MCEvent(); + if (!fMCevent) { + Printf("ERROR: Could not retrieve MC event"); + return; + } + + fMCstack = fMCevent->Stack(); + + for(Int_t iMC = 0; iMC < fMCstack->GetNtrack(); iMC++) + { + + fMCtrack = fMCstack->Particle(iMC); + Int_t pdg = fMCtrack->GetPdgCode(); + if(TMath::Abs(pdg)==111) fPtMCpi0->Fill(fMCtrack->Pt()); + + if(!fMCstack->IsPhysicalPrimary(iMC)) continue; + + + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax) + { + if( TMath::Abs(pdg) == 211 || TMath::Abs(pdg) == 2212 || TMath::Abs(pdg) == 321 || TMath::Abs(pdg) == 11 || TMath::Abs(pdg) == 13 ) + { + fPtMCparticleAll->Fill(fMCtrack->Pt()); + + Bool_t MotherFound = FindMother(iMC); + if(MotherFound) + { + if(fIsHFE1) fPtMCparticleAllHfe1->Fill(fMCtrack->Pt());//denominator for total efficiency + if(fIsHFE2) fPtMCparticleAllHfe2->Fill(fMCtrack->Pt()); + } + } + } + } + } + } + +//______________________________________________________________________ +//EMCal Trigger Selection (Threshould selection) + TString firedTrigger; + TString TriggerEG1("EG1"); + TString TriggerEG2("EG2"); + + if(fAOD) firedTrigger = fAOD->GetFiredTriggerClasses(); + else if(fESD) firedTrigger = fESD->GetFiredTriggerClasses(); + + fNevent->Fill(0); + if(firedTrigger.Contains(TriggerEG1)) fNevent->Fill(1); + if(firedTrigger.Contains(TriggerEG2)) fNevent->Fill(2); + + //EG1 + if(firedTrigger.Contains(TriggerEG1)) + { + fNevent->Fill(3); + } + else + { + if(fEMCEG1) return; + } + + //EG2 + if(firedTrigger.Contains(TriggerEG2)) + { + fNevent->Fill(4); + } + else + { + if(fEMCEG2) return; + } + +//______________________________________________________________________ + + Int_t ClsNo = -999; + if(!fIsAOD) ClsNo = fESD->GetNumberOfCaloClusters(); + else ClsNo = fAOD->GetNumberOfCaloClusters(); + +//______________________________________________________________________ + +///______________________________________________________________________ +///Track loop + for(Int_t iTracks = 0; iTracks < fVevent->GetNumberOfTracks(); iTracks++) + { + AliVParticle* Vtrack = fVevent->GetTrack(iTracks); + if (!Vtrack) + { + printf("ERROR: Could not receive track %d\n", iTracks); + continue; + } + + AliVTrack *track = dynamic_cast(Vtrack); + AliESDtrack *etrack = dynamic_cast(Vtrack); + AliAODTrack *atrack = dynamic_cast(Vtrack); + + Double_t fTPCnSigma = -999; + Double_t fTPCnSigma_pion = -999; + Double_t fTPCnSigma_proton = -999; + Double_t fTPCnSigma_kaon = -999; + Double_t fTPCsignal = -999; + Double_t fPt = -999; + Double_t fP = -999; + + ///_____________________________________________________________________________ + ///Fill QA plots without track selection + fPt = track->Pt(); + fP = TMath::Sqrt((track->Pt())*(track->Pt()) + (track->Pz())*(track->Pz())); + + fTPCsignal = track->GetTPCsignal(); + fTPCnSigma = fPidResponse->NumberOfSigmasTPC(track, AliPID::kElectron); + fTPCnSigma_pion = fPidResponse->NumberOfSigmasTPC(track, AliPID::kPion); + fTPCnSigma_proton = fPidResponse->NumberOfSigmasTPC(track, AliPID::kProton); + fTPCnSigma_kaon = fPidResponse->NumberOfSigmasTPC(track, AliPID::kKaon); + + fTPC_p[0]->Fill(fPt,fTPCsignal); + fTPCnsigma_p[0]->Fill(fP,fTPCnSigma); + + + Float_t TPCNcls = track->GetTPCNcls(); + Float_t pos[3]={0,0,0}; + + Double_t fEMCflag = kFALSE; + if(track->GetEMCALcluster()>0) + { + fClus = fVevent->GetCaloCluster(track->GetEMCALcluster()); + if(fClus->IsEMCAL()) + { + if(TMath::Abs(fClus->GetTrackDx())<=0.05 && TMath::Abs(fClus->GetTrackDz())<=0.05) + { + fEMCflag = kTRUE; + fEoverP_pt[0]->Fill(fPt,(fClus->E() / fP)); + + + Float_t Energy = fClus->E(); + Float_t EoverP = Energy/track->P(); + //Float_t M02 = fClus->GetM02(); + //Float_t M20 = fClus->GetM20(); + + /////////////// for Eta Phi distribution + fClus->GetPosition(pos); + TVector3 vpos(pos[0],pos[1],pos[2]); + Double_t cphi = vpos.Phi(); + Double_t ceta = vpos.Eta(); + fEtaPhi[0]->Fill(cphi,ceta); + + fECluster[0]->Fill(Energy); + fTPCNcls_EoverP[0]->Fill(TPCNcls, EoverP); + } + } + } + + //______________________________________________________________ + // Vertex + + fVtxZ[0]->Fill(fZvtx); + fNTracks[0]->Fill(fNOtrks); + fNClusters[0]->Fill(ClsNo); + + //______________________________________________________________ + + ///Fill QA plots without track selection + ///_____________________________________________________________________________ +//______________________________________________________________________________________ +//Track Selection Cuts + +//AOD (Test Filter Bit) + if(fIsAOD) + { + // standard cuts with very loose DCA - BIT(4) + // Description: + /* + GetStandardITSTPCTrackCuts2011(kFALSE) + SetMaxChi2PerClusterTPC(4); + SetAcceptKinkDaughters(kFALSE); + SetRequireTPCRefit(kTRUE); + SetRequireITSRefit(kTRUE); + SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny); + SetMaxDCAToVertexZ(2); + SetMaxDCAToVertex2D(kFALSE); + SetRequireSigmaToVertex(kFALSE); + SetMaxChi2PerClusterITS(36); + SetMaxDCAToVertexXY(2.4) + SetMaxDCAToVertexZ(3.2) + SetDCaToVertex2D(kTRUE) + */ + + if(!atrack->TestFilterMask(AliAODTrack::kTrkGlobalNoDCA)) continue; + } + +//RecKine: ITSTPC cuts + if(!ProcessCutStep(AliHFEcuts::kStepRecKineITSTPC, track)) continue; +//RecKink + if(fRejectKinkMother) + { + if(fIsAOD) + { + Bool_t kinkmotherpass = kTRUE; + for(Int_t kinkmother = 0; kinkmother < fNumberOfMotherkink; kinkmother++) + { + if(track->GetID() == fListOfmotherkink[kinkmother]) + { + kinkmotherpass = kFALSE; + continue; + } + } + if(!kinkmotherpass) continue; + } + else + { + if(etrack->GetKinkIndex(0) != 0) continue; + } + } + +//RecPrim + if(!fIsAOD) + { + if(!ProcessCutStep(AliHFEcuts::kStepRecPrim, track)) continue; + } + +//HFEcuts: ITS layers cuts + if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsITS, track)) continue; + +//HFE cuts: TPC PID cleanup + if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTPC, track)) continue; +//______________________________________________________________________________________ + + ///_____________________________________________________________ + ///QA plots after track selection + if(fIsMC) + { + if(track->GetLabel()>=0) fPtMCWithLabel->Fill(fPt); + else fPtMCWithoutLabel->Fill(fPt); + } + + if(fIsMC && fIsAOD && track->GetLabel()>=0) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + + if(fMCparticle->IsPhysicalPrimary()) fPtIsPhysicaPrimary->Fill(fPt); + + Int_t pdg = fMCparticle->GetPdgCode(); + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax && fMCparticle->Charge()!=0) + { + if( TMath::Abs(pdg) == 211 || TMath::Abs(pdg) == 2212 || TMath::Abs(pdg) == 321 || TMath::Abs(pdg) == 11 || TMath::Abs(pdg) == 13 ) + { + if(fMCparticle->IsPhysicalPrimary()) + { + fPtMCparticleReco->Fill(fMCparticle->Pt()); + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound) + { + if(fIsHFE1) fPtMCparticleRecoHfe1->Fill(fMCparticle->Pt()); + if(fIsHFE2) fPtMCparticleRecoHfe2->Fill(fMCparticle->Pt()); + } + } + } + } + } + else if(fIsMC && track->GetLabel()>=0) + { + if(fMCstack->IsPhysicalPrimary(track->GetLabel())) + { + fPtIsPhysicaPrimary->Fill(fPt); + fMCtrack = fMCstack->Particle(track->GetLabel()); + + Int_t pdg = fMCtrack->GetPdgCode(); + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax) + { + if( TMath::Abs(pdg) == 211 || TMath::Abs(pdg) == 2212 || TMath::Abs(pdg) == 321 || TMath::Abs(pdg) == 11 || TMath::Abs(pdg) == 13 ) + { + fPtMCparticleReco->Fill(fMCtrack->Pt()); + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound) + { + if(fIsHFE1) fPtMCparticleRecoHfe1->Fill(fMCtrack->Pt()); + if(fIsHFE2) fPtMCparticleRecoHfe2->Fill(fMCtrack->Pt()); + } + } + } + } + } + + fTPC_p[1]->Fill(fPt,fTPCsignal); + fTPCnsigma_p[1]->Fill(fP,fTPCnSigma); + Double_t fPtBin[6] = {2,4,6,8,10,15}; + + TPCNcls = track->GetTPCNcls(); + Float_t pos2[3]={0,0,0}; + + if(track->GetEMCALcluster()>0) + { + fClus = fVevent->GetCaloCluster(track->GetEMCALcluster()); + if(fClus->IsEMCAL()) + { + if(TMath::Abs(fClus->GetTrackDx())<=0.05 && TMath::Abs(fClus->GetTrackDz())<=0.05) + { + fEoverP_pt[1]->Fill(fPt,(fClus->E() / fP)); + + Float_t Energy = fClus->E(); + Float_t EoverP = Energy/track->P(); + //Float_t M02 = fClus->GetM02(); + //Float_t M20 = fClus->GetM20(); + Float_t ncells = fClus->GetNCells(); + + /////////////// for Eta Phi distribution + fClus->GetPosition(pos2); + TVector3 vpos(pos2[0],pos2[1],pos2[2]); + Double_t cphi = vpos.Phi(); + Double_t ceta = vpos.Eta(); + fEtaPhi[1]->Fill(cphi,ceta); + + fECluster[1]->Fill(Energy); + fTPCNcls_EoverP[1]->Fill(TPCNcls, EoverP); + + // 31/05/2103 + //for EMCal triggger performance + if(EoverP > 0.9){ + ftpc_p_EoverPcut->Fill(track->P(), fTPCsignal); + fnsigma_p_EoverPcut->Fill(track->P(), fTPCnSigma); + + } + + + //for hadron contamination calculations + + // 31/05/2103 + // EtaCut -> dados + if(track->Eta()>=fEtaCutMin && track->Eta()<=fEtaCutMax ){ + //main + if(TMath::Abs(fTPCnSigma_pion)<3 || TMath::Abs(fTPCnSigma_proton)<3 || TMath::Abs(fTPCnSigma_kaon)<3 ){ + + if(fTPCnSigma<-3.5){ + fEoverP_pt_hadrons->Fill(fPt,EoverP); + } + } + + if(fTPCnSigma < -3.5){ + fEoverP_pt_pions->Fill(fPt, EoverP); + + } + + if(fTPCnSigma < -3.5 && fTPCnSigma > -10){ + fEoverP_pt_pions2->Fill(fPt, EoverP); + + } + } + + for(Int_t i = 0; i < 5; i++) + { + if(fPt>=fPtBin[i] && fPt -10){ + fNcells_hadrons[i]->Fill(ncells); + } + //hadrons selection using E/p + if((fClus->E() / fP) >= 0.2 && (fClus->E() / fP) <= 0.7){ + fTPCnsigma_eta_hadrons[i]->Fill(fTPCnSigma, ceta); + } + //electrons selection using E/p + if((fClus->E() / fP) >= fEoverPCutMin && (fClus->E() / fP) <= fEoverPCutMax) { + fTPCnsigma_eta_electrons[i]->Fill(fTPCnSigma, ceta); + } + } + } + + if((fClus->E() / fP) >= fEoverPCutMin && (fClus->E() / fP) <= fEoverPCutMax) + { + fTPCnsigma_eta->Fill(track->Eta(),fTPCnSigma); + fTPCnsigma_phi->Fill(track->Phi(),fTPCnSigma); + + if(fUseEMCal) + { + if(fTPCnSigma < 3.5 && fCorrelationFlag) + { + fPtTrigger_Inc->Fill(fPt); + DiHadronCorrelation(track, iTracks); + } + } + } + + } + } + } + + //______________________________________________________________ + // Vertex + + fVtxZ[1]->Fill(fZvtx); + fNTracks[1]->Fill(fNOtrks); + fNClusters[1]->Fill(ClsNo); + //______________________________________________________________ + + ///______________________________________________________________________ + ///Histograms for PID Studies + //Double_t fPtBin[6] = {2,4,6,8,10,15}; + + for(Int_t i = 0; i < 5; i++) + { + if(fPt>=fPtBin[i] && fPtFill(fTPCnSigma,(fClus->E() / fP)); + + + fTPC_pt[i]->Fill(fTPCsignal); + fTPCnsigma_pt[i]->Fill(fTPCnSigma); + + } + } + + ///QA plots after track selection + ///_____________________________________________________________ + + //_______________________________________________________ + //Correlation Analysis - DiHadron + if(!fUseEMCal) + { + if(fTPCnSigma < 3.5 && fCorrelationFlag) + { + fPtTrigger_Inc->Fill(fPt); + DiHadronCorrelation(track, iTracks); + } + } + //_______________________________________________________ + + + //!////////////////////////////////////////////////////////////////// + ///TPC - efficiency calculation // + + /// changing start here + if(fIsMC && fIsAOD && track->GetLabel()>=0) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + Int_t pdg = fMCparticle->GetPdgCode(); + + // + if(fMCparticle->IsPhysicalPrimary()){ + + + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax ){ + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound){ + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + if( TMath::Abs(pdg) == 11 && fMCparticleMother->GetPdgCode()!=22 ){ + if(fIsHFE1) fPtMC_TPC_All->Fill(fMCparticle->Pt()); + } + } + } + } + }///until here + + else if(fIsMC && track->GetLabel()>=0)//ESD + { + + if(fMCstack->IsPhysicalPrimary(track->GetLabel())){ + fMCtrack = fMCstack->Particle(track->GetLabel()); + + Int_t pdg = fMCtrack->GetPdgCode(); + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax ){ + + if(fMCtrack->GetFirstMother()>0){ + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + if( TMath::Abs(pdg) == 11 && fMCtrackMother->GetPdgCode()!=22 ){ + if(fIsHFE1) fPtMC_TPC_All->Fill(fMCtrack->Pt()); + } + } + } + } + } + + +///________________________________________________________________________ +///PID +///Here the PID cuts defined in the file "ConfigEMCalHFEpA.C" is applied + Int_t pidpassed = 1; + AliHFEpidObject hfetrack; + hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis); + hfetrack.SetRecTrack(track); + hfetrack.SetPP(); //proton-proton analysis + if(!fPID->IsSelected(&hfetrack, NULL, "", fPIDqa)) pidpassed = 0; + fpid->Fill(pidpassed); + + if(pidpassed==0) continue; +///________________________________________________________________________ + + + //!////////////////////////////////////////////////////////////////// + ///TPC efficiency calculations + + /// changing start here + if(fIsMC && fIsAOD && track->GetLabel()>=0) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + Int_t pdg = fMCparticle->GetPdgCode(); + + // + if(fMCparticle->IsPhysicalPrimary()){ + + + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax ){ + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound){ + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + if( TMath::Abs(pdg) == 11 && fMCparticleMother->GetPdgCode()!=22 ){ + if(fIsHFE1) fPtMC_TPC_Selected->Fill(fMCparticle->Pt()); + } + } + } + } + }///until here + + else if(fIsMC && track->GetLabel()>=0)//ESD + { + + if(fMCstack->IsPhysicalPrimary(track->GetLabel())){ + fMCtrack = fMCstack->Particle(track->GetLabel()); + + Int_t pdg = fMCtrack->GetPdgCode(); + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax ){ + + if(fMCtrack->GetFirstMother()>0){ + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + if( TMath::Abs(pdg) == 11 && fMCtrackMother->GetPdgCode()!=22 ){ + if(fIsHFE1) fPtMC_TPC_Selected->Fill(fMCtrack->Pt()); + } + } + } + } + } + + //Eta Cut for TPC only + if(track->Eta()>=fEtaCutMin && track->Eta()<=fEtaCutMax ){ + fTPCnsigma_pt_2D->Fill(fPt,fTPCnSigma); + } + + fTPCnsigma_p[2]->Fill(fP,fTPCnSigma); + fTPC_p[2]->Fill(fP,fTPCsignal); + TPCNcls = track->GetTPCNcls(); + Float_t pos3[3]={0,0,0}; + + if(track->GetEMCALcluster()>0) + { + fClus = fVevent->GetCaloCluster(track->GetEMCALcluster()); + if(fClus->IsEMCAL()) + { + + + /////////////// Residuals + Double_t Dx = fClus->GetTrackDx(); + Double_t Dz = fClus->GetTrackDz(); + Double_t R=TMath::Sqrt(Dx*Dx+Dz*Dz); + for(Int_t i = 0; i < 5; i++) + { + if(fPt>=fPtBin[i] && fPtFill(Dz); + fPhi[i]->Fill(Dx); + fR[i]->Fill(R); + } + } + + if(TMath::Abs(fClus->GetTrackDx())<=0.05 && TMath::Abs(fClus->GetTrackDz())<=0.05) + { + + // EtaCut + if(track->Eta()>=fEtaCutMin && track->Eta()<=fEtaCutMax ){ + fEoverP_pt[2]->Fill(fPt,(fClus->E() / fP)); + } + + Float_t Energy = fClus->E(); + Float_t EoverP = Energy/track->P(); + Float_t M02 = fClus->GetM02(); + Float_t M20 = fClus->GetM20(); + Float_t ncells = fClus->GetNCells(); + + + /////////////// for Eta Phi distribution + fClus->GetPosition(pos3); + TVector3 vpos(pos3[0],pos3[1],pos3[2]); + Double_t cphi = vpos.Phi(); + Double_t ceta = vpos.Eta(); + fEtaPhi[2]->Fill(cphi,ceta); + + + + fTPCNcls_EoverP[2]->Fill(TPCNcls, EoverP); + + for(Int_t i = 0; i < 5; i++) + { + if(fPt>=fPtBin[i] && fPtFill(R, EoverP); + fNcells[i]->Fill(ncells); + fNcells_EoverP[i]->Fill(EoverP, ncells); + fM02_EoverP[i]->Fill(M02,EoverP); + fM20_EoverP[i]->Fill(M20,EoverP); + fECluster_ptbins[i]->Fill(Energy); + fEoverP_ptbins[i]->Fill(EoverP); + + if((fClus->E() / fP) >= fEoverPCutMin && (fClus->E() / fP) <= fEoverPCutMax) { + fNcells_electrons[i]->Fill(ncells); + } + + if(M02<0.5 && M20<0.3) { + fEoverP_wSSCut[i]->Fill(EoverP); + } + } + } + + fNcells_pt->Fill(fPt, ncells); + + + //!////////////////////////////////////////////////////////////////// + ///EMCal - Efficiency calculations + + /// changing start here + if(fIsMC && fIsAOD && track->GetLabel()>=0) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + Int_t pdg = fMCparticle->GetPdgCode(); + + // + if(fMCparticle->IsPhysicalPrimary()){ + + + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax ){ + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound){ + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + if( TMath::Abs(pdg) == 11 && fMCparticleMother->GetPdgCode()!=22 ){ + fPtMC_EMCal_All->Fill(fMCparticle->Pt()); + } + } + } + } + }///until here + + else if(fIsMC && track->GetLabel()>=0)//ESD + { + + if(fMCstack->IsPhysicalPrimary(track->GetLabel())) + { + + fMCtrack = fMCstack->Particle(track->GetLabel()); + + Int_t pdg = fMCtrack->GetPdgCode(); + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax && fMCtrack->Phi()>=(TMath::Pi()*80/180) && fMCtrack->Phi()<=TMath::Pi()) + { + if(fMCtrack->GetFirstMother()>0){ + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + if( TMath::Abs(pdg) == 11 && fMCtrackMother->GetPdgCode()!=22 ){ + + fPtMC_EMCal_All->Fill(fMCtrack->Pt()); + } + } + } + } + } + + + if((fClus->E() / fP) >= fEoverPCutMin && (fClus->E() / fP) <= fEoverPCutMax) + { + + fECluster[2]->Fill(Energy); + //_______________________________________________________ + //Correlation Analysis + if(fUseEMCal) + { + fPtElec_Inc->Fill(fPt); + + if(fCorrelationFlag) + { + ElectronHadronCorrelation(track, iTracks, Vtrack); + } + } + //_______________________________________________________ + + //////////////////////////////////////////////////////////////////// + ///EMCal - efficiency calculations + + /// changing start here + if(fIsMC && fIsAOD && track->GetLabel()>=0) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + Int_t pdg = fMCparticle->GetPdgCode(); + + // + if(fMCparticle->IsPhysicalPrimary()){ + + + if(fMCparticle->Eta()>=fEtaCutMin && fMCparticle->Eta()<=fEtaCutMax ){ + //check + if(fIsHFE1) fPtMCelectronAfterAll->Fill(fMCparticle->Pt()); //numerator for the total efficiency + + + Bool_t MotherFound = FindMother(track->GetLabel()); + if(MotherFound){ + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + if( TMath::Abs(pdg) == 11 && fMCparticleMother->GetPdgCode()!=22 ){ + fPtMC_EMCal_Selected->Fill(fMCparticle->Pt()); + } + } + } + } + }///until here + + else if(fIsMC && track->GetLabel()>=0)//ESD + { + if(fMCstack->IsPhysicalPrimary(track->GetLabel())) + { + Bool_t MotherFound = FindMother(track->GetLabel()); + fMCtrack = fMCstack->Particle(track->GetLabel()); + if(MotherFound) + { + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax){ + if(fIsHFE1) fPtMCelectronAfterAll->Fill(fMCtrack->Pt()); //numerator for the total efficiency + } + } + + + + Int_t pdg = fMCtrack->GetPdgCode(); + if(fMCtrack->Eta()>=fEtaCutMin && fMCtrack->Eta()<=fEtaCutMax && fMCtrack->Phi()>=(TMath::Pi()*80/180) && fMCtrack->Phi()<=TMath::Pi()) + { + if(fMCtrack->GetFirstMother()>0){ + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + if( TMath::Abs(pdg) == 11 && fMCtrackMother->GetPdgCode()!=22 ){ + + fPtMC_EMCal_Selected->Fill(fMCtrack->Pt()); + } + } + } + } + } + /////////////////////////////////////////////////////////////////// + + + } + } + } + } + + //______________________________________________________________ + // Vertex + + fVtxZ[2]->Fill(fZvtx); + fNTracks[2]->Fill(fNOtrks); + fNClusters[2]->Fill(ClsNo); + + //______________________________________________________________ + + //_______________________________________________________ + //Correlation Analysis + if(!fUseEMCal) + { + fPtElec_Inc->Fill(fPt); + + if(fCorrelationFlag) + { + ElectronHadronCorrelation(track, iTracks, Vtrack); + } + } + //_______________________________________________________ + +///________________________________________________________________________ + } + + //__________________________________________________________________ + //Event Mixing Analysis + //Filling pool + if(fEventMixingFlag) + { + fPool = fPoolMgr->GetEventPool(fCentrality->GetCentralityPercentile("V0A"), fZvtx); // Get the buffer associated with the current centrality and z-vtx + + if(!fPool) AliFatal(Form("No pool found for centrality = %f, zVtx = %f", fCentrality->GetCentralityPercentile("V0A"), fZvtx)); + + fPool->UpdatePool(SelectedHadrons()); // fill the tracks in the event buffer. The ownership is handed over to the event mixing class. We are not allowed to delete tracksClone anymore! + + + } + + //__________________________________________________________________ + + delete fListOfmotherkink; + PostData(1, fOutputList); +} + +//______________________________________________________________________ +void AliAnalysisTaskEMCalHFEpA::Terminate(Option_t *) +{ +//Draw result to the screen +//Called once at the end of the query + + fOutputList = dynamic_cast (GetOutputData(1)); + + if(!fOutputList) + { + printf("ERROR: Output list not available\n"); + return; + } +} + +//______________________________________________________________________ +Bool_t AliAnalysisTaskEMCalHFEpA::ProcessCutStep(Int_t cutStep, AliVParticle *track) +{ +//Check single track cuts for a given cut step +//Note this function is called inside the UserExec function + const Int_t kMCOffset = AliHFEcuts::kNcutStepsMCTrack; + if(!fCFM->CheckParticleCuts(cutStep + kMCOffset, track)) return kFALSE; + return kTRUE; +} + +//______________________________________________________________________ +void AliAnalysisTaskEMCalHFEpA::ElectronHadronCorrelation(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack) +{ + ///_________________________________________________________________ + ///MC analysis + if(fIsMC) + { + if(track->GetLabel() < 0) + { + AliWarning(Form("The track %d does not have a valid MC label",trackIndex)); + return; + } + + if(fIsAOD) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(track->GetLabel()); + + if(fMCparticle->GetMother()<0) return; + + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + + if(TMath::Abs(fMCparticle->GetPdgCode())==11 && (TMath::Abs(fMCparticleMother->GetPdgCode())==22 || TMath::Abs(fMCparticleMother->GetPdgCode())==111 || TMath::Abs(fMCparticleMother->GetPdgCode())==221)) + { + //Is Background + fPtBackgroundBeforeReco->Fill(track->Pt()); + } + } + else + { + fMCtrack = fMCstack->Particle(track->GetLabel()); + + if(fMCtrack->GetFirstMother()<0) return; + + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + + if(TMath::Abs(fMCtrack->GetPdgCode())==11 && (TMath::Abs(fMCtrackMother->GetPdgCode())==22 || TMath::Abs(fMCtrackMother->GetPdgCode())==111 || TMath::Abs(fMCtrackMother->GetPdgCode())==221)) + { + //Is Background + fPtBackgroundBeforeReco->Fill(track->Pt()); + } + } + } + ///_________________________________________________________________ + + //________________________________________________ + //Associated particle cut + fPartnerCuts->SetAcceptKinkDaughters(kFALSE); + fPartnerCuts->SetRequireITSRefit(kTRUE); + fPartnerCuts->SetRequireTPCRefit(kTRUE); + fPartnerCuts->SetEtaRange(-0.9,0.9); + fPartnerCuts->SetMaxChi2PerClusterTPC(4.0); + fPartnerCuts->SetMinNClustersTPC(80); + fPartnerCuts->SetPtRange(0.3,1e10); + //fPartnerCuts->SetRequireSigmaToVertex(kTRUE); + //fPartnerCuts->SetMaxDCAToVertexXY(1); + //fPartnerCuts->SetMaxDCAToVertexZ(3); + //_________________________________________________ + + ///################################################################# + //Non-HFE reconstruction + fNonHFE = new AliSelectNonHFE(); + fNonHFE->SetAODanalysis(fIsAOD); + if(fMassCutFlag) fNonHFE->SetInvariantMassCut(fMassCut); + if(fAngleCutFlag) fNonHFE->SetOpeningAngleCut(fAngleCut); + if(fChi2CutFlag) fNonHFE->SetChi2OverNDFCut(fChi2Cut); + if(fDCAcutFlag) fNonHFE->SetDCACut(fDCAcut); + fNonHFE->SetAlgorithm("DCA"); //KF + fNonHFE->SetPIDresponse(fPidResponse); + fNonHFE->SetTrackCuts(-3.5,3.5,fPartnerCuts); + + fNonHFE->SetHistAngleBack(fOpAngleBack); + fNonHFE->SetHistAngle(fOpAngle); + fNonHFE->SetHistDCABack(fDCABack); + fNonHFE->SetHistDCA(fDCA); + fNonHFE->SetHistMassBack(fInvMassBack); + fNonHFE->SetHistMass(fInvMass); + + fNonHFE->FindNonHFE(trackIndex,vtrack,fVevent); + + Int_t *fUlsPartner = fNonHFE->GetPartnersULS(); + Int_t *fLsPartner = fNonHFE->GetPartnersLS(); + Bool_t fUlsIsPartner = kFALSE; + Bool_t fLsIsPartner = kFALSE; + ///################################################################# + + //Electron Information + Double_t fPhiE = -999; + Double_t fEtaE = -999; + Double_t fPhiH = -999; + Double_t fEtaH = -999; + Double_t fDphi = -999; + Double_t fDeta = -999; + Double_t fPtE = -999; + Double_t fPtH = -999; + + Double_t pi = TMath::Pi(); + + fPhiE = track->Phi(); + fEtaE = track->Eta(); + fPtE = track->Pt(); + + ///_________________________________________________________________ + ///MC analysis + if(fIsMC) + { + if(fIsAOD) + { + if(TMath::Abs(fMCparticle->GetPdgCode())==11 && (TMath::Abs(fMCparticleMother->GetPdgCode())==22 || TMath::Abs(fMCparticleMother->GetPdgCode())==111 || TMath::Abs(fMCparticleMother->GetPdgCode())==221)) + { + if(fNonHFE->IsULS()) fPtElec_ULS->Fill(fPtE,fNonHFE->GetNULS()); + if(fNonHFE->IsLS()) fPtElec_LS->Fill(fPtE,fNonHFE->GetNLS()); + } + } + else + { + if(TMath::Abs(fMCtrack->GetPdgCode())==11 && (TMath::Abs(fMCtrackMother->GetPdgCode())==22 || TMath::Abs(fMCtrackMother->GetPdgCode())==111 || TMath::Abs(fMCtrackMother->GetPdgCode())==221)) + { + if(fNonHFE->IsULS()) fPtElec_ULS->Fill(fPtE,fNonHFE->GetNULS()); + if(fNonHFE->IsLS()) fPtElec_LS->Fill(fPtE,fNonHFE->GetNLS()); + } + } + } + ///_________________________________________________________________ + else + { + if(fNonHFE->IsULS()) fPtElec_ULS->Fill(fPtE,fNonHFE->GetNULS()); + if(fNonHFE->IsLS()) fPtElec_LS->Fill(fPtE,fNonHFE->GetNLS()); + } + + //__________________________________________________________________ + //Event Mixing Analysis - Hadron Loop + //Retrieve + if(fEventMixingFlag) + { + fPool = fPoolMgr->GetEventPool(fCentrality->GetCentralityPercentile("V0A"), fZvtx); // Get the buffer associated with the current centrality and z-vtx + + if(!fPool) AliFatal(Form("No pool found for centrality = %f, zVtx = %f",fCentrality->GetCentralityPercentile("V0A"), fZvtx)); + + if(fPool->GetCurrentNEvents() >= 5) // start mixing when 5 events are in the buffer + { + fPoolNevents->Fill(fPool->GetCurrentNEvents()); + + for (Int_t jMix = 0; jMix < fPool->GetCurrentNEvents(); jMix++) // mix with each event in the buffer + { + TObjArray* bgTracks = fPool->GetEvent(jMix); + + for (Int_t kMix = 0; kMix < bgTracks->GetEntriesFast(); kMix++) // mix with each track in the event + { + const AliEHCParticle* MixedTrack(dynamic_cast(bgTracks->At(kMix))); + if (NULL == MixedTrack) continue; + + fPhiH = MixedTrack->Phi(); + fEtaH = MixedTrack->Eta(); + fPtH = MixedTrack->Pt(); + + if(fPtH>fPtE || fPtH<1) continue; + + fDphi = fPhiE - fPhiH; + + if (fDphi > 3*pi/2) fDphi = fDphi - 2*pi; + if (fDphi < -pi/2) fDphi = fDphi + 2*pi; + + fDeta = fEtaE - fEtaH; + + Double_t fPtBin[6] = {2,4,6,8,10,15}; + + for(Int_t i = 0; i < 5; i++) + { + if(fPtE>=fPtBin[i] && fPtEFill(fDphi,fDeta); + + if(fNonHFE->IsULS()) fCEtaPhi_ULS_EM[i]->Fill(fDphi,fDeta); + if(fNonHFE->IsLS()) fCEtaPhi_LS_EM[i]->Fill(fDphi,fDeta); + + if(fNonHFE->IsULS()) fCEtaPhi_ULS_Weight_EM[i]->Fill(fDphi,fDeta,fNonHFE->GetNULS()); + if(fNonHFE->IsLS()) fCEtaPhi_LS_Weight_EM[i]->Fill(fDphi,fDeta,fNonHFE->GetNLS()); + } + } + + // TODO your code: do event mixing with current event and bgTracks + // note that usually the content filled now is weighted by 1 / pool->GetCurrentNEvents() + } + } + } + } + //__________________________________________________________________ + + //__________________________________________________________________ + //Same Event Analysis - Hadron Loop + for(Int_t iTracks = 0; iTracks < fVevent->GetNumberOfTracks(); iTracks++) + { + if(trackIndex==iTracks) continue; + + AliVParticle* Vtrack2 = fVevent->GetTrack(iTracks); + if (!Vtrack2) + { + printf("ERROR: Could not receive track %d\n", iTracks); + continue; + } + + AliVTrack *track2 = dynamic_cast(Vtrack2); + + if(fIsAOD) + { + AliAODTrack *atrack2 = dynamic_cast(Vtrack2); + if(!atrack2->TestFilterMask(AliAODTrack::kTrkTPCOnly)) continue; + if((!(atrack2->GetStatus()&AliESDtrack::kITSrefit)|| (!(atrack2->GetStatus()&AliESDtrack::kTPCrefit)))) continue; + if(atrack2->GetTPCNcls() < 80) continue; + } + else + { + AliESDtrack *etrack2 = dynamic_cast(Vtrack2); + if(!fPartnerCuts->AcceptTrack(etrack2)) continue; + } + + fPhiH = track2->Phi(); + fEtaH = track2->Eta(); + fPtH = track2->Pt(); + + if(fPtH>fPtE || fPtH<1) continue; + + fDphi = fPhiE - fPhiH; + + if (fDphi > 3*pi/2) fDphi = fDphi - 2*pi; + if (fDphi < -pi/2) fDphi = fDphi + 2*pi; + + fDeta = fEtaE - fEtaH; + + Double_t fPtBin[6] = {2,4,6,8,10,15}; + + //______________________________________________________________ + //Check if this track is a Non-HFE partner + for(Int_t i = 0; i < fNonHFE->GetNULS(); i++) + { + if(fUlsPartner[i]==iTracks) fUlsIsPartner=kTRUE; + } + for(Int_t i = 0; i < fNonHFE->GetNLS(); i++) + { + if(fLsPartner[i]==iTracks) fLsIsPartner=kTRUE; + } + //______________________________________________________________ + + for(Int_t i = 0; i < 5; i++) + { + if(fPtE>=fPtBin[i] && fPtEFill(fDphi,fDeta); + + if(fNonHFE->IsULS()) fCEtaPhi_ULS[i]->Fill(fDphi,fDeta); + if(fNonHFE->IsLS()) fCEtaPhi_LS[i]->Fill(fDphi,fDeta); + if(fNonHFE->IsULS() && !fUlsIsPartner) fCEtaPhi_ULS_NoP[i]->Fill(fDphi,fDeta); + if(fNonHFE->IsLS() && !fLsIsPartner) fCEtaPhi_LS_NoP[i]->Fill(fDphi,fDeta); + + if(fNonHFE->IsULS()) fCEtaPhi_ULS_Weight[i]->Fill(fDphi,fDeta,fNonHFE->GetNULS()); + if(fNonHFE->IsLS()) fCEtaPhi_LS_Weight[i]->Fill(fDphi,fDeta,fNonHFE->GetNLS()); + if(fNonHFE->IsULS() && !fUlsIsPartner) fCEtaPhi_ULS_NoP_Weight[i]->Fill(fDphi,fDeta,fNonHFE->GetNULS()); + if(fNonHFE->IsLS() && !fLsIsPartner) fCEtaPhi_LS_NoP_Weight[i]->Fill(fDphi,fDeta,fNonHFE->GetNLS()); + } + } + } +} + +//____________________________________________________________________________________________________________ +//Create a TObjArray with selected hadrons, for the mixed event analysis +TObjArray* AliAnalysisTaskEMCalHFEpA::SelectedHadrons() +{ + fTracksClone = new TObjArray; + fTracksClone->SetOwner(kTRUE); + + for(Int_t iTracks = 0; iTracks < fVevent->GetNumberOfTracks(); iTracks++) + { + AliVParticle* Vtrack2 = fVevent->GetTrack(iTracks); + if (!Vtrack2) + { + printf("ERROR: Could not receive track %d\n", iTracks); + continue; + } + + AliVTrack *track2 = dynamic_cast(Vtrack2); + + if(fIsAOD) + { + AliAODTrack *atrack2 = dynamic_cast(Vtrack2); + if(!atrack2->TestFilterMask(AliAODTrack::kTrkTPCOnly)) continue; + if((!(atrack2->GetStatus()&AliESDtrack::kITSrefit)|| (!(atrack2->GetStatus()&AliESDtrack::kTPCrefit)))) continue; + if(atrack2->GetTPCNcls() < 80) continue; + } + else + { + AliESDtrack *etrack2 = dynamic_cast(Vtrack2); + if(!fPartnerCuts->AcceptTrack(etrack2)) continue; + } + + fTracksClone->Add(new AliEHCParticle(track2->Eta(), track2->Phi(), track2->Pt())); + } + return fTracksClone; +} +//____________________________________________________________________________________________________________ + +//______________________________________________________________________ +void AliAnalysisTaskEMCalHFEpA::DiHadronCorrelation(AliVTrack *track, Int_t trackIndex) +{ + //________________________________________________ + //Associated particle cut + fPartnerCuts->SetAcceptKinkDaughters(kFALSE); + fPartnerCuts->SetRequireITSRefit(kTRUE); + fPartnerCuts->SetRequireTPCRefit(kTRUE); + fPartnerCuts->SetEtaRange(-0.9,0.9); + fPartnerCuts->SetMaxChi2PerClusterTPC(4.0); + fPartnerCuts->SetMinNClustersTPC(80); + fPartnerCuts->SetPtRange(0.3,1e10); + //fPartnerCuts->SetRequireSigmaToVertex(kTRUE); + //fPartnerCuts->SetMaxDCAToVertexXY(1); + //fPartnerCuts->SetMaxDCAToVertexZ(3); + //_________________________________________________ + + //Electron Information + Double_t fPhiE = -999; + Double_t fEtaE = -999; + Double_t fPhiH = -999; + Double_t fEtaH = -999; + Double_t fDphi = -999; + Double_t fDeta = -999; + Double_t fPtE = -999; + Double_t fPtH = -999; + + Double_t pi = TMath::Pi(); + + fPhiE = track->Phi(); + fEtaE = track->Eta(); + fPtE = track->Pt(); + + //__________________________________________________________________ + //Same Event Analysis - Hadron Loop + for(Int_t iTracks = 0; iTracks < fVevent->GetNumberOfTracks(); iTracks++) + { + if(trackIndex==iTracks) continue; + + AliVParticle* Vtrack2 = fVevent->GetTrack(iTracks); + if (!Vtrack2) + { + printf("ERROR: Could not receive track %d\n", iTracks); + continue; + } + + AliVTrack *track2 = dynamic_cast(Vtrack2); + + if(fIsAOD) + { + AliAODTrack *atrack2 = dynamic_cast(Vtrack2); + if(!atrack2->TestFilterMask(AliAODTrack::kTrkTPCOnly)) continue; + if((!(atrack2->GetStatus()&AliESDtrack::kITSrefit)|| (!(atrack2->GetStatus()&AliESDtrack::kTPCrefit)))) continue; + if(atrack2->GetTPCNcls() < 80) continue; + } + else + { + AliESDtrack *etrack2 = dynamic_cast(Vtrack2); + if(!fPartnerCuts->AcceptTrack(etrack2)) continue; + } + + fPhiH = track2->Phi(); + fEtaH = track2->Eta(); + fPtH = track2->Pt(); + + if(fPtH>fPtE || fPtH<1) continue; + + fDphi = fPhiE - fPhiH; + + if (fDphi > 3*pi/2) fDphi = fDphi - 2*pi; + if (fDphi < -pi/2) fDphi = fDphi + 2*pi; + + fDeta = fEtaE - fEtaH; + + Double_t fPtBin[6] = {2,4,6,8,10,15}; + + for(Int_t i = 0; i < 5; i++) + { + if(fPtE>=fPtBin[i] && fPtEFill(fDphi,fDeta); + } + } + } +} +//____________________________________________________________________________________________________________ + +//______________________________________________________________________ +Bool_t AliAnalysisTaskEMCalHFEpA::FindMother(Int_t mcIndex) +{ + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + + if(mcIndex < 0 || !fIsMC) + { + return kFALSE; + } + + Int_t pdg = -99999; + Int_t mpdg = -99999; + Int_t gmpdg = -99999; + Int_t ggmpdg = -99999; + Int_t gggmpdg = -99999; + + if(fIsAOD) + { + fMCparticle = (AliAODMCParticle*) fMCarray->At(mcIndex); + + pdg = TMath::Abs(fMCparticle->GetPdgCode()); + + + if(pdg!=11) + { + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + return kFALSE; + } + + if(fMCparticle->GetMother()<0) + { + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + return kFALSE; + } + + fMCparticleMother = (AliAODMCParticle*) fMCarray->At(fMCparticle->GetMother()); + mpdg = TMath::Abs(fMCparticleMother->GetPdgCode()); + + if(fMCparticleMother->GetMother()<0) + { + gmpdg = 0; + ggmpdg = 0; + gggmpdg = 0; + } + else + { + fMCparticleGMother = (AliAODMCParticle*) fMCarray->At(fMCparticleMother->GetMother()); + gmpdg = TMath::Abs(fMCparticleGMother->GetPdgCode()); + if(fMCparticleGMother->GetMother()<0) + { + ggmpdg = 0; + gggmpdg = 0; + } + else + { + fMCparticleGGMother = (AliAODMCParticle*) fMCarray->At(fMCparticleGMother->GetMother()); + ggmpdg = TMath::Abs(fMCparticleGGMother->GetPdgCode()); + if(fMCparticleGGMother->GetMother()<0) + { + gggmpdg = 0; + } + else + { + fMCparticleGGGMother = (AliAODMCParticle*) fMCarray->At(fMCparticleGGMother->GetMother()); + gggmpdg = TMath::Abs(fMCparticleGGGMother->GetPdgCode()); + } + } + } + } + else + { + fMCtrack = fMCstack->Particle(mcIndex); + + pdg = TMath::Abs(fMCtrack->GetPdgCode()); + + if(pdg!=11) + { + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + return kFALSE; + } + + if(fMCtrack->GetFirstMother()<0) + { + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + return kFALSE; + } + + fMCtrackMother = fMCstack->Particle(fMCtrack->GetFirstMother()); + mpdg = TMath::Abs(fMCtrackMother->GetPdgCode()); + + if(fMCtrackMother->GetFirstMother()<0) + { + gmpdg = 0; + ggmpdg = 0; + gggmpdg = 0; + } + else + { + fMCtrackGMother = fMCstack->Particle(fMCtrackMother->GetFirstMother()); + gmpdg = TMath::Abs(fMCtrackGMother->GetPdgCode()); + + if(fMCtrackGMother->GetFirstMother()<0) + { + ggmpdg = 0; + gggmpdg = 0; + } + else + { + fMCtrackGGMother = fMCstack->Particle(fMCtrackGMother->GetFirstMother()); + ggmpdg = TMath::Abs(fMCtrackGGMother->GetPdgCode()); + + if(fMCtrackGGMother->GetFirstMother()<0) + { + gggmpdg = 0; + } + else + { + fMCtrackGGGMother = fMCstack->Particle(fMCtrackGGMother->GetFirstMother()); + gggmpdg = TMath::Abs(fMCtrackGGGMother->GetPdgCode()); + } + } + } + } + + //Tag Electron Source + if(mpdg==111 || mpdg==221 || mpdg==22) + { + fIsHFE1 = kFALSE; + fIsHFE2 = kFALSE; + fIsNonHFE = kTRUE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + + if(mpdg==111) fIsFromPi0 = kFALSE; + if(mpdg==221)fIsFromEta = kFALSE; + if(mpdg==22) fIsFromGamma = kFALSE; + + return kTRUE; + } + else + { + fIsHFE1 = kFALSE; + fIsHFE2 = kTRUE; + + fIsFromPi0 = kFALSE; + fIsFromEta = kFALSE; + fIsFromGamma = kFALSE; + + fIsNonHFE = kFALSE; + + fIsFromD = kFALSE; + fIsFromB = kFALSE; + + if(mpdg>400 && mpdg<500) + { + if((gmpdg>500 && gmpdg<600) || (ggmpdg>500 && ggmpdg<600) || (gggmpdg>500 && gggmpdg<600)) + { + fIsHFE1 = kTRUE; + fIsFromD = kFALSE; + fIsFromB = kTRUE; + return kTRUE; + } + else + { + fIsHFE1 = kTRUE; + fIsFromD = kTRUE; + fIsFromB = kFALSE; + return kTRUE; + } + } + else if(mpdg>500 && mpdg<600) + { + fIsHFE1 = kTRUE; + fIsFromD = kFALSE; + fIsFromB = kTRUE; + return kTRUE; + } + else + { + fIsHFE1 = kFALSE; + fIsFromD = kFALSE; + fIsFromB = kFALSE; + return kFALSE; + } + } +} diff --git a/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h b/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h new file mode 100644 index 00000000000..059e883a5b3 --- /dev/null +++ b/PWGHF/hfe/AliAnalysisTaskEMCalHFEpA.h @@ -0,0 +1,334 @@ +#ifndef AliAnalysisTaskEMCalHFEpA_cxx +#define AliAnalysisTaskEMCalHFEpA_cxx + +/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * + * See cxx source for full Copyright notice */ + +//////////////////////////////////////////////////////////////////////// +// // +// Task for Heavy-flavour electron analysis in pPb collisions // +// (+ Electron-Hadron Jetlike Azimuthal Correlation) // +// // +// v1.0 // +// // +// Authors // +// Elienos Pereira de Oliveira Filho (epereira@cern.ch) // +// Cristiane Jahnke (cristiane.jahnke@cern.ch) // +// // +//////////////////////////////////////////////////////////////////////// + +class TH1F; +class TH2F; +class AliESDEvent; +class AliESDtrackCuts; +class AliESDtrack; +class AliHFEcontainer; +class AliHFEcuts; +class AliHFEpid; +class AliHFEpidQAmanager; +class AliCFManager; +class AliPIDResponse; +class AliCentrality; +class AliAODEvent; +class AliVEvent; +class AliAODMCHeader; +class AliSelectNonHFE; +class AliEventPoolManager; +class AliEventPool; +class TObjArray; + +//______________________________________________________________________ +//Library +#include "AliAnalysisTaskSE.h" +#include "AliHFEpid.h" +#include "AliLog.h" +//______________________________________________________________________ + +//______________________________________________________________________ +class AliAnalysisTaskEMCalHFEpA : public AliAnalysisTaskSE +{ +//______________________________________________________________________ + public: + AliAnalysisTaskEMCalHFEpA(); + AliAnalysisTaskEMCalHFEpA(const char *name); + virtual ~AliAnalysisTaskEMCalHFEpA(); + + virtual void UserCreateOutputObjects(); + virtual void UserExec(Option_t *option); + virtual void Terminate(Option_t *); + + //Setters + void SetHFECuts(AliHFEcuts * const cuts) {fCuts = cuts;}; + void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) {fRejectKinkMother = rejectKinkMother;}; + void SetCorrelationAnalysis() {fCorrelationFlag = kTRUE;}; + void SetMCanalysis() {fIsMC = kTRUE;}; + void SetCentrality(Double_t CentralityMin, Double_t CentralityMax) { fCentralityMin = CentralityMin; fCentralityMax = CentralityMax; fHasCentralitySelection = kTRUE; }; + void SetAODanalysis(Bool_t IsAOD) {fIsAOD = IsAOD;}; + void SetEventMixing(Bool_t EventMixingFlag) { fEventMixingFlag = EventMixingFlag;}; + void SetNonHFEmassCut(Double_t MassCut) { fMassCut = MassCut; fMassCutFlag = kTRUE;}; + void SetEtaCut(Double_t EtaCutMin,Double_t EtaCutMax ) { fEtaCutMin = EtaCutMin; fEtaCutMax = EtaCutMax; }; + void SetEoverPCut(Double_t EoverPCutMin,Double_t EoverPCutMax ) { fEoverPCutMin = EoverPCutMin; fEoverPCutMax = EoverPCutMax; }; + void SetNonHFEangleCut(Double_t AngleCut) { fAngleCut = AngleCut; fAngleCutFlag = kTRUE;}; + void SetNonHFEchi2Cut(Double_t Chi2Cut) { fChi2Cut = Chi2Cut; fChi2CutFlag = kTRUE;}; + void SetNonHFEdcaCut(Double_t DCAcut) { fDCAcut = DCAcut; fDCAcutFlag = kTRUE;}; + void SetUseEMCal() { fUseEMCal=kTRUE;}; + void SetEMCalTriggerEG1() { fEMCEG1=kTRUE; }; + void SetEMCalTriggerEG2() { fEMCEG2=kTRUE; }; + void SetCentralityEstimator(Int_t Estimator) { fEstimator=Estimator; }; //0 = V0A, 1 = Other + + //Getters + AliHFEpid *GetPID() const {return fPID;}; +//______________________________________________________________________ + +//______________________________________________________________________ + private: + +//Function to process track cuts + Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track); +//Function to process eh analysis + void ElectronHadronCorrelation(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack); +//Selected Hadrons, for mixed event analysis + TObjArray* SelectedHadrons(); +//DiHadron Correlation Background + void DiHadronCorrelation(AliVTrack *track, Int_t trackIndex); +//Find Mothers (Finde HFE and NonHFE from MC information) + Bool_t FindMother(Int_t mcIndex); + +//Flags for specifics analysis + Bool_t fCorrelationFlag; + Bool_t fIsMC; + Bool_t fUseEMCal; + Bool_t fEMCEG1; + Bool_t fEMCEG2; + +//Used in the function FindMother + Bool_t fIsHFE1; + Bool_t fIsHFE2; + Bool_t fIsNonHFE; + Bool_t fIsFromD; + Bool_t fIsFromB; + Bool_t fIsFromPi0; + Bool_t fIsFromEta; + Bool_t fIsFromGamma; + +//General variables + AliESDEvent *fESD; + AliAODEvent *fAOD; /// new + AliVEvent *fVevent; /// new + AliESDtrackCuts *fPartnerCuts; + TList *fOutputList; + AliPIDResponse *fPidResponse; + AliSelectNonHFE *fNonHFE; + +//For the case of AOD analysis + Bool_t fIsAOD; //flag for AOD analysis + +//For Centrality Selection + AliCentrality *fCentrality; + Double_t fCentralityMin; + Double_t fCentralityMax; + Bool_t fHasCentralitySelection; + TH1F *fCentralityHist; + TH1F *fCentralityHistPass; + Float_t fZvtx; + Int_t fEstimator; + +//EMCal + //AliESDCaloCluster *fClus; + AliVCluster *fClus; + +//Histograms + TH1F *fNevent; + TH1F *fPtElec_Inc; + TH1F *fPtElec_ULS; + TH1F *fPtElec_LS; + + //PID Histograms + TH1F *fpid; + + TH2F **fEoverP_pt; + TH2F **fEoverP_tpc; + + TH1F **fTPC_pt; + TH2F **fTPC_p; + + TH1F **fTPCnsigma_pt; + TH2F **fTPCnsigma_p; + TH2F *fTPCnsigma_pt_2D; + + TH2F *fTPCnsigma_eta; + TH2F *fTPCnsigma_phi; + + + TH1F **fECluster; + TH2F **fEtaPhi; + TH1F **fVtxZ; + TH1F **fNTracks; + TH1F **fNClusters; + TH2F **fTPCNcls_EoverP; + + TH1F **fEta; + TH1F **fPhi; + TH1F **fR; + TH2F **fR_EoverP; + TH1F **fNcells; + TH2F **fNcells_EoverP; + TH1F **fNcells_electrons; + TH1F **fNcells_hadrons; + TH1F **fECluster_ptbins; + TH1F **fEoverP_ptbins; + TH1F **fEoverP_wSSCut; + TH2F **fM02_EoverP; + TH2F **fM20_EoverP; + TH2F **fTPCnsigma_eta_electrons; + TH2F **fTPCnsigma_eta_hadrons; + + TH2F *fEoverP_pt_pions; + + TH2F *ftpc_p_EoverPcut; + TH2F *fnsigma_p_EoverPcut; + + TH2F *fEoverP_pt_pions2; + TH2F *fNcells_pt; + TH2F *fEoverP_pt_hadrons; + + //Electron-Hadron Correlation Histograms + TH2F **fCEtaPhi_Inc; + + TH2F **fCEtaPhi_ULS; + TH2F **fCEtaPhi_LS; + TH2F **fCEtaPhi_ULS_NoP; + TH2F **fCEtaPhi_LS_NoP; + + TH2F **fCEtaPhi_ULS_Weight; + TH2F **fCEtaPhi_LS_Weight; + TH2F **fCEtaPhi_ULS_NoP_Weight; + TH2F **fCEtaPhi_LS_NoP_Weight; + + TH1F *fInvMass; + TH1F *fInvMassBack; + TH1F *fDCA; + TH1F *fDCABack; + TH1F *fOpAngle; + TH1F *fOpAngleBack; + + Double_t fMassCut; + Double_t fEtaCutMin; + Double_t fEtaCutMax; + Double_t fEoverPCutMin; + Double_t fEoverPCutMax; + Double_t fAngleCut; + Double_t fChi2Cut; + Double_t fDCAcut; + Bool_t fMassCutFlag; + Bool_t fAngleCutFlag; + Bool_t fChi2CutFlag; + Bool_t fDCAcutFlag; + + //Non-HFE reconstruction efficiency + TH1F *fPtBackgroundBeforeReco; + TH1F *fPtBackgroundAfterReco; + //Tracking Efficiency + TH1F *fPtMCparticleAll; + TH1F *fPtMCparticleReco; + TH1F *fPtMCparticleAllHfe1; + TH1F *fPtMCparticleRecoHfe1; + TH1F *fPtMCparticleAllHfe2; + TH1F *fPtMCparticleRecoHfe2; + TH1F *fPtMCelectronAfterAll; + + TH1F *fPtMCpi0; + + TH1F *fPtMC_EMCal_All; + TH1F *fPtMC_EMCal_Selected; + TH1F *fPtMC_TPC_All; + TH1F *fPtMC_TPC_Selected; + + TH1F *fPtMCWithLabel; + TH1F *fPtMCWithoutLabel; + TH1F *fPtIsPhysicaPrimary; + +//For the HFE package + AliHFEcuts *fCuts; // Cut Collection for HFE + AliCFManager *fCFM; // Correction Framework Manager + AliHFEpid *fPID; // PID + AliHFEpidQAmanager *fPIDqa; // PID QA manager + +//Others + AliStack *fMCstack; // + Bool_t fRejectKinkMother; // + TParticle *fMCtrack; + TParticle *fMCtrackMother; + TParticle *fMCtrackGMother; + TParticle *fMCtrackGGMother; + TParticle *fMCtrackGGGMother; + TClonesArray *fMCarray; + AliAODMCHeader *fMCheader; + AliAODMCParticle *fMCparticle; + AliAODMCParticle *fMCparticleMother; + AliAODMCParticle *fMCparticleGMother; + AliAODMCParticle *fMCparticleGGMother; + AliAODMCParticle *fMCparticleGGGMother; + AliMCEventHandler *fEventHandler; + AliMCEvent *fMCevent; + +//______________________________________________________________________ +//Mixed event analysis + AliEventPoolManager *fPoolMgr; + AliEventPool *fPool; + TObjArray *fTracksClone; + TObjArray *fTracks; + + TH2F **fCEtaPhi_Inc_EM; + + TH2F **fCEtaPhi_ULS_EM; + TH2F **fCEtaPhi_LS_EM; + + TH2F **fCEtaPhi_ULS_Weight_EM; + TH2F **fCEtaPhi_LS_Weight_EM; + + TH1F *fPoolNevents; + + Bool_t fEventMixingFlag; +//______________________________________________________________________ + +//______________________________________________________________________ +//Di-hadron correlation + TH2F **fCEtaPhi_Inc_DiHadron; + TH1F *fPtTrigger_Inc; +//______________________________________________________________________ + + AliAnalysisTaskEMCalHFEpA(const AliAnalysisTaskEMCalHFEpA&); // not implemented + AliAnalysisTaskEMCalHFEpA& operator=(const AliAnalysisTaskEMCalHFEpA&); // not implemented + + ClassDef(AliAnalysisTaskEMCalHFEpA, 1); // example of analysis +//______________________________________________________________________ +}; + +///_________________________________________________________________________________________________ +///Class copied from : $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskLongRangeCorrelations.h +///Author: Christoph Mayer +class AliEHCParticle : public TObject { +public: + AliEHCParticle(Double_t eta=0, Double_t phi=0, Double_t pt=0) + : fEta(eta), fPhi(phi), fPt(pt) {} + virtual ~AliEHCParticle() {} + + Double_t Eta() const { return fEta; } + Double_t Phi() const { return fPhi; } + Double_t Pt() const { return fPt; } + +protected: +private: + AliEHCParticle(const AliEHCParticle&); + AliEHCParticle& operator=(const AliEHCParticle&); + + Double_t fEta; + Double_t fPhi; + Double_t fPt; + + ClassDef(AliEHCParticle, 1); +} ; +///_________________________________________________________________________________________________ + +#endif diff --git a/PWGHF/hfe/macros/AddTaskEMCalHFEpA.C b/PWGHF/hfe/macros/AddTaskEMCalHFEpA.C new file mode 100644 index 00000000000..b43da7e2efc --- /dev/null +++ b/PWGHF/hfe/macros/AddTaskEMCalHFEpA.C @@ -0,0 +1,59 @@ +AliAnalysisTask AddTaskEMCalHFEpA( + + Bool_t isMC = kFALSE, + Int_t triggerIndex = 0, + Int_t configIndex = 0, + Int_t centralityIndex = 0, + Bool_t isAOD = kFALSE, + Bool_t isEMCal = kFALSE, + char * period = "b", + Int_t EMCalThreshould = 0 + ) +{ + AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); + + if (!mgr) { + ::Error("AddTaskEMCalHFEpA", "No analysis manager to connect to."); + return NULL; + } + + if (!mgr->GetInputEventHandler()) { + ::Error("AddTaskEMCalHFEpA", "This task requires an input event handler"); + return NULL; + } + + //_______________________ + //Config Task + //gROOT->LoadMacro("ConfigEMCalHFEpA.C"); + gROOT->LoadMacro("$ALICE_ROOT/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpA.C"); + //AliAnalysisTaskEMCalHFEpA *task = ConfigEMCalHFEpA(isMC,triggerIndex,configIndex,centralityIndex,isAOD,isEMCal, EMCalThreshould, period);//test (period) + AliAnalysisTaskEMCalHFEpA *task = ConfigEMCalHFEpA(isMC,triggerIndex,configIndex,centralityIndex,isAOD,isEMCal, EMCalThreshould); + + //_______________________ + //Trigger + if(!isMC) + { + if(triggerIndex == 0) task->SelectCollisionCandidates(AliVEvent::kINT7); + if(triggerIndex == 1) task->SelectCollisionCandidates(AliVEvent::kEMC7); + if(triggerIndex == 2) task->SelectCollisionCandidates(AliVEvent::kEMCEGA); + + //if(triggerIndex == 3) task->SelectCollisionCandidates(AliVEvent::kEMC8); + //if(triggerIndex == 4) task->SelectCollisionCandidates(AliVEvent::kEMCEJE); //Jet Trigger + } + + mgr->AddTask(task); + + //Create containers for input/output + + TString containerName = mgr->GetCommonFileName(); + containerName += ":HFE_EMCal_pPb_cris"; + + AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer(); + AliAnalysisDataContainer *coutput = mgr->CreateContainer(Form("chist_RpPb_%d_%d_%d_%d",triggerIndex,configIndex,centralityIndex, EMCalThreshould), TList::Class(), AliAnalysisManager::kOutputContainer, containerName.Data()); + + //Connect input/output + mgr->ConnectInput(task, 0, cinput); + mgr->ConnectOutput(task, 1, coutput); + + return task; +} diff --git a/PWGHF/hfe/macros/AddTaskEMCalHFEpACorrelation.C b/PWGHF/hfe/macros/AddTaskEMCalHFEpACorrelation.C new file mode 100644 index 00000000000..e9d223e171f --- /dev/null +++ b/PWGHF/hfe/macros/AddTaskEMCalHFEpACorrelation.C @@ -0,0 +1,62 @@ +AliAnalysisTask AddTaskEMCalHFEpACorrelation( + + Bool_t isMC = kFALSE, + Int_t triggerIndex = 0, + Int_t configIndex = 0, + Int_t centralityIndex = 0, + Bool_t isAOD = kFALSE, + Bool_t isEMCal = kFALSE, + char * period = "b", + Int_t EMCalThreshould = 0 + ) +{ + AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); + + if (!mgr) { + ::Error("AddTaskEMCalHFEpA", "No analysis manager to connect to."); + return NULL; + } + + if (!mgr->GetInputEventHandler()) { + ::Error("AddTaskEMCalHFEpA", "This task requires an input event handler"); + return NULL; + } + + //_______________________ + //Config Task + //gROOT->LoadMacro("ConfigEMCalHFEpACorrelation.C"); + gROOT->LoadMacro("$ALICE_ROOT/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpACorrelation.C"); + AliAnalysisTaskEMCalHFEpA *task = ConfigEMCalHFEpACorrelation(isMC,triggerIndex,configIndex,centralityIndex,isAOD,isEMCal,EMCalThreshould); + + //_______________________ + //Trigger + if(!isMC && (period=="d" || period=="e" || period=="f")) + { + if(triggerIndex == 0) task->SelectCollisionCandidates(AliVEvent::kINT7); + if(triggerIndex == 1) task->SelectCollisionCandidates(AliVEvent::kEMC7); + if(triggerIndex == 2) task->SelectCollisionCandidates(AliVEvent::kEMCEGA); + } + else if(!isMC) + { + if(triggerIndex == 0) task->SelectCollisionCandidates(AliVEvent::kINT7); + if(triggerIndex == 1) task->SelectCollisionCandidates(AliVEvent::kEMC7); + if(triggerIndex == 2) task->SelectCollisionCandidates(AliVEvent::kEMCEGA); + //if(triggerIndex == 3) task->SelectCollisionCandidates(AliVEvent::kEMC8); + //if(triggerIndex == 4) task->SelectCollisionCandidates(AliVEvent::kEMCEJE); //Jet Trigger + } + + mgr->AddTask(task); + + TString containerName = mgr->GetCommonFileName(); + containerName += ":HFE_EMCal_pPb_elienos"; + + //Create containers for input/output + AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer(); + AliAnalysisDataContainer *coutput = mgr->CreateContainer(Form("chist_eh_%d_%d_%d_%d",triggerIndex,configIndex,centralityIndex,EMCalThreshould), TList::Class(), AliAnalysisManager::kOutputContainer, containerName.Data()); + + //Connect input/output + mgr->ConnectInput(task, 0, cinput); + mgr->ConnectOutput(task, 1, coutput); + + return task; +} diff --git a/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpA.C b/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpA.C new file mode 100644 index 00000000000..82fc014024c --- /dev/null +++ b/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpA.C @@ -0,0 +1,134 @@ +///******************************************************* +///Config Description +//configIndex = 0 ---> Default cuts and PID +//configIndex = 1 ---> Non-HFE, Op Angle < 0.1 +//configIndex = 2 ---> +///******************************************************* + +AliAnalysisTaskEMCalHFEpA* ConfigEMCalHFEpA( + +Bool_t isMC=kFALSE, +Int_t triggerIndex=0, +Int_t configIndex=0, +Int_t centralityIndex=0, +Bool_t isAOD = kFALSE, +Bool_t isEMCal = kFALSE, +Int_t EMCalThreshould = 0 //0 == EG1, 1 == EG2 +) + +{ +///_______________________________________________________________________________________________________________ +///Track selection: Cuts used to ensure a minimum quality level of the tracks selected to perform the analysis + AliHFEcuts *hfecuts = new AliHFEcuts("hfeCutsMinBias","HFE Cuts"); + hfecuts->CreateStandardCuts(); + + //TPC Cuts + hfecuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable); + if(configIndex==2) hfecuts->SetMinNClustersTPC(90); //Minimum number of clusters on TPC + else if(configIndex==3) hfecuts->SetMinNClustersTPC(110); //Minimum number of clusters on TPC + else hfecuts->SetMinNClustersTPC(100); //Minimum number of clusters on TPC + + hfecuts->SetMinNClustersTPCPID(80); //Minimum number of clusters for dE/dx + hfecuts->SetMinRatioTPCclusters(0.6); //Number of clusters (Found/Findable) + + //ITS + //hfecuts->SetCutITSpixel(AliHFEextraCuts::kBoth); //Require at least one cluster on SPD + hfecuts->SetCutITSpixel(AliHFEextraCuts::kAny); //Require at least one cluster on SPD + //hfecuts->SetCutITSdrift(AliHFEextraCuts::kAny); //Require at least one cluster on SDD + hfecuts->SetCheckITSLayerStatus(kFALSE); + + if(configIndex==4) hfecuts->SetMinNClustersITS(2); //Minimum number of clusters on ITS + else if(configIndex==5) hfecuts->SetMinNClustersITS(4); //Minimum number of clusters on ITS + else hfecuts->SetMinNClustersITS(3); //Minimum number of clusters on ITS + + //Additional Cuts + hfecuts->SetPtRange(2, 1e6); //Transversal momentum range in GeV/c + //hfecuts->SetMaxImpactParam(1,2); //DCA to vertex + + //Event Selection + hfecuts->SetVertexRange(10.); // + //hfecuts->SetProductionVertex(0,0.3,0,0.3); // +///_______________________________________________________________________________________________________________ + +///_________________________________________________________________________________________________________________________ +///Task config + AliAnalysisTaskEMCalHFEpA *task = new AliAnalysisTaskEMCalHFEpA(Form("HFECuts%d_%d_%d",triggerIndex,configIndex,centralityIndex)); + printf("task ------------------------ %p\n ", task); + task->SetHFECuts(hfecuts); + task->SetCorrelationAnalysis(); + task->SetAODanalysis(isAOD); + task->SetEventMixing(kTRUE); + + //to separate trigger threshould + if(EMCalThreshould==0 && triggerIndex==2) task->SetEMCalTriggerEG1(); + if(EMCalThreshould==1 && triggerIndex==2) task->SetEMCalTriggerEG2(); + + if(isEMCal) task->SetUseEMCal(); + + if(configIndex==6) task->SetNonHFEmassCut(0.05); + else task->SetNonHFEmassCut(0.1); + + //if(isEMCal) task->SetEtaCut(-0.6,0.6); + //else task->SetEtaCut(-0.9,0.9); + + task->SetEtaCut(-0.6,0.6); + + if (configIndex==7) task->SetEoverPCut(0.85,1.2); + else if (configIndex==8) task->SetEoverPCut(0.75,1.25); + else task->SetEoverPCut(0.8,1.2); + + if(configIndex==1) task->SetNonHFEangleCut(0.1); + + if(centralityIndex==0) task->SetCentrality(0,20); + if(centralityIndex==1) task->SetCentrality(20,40); + if(centralityIndex==2) task->SetCentrality(40,60); + if(centralityIndex==3) task->SetCentrality(60,80); + if(centralityIndex==4) task->SetCentrality(80,100); + if(centralityIndex==5) task->SetCentrality(0,100); +///_______________________________________________________________________________________________________________ + +///_______________________________________________________________________________________________________________ +///Particle identification + AliHFEpid *pid = task->GetPID(); + +//______________________________________ +//In the case of a simulation + if(isMC) + { + pid->SetHasMCData(kTRUE); + task->SetMCanalysis(); + } +//______________________________________ + +//______________________________________________________ +//Configure PID + //_________________________ + //TPC PID + pid->AddDetector("TPC", 1); //Add TPC PID + + //_________________________ + //Configure TPC cut + //Defaul = -1 to 3 sigmas + //Note that it is also possible to define a model instead of a constant + //--------->For this change the "cut model" + + Double_t params[4]; + char *cutmodel; + cutmodel = "pol0"; + + if(configIndex==9) params[0] = -1.5; + else if (configIndex==10) params[0] = -0.5; + else if (configIndex==11) params[0] = 0; + else params[0] = -1; + + pid->ConfigureTPCdefaultCut(cutmodel,params,3.0); +//_______________________________________________________ +///_______________________________________________________________________________________________________________ + + printf("*************************************\n"); + printf("Configuring standard Task:\n"); + pid->PrintStatus(); + printf("*************************************\n"); + + return task; +} diff --git a/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpACorrelation.C b/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpACorrelation.C new file mode 100644 index 00000000000..a21f89d0a68 --- /dev/null +++ b/PWGHF/hfe/macros/configs/pPb/ConfigEMCalHFEpACorrelation.C @@ -0,0 +1,152 @@ +///******************************************************* +///Config Description +//configIndex = 0 ---> Default cuts and PID +//configIndex = 1 ---> TPC Ncls = 100 +//configIndex = 2 ---> TPC Ncls = 60 +//configIndex = 3 ---> SPD kBoth + 3 ITS cls +//configIndex = 4 ---> SPD kBoth + 4 ITS cls +//configIndex = 5 ---> SPD kAny + 3 ITS cls +//configIndex = 6 ---> Mass < 0.05 +//configIndex = 7 ---> Mass < 0.15 +//configIndex = 8 ---> Op Angle < 0.1 +//configIndex = 9 ---> TPC PID: -0.5 to 3.0 +//configIndex = 10 ---> V0A -> other +//configIndex = 11 ---> +///******************************************************* + +AliAnalysisTaskEMCalHFEpA* ConfigEMCalHFEpACorrelation( +Bool_t isMC=kFALSE, +Int_t triggerIndex=0, +Int_t configIndex=0, +Int_t centralityIndex=0, +Bool_t isAOD = kFALSE, +Bool_t isEMCal = kFALSE, +Int_t EMCalThreshould = 0 //0 == EG1, 1 == EG2 +) + +{ +///_______________________________________________________________________________________________________________ +///Track selection: Cuts used to ensure a minimum quality level of the tracks selected to perform the analysis + AliHFEcuts *hfecuts = new AliHFEcuts("hfeCutsMinBias","HFE Cuts"); + hfecuts->CreateStandardCuts(); + + //TPC Cuts + + hfecuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable); + if(configIndex==1) hfecuts->SetMinNClustersTPC(100); //Minimum number of clusters on TPC = 100 + else if(configIndex==2) hfecuts->SetMinNClustersTPC(60); //Minimum number of clusters on TPC = 60 + else hfecuts->SetMinNClustersTPC(80); //Minimum number of clusters on TPC = 80 + + hfecuts->SetMinNClustersTPCPID(80); //Minimum number of clusters for dE/dx + hfecuts->SetMinRatioTPCclusters(0.6); //Number of clusters (Found/Findable) + + //ITS + if(configIndex==3) + { + hfecuts->SetCutITSpixel(AliHFEextraCuts::kBoth); //Require 2 cluster on SPD + hfecuts->SetMinNClustersITS(3); //Minimum number of clusters on ITS + } + else if(configIndex==4) + { + hfecuts->SetCutITSpixel(AliHFEextraCuts::kBoth); //Require 2 cluster on SPD + hfecuts->SetMinNClustersITS(4); //Minimum number of clusters on ITS + } + else if(configIndex==5) + { + hfecuts->SetCutITSpixel(AliHFEextraCuts::kAny); //Require at least one cluster on SPD + hfecuts->SetMinNClustersITS(3); //Minimum number of clusters on ITS + } + else + { + hfecuts->SetCutITSpixel(AliHFEextraCuts::kAny); //Require at least one cluster on SPD + hfecuts->SetMinNClustersITS(2); //Minimum number of clusters on ITS + } + + hfecuts->SetCheckITSLayerStatus(kFALSE); + + //Additional Cuts + hfecuts->SetPtRange(2, 1e6); //Transversal momentum range in GeV/c + //hfecuts->SetMaxImpactParam(1,2); //DCA to vertex + + //Event Selection + hfecuts->SetVertexRange(10.); // + //hfecuts->SetProductionVertex(0,0.3,0,0.3); // +///_______________________________________________________________________________________________________________ + +///_________________________________________________________________________________________________________________________ +///Task config + AliAnalysisTaskEMCalHFEpA *task = new AliAnalysisTaskEMCalHFEpA(Form("HFECuts%d_%d_%d_%d",triggerIndex,configIndex,centralityIndex,EMCalThreshould)); + printf("task ------------------------ %p\n ", task); + + task->SetHFECuts(hfecuts); + task->SetCorrelationAnalysis(); + task->SetAODanalysis(isAOD); + task->SetEventMixing(kTRUE); + + if(configIndex==10) task->SetCentralityEstimator(1); + else task->SetCentralityEstimator(0); + + if(EMCalThreshould==0 && triggerIndex==2) task->SetEMCalTriggerEG1(); + if(EMCalThreshould==1 && triggerIndex==2) task->SetEMCalTriggerEG2(); + + if(isEMCal) task->SetUseEMCal(); + + if(configIndex==6) task->SetNonHFEmassCut(0.05); + else if(configIndex==7) task->SetNonHFEmassCut(0.15); + else task->SetNonHFEmassCut(0.1); + + if(isEMCal) task->SetEtaCut(-0.6,0.6); + else task->SetEtaCut(-0.9,0.9); + + task->SetEoverPCut(0.8,1.2); //Will work only in case isEMCal = kTRUE + + if(configIndex==8) task->SetNonHFEangleCut(0.1); + + if(centralityIndex==0) task->SetCentrality(0,20); + if(centralityIndex==1) task->SetCentrality(20,60); + if(centralityIndex==2) task->SetCentrality(60,100); +///_______________________________________________________________________________________________________________ + +///_______________________________________________________________________________________________________________ +///Particle identification + AliHFEpid *pid = task->GetPID(); + +//______________________________________ +//In the case of a simulation + if(isMC) + { + pid->SetHasMCData(kTRUE); + task->SetMCanalysis(); + } +//______________________________________ + +//______________________________________________________ +//Configure PID + //_________________________ + //TPC PID + pid->AddDetector("TPC", 1); //Add TPC PID + + //_________________________ + //Configure TPC cut + //Defaul = -1 to 3 sigmas + //Note that it is also possible to define a model instead of a constant + //--------->For this change the "cut model" + + Double_t params[4]; + char *cutmodel; + cutmodel = "pol0"; + + if(configIndex==9) params[0] = -0.5; + else params[0] = -1; + + pid->ConfigureTPCdefaultCut(cutmodel,params,3.0); +//_______________________________________________________ +///_______________________________________________________________________________________________________________ + + printf("*************************************\n"); + printf("Configuring standard Task:\n"); + pid->PrintStatus(); + printf("*************************************\n"); + + return task; +} -- 2.43.5