]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGHF/hfe/AliAnalysisTaskHFECal.cxx
update
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.cxx
index 3a1cd20d5ef1a325530256d9d51799da90e89adf..706cf45386ff3767b8afdee085f1ae384c340c1a 100644 (file)
@@ -86,9 +86,10 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal(const char *name)
   : AliAnalysisTaskSE(name)
   ,fESD(0)
   ,fMC(0)
+  ,stack(0)
   ,fGeom(0)
   ,fOutputList(0)
-  ,fqahist(1
+  ,fqahist(0
   ,fTrackCuts(0)
   ,fCuts(0)
   ,fIdentifiedAsOutInz(kFALSE)
@@ -112,6 +113,16 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal(const char *name)
   ,fIncpTM20(0)        
   ,fInvmassLS(0)               
   ,fInvmassULS(0)              
+  ,fInvmassLSmc(0)             
+  ,fInvmassULSmc(0)            
+  ,fInvmassLSmc0(0)            
+  ,fInvmassLSmc1(0)            
+  ,fInvmassLSmc2(0)            
+  ,fInvmassLSmc3(0)            
+  ,fInvmassULSmc0(0)           
+  ,fInvmassULSmc1(0)           
+  ,fInvmassULSmc2(0)           
+  ,fInvmassULSmc3(0)           
   ,fOpeningAngleLS(0)  
   ,fOpeningAngleULS(0) 
   ,fPhotoElecPt(0)
@@ -142,12 +153,26 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal(const char *name)
   ,fIncpTMChfeAll(0)   
   ,fIncpTMCM20hfe(0)   
   ,fIncpTMCM20hfeAll(0)        
+  ,fIncpTMCM20hfeCheck(0)      
   ,fIncpTMCpho(0)      
   ,fIncpTMCM20pho(0)   
   ,fPhoElecPtMC(0)
   ,fPhoElecPtMCM20(0)
   ,fSameElecPtMC(0)
   ,fSameElecPtMCM20(0)
+  ,fIncpTMCM20pho_pi0e(0)      
+  ,fPhoElecPtMCM20_pi0e(0)
+  ,fSameElecPtMCM20_pi0e(0)
+  ,fIncpTMCM20pho_eta(0)       
+  ,fPhoElecPtMCM20_eta(0)
+  ,fSameElecPtMCM20_eta(0)
+  ,CheckNclust(0)
+  ,CheckNits(0)
+  ,Hpi0pTcheck(0)
+  ,HETApTcheck(0)
+  ,HphopTcheck(0)
+  ,fpTCheck(0)
+  ,fMomDtoE(0) 
 {
   //Named constructor
   
@@ -169,9 +194,10 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal()
   : AliAnalysisTaskSE("DefaultAnalysis_AliAnalysisTaskHFECal")
   ,fESD(0)
   ,fMC(0)
+  ,stack(0)
   ,fGeom(0)
   ,fOutputList(0)
-  ,fqahist(1)
+  ,fqahist(0)
   ,fTrackCuts(0)
   ,fCuts(0)
   ,fIdentifiedAsOutInz(kFALSE)
@@ -195,6 +221,16 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal()
   ,fIncpTM20(0)         
   ,fInvmassLS(0)               
   ,fInvmassULS(0)              
+  ,fInvmassLSmc(0)             
+  ,fInvmassULSmc(0)            
+  ,fInvmassLSmc0(0)            
+  ,fInvmassLSmc1(0)            
+  ,fInvmassLSmc2(0)            
+  ,fInvmassLSmc3(0)            
+  ,fInvmassULSmc0(0)           
+  ,fInvmassULSmc1(0)           
+  ,fInvmassULSmc2(0)           
+  ,fInvmassULSmc3(0)           
   ,fOpeningAngleLS(0)  
   ,fOpeningAngleULS(0) 
   ,fPhotoElecPt(0)
@@ -225,12 +261,26 @@ AliAnalysisTaskHFECal::AliAnalysisTaskHFECal()
   ,fIncpTMChfeAll(0)   
   ,fIncpTMCM20hfe(0)   
   ,fIncpTMCM20hfeAll(0)        
+  ,fIncpTMCM20hfeCheck(0)      
   ,fIncpTMCpho(0)      
   ,fIncpTMCM20pho(0)   
   ,fPhoElecPtMC(0)
   ,fPhoElecPtMCM20(0)
   ,fSameElecPtMC(0)
   ,fSameElecPtMCM20(0)
+  ,fIncpTMCM20pho_pi0e(0)      
+  ,fPhoElecPtMCM20_pi0e(0)
+  ,fSameElecPtMCM20_pi0e(0)
+  ,fIncpTMCM20pho_eta(0)       
+  ,fPhoElecPtMCM20_eta(0)
+  ,fSameElecPtMCM20_eta(0)
+  ,CheckNclust(0)
+  ,CheckNits(0)
+  ,Hpi0pTcheck(0)
+  ,HETApTcheck(0)
+  ,HphopTcheck(0)
+  ,fpTCheck(0)
+  ,fMomDtoE(0)
 {
        //Default constructor
        fPID = new AliHFEpid("hfePid");
@@ -256,6 +306,7 @@ AliAnalysisTaskHFECal::~AliAnalysisTaskHFECal()
   delete fOutputList;
   delete fGeom;
   delete fPID;
+  delete fCuts;
   delete fCFM;
   delete fPIDqa;
   delete fTrackCuts;
@@ -286,7 +337,7 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
   }
  
   if(fmcData)fMC = MCEvent();
-  AliStack* stack = NULL;
+  //AliStack* stack = NULL;
   if(fmcData && fMC)stack = fMC->Stack();
 
   Float_t cent = -1.;
@@ -308,6 +359,19 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
       Bool_t mcInDtoE= kFALSE;
       Bool_t mcInBtoE= kFALSE;
 
+      Bool_t MChijing = fMC->IsFromBGEvent(iParticle);
+      //if(!MChijing)printf("not MC hijing");
+      int iHijing = 1;
+      if(!MChijing)iHijing = 0;
+      if(fPDG==111)Hpi0pTcheck->Fill(pTMC,iHijing);
+      if(fPDG==221)HETApTcheck->Fill(pTMC,iHijing);
+
+      if(particle->GetFirstMother()>-1 && fPDG==22)
+        {
+        int parentPID = stack->Particle(particle->GetFirstMother())->GetPdgCode();  
+         if(parentPID==111 || parentPID==221)HphopTcheck->Fill(pTMC,iHijing); // pi0->g & eta->g
+        } 
+
       if(particle->GetFirstMother()>-1 && fabs(fPDG)==11)
         {
            int parentPID = stack->Particle(particle->GetFirstMother())->GetPdgCode();  
@@ -383,32 +447,116 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
       continue;
     }
    
+    int parentlabel = 99999;
+    int parentPID = 99999;
+    int grand_parentlabel = 99999;
+    int grand_parentPID = 99999;
     Bool_t mcPho = kFALSE;
     Bool_t mcDtoE= kFALSE;
     Bool_t mcBtoE= kFALSE;
+    Bool_t mcOrgPi0 = kFALSE;
+    Bool_t mcOrgEta = kFALSE;
     double mcele = -1.;
     double mcpT = 0.0;
     double mcMompT = 0.0;
+    //double mcGrandMompT = 0.0;
+    double mcWeight = -10.0;
+
+    int iHijing = 1;
+
     if(fmcData && fMC && stack)
       {
        Int_t label = TMath::Abs(track->GetLabel());
+
+       Bool_t MChijing = fMC->IsFromBGEvent(label);
+       if(!MChijing)iHijing = 0;
+
        TParticle* particle = stack->Particle(label);
        int mcpid = particle->GetPdgCode();
        mcpT = particle->Pt();
        //printf("MCpid = %d",mcpid);
        if(particle->GetFirstMother()>-1)
          {
-          int parentPID = stack->Particle(particle->GetFirstMother())->GetPdgCode();
-          mcMompT = stack->Particle(particle->GetFirstMother())->Pt();
+          //int parentlabel = particle->GetFirstMother();
+          //int parentPID = stack->Particle(particle->GetFirstMother())->GetPdgCode();
+          //mcMompT = stack->Particle(particle->GetFirstMother())->Pt();
+          FindMother(particle, parentlabel, parentPID);
+          mcMompT = stack->Particle(parentlabel)->Pt();
           if((parentPID==22 || parentPID==111 || parentPID==221)&& fabs(mcpid)==11)mcPho = kTRUE;
           if((fabs(parentPID)==411 || fabs(parentPID)==413 || fabs(parentPID)==421 || fabs(parentPID)==423 || fabs(parentPID)==431)&& fabs(mcpid)==11)mcDtoE = kTRUE;
           if((fabs(parentPID)==511 || fabs(parentPID)==513 || fabs(parentPID)==521 || fabs(parentPID)==523 || fabs(parentPID)==531)&& fabs(mcpid)==11)mcBtoE = kTRUE;
+
+          // make D->e pT correlation
+          if(mcDtoE)fMomDtoE->Fill(mcpT,mcMompT); 
+
+           //cout << "check PID = " << parentPID << endl;
+           //cout << "check pho = " << mcPho << endl;
+           //cout << "check D or B = " << mcDtoE << endl;
+          // pi->e (Dalitz)
+          if(parentPID==111 && fabs(mcpid)==11 && mcMompT>0.0)
+              {
+               //cout << "find pi0->e " <<  endl;
+               mcOrgPi0 = kTRUE;
+               mcWeight = GetMCweight(mcMompT); 
+              }
+          // eta->e (Dalitz)
+          if(parentPID==221 && fabs(mcpid)==11 && mcMompT>0.0)
+              {
+               //cout << "find Eta->e " <<  endl;
+               mcOrgEta = kTRUE;
+               mcWeight = GetMCweightEta(mcMompT); 
+              }
+
+          // access grand parent 
+          TParticle* particle_parent = stack->Particle(parentlabel); // get parent pointer
+          //if(particle_parent->GetFirstMother()>-1 && parentPID==22 && fabs(mcpid)==11) // get grand parent g->e
+          if(particle_parent->GetFirstMother()>-1 && (parentPID==22 || parentPID==111) && fabs(mcpid)==11) // get grand parent g->e
+            {
+             //int grand_parentPID = stack->Particle(particle_parent->GetFirstMother())->GetPdgCode();
+             //double pTtmp = stack->Particle(particle_parent->GetFirstMother())->Pt();
+             FindMother(particle_parent, grand_parentlabel, grand_parentPID);
+             double mcGrandpT = stack->Particle(grand_parentlabel)->Pt();
+             if(grand_parentPID==111 && mcGrandpT>0.0)
+                {
+                  // check eta->pi0 decay !
+                   int grand2_parentlabel = 99999; int grand2_parentPID = 99999;
+                   TParticle* particle_grand = stack->Particle(grand_parentlabel); // get parent pointer
+                   FindMother(particle_grand, grand2_parentlabel, grand2_parentPID);
+                   if(grand2_parentPID==221)
+                     {
+                      //cout << "find Eta->e " <<  endl;
+                      double mcGrandpT2 = stack->Particle(grand2_parentlabel)->Pt();
+                      mcOrgEta = kTRUE;
+                      mcWeight = GetMCweight(mcGrandpT2);  
+                      mcMompT = mcGrandpT2; 
+                     }
+                   else
+                     {
+                      //cout << "find pi0->e " <<  endl;
+                      mcOrgPi0 = kTRUE;
+                      mcWeight = GetMCweight(mcGrandpT);  
+                      mcMompT = mcGrandpT; 
+                     }
+                }
+
+             if(grand_parentPID==221 && mcGrandpT>0.0)
+                {
+                   //cout << "find Eta->e " <<  endl;
+                   mcOrgEta = kTRUE;
+                   mcOrgPi0 = kFALSE;
+                   mcWeight = GetMCweightEta(mcGrandpT); 
+                   mcMompT = mcGrandpT; 
+                }
+            }
          } 
        if(fabs(mcpid)==11 && mcDtoE)mcele= 1.; 
        if(fabs(mcpid)==11 && mcBtoE)mcele= 2.; 
        if(fabs(mcpid)==11 && mcPho)mcele= 3.; 
       }
  
+    //cout << "Pi0 = " << mcOrgPi0 << " ; Eta = " << mcOrgEta << endl; 
+    //printf("weight = %f\n",mcWeight);
+
     if(TMath::Abs(track->Eta())>0.6) continue;
     if(TMath::Abs(track->Pt()<2.0)) continue;
     
@@ -430,6 +578,9 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
     // HFE cuts: TPC PID cleanup
     if(!ProcessCutStep(AliHFEcuts::kStepHFEcutsTPC, track)) continue;
 
+    int nTPCcl = track->GetTPCNcls();
+    int nTPCclF = track->GetTPCNclsF();
+    int nITS = track->GetNcls(0);
     
     fTrackPtAftTrkCuts->Fill(track->Pt());             
     
@@ -477,23 +628,29 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
 
                if(fTPCnSigma>-1.5 && fTPCnSigma<3.0)
                {
-                 SelectPhotonicElectron(iTracks,cent,track,fFlagPhotonicElec,fFlagConvinatElec,fTPCnSigma,m20,eop,mcele);
+                 SelectPhotonicElectron(iTracks,cent,track,fFlagPhotonicElec,fFlagConvinatElec,fTPCnSigma,m20,eop,mcele,mcWeight,iHijing,mcOrgPi0,mcOrgEta);
                }
                if(fFlagPhotonicElec)oppstatus = 1.0;
                if(fFlagConvinatElec)oppstatus = 2.0;
                if(fFlagPhotonicElec && fFlagConvinatElec)oppstatus = 3.0;
 
                  double valdedx[16];
-                 valdedx[0] = pt; valdedx[1] = dEdx; valdedx[2] = phi; valdedx[3] = eta; valdedx[4] = fTPCnSigma;
-                 valdedx[5] = eop; valdedx[6] = rmatch; valdedx[7] = ncells,  valdedx[8] = m02; valdedx[9] = m20; valdedx[10] = mcpT;
-                 valdedx[11] = cent; valdedx[12] = charge; valdedx[13] = oppstatus; valdedx[14] = clust->Chi2();
+                 valdedx[0] = pt; valdedx[1] = nITS; valdedx[2] = phi; valdedx[3] = eta; valdedx[4] = fTPCnSigma;
+                 valdedx[5] = eop; valdedx[6] = rmatch; valdedx[7] = ncells,  valdedx[8] = nTPCclF; valdedx[9] = m20; valdedx[10] = mcpT;
+                 valdedx[11] = cent; valdedx[12] = charge; valdedx[13] = oppstatus; valdedx[14] = nTPCcl;
                   valdedx[15] = mcele;
                   if(fqahist==1)fEleInfo->Fill(valdedx);
                  
 
       }
     }
-        
+     
+    if(nITS<2.5)continue;
+    if(nTPCcl<100)continue;
+   
+    CheckNclust->Fill(nTPCcl); 
+    CheckNits->Fill(nITS); 
+
     fdEdxBef->Fill(mom,fTPCnSigma);
     fTPCnsigma->Fill(mom,fTPCnSigma);
     if(fTPCnSigma >= -1.0 && fTPCnSigma <= 3)fTrkEovPBef->Fill(pt,eop);
@@ -540,23 +697,27 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
      }
     
     // MC
