Added AlidNdPtAnalysisPbPbAOD.{h,cxx}

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / ChargedHadrons / dNdPt / AlidNdPtAnalysisPbPbAOD.cxx

/**************************************************************************
* 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.                  *
**************************************************************************/
//------------------------------------------------------------------------------
// AlidNdPtAnalysisPbPbAOD class. 
// 
// Author: P. Luettig, 15.05.2013
//------------------------------------------------------------------------------


#include "AlidNdPtAnalysisPbPbAOD.h"


using namespace std;

ClassImp(AlidNdPtAnalysisPbPbAOD)

// dummy constructor
AlidNdPtAnalysisPbPbAOD::AlidNdPtAnalysisPbPbAOD() : AliAnalysisTaskSE(),
fOutputList(0),
// Histograms
hPt(0),
hMCPt(0),
hnZvPtEtaCent(0),
hnMCRecPrimZvPtEtaCent(0),
hnMCGenZvPtEtaCent(0),
hnMCSecZvPtEtaCent(0),
hEventStatistics(0),
hEventStatisticsCentrality(0),
hAllEventStatisticsCentrality(0),
hnZvMultCent(0),
hTriggerStatistics(0),
hMCTrackPdgCode(0),
hMCTrackStatusCode(0),
hCharge(0),
hMCCharge(0),
hMCPdgPt(0),
hMCHijingPrim(0),
hAccNclsTPC(0),
hAccCrossedRowsTPC(0),
//global
bIsMonteCarlo(0),
// event cut variables
dCutMaxZVertex(10.),  
// track kinematic cut variables
dCutPtMin(0.15),
dCutPtMax(1000.),
dCutEtaMin(-0.8),
dCutEtaMax(0.8),    
// track quality cut variables
bCutRequireTPCRefit(kTRUE),
dCutMinNumberOfCrossedRows(120.),
dCutMinRatioCrossedRowsOverFindableClustersTPC(0.8),
dCutMaxChi2PerClusterTPC(4.),
dCutMaxFractionSharedTPCClusters(0.4),
dCutMaxDCAToVertexZ(3.0),
dCutMaxDCAToVertexXY(3.0),
bCutRequireITSRefit(kTRUE),
dCutMaxChi2PerClusterITS(36.),
dCutDCAToVertex2D(kFALSE),
dCutRequireSigmaToVertex(kFALSE),
dCutMaxDCAToVertexXYPtDepPar0(0.0182),
dCutMaxDCAToVertexXYPtDepPar1(0.0350),
dCutMaxDCAToVertexXYPtDepPar2(1.01),
bCutAcceptKinkDaughters(kFALSE),
dCutMaxChi2TPCConstrainedGlobal(36.),
// binning for THnSparse
fMultNbins(0),
fPtNbins(0),
fPtCorrNbins(0),
fEtaNbins(0),
fZvNbins(0),
fCentralityNbins(0),
fBinsMult(0),
fBinsPt(0),
fBinsPtCorr(0),
fBinsEta(0),
fBinsZv(0),
fBinsCentrality(0)
{
  
  fMultNbins = 0;
  fPtNbins = 0;
  fPtCorrNbins = 0;
  fEtaNbins = 0;
  fZvNbins = 0;
  fCentralityNbins = 0;
  fBinsMult = 0;
  fBinsPt = 0;
  fBinsPtCorr = 0;
  fBinsEta = 0;
  fBinsEta = 0;
  fBinsZv = 0;
  fBinsCentrality = 0;
  
}

