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

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: ALICE Offline.                                                 *
 * 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                       //
//          Dealing with Wide pT Addition (Test Only - not for use)
//=========================================================================//

#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 "AliPIDCombined.h"
#include "AliAODHeader.h"
#include "AliAODpidUtil.h"
#include "AliHelperPID.h"
using std::endl;
using std::cout;
#include "AliEbyEPidTTask.h"

ClassImp(AliEbyEPidTTask)

//-----------------------------------------------------------------------
AliEbyEPidTTask::AliEbyEPidTTask( const char *name ) : AliAnalysisTaskSE( name ), 
  fThnList(NULL), 
  isMC(kFALSE), 
  fCentralityEstimator("V0M"), 
  fAODtrackCutBit(128),
  fHelperPID(0x0),
  fEventCounter(NULL), 
  fEventTree(NULL),
  isQA(kFALSE), 
  fDebug(kFALSE), 
  fRunNumber(0),
  fNumberOfTracks(8000),
  fNumberOfTracksM(8000),
  fFilterBit(99),
  fCentPercentile(-1),
  fVertexX(-99),
  fVertexY(-99),
  fVertexZ(-99) {
  /*
    fTrackPt[kTrack];
    fTrackEta[kTrack];
    fTrackPhi[kTrack];
    fTrackPtM[kTrack];
    fTrackEtaM[kTrack];
    fTrackPhiM[kTrack];
    fTrackCharge[kTrack];
    fTrackPid[kTrack];
    fTrackChargeM[kTrack];
    fTrackPidM[kTrack];
  */
 
  DefineOutput(1, TList::Class()); //! Connect Outpput....
  DefineOutput(2, TTree::Class()); //! Connect Outpput....
}

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

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

  fEventCounter = new TH1D("fEventCounter","EventCounter", 70, 0.5,70.5);
  fThnList->Add(fEventCounter);

  TList *ll = (TList*)fHelperPID->GetOutputList();
  for (Int_t ikey = 0; ikey < ll->GetEntries(); ikey++) {
    fThnList->Add(ll->At(ikey));
  }
  TDirectory *owd = gDirectory;
  OpenFile(1);
  fEventTree = new TTree("fEventTree","fEventTree");
  owd->cd();

  fEventTree->Branch("fRunNumber",      &fRunNumber,     "fRunNumber/I");
  fEventTree->Branch("fFilterBit",      &fFilterBit,     "fFilterBit/I");
  fEventTree->Branch("fNumberOfTracks", &fNumberOfTracks,"fNumberOfTracks/I");
  fEventTree->Branch("fCentPercentile", &fCentPercentile,"fCentPercentile/F");
  
  fEventTree->Branch("fVertexX",        &fVertexX,       "fVertexX/F");
  fEventTree->Branch("fVertexY",        &fVertexY,       "fVertexY/F");
  fEventTree->Branch("fVertexZ",        &fVertexZ,       "fVertexZ/F");
  
  fEventTree->Branch("fTrackPt",        fTrackPt,        "fTrackPt[fNumberOfTracks]/F");
  fEventTree->Branch("fTrackPhi",       fTrackPhi,       "fTrackPhi[fNumberOfTracks]/F");
  fEventTree->Branch("fTrackEta",       fTrackEta,       "fTrackEta[fNumberOfTracks]/F");
  fEventTree->Branch("fTrackCharge",    fTrackCharge,    "fTrackCharge[fNumberOfTracks]/I");
  fEventTree->Branch("fTrackPid",       fTrackPid,       "fTrackPid[fNumberOfTracks]/I");
  
  if(isMC) {
    fEventTree->Branch("fNumberOfTracksM", &fNumberOfTracksM, "fNumberOfTracksM/I");
    fEventTree->Branch("fTrackPtM",        fTrackPtM,        "fTrackPtM[fNumberOfTracksM]/F");
    fEventTree->Branch("fTrackPhiM",       fTrackPhiM,       "fTrackPhiM[fNumberOfTracksM]/F");
    fEventTree->Branch("fTrackEtaM",       fTrackEtaM,       "fTrackEtaM[fNumberOfTracksM]/F");
    fEventTree->Branch("fTrackChargeM",    fTrackChargeM,    "fTrackChargeM[fNumberOfTracksM]/I");
    fEventTree->Branch("fTrackPidM",       fTrackPidM,       "fTrackPidM[fNumberOfTracksM]/I");
  }

  PostData(1, fThnList);
  PostData(2, fEventTree);  
}

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

   fEventCounter->Fill(1);

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

  fEventCounter->Fill(2);

  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());
 // gRefMul = aodHeader->GetRefMultiplicity();
  if (gCent < 0 || gCent > 100) return;
  fEventCounter->Fill(3);  

  const AliAODVertex *vertex = event->GetPrimaryVertex();
  if(!vertex) return;
  fEventCounter->Fill(4);
  Bool_t vtest = kFALSE;
  Double32_t fCov[6];
  vertex->GetCovarianceMatrix(fCov);
  if(vertex->GetNContributors() > 0) {
      if(fCov[5] != 0) {
	vtest = kTRUE;
      }
  }
  if(!vtest)return;
  
  fEventCounter->Fill(5);

  AliCentrality *centrality = event->GetCentrality();
  if (centrality->GetQuality() != 0) return;

  fEventCounter->Fill(6);

  fRunNumber = event->GetRunNumber();
  fFilterBit = fAODtrackCutBit;
  fCentPercentile = gCent;
  fVertexX = vertex->GetX();
  fVertexY = vertex->GetY();
  fVertexZ = vertex->GetZ();
  
