X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PWGHF%2Fhfe%2FAliHFEvarManager.cxx;h=f23ffd9888d5f63fec71c615022b6e7f94acf95b;hb=40e62b79a9a450e2be559df25c3105fea66c2e9a;hp=827313e218ee1898db945833e3610d6b865512af;hpb=71925dea2c01e91a2197290743308c44e4f4ec87;p=u%2Fmrichter%2FAliRoot.git

diff --git a/PWGHF/hfe/AliHFEvarManager.cxx b/PWGHF/hfe/AliHFEvarManager.cxx
index 827313e218e..f23ffd9888d 100644
--- 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;