AlidNdPtAnalysisPbPbAOD::AlidNdPtAnalysisPbPbAOD(const char *name) : AliAnalysisTaskSE(name),
fOutputList(0),
// Histograms
hPt(0),
hMCPt(0),
hnZvPtEtaCent(0),
hnMCRecPrimZvPtEtaCent(0),
hnMCGenZvPtEtaCent(0),
hnMCSecZvPtEtaCent(0),
hEventStatistics(0),
hEventStatisticsCentrality(0),
hAllEventStatisticsCentrality(0),
hnZvMultCent(0),
hTriggerStatistics(0),
hMCTrackPdgCode(0),
hMCTrackStatusCode(0),
hCharge(0),
hMCCharge(0),
hMCPdgPt(0),
hMCHijingPrim(0),
hAccNclsTPC(0),
hAccCrossedRowsTPC(0),
//global
bIsMonteCarlo(0),
// event cut variables
dCutMaxZVertex(10.),  
// track kinematic cut variables
dCutPtMin(0.15),
dCutPtMax(200.),
dCutEtaMin(-0.8),
dCutEtaMax(0.8),    
// track quality cut variables
bCutRequireTPCRefit(kTRUE),
dCutMinNumberOfCrossedRows(120.),
dCutMinRatioCrossedRowsOverFindableClustersTPC(0.8),
dCutMaxChi2PerClusterTPC(4.),
dCutMaxFractionSharedTPCClusters(0.4),
dCutMaxDCAToVertexZ(3.0),
dCutMaxDCAToVertexXY(3.0),
bCutRequireITSRefit(kTRUE),
dCutMaxChi2PerClusterITS(36.),
dCutDCAToVertex2D(kFALSE),
dCutRequireSigmaToVertex(kFALSE),
dCutMaxDCAToVertexXYPtDepPar0(0.0182),
dCutMaxDCAToVertexXYPtDepPar1(0.0350),
dCutMaxDCAToVertexXYPtDepPar2(1.01),
bCutAcceptKinkDaughters(kFALSE),
dCutMaxChi2TPCConstrainedGlobal(36.),
// binning for THnSparse
fMultNbins(0),
fPtNbins(0),
fPtCorrNbins(0),
fEtaNbins(0),
fZvNbins(0),
fCentralityNbins(0),
fBinsMult(0),
fBinsPt(0),
fBinsPtCorr(0),
fBinsEta(0),
fBinsZv(0),
fBinsCentrality(0)
{
  fMultNbins = 0;
  fPtNbins = 0;
  fPtCorrNbins = 0;
  fEtaNbins = 0;
  fZvNbins = 0;
  fCentralityNbins = 0;
  fBinsMult = 0;
  fBinsPt = 0;
  fBinsPtCorr = 0;
  fBinsEta = 0;
  fBinsEta = 0;
  fBinsZv = 0;
  fBinsCentrality = 0;
  
  DefineOutput(1, TList::Class());
}

// destructor
AlidNdPtAnalysisPbPbAOD::~AlidNdPtAnalysisPbPbAOD()
{
  if(hnZvPtEtaCent) delete hnZvPtEtaCent; hnZvPtEtaCent = 0;
  if(hPt) delete hPt; hPt = 0;
}

