Changes in modules/MomentumSmearing.cc [341014c:27197df] in git

File:

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

modules/MomentumSmearing.cc

@@ -78 +78 @@
   fItInputArray = fInputArray->MakeIterator();
 
+  // switch to compute momentum smearing based on momentum vector eta, phi
+  fUseMomentumVector = GetBool("UseMomentumVector", false);
+
   // create output array
 
@@ -95 +98 @@
 {
   Candidate *candidate, *mother;
-  Double_t pt, eta, phi, e, res;
+  Double_t pt, eta, phi, e, m, res;
 
   fItInputArray->Reset();
@@ -104 +107 @@
     eta = candidatePosition.Eta();
     phi = candidatePosition.Phi();
+
+    if (fUseMomentumVector){
+      eta = candidateMomentum.Eta();
+      phi = candidateMomentum.Phi();
+    }
+
     pt = candidateMomentum.Pt();
     e = candidateMomentum.E();
-    res = fFormula->Eval(pt, eta, phi, e);
+    m = candidateMomentum.M();
+    res = fFormula->Eval(pt, eta, phi, e, candidate);
 
     // apply smearing formula
@@ -121 +131 @@
     eta = candidateMomentum.Eta();
     phi = candidateMomentum.Phi();
-    candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt * TMath::CosH(eta));
+    candidate->Momentum.SetPtEtaPhiM(pt, eta, phi, m);
     //candidate->TrackResolution = fFormula->Eval(pt, eta, phi, e);
     candidate->TrackResolution = res;