[u/mrichter/AliRoot.git] / PWGCF / EBYE / PIDFluctuation / task / AliEbyEParticleRatioFluctuationTask.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: Satyajit Jena.                                                 *
 * 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.                  *
 **************************************************************************/


//=========================================================================//
//             AliEbyE Analysis for Particle Ratio Fluctuation             //
//                   Deepika Rathee  | Satyajit Jena                       //
//                   drathee@cern.ch | sjena@cern.ch                       //
//
//             V0.1 2013/03/25 Using THnSparse
//             V0.2 2013/04/03 Cleanup
//             V1.0 2013/04/10 Cleanup Bins for Less Memory
//             V1.1 2013/04/15 Bins Added 
//   Todo: pp and pA, Mix Events
//=========================================================================//

#include "TChain.h"
#include "TList.h"
#include "TFile.h"
#include "TTree.h"
#include "TH1D.h"
#include "TH2F.h"
#include "TH3F.h"
#include "TCanvas.h"
#include "AliAnalysisTask.h"
#include "AliAnalysisManager.h"
#include "AliVEvent.h"
#include "AliESD.h"
#include "AliESDEvent.h"
#include "AliAODEvent.h"
#include "AliAODMCParticle.h"
#include "AliAODMCHeader.h"
#include "AliPIDResponse.h"
#include "AliAODHeader.h"
#include "AliAODpidUtil.h"
#include "AliHelperPID.h"
using std::endl;
using std::cout;
#include "AliEbyEParticleRatioFluctuationTask.h"

ClassImp(AliEbyEParticleRatioFluctuationTask)

//-----------------------------------------------------------------------
AliEbyEParticleRatioFluctuationTask::AliEbyEParticleRatioFluctuationTask( const char *name ) : AliAnalysisTaskSE( name ), 
  fThnList(NULL), 
  fAnalysisType("AOD"), 
  fAnalysisData("PbPb"), 
  fCentralityEstimator("V0M"), 
  fVxMax(3.), 
  fVyMax(3.), 
  fVzMax(10.), 
  fDCAxy(2.4),  
  fDCAz(3.2), 
  fPtLowerLimit(0.2), 
  fPtHigherLimit(5.), 
  fEtaLowerLimit(-1.), 
  fEtaHigherLimit(1.), 
  fTPCNClus(80),
  fAODtrackCutBit(128), 
  isQA(kFALSE), 
  fDebug(kFALSE), 
  fHelperPID(0x0),
  fEventCounter(NULL), 
  fHistoCorrelation(NULL) { 
  for(Int_t i = 0; i < 14; i++ ) fHistQA[i] = NULL;
  DefineOutput(1, TList::Class()); //! Connect Outpput....
}

AliEbyEParticleRatioFluctuationTask::~AliEbyEParticleRatioFluctuationTask() {
  //!   Cleaning up
  if (fThnList)   delete fThnList;
  if (fHelperPID) delete fHelperPID;
}