void AlidNdPtAnalysisPbPbAOD::UserCreateOutputObjects()
{
  // create all output histograms here
  OpenFile(1, "RECREATE");
  
  fOutputList = new TList();
  fOutputList->SetOwner();
  
  //define default binning
  Double_t binsMultDefault[48] = {-0.5, 0.5 , 1.5 , 2.5 , 3.5 , 4.5 , 5.5 , 6.5 , 7.5 , 8.5,9.5, 10.5, 11.5, 12.5, 13.5, 14.5, 15.5, 16.5, 17.5, 18.5,19.5, 20.5, 30.5, 40.5 , 50.5 , 60.5 , 70.5 , 80.5 , 90.5 , 100.5,200.5, 300.5, 400.5, 500.5, 600.5, 700.5, 800.5, 900.5, 1000.5, 2000.5, 3000.5, 4000.5, 5000.5, 6000.5, 7000.5, 8000.5, 9000.5, 10000.5 }; 
  Double_t binsPtDefault[82] = {0.0, 0.05, 0.1, 0.15, 0.2, 0.25, 0.3, 0.35, 0.4, 0.45, 0.5, 0.55, 0.6, 0.65, 0.7, 0.75, 0.8, 0.85, 0.9, 0.95, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2.0, 2.2, 2.4, 2.6, 2.8, 3.0, 3.2, 3.4, 3.6, 3.8, 4.0, 4.5, 5.0, 5.5, 6.0, 6.5, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 18.0, 20.0, 22.0, 24.0, 26.0, 28.0, 30.0, 32.0, 34.0, 36.0, 40.0, 45.0, 50.0, 60.0, 70.0, 80.0, 90.0, 100.0, 110.0, 120.0, 130.0, 140.0, 150.0, 160.0, 180.0, 200.0};
  Double_t binsPtCorrDefault[37] = {0.,0.05,0.1,0.15,0.2,0.25,0.3,0.35,0.4,0.45,0.5,0.55,0.6,0.65,0.7,0.75,0.8,0.85,0.9,0.95,1.0,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9,2.0,2.2,2.4,2.6,3.0,4.0,200.0};    
  Double_t binsEtaDefault[31] = {-1.5,-1.4,-1.3,-1.2,-1.1,-1.0,-0.9,-0.8,-0.7,-0.6,-0.5,-0.4,-0.3,-0.2,-0.1,0.,0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,1.0,1.1,1.2,1.3,1.4,1.5};
  Double_t binsZvDefault[13] = {-30.,-25.,-20.,-15.,-10.,-5.,0.,5.,10.,15.,20.,25.,30.};
  Double_t binsCentralityDefault[12] = {0., 5., 10., 20., 30., 40., 50., 60., 70., 80., 90., 100.};  
  
  // if no binning is set, use the default
  if (!fBinsMult)	{ SetBinsMult(48,binsMultDefault); }
  if (!fBinsPt)		{ SetBinsPt(82,binsPtDefault); }
  if (!fBinsPtCorr)	{ SetBinsPtCorr(37,binsPtCorrDefault); }
  if (!fBinsEta)	{ SetBinsEta(31,binsEtaDefault); }
  if (!fBinsZv)		{ SetBinsZv(13,binsZvDefault); }  
  if (!fBinsCentrality)	{ SetBinsCentrality(12,binsCentralityDefault); }
  
  Int_t binsZvPtEtaCent[4]={fZvNbins-1,fPtNbins-1,fEtaNbins-1,fCentralityNbins-1};
  Int_t binsZvMultCent[3]={fZvNbins-1,fMultNbins-1,fCentralityNbins-1};
  
  // define Histograms
  hnZvPtEtaCent = new THnSparseF("hnZvPtEtaCent","Zv:Pt:Eta:Centrality",4,binsZvPtEtaCent);
  hnZvPtEtaCent->SetBinEdges(0,fBinsZv);
  hnZvPtEtaCent->SetBinEdges(1,fBinsPt);
  hnZvPtEtaCent->SetBinEdges(2,fBinsEta);
  hnZvPtEtaCent->SetBinEdges(3,fBinsCentrality);
  hnZvPtEtaCent->GetAxis(0)->SetTitle("Zv (cm)");
  hnZvPtEtaCent->GetAxis(1)->SetTitle("Pt (GeV/c)");
  hnZvPtEtaCent->GetAxis(2)->SetTitle("Eta");
  hnZvPtEtaCent->GetAxis(3)->SetTitle("Centrality");
  hnZvPtEtaCent->Sumw2();
  
  hnMCRecPrimZvPtEtaCent = new THnSparseF("hnMCRecPrimZvPtEtaCent","mcZv:mcPt:mcEta:Centrality",4,binsZvPtEtaCent);
  hnMCRecPrimZvPtEtaCent->SetBinEdges(0,fBinsZv);
  hnMCRecPrimZvPtEtaCent->SetBinEdges(1,fBinsPt);
  hnMCRecPrimZvPtEtaCent->SetBinEdges(2,fBinsEta);
  hnMCRecPrimZvPtEtaCent->SetBinEdges(3,fBinsCentrality);
  hnMCRecPrimZvPtEtaCent->GetAxis(0)->SetTitle("MC Zv (cm)");
  hnMCRecPrimZvPtEtaCent->GetAxis(1)->SetTitle("MC Pt (GeV/c)");
  hnMCRecPrimZvPtEtaCent->GetAxis(2)->SetTitle("MC Eta");
  hnMCRecPrimZvPtEtaCent->GetAxis(3)->SetTitle("Centrality");
  hnMCRecPrimZvPtEtaCent->Sumw2();
  
  hnMCGenZvPtEtaCent = new THnSparseF("hnMCGenZvPtEtaCent","mcZv:mcPt:mcEta:Centrality",4,binsZvPtEtaCent);
  hnMCGenZvPtEtaCent->SetBinEdges(0,fBinsZv);
  hnMCGenZvPtEtaCent->SetBinEdges(1,fBinsPt);
  hnMCGenZvPtEtaCent->SetBinEdges(2,fBinsEta);
  hnMCGenZvPtEtaCent->SetBinEdges(3,fBinsCentrality);
  hnMCGenZvPtEtaCent->GetAxis(0)->SetTitle("MC Zv (cm)");
  hnMCGenZvPtEtaCent->GetAxis(1)->SetTitle("MC Pt (GeV/c)");
  hnMCGenZvPtEtaCent->GetAxis(2)->SetTitle("MC Eta");
  hnMCGenZvPtEtaCent->GetAxis(3)->SetTitle("Centrality");
  hnMCGenZvPtEtaCent->Sumw2();
  
  hnMCSecZvPtEtaCent = new THnSparseF("hnMCSecZvPtEtaCent","mcZv:mcPt:mcEta:Centrality",4,binsZvPtEtaCent);
  hnMCSecZvPtEtaCent->SetBinEdges(0,fBinsZv);
  hnMCSecZvPtEtaCent->SetBinEdges(1,fBinsPt);
  hnMCSecZvPtEtaCent->SetBinEdges(2,fBinsEta);
  hnMCSecZvPtEtaCent->SetBinEdges(3,fBinsCentrality);
  hnMCSecZvPtEtaCent->GetAxis(0)->SetTitle("MC Sec Zv (cm)");
  hnMCSecZvPtEtaCent->GetAxis(1)->SetTitle("MC Sec Pt (GeV/c)");
  hnMCSecZvPtEtaCent->GetAxis(2)->SetTitle("MC Sec Eta");
  hnMCSecZvPtEtaCent->GetAxis(3)->SetTitle("Centrality");
  hnMCSecZvPtEtaCent->Sumw2();
  
  hPt = new TH1F("hPt","hPt",2000,0,200);
  hPt->GetXaxis()->SetTitle("p_{T} (GeV/c)");
  hPt->GetYaxis()->SetTitle("dN/dp_{T}");
  hPt->Sumw2();
  
  hMCPt = new TH1F("hMCPt","hMCPt",2000,0,200);
  hMCPt->GetXaxis()->SetTitle("MC p_{T} (GeV/c)");
  hMCPt->GetYaxis()->SetTitle("dN/dp_{T}");
  hMCPt->Sumw2();
  
  hEventStatistics = new TH1F("hEventStatistics","hEventStatistics",10,0,10);
  hEventStatistics->GetYaxis()->SetTitle("number of events");
  hEventStatistics->SetBit(TH1::kCanRebin);
  
  hEventStatisticsCentrality = new TH1F("hEventStatisticsCentrality","hEventStatisticsCentrality",fCentralityNbins-1, fBinsCentrality);
  hEventStatisticsCentrality->GetYaxis()->SetTitle("number of events");
  
  hAllEventStatisticsCentrality = new TH1F("hAllEventStatisticsCentrality","hAllEventStatisticsCentrality",fCentralityNbins-1, fBinsCentrality);
  hAllEventStatisticsCentrality->GetYaxis()->SetTitle("number of events");
  
  hnZvMultCent = new THnSparseF("hnZvMultCent","Zv:mult:Centrality",3,binsZvMultCent);
  hnZvMultCent->SetBinEdges(0,fBinsZv);
  hnZvMultCent->SetBinEdges(1,fBinsMult);
  hnZvMultCent->SetBinEdges(2,fBinsCentrality);
  hnZvMultCent->GetAxis(0)->SetTitle("Zv (cm)");
  hnZvMultCent->GetAxis(1)->SetTitle("N_{acc}");
  hnZvMultCent->GetAxis(2)->SetTitle("Centrality");
  hnZvMultCent->Sumw2();
  
  hTriggerStatistics = new TH1F("hTriggerStatistics","hTriggerStatistics",10,0,10);
  hTriggerStatistics->GetYaxis()->SetTitle("number of events");
  
  hMCTrackPdgCode = new TH1F("hMCTrackPdgCode","hMCTrackPdgCode",100,0,10);
  hMCTrackPdgCode->GetYaxis()->SetTitle("number of tracks");
  hMCTrackPdgCode->SetBit(TH1::kCanRebin);
  
  hMCTrackStatusCode = new TH1F("hMCTrackStatusCode","hMCTrackStatusCode",100,0,10);
  hMCTrackStatusCode->GetYaxis()->SetTitle("number of tracks");
  hMCTrackStatusCode->SetBit(TH1::kCanRebin);
  
  hCharge = new TH1F("hCharge","hCharge",30, -5, 5);
  hCharge->GetXaxis()->SetTitle("Charge");
  hCharge->GetYaxis()->SetTitle("number of tracks");
  
  hMCCharge = new TH1F("hMCCharge","hMCCharge",30, -5, 5);
  hMCCharge->GetXaxis()->SetTitle("MC Charge");
  hMCCharge->GetYaxis()->SetTitle("number of tracks");
  
  hMCPdgPt = new TH2F("hMCPdgPt","hMCPdgPt",fPtNbins-1, fBinsPt, 100,0,100);
  hMCPdgPt->GetYaxis()->SetTitle("particle");
  hMCPdgPt->GetXaxis()->SetTitle("Pt (GeV/c)");
  
  hMCHijingPrim = new TH1F("hMCHijingPrim","hMCHijingPrim",2,0,2);
  hMCPdgPt->GetYaxis()->SetTitle("number of particles");
  
  hAccNclsTPC = new TH1F("hAccNclsTPC","hAccNclsTPC",160,0,159);
  hAccNclsTPC->GetXaxis()->SetTitle("number of clusters per track after cut");
  
  hAccCrossedRowsTPC = new TH1F("hAccCrossedRowsTPC","hAccCrossedRowsTPC",160,0,159);
  hAccCrossedRowsTPC->GetXaxis()->SetTitle("number of crossed rows per track after cut");
  
  
  // Add Histos, Profiles etc to List
  fOutputList->Add(hnZvPtEtaCent);
  fOutputList->Add(hPt);
  fOutputList->Add(hnMCRecPrimZvPtEtaCent);
  fOutputList->Add(hnMCGenZvPtEtaCent);
  fOutputList->Add(hnMCSecZvPtEtaCent);
  fOutputList->Add(hMCPt);
  fOutputList->Add(hEventStatistics);
  fOutputList->Add(hEventStatisticsCentrality);
  fOutputList->Add(hAllEventStatisticsCentrality);
  fOutputList->Add(hnZvMultCent);
  fOutputList->Add(hTriggerStatistics);
  fOutputList->Add(hMCTrackPdgCode);
  fOutputList->Add(hMCTrackStatusCode);
  fOutputList->Add(hCharge);
  fOutputList->Add(hMCCharge);
  fOutputList->Add(hMCPdgPt);
  fOutputList->Add(hMCHijingPrim);
  fOutputList->Add(hAccNclsTPC);
  fOutputList->Add(hAccCrossedRowsTPC);
  
  PostData(1, fOutputList);
}

