--- /dev/null
+/**************************************************************************
+ * 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. *
+ **************************************************************************/
+
+// Extension to Pi0FLOw, mimicing AliPHOSHijingEfficiency
+// by Dmitri Peressounko, 05.02.2013
+// Authors: Henrik Qvigstad, Dmitri Peressounko
+// Date : 05.04.2013
+/* $Id$ */
+
+#include "TChain.h"
+#include "TTree.h"
+#include "TObjArray.h"
+#include "TF1.h"
+#include "TFile.h"
+#include "TH1F.h"
+#include "TH2F.h"
+#include "TH2I.h"
+#include "TH3F.h"
+#include "TParticle.h"
+#include "TCanvas.h"
+#include "TStyle.h"
+#include "TRandom.h"
+#include "TROOT.h"
+#include "THashList.h"
+
+
+#include "AliAnalysisManager.h"
+#include "AliMCEventHandler.h"
+#include "AliMCEvent.h"
+#include "AliStack.h"
+#include "AliAnalysisTaskSE.h"
+#include "AliPHOSHijingEfficiency.h"
+#include "AliCaloPhoton.h"
+#include "AliPHOSGeometry.h"
+#include "AliPHOSEsdCluster.h"
+#include "AliPHOSCalibData.h"
+#include "AliESDEvent.h"
+#include "AliESDCaloCells.h"
+#include "AliESDVertex.h"
+#include "AliESDtrackCuts.h"
+#include "AliLog.h"
+#include "AliPID.h"
+#include "AliCDBManager.h"
+#include "AliCentrality.h"
+#include "AliESDtrackCuts.h"
+#include "AliEventplane.h"
+#include "TProfile.h"
+#include <TPDGCode.h>
+#include "AliOADBContainer.h"
+
+
+#include "AliAnalysisTaskPi0FlowMC.h"
+
+ClassImp(AliAnalysisTaskPi0FlowMC);
+
+//TODO: rnlin?
+
+AliAnalysisTaskPi0FlowMC::AliAnalysisTaskPi0FlowMC(const char* name, AliAnalysisTaskPi0Flow::Period period)
+: AliAnalysisTaskPi0Flow(name, period),
+ fStack(0x0)
+{
+}
+
+AliAnalysisTaskPi0FlowMC::~AliAnalysisTaskPi0FlowMC()
+{
+}
+
+void AliAnalysisTaskPi0FlowMC::MakeMCHistograms()
+{
+ //AliAnalysisTaskPi0Flow::MakeMCHistograms();
+
+ // MC Generated histograms
+ char key[55];
+ for(Int_t cent=0; cent < fCentEdges.GetSize()-1; cent++){
+ snprintf(key,55,"hMC_rap_gamma_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity pi0",200,-1.,1.)) ;
+ snprintf(key,55,"hMC_rap_pi0_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity pi0",200,-1.,1.)) ;
+ snprintf(key,55,"hMC_rap_eta_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity eta",200,-1.,1.)) ;
+ snprintf(key,55,"hMC_phi_gamma_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Phi pi0",200,0.,TMath::TwoPi())) ;
+ snprintf(key,55,"hMC_phi_pi0_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Phi pi0",200,0.,TMath::TwoPi())) ;
+ snprintf(key,55,"hMC_phi_eta_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Phi eta",200,0.,TMath::TwoPi())) ;
+ snprintf(key,55,"hMC_all_gamma_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity photon",250,0.,25.)) ;
+ snprintf(key,55,"hMC_all_pi0_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity pi0",250,0.,25.)) ;
+ snprintf(key,55,"hMC_all_eta_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity eta",250,0.,25.)) ;
+ snprintf(key,55,"hMC_unitEta_gamma_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Pt photon",250,0.,25.)) ;
+ snprintf(key,55,"hMC_unitEta_pi0_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity eta",250,0.,25.)) ;
+ snprintf(key,55,"hMC_unitEta_eta_cen%d",cent) ;
+ fOutputContainer->Add(new TH1F(key,"Rapidity eta",250,0.,25.)) ;
+ }
+ fOutputContainer->Add(new TH2F("hMC_gamma_vertex","Creation vertex",25,0.,25.,1000,0.,500.)) ;
+ fOutputContainer->Add(new TH2F("hMC_pi0_vertex","Creation vertex",25,0.,25.,1000,0.,500.)) ;
+ fOutputContainer->Add(new TH2F("hMC_eta_vertex","Creation vertex",25,0.,25.,1000,0.,500.)) ;
+
+
+ Int_t nPt = 200;
+ Double_t ptMin = 0;
+ Double_t ptMax = 20;
+ fOutputContainer->Add(new TH2F("Vertex","Pi0 creation vertex",nPt,ptMin,ptMax,5000,0.,500.));
+ fOutputContainer->Add(new TH3F("hSecondPi0RphiZ","Secondary pi0 vertex",450,0.,450.,100,0.,TMath::TwoPi(),200,-100.,100.));
+ fOutputContainer->Add(new TH2F("hSecondPi0RE","Secondary pi0 vertex",450,0.,450.,200,0.,20.));
+ fOutputContainer->Add(new TH3F("hMass_R","Mass vs radius any parent",50,0.,0.25,100,0.,10.,300,0.,600.));
+ fOutputContainer->Add(new TH3F("Real_pi_R","All clusters",50,0.,0.25,100,0.,10.,250,0.,500.));
+ fOutputContainer->Add(new TH3F("Real_pi_Z","All clusters",50,0.,0.25,100,0.,10.,100,-100.,100.));
+// fOutputContainer->Add(new TH2F(Form("Real_npnp_RZ"),"All clusters",250,0.,500.,100,-100.,100.));
+// fOutputContainer->Add(new TH3F(Form("Real_mass_R"),"All clusters",50,0.,0.25,100,0.,10.,300,0.,600.));
+
+ const Int_t nM = 500;
+ const Double_t mMin = 0.0;
+ const Double_t mMax = 1.0;
+
+ for(Int_t cen=0; cen < fCentEdges.GetSize()-1; cen++){
+ fOutputContainer->Add(new TH1F(Form("hPrimPhot_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimEl_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimPi0_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimEta_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimPipm_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimP_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimPbar_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimN_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimNbar_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimK0S_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimK0L_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimKpm_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH1F(Form("hPrimOther_cen%d",cen),"Primary spetrum",nPt,ptMin,ptMax));
+
+ //pairs from common parents
+ fOutputContainer->Add(new TH2F(Form("hParentAll_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentK0s_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentGamma_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentEl_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentOther_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentDirPi0_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+
+ //common parent - pi0
+ fOutputContainer->Add(new TH2F(Form("hParentPi0NoPrim_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Eta_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Omega_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Pipm_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Kpm_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Ks_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0Kl_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0pn_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+ fOutputContainer->Add(new TH2F(Form("hParentPi0antipn_cen%d",cen),"All clusters",nM,mMin,mMax,nPt,ptMin,ptMax));
+
+ }
+
+
+ //Photon contaminations
+ fOutputContainer->Add(new TH2F("hPipmGammaConvR","Conversion radius" ,200,0.,20.,1000,0.,500.));
+ fOutputContainer->Add(new TH2F("hPipmElConvR","Conversion radius" ,200,0.,20.,1000,0.,500.));
+ fOutputContainer->Add(new TH2F("hPipmNConvR","Conversion radius" ,200,0.,20.,1000,0.,500.));
+ fOutputContainer->Add(new TH2F("hPipmOtherConvR","Conversion radius" ,200,0.,20.,1000,0.,500.));
+ fOutputContainer->Add(new TH2F("hPipmGammaConvRZ","Conversion radius" ,400,-200.,200.,1000,0.,500.));
+
+ const Int_t nTypes=24 ;
+ char partTypes[nTypes][55] ;
+ snprintf(partTypes[0],55,"hGammaNoPrim") ; //
+ snprintf(partTypes[1],55,"hGammaPhot") ; //
+ snprintf(partTypes[2],55,"hGammaEl") ; //
+ snprintf(partTypes[3],55,"hGammaPi0") ; //
+ snprintf(partTypes[4],55,"hGammaEta") ; //
+ snprintf(partTypes[5],55,"hhGammaOmega") ; //
+ snprintf(partTypes[6],55,"hGammaPipm") ; //
+ snprintf(partTypes[7],55,"hGammaP") ; //
+ snprintf(partTypes[8],55,"hGammaPbar") ; //
+ snprintf(partTypes[9],55,"hGammaN") ; //
+ snprintf(partTypes[10],55,"hGammaNbar") ; //
+ snprintf(partTypes[11],55,"hGammaK0S") ; //
+ snprintf(partTypes[12],55,"hGammaK0L") ; //
+ snprintf(partTypes[13],55,"hGammaKpm") ; //
+ snprintf(partTypes[14],55,"hGammaKstar") ; //
+ snprintf(partTypes[15],55,"hGammaDelta") ; //
+ snprintf(partTypes[16],55,"hGammaOtherCharged") ; //
+ snprintf(partTypes[17],55,"hGammaOtherNeutral") ; //
+ snprintf(partTypes[18],55,"hGammaPipmGamma") ; //
+ snprintf(partTypes[19],55,"hGammaPipmEl") ; //
+ snprintf(partTypes[20],55,"hGammaPipmOther") ; //
+ snprintf(partTypes[21],55,"hGammaPipmDirect") ; //
+ snprintf(partTypes[22],55,"hGammaPipmp") ; //
+ snprintf(partTypes[23],55,"hGammaPipmn") ; //
+
+ const Int_t nPID=12 ;
+ char cPID[25][12] ;
+ snprintf(cPID[0],25,"All") ;
+ snprintf(cPID[1],25,"Allcore") ;
+ snprintf(cPID[2],25,"CPV") ;
+ snprintf(cPID[3],25,"CPVcore") ;
+ snprintf(cPID[4],25,"CPV2") ;
+ snprintf(cPID[5],25,"CPV2core") ;
+ snprintf(cPID[6],25,"Disp") ;
+ snprintf(cPID[7],25,"Dispcore") ;
+ snprintf(cPID[8],25,"Disp2") ;
+ snprintf(cPID[9],25,"Disp2core") ;
+ snprintf(cPID[10],25,"Both") ;
+ snprintf(cPID[11],25,"Bothcore") ;
+
+ for(Int_t itype=0; itype<nTypes; itype++){
+ for(Int_t iPID=0; iPID<nPID; iPID++){
+ for(Int_t cen=0; cen<5; cen++){
+ fOutputContainer->Add(new TH1F(Form("%s_%s_cen%d",partTypes[itype],cPID[iPID],cen),"Cluster parents",nPt,ptMin,ptMax));
+ }
+ }
+ }
+}
+
+void AliAnalysisTaskPi0FlowMC::DoMC()
+{
+ //AliAnalysisTaskPi0Flow::DoMC();
+ fStack = GetMCStack();
+
+ //TODO: Geometry IHEP?
+ //TODO: decalib.?
+ //TODO: PHOS matrix?
+ //TODO: Centrality.
+
+ FillMCHist();
+
+ // Proccess all selected clusters
+ for (Int_t i1=0; i1<fCaloPhotonsPHOS->GetEntriesFast(); i1++) {
+ //Bool_t sure= kTRUE;
+ //Int_t primary=FindPrimary(clu,sure) ; //номер праймари частицы в стеке
+ //ph->SetPrimary(primary) ;
+ //ph->SetWeight(PrimaryWeight(primary)) ;
+ }
+
+ FillSecondaries() ;
+}
+
+
+
+//___________________________________________________________________________
+void AliAnalysisTaskPi0FlowMC::FillMCHist(){
+ //fill histograms for efficiensy etc. calculation
+
+ //---------First pi0/eta-----------------------------
+ char partName[10] ;
+ char hkey[55] ;
+
+ if(!fStack) return ;
+ for(Int_t i=0;i<fStack->GetNtrack();i++){
+ TParticle* particle = fStack->Particle(i);
+ if(particle->GetPdgCode() == kPi0)
+ snprintf(partName,10,"pi0") ;
+ else
+ if(particle->GetPdgCode() == kEta)
+ snprintf(partName,10,"eta") ;
+ else
+ if(particle->GetPdgCode() == kGamma)
+ snprintf(partName,10,"gamma") ;
+ else
+ continue ;
+
+ //Primary particle
+ Double_t r=particle->R() ;
+ Double_t pt = particle->Pt() ;
+ //Distribution over vertex
+ FillHistogram(Form("hMC_%s_vertex",partName),pt,r) ;
+
+ if(r >kRCut)
+ continue ;
+
+ //Total number of pi0 with creation radius <1 cm
+ Double_t weight = PrimaryParticleWeight(particle) ;
+ snprintf(hkey,55,"hMC_all_%s_cen%d",partName,fCentBin) ;
+ FillHistogram(hkey,pt,weight) ;
+ if(TMath::Abs(particle->Y())<0.12){
+ snprintf(hkey,55,"hMC_unitEta_%s_cen%d",partName,fCentBin) ;
+ FillHistogram(hkey,pt,weight) ;
+ }
+
+ snprintf(hkey,55,"hMC_rap_%s_cen%d",partName,fCentBin) ;
+ FillHistogram(hkey,particle->Y(),weight) ;
+
+ Double_t phi=particle->Phi() ;
+ while(phi<0.)phi+=TMath::TwoPi() ;
+ while(phi>TMath::TwoPi())phi-=TMath::TwoPi() ;
+ snprintf(hkey,55,"hMC_phi_%s_cen%d",partName,fCentBin) ;
+ FillHistogram(hkey,phi,weight) ;
+ }
+}
+
+//________________________________________________________________________
+void AliAnalysisTaskPi0FlowMC::FillSecondaries(){
+ //Sort secondaires
+
+ //Fill spectra of primary particles
+ //with proper weight
+ std::cout << fStack << std::endl;
+ AliInfo("start");
+ for(Int_t i=0; i<fStack->GetNtrack(); i++){
+ TParticle * p = fStack->Particle(i) ;
+ if(p->R()>kRCut)
+ continue ;
+ if(TMath::Abs(p->Y())>0.5)
+ continue ;
+ Double_t w = PrimaryParticleWeight(p) ;
+ Int_t primPdgCode=p->GetPdgCode() ;
+ switch(primPdgCode){
+ case kGamma: FillHistogram(Form("hPrimPhot_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kElectron:
+ case -kElectron:
+ FillHistogram(Form("hPrimEl_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kPi0:
+ FillHistogram(Form("hPrimPi0_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kEta:
+ FillHistogram(Form("hPrimEta_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kPiPlus:
+ case kPiMinus:
+ FillHistogram(Form("hPrimPipm_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kProton: //p
+ FillHistogram(Form("hPrimP_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kProtonBar: //pbar
+ FillHistogram(Form("hPrimPbar_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kNeutron: //n
+ FillHistogram(Form("hPrimN_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case kNeutronBar: //nbar
+ FillHistogram(Form("hPrimNbar_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case 310: //nbar
+ FillHistogram(Form("hPrimK0S_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case 130: //nbar
+ FillHistogram(Form("hPrimK0L_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ case 321: //K+
+ case -321: //K-
+ FillHistogram(Form("hPrimKpm_cen%d",fCentBin),p->Pt(),w);
+ break ;
+ default: //other
+ FillHistogram(Form("hPrimOther_cen%d",fCentBin),p->Pt(),w);
+ }
+ }
+ AliInfo("Origins of secondary pi0s");
+ //Origins of secondary pi0s
+ for(Int_t i=0; i<fStack->GetNtrack(); i++){
+ TParticle * p = fStack->Particle(i) ;
+ if(p->GetPdgCode()!=111)
+ continue ;
+ FillHistogram("Vertex",p->Pt(),p->R());
+ if(p->R()<kRCut)
+ continue ;
+ Double_t phi=p->Phi() ;
+ while(phi<0.)phi+=TMath::TwoPi() ;
+ while(phi>TMath::TwoPi())phi-=TMath::TwoPi() ;
+ FillHistogram("hSecondPi0RphiZ",p->R(),phi,p->Vz()) ;
+ Double_t w = PrimaryParticleWeight(p) ;
+ FillHistogram("hSecondPi0RE",p->R(),p->Pt(),w) ;
+ }
+
+ TLorentzVector p1;
+
+ Int_t inPHOS=fCaloPhotonsPHOS->GetEntries() ;
+ for (Int_t i1=0; i1<inPHOS-1; i1++) {
+ AliCaloPhoton * ph1=(AliCaloPhoton*)fCaloPhotonsPHOS->At(i1) ;
+ Double_t w1=ph1->GetWeight() ;
+ for (Int_t i2=i1+1; i2<inPHOS; i2++) {
+ AliCaloPhoton * ph2=(AliCaloPhoton*)fCaloPhotonsPHOS->At(i2) ;
+ TLorentzVector p12 = *ph1 + *ph2;
+ Double_t w2=ph2->GetWeight() ;
+ Double_t w = TMath::Sqrt(w1*w2) ;
+ FillHistogram(Form("hParentAll_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ Int_t prim=FindCommonParent(ph1->GetPrimary(),ph2->GetPrimary()) ;
+ if(prim>-1){
+ TParticle * particle = (TParticle *)fStack->Particle(prim);
+ FillHistogram("hMass_R",p12.M(),p12.Pt(),TMath::Sqrt(particle->R()*particle->R()+particle->Vz()*particle->Vz())) ;
+
+
+ Int_t pdgCode=particle->GetPdgCode() ;
+ if(pdgCode!=111){ //common parent - not pi0
+ if(pdgCode==22)
+ FillHistogram(Form("hParentGamma_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ else{
+ if(pdgCode==11 || pdgCode==-11){
+ FillHistogram(Form("hParentEl_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ }
+ else{
+ if(InPi0mass(p12.M() ,p12.Pt())){
+ printf("Common parent: %d \n",pdgCode) ;
+ }
+ FillHistogram(Form("hParentOther_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ }
+ }//Not photons
+ }//Common parent not pi0
+ else{ //common parent - pi0
+ FillHistogram(Form("hParentPi0_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ FillHistogram(Form("Real_pi_R"),p12.M(),p12.Pt(),particle->R(),w) ;
+ FillHistogram(Form("Real_pi_Z"),p12.M(),p12.Pt(),particle->Vz(),w) ;
+ if(particle->R()<kRCut && TMath::Abs(particle->Vz())<fMaxAbsVertexZ){
+ FillHistogram(Form("hParentDirPi0_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ continue ;
+ }
+ //Common particle pi0, created off-vertex
+ Int_t primPi0=particle->GetFirstMother();
+ if(primPi0==-1){
+ FillHistogram(Form("hParentPi0NoPrim_cen%d",fCentBin),p12.M(),p12.Pt(),w) ;
+ }
+ else{
+ Int_t primPdgCode=fStack->Particle(primPi0)->GetPdgCode() ;
+ switch(primPdgCode){
+ case 221: FillHistogram(Form("hParentPi0Eta_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; //eta
+ break ;
+ case 223: FillHistogram(Form("hParentPi0Omega_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; //omega
+ break ;
+ case 211: //pi+-
+ case -211: FillHistogram(Form("hParentPi0Pipm_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; //
+ break ;
+ case 321: //K+-
+ case -321: FillHistogram(Form("hParentPi0Kpm_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; //
+ break ;
+ case 310: FillHistogram(Form("hParentPi0Ks_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; // K0S
+ break ;
+ case 130: FillHistogram(Form("hParentPi0Kl_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; // K0L
+ break ;
+ case 2212: //p
+ case 2112: //n
+ FillHistogram(Form("hParentPi0pn_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; // pn
+ break ;
+ case -2212: //pbar
+ case -2112: //nbar
+ FillHistogram(Form("hParentPi0antipn_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; // pn
+ break ;
+ default: //other
+ FillHistogram(Form("hParentPi0Other_cen%d",fCentBin),p12.M(),p12.Pt(),w) ; //
+ }//switch
+ }//pi0 with primary
+ }//common parent - pi0
+ }//there is common primary
+ }//seond photon loop
+ }//first photon loop
+
+
+ //Now look at photon contaiminations
+ for (Int_t i1=0; i1<inPHOS-1; i1++) {
+ AliCaloPhoton * ph1=(AliCaloPhoton*)fCaloPhotonsPHOS->At(i1) ;
+ Int_t iprim = ph1->GetPrimary() ;
+ if(iprim<0)
+ FillAllHistograms(Form("hGammaNoPrim_cen%d",fCentBin),ph1) ; //
+ else{
+ //Find primary at vertex
+ TParticle * primPHOS = fStack->Particle(iprim) ;
+ Int_t iprimV=primPHOS->GetFirstMother();
+ TParticle * primVtx = primPHOS ;
+ while((iprimV>-1) && primVtx->R()>kRCut){
+ primVtx = fStack->Particle(iprimV) ;
+ iprimV=primVtx->GetFirstMother();
+ }
+
+ //photon
+ Int_t primPdgCode=primVtx->GetPdgCode() ;
+ switch(primPdgCode){
+ case 22: FillAllHistograms("hGammaPhot",ph1);
+ break ;
+ case 11:
+ case -11:
+ FillAllHistograms("hGammaEl",ph1);
+ break ;
+ case 111:
+ FillAllHistograms("hGammaPi0",ph1);
+ break ;
+ case 221:
+ FillAllHistograms("hGammaEta",ph1);
+ break ;
+ case 223: FillAllHistograms("hGammaOmega",ph1) ; //omega
+ break ;
+ case 211:
+ case -211:
+ FillAllHistograms("hGammaPipm",ph1);
+ //Find particle entered PHOS
+ if(primVtx == primPHOS)
+ FillAllHistograms("hGammaPipmDirect",ph1);
+ else{
+ Int_t primPdgPHOS=primPHOS->GetPdgCode() ;
+ if(primPdgPHOS==22){
+ FillAllHistograms("hGammaPipmGamma",ph1);
+ FillHistogram("hPipmGammaConvR",ph1->Pt(),primPHOS->R());
+ FillHistogram("hPipmGammaConvRZ",primPHOS->Vz(),primPHOS->R());
+ break ;
+ }
+ if(TMath::Abs(primPdgPHOS)==11){
+ FillAllHistograms("hGammaPipmEl",ph1);
+ FillHistogram("hPipmElConvR",ph1->Pt(),primPHOS->R());
+ break ;
+ }
+ if(TMath::Abs(primPdgPHOS)==2212){
+ FillAllHistograms("hGammaPipmp",ph1);
+ FillHistogram("hPipmNConvR",ph1->Pt(),primPHOS->R());
+ break ;
+ }
+ if(TMath::Abs(primPdgPHOS)==2112){
+ FillAllHistograms("hGammaPipmn",ph1);
+ FillHistogram("hPipmNConvR",ph1->Pt(),primPHOS->R());
+ break ;
+ }
+ FillAllHistograms("hGammaPipmOther",ph1);
+ FillHistogram("hPipmOtherConvR",ph1->Pt(),primPHOS->R());
+ }
+ break ;
+ case 2212: //p
+ FillAllHistograms("hGammaP",ph1);
+ break ;
+ case -2212: //pbar
+ FillAllHistograms("hGammaPbar",ph1);
+ break ;
+ case 2112: //n
+ FillAllHistograms("hGammaN",ph1);
+ break ;
+ case -2112: //nbar
+ FillAllHistograms("hGammaNbar",ph1) ; // pn
+ break ;
+ case 310: //nbar
+ FillAllHistograms("hGammaK0S",ph1) ; // pn
+ break ;
+ case 130: //nbar
+ FillAllHistograms("hGammaK0L",ph1) ; // pn
+ break ;
+ case 321: //K+
+ case -321: //K-
+ FillAllHistograms("hGammaKpm",ph1) ; // pn
+ break ;
+ case -323:
+ case 323:
+ case -313:
+ case 313: FillAllHistograms("hGammaKstar",ph1) ; // K*(892)
+ break ;
+
+ case -2224 : //Deltas
+ case 2224 : //Deltas
+ case -2214 : //Deltas
+ case 2214 : //Deltas
+ case -2114 : //Deltas
+ case 2114 : //Deltas
+ case -1114 : //Deltas
+ case 1114 : //Deltas
+ FillAllHistograms("hGammaDelta",ph1) ; // pn
+ break ;
+ default: //other
+ if(primVtx->GetPDG()->Charge())
+ FillAllHistograms("hGammaOtherCharged",ph1) ; //
+ else
+ FillAllHistograms("hGammaOtherNeutral",ph1) ; //
+ }
+ }
+
+ }//single photons
+}
+
+//_____________________________________________________________________________
+void AliAnalysisTaskPi0FlowMC::FillAllHistograms(const char * particleName,AliCaloPhoton * ph)
+{
+ //Fill All PID histograms
+
+ Double_t w=ph->GetWeight() ;
+ Double_t pt = ph->Pt() ;
+ Double_t ptC=ph->GetMomV2()->Pt() ;
+ FillHistogram(Form("%s_All_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_Allcore_cen%d",particleName,fCentBin),ptC,w) ;
+ if(ph->IsCPVOK()){
+ FillHistogram(Form("%s_CPV_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_CPVcore_cen%d",particleName,fCentBin),ptC,w) ;
+ }
+ if(ph->IsCPV2OK()){
+ FillHistogram(Form("%s_CPV2_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_CPV2core_cen%d",particleName,fCentBin),ptC,w) ;
+ }
+ if(ph->IsDispOK()){
+ FillHistogram(Form("%s_Disp_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_Dispcore_cen%d",particleName,fCentBin),ptC,w) ;
+ if(ph->IsDisp2OK()){
+ FillHistogram(Form("%s_Disp2_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_Disp2core_cen%d",particleName,fCentBin),ptC,w) ;
+ }
+ if(ph->IsCPVOK()){
+ FillHistogram(Form("%s_Both_cen%d",particleName,fCentBin),pt,w) ;
+ FillHistogram(Form("%s_Bothcore_cen%d",particleName,fCentBin),ptC,w) ;
+ }
+ }
+}
+
+
+//___________________________________________________________________________
+Double_t AliAnalysisTaskPi0FlowMC::PrimaryWeight(Int_t primary){
+ //Check who is the primary and introduce weight to correct primary spectrum
+
+ if(primary<0 || primary>=fStack->GetNtrack())
+ return 1 ;
+ //trace primaries up to IP
+ TParticle* particle = fStack->Particle(primary);
+ Double_t r=particle->R() ;
+ Int_t mother = particle->GetFirstMother() ;
+ while(mother>-1){
+ if(r<1. && particle->GetPdgCode()==111)
+ break ;
+ particle = fStack->Particle(mother);
+ mother = particle->GetFirstMother() ;
+ r=particle->R() ;
+ }
+
+ return TMath::Max(0.,PrimaryParticleWeight(particle)) ;
+}
+//________________________________________________________________________
+Double_t AliAnalysisTaskPi0FlowMC::PrimaryParticleWeight(TParticle * particle){
+ return 1.; //TODO: use weight.
+
+ Int_t pdg = particle->GetPdgCode() ;
+ Int_t type=0 ;
+ if(pdg == 111 || TMath::Abs(pdg)==211){
+ type =1 ;
+ }
+ else{
+ if(TMath::Abs(pdg)<1000){ //Kaon-like
+ type =2 ;
+ }
+ else
+ type = 3; //baryons
+ }
+
+ Double_t pt = particle->Pt() ;
+ if(type==1){
+ if(fCentBin==0) //0-5
+ return (1.662990+1.140890*pt-0.192088*pt*pt)/(1.-0.806630*pt+0.304771*pt*pt)+0.141690*pt ;
+ if(fCentBin==1) //5-10
+ return (1.474351+0.791492*pt-0.066369*pt*pt)/(1.-0.839338*pt+0.317312*pt*pt)+0.093289*pt ;
+ if(fCentBin==2) //10-20
+ return (1.174728+0.959681*pt-0.137695*pt*pt)/(1.-0.788873*pt+0.299538*pt*pt)+0.128759*pt ;
+ if(fCentBin==3) //20-40
+ return (0.927335+0.475349*pt+0.004364*pt*pt)/(1.-0.817966*pt+0.309787*pt*pt)+0.086899*pt ;
+ if(fCentBin==4) //40-60
+ return (0.676878+0.190680*pt+0.077031*pt*pt)/(1.-0.790623*pt+0.305183*pt*pt)+0.064510*pt ;
+ if(fCentBin==5) //60-80
+ return (0.684726-0.606262*pt+0.409963*pt*pt)/(1.-1.080061*pt+0.456933*pt*pt)+0.005151*pt ;
+ }
+ if(type==2){
+ if(fCentBin==0) //0-5
+ return (-0.417131+2.253936*pt-0.337731*pt*pt)/(1.-0.909892*pt+0.316820*pt*pt)+0.157312*pt ;
+ if(fCentBin==1) //5-10
+ return (-0.352275+1.844466*pt-0.248598*pt*pt)/(1.-0.897048*pt+0.316462*pt*pt)+0.132461*pt ;
+ if(fCentBin==2) //10-20
+ return (-0.475481+1.975108*pt-0.336013*pt*pt)/(1.-0.801028*pt+0.276705*pt*pt)+0.188164*pt ;
+ if(fCentBin==3) //20-40
+ return (-0.198954+1.068789*pt-0.103540*pt*pt)/(1.-0.848354*pt+0.299209*pt*pt)+0.112939*pt ;
+ if(fCentBin==4) //40-60
+ return (-0.111052+0.664041*pt-0.019717*pt*pt)/(1.-0.804916*pt+0.300779*pt*pt)+0.095784*pt ;
+ if(fCentBin==5) //0-5
+ return (0.202788-0.439832*pt+0.564585*pt*pt)/(1.-1.254029*pt+0.679444*pt*pt)+0.016235*pt ;
+ }
+ if(type==3){
+ if(fCentBin==0) //0-5
+ return (-1.312732+2.743568*pt-0.375775*pt*pt)/(1.-0.717533*pt+0.164694*pt*pt)+0.164445*pt ;
+ if(fCentBin==1) //5-10
+ return (-1.229425+2.585889*pt-0.330164*pt*pt)/(1.-0.715892*pt+0.167386*pt*pt)+0.133085*pt ;
+ if(fCentBin==2) //10-20
+ return (-1.135677+2.397489*pt-0.320355*pt*pt)/(1.-0.709312*pt+0.164350*pt*pt)+0.146095*pt ;
+ if(fCentBin==3) //20-40
+ return (-0.889993+1.928263*pt-0.220785*pt*pt)/(1.-0.715991*pt+0.174729*pt*pt)+0.095098*pt ;
+ if(fCentBin==4) //40-60
+ return (-0.539237+1.329118*pt-0.115439*pt*pt)/(1.-0.722906*pt+0.186832*pt*pt)+0.059267*pt ;
+ if(fCentBin==5) //60-80
+ return (-0.518126+1.327628*pt-0.130881*pt*pt)/(1.-0.665649*pt+0.184300*pt*pt)+0.081701*pt ;
+ }
+ return 1. ;
+}
+
+//___________________________________________________________________________
+AliStack* AliAnalysisTaskPi0FlowMC::GetMCStack()
+{
+ fStack = 0;
+ AliVEventHandler* eventHandler = AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler();
+ if(eventHandler){
+ AliMCEventHandler* mcEventHandler = dynamic_cast<AliMCEventHandler*> (eventHandler);
+ if( mcEventHandler)
+ fStack = static_cast<AliMCEventHandler*>(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler())->MCEvent()->Stack();
+ }
+ return fStack;
+}
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff