Changes in modules/MomentumSmearing.cc [341014c:84a097e] in git

File:

• modules/MomentumSmearing.cc (modified) (3 diffs)

modules/MomentumSmearing.cc

@@ -95 +95 @@
 {
   Candidate *candidate, *mother;
-  Double_t pt, eta, phi, e, res;
+  Double_t pt, eta, phi, e, m, res;
 
   fItInputArray->Reset();
@@ -106 +106 @@
     pt = candidateMomentum.Pt();
     e = candidateMomentum.E();
+    m = candidateMomentum.M();
+
     res = fFormula->Eval(pt, eta, phi, e);
-
-    // apply smearing formula
-    //pt = gRandom->Gaus(pt, fFormula->Eval(pt, eta, phi, e) * pt);
-
     res = (res > 1.0) ? 1.0 : res;
 
     pt = LogNormal(pt, res * pt);
-
-    //if(pt <= 0.0) continue;
 
     mother = candidate;
@@ -121 +117 @@
     eta = candidateMomentum.Eta();
     phi = candidateMomentum.Phi();
-    candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt * TMath::CosH(eta));
-    //candidate->TrackResolution = fFormula->Eval(pt, eta, phi, e);
+    candidate->Momentum.SetPtEtaPhiM(pt, eta, phi, m);
+    candidate->ErrorPT = res*pt;
+    candidate->PT = pt;
+    
     candidate->TrackResolution = res;
     candidate->AddCandidate(mother);