void AlidNdPtAnalysisPbPbAOD::UserExec(Option_t *option)
{
  // Main Loop
  // called for each event
  hEventStatistics->Fill("all events",1);
  
  // set ZERO pointers:
  AliInputEventHandler *inputHandler = NULL;
  AliAODTrack *track = NULL;
  AliAODMCParticle *mcPart = NULL;
  AliAODMCHeader *mcHdr = NULL;
  AliGenHijingEventHeader *genHijingHeader = NULL;
  AliGenPythiaEventHeader *genPythiaHeader = NULL;
  
  Bool_t bIsEventSelectedMB = kFALSE;
  Bool_t bIsEventSelectedSemi = kFALSE;
  Bool_t bIsEventSelectedCentral = kFALSE;
  Bool_t bIsEventSelected = kFALSE;
  Bool_t bIsPrimary = kFALSE;
  Bool_t bIsHijingParticle = kFALSE;
  Bool_t bIsPythiaParticle = kFALSE;
  Bool_t bEventHasATrack = kFALSE;
  Bool_t bEventHasATrackInRange = kFALSE;
  Int_t nTriggerFired = 0;
  
  
  Double_t dMCTrackZvPtEtaCent[4] = {0};
  Double_t dTrackZvPtEtaCent[4] = {0};
  
  Double_t dMCEventZv = -100;
  Double_t dEventZv = -100;
  Int_t iAcceptedMultiplicity = 0;
  
  bIsMonteCarlo = kFALSE;
  
  AliAODEvent *eventAOD = 0x0;
  eventAOD = dynamic_cast<AliAODEvent*>( InputEvent() );
  if (!eventAOD) {
    AliWarning("ERROR: eventAOD not available \n");
    return;
  }
  
  // check, which trigger has been fired
  inputHandler = (AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
  bIsEventSelectedMB = ( inputHandler->IsEventSelected() & AliVEvent::kMB);
  bIsEventSelectedSemi = ( inputHandler->IsEventSelected() & AliVEvent::kSemiCentral);
  bIsEventSelectedCentral = ( inputHandler->IsEventSelected() & AliVEvent::kCentral);
  
  if(bIsEventSelectedMB || bIsEventSelectedSemi || bIsEventSelectedCentral) hTriggerStatistics->Fill("all triggered events",1);
  if(bIsEventSelectedMB) { hTriggerStatistics->Fill("MB trigger",1); nTriggerFired++; }
  if(bIsEventSelectedSemi) { hTriggerStatistics->Fill("SemiCentral trigger",1); nTriggerFired++; }
  if(bIsEventSelectedCentral) { hTriggerStatistics->Fill("Central trigger",1); nTriggerFired++; }
  if(nTriggerFired == 0) { hTriggerStatistics->Fill("No trigger",1); }
  
  bIsEventSelected = ( inputHandler->IsEventSelected() & GetCollisionCandidates() );
  
  // only take tracks of events, which are triggered
  if(nTriggerFired == 0) { return; } 
  
  //   if( !bIsEventSelected || nTriggerFired>1 ) return;
  
  //   hEventStatistics->Fill("events with only coll. cand.", 1);
  
  
  
  // check if there is a stack, if yes, then do MC loop
  TList *list = eventAOD->GetList();
  TClonesArray *stack = 0x0;
  stack = (TClonesArray*)list->FindObject(AliAODMCParticle::StdBranchName());
  
  if( stack )
  {
    bIsMonteCarlo = kTRUE;
    
    mcHdr = (AliAODMCHeader*)list->FindObject(AliAODMCHeader::StdBranchName());
    
    genHijingHeader = GetHijingEventHeader(mcHdr);
    //     genPythiaHeader = GetPythiaEventHeader(mcHdr);
    
    if(!genHijingHeader) { return; }
    
    //     if(!genPythiaHeader)  { return; }
    
    dMCEventZv = mcHdr->GetVtxZ();
    dMCTrackZvPtEtaCent[0] = dMCEventZv;
    hEventStatistics->Fill("MC all events",1);
  }
  
  AliCentrality* aCentrality = eventAOD->GetCentrality();
  Double_t dCentrality = aCentrality->GetCentralityPercentile("V0M");
  
  if( dCentrality < 0 ) return;
  hEventStatistics->Fill("after centrality selection",1);
  
  
  
  // start with MC truth analysis
  if(bIsMonteCarlo)
  {
    
    if( dMCEventZv > dCutMaxZVertex )  { return; }
    
    dMCTrackZvPtEtaCent[0] = dMCEventZv;
    
    hEventStatistics->Fill("MC afterZv cut",1);
    
    for(Int_t iMCtrack = 0; iMCtrack < stack->GetEntriesFast(); iMCtrack++)
    {
      mcPart =(AliAODMCParticle*)stack->At(iMCtrack);
      
      // check for charge
      if( !(IsMCTrackAccepted(mcPart)) ) continue;
      
      if(!IsHijingParticle(mcPart, genHijingHeader)) { continue; }
      
      if(mcPart->IsPhysicalPrimary() ) 
      {
	hMCHijingPrim->Fill("IsPhysicalPrimary",1);
      }
      else
      {
	hMCHijingPrim->Fill("NOT a primary",1);
	continue;
      }
      
        
      //       
      // ======================== fill histograms ========================
      dMCTrackZvPtEtaCent[1] = mcPart->Pt();
      dMCTrackZvPtEtaCent[2] = mcPart->Eta();
      dMCTrackZvPtEtaCent[3] = dCentrality;
      
      bEventHasATrack = kTRUE;
      
      hnMCGenZvPtEtaCent->Fill(dMCTrackZvPtEtaCent);
      
      if( (dMCTrackZvPtEtaCent[1] > dCutPtMin) &&
	(dMCTrackZvPtEtaCent[1] < dCutPtMax) &&
	(dMCTrackZvPtEtaCent[2] > dCutEtaMin) &&
	(dMCTrackZvPtEtaCent[2] < dCutEtaMax) )
      {
	hMCPt->Fill(mcPart->Pt());
	hMCCharge->Fill(mcPart->Charge()/3.);
	bEventHasATrackInRange = kTRUE;
      }
      
    }
  } // isMonteCarlo
  if(bEventHasATrack) { hEventStatistics->Fill("MC events with tracks",1); }
  if(bEventHasATrackInRange) { hEventStatistics->Fill("MC events with tracks in range",1); }
  bEventHasATrack = kFALSE;
  bEventHasATrackInRange = kFALSE;
  
  
  
  // Loop over recontructed tracks
  
  dEventZv = eventAOD->GetPrimaryVertex()->GetZ();
  if( TMath::Abs(dEventZv) > dCutMaxZVertex ) return;
  
  hAllEventStatisticsCentrality->Fill(dCentrality/*, nTriggerFired*/);
  
  hEventStatistics->Fill("after Zv cut",1);
  
  dTrackZvPtEtaCent[0] = dEventZv;
  
  for(Int_t itrack = 0; itrack < eventAOD->GetNumberOfTracks(); itrack++)
  {
    track = eventAOD->GetTrack(itrack);
    if(!track) continue;
    
    mcPart = NULL;
    dMCTrackZvPtEtaCent[1] = 0;
    dMCTrackZvPtEtaCent[2] = 0;
    dMCTrackZvPtEtaCent[3] = 0;
    
    bIsPrimary = kFALSE;
    
    if( !(IsTrackAccepted(track)) ) continue;
    
    if( bIsMonteCarlo )
    {
      mcPart = (AliAODMCParticle*)stack->At(TMath::Abs(track->GetLabel()));
      if( !mcPart ) { continue; }
      
      // check for charge
      if( !(IsMCTrackAccepted(mcPart)) ) {  continue; } 
      
      bIsHijingParticle = IsHijingParticle(mcPart, genHijingHeader);
      //       bIsPythiaParticle = IsPythiaParticle(mcPart, genPythiaHeader);
      
      //       if(!bIsHijingParticle) continue; // only take real tracks, not injected ones
      
      bIsPrimary = mcPart->IsPhysicalPrimary();
      
      dMCTrackZvPtEtaCent[1] = mcPart->Pt();
      dMCTrackZvPtEtaCent[2] = mcPart->Eta();
      dMCTrackZvPtEtaCent[3] = dCentrality;
      
      if(bIsPrimary && bIsHijingParticle)
      {
	hnMCRecPrimZvPtEtaCent->Fill(dMCTrackZvPtEtaCent);
      }
      
      if(!bIsPrimary /*&& !bIsHijingParticle*/)
      {
	Int_t indexMoth = mcPart->GetMother(); 
	if(indexMoth >= 0)
	{
	  AliAODMCParticle* moth = (AliAODMCParticle*)stack->At(indexMoth);
	  Bool_t bMotherIsHijingParticle = IsHijingParticle(moth, genHijingHeader);
	  
	  if(bMotherIsHijingParticle) // only store secondaries, which come from a not embedded signal!
	  {
	    hMCTrackStatusCode->Fill(Form("%d",mcPart->GetStatus()), 1);
	    if(TMath::Abs(mcPart->Eta()) < 0.8) { hMCPdgPt->Fill(mcPart->Pt(), Form("%s",GetParticleName(mcPart->GetPdgCode())), 1); }
	    
	    hnMCSecZvPtEtaCent->Fill(dMCTrackZvPtEtaCent);
	    hMCTrackPdgCode->Fill(Form("%s_H%i_H%i",GetParticleName(moth->GetPdgCode()),bMotherIsHijingParticle, bIsHijingParticle), 1);
	    // 	  delete moth;
	  }
	} 	
      }
    } // end isMonteCarlo 
    
    // ======================== fill histograms ========================
    
//     if(bIsMonteCarlo && !bIsHijingParticle) 
//     { 
//       continue;  //only store reco tracks, which do not come from embedded signal
//     }
    
    dTrackZvPtEtaCent[1] = track->Pt();
    dTrackZvPtEtaCent[2] = track->Eta();
    dTrackZvPtEtaCent[3] = dCentrality;
    
    bEventHasATrack = kTRUE;
    
    hnZvPtEtaCent->Fill(dTrackZvPtEtaCent);
    
    if( (dTrackZvPtEtaCent[1] > dCutPtMin) &&
      (dTrackZvPtEtaCent[1] < dCutPtMax) &&
      (dTrackZvPtEtaCent[2] > dCutEtaMin) &&
      (dTrackZvPtEtaCent[2] < dCutEtaMax) )
    {
      iAcceptedMultiplicity++;
      bEventHasATrackInRange = kTRUE;
      hPt->Fill(track->Pt());
      hCharge->Fill(track->Charge());
    }
  } // end track loop
  
  if(bEventHasATrack) { hEventStatistics->Fill("events with tracks",1); bEventHasATrack = kFALSE; }
  
  if(bEventHasATrackInRange) 
  { 
    hEventStatistics->Fill("events with tracks in range",1); 
    hEventStatisticsCentrality->Fill(dCentrality); 
    bEventHasATrackInRange = kFALSE; 
  }
  
  Double_t dEventZvMultCent[3] = {dEventZv, iAcceptedMultiplicity, dCentrality};
  hnZvMultCent->Fill(dEventZvMultCent);
  
  
  
  PostData(1, fOutputList);
  
}

Bool_t AlidNdPtAnalysisPbPbAOD::IsTrackAccepted(AliAODTrack *tr)
{
  if(!tr) return kFALSE;
  
  if(tr->Charge()==0) { return kFALSE; }
  
  if(!(tr->TestFilterBit(AliAODTrack::kTrkGlobal)) ) { return kFALSE; }
  
  
  
  Double_t dNClustersTPC = tr->GetTPCNcls();
  Double_t dCrossedRowsTPC = tr->GetTPCClusterInfo(2,1);
  
  hAccNclsTPC->Fill(dNClustersTPC);
  hAccCrossedRowsTPC->Fill(dCrossedRowsTPC);
  //   Double_t dFindableClustersTPC = tr->GetTPCNclsF();
  //   Double_t dChi2PerClusterTPC = (dNClustersTPC>0)?tr->Chi2perNDF()*(dNClustersTPC-5)/dNClustersTPC:-1.; // see AliDielectronVarManager.h
  //   
  //   Bool_t bIsFromKink = kFALSE;
  //   if(tr->GetProdVertex()->GetType() == AliAODVertex::kKink) bIsFromKink = kTRUE;
  //   
  //   // from AliAnalysisTaskPIDqa.cxx
  //   ULong_t uStatus = tr->GetStatus();
  //   Bool_t bHasRefitTPC = kFALSE;
  //   Bool_t bHasRefitITS = kFALSE;
  //   
  //   if ((uStatus & AliVTrack::kTPCrefit) == AliVTrack::kTPCrefit) bHasRefitTPC = kTRUE;
  //   if ((uStatus & AliVTrack::kITSrefit) == AliVTrack::kITSrefit) bHasRefitITS = kTRUE;
  //   
  //   // from AliDielectronVarManager.h
  //   Bool_t bHasHitInSPD = kFALSE;
  //   for (Int_t iC=0; iC<2; iC++) 
  //   {
    //     if (((tr->GetITSClusterMap()) & (1<<(iC))) > 0) {  bHasHitInSPD = kTRUE;  }
    //   }
    //   
    //   Double_t dNClustersITS = tr->GetITSNcls();
    
    // cuts to be done:
    // TPC  
    //   esdTrackCuts->SetMinNCrossedRowsTPC(70);
    //   esdTrackCuts->SetMinRatioCrossedRowsOverFindableClustersTPC(0.8);
    //   
    //   esdTrackCuts->SetMaxChi2PerClusterTPC(4);
    //   esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
    //   esdTrackCuts->SetRequireTPCRefit(kTRUE);
    // ITS
    //   esdTrackCuts->SetRequireITSRefit(kTRUE);
    //   esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,	 AliESDtrackCuts::kAny);
    //   
    //   esdTrackCuts->SetMaxDCAToVertexXYPtDep("0.0105+0.0350/pt^1.1");
    //   esdTrackCuts->SetMaxChi2TPCConstrainedGlobal(36);
    //   
    //   esdTrackCuts->SetMaxDCAToVertexZ(2);
    //   esdTrackCuts->SetDCAToVertex2D(kFALSE);
    //   esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
    //   
    //   esdTrackCuts->SetMaxChi2PerClusterITS(36);
    
    
    return kTRUE;
}

Bool_t AlidNdPtAnalysisPbPbAOD::IsMCTrackAccepted(AliAODMCParticle *part)
{
  if(!part) return kFALSE;
  
  Double_t charge = part->Charge()/3.;
  if (TMath::Abs(charge) < 0.001) return kFALSE;
  
  return kTRUE;
}

const char * AlidNdPtAnalysisPbPbAOD::GetParticleName(Int_t pdg) 
{
  TParticlePDG * p1 = TDatabasePDG::Instance()->GetParticle(pdg);
  if(p1) return p1->GetName();
  return Form("%d", pdg);
}

AliGenHijingEventHeader* AlidNdPtAnalysisPbPbAOD::GetHijingEventHeader(AliAODMCHeader *header)
{
  //
  // inspired by PWGJE/AliPWG4HighPtSpectra.cxx
  //
  
  if(!header) return 0x0;
  AliGenHijingEventHeader* hijingGenHeader = NULL;
  
  TList* headerList = header->GetCocktailHeaders();
  
  for(Int_t i = 0; i < headerList->GetEntries(); i++)
  {
    hijingGenHeader = dynamic_cast<AliGenHijingEventHeader*>(headerList->At(i));
    if(hijingGenHeader) break;
  }
  
  if(!hijingGenHeader) return 0x0;
  
  return hijingGenHeader;
}

AliGenPythiaEventHeader* AlidNdPtAnalysisPbPbAOD::GetPythiaEventHeader(AliAODMCHeader *header)
{
  //
  // inspired by PWGJE/AliPWG4HighPtSpectra.cxx
  //
  
  if(!header) return 0x0;
  AliGenPythiaEventHeader* PythiaGenHeader = NULL;
  
  TList* headerList = header->GetCocktailHeaders();
  
  for(Int_t i = 0; i < headerList->GetEntries(); i++)
  {
    PythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(headerList->At(i));
    if(PythiaGenHeader) break;
  }
  
  if(!PythiaGenHeader) return 0x0;
  
  return PythiaGenHeader;
}

//________________________________________________________________________
Bool_t AlidNdPtAnalysisPbPbAOD::IsHijingParticle(const AliAODMCParticle *part, AliGenHijingEventHeader* hijingGenHeader){
  
  // Check whether a particle is from Hijing or some injected 
  
  if(part->Label() > (hijingGenHeader->NProduced()-1)) return kFALSE;
  return kTRUE;
}

//________________________________________________________________________
Bool_t AlidNdPtAnalysisPbPbAOD::IsPythiaParticle(const AliAODMCParticle *part, AliGenPythiaEventHeader* pythiaGenHeader){
  
  // Check whether a particle is from Pythia or some injected 
  
  if(part->Label() > (pythiaGenHeader->NProduced()-1)) return kFALSE;
  return kTRUE;
}

Bool_t AlidNdPtAnalysisPbPbAOD::IsMCSecondary(AliAODMCParticle *part, TClonesArray *arrayMC)
{
  //
  // from AliAnalysisTaskSpectraAOD.cxx
  //
  
  if(!part) return kFALSE; 
  
  if( part->IsPhysicalPrimary() ) return kFALSE;
  
  Bool_t isSecondaryMaterial = kFALSE; 
  Bool_t isSecondaryWeak     = kFALSE; 
  Int_t mfl = -999;
  Int_t codemoth = -999;
  Int_t indexMoth = part->GetMother(); // FIXME ignore fakes? TO BE CHECKED, on ESD is GetFirstMother()
  if(indexMoth >= 0)
  {
    AliAODMCParticle* moth = (AliAODMCParticle*) arrayMC->At(indexMoth);
    codemoth = TMath::Abs(moth->GetPdgCode());
    mfl = Int_t (codemoth/ TMath::Power(10, Int_t(TMath::Log10(codemoth))));
  } 
  // add if(partMC->GetStatus() & kPDecay)? FIXME
  if(mfl==3) isSecondaryWeak     = kTRUE;     
  else       isSecondaryMaterial = kTRUE; 
  
  if( isSecondaryMaterial || isSecondaryWeak ) return kTRUE;
  
  return kFALSE;
}



void AlidNdPtAnalysisPbPbAOD::Terminate(Option_t *)
{
  
}

Double_t* AlidNdPtAnalysisPbPbAOD::GetArrayClone(Int_t n, Double_t* source)
{
  if (!source || n==0) return 0;
  Double_t* dest = new Double_t[n];
  for (Int_t i=0; i<n ; i++) { dest[i] = source[i]; }
  return dest;
}