//---------------------------------------------------------------------------------
void AliEbyEParticleRatioFluctuationTask::UserCreateOutputObjects() {
  fThnList = new TList();
  fThnList->SetOwner(kTRUE);

  fEventCounter = new TH1D("fEventCounter","EventCounter", 300, 0.5,300.5);
  if (isQA) fThnList->Add(fEventCounter);
  
  fHistQA[0] = new TH2F("fHistQAvx", "Histo Vx Selected;Centrality;Vx", 100,0,100, 5000, -5., 5.);
  fHistQA[1] = new TH2F("fHistQAvy", "Histo Vy Selected;Centrality;Vy", 100,0,100, 5000, -5., 5.);
  fHistQA[2] = new TH2F("fHistQAvz", "Histo Vz Selected;Centrality;Vz", 100,0,100, 5000, -25., 25.);  
  fHistQA[3] = new TH2F("fHistQAvxA", "Histo Vx;Centrality;Vx", 100,0,100, 5000, -5., 5.);
  fHistQA[4] = new TH2F("fHistQAvyA", "Histo Vy;Centrality;Vy", 100,0,100, 5000, -5., 5.);
  fHistQA[5] = new TH2F("fHistQAvzA", "Histo Vz;Centrality;Vz", 100,0,100, 5000, -25., 25.);

  fHistQA[6] = new TH2F("fHistQADcaXyA", "Histo DCAxy;Centrality;DCAxy",100,0,100, 600, -15., 15.);
  fHistQA[7] = new TH2F("fHistQADcaZA", "Histo DCAz;Centrality;DCAz ",100,0,100, 600, -15., 15.);   
  fHistQA[8] = new TH2F("fHistQAPtA","p_{T} distribution;Centrality;p_{T}",100,0,100,1000,0,10);
  fHistQA[9] = new TH2F("fHistQAEtaA","#eta distribution;Centrality;#eta",100,0,100,240,-1.2,1.2);

  fHistQA[10] = new TH2F("fHistQADcaXy", "Histo DCAxy after Selected;Centrality;DCAxy", 100,0,100,600, -15., 15.);
  fHistQA[11] = new TH2F("fHistQADcaZ", "Histo DCAz Selected;Centrality;DCAz", 100,0,100,600, -15., 15.);   
  fHistQA[12] = new TH2F("fHistQAPt","p_{T} distribution Selected;Centrality;p_{T}",100,0,100,1000,0,10);
  fHistQA[13] = new TH2F("fHistQAEta","#eta distribution Selected;Centrality;#eta",100,0,100, 240,-1.2,1.2);
  
  if (isQA) for(Int_t i = 0; i < 14; i++) fThnList->Add(fHistQA[i]);
  
  Int_t fgSparseDataBins[kNSparseData]   = {100, 5000, 5000, 2500, 2500, 3000, 1500, 1500, 1000, 500, 500, 500, 250, 250};
  Double_t fgSparseDataMin[kNSparseData] = {0.,  0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,   0.,  0.,  0.,  0.,  0.};
  Double_t fgSparseDataMax[kNSparseData] = {100.,5000.,5000.,2500.,2500.,3000.,1500.,1500.,1000.,500.,500.,500.,250.,250.};
  
  const Char_t *fgkSparseDataTitle[] = {"centrality","RefMult","N_{ch}", "N_{+}","N_{-}","N_{#pi}", "N_{#pi^{+}}","N_{#pi^{-}}","N_{K}","N_{K^{+}}", "N_{K^{-}}","N_{p}","N_{p}","N_{#bar{p}}"};
    
  fHistoCorrelation = new THnSparseI("fThnCorr", "", kNSparseData, fgSparseDataBins, fgSparseDataMin, fgSparseDataMax);
  for (Int_t iaxis = 0; iaxis < kNSparseData; iaxis++)
    fHistoCorrelation->GetAxis(iaxis)->SetTitle(fgkSparseDataTitle[iaxis]);
  
  if(!isQA) fThnList->Add(fHistoCorrelation);
  if(isQA)  
    if (fHelperPID)
      fThnList->Add(fHelperPID);
  
  PostData(1, fThnList);
}