//Default TPC priors
  if(fHelperPID->GetPIDType()==kBayes)fHelperPID->GetPIDCombined()->SetDefaultTPCPriors();//FIXME maybe this can go in the UserCreateOutputObject?


  Int_t iTracks = 0; 
  for (Int_t itrk = 0; itrk < event->GetNumberOfTracks(); itrk++) {
    AliAODTrack* track = dynamic_cast<AliAODTrack *>(event->GetTrack(itrk));
    fEventCounter->Fill(10);
    if (!track) continue;
    fEventCounter->Fill(11);
    if (!AcceptTrack(track)) continue;
    fEventCounter->Fill(12);
    Int_t a = fHelperPID->GetParticleSpecies((AliVTrack*) track,kTRUE);
    //  if(a < 0 || a > 2) continue;
    fEventCounter->Fill(13);
    
    Int_t b = -999;
    if (a == 0 ) b = 1;
    else if (a == 1 ) b = 2;
    else if (a == 2 ) b = 3;
    else b = 4;
    
    if (track->Charge()  < 0 ) b = -1*b;
    
    //    Int_t icharge = track->Charge() > 0 ? 0 : 1;

    // cout << icharge << "  " << track->Charge() << endl;

    fTrackPt[iTracks]     = (Float_t)track->Pt();
    fTrackPhi[iTracks]    = (Float_t)track->Phi();
    fTrackEta[iTracks]    = (Float_t)track->Eta();
    fTrackCharge[iTracks] = track->Charge();
    fTrackPid[iTracks] = b;
    iTracks++;
  }
  fNumberOfTracks = iTracks;

  if(isMC) {
    fEventCounter->Fill(21);
    TClonesArray *arrayMC= 0; 
    arrayMC = dynamic_cast<TClonesArray*> (event->GetList()->FindObject(AliAODMCParticle::StdBranchName()));
    if (!arrayMC) {
      Printf("Error: MC particles branch not found!\n");
      return;
    }
    fEventCounter->Fill(22);
    AliAODMCHeader *mcHdr=0;
    mcHdr=(AliAODMCHeader*)event->GetList()->FindObject(AliAODMCHeader::StdBranchName());  
    if(!mcHdr) {
      Printf("MC header branch not found!\n");
      return;
    }
    
    fEventCounter->Fill(23);
    
    Int_t nMC = arrayMC->GetEntries();
    Int_t mTracks = 0; 
    for (Int_t iMC = 0; iMC < nMC; iMC++) {
      fEventCounter->Fill(24);
      AliAODMCParticle *partMC = (AliAODMCParticle*) arrayMC->At(iMC);
      if(!AcceptMCTrack(partMC)) continue;
      
      fEventCounter->Fill(25);
      Int_t a  = partMC->GetPdgCode();
      
      // Int_t a = fHelperPID->GetMCParticleSpecie((AliVEvent*) event,(AliVTrack*)partMC,1);
      
      //if(a < 0 || a > 2) continue;
      Int_t icharge = a > 0 ? 0 : 1;
      
      //  cout << a << "  " << icharge << endl;
      
      fTrackPtM[mTracks]     = (Float_t)partMC->Pt();
      fTrackPhiM[mTracks]    = (Float_t)partMC->Phi();
      fTrackEtaM[mTracks]    = (Float_t)partMC->Eta();
      fTrackChargeM[mTracks] = icharge;
      fTrackPidM[mTracks] = a;
      
      mTracks++;
    }
    fEventCounter->Fill(26);
    fNumberOfTracksM = mTracks;
  }
  
  fEventCounter->Fill(30);
  fEventTree->Fill();
  
  PostData(1, fThnList);  
  PostData(2, fEventTree);
}

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

