Update HFE code

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliHFEreducedEventCreatorESD.cxx

diff --git a/PWGHF/hfe/AliHFEreducedEventCreatorESD.cxx b/PWGHF/hfe/AliHFEreducedEventCreatorESD.cxx
index a3e4d95377be1fadda59982c5d7ef3c658afb161..039e097198c6274c32e439b6768cca6fe32a88b2 100644 (file)
--- a/PWGHF/hfe/AliHFEreducedEventCreatorESD.cxx
+++ b/PWGHF/hfe/AliHFEreducedEventCreatorESD.cxx
@@ -367,7 +367,7 @@ void AliHFEreducedEventCreatorESD::UserExec(Option_t *){
       hfemcpart.SetSource(source);
       */
       hfemcpart.SetSource(static_cast<Int_t>(fSignalCuts->GetSignalSource(mctrack)));
-      hfemcpart.SetElectronSource(fSignalCuts->GetMCQAObject()->GetElecSource(mctrack));
+      hfemcpart.SetElectronSource(fSignalCuts->GetMCQAObject()->GetElecSource(mctrack,kTRUE));
       fHFEevent->AddMCParticle(&hfemcpart);
     }
   }
@@ -467,7 +467,7 @@ void AliHFEreducedEventCreatorESD::UserExec(Option_t *){
         hfetrack.SetMCSource(source); 
         */
         hfetrack.SetMCSource(static_cast<Int_t>(fSignalCuts->GetSignalSource(track))); 
-        hfetrack.SetMCElectronSource(fSignalCuts->GetMCQAObject()->GetElecSource(mctrack));
+        hfetrack.SetMCElectronSource(fSignalCuts->GetMCQAObject()->GetElecSource(mctrack,kTRUE));
       } else {
         AliDebug(2, "Associated MC particle not found");
       }