//----------------------------------------------------------------------------------
void AliEbyEParticleRatioFluctuationTask::UserExec( Option_t * ){

  if (isQA) fEventCounter->Fill(1);

  AliAODEvent* event = dynamic_cast<AliAODEvent*>(InputEvent());
  if (!event) {
    Printf("ERROR 01: AOD not found ");
    return;
  }

  Int_t gCent   = -1;
  Float_t gRefMul = -1;
  
  AliAODHeader *aodHeader = dynamic_cast<AliAODHeader*>(event->GetHeader());
  if(!aodHeader) AliFatal("Not a standard AOD");
  gCent = (Int_t)aodHeader->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data()); // V0M`
  gRefMul = aodHeader->GetRefMultiplicity();
  if (gCent < 0 || gCent > 100) return;
  if (isQA) fEventCounter->Fill(2);  

  if (!AcceptEvent(event,gCent)) return;
  
  Int_t icharge = -1;
  Int_t gCharge[2];
  Int_t gPid[3][2];
  
  for (Int_t i = 0; i < 2; i++) {
    gCharge[i] = 0;
    for (Int_t ii = 0; ii < 3; ii++) {
      gPid[ii][i] = 0;
    }
  }


  if(fAnalysisType == "AOD") {
    if (isQA) {
	fEventCounter->Fill(5);
	fEventCounter->Fill(50+gCent);
    }
    
    for (Int_t itrk = 0; itrk < event->GetNumberOfTracks(); itrk++) {
      AliAODTrack* track = dynamic_cast<AliAODTrack *>(event->GetTrack(itrk));
      if (!track) continue;
      
      if (!AcceptTrack(track, gCent)) continue;
            
      Int_t a = fHelperPID->GetParticleSpecies((AliVTrack*) track,kTRUE);

      if(a < 0 || a > 2) continue;
      icharge = track->Charge() > 0 ? 0 : 1;
      gCharge[icharge]++;
      gPid[a][icharge]++;
    }
  }
  else if(fAnalysisType == "MCAOD") {
    TClonesArray *arrayMC= 0; 
    arrayMC = dynamic_cast<TClonesArray*> (event->GetList()->FindObject(AliAODMCParticle::StdBranchName()));
    if (!arrayMC) {
      Printf("Error: MC particles branch not found!\n");
      return;
    }
    AliAODMCHeader *mcHdr=0;
    mcHdr=(AliAODMCHeader*)event->GetList()->FindObject(AliAODMCHeader::StdBranchName());  
    if(!mcHdr) {
      Printf("MC header branch not found!\n");
      return;
    }
    
    if (isQA) {
      fEventCounter->Fill(5);
      fEventCounter->Fill(50+gCent);
    }
    
    Int_t nMC = arrayMC->GetEntries();
    for (Int_t iMC = 0; iMC < nMC; iMC++) {
      AliAODMCParticle *partMC = (AliAODMCParticle*) arrayMC->At(iMC);
      if(!AcceptMCTrack(partMC, gCent)) continue;
      Int_t a = fHelperPID->GetMCParticleSpecie((AliVEvent*) event,(AliVTrack*)partMC,1);
      if(a < 0 || a > 2) continue;
      icharge = partMC->Charge() > 0 ? 0 : 1;
      gCharge[icharge]++;
      gPid[a][icharge]++;
    }
  }
  else {
    printf(" No Event Type is Defined ");
    return;
  }
  
  if( (gCharge[0] + gCharge[1]) != 0 ) {
    if (isQA) {
      fEventCounter->Fill(6); 
      fEventCounter->Fill(160 + gCent);
    }
    else { 
      Double_t vsparse[kNSparseData];
      vsparse[0]   = gCent;
      vsparse[1]   = gRefMul;
      vsparse[2]   = gCharge[0] + gCharge[1];
      vsparse[3]   = gCharge[0];
      vsparse[4]   = gCharge[1];
      vsparse[5]   = gPid[0][0] + gPid[0][1];
      vsparse[6]   = gPid[0][0];
      vsparse[7]   = gPid[0][1];
      vsparse[8]   = gPid[1][0] + gPid[1][0];
      vsparse[9]   = gPid[1][0];
      vsparse[10]  = gPid[1][1];
      vsparse[11]  = gPid[2][0] + gPid[2][1];
      vsparse[12]  = gPid[2][0];
      vsparse[13]  = gPid[2][1];
      fHistoCorrelation->Fill(vsparse);
    }
  }
  
  if(fDebug && isQA)  Printf(" %6d  %6d %6d  %6d %6d  %6d %6d %6d %6d %6d %6d %6d %6d", 
			     (Int_t)fEventCounter->GetBinContent(1), 
			     (Int_t)fEventCounter->GetBinContent(2), 
			     (Int_t)fEventCounter->GetBinContent(3), 
			     (Int_t)gCent, (Int_t)gRefMul, 
			     gCharge[0], gCharge[1], 
			     gPid[0][0], gPid[0][1],  gPid[1][0], 
			     gPid[1][1],  gPid[2][0], gPid[2][1]);
  
  PostData(1, fThnList);
}

void AliEbyEParticleRatioFluctuationTask::Terminate( Option_t * ){
  Info("AliEbyEParticleRatioFluctuationTask"," Task Successfully finished");
}