-    if(mcele>0)
+    if(mcele>0) // select MC electrons
       {
 
           fIncpTMChfeAll->Fill(cent,pt);    
           if(m20>0.0 && m20<0.3)fIncpTMCM20hfeAll->Fill(cent,pt);    
 
-       if(mcBtoE || mcDtoE)
+       if(mcBtoE || mcDtoE) // select B->e & D->e
          {
           fIncpTMChfe->Fill(cent,pt);    
           if(m20>0.0 && m20<0.3)fIncpTMCM20hfe->Fill(cent,pt);    
+          if(m20>0.0 && m20<0.3)fIncpTMCM20hfeCheck->Fill(cent,mcpT);    
          }
-       if(mcPho)
+      
+       if(mcPho) // select photonic electrons
         {
-         double phoval[3];
+         double phoval[5];
          phoval[0] = cent;
          phoval[1] = pt;
          phoval[2] = fTPCnSigma;
+         phoval[3] = iHijing;
+         phoval[4] = mcMompT;
 
          fIncpTMCpho->Fill(phoval);    
          if(fFlagPhotonicElec) fPhoElecPtMC->Fill(phoval);
@@ -567,6 +728,30 @@ void AliAnalysisTaskHFECal::UserExec(Option_t*)
             fIncpTMCM20pho->Fill(phoval);    
             if(fFlagPhotonicElec) fPhoElecPtMCM20->Fill(phoval);
             if(fFlagConvinatElec) fSameElecPtMCM20->Fill(phoval);
+            // pi0->g->e
+            if(mcWeight>-1)
+              {
+               if(iHijing==1)mcWeight = 1.0; 
+               if(mcOrgPi0)
+                 {
+                  //fIncpTMCM20pho_pi0e->Fill(phoval,mcWeight);    
+                  //if(fFlagPhotonicElec) fPhoElecPtMCM20_pi0e->Fill(phoval,mcWeight);
+                  //if(fFlagConvinatElec) fSameElecPtMCM20_pi0e->Fill(phoval,mcWeight);
+                  fIncpTMCM20pho_pi0e->Fill(phoval);    
+                  if(fFlagPhotonicElec) fPhoElecPtMCM20_pi0e->Fill(phoval);
+                  if(fFlagConvinatElec) fSameElecPtMCM20_pi0e->Fill(phoval);
+                 }
+               if(mcOrgEta)
+                 {
+                  //fIncpTMCM20pho_eta->Fill(phoval,mcWeight);    
+                  //if(fFlagPhotonicElec) fPhoElecPtMCM20_eta->Fill(phoval,mcWeight);
+                  //if(fFlagConvinatElec) fSameElecPtMCM20_eta->Fill(phoval,mcWeight);
+                  fIncpTMCM20pho_eta->Fill(phoval);    
+                  if(fFlagPhotonicElec) fPhoElecPtMCM20_eta->Fill(phoval);
+                  if(fFlagConvinatElec) fSameElecPtMCM20_eta->Fill(phoval);
+                 }
+               // --- eta
+              }
            }
         } 
       } 
