1) K0S histogram is added

2) Primary particle selection now adds e+ and e- from any vertices (Boris)

diff --git a/PWGGA/PHOSTasks/PHOS_PbPb/AliAnalysisTaskPi0FlowMC.cxx b/PWGGA/PHOSTasks/PHOS_PbPb/AliAnalysisTaskPi0FlowMC.cxx
index afd5f5b54b3f6c046db145392ad9f69623e90c69..3b5e866c54bd4191718a6bcda0bfb2018ed7e007 100644 (file)
--- a/PWGGA/PHOSTasks/PHOS_PbPb/AliAnalysisTaskPi0FlowMC.cxx
+++ b/PWGGA/PHOSTasks/PHOS_PbPb/AliAnalysisTaskPi0FlowMC.cxx
@@ -63,6 +63,7 @@
 #include "AliOADBContainer.h"
 
 
+#include "AliAnalysisTaskPi0Flow.h"
 #include "AliAnalysisTaskPi0FlowMC.h"
 
 ClassImp(AliAnalysisTaskPi0FlowMC);
@@ -117,6 +118,13 @@ void AliAnalysisTaskPi0FlowMC::UserCreateOutputObjects()
     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.)) ;
+
+    snprintf(key,55,"hMC_all_K0S_cen%d",cent) ;
+    fOutputContainer->Add(new TH1F(key,"Rapidity photon",250,0.,25.)) ;
+
+    snprintf(key,55,"hMC_unitEta_K0S_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.)) ;
@@ -256,7 +264,14 @@ void AliAnalysisTaskPi0FlowMC::SelectPhotonClusters()
     Int_t primary=FindPrimary(cluster,sure) ;
     photon->SetPrimary(primary);
     photon->SetWeight(PrimaryWeight(primary)) ;
-  }
+
+    TParticle* p = fStack->Particle(primary);
+    
+    if(p->R() >kRCut) {
+      if(p->GetPdgCode()==11 || p->GetPdgCode()==-11) continue;
+      else { fCaloPhotonsPHOS->Remove(photon); fCaloPhotonsPHOS->Compress(); } 
+    } 
+  }  
 }
 
 void AliAnalysisTaskPi0FlowMC::FillSelectedClusterHistograms()
@@ -933,7 +948,10 @@ void AliAnalysisTaskPi0FlowMC::FillMCHist(){
         if(particle->GetPdgCode() == kGamma)
            snprintf(partName,10,"gamma") ;
        else
-           continue ;
+         if(particle->GetPdgCode() == 310)
+           snprintf(partName,10,"K0S") ;
+         else
+           continue ;
 
     //Primary particle
     Double_t r=particle->R() ;
@@ -944,23 +962,29 @@ void AliAnalysisTaskPi0FlowMC::FillMCHist(){
     if(r >kRCut)
       continue ;
 
+    Double_t phi=particle->Phi() ;
+    while(phi<0.)phi+=TMath::TwoPi() ;
+    while(phi>TMath::TwoPi())phi-=TMath::TwoPi() ;
+
+    Double_t phig = 180./TMath::Pi()*phi; // phi in deg
+    
     //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())<1.){
+    
+    if(TMath::Abs(particle->Y())<0.135 && phig>260. && phig<320.){
       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) ;
+    
+      snprintf(hkey,55,"hMC_phi_%s_cen%d",partName,fCentBin) ;
+      FillHistogram(hkey,phi,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) ;
   }
 }