//___________________________________________________________
Bool_t AliEbyEParticleRatioFluctuationTask::AcceptEvent(AliAODEvent *event, Int_t cent) const {
  Bool_t ver = kFALSE;
  const AliAODVertex *vertex = event->GetPrimaryVertex();
  if(vertex) {
    Double32_t fCov[6];
    vertex->GetCovarianceMatrix(fCov);
    if(vertex->GetNContributors() > 0) {
      if(fCov[5] != 0) {
	
	if(isQA) {	
	  fEventCounter->Fill(3);
	  fHistQA[3]->Fill(cent,vertex->GetX());
	  fHistQA[4]->Fill(cent,vertex->GetY());
	  fHistQA[5]->Fill(cent,vertex->GetZ());
	}
	
	if(TMath::Abs(vertex->GetX()) < fVxMax) {
	  if(TMath::Abs(vertex->GetY()) < fVyMax) {
	    if(TMath::Abs(vertex->GetZ()) < fVzMax) {
	      ver = kTRUE;
	      if(isQA) {	
		fEventCounter->Fill(4);
		fHistQA[0]->Fill(cent,vertex->GetX());
		fHistQA[1]->Fill(cent,vertex->GetY());
		fHistQA[2]->Fill(cent,vertex->GetZ());
	      }
	    }
	  }
	}
      }
    }
  }
  
  AliCentrality *centrality = event->GetCentrality();
  if (centrality->GetQuality() != 0) ver = kFALSE;
  return ver;
}


//___________________________________________________________
Bool_t AliEbyEParticleRatioFluctuationTask::AcceptTrack(AliAODTrack *track, Int_t cent) const {
  if(!track)                                  return kFALSE;
  if (track->Charge() == 0 )                  return kFALSE;
  
  if(isQA) {
    fHistQA[6]->Fill(cent,track->DCA());
    fHistQA[7]->Fill(cent,track->ZAtDCA());
    fHistQA[8]->Fill(cent,track->Pt());
    fHistQA[9]->Fill(cent,track->Eta());
  }
  
  if (!track->TestFilterBit(fAODtrackCutBit)) return kFALSE;

  if (track->Eta() < fEtaLowerLimit ||
      track->Eta() > fEtaHigherLimit)         return kFALSE;
  if (track->Pt() < fPtLowerLimit ||
      track->Pt() > fPtHigherLimit)           return kFALSE;  
  if ( track->DCA() > fDCAxy )                return kFALSE; 
  if ( track->ZAtDCA() > fDCAz )              return kFALSE;
   
  if(isQA) {
    fHistQA[10]->Fill(cent,track->DCA());
    fHistQA[11]->Fill(cent,track->ZAtDCA());
    fHistQA[12]->Fill(cent,track->Pt());
    fHistQA[13]->Fill(cent,track->Eta());
  }
 
  return kTRUE;
}


