Merge branch 'feature-movesplit'

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

diff --git a/PWGHF/hfe/AliHFEvarManager.cxx b/PWGHF/hfe/AliHFEvarManager.cxx
index 827313e218ee1898db945833e3610d6b865512af..f23ffd9888d5f63fec71c615022b6e7f94acf95b 100644 (file)
--- a/PWGHF/hfe/AliHFEvarManager.cxx
+++ b/PWGHF/hfe/AliHFEvarManager.cxx
@@ -140,9 +140,9 @@ void AliHFEvarManager::Copy(TObject &o) const{
   target.fContentMC = new Double_t[sizeof(fContentMC)/sizeof(Double_t)];
   target.fWeightFactor = fWeightFactor;
   target.fSignalTrack = fSignalTrack;
-       target.fWeighting = fWeighting;
+  target.fWeighting = fWeighting;
   target.fSignal = fSignal;
-       target.fWeightFactors = fWeightFactors;
+  target.fWeightFactors = fWeightFactors;
   target.fWeightFactorsFunction = fWeightFactorsFunction; 
   target.SetOwner(kFALSE);
 }
@@ -320,14 +320,17 @@ Double_t AliHFEvarManager::GetValue(AliVParticle *track, UInt_t code, Float_t ce
     }
     case kSource:{
       if(fSignal){
-       if(fSignal->IsCharmElectron(track)) value = 0;
-       else if(fSignal->IsBeautyElectron(track)) value = 1;
-       else if(fSignal->IsGammaElectron(track)) value = 2;
+        value = static_cast<Int_t>(fSignal->GetSignalSource(track));
+        /*
+             if(fSignal->IsCharmElectron(track)) value = 0;
+             else if(fSignal->IsBeautyElectron(track)) value = 1;
+             else if(fSignal->IsGammaElectron(track)) value = 2;
         else if(fSignal->IsNonHFElectron(track)) value = 3;
         else if(fSignal->IsJpsiElectron(track)) value = 4;
         else if(fSignal->IsB2JpsiElectron(track)) value = 5;
         else if(fSignal->IsKe3Electron(track)) value = 6;
-       else value = 7;
+             else value = 7;
+        */
       }
       AliDebug(2, Form("source: %f", value));
       break;