- AliAnalysisTaskSAQA: new QA histograms

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliHadCorrTask.cxx

diff --git a/PWGJE/EMCALJetTasks/AliHadCorrTask.cxx b/PWGJE/EMCALJetTasks/AliHadCorrTask.cxx
index 4f34a8419dd8682c63ddd80dcb58c9374ae77048..69aa600fddb66aea6af38758db8811a56f5f8cce 100644 (file)
--- a/PWGJE/EMCALJetTasks/AliHadCorrTask.cxx
+++ b/PWGJE/EMCALJetTasks/AliHadCorrTask.cxx
@@ -573,20 +573,21 @@ void AliHadCorrTask::DoMatchedTracksLoop(AliEmcalParticle *emccluster, Double_t
     AliEmcalParticle *emctrack = static_cast<AliEmcalParticle*>(fTracks->At(iTrack));
     if (!emctrack)
       continue;
+
+    // check if track also points to cluster
+    if (fDoTrackClus && (emctrack->GetMatchedObjId(0)) != iClus)
+      continue;
+
     AliVTrack *track = emctrack->GetTrack();
     if (!track)
       continue;
     if (!AcceptTrack(track))
       continue;
 
-    // check if track also points to cluster
-    if (fDoTrackClus && (track->GetEMCALcluster()) != iClus)
-      continue;
-
     Double_t etadiff = 999;
     Double_t phidiff = 999;
     AliPicoTrack::GetEtaPhiDiff(track, cluster, phidiff, etadiff);
-    
+
     Double_t mom       = track->P();
     Int_t    mombin    = GetMomBin(mom); 
     Int_t    centbinch = fCentBin;
@@ -877,8 +878,13 @@ Double_t AliHadCorrTask::ApplyHadCorrAllTracks(AliEmcalParticle *emccluster, Dou
       else if (trkPMCfrac > 0.95)
        fHistEmbTrackMatchesOversub[fCentBin]->Fill(energyclus, overSub);
 
-      if (fDoExact)
+      if (fDoExact) {
        Esub -= overSub;
+       if (EclusBkgcorr + EclusMCcorr > 0) {
+         Double_t newfrac = EclusMCcorr / (EclusBkgcorr + EclusMCcorr);
+         cluster->SetMCEnergyFraction(newfrac);
+       }
+      }
     }
   }