@@ -623,7 +808,23 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   fPID->SortDetectors(); 
   fPIDqa = new AliHFEpidQAmanager();
   fPIDqa->Initialize(fPID);
-  
+  //------- fcut --------------  
+  fCuts = new AliHFEcuts();
+  fCuts->CreateStandardCuts();
+  //fCuts->SetMinNClustersTPC(100);
+  fCuts->SetMinNClustersTPC(90);
+  fCuts->SetMinRatioTPCclusters(0.6);
+  fCuts->SetTPCmodes(AliHFEextraCuts::kFound, AliHFEextraCuts::kFoundOverFindable);
+  //fCuts->SetMinNClustersITS(3);
+  fCuts->SetMinNClustersITS(2);
+  fCuts->SetCutITSpixel(AliHFEextraCuts::kAny);
+  fCuts->SetCheckITSLayerStatus(kFALSE);
+  fCuts->SetVertexRange(10.);
+  fCuts->SetTOFPIDStep(kFALSE);
+  fCuts->SetPtRange(2, 50);
+  fCuts->SetMaxImpactParam(3.,3.);
+
   //--------Initialize correction Framework and Cuts
   fCFM = new AliCFManager;
   const Int_t kNcutSteps = AliHFEcuts::kNcutStepsMCTrack + AliHFEcuts::kNcutStepsRecTrack + AliHFEcuts::kNcutStepsDETrack;