//___________________________________________________________
Bool_t AliEbyEPidTTask::AcceptTrack(AliAODTrack *track) const {
  if(!track)                                  return kFALSE;
  if (track->Charge() == 0 )                  return kFALSE;
  if (!track->TestFilterBit(fAODtrackCutBit)) return kFALSE;
  return kTRUE;
}


//___________________________________________________________
Bool_t AliEbyEPidTTask::AcceptMCTrack(AliAODMCParticle *track) const {
  if(!track) return kFALSE;
  if(!track->IsPhysicalPrimary()) return kFALSE;
  if (track->Charge() == 0 ) return kFALSE;
  return kTRUE;
}
Commit	Line	Data
3435d75f	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: ALICE Offline. *
	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	// Dealing with Wide pT Addition (Test Only - not for use)
	22	//=========================================================================//
	23
	24	#include "TChain.h"
	25	#include "TList.h"
	26	#include "TFile.h"
	27	#include "TTree.h"
	28	#include "TH1D.h"
	29	#include "TH2F.h"
	30	#include "TH3F.h"
	31	#include "TCanvas.h"
	32	#include "AliAnalysisTask.h"
	33	#include "AliAnalysisManager.h"
	34	#include "AliVEvent.h"
	35	#include "AliESD.h"
	36	#include "AliESDEvent.h"
	37	#include "AliAODEvent.h"
	38	#include "AliAODMCParticle.h"
	39	#include "AliAODMCHeader.h"
	40	#include "AliPIDResponse.h"