//___________________________________________________________
Bool_t AliEbyEParticleRatioFluctuationTask::AcceptMCTrack(AliAODMCParticle *track, Int_t cent) const {
  if(!track) return kFALSE;
  if(!track->IsPhysicalPrimary()) return kFALSE;
  if (track->Charge() == 0 ) return kFALSE;
  if(isQA) {
    fHistQA[8]->Fill(cent,track->Pt());
    fHistQA[9]->Fill(cent,track->Eta());
  }

  if (track->Eta() < fEtaLowerLimit ||
      track->Eta() > fEtaHigherLimit) return kFALSE;
  if (track->Pt() < fPtLowerLimit ||
      track->Pt() > fPtHigherLimit) return kFALSE;  
    
  if(isQA) {
    fHistQA[12]->Fill(cent,track->Pt());
    fHistQA[13]->Fill(cent,track->Eta());
  }
 
  return kTRUE;
}
Commit	Line	Data
c683985a	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: Satyajit Jena. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16
	17	//=========================================================================//
	18	// AliEbyE Analysis for Particle Ratio Fluctuation //
	19	// Deepika Rathee \| Satyajit Jena //
	20	// drathee@cern.ch \| sjena@cern.ch //
	21	//
	22	// V0.1 2013/03/25 Using THnSparse
	23	// V0.2 2013/04/03 Cleanup
	24	// V1.0 2013/04/10 Cleanup Bins for Less Memory
	25	// V1.1 2013/04/15 Bins Added
	26	// Todo: pp and pA, Mix Events
	27	//=========================================================================//
	28
	29	#include "TChain.h"
	30	#include "TList.h"
	31	#include "TFile.h"
	32	#include "TTree.h"
	33	#include "TH1D.h"
	34	#include "TH2F.h"
	35	#include "TH3F.h"
	36	#include "TCanvas.h"
	37	#include "AliAnalysisTask.h"
	38	#include "AliAnalysisManager.h"
	39	#include "AliVEvent.h"
	40	#include "AliESD.h"
	41	#include "AliESDEvent.h"
	42	#include "AliAODEvent.h"
	43	#include "AliAODMCParticle.h"
	44	#include "AliAODMCHeader.h"
	45	#include "AliPIDResponse.h"
	46	#include "AliAODHeader.h"
	47	#include "AliAODpidUtil.h"
	48	#include "AliHelperPID.h"
	49	using std::endl;
	50	using std::cout;
	51	#include "AliEbyEParticleRatioFluctuationTask.h"
	52
	53	ClassImp(AliEbyEParticleRatioFluctuationTask)
	54
	55	//-----------------------------------------------------------------------
	56	AliEbyEParticleRatioFluctuationTask::AliEbyEParticleRatioFluctuationTask( const char *name ) : AliAnalysisTaskSE( name ),
	57	fThnList(NULL),
	58	fAnalysisType("AOD"),
	59	fAnalysisData("PbPb"),
	60	fCentralityEstimator("V0M"),
	61	fVxMax(3.),
	62	fVyMax(3.),
	63	fVzMax(10.),
	64	fDCAxy(2.4),