@@ -646,7 +847,7 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   fNoEvents = new TH1F("fNoEvents","",4,-0.5,3.5) ;
   fOutputList->Add(fNoEvents);
   
-  Int_t binsE[5] =    {250, 100,  1000, 100,   10};
+  Int_t binsE[5] =    {250, 100,  1000, 200,   10};
   Double_t xminE[5] = {1.0,  -1,   0.0,   0, -0.5}; 
   Double_t xmaxE[5] = {3.5,   1, 100.0, 100,  9.5}; 
   fEMCAccE = new THnSparseD("fEMCAccE","EMC acceptance & E;#eta;#phi;Energy;Centrality;trugCondition;",5,binsE,xminE,xmaxE);
@@ -676,22 +877,60 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   fdEdxAft = new TH2F("fdEdxAft","track dEdx vs p after HFE pid",100,0,50,200,-10,10);
   fOutputList->Add(fdEdxAft);
   
-  fIncpT = new TH2F("fIncpT","HFE pid electro vs. centrality",100,0,100,100,0,50);
+  fIncpT = new TH2F("fIncpT","HFE pid electro vs. centrality",200,0,100,100,0,50);
   fOutputList->Add(fIncpT);
 
-  fIncpTM20 = new TH2F("fIncpTM20","HFE pid electro vs. centrality with M20",100,0,100,100,0,50);
+  fIncpTM20 = new TH2F("fIncpTM20","HFE pid electro vs. centrality with M20",200,0,100,100,0,50);
   fOutputList->Add(fIncpTM20);
   
-  Int_t nBinspho[9] =  { 100, 100, 500, 12,   50,    4, 200,   8, 100};
+  Int_t nBinspho[9] =  { 200, 100, 500, 12,   50,    4, 200,   8, 100};
   Double_t minpho[9] = {  0.,  0.,  0., -2.5,  0, -0.5,   0,-1.5,   0};   
   Double_t maxpho[9] = {100., 50., 0.5, 3.5,   1,  3.5,   2, 6.5,  50};   
 
   fInvmassLS = new THnSparseD("fInvmassLS", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2); nSigma; angle; m20cut; eop; Mcele;", 9, nBinspho,minpho, maxpho);
-  fOutputList->Add(fInvmassLS);
+  if(fqahist==1)fOutputList->Add(fInvmassLS);
   
   fInvmassULS = new THnSparseD("fInvmassULS", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2); nSigma; angle; m20cut; eop; MCele", 9, nBinspho,minpho, maxpho);
