]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGHF/hfe/AliAnalysisTaskHFECal.cxx
update
[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.cxx
index 0f3bd33ae6041adb2abb1a1c09f4f96610b946b9..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,14 +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
   
@@ -171,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)
@@ -197,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)
@@ -227,14 +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");
@@ -291,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.;
@@ -313,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();  
@@ -388,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;
     
@@ -485,7 +628,7 @@ 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;
@@ -554,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);
@@ -581,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
+              }
            }
         } 
       } 
@@ -717,11 +888,49 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   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);
   
@@ -811,35 +1020,77 @@ void AliAnalysisTaskHFECal::UserCreateOutputObjects()
   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] =  { 200, 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);
 }
 
@@ -861,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
   
@@ -875,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);
@@ -886,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();
@@ -924,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);
@@ -953,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;
+}
 
 
 //_________________________________________