e69ef551	41	#include "AliPIDCombined.h"
3435d75f	42	#include "AliAODHeader.h"
	43	#include "AliAODpidUtil.h"
	44	#include "AliHelperPID.h"
	45	using std::endl;
	46	using std::cout;
	47	#include "AliEbyEPidTTask.h"
	48
	49	ClassImp(AliEbyEPidTTask)
	50
	51	//-----------------------------------------------------------------------
	52	AliEbyEPidTTask::AliEbyEPidTTask( const char *name ) : AliAnalysisTaskSE( name ),
	53	fThnList(NULL),
	54	isMC(kFALSE),
	55	fCentralityEstimator("V0M"),
	56	fAODtrackCutBit(128),
	57	fHelperPID(0x0),
	58	fEventCounter(NULL),
	59	fEventTree(NULL),
	60	isQA(kFALSE),
	61	fDebug(kFALSE),
	62	fRunNumber(0),
	63	fNumberOfTracks(8000),
	64	fNumberOfTracksM(8000),
	65	fFilterBit(99),
	66	fCentPercentile(-1),
	67	fVertexX(-99),
	68	fVertexY(-99),
	69	fVertexZ(-99) {
	70	/*
	71	fTrackPt[kTrack];
	72	fTrackEta[kTrack];
	73	fTrackPhi[kTrack];
	74	fTrackPtM[kTrack];
	75	fTrackEtaM[kTrack];
	76	fTrackPhiM[kTrack];
	77	fTrackCharge[kTrack];
	78	fTrackPid[kTrack];
	79	fTrackChargeM[kTrack];
	80	fTrackPidM[kTrack];
	81	*/
	82
	83	DefineOutput(1, TList::Class()); //! Connect Outpput....
	84	DefineOutput(2, TTree::Class()); //! Connect Outpput....
	85	}
	86
	87	AliEbyEPidTTask::~AliEbyEPidTTask() {
	88	//! Cleaning up
	89	if (fThnList) delete fThnList;
	90	if (fHelperPID) delete fHelperPID;
e69ef551	91	if (fEventTree) delete fEventTree;
3435d75f	92	}
	93
	94	//---------------------------------------------------------------------------------
	95	void AliEbyEPidTTask::UserCreateOutputObjects() {
	96	fThnList = new TList();
	97	fThnList->SetOwner(kTRUE);
	98
be2d8a61	99	fEventCounter = new TH1D("fEventCounter","EventCounter", 70, 0.5,70.5);
3435d75f	100	fThnList->Add(fEventCounter);
	101
	102	TList ll = (TList)fHelperPID->GetOutputList();
	103	for (Int_t ikey = 0; ikey < ll->GetEntries(); ikey++) {
	104	fThnList->Add(ll->At(ikey));
	105	}
e69ef551	106	TDirectory *owd = gDirectory;
e69ef551	107	OpenFile(1);
3435d75f	108	fEventTree = new TTree("fEventTree","fEventTree");
e69ef551	109	owd->cd();
e69ef551	110
3435d75f	111	fEventTree->Branch("fRunNumber", &fRunNumber, "fRunNumber/I");
	112	fEventTree->Branch("fFilterBit", &fFilterBit, "fFilterBit/I");
	113	fEventTree->Branch("fNumberOfTracks", &fNumberOfTracks,"fNumberOfTracks/I");
	114	fEventTree->Branch("fCentPercentile", &fCentPercentile,"fCentPercentile/F");
	115
	116	fEventTree->Branch("fVertexX", &fVertexX, "fVertexX/F");
	117	fEventTree->Branch("fVertexY", &fVertexY, "fVertexY/F");
	118	fEventTree->Branch("fVertexZ", &fVertexZ, "fVertexZ/F");
	119
	120	fEventTree->Branch("fTrackPt", fTrackPt, "fTrackPt[fNumberOfTracks]/F");
	121	fEventTree->Branch("fTrackPhi", fTrackPhi, "fTrackPhi[fNumberOfTracks]/F");
	122	fEventTree->Branch("fTrackEta", fTrackEta, "fTrackEta[fNumberOfTracks]/F");
	123	fEventTree->Branch("fTrackCharge", fTrackCharge, "fTrackCharge[fNumberOfTracks]/I");
	124	fEventTree->Branch("fTrackPid", fTrackPid, "fTrackPid[fNumberOfTracks]/I");
	125
	126	if(isMC) {
	127	fEventTree->Branch("fNumberOfTracksM", &fNumberOfTracksM, "fNumberOfTracksM/I");
	128	fEventTree->Branch("fTrackPtM", fTrackPtM, "fTrackPtM[fNumberOfTracksM]/F");
	129	fEventTree->Branch("fTrackPhiM", fTrackPhiM, "fTrackPhiM[fNumberOfTracksM]/F");
	130	fEventTree->Branch("fTrackEtaM", fTrackEtaM, "fTrackEtaM[fNumberOfTracksM]/F");
	131	fEventTree->Branch("fTrackChargeM", fTrackChargeM, "fTrackChargeM[fNumberOfTracksM]/I");
	132	fEventTree->Branch("fTrackPidM", fTrackPidM, "fTrackPidM[fNumberOfTracksM]/I");
	133	}
	134
	135	PostData(1, fThnList);
	136	PostData(2, fEventTree);
	137	}
	138
	139	//----------------------------------------------------------------------------------
	140	void AliEbyEPidTTask::UserExec( Option_t * ){
	141
	142	fEventCounter->Fill(1);
	143
	144	AliAODEvent* event = dynamic_cast<AliAODEvent*>(InputEvent());
	145	if (!event) {
	146	Printf("ERROR 01: AOD not found ");
	147	return;
	148	}
	149
be2d8a61	150	fEventCounter->Fill(2);
be2d8a61	151
3435d75f	152	Int_t gCent = -1;
638ae862	153	// Float_t gRefMul = -1;
3435d75f	154
0a918d8d	155	AliAODHeader aodHeader = dynamic_cast<AliAODHeader>(event->GetHeader());
0a918d8d	156	if(!aodHeader) AliFatal("Not a standard AOD");
3435d75f	157	gCent = (Int_t)aodHeader->GetCentralityP()->GetCentralityPercentile(fCentralityEstimator.Data());
638ae862	158	// gRefMul = aodHeader->GetRefMultiplicity();
3435d75f	159	if (gCent < 0 \|\| gCent > 100) return;
be2d8a61	160	fEventCounter->Fill(3);
3435d75f	161
	162	const AliAODVertex *vertex = event->GetPrimaryVertex();
	163	if(!vertex) return;
be2d8a61	164	fEventCounter->Fill(4);
3435d75f	165	Bool_t vtest = kFALSE;
	166	Double32_t fCov[6];
	167	vertex->GetCovarianceMatrix(fCov);
	168	if(vertex->GetNContributors() > 0) {
	169	if(fCov[5] != 0) {
	170	vtest = kTRUE;
	171	}
	172	}
	173	if(!vtest)return;
	174
be2d8a61	175	fEventCounter->Fill(5);
be2d8a61	176
3435d75f	177	AliCentrality *centrality = event->GetCentrality();
	178	if (centrality->GetQuality() != 0) return;
	179
be2d8a61	180	fEventCounter->Fill(6);
be2d8a61	181
3435d75f	182	fRunNumber = event->GetRunNumber();
	183	fFilterBit = fAODtrackCutBit;
	184	fCentPercentile = gCent;
	185	fVertexX = vertex->GetX();
	186	fVertexY = vertex->GetY();
	187	fVertexZ = vertex->GetZ();
	188
e69ef551	189	//Default TPC priors
	190	if(fHelperPID->GetPIDType()==kBayes)fHelperPID->GetPIDCombined()->SetDefaultTPCPriors();//FIXME maybe this can go in the UserCreateOutputObject?
	191
	192
	193
	194
3435d75f	195	Int_t iTracks = 0;
	196	for (Int_t itrk = 0; itrk < event->GetNumberOfTracks(); itrk++) {
	197	AliAODTrack* track = dynamic_cast<AliAODTrack *>(event->GetTrack(itrk));
be2d8a61	198	fEventCounter->Fill(10);
3435d75f	199	if (!track) continue;
be2d8a61	200	fEventCounter->Fill(11);
3435d75f	201	if (!AcceptTrack(track)) continue;
be2d8a61	202	fEventCounter->Fill(12);
3435d75f	203	Int_t a = fHelperPID->GetParticleSpecies((AliVTrack*) track,kTRUE);
122cbc27	204	// if(a < 0 \|\| a > 2) continue;
be2d8a61	205	fEventCounter->Fill(13);
122cbc27	206
	207	Int_t b = -999;
	208	if (a == 0 ) b = 1;
	209	else if (a == 1 ) b = 2;
	210	else if (a == 2 ) b = 3;
	211	else b = 4;
	212
	213	if (track->Charge() < 0 ) b = -1*b;
	214
e69ef551	215	// Int_t icharge = track->Charge() > 0 ? 0 : 1;
3435d75f	216
	217	// cout << icharge << " " << track->Charge() << endl;
	218
	219	fTrackPt[iTracks] = (Float_t)track->Pt();
	220	fTrackPhi[iTracks] = (Float_t)track->Phi();
	221	fTrackEta[iTracks] = (Float_t)track->Eta();
e69ef551	222	fTrackCharge[iTracks] = track->Charge();
122cbc27	223	fTrackPid[iTracks] = b;
3435d75f	224	iTracks++;
	225	}
	226	fNumberOfTracks = iTracks;
	227
	228	if(isMC) {
be2d8a61	229	fEventCounter->Fill(21);
3435d75f	230	TClonesArray *arrayMC= 0;
	231	arrayMC = dynamic_cast<TClonesArray*> (event->GetList()->FindObject(AliAODMCParticle::StdBranchName()));
	232	if (!arrayMC) {
	233	Printf("Error: MC particles branch not found!\n");
	234	return;
	235	}
be2d8a61	236	fEventCounter->Fill(22);
3435d75f	237	AliAODMCHeader *mcHdr=0;
	238	mcHdr=(AliAODMCHeader*)event->GetList()->FindObject(AliAODMCHeader::StdBranchName());
	239	if(!mcHdr) {
	240	Printf("MC header branch not found!\n");
	241	return;
	242	}
122cbc27	243
be2d8a61	244	fEventCounter->Fill(23);
3435d75f	245
122cbc27	246	Int_t nMC = arrayMC->GetEntries();
122cbc27	247	Int_t mTracks = 0;
3435d75f	248	for (Int_t iMC = 0; iMC < nMC; iMC++) {
be2d8a61	249	fEventCounter->Fill(24);
122cbc27	250	AliAODMCParticle partMC = (AliAODMCParticle) arrayMC->At(iMC);
	251	if(!AcceptMCTrack(partMC)) continue;
	252
	253	fEventCounter->Fill(25);
	254	Int_t a = partMC->GetPdgCode();
	255
	256	// Int_t a = fHelperPID->GetMCParticleSpecie((AliVEvent) event,(AliVTrack)partMC,1);
	257
	258	//if(a < 0 \|\| a > 2) continue;
	259	Int_t icharge = a > 0 ? 0 : 1;
	260
	261	// cout << a << " " << icharge << endl;
	262
	263	fTrackPtM[mTracks] = (Float_t)partMC->Pt();
	264	fTrackPhiM[mTracks] = (Float_t)partMC->Phi();
	265	fTrackEtaM[mTracks] = (Float_t)partMC->Eta();
	266	fTrackChargeM[mTracks] = icharge;
	267	fTrackPidM[mTracks] = a;
	268
	269	mTracks++;
3435d75f	270	}
be2d8a61	271	fEventCounter->Fill(26);
3435d75f	272	fNumberOfTracksM = mTracks;
	273	}
	274
be2d8a61	275	fEventCounter->Fill(30);
be2d8a61	276	fEventTree->Fill();
3435d75f	277
	278	PostData(1, fThnList);
	279	PostData(2, fEventTree);
	280	}
	281
	282	//___________________________________________________________
	283	void AliEbyEPidTTask::Terminate( Option_t * ){
	284	Info("AliEbyEPidTTask"," Task Successfully finished");
	285	}
	286
	287	//___________________________________________________________
	288	Bool_t AliEbyEPidTTask::AcceptTrack(AliAODTrack *track) const {
	289	if(!track) return kFALSE;
	290	if (track->Charge() == 0 ) return kFALSE;
	291	if (!track->TestFilterBit(fAODtrackCutBit)) return kFALSE;
	292	return kTRUE;
	293	}
	294
	295
	296	//___________________________________________________________
	297	Bool_t AliEbyEPidTTask::AcceptMCTrack(AliAODMCParticle *track) const {
	298	if(!track) return kFALSE;
	299	if(!track->IsPhysicalPrimary()) return kFALSE;
	300	if (track->Charge() == 0 ) return kFALSE;
	301	return kTRUE;
	302	}