-  fOutputList->Add(fInvmassULS);
+  if(fqahist==1)fOutputList->Add(fInvmassULS);
+  
+  fInvmassLSmc = new THnSparseD("fInvmassLSmc", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2); nSigma; angle; m20cut; eop; Mcele;", 9, nBinspho,minpho, maxpho);
+  if(fqahist==1)fOutputList->Add(fInvmassLSmc);
+  
+  fInvmassULSmc = new THnSparseD("fInvmassULSmc", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2); nSigma; angle; m20cut; eop; MCele", 9, nBinspho,minpho, maxpho);
+  if(fqahist==1)fOutputList->Add(fInvmassULSmc);
+
+  fInvmassLSmc0 = new TH2D("fInvmassLSmc0", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassLSmc0->Sumw2();
+  fOutputList->Add(fInvmassLSmc0);
   
+  fInvmassLSmc1 = new TH2D("fInvmassLSmc1", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassLSmc1->Sumw2();
+  fOutputList->Add(fInvmassLSmc1);
+
+  fInvmassLSmc2 = new TH2D("fInvmassLSmc2", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassLSmc2->Sumw2();
+  fOutputList->Add(fInvmassLSmc2);
+
+  fInvmassLSmc3 = new TH2D("fInvmassLSmc3", "Inv mass of LS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassLSmc3->Sumw2();
+  fOutputList->Add(fInvmassLSmc3);
+
+  fInvmassULSmc0 = new TH2D("fInvmassULSmc0", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassULSmc0->Sumw2();
+  fOutputList->Add(fInvmassULSmc0);
+
+  fInvmassULSmc1 = new TH2D("fInvmassULSmc1", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassULSmc1->Sumw2();
+  fOutputList->Add(fInvmassULSmc1);
+
+  fInvmassULSmc2 = new TH2D("fInvmassULSmc2", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassULSmc2->Sumw2();
+  fOutputList->Add(fInvmassULSmc2);
+
+  fInvmassULSmc3 = new TH2D("fInvmassULSmc3", "Inv mass of ULS (e,e); cent; p_{T} (GeV/c); mass(GeV/c^2)",20,0,20,500,0,0.5 );
+  fInvmassULSmc3->Sumw2();
+  fOutputList->Add(fInvmassULSmc3);
+
   fOpeningAngleLS = new TH1F("fOpeningAngleLS","Opening angle for LS pairs",100,0,1);
   fOutputList->Add(fOpeningAngleLS);
   
@@ -701,32 +940,32 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   fPhotoElecPt = new TH1F("fPhotoElecPt", "photonic electron pt",100,0,50);
   fOutputList->Add(fPhotoElecPt);
   
-  fPhoElecPt = new TH2F("fPhoElecPt", "Pho-inclusive electron pt",100,0,100,100,0,50);
+  fPhoElecPt = new TH2F("fPhoElecPt", "Pho-inclusive electron pt",200,0,100,100,0,50);
   fOutputList->Add(fPhoElecPt);
   
-  fPhoElecPtM20 = new TH2F("fPhoElecPtM20", "Pho-inclusive electron pt with M20",100,0,100,100,0,50);
+  fPhoElecPtM20 = new TH2F("fPhoElecPtM20", "Pho-inclusive electron pt with M20",200,0,100,100,0,50);
   fOutputList->Add(fPhoElecPtM20);
 
-  fSameElecPt = new TH2F("fSameElecPt", "Same-inclusive electron pt",100,0,100,100,0,50);
+  fSameElecPt = new TH2F("fSameElecPt", "Same-inclusive electron pt",200,0,100,100,0,50);
   fOutputList->Add(fSameElecPt);
 
-  fSameElecPtM20 = new TH2F("fSameElecPtM20", "Same-inclusive electron pt with M20",100,0,100,100,0,50);
+  fSameElecPtM20 = new TH2F("fSameElecPtM20", "Same-inclusive electron pt with M20",200,0,100,100,0,50);
   fOutputList->Add(fSameElecPtM20);
 
-  fCent = new TH1F("fCent","Centrality",100,0,100) ;
+  fCent = new TH1F("fCent","Centrality",200,0,100) ;
   fOutputList->Add(fCent);
  
   // Make common binning
   const Double_t kMinP = 0.;
   const Double_t kMaxP = 50.;
-  const Double_t kTPCSigMim = 40.;
-  const Double_t kTPCSigMax = 140.;
+  //const Double_t kTPCSigMim = 40.;
+  //const Double_t kTPCSigMax = 140.;
 
   // 1st histogram: TPC dEdx with/without EMCAL (p, pT, TPC Signal, phi, eta,  Sig,  e/p,  ,match, cell, M02, M20, Disp, Centrality, select)
-  Int_t nBins[16] =  {  250,        200,   60,    20,   100,  300,  50,   40,   200, 200,  250, 100,    3,    5,   10,    8};
-  Double_t min[16] = {kMinP,  kTPCSigMim, 1.0,  -1.0,  -6.0,    0,   0,    0,   0.0, 0.0,  0.0,   0, -1.5, -0.5, -0.5, -1.5};
-  Double_t max[16] = {kMaxP,  kTPCSigMax, 4.0,   1.0,   4.0,  3.0, 0.1,   40,   2.0, 2.0, 50.0, 100,  1.5,  4.5,  9.5,  6.5};
-  fEleInfo = new THnSparseD("fEleInfo", "Electron Info; pT [GeV/c]; TPC signal;phi;eta;nSig; E/p;Rmatch;Ncell;M02;M20;mcpT;Centrality;charge;opp;same;trigCond;MCele", 16, nBins, min, max);
+  Int_t nBins[16] =  {  250,    10,   60,    20,   100,  300,  50,   40,   200, 200,  250, 200,    3,    5, 100,    8};
+  Double_t min[16] = {kMinP,  -0.5, 1.0,  -1.0,  -6.0,    0,   0,    0,   0.0, 0.0,  0.0,   0, -1.5, -0.5,  80, -1.5};
+  Double_t max[16] = {kMaxP,   9.5, 4.0,   1.0,   4.0,  3.0, 0.1,   40,   200, 2.0, 50.0, 100,  1.5,  4.5, 180,  6.5};
+  fEleInfo = new THnSparseD("fEleInfo", "Electron Info; pT [GeV/c]; TPC signal;phi;eta;nSig; E/p;Rmatch;Ncell;clsF;M20;mcpT;Centrality;charge;opp;same;trigCond;MCele", 16, nBins, min, max);
   if(fqahist==1)fOutputList->Add(fEleInfo);
 
   //<---  Trigger info
@@ -763,47 +1002,94 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   */
   //<---------- MC
 
-  fInputHFEMC = new TH2F("fInputHFEMC","Input MC HFE pid electro vs. centrality",100,0,100,100,0,50);
+  fInputHFEMC = new TH2F("fInputHFEMC","Input MC HFE pid electro vs. centrality",200,0,100,100,0,50);
   fOutputList->Add(fInputHFEMC);
 
-  fInputAlle = new TH2F("fInputAlle","Input MC electro vs. centrality",100,0,100,100,0,50);
+  fInputAlle = new TH2F("fInputAlle","Input MC electro vs. centrality",200,0,100,100,0,50);
   fOutputList->Add(fInputAlle);
 
-  fIncpTMChfe = new TH2F("fIncpTMChfe","MC HFE pid electro vs. centrality",100,0,100,100,0,50);
+  fIncpTMChfe = new TH2F("fIncpTMChfe","MC HFE pid electro vs. centrality",200,0,100,100,0,50);
   fOutputList->Add(fIncpTMChfe);
 
-  fIncpTMChfeAll = new TH2F("fIncpTMChfeAll","MC Alle pid electro vs. centrality",100,0,100,100,0,50);
+  fIncpTMChfeAll = new TH2F("fIncpTMChfeAll","MC Alle pid electro vs. centrality",200,0,100,100,0,50);
   fOutputList->Add(fIncpTMChfeAll);
 
-  fIncpTMCM20hfe = new TH2F("fIncpTMCM20hfe","MC HFE pid electro vs. centrality with M20",100,0,100,100,0,50);
+  fIncpTMCM20hfe = new TH2F("fIncpTMCM20hfe","MC HFE pid electro vs. centrality with M20",200,0,100,100,0,50);
   fOutputList->Add(fIncpTMCM20hfe);
 
-  fIncpTMCM20hfeAll = new TH2F("fIncpTMCM20hfeAll","MC Alle pid electro vs. centrality with M20",100,0,100,100,0,50);
+  fIncpTMCM20hfeAll = new TH2F("fIncpTMCM20hfeAll","MC Alle pid electro vs. centrality with M20",200,0,100,100,0,50);
   fOutputList->Add(fIncpTMCM20hfeAll);
 
+  fIncpTMCM20hfeCheck = new TH2F("fIncpTMCM20hfeCheck","MC HFE pid electro vs. centrality with M20 Check",200,0,100,100,0,50);
+  fOutputList->Add(fIncpTMCM20hfeCheck);
 
-  Int_t nBinspho2[3] =  { 100, 100,    7};
-  Double_t minpho2[3] = {  0.,  0., -2.5};   
-  Double_t maxpho2[3] = {100., 50.,  4.5};   
+  Int_t nBinspho2[5] =  { 200, 100,    7,    3, 700};
+  Double_t minpho2[5] = {  0.,  0., -2.5, -0.5, 0.};   
+  Double_t maxpho2[5] = {100., 50.,  4.5,  2.5, 70.};   
 
-  fIncpTMCpho = new THnSparseD("fIncpTMCpho","MC Pho pid electro vs. centrality",3,nBinspho2,minpho2,maxpho2);
+  fIncpTMCpho = new THnSparseD("fIncpTMCpho","MC Pho pid electro vs. centrality",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fIncpTMCpho);
 
-  fIncpTMCM20pho = new THnSparseD("fIncpTMCM20pho","MC Pho pid electro vs. centrality with M20",3,nBinspho2,minpho2,maxpho2);
+  fIncpTMCM20pho = new THnSparseD("fIncpTMCM20pho","MC Pho pid electro vs. centrality with M20",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fIncpTMCM20pho);
 
-  fPhoElecPtMC = new THnSparseD("fPhoElecPtMC", "MC Pho-inclusive electron pt",3,nBinspho2,minpho2,maxpho2);
+  fPhoElecPtMC = new THnSparseD("fPhoElecPtMC", "MC Pho-inclusive electron pt",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fPhoElecPtMC);
   
-  fPhoElecPtMCM20 = new THnSparseD("fPhoElecPtMCM20", "MC Pho-inclusive electron pt with M20",3,nBinspho2,minpho2,maxpho2);
+  fPhoElecPtMCM20 = new THnSparseD("fPhoElecPtMCM20", "MC Pho-inclusive electron pt with M20",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fPhoElecPtMCM20);
 
-  fSameElecPtMC = new THnSparseD("fSameElecPtMC", "MC Same-inclusive electron pt",3,nBinspho2,minpho2,maxpho2);
+  fSameElecPtMC = new THnSparseD("fSameElecPtMC", "MC Same-inclusive electron pt",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fSameElecPtMC);
 
-  fSameElecPtMCM20 = new THnSparseD("fSameElecPtMCM20", "MC Same-inclusive electron pt with M20",3,nBinspho2,minpho2,maxpho2);
+  fSameElecPtMCM20 = new THnSparseD("fSameElecPtMCM20", "MC Same-inclusive electron pt with M20",5,nBinspho2,minpho2,maxpho2);
   fOutputList->Add(fSameElecPtMCM20);
 
+  fIncpTMCM20pho_pi0e = new THnSparseD("fIncpTMCM20pho_pi0e","MC Pho pi0->e pid electro vs. centrality with M20",5,nBinspho2,minpho2,maxpho2);
+  fIncpTMCM20pho_pi0e->Sumw2();
+  fOutputList->Add(fIncpTMCM20pho_pi0e);
+
+  fPhoElecPtMCM20_pi0e = new THnSparseD("fPhoElecPtMCM20_pi0e", "MC Pho-inclusive electron pt with M20 pi0->e",5,nBinspho2,minpho2,maxpho2);
+  fPhoElecPtMCM20_pi0e->Sumw2();
+  fOutputList->Add(fPhoElecPtMCM20_pi0e);
+
+  fSameElecPtMCM20_pi0e = new THnSparseD("fSameElecPtMCM20_pi0e", "MC Same-inclusive electron pt pi0->e",5,nBinspho2,minpho2,maxpho2);
+  fSameElecPtMCM20_pi0e->Sumw2();
+  fOutputList->Add(fSameElecPtMCM20_pi0e);
+ //
+  fIncpTMCM20pho_eta = new THnSparseD("fIncpTMCM20pho_eta","MC Pho pi0->e pid electro vs. centrality with M20",5,nBinspho2,minpho2,maxpho2);
+  fIncpTMCM20pho_eta->Sumw2();
+  fOutputList->Add(fIncpTMCM20pho_eta);
+
+  fPhoElecPtMCM20_eta = new THnSparseD("fPhoElecPtMCM20_eta", "MC Pho-inclusive electron pt with M20 pi0->e",5,nBinspho2,minpho2,maxpho2);
+  fPhoElecPtMCM20_eta->Sumw2();
+  fOutputList->Add(fPhoElecPtMCM20_eta);
+
+  fSameElecPtMCM20_eta = new THnSparseD("fSameElecPtMCM20_eta", "MC Same-inclusive electron pt pi0->e",5,nBinspho2,minpho2,maxpho2);
+  fSameElecPtMCM20_eta->Sumw2();
+  fOutputList->Add(fSameElecPtMCM20_eta);
+
+
+  CheckNclust = new TH1D("CheckNclust","cluster check",200,0,200);
+  fOutputList->Add(CheckNclust);
+
+  CheckNits = new TH1D("CheckNits","ITS cluster check",8,-0.5,7.5);
+  fOutputList->Add(CheckNits);
+
+  Hpi0pTcheck = new TH2D("Hpi0pTcheck","Pi0 pT from Hijing",100,0,50,3,-0.5,2.5);
+  fOutputList->Add(Hpi0pTcheck);
+
+  HETApTcheck = new TH2D("HETApTcheck","Eta pT from Hijing",100,0,50,3,-0.5,2.5);
+  fOutputList->Add(HETApTcheck);
+
+  HphopTcheck = new TH2D("HphopTcheck","Pho pT from Hijing",100,0,50,3,-0.5,2.5);
+  fOutputList->Add(HphopTcheck);
+
+  fpTCheck = new TH1D("fpTCheck","pT check",500,0,50);
+  fOutputList->Add(fpTCheck);
+
+  fMomDtoE = new TH2D("fMomDtoE","D->E pT correlations;e p_{T} GeV/c;D p_{T} GeV/c",400,0,40,400,0,40);
+  fOutputList->Add(fMomDtoE);
 
   PostData(1,fOutputList);
 }
@@ -826,7 +1112,7 @@ Bool_t AliAnalysisTaskHFECal::ProcessCutStep(Int_t cutStep, AliVParticle *track)
 //_________________________________________
 //void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec)
 //void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig)
-void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce)
+void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta)
 {
   //Identify non-heavy flavour electrons using Invariant mass method
   
@@ -840,9 +1126,23 @@ void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent,
   
   const AliESDVertex *pVtx = fESD->GetPrimaryVertex();
   
+  double ptEle = track->Pt();  //add
+  if(ibgevent==0 && w > 0.0)
+     {
+      fpTCheck->Fill(ptEle,w);
+     }
+
   Bool_t flagPhotonicElec = kFALSE;
   Bool_t flagConvinatElec = kFALSE;
   
+  int p1 = 0;
+  if(mce==3)
+     {
+       Int_t label = TMath::Abs(track->GetLabel());
+       TParticle* particle = stack->Particle(label);
+       p1 = particle->GetFirstMother();
+     }
+
   //for(Int_t jTracks = itrack+1; jTracks<fESD->GetNumberOfTracks(); jTracks++){
   for(Int_t jTracks = 0; jTracks<fESD->GetNumberOfTracks(); jTracks++){
     AliESDtrack* trackAsso = fESD->GetTrack(jTracks);
@@ -851,12 +1151,25 @@ void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent,
       continue;
     }
     if(itrack==jTracks)continue;    
+    int jbgevent = 0;    
+
+    int p2 = 0;
+    if(mce==3)
+    {
+      Int_t label2 = TMath::Abs(trackAsso->GetLabel());
+      TParticle* particle2 = stack->Particle(label2);
+      Bool_t MChijing_ass = fMC->IsFromBGEvent(label2);
+      if(MChijing_ass)jbgevent =1;
+      if(particle2->GetFirstMother()>-1)
+         p2 = particle2->GetFirstMother();
+    }
 
     Double_t dEdxAsso = -999., ptPrim=-999., ptAsso=-999., openingAngle = -999.;
     Double_t mass=999., width = -999;
     Bool_t fFlagLS=kFALSE, fFlagULS=kFALSE;
     
-    ptPrim = track->Pt();
+    //ptPrim = track->Pt();
+    ptPrim = ptEle;
 
     dEdxAsso = trackAsso->GetTPCsignal();
     ptAsso = trackAsso->Pt();
@@ -889,6 +1202,7 @@ void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent,
     Double_t chi2recg = recg.GetChi2()/recg.GetNDF();
     if(TMath::Sqrt(TMath::Abs(chi2recg))>3.) continue;
     
+    //  for v5-03-70-AN comment
     AliKFVertex primV(*pVtx);
     primV += recg;
     recg.SetProductionVertex(primV);
@@ -918,24 +1232,97 @@ void AliAnalysisTaskHFECal::SelectPhotonicElectron(Int_t itrack, Double_t cent,
 
     if(fFlagLS) fInvmassLS->Fill(phoinfo);
     if(fFlagULS) fInvmassULS->Fill(phoinfo);
-    
+    if(fFlagLS && ibgevent==0 && jbgevent==0) fInvmassLSmc->Fill(phoinfo,w);
+    if(fFlagULS && ibgevent==0 && jbgevent==0)
+       {
+         fInvmassULSmc->Fill(phoinfo,w);
+       }
     //printf("fInvmassCut %f\n",fInvmassCut);
     //printf("openingAngle %f\n",fOpeningAngleCut);
 
     if(openingAngle > fOpeningAngleCut) continue;
-      
-    if(mass<fInvmassCut && fFlagULS && !flagPhotonicElec){
-      flagPhotonicElec = kTRUE;
-    }
-    if(mass<fInvmassCut && fFlagLS && !flagConvinatElec){
-      flagConvinatElec = kTRUE;
-    }
     
+    // for real data  
+    //printf("mce =%f\n",mce);
+    if(mce<-0.5) // mce==-1. is real
+       {
+         //printf("Real data\n");
+        if(mass<fInvmassCut && fFlagULS && !flagPhotonicElec){
+        //if(mass<fInvmassCut && fFlagULS && !flagPhotonicElec && (p1==p2)){ <--- only MC train (55,56) v5-03-68-AN & 69 for check
+              flagPhotonicElec = kTRUE;
+             }
+        if(mass<fInvmassCut && fFlagLS && !flagConvinatElec){
+              flagConvinatElec = kTRUE;
+             }
+        }
+    // for MC data  
+    else
+       {
+         //printf("MC data\n");
+
+         if(w>0.0)
+           {
+           //cout << "tagpi0 = " << tagpi0 << " ; tageta = " << tageta << endl;
+           if(fFlagLS && ibgevent==0 && jbgevent==0 && tagpi0) fInvmassLSmc0->Fill(ptPrim,mass,w);
+           if(fFlagULS && ibgevent==0 && jbgevent==0 && tagpi0) fInvmassULSmc0->Fill(ptPrim,mass,w);
+           if(fFlagLS && ibgevent==0 && jbgevent==0 && tageta) fInvmassLSmc1->Fill(ptPrim,mass,w);
+           if(fFlagULS && ibgevent==0 && jbgevent==0 && tageta) fInvmassULSmc1->Fill(ptPrim,mass,w);
+           if(fFlagLS && ibgevent==0 && jbgevent==0 && (p1==p2) && tagpi0) fInvmassLSmc2->Fill(ptPrim,mass,w);
+           if(fFlagULS && ibgevent==0 && jbgevent==0 && (p1==p2) && tagpi0) fInvmassULSmc2->Fill(ptPrim,mass,w);
+           if(fFlagLS && ibgevent==0 && jbgevent==0 && (p1==p2) && tageta) fInvmassLSmc3->Fill(ptPrim,mass,w);
+           if(fFlagULS && ibgevent==0 && jbgevent==0 && (p1==p2) && tageta) fInvmassULSmc3->Fill(ptPrim,mass,w);
+          }
+        if(mass<fInvmassCut && fFlagULS && !flagPhotonicElec && (ibgevent==jbgevent)){
+        //if(mass<fInvmassCut && fFlagULS && !flagPhotonicElec && (p1==p2)){ <--- only MC train (55,56) v5-03-68-AN & 69 for check
+              flagPhotonicElec = kTRUE;
+             }
+        if(mass<fInvmassCut && fFlagLS && !flagConvinatElec && (ibgevent==jbgevent)){
+              flagConvinatElec = kTRUE;
+             }
+        }
+
   }
   fFlagPhotonicElec = flagPhotonicElec;
   fFlagConvinatElec = flagConvinatElec;
   
 }
+//-------------------------------------------
+
+void AliAnalysisTaskHFECal::FindMother(TParticle* part, int &label, int &pid)
+{
+ //int label = 99999;
+ //int pid = 99999;
+
+ if(part->GetFirstMother()>-1)
+   {
+    label = part->GetFirstMother();
+    pid = stack->Particle(label)->GetPdgCode();
+   }
+   cout << "Find Mother : label = " << label << " ; pid" << pid << endl;
+}
+
+double AliAnalysisTaskHFECal::GetMCweight(double mcPi0pT)
+{
+        double weight = 1.0;
+
+       if(mcPi0pT>0.0 && mcPi0pT<5.0)
+       {
+               weight = 0.323*mcPi0pT/(TMath::Exp(-1.6+0.767*mcPi0pT+0.0285*mcPi0pT*mcPi0pT));
+       }
+       else
+       {
+               weight = 115.0/(0.718*mcPi0pT*TMath::Power(mcPi0pT,3.65));
+       }
+  return weight;
+}
+
+double AliAnalysisTaskHFECal::GetMCweightEta(double mcEtapT)
+{
+  double weight = 1.0;
+
+  weight = 223.3/TMath::Power((TMath::Exp(-0.17*mcEtapT-0.0322*mcEtapT*mcEtapT)+mcEtapT/1.69),5.65);
+  return weight;
+}
 
 
 //_________________________________________