/************************************************************************** * Copyright(c) 1998-2008, 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. * **************************************************************************/ /* $Id$ */ //************************************************************************* // Class AliAnalysisTaskSEDvsMultiplicity // AliAnalysisTaskSE for the D meson vs. multiplcity analysis // Authors: Renu Bala, Zaida Conesa del Valle, Francesco Prino ///////////////////////////////////////////////////////////// #include #include #include #include #include #include #include #include #include #include #include "AliAnalysisManager.h" #include "AliRDHFCuts.h" #include "AliRDHFCutsDplustoKpipi.h" #include "AliRDHFCutsDStartoKpipi.h" #include "AliRDHFCutsD0toKpi.h" #include "AliAODHandler.h" #include "AliAODEvent.h" #include "AliAODVertex.h" #include "AliAODTrack.h" #include "AliAODRecoDecayHF.h" #include "AliAODRecoCascadeHF.h" #include "AliAnalysisVertexingHF.h" #include "AliAnalysisTaskSE.h" #include "AliAnalysisTaskSEDvsMultiplicity.h" #include "AliNormalizationCounter.h" #include "AliVertexingHFUtils.h" #include "AliAODVZERO.h" ClassImp(AliAnalysisTaskSEDvsMultiplicity) //________________________________________________________________________ AliAnalysisTaskSEDvsMultiplicity::AliAnalysisTaskSEDvsMultiplicity(): AliAnalysisTaskSE(), fOutput(0), fListCuts(0), fOutputCounters(0), fListProfiles(0), fHistNEvents(0), fHistNtrEta16vsNtrEta1EvSel(0), fHistNtrEta05vsNtrEta1EvSel(0), fHistNtrEta03vsNtrEta1EvSel(0), fHistNtrEtaV0AvsNtrEta1EvSel(0), fHistNtrEtaV0MvsNtrEta1EvSel(0), fHistNtrCorrEta1vsNtrRawEta1EvSel(0), fHistNtrEta16vsNtrEta1EvWithCand(0), fHistNtrEta05vsNtrEta1EvWithCand(0), fHistNtrEta03vsNtrEta1EvWithCand(0), fHistNtrEtaV0AvsNtrEta1EvWithCand(0), fHistNtrEtaV0MvsNtrEta1EvWithCand(0), fHistNtrCorrEta1vsNtrRawEta1EvWithCand(0), fHistNtrEta16vsNtrEta1EvWithD(0), fHistNtrEta05vsNtrEta1EvWithD(0), fHistNtrEta03vsNtrEta1EvWithD(0), fHistNtrEtaV0AvsNtrEta1EvWithD(0), fHistNtrEtaV0MvsNtrEta1EvWithD(0), fHistNtrCorrEta1vsNtrRawEta1EvWithD(0), fHistNtrVsZvtx(0), fHistNtrCorrVsZvtx(0), fHistNtrVsNchMC(0), fHistNtrCorrVsNchMC(0), fHistNtrVsNchMCPrimary(0), fHistNtrCorrVsNchMCPrimary(0), fHistNtrVsNchMCPhysicalPrimary(0), fHistNtrCorrVsNchMCPhysicalPrimary(0), fHistGenPrimaryParticlesInelGt0(0), fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary(0), fHistNtrUnCorrPSSel(0), fHistNtrUnCorrEvSel(0), fHistNtrUnCorrEvWithCand(0), fHistNtrUnCorrEvWithD(0), fHistNtrCorrPSSel(0), fHistNtrCorrEvSel(0), fHistNtrCorrEvWithCand(0), fHistNtrCorrEvWithD(0), fPtVsMassVsMult(0), fPtVsMassVsMultNoPid(0), fPtVsMassVsMultUncorr(0), fPtVsMassVsMultPart(0), fPtVsMassVsMultAntiPart(0), fPtVsMassVsMultMC(0), fUpmasslimit(1.965), fLowmasslimit(1.765), fNMassBins(200), fRDCutsAnalysis(0), fCounter(0), fCounterU(0), fDoImpPar(kFALSE), fNImpParBins(400), fLowerImpPar(-2000.), fHigherImpPar(2000.), fReadMC(kFALSE), fMCOption(0), fisPPbData(kFALSE), fUseBit(kTRUE), fSubtractTrackletsFromDau(kFALSE), fKeepCorrPlots(kFALSE), fUseNchWeight(kFALSE), fHistoMCNch(0), fHistoMeasNch(0), fRefMult(9.26), fPdgMeson(411), fMultiplicityEstimator(kNtrk10), fMCPrimariesEstimator(kEta10) { // Default constructor for(Int_t i=0; i<5; i++) fHistMassPtImpPar[i]=0; for(Int_t i=0; i<4; i++) fMultEstimatorAvg[i]=0; } //________________________________________________________________________ AliAnalysisTaskSEDvsMultiplicity::AliAnalysisTaskSEDvsMultiplicity(const char *name, Int_t pdgMeson,AliRDHFCuts *cuts, Bool_t switchPPb): AliAnalysisTaskSE(name), fOutput(0), fListCuts(0), fOutputCounters(0), fListProfiles(0), fHistNEvents(0), fHistNtrEta16vsNtrEta1EvSel(0), fHistNtrEta05vsNtrEta1EvSel(0), fHistNtrEta03vsNtrEta1EvSel(0), fHistNtrEtaV0AvsNtrEta1EvSel(0), fHistNtrEtaV0MvsNtrEta1EvSel(0), fHistNtrCorrEta1vsNtrRawEta1EvSel(0), fHistNtrEta16vsNtrEta1EvWithCand(0), fHistNtrEta05vsNtrEta1EvWithCand(0), fHistNtrEta03vsNtrEta1EvWithCand(0), fHistNtrEtaV0AvsNtrEta1EvWithCand(0), fHistNtrEtaV0MvsNtrEta1EvWithCand(0), fHistNtrCorrEta1vsNtrRawEta1EvWithCand(0), fHistNtrEta16vsNtrEta1EvWithD(0), fHistNtrEta05vsNtrEta1EvWithD(0), fHistNtrEta03vsNtrEta1EvWithD(0), fHistNtrEtaV0AvsNtrEta1EvWithD(0), fHistNtrEtaV0MvsNtrEta1EvWithD(0), fHistNtrCorrEta1vsNtrRawEta1EvWithD(0), fHistNtrVsZvtx(0), fHistNtrCorrVsZvtx(0), fHistNtrVsNchMC(0), fHistNtrCorrVsNchMC(0), fHistNtrVsNchMCPrimary(0), fHistNtrCorrVsNchMCPrimary(0), fHistNtrVsNchMCPhysicalPrimary(0), fHistNtrCorrVsNchMCPhysicalPrimary(0), fHistGenPrimaryParticlesInelGt0(0), fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary(0), fHistNtrUnCorrPSSel(0), fHistNtrUnCorrEvSel(0), fHistNtrUnCorrEvWithCand(0), fHistNtrUnCorrEvWithD(0), fHistNtrCorrPSSel(0), fHistNtrCorrEvSel(0), fHistNtrCorrEvWithCand(0), fHistNtrCorrEvWithD(0), fPtVsMassVsMult(0), fPtVsMassVsMultNoPid(0), fPtVsMassVsMultUncorr(0), fPtVsMassVsMultPart(0), fPtVsMassVsMultAntiPart(0), fPtVsMassVsMultMC(0), fUpmasslimit(1.965), fLowmasslimit(1.765), fNMassBins(200), fRDCutsAnalysis(cuts), fCounter(0), fCounterU(0), fDoImpPar(kFALSE), fNImpParBins(400), fLowerImpPar(-2000.), fHigherImpPar(2000.), fReadMC(kFALSE), fMCOption(0), fisPPbData(switchPPb), fUseBit(kTRUE), fSubtractTrackletsFromDau(kFALSE), fKeepCorrPlots(kFALSE), fUseNchWeight(kFALSE), fHistoMCNch(0), fHistoMeasNch(0), fRefMult(9.26), fPdgMeson(pdgMeson), fMultiplicityEstimator(kNtrk10), fMCPrimariesEstimator(kEta10) { // // Standard constructor // for(Int_t i=0; i<5; i++) fHistMassPtImpPar[i]=0; for(Int_t i=0; i<4; i++) fMultEstimatorAvg[i]=0; if(fPdgMeson==413){ fNMassBins=200; SetMassLimits(0.12,0.2); }else{ fNMassBins=200; SetMassLimits(fPdgMeson,0.1); } // Default constructor // Otput slot #1 writes into a TList container DefineOutput(1,TList::Class()); //My private output // Output slot #2 writes cut to private output DefineOutput(2,TList::Class()); // Output slot #3 writes cut to private output DefineOutput(3,TList::Class()); // Output slot #4 writes cut to private output DefineOutput(4,TList::Class()); } //________________________________________________________________________ AliAnalysisTaskSEDvsMultiplicity::~AliAnalysisTaskSEDvsMultiplicity() { // // Destructor // delete fOutput; delete fHistNEvents; delete fListCuts; delete fListProfiles; delete fRDCutsAnalysis; delete fCounter; delete fCounterU; for(Int_t i=0; i<4; i++) { if (fMultEstimatorAvg[i]) delete fMultEstimatorAvg[i]; } for(Int_t i=0; i<5; i++){ delete fHistMassPtImpPar[i]; } if(fHistoMCNch) delete fHistoMCNch; if(fHistoMeasNch) delete fHistoMeasNch; } //_________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::SetMassLimits(Double_t lowlimit, Double_t uplimit){ // set invariant mass limits if(uplimit>lowlimit){ fLowmasslimit = lowlimit; fUpmasslimit = uplimit; }else{ AliError("Wrong mass limits: upper value should be larger than lower one"); } } //_________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::SetMassLimits(Int_t pdg, Double_t range){ // set invariant mass limits Double_t mass=TDatabasePDG::Instance()->GetParticle(TMath::Abs(pdg))->Mass(); SetMassLimits(mass-range,mass+range); } //________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::Init(){ // // Initialization // printf("AnalysisTaskSEDvsMultiplicity::Init() \n"); if(fUseNchWeight && !fReadMC){ AliFatal("Nch weights can only be used in MC mode"); return; } if(fUseNchWeight && !fHistoMCNch){ AliFatal("Nch weights can only be used without histogram"); return; } fListCuts=new TList(); fListCuts->SetOwner(); fListCuts->SetName("CutsList"); if(fPdgMeson==411){ AliRDHFCutsDplustoKpipi* copycut=new AliRDHFCutsDplustoKpipi(*(static_cast(fRDCutsAnalysis))); copycut->SetName("AnalysisCutsDplus"); fListCuts->Add(copycut); }else if(fPdgMeson==421){ AliRDHFCutsD0toKpi* copycut=new AliRDHFCutsD0toKpi(*(static_cast(fRDCutsAnalysis))); copycut->SetName("AnalysisCutsDzero"); fListCuts->Add(copycut); }else if(fPdgMeson==413){ AliRDHFCutsDStartoKpipi* copycut=new AliRDHFCutsDStartoKpipi(*(static_cast(fRDCutsAnalysis))); copycut->SetName("AnalysisCutsDStar"); fListCuts->Add(copycut); } PostData(2,fListCuts); fListProfiles = new TList(); fListProfiles->SetOwner(); TString period[4]; Int_t nProfiles=4; if (fisPPbData) {period[0]="LHC13b"; period[1]="LHC13c"; nProfiles = 2;} else {period[0]="LHC10b"; period[1]="LHC10c"; period[2]="LHC10d"; period[3]="LHC10e"; nProfiles = 4;} for(Int_t i=0; iSetName(Form("ProfileTrkVsZvtx%s\n",period[i].Data())); fListProfiles->Add(hprof); } } PostData(4,fListProfiles); return; } //________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::UserCreateOutputObjects() { // Create the output container // if(fDebug > 1) printf("AnalysisTaskSEDvsMultiplicity::UserCreateOutputObjects() \n"); // Several histograms are more conveniently managed in a TList fOutput = new TList(); fOutput->SetOwner(); fOutput->SetName("OutputHistos"); Int_t nMultBins = 200; Float_t firstMultBin = -0.5; Float_t lastMultBin = 199.5; Int_t nMultBinsNtrk = nMultBins; Float_t lastMultBinNtrk = lastMultBin; Int_t nMultBinsV0 = 400; Float_t lastMultBinV0 = 799.5; if(fisPPbData) { nMultBinsNtrk = 375; lastMultBinNtrk = 374.5; nMultBins = nMultBinsNtrk; lastMultBin = lastMultBinNtrk; } if(fMultiplicityEstimator==kVZERO || fMultiplicityEstimator==kVZEROA) { nMultBins = nMultBinsV0; lastMultBin = lastMultBinV0; } fHistNtrUnCorrPSSel = new TH1F("hNtrUnCorrPSSel","Uncorrected tracklets multiplicity for PS selected events; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); fHistNtrUnCorrEvSel = new TH1F("hNtrUnCorrEvSel","Uncorrected tracklets multiplicity for selected events; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); fHistNtrUnCorrEvWithCand = new TH1F("hNtrUnCorrEvWithCand", "Uncorrected Tracklets multiplicity for events with D candidates; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin);// Total multiplicity fHistNtrUnCorrEvWithD = new TH1F("hNtrUnCorrEvWithD","Uncorrected Tracklets multiplicity for events with D in mass region ; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); // fHistNtrCorrPSSel = new TH1F("hNtrCorrPSSel","Corrected tracklets multiplicity for PS selected events; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); fHistNtrCorrEvSel = new TH1F("hNtrCorrEvSel","Corrected tracklets multiplicity for selected events; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); fHistNtrCorrEvWithCand = new TH1F("hNtrCorrEvWithCand", "Tracklets multiplicity for events with D candidates; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin);// Total multiplicity fHistNtrCorrEvWithD = new TH1F("hNtrCorrEvWithD", "Tracklets multiplicity for events with D in mass region ; Tracklets ; Entries",nMultBins,firstMultBin,lastMultBin); // if(fKeepCorrPlots){ fHistNtrEta16vsNtrEta1EvSel = new TH2F("hNtrEta16vsNtrEta1EvSel","Uncorrected Eta1.6 vs Eta1.0 (events selected); Ntracklets #eta<1.0; Ntracklets #eta<1.6",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram fHistNtrEta05vsNtrEta1EvSel = new TH2F("hNtrEta05vsNtrEta1EvSel","Uncorrected Eta0.5 vs Eta1.0 (events selected); Ntracklets #eta<1.0; Ntracklets #eta<0.5",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.5 vs eta 1.0 histogram fHistNtrEta03vsNtrEta1EvSel = new TH2F("hNtrEta03vsNtrEta1EvSel","Uncorrected Eta0.3 vs Eta1.0 (events selected); Ntracklets #eta<1.0; Ntracklets #eta<0.3",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.3 vs eta 1.0 histogram fHistNtrEtaV0AvsNtrEta1EvSel = new TH2F("hNtrEtaV0AvsNtrEta1EvSel","Uncorrected Eta-V0A vs Eta1.0 (events selected); Ntracklets #eta<1.0; Multiplicity V0A",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0A vs eta 1.0 histogram fHistNtrEtaV0MvsNtrEta1EvSel = new TH2F("hNtrEtaV0MvsNtrEta1EvSel","Uncorrected Eta-V0M vs Eta1.0 (events selected); Ntracklets #eta<1.0; Multiplicity V0A+V0C",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0M vs eta 1.0 histogram fHistNtrCorrEta1vsNtrRawEta1EvSel = new TH2F("hNtrCorrEta1vsNtrRawEta1EvSel","Corrected Eta1 vs Eta1.0 (events selected); Ntracklets #eta<1.0 corrected; Ntracklets #eta<1",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram fHistNtrEta16vsNtrEta1EvWithCand = new TH2F("hNtrEta16vsNtrEta1EvWithCand","Uncorrected Eta1.6 vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0; Ntracklets #eta<1.6",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram fHistNtrEta05vsNtrEta1EvWithCand = new TH2F("hNtrEta05vsNtrEta1EvWithCand","Uncorrected Eta0.5 vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0; Ntracklets #eta<0.5",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.5 vs eta 1.0 histogram fHistNtrEta03vsNtrEta1EvWithCand = new TH2F("hNtrEta03vsNtrEta1EvWithCand","Uncorrected Eta0.3 vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0; Ntracklets #eta<0.3",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.3 vs eta 1.0 histogram fHistNtrEtaV0AvsNtrEta1EvWithCand = new TH2F("hNtrEtaV0AvsNtrEta1EvWithCand","Uncorrected Eta-V0A vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0; Multiplicity V0A",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0A vs eta 1.0 histogram fHistNtrEtaV0MvsNtrEta1EvWithCand = new TH2F("hNtrEtaV0MvsNtrEta1EvWithCand","Uncorrected Eta-V0M vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0; Multiplicity V0A+V0C",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0M vs eta 1.0 histogram fHistNtrCorrEta1vsNtrRawEta1EvWithCand = new TH2F("hNtrCorrEta1vsNtrRawEta1EvWithCand","Corrected Eta1 vs Eta1.0 (events selected with a D candidate); Ntracklets #eta<1.0 corrected; Ntracklets #eta<1",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram fHistNtrEta16vsNtrEta1EvWithD = new TH2F("hNtrEta16vsNtrEta1EvWithD","Uncorrected Eta1.6 vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0; Ntracklets #eta<1.6",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram fHistNtrEta05vsNtrEta1EvWithD = new TH2F("hNtrEta05vsNtrEta1EvWithD","Uncorrected Eta0.5 vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0; Ntracklets #eta<0.5",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.5 vs eta 1.0 histogram fHistNtrEta03vsNtrEta1EvWithD = new TH2F("hNtrEta03vsNtrEta1EvWithD","Uncorrected Eta0.3 vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0; Ntracklets #eta<0.3",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 0.3 vs eta 1.0 histogram fHistNtrEtaV0AvsNtrEta1EvWithD = new TH2F("hNtrEtaV0AvsNtrEta1EvWithD","Uncorrected Eta-V0A vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0; Multiplicity V0A",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0A vs eta 1.0 histogram fHistNtrEtaV0MvsNtrEta1EvWithD = new TH2F("hNtrEtaV0MvsNtrEta1EvWithD","Uncorrected Eta-V0M vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0; Multiplicity V0A+V0C",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsV0,firstMultBin,lastMultBinV0); //eta V0M vs eta 1.0 histogram fHistNtrCorrEta1vsNtrRawEta1EvWithD = new TH2F("hNtrCorrEta1vsNtrRawEta1EvWithD","Corrected Eta1 vs Eta1.0 (events selected with D in mass range); Ntracklets #eta<1.0 corrected; Ntracklets #eta<1",nMultBinsNtrk,firstMultBin,lastMultBinNtrk,nMultBinsNtrk,firstMultBin,lastMultBinNtrk); //eta 1.6 vs eta 1.0 histogram } fHistNtrVsZvtx = new TH2F("hNtrVsZvtx","Ntracklet vs VtxZ; VtxZ;N_{tracklet};",300,-15,15,nMultBins,firstMultBin,lastMultBin); // fHistNtrCorrVsZvtx = new TH2F("hNtrCorrVsZvtx","Ntracklet vs VtxZ; VtxZ;N_{tracklet};",300,-15,15,nMultBins,firstMultBin,lastMultBin); // fHistNtrVsNchMC = new TH2F("hNtrVsNchMC","Ntracklet vs NchMC; Nch;N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistNtrCorrVsNchMC = new TH2F("hNtrCorrVsNchMC","Ntracklet vs Nch; Nch;N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistNtrVsNchMCPrimary = new TH2F("hNtrVsNchMCPrimary","Ntracklet vs Nch (Primary); Nch (Primary);N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistNtrCorrVsNchMCPrimary = new TH2F("hNtrCorrVsNchMCPrimary","Ntracklet vs Nch (Primary); Nch(Primary) ;N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistNtrVsNchMCPhysicalPrimary = new TH2F("hNtrVsNchMCPhysicalPrimary","Ntracklet vs Nch (Physical Primary); Nch (Physical Primary);N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistNtrCorrVsNchMCPhysicalPrimary = new TH2F("hNtrCorrVsMCPhysicalPrimary","Ntracklet vs Nch (Physical Primary); Nch (Physical Primary);N_{tracklet};",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); // fHistGenPrimaryParticlesInelGt0 = new TH1F("hGenPrimaryParticlesInelGt0","Multiplcity of generated charged particles ; Nparticles ; Entries",nMultBins,firstMultBin,lastMultBin); fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary = new TH3F("fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary", "MC: Nch (Physical Primary) vs Nch (Primary) vs Nch (Generated); Nch (Generated); Nch (Primary); Nch (Physical Primary)",nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin,nMultBins,firstMultBin,lastMultBin); fHistNtrUnCorrPSSel->Sumw2(); fHistNtrUnCorrEvSel->Sumw2(); fHistNtrUnCorrEvWithCand->Sumw2(); fHistNtrUnCorrEvWithD->Sumw2(); fHistNtrCorrPSSel->Sumw2(); fHistNtrCorrEvSel->Sumw2(); fHistNtrCorrEvWithCand->Sumw2(); fHistNtrCorrEvWithD->Sumw2(); fHistGenPrimaryParticlesInelGt0->Sumw2(); fOutput->Add(fHistNtrUnCorrPSSel); fOutput->Add(fHistNtrUnCorrEvSel); fOutput->Add(fHistNtrUnCorrEvWithCand); fOutput->Add(fHistNtrUnCorrEvWithD); fOutput->Add(fHistNtrCorrPSSel); fOutput->Add(fHistNtrCorrEvSel); fOutput->Add(fHistNtrCorrEvWithCand); fOutput->Add(fHistNtrCorrEvWithD); if(fKeepCorrPlots){ fOutput->Add(fHistNtrEta16vsNtrEta1EvSel); fOutput->Add(fHistNtrEta05vsNtrEta1EvSel); fOutput->Add(fHistNtrEta03vsNtrEta1EvSel); fOutput->Add(fHistNtrEtaV0AvsNtrEta1EvSel); fOutput->Add(fHistNtrEtaV0MvsNtrEta1EvSel); fOutput->Add(fHistNtrCorrEta1vsNtrRawEta1EvSel); fOutput->Add(fHistNtrEta16vsNtrEta1EvWithCand); fOutput->Add(fHistNtrEta05vsNtrEta1EvWithCand); fOutput->Add(fHistNtrEta03vsNtrEta1EvWithCand); fOutput->Add(fHistNtrEtaV0AvsNtrEta1EvWithCand); fOutput->Add(fHistNtrEtaV0MvsNtrEta1EvWithCand); fOutput->Add(fHistNtrCorrEta1vsNtrRawEta1EvWithCand); fOutput->Add(fHistNtrEta16vsNtrEta1EvWithD); fOutput->Add(fHistNtrEta05vsNtrEta1EvWithD); fOutput->Add(fHistNtrEta03vsNtrEta1EvWithD); fOutput->Add(fHistNtrEtaV0AvsNtrEta1EvWithD); fOutput->Add(fHistNtrEtaV0MvsNtrEta1EvWithD); fOutput->Add(fHistNtrCorrEta1vsNtrRawEta1EvWithD); } fOutput->Add(fHistNtrVsZvtx); fOutput->Add(fHistNtrCorrVsZvtx); fOutput->Add(fHistNtrVsNchMC); fOutput->Add(fHistNtrCorrVsNchMC); fOutput->Add(fHistNtrVsNchMCPrimary); fOutput->Add(fHistNtrCorrVsNchMCPrimary); fOutput->Add(fHistNtrVsNchMCPhysicalPrimary); fOutput->Add(fHistNtrCorrVsNchMCPhysicalPrimary); fOutput->Add(fHistGenPrimaryParticlesInelGt0); fOutput->Add(fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary); fHistNEvents = new TH1F("fHistNEvents", "number of events ",11,-0.5,10.5); fHistNEvents->GetXaxis()->SetBinLabel(1,"nEvents total"); fHistNEvents->GetXaxis()->SetBinLabel(2,"nEvents with Z vertex"); fHistNEvents->GetXaxis()->SetBinLabel(3,"nEvents selected"); fHistNEvents->GetXaxis()->SetBinLabel(4,"Rejected due to trigger"); fHistNEvents->GetXaxis()->SetBinLabel(5,"Rejected due to phys sel"); fHistNEvents->GetXaxis()->SetBinLabel(6,"Rejected due to vertex cuts"); fHistNEvents->GetXaxis()->SetBinLabel(7,"Rejected due to pileup"); fHistNEvents->GetXaxis()->SetBinLabel(8,"Total no. of candidate"); fHistNEvents->GetXaxis()->SetBinLabel(9,"no. of cand wo bitmask"); fHistNEvents->GetXaxis()->SetBinLabel(10,"D after cuts (No PID)"); fHistNEvents->GetXaxis()->SetBinLabel(11,"D after cuts + PID)"); fHistNEvents->GetXaxis()->SetNdivisions(1,kFALSE); fHistNEvents->Sumw2(); fHistNEvents->SetMinimum(0); fOutput->Add(fHistNEvents); fPtVsMassVsMult=new TH3F("hPtVsMassvsMult", "D candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fPtVsMassVsMultNoPid=new TH3F("hPtVsMassvsMultNoPid", "D candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fPtVsMassVsMultUncorr=new TH3F("hPtVsMassvsMultUncorr", "D candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fPtVsMassVsMultPart=new TH3F("hPtVsMassvsMultPart", "D candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fPtVsMassVsMultAntiPart=new TH3F("hPtVsMassvsMultAntiPart", "D candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fPtVsMassVsMultMC=new TH3F("hPtVsMassvsMultMC", "D true candidates: p_{t} vs mass vs tracklets multiplicity; Tracklets; Mass M [GeV/c^{2}]; p_{t} [GeV/c]",nMultBins,firstMultBin,lastMultBin,fNMassBins,fLowmasslimit,fUpmasslimit,48,0.,24.); fOutput->Add(fPtVsMassVsMult); fOutput->Add(fPtVsMassVsMultUncorr); fOutput->Add(fPtVsMassVsMultNoPid); fOutput->Add(fPtVsMassVsMultPart); fOutput->Add(fPtVsMassVsMultAntiPart); fOutput->Add(fPtVsMassVsMultMC); if(fDoImpPar) CreateImpactParameterHistos(); fCounter = new AliNormalizationCounter("NormCounterCorrMult"); fCounter->SetStudyMultiplicity(kTRUE,1.); fCounter->Init(); fCounterU = new AliNormalizationCounter("NormCounterUnCorrMult"); fCounterU->SetStudyMultiplicity(kTRUE,1.); fCounterU->Init(); fOutputCounters = new TList(); fOutputCounters->SetOwner(); fOutputCounters->SetName("OutputCounters"); fOutputCounters->Add(fCounter); fOutputCounters->Add(fCounterU); PostData(1,fOutput); PostData(2,fListCuts); PostData(3,fOutputCounters); PostData(4,fListProfiles); if(fUseNchWeight) CreateMeasuredNchHisto(); return; } //________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::UserExec(Option_t */*option*/) { // Execute analysis for current event: // heavy flavor candidates association to MC truth AliAODEvent *aod = dynamic_cast (InputEvent()); // AliAODTracklets* tracklets = aod->GetTracklets(); //Int_t ntracklets = tracklets->GetNumberOfTracklets(); TClonesArray *arrayCand = 0; TString arrayName=""; UInt_t pdgDau[3]; Int_t nDau=0; Int_t selbit=0; if(fPdgMeson==411){ arrayName="Charm3Prong"; pdgDau[0]=211; pdgDau[1]=321; pdgDau[2]=211; nDau=3; selbit=AliRDHFCuts::kDplusCuts; }else if(fPdgMeson==421){ arrayName="D0toKpi"; pdgDau[0]=211; pdgDau[1]=321; pdgDau[2]=0; nDau=2; selbit=AliRDHFCuts::kD0toKpiCuts; }else if(fPdgMeson==413){ arrayName="Dstar"; pdgDau[0]=321; pdgDau[1]=211; pdgDau[2]=0; // Quoting here D0 daughters (D* ones on another variable later) nDau=2; selbit=AliRDHFCuts::kDstarCuts; } if(!aod && AODEvent() && IsStandardAOD()) { // In case there is an AOD handler writing a standard AOD, use the AOD // event in memory rather than the input (ESD) event. aod = dynamic_cast (AODEvent()); // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root) // have to taken from the AOD event hold by the AliAODExtension AliAODHandler* aodHandler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); if(aodHandler->GetExtensions()) { AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root"); AliAODEvent *aodFromExt = ext->GetAOD(); arrayCand=(TClonesArray*)aodFromExt->GetList()->FindObject(arrayName.Data()); } } else if(aod) { arrayCand=(TClonesArray*)aod->GetList()->FindObject(arrayName.Data()); } if(!aod || !arrayCand) { printf("AliAnalysisTaskSEDvsMultiplicity::UserExec: Charm3Prong branch not found!\n"); return; } if(fisPPbData && fReadMC){ Int_t runnumber = aod->GetRunNumber(); if(aod->GetTriggerMask()==0 && (runnumber>=195344 && runnumber<=195677)){ AliDebug(3,"Event rejected because of null trigger mask"); return; } } // fix for temporary bug in ESDfilter // the AODs with null vertex pointer didn't pass the PhysSel if(!aod->GetPrimaryVertex()||TMath::Abs(aod->GetMagneticField())<0.001) return; // Int_t countTreta1=AliVertexingHFUtils::GetNumberOfTrackletsInEtaRange(aod,-1.,1.); // Int_t countTreta03=AliVertexingHFUtils::GetNumberOfTrackletsInEtaRange(aod,-0.3,0.3); // Int_t countTreta05=AliVertexingHFUtils::GetNumberOfTrackletsInEtaRange(aod,-0.5,0.5); // Int_t countTreta16=AliVertexingHFUtils::GetNumberOfTrackletsInEtaRange(aod,-1.6,1.6); Int_t countTreta1=0, countTreta03=0, countTreta05=0, countTreta16=0; AliAODTracklets* tracklets=aod->GetTracklets(); Int_t nTr=tracklets->GetNumberOfTracklets(); for(Int_t iTr=0; iTrGetTheta(iTr); Double_t eta=-TMath::Log(TMath::Tan(theta/2.)); if(eta>-0.3 && eta<0.3) countTreta03++; if(eta>-0.5 && eta<0.5) countTreta05++; if(eta>-1.0 && eta<1.0) countTreta1++; if(eta>-1.6 && eta<1.6) countTreta16++; } Int_t vzeroMult=0; Int_t vzeroMultA=0; AliAODVZERO *vzeroAOD = (AliAODVZERO*)aod->GetVZEROData(); if(vzeroAOD) { vzeroMultA = vzeroAOD->GetMTotV0A(); vzeroMult = vzeroMultA + vzeroAOD->GetMTotV0C(); } Int_t countMult = countTreta1; if(fMultiplicityEstimator==kNtrk03) { countMult = countTreta03; } else if(fMultiplicityEstimator==kNtrk05) { countMult = countTreta05; } else if(fMultiplicityEstimator==kNtrk10to16) { countMult = countTreta16 - countTreta1; } else if(fMultiplicityEstimator==kVZERO) { countMult = vzeroMult; } else if(fMultiplicityEstimator==kVZEROA) { countMult = vzeroMultA; } fCounterU->StoreEvent(aod,fRDCutsAnalysis,fReadMC,countMult); fHistNEvents->Fill(0); // count event Double_t countTreta1corr=countTreta1; Double_t countCorr=countMult; AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex(); // if(vtx1){ // FIX ME: No correction to the VZERO !! if(vtx1 && (fMultiplicityEstimator!=kVZERO) && (fMultiplicityEstimator!=kVZEROA)){ if(vtx1->GetNContributors()>0){ fHistNEvents->Fill(1); TProfile* estimatorAvg = GetEstimatorHistogram(aod); if(estimatorAvg){ countTreta1corr=AliVertexingHFUtils::GetCorrectedNtracklets(estimatorAvg,countTreta1,vtx1->GetZ(),fRefMult); countCorr=AliVertexingHFUtils::GetCorrectedNtracklets(estimatorAvg,countMult,vtx1->GetZ(),fRefMult); } } } Bool_t isEvSel=fRDCutsAnalysis->IsEventSelected(aod); if(fRDCutsAnalysis->GetWhyRejection()==5) fHistNEvents->Fill(3); if(fRDCutsAnalysis->GetWhyRejection()==7) fHistNEvents->Fill(4); if(fRDCutsAnalysis->GetWhyRejection()==6) fHistNEvents->Fill(5); if(fRDCutsAnalysis->GetWhyRejection()==1) fHistNEvents->Fill(6); Bool_t isEvPSRejected = fRDCutsAnalysis->IsEventRejectedDuePhysicsSelection(); if(!isEvPSRejected){ fHistNtrUnCorrPSSel->Fill(countMult); fHistNtrCorrPSSel->Fill(countCorr); } if(!isEvSel)return; if(fKeepCorrPlots){ fHistNtrEta16vsNtrEta1EvSel->Fill(countTreta1,countTreta16); fHistNtrEta05vsNtrEta1EvSel->Fill(countTreta1,countTreta05); fHistNtrEta03vsNtrEta1EvSel->Fill(countTreta1,countTreta03); fHistNtrEtaV0AvsNtrEta1EvSel->Fill(countTreta1,vzeroMultA); fHistNtrEtaV0MvsNtrEta1EvSel->Fill(countTreta1,vzeroMult); fHistNtrCorrEta1vsNtrRawEta1EvSel->Fill(countTreta1,countTreta1corr); } if(vtx1){ fHistNtrVsZvtx->Fill(vtx1->GetZ(),countMult); fHistNtrCorrVsZvtx->Fill(vtx1->GetZ(),countCorr); } TClonesArray *arrayMC=0; AliAODMCHeader *mcHeader=0; Double_t nchWeight=1.0; // load MC particles if(fReadMC){ arrayMC = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName()); if(!arrayMC) { printf("AliAnalysisTaskSEDvsMultiplicity::UserExec: MC particles branch not found!\n"); return; } // load MC header mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName()); if(!mcHeader) { printf("AliAnalysisTaskSEDvsMultiplicity::UserExec: MC header branch not found!\n"); return; } // Int_t nChargedMC=AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,-1.0,1.0); // Int_t nChargedMCPrimary=AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,-1.0,1.0); // Int_t nChargedMCPhysicalPrimary=AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,-1.0,1.0); // Int_t nChargedMCEta10=0, nChargedMCEta03=0, nChargedMCEta05=0, nChargedMCEta16=0, nChargedMCEtam37tm17=0, nChargedMCEta28t51=0; Int_t nChargedMCPrimaryEta10=0, nChargedMCPrimaryEta03=0, nChargedMCPrimaryEta05=0, nChargedMCPrimaryEta16=0, nChargedMCPrimaryEtam37tm17=0, nChargedMCPrimaryEta28t51=0; Int_t nChargedMCPhysicalPrimaryEta10=0, nChargedMCPhysicalPrimaryEta03=0, nChargedMCPhysicalPrimaryEta05=0, nChargedMCPhysicalPrimaryEta16=0, nChargedMCPhysicalPrimaryEtam37tm17=0, nChargedMCPhysicalPrimaryEta28t51=0; for(Int_t i=0; iGetEntriesFast(); i++){ AliAODMCParticle *part=(AliAODMCParticle*)arrayMC->UncheckedAt(i); Int_t charge = part->Charge(); Double_t eta = part->Eta(); Bool_t isPrim = part->IsPrimary(); Bool_t isPhysPrim = part->IsPhysicalPrimary(); if(charge!=0) { if(eta>-0.3 && eta< 0.3) { nChargedMCEta03++; if(isPrim) nChargedMCPrimaryEta03++; if(isPhysPrim) nChargedMCPhysicalPrimaryEta03++; } if(eta>-0.5 && eta< 0.5) { nChargedMCEta05++; if(isPrim) nChargedMCPrimaryEta05++; if(isPhysPrim) nChargedMCPhysicalPrimaryEta05++; } if(eta>-1.0 && eta< 1.0) { nChargedMCEta10++; if(isPrim) nChargedMCPrimaryEta10++; if(isPhysPrim) nChargedMCPhysicalPrimaryEta10++; } if(eta>-1.6 && eta< 1.6) { nChargedMCEta16++; if(isPrim) nChargedMCPrimaryEta16++; if(isPhysPrim) nChargedMCPhysicalPrimaryEta16++; } if(eta>-3.7 && eta<-1.7) { nChargedMCEtam37tm17++; if(isPrim) nChargedMCPrimaryEtam37tm17++; if(isPhysPrim) nChargedMCPhysicalPrimaryEtam37tm17++; } if(eta> 2.8 && eta< 5.1) { nChargedMCEta28t51++; if(isPrim) nChargedMCPrimaryEta28t51++; if(isPhysPrim) nChargedMCPhysicalPrimaryEta28t51++; } } } Int_t nChargedMC=nChargedMCEta10; Int_t nChargedMCPrimary=nChargedMCPrimaryEta10; Int_t nChargedMCPhysicalPrimary=nChargedMCPhysicalPrimaryEta10; // Compute the Nch weights (reference is Ntracklets within |eta|<1.0) if(fUseNchWeight){ Double_t tmpweight = 1.0; if(nChargedMCPhysicalPrimary<=0) tmpweight = 0.0; else{ Double_t pMeas = fHistoMeasNch->GetBinContent(fHistoMeasNch->FindBin(nChargedMCPhysicalPrimary)); // printf(" pMeas=%2.2f and histo MCNch %s \n",pMeas,fHistoMCNch); Double_t pMC = fHistoMCNch->GetBinContent(fHistoMCNch->FindBin(nChargedMCPhysicalPrimary)); tmpweight = pMC>0 ? pMeas/pMC : 0.; } nchWeight *= tmpweight; AliDebug(2,Form("Using Nch weights, Mult=%d Weight=%f\n",nChargedMCPhysicalPrimary,nchWeight)); } // Now recompute the variables in case another MC estimator is considered // Int_t nChargedMCEta10 = nChargedMC; // Int_t nChargedMCEta16 = AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,-1.6,1.6); // Int_t nChargedMCEta05 = AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,-0.5,0.5); // Int_t nChargedMCEta03 = AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,-0.3,0.3); // Int_t nChargedMCEtam37tm17 = AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,-3.7,-1.7); // Int_t nChargedMCEta28t51 = AliVertexingHFUtils::GetGeneratedMultiplicityInEtaRange(arrayMC,2.8,5.1); // Int_t nChargedMCPrimaryEta10 = nChargedMCPrimary; // Int_t nChargedMCPrimaryEta16 = AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,-1.6,1.6); // Int_t nChargedMCPrimaryEta05 = AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,-0.5,0.5); // Int_t nChargedMCPrimaryEta03 = AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,-0.3,0.3); // Int_t nChargedMCPrimaryEtam37tm17 = AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,-3.7,-1.7); // Int_t nChargedMCPrimaryEta28t51 = AliVertexingHFUtils::GetGeneratedPrimariesInEtaRange(arrayMC,2.8,5.1); // Int_t nChargedMCPhysicalPrimaryEta10 = nChargedMCPhysicalPrimary; // Int_t nChargedMCPhysicalPrimaryEta16 = AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,-1.6,1.6); // Int_t nChargedMCPhysicalPrimaryEta05 = AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,-0.5,0.5); // Int_t nChargedMCPhysicalPrimaryEta03 = AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,-0.3,0.3); // Int_t nChargedMCPhysicalPrimaryEtam37tm17 = AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,-3.7,-1.7); // Int_t nChargedMCPhysicalPrimaryEta28t51 = AliVertexingHFUtils::GetGeneratedPhysicalPrimariesInEtaRange(arrayMC,2.8,5.1); if(fMCPrimariesEstimator==kEta10to16){ nChargedMC = nChargedMCEta16 - nChargedMCEta10; nChargedMCPrimary = nChargedMCPrimaryEta16 - nChargedMCPrimaryEta10; nChargedMCPhysicalPrimary = nChargedMCPhysicalPrimaryEta16 - nChargedMCPhysicalPrimaryEta10; } else if(fMCPrimariesEstimator==kEta05){ nChargedMC = nChargedMCEta05; nChargedMCPrimary = nChargedMCPrimaryEta05; nChargedMCPhysicalPrimary = nChargedMCPhysicalPrimaryEta05; } else if(fMCPrimariesEstimator==kEta03){ nChargedMC = nChargedMCEta03; nChargedMCPrimary = nChargedMCPrimaryEta03; nChargedMCPhysicalPrimary = nChargedMCPhysicalPrimaryEta03; } else if(fMCPrimariesEstimator==kEtaVZERO){ nChargedMC = nChargedMCEtam37tm17 + nChargedMCEta28t51; nChargedMCPrimary = nChargedMCPrimaryEtam37tm17 + nChargedMCPrimaryEta28t51; nChargedMCPhysicalPrimary = nChargedMCPhysicalPrimaryEtam37tm17 + nChargedMCPhysicalPrimaryEta28t51; } else if(fMCPrimariesEstimator==kEtaVZEROA){ nChargedMC = nChargedMCEta28t51; nChargedMCPrimary = nChargedMCPrimaryEta28t51; nChargedMCPhysicalPrimary = nChargedMCPhysicalPrimaryEta28t51; } // Here fill the MC correlation plots if(nChargedMCPhysicalPrimary>0){ // INEL>0 for |eta|<1 fHistGenPrimaryParticlesInelGt0->Fill(nChargedMCPhysicalPrimary,nchWeight); } fHistNtrVsNchMC->Fill(nChargedMC,countMult,nchWeight); fHistNtrCorrVsNchMC->Fill(nChargedMC,countCorr,nchWeight); fHistNtrVsNchMCPrimary->Fill(nChargedMCPrimary,countMult,nchWeight); fHistNtrCorrVsNchMCPrimary->Fill(nChargedMCPrimary,countCorr,nchWeight); fHistNtrVsNchMCPhysicalPrimary->Fill(nChargedMCPhysicalPrimary,countMult,nchWeight); fHistNtrCorrVsNchMCPhysicalPrimary->Fill(nChargedMCPhysicalPrimary,countCorr,nchWeight); fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary->Fill(nChargedMC,nChargedMCPrimary,nChargedMCPhysicalPrimary,nchWeight); } Int_t nCand = arrayCand->GetEntriesFast(); Int_t nSelectedNoPID=0,nSelectedPID=0,nSelectedInMassPeak=0; Double_t mD0PDG = TDatabasePDG::Instance()->GetParticle(421)->Mass(); Double_t mDplusPDG = TDatabasePDG::Instance()->GetParticle(411)->Mass(); Double_t mDstarPDG = TDatabasePDG::Instance()->GetParticle(413)->Mass(); // pdg of daughters needed for D* too UInt_t pdgDgDStartoD0pi[2]={421,211}; Double_t aveMult=0.; Double_t nSelCand=0.; for (Int_t iCand = 0; iCand < nCand; iCand++) { AliAODRecoDecayHF *d = (AliAODRecoDecayHF*)arrayCand->UncheckedAt(iCand); AliAODRecoCascadeHF *dCascade = NULL; if(fPdgMeson==413) dCascade = (AliAODRecoCascadeHF*)d; fHistNEvents->Fill(7); if(fUseBit && !d->HasSelectionBit(selbit)){ fHistNEvents->Fill(8); continue; } Double_t ptCand = d->Pt(); Double_t rapid=d->Y(fPdgMeson); Bool_t isFidAcc=fRDCutsAnalysis->IsInFiducialAcceptance(ptCand,rapid); if(!isFidAcc) continue; Int_t labD=-1; if(fReadMC) { if(fPdgMeson==413){ labD = dCascade->MatchToMC(fPdgMeson,421,(Int_t*)pdgDgDStartoD0pi,(Int_t*)pdgDau,arrayMC); } else { labD = d->MatchToMC(fPdgMeson,arrayMC,nDau,(Int_t*)pdgDau); } FillMCMassHistos(arrayMC,labD, countMult,nchWeight); } Int_t passAllCuts=fRDCutsAnalysis->IsSelected(d,AliRDHFCuts::kAll,aod); Int_t passTopolCuts=fRDCutsAnalysis->GetIsSelectedCuts(); if(passTopolCuts==0) continue; nSelectedNoPID++; fHistNEvents->Fill(9); if(passAllCuts){ nSelectedPID++; fHistNEvents->Fill(10); } Double_t multForCand = countCorr; if(fSubtractTrackletsFromDau){ // For the D* case, subtract only the D0 daughter tracks <=== FIXME !! AliAODRecoDecayHF2Prong* d0fromDstar = NULL; if(fPdgMeson==413) d0fromDstar = (AliAODRecoDecayHF2Prong*)dCascade->Get2Prong(); for(Int_t iDau=0; iDauGetDaughter(iDau); } else{ t = (AliAODTrack*)d->GetDaughter(iDau); } if(!t) continue; if(t->HasPointOnITSLayer(0) && t->HasPointOnITSLayer(1)){ if(multForCand>0) multForCand-=1; } } } Bool_t isPrimary=kTRUE; Double_t trueImpParXY=9999.; Double_t impparXY=d->ImpParXY()*10000.; Double_t dlen=0.1; //FIXME Double_t mass[2]; if(fPdgMeson==411){ mass[0]=d->InvMass(nDau,pdgDau); mass[1]=-1.; if(TMath::Abs(mass[0]-mDplusPDG)<0.02) nSelectedInMassPeak++; //20 MeV for now... FIXME }else if(fPdgMeson==421){ UInt_t pdgdaughtersD0[2]={211,321};//pi,K UInt_t pdgdaughtersD0bar[2]={321,211};//K,pi mass[0]=d->InvMass(2,pdgdaughtersD0); mass[1]=d->InvMass(2,pdgdaughtersD0bar); if(TMath::Abs(mass[0]-mD0PDG)<0.02 || TMath::Abs(mass[1]-mD0PDG)<0.02 ) nSelectedInMassPeak++; //20 MeV for now... FIXME }else if(fPdgMeson==413){ // FIXME mass[0]=dCascade->DeltaInvMass(); mass[1]=-1.; if(TMath::Abs(mass[0]-(mDstarPDG-mD0PDG))<0.0015) nSelectedInMassPeak++; //1 MeV for now... FIXME } for(Int_t iHyp=0; iHyp<2; iHyp++){ if(mass[iHyp]<0.) continue; // for D+ and D* we have 1 mass hypothesis Double_t invMass=mass[iHyp]; Double_t arrayForSparse[5]={invMass,ptCand,impparXY,dlen,multForCand}; if(fReadMC){ if(fPdgMeson==413){ labD = dCascade->MatchToMC(fPdgMeson,421,(Int_t*)pdgDgDStartoD0pi,(Int_t*)pdgDau,arrayMC); } else { labD = d->MatchToMC(fPdgMeson,arrayMC,nDau,(Int_t*)pdgDau); } Bool_t fillHisto=fDoImpPar; if(labD>=0){ AliAODMCParticle *partD = (AliAODMCParticle*)arrayMC->At(labD); Int_t code=partD->GetPdgCode(); if(CheckOrigin(arrayMC,partD)==5) isPrimary=kFALSE; if(code<0 && iHyp==0) fillHisto=kFALSE; if(code>0 && iHyp==1) fillHisto=kFALSE; if(!isPrimary){ if(fPdgMeson==411){ trueImpParXY=AliVertexingHFUtils::GetTrueImpactParameterDplus(mcHeader,arrayMC,partD)*10000.; }else if(fPdgMeson==421){ trueImpParXY=AliVertexingHFUtils::GetTrueImpactParameterDzero(mcHeader,arrayMC,partD)*10000.; }else if(fPdgMeson==413){ trueImpParXY=0.; /// FIXME } Double_t arrayForSparseTrue[5]={invMass,ptCand,trueImpParXY,dlen,multForCand}; if(fillHisto && passAllCuts){ fHistMassPtImpPar[2]->Fill(arrayForSparse); fHistMassPtImpPar[3]->Fill(arrayForSparseTrue); } }else{ if(fillHisto && passAllCuts) fHistMassPtImpPar[1]->Fill(arrayForSparse); } }else{ if(fillHisto && passAllCuts)fHistMassPtImpPar[4]->Fill(arrayForSparse); } if(TMath::Abs(labD)==fPdgMeson && fMCOption==2) continue; if(TMath::Abs(labD)!=fPdgMeson && fMCOption==1) continue; } if(fPdgMeson==421){ if(iHyp==0 && !(passTopolCuts&1)) continue; // candidate not passing as D0 if(iHyp==1 && !(passTopolCuts&2)) continue; // candidate not passing as D0bar } fPtVsMassVsMultNoPid->Fill(multForCand,invMass,ptCand); if(fPdgMeson==421){ if(iHyp==0 && !(passAllCuts&1)) continue; // candidate not passing as D0 if(iHyp==1 && !(passAllCuts&2)) continue; // candidate not passing as D0bar } if(passAllCuts){ aveMult+=multForCand; nSelCand+=1.; fPtVsMassVsMult->Fill(multForCand,invMass,ptCand,nchWeight); fPtVsMassVsMultUncorr->Fill(countTreta1,invMass,ptCand,nchWeight); // Add separation between part antipart if(fPdgMeson==411){ if(d->GetCharge()>0) fPtVsMassVsMultPart->Fill(multForCand,invMass,ptCand,nchWeight); else fPtVsMassVsMultAntiPart->Fill(multForCand,invMass,ptCand,nchWeight); }else if(fPdgMeson==421){ if(passAllCuts&1) fPtVsMassVsMultPart->Fill(multForCand,invMass,ptCand,nchWeight); if(passAllCuts&2) fPtVsMassVsMultAntiPart->Fill(multForCand,invMass,ptCand,nchWeight); }else if(fPdgMeson==413){ if(d->GetCharge()>0) fPtVsMassVsMultPart->Fill(multForCand,invMass,ptCand,nchWeight); else fPtVsMassVsMultAntiPart->Fill(multForCand,invMass,ptCand,nchWeight); } if(fDoImpPar){ fHistMassPtImpPar[0]->Fill(arrayForSparse); } } } } if(fSubtractTrackletsFromDau && nSelCand>0){ aveMult/=nSelCand; fCounter->StoreEvent(aod,fRDCutsAnalysis,fReadMC,(Int_t)(aveMult+0.5001)); }else{ fCounter->StoreEvent(aod,fRDCutsAnalysis,fReadMC,(Int_t)countCorr); } fCounter->StoreCandidates(aod,nSelectedNoPID,kTRUE); fCounter->StoreCandidates(aod,nSelectedPID,kFALSE); fHistNtrUnCorrEvSel->Fill(countMult,nchWeight); fHistNtrCorrEvSel->Fill(countCorr,nchWeight); if(nSelectedPID>0) { fHistNtrUnCorrEvWithCand->Fill(countMult,nchWeight); fHistNtrCorrEvWithCand->Fill(countCorr,nchWeight); if(fKeepCorrPlots){ fHistNtrEta16vsNtrEta1EvWithCand->Fill(countTreta1,countTreta16); fHistNtrEta05vsNtrEta1EvWithCand->Fill(countTreta1,countTreta05); fHistNtrEta03vsNtrEta1EvWithCand->Fill(countTreta1,countTreta03); fHistNtrEtaV0AvsNtrEta1EvWithCand->Fill(countTreta1,vzeroMultA); fHistNtrEtaV0MvsNtrEta1EvWithCand->Fill(countTreta1,vzeroMult); fHistNtrCorrEta1vsNtrRawEta1EvWithCand->Fill(countTreta1,countTreta1corr); } } if(nSelectedInMassPeak>0) { fHistNtrUnCorrEvWithD->Fill(countMult,nchWeight); fHistNtrCorrEvWithD->Fill(countCorr,nchWeight); if(fKeepCorrPlots){ fHistNtrEta16vsNtrEta1EvWithD->Fill(countTreta1,countTreta16); fHistNtrEta05vsNtrEta1EvWithD->Fill(countTreta1,countTreta05); fHistNtrEta03vsNtrEta1EvWithD->Fill(countTreta1,countTreta03); fHistNtrEtaV0AvsNtrEta1EvWithD->Fill(countTreta1,vzeroMultA); fHistNtrEtaV0MvsNtrEta1EvWithD->Fill(countTreta1,vzeroMult); fHistNtrCorrEta1vsNtrRawEta1EvWithD->Fill(countTreta1,countTreta1corr); } } PostData(1,fOutput); PostData(2,fListCuts); PostData(3,fOutputCounters); return; } //________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::CreateImpactParameterHistos(){ // Histos for impact paramter study // mass . pt , impact parameter , decay length , multiplicity Int_t nbins[5]={fNMassBins,200,fNImpParBins,50,100}; Double_t xmin[5]={fLowmasslimit,0.,fLowerImpPar,0.,0.}; Double_t xmax[5]={fUpmasslimit,20.,fHigherImpPar,1.,100.}; fHistMassPtImpPar[0]=new THnSparseF("hMassPtImpParAll", "Mass vs. pt vs.imppar - All", 5,nbins,xmin,xmax); fHistMassPtImpPar[1]=new THnSparseF("hMassPtImpParPrompt", "Mass vs. pt vs.imppar - promptD", 5,nbins,xmin,xmax); fHistMassPtImpPar[2]=new THnSparseF("hMassPtImpParBfeed", "Mass vs. pt vs.imppar - DfromB", 5,nbins,xmin,xmax); fHistMassPtImpPar[3]=new THnSparseF("hMassPtImpParTrueBfeed", "Mass vs. pt vs.true imppar -DfromB", 5,nbins,xmin,xmax); fHistMassPtImpPar[4]=new THnSparseF("hMassPtImpParBkg", "Mass vs. pt vs.imppar - backgr.", 5,nbins,xmin,xmax); for(Int_t i=0; i<5;i++){ fOutput->Add(fHistMassPtImpPar[i]); } } //________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::Terminate(Option_t */*option*/) { // Terminate analysis // if(fDebug > 1) printf("AnalysisTaskSEDvsMultiplicity: Terminate() \n"); fOutput = dynamic_cast (GetOutputData(1)); if (!fOutput) { printf("ERROR: fOutput not available\n"); return; } fHistNEvents = dynamic_cast(fOutput->FindObject("fHistNEvents")); if(!fHistNEvents){ printf("ERROR: fHistNEvents not available\n"); return; } printf("Number of Analyzed Events = %d\n",(Int_t)fHistNEvents->GetBinContent(3)); return; } //_________________________________________________________________________________________________ Int_t AliAnalysisTaskSEDvsMultiplicity::CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const { // // checking whether the mother of the particles come from a charm or a bottom quark // Int_t pdgGranma = 0; Int_t mother = 0; mother = mcPartCandidate->GetMother(); Int_t istep = 0; Int_t abspdgGranma =0; Bool_t isFromB=kFALSE; // Bool_t isQuarkFound=kFALSE; while (mother >0 ){ istep++; AliAODMCParticle* mcGranma = dynamic_cast(arrayMC->At(mother)); if (mcGranma){ pdgGranma = mcGranma->GetPdgCode(); abspdgGranma = TMath::Abs(pdgGranma); if ((abspdgGranma > 500 && abspdgGranma < 600) || (abspdgGranma > 5000 && abspdgGranma < 6000)){ isFromB=kTRUE; } // if(abspdgGranma==4 || abspdgGranma==5) isQuarkFound=kTRUE; mother = mcGranma->GetMother(); }else{ AliError("Failed casting the mother particle!"); break; } } if(isFromB) return 5; else return 4; } //____________________________________________________________________________ TProfile* AliAnalysisTaskSEDvsMultiplicity::GetEstimatorHistogram(const AliVEvent* event){ // Get Estimator Histogram from period event->GetRunNumber(); // // If you select SPD tracklets in |eta|<1 you should use type == 1 // Int_t runNo = event->GetRunNumber(); Int_t period = -1; // pp: 0-LHC10b, 1-LHC10c, 2-LHC10d, 3-LHC10e // pPb: 0-LHC13b, 1-LHC13c if (fisPPbData) { if (runNo>195343 && runNo<195484) period = 0; if (runNo>195528 && runNo<195678) period = 1; if (period < 0 || period > 1) return 0; } else { if(runNo>114930 && runNo<117223) period = 0; if(runNo>119158 && runNo<120830) period = 1; if(runNo>122373 && runNo<126438) period = 2; if(runNo>127711 && runNo<130841) period = 3; if(period<0 || period>3) return 0; } return fMultEstimatorAvg[period]; } //__________________________________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::CreateMeasuredNchHisto(){ // creates historgam with measured multiplcity distribution in pp 7 TeV collisions (from Eur. Phys. J. C (2010) 68: 345–354) // // for Nch > 70 the points were obtainedwith a double NBD distribution // TF1 *fit1 = new TF1("fit1","[0]*(TMath::Gamma(x+[1])/(TMath::Gamma(x+1)*TMath::Gamma([1])))*(TMath::Power(([2]/[1]),x))*(TMath::Power((1+([2]/[1])),-x-[1]))"); fit1->SetParameter(0,1.);// normalization constant // fit1->SetParameter(1,1.63); // k parameter // fit1->SetParameter(2,12.8); // mean multiplicity Double_t nchbins[82]={0.50,1.50,2.50,3.50,4.50,5.50,6.50,7.50,8.50,9.50, 10.50,11.50,12.50,13.50,14.50,15.50,16.50,17.50,18.50,19.50, 20.50,21.50,22.50,23.50,24.50,25.50,26.50,27.50,28.50,29.50, 30.50,31.50,32.50,33.50,34.50,35.50,36.50,37.50,38.50,39.50, 40.50,41.50,42.50,43.50,44.50,45.50,46.50,47.50,48.50,49.50, 50.50,51.50,52.50,53.50,54.50,55.50,56.50,57.50,58.50,59.50, 60.50,62.50,64.50,66.50,68.50,70.50,72.50,74.50,76.50,78.50, 80.50,82.50,84.50,86.50,88.50,90.50,92.50,94.50,96.50,98.50, 100.50,102.50}; Double_t pch[81]={0.062011,0.072943,0.070771,0.067245,0.062834,0.057383,0.051499,0.04591,0.041109,0.036954, 0.03359,0.030729,0.028539,0.026575,0.024653,0.0229,0.021325,0.019768,0.018561,0.017187, 0.01604,0.014836,0.013726,0.012576,0.011481,0.010393,0.009502,0.008776,0.008024,0.007452, 0.006851,0.006428,0.00594,0.005515,0.005102,0.00469,0.004162,0.003811,0.003389,0.003071, 0.002708,0.002422,0.002184,0.001968,0.00186,0.00165,0.001577,0.001387,0.001254,0.001118, 0.001037,0.000942,0.000823,0.000736,0.000654,0.000579,0.000512,0.00049,0.00045,0.000355, 0.000296,0.000265,0.000193,0.00016,0.000126,0.0000851, 0.0000676,0.0000537,0.0000426, 0.0000338, 0.0000268,0.0000213,0.0000166,0.0000133,0.0000106,0.00000837,0.00000662, 0.00000524,0.00000414, 0.00000327, 0.00000258}; if(fHistoMeasNch) delete fHistoMeasNch; fHistoMeasNch=new TH1F("hMeaseNch","",81,nchbins); for(Int_t i=0; i<81; i++){ fHistoMeasNch->SetBinContent(i+1,pch[i]); fHistoMeasNch->SetBinError(i+1,0.); } } //__________________________________________________________________________________________________ void AliAnalysisTaskSEDvsMultiplicity::FillMCMassHistos(TClonesArray *arrayMC, Int_t labD, Int_t countMult,Double_t nchWeight) { // // Function to fill the true MC signal // if(labD>=0){ AliAODMCParticle *partD = (AliAODMCParticle*)arrayMC->At(labD); Double_t mass = partD->M(); Double_t pt = partD->Pt(); fPtVsMassVsMultMC->Fill(countMult,mass,pt,nchWeight); } }