65	fDCAz(3.2),
66	fPtLowerLimit(0.2),
67	fPtHigherLimit(5.),
68	fEtaLowerLimit(-1.),
69	fEtaHigherLimit(1.),
70	fTPCNClus(80),
71	fAODtrackCutBit(128),
72	isQA(kFALSE),
73	fDebug(kFALSE),
74	fHelperPID(0x0),
75	fEventCounter(NULL),
76	fHistoCorrelation(NULL) {
77	for(Int_t i = 0; i < 14; i++ ) fHistQA[i] = NULL;
78	DefineOutput(1, TList::Class()); //! Connect Outpput....
79	}
80
81	AliEbyEParticleRatioFluctuationTask::~AliEbyEParticleRatioFluctuationTask() {
82	//! Cleaning up
83	if (fThnList) delete fThnList;
84	if (fHelperPID) delete fHelperPID;
85	}
86
87	//---------------------------------------------------------------------------------
88	void AliEbyEParticleRatioFluctuationTask::UserCreateOutputObjects() {
89	fThnList = new TList();
90	fThnList->SetOwner(kTRUE);
91
92	fEventCounter = new TH1D("fEventCounter","EventCounter", 300, 0.5,300.5);
93	if (isQA) fThnList->Add(fEventCounter);
94
95	fHistQA[0] = new TH2F("fHistQAvx", "Histo Vx Selected;Centrality;Vx", 100,0,100, 5000, -5., 5.);
96	fHistQA[1] = new TH2F("fHistQAvy", "Histo Vy Selected;Centrality;Vy", 100,0,100, 5000, -5., 5.);
97	fHistQA[2] = new TH2F("fHistQAvz", "Histo Vz Selected;Centrality;Vz", 100,0,100, 5000, -25., 25.);
98	fHistQA[3] = new TH2F("fHistQAvxA", "Histo Vx;Centrality;Vx", 100,0,100, 5000, -5., 5.);
99	fHistQA[4] = new TH2F("fHistQAvyA", "Histo Vy;Centrality;Vy", 100,0,100, 5000, -5., 5.);
100	fHistQA[5] = new TH2F("fHistQAvzA", "Histo Vz;Centrality;Vz", 100,0,100, 5000, -25., 25.);
101
102	fHistQA[6] = new TH2F("fHistQADcaXyA", "Histo DCAxy;Centrality;DCAxy",100,0,100, 600, -15., 15.);
103	fHistQA[7] = new TH2F("fHistQADcaZA", "Histo DCAz;Centrality;DCAz ",100,0,100, 600, -15., 15.);
104	fHistQA[8] = new TH2F("fHistQAPtA","p_{T} distribution;Centrality;p_{T}",100,0,100,1000,0,10);
105	fHistQA[9] = new TH2F("fHistQAEtaA","#eta distribution;Centrality;#eta",100,0,100,240,-1.2,1.2);
106
107	fHistQA[10] = new TH2F("fHistQADcaXy", "Histo DCAxy after Selected;Centrality;DCAxy", 100,0,100,600, -15., 15.);
108	fHistQA[11] = new TH2F("fHistQADcaZ", "Histo DCAz Selected;Centrality;DCAz", 100,0,100,600, -15., 15.);
109	fHistQA[12] = new TH2F("fHistQAPt","p_{T} distribution Selected;Centrality;p_{T}",100,0,100,1000,0,10);
110	fHistQA[13] = new TH2F("fHistQAEta","#eta distribution Selected;Centrality;#eta",100,0,100, 240,-1.2,1.2);
111
112	if (isQA) for(Int_t i = 0; i < 14; i++) fThnList->Add(fHistQA[i]);
113
114	Int_t fgSparseDataBins[kNSparseData] = {100, 5000, 5000, 2500, 2500, 3000, 1500, 1500, 1000, 500, 500, 500, 250, 250};
115	Double_t fgSparseDataMin[kNSparseData] = {0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.};
116	Double_t fgSparseDataMax[kNSparseData] = {100.,5000.,5000.,2500.,2500.,3000.,1500.,1500.,1000.,500.,500.,500.,250.,250.};
117
118	const Char_t *fgkSparseDataTitle[] = {"centrality","RefMult","N_{ch}", "N_{+}","N_{-}","N_{#pi}", "N_{#pi^{+}}","N_{#pi^{-}}","N_{K}","N_{K^{+}}", "N_{K^{-}}","N_{p}","N_{p}","N_{#bar{p}}"};
119
120	fHistoCorrelation = new THnSparseI("fThnCorr", "", kNSparseData, fgSparseDataBins, fgSparseDataMin, fgSparseDataMax);
121	for (Int_t iaxis = 0; iaxis < kNSparseData; iaxis++)
122	fHistoCorrelation->GetAxis(iaxis)->SetTitle(fgkSparseDataTitle[iaxis]);
123
124	if(!isQA) fThnList->Add(fHistoCorrelation);
125	if(isQA)
126	if (fHelperPID)
127	fThnList->Add(fHelperPID);
128
129	PostData(1, fThnList);
130	}
131
132	//----------------------------------------------------------------------------------
133	void AliEbyEParticleRatioFluctuationTask::UserExec( Option_t * ){
134
135	if (isQA) fEventCounter->Fill(1);
136
137	AliAODEvent* event = dynamic_cast<AliAODEvent*>(InputEvent());
138	if (!event) {
139	Printf("ERROR 01: AOD not found ");
140	return;
141	}
142
143	Int_t gCent = -1;
144	Float_t gRefMul = -1;
145
0a918d8d	146	AliAODHeader aodHeader = dynamic_cast<AliAODHeader>(event->GetHeader());
0a918d8d	147	if(!aodHeader) AliFatal("Not a standard AOD");
ce85dc2c	148	gCent = (Int_t)aodHeader->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data()); // V0M`
c683985a	149	gRefMul = aodHeader->GetRefMultiplicity();
	150	if (gCent < 0 \|\| gCent > 100) return;
	151	if (isQA) fEventCounter->Fill(2);
	152
	153	if (!AcceptEvent(event,gCent)) return;
	154
	155	Int_t icharge = -1;
	156	Int_t gCharge[2];
	157	Int_t gPid[3][2];
	158
	159	for (Int_t i = 0; i < 2; i++) {
	160	gCharge[i] = 0;
	161	for (Int_t ii = 0; ii < 3; ii++) {
	162	gPid[ii][i] = 0;
	163	}
	164	}
	165
	166
	167	if(fAnalysisType == "AOD") {
	168	if (isQA) {
	169	fEventCounter->Fill(5);
	170	fEventCounter->Fill(50+gCent);
	171	}
	172
	173	for (Int_t itrk = 0; itrk < event->GetNumberOfTracks(); itrk++) {
	174	AliAODTrack* track = dynamic_cast<AliAODTrack *>(event->GetTrack(itrk));
	175	if (!track) continue;
	176
	177	if (!AcceptTrack(track, gCent)) continue;
	178
	179	Int_t a = fHelperPID->GetParticleSpecies((AliVTrack*) track,kTRUE);
	180
	181	if(a < 0 \|\| a > 2) continue;
	182	icharge = track->Charge() > 0 ? 0 : 1;
	183	gCharge[icharge]++;
	184	gPid[a][icharge]++;
	185	}
	186	}
	187	else if(fAnalysisType == "MCAOD") {
	188	TClonesArray *arrayMC= 0;
	189	arrayMC = dynamic_cast<TClonesArray*> (event->GetList()->FindObject(AliAODMCParticle::StdBranchName()));
	190	if (!arrayMC) {
	191	Printf("Error: MC particles branch not found!\n");
	192	return;
	193	}
	194	AliAODMCHeader *mcHdr=0;
	195	mcHdr=(AliAODMCHeader*)event->GetList()->FindObject(AliAODMCHeader::StdBranchName());
	196	if(!mcHdr) {
	197	Printf("MC header branch not found!\n");
	198	return;
	199	}
	200
	201	if (isQA) {
	202	fEventCounter->Fill(5);
	203	fEventCounter->Fill(50+gCent);
	204	}
	205
	206	Int_t nMC = arrayMC->GetEntries();
	207	for (Int_t iMC = 0; iMC < nMC; iMC++) {
	208	AliAODMCParticle partMC = (AliAODMCParticle) arrayMC->At(iMC);
	209	if(!AcceptMCTrack(partMC, gCent)) continue;
	210	Int_t a = fHelperPID->GetMCParticleSpecie((AliVEvent) event,(AliVTrack)partMC,1);
	211	if(a < 0 \|\| a > 2) continue;
	212	icharge = partMC->Charge() > 0 ? 0 : 1;
213	gCharge[icharge]++;
214	gPid[a][icharge]++;
215	}
216	}
217	else {
218	printf(" No Event Type is Defined ");
219	return;
220	}
221
222	if( (gCharge[0] + gCharge[1]) != 0 ) {
223	if (isQA) {
224	fEventCounter->Fill(6);
225	fEventCounter->Fill(160 + gCent);
226	}
227	else {
228	Double_t vsparse[kNSparseData];
229	vsparse[0] = gCent;
230	vsparse[1] = gRefMul;
231	vsparse[2] = gCharge[0] + gCharge[1];
232	vsparse[3] = gCharge[0];
233	vsparse[4] = gCharge[1];
234	vsparse[5] = gPid[0][0] + gPid[0][1];
235	vsparse[6] = gPid[0][0];
236	vsparse[7] = gPid[0][1];
237	vsparse[8] = gPid[1][0] + gPid[1][0];
238	vsparse[9] = gPid[1][0];
239	vsparse[10] = gPid[1][1];
240	vsparse[11] = gPid[2][0] + gPid[2][1];
241	vsparse[12] = gPid[2][0];
242	vsparse[13] = gPid[2][1];
243	fHistoCorrelation->Fill(vsparse);
244	}
245	}
246
247	if(fDebug && isQA) Printf(" %6d %6d %6d %6d %6d %6d %6d %6d %6d %6d %6d %6d %6d",
248	(Int_t)fEventCounter->GetBinContent(1),
249	(Int_t)fEventCounter->GetBinContent(2),
250	(Int_t)fEventCounter->GetBinContent(3),
251	(Int_t)gCent, (Int_t)gRefMul,
252	gCharge[0], gCharge[1],
253	gPid[0][0], gPid[0][1], gPid[1][0],
254	gPid[1][1], gPid[2][0], gPid[2][1]);
255
256	PostData(1, fThnList);
257	}
258
259	void AliEbyEParticleRatioFluctuationTask::Terminate( Option_t * ){
260	Info("AliEbyEParticleRatioFluctuationTask"," Task Successfully finished");
261	}
262
263	//___________________________________________________________
264	Bool_t AliEbyEParticleRatioFluctuationTask::AcceptEvent(AliAODEvent *event, Int_t cent) const {
265	Bool_t ver = kFALSE;
266	const AliAODVertex *vertex = event->GetPrimaryVertex();
267	if(vertex) {
268	Double32_t fCov[6];
269	vertex->GetCovarianceMatrix(fCov);
270	if(vertex->GetNContributors() > 0) {
271	if(fCov[5] != 0) {
272
273	if(isQA) {
274	fEventCounter->Fill(3);
275	fHistQA[3]->Fill(cent,vertex->GetX());
276	fHistQA[4]->Fill(cent,vertex->GetY());
277	fHistQA[5]->Fill(cent,vertex->GetZ());
278	}
279
280	if(TMath::Abs(vertex->GetX()) < fVxMax) {
281	if(TMath::Abs(vertex->GetY()) < fVyMax) {
282	if(TMath::Abs(vertex->GetZ()) < fVzMax) {
283	ver = kTRUE;
284	if(isQA) {
285	fEventCounter->Fill(4);
286	fHistQA[0]->Fill(cent,vertex->GetX());
287	fHistQA[1]->Fill(cent,vertex->GetY());
288	fHistQA[2]->Fill(cent,vertex->GetZ());
289	}
290	}
291	}
292	}
293	}
294	}
295	}
296
297	AliCentrality *centrality = event->GetCentrality();
298	if (centrality->GetQuality() != 0) ver = kFALSE;
299	return ver;
300	}
301
302
303	//___________________________________________________________
304	Bool_t AliEbyEParticleRatioFluctuationTask::AcceptTrack(AliAODTrack *track, Int_t cent) const {
305	if(!track) return kFALSE;
306	if (track->Charge() == 0 ) return kFALSE;
307
308	if(isQA) {
309	fHistQA[6]->Fill(cent,track->DCA());
310	fHistQA[7]->Fill(cent,track->ZAtDCA());
311	fHistQA[8]->Fill(cent,track->Pt());
312	fHistQA[9]->Fill(cent,track->Eta());
313	}
314
315	if (!track->TestFilterBit(fAODtrackCutBit)) return kFALSE;
316
317	if (track->Eta() < fEtaLowerLimit \|\|
318	track->Eta() > fEtaHigherLimit) return kFALSE;
319	if (track->Pt() < fPtLowerLimit \|\|
320	track->Pt() > fPtHigherLimit) return kFALSE;
321	if ( track->DCA() > fDCAxy ) return kFALSE;
322	if ( track->ZAtDCA() > fDCAz ) return kFALSE;
323
324	if(isQA) {
325	fHistQA[10]->Fill(cent,track->DCA());
326	fHistQA[11]->Fill(cent,track->ZAtDCA());
327	fHistQA[12]->Fill(cent,track->Pt());
328	fHistQA[13]->Fill(cent,track->Eta());
329	}
330
331	return kTRUE;
332	}
333
334
335	//___________________________________________________________
336	Bool_t AliEbyEParticleRatioFluctuationTask::AcceptMCTrack(AliAODMCParticle *track, Int_t cent) const {
337	if(!track) return kFALSE;
338	if(!track->IsPhysicalPrimary()) return kFALSE;
339	if (track->Charge() == 0 ) return kFALSE;
340	if(isQA) {
341	fHistQA[8]->Fill(cent,track->Pt());
342	fHistQA[9]->Fill(cent,track->Eta());
343	}
344
345	if (track->Eta() < fEtaLowerLimit \|\|
346	track->Eta() > fEtaHigherLimit) return kFALSE;
347	if (track->Pt() < fPtLowerLimit \|\|
348	track->Pt() > fPtHigherLimit) return kFALSE;
349
350	if(isQA) {
351	fHistQA[12]->Fill(cent,track->Pt());
352	fHistQA[13]->Fill(cent,track->Eta());
353	}
354
355	return kTRUE;
356	}