From 5d6a26219c3a80dfb2d0a76a068172a69049667c Mon Sep 17 00:00:00 2001
From: Ziyue Zhang <Ziyue_Zhang@localhost.localdomain>
Date: Thu, 4 Feb 2021 14:24:26 -0600
Subject: [PATCH] Add proton mass; re-implement inv_quant_rec in the script

---
 benchmarks/dvmp/analysis/vm_invar.cxx | 4 ++++
 include/util.h                        | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/benchmarks/dvmp/analysis/vm_invar.cxx b/benchmarks/dvmp/analysis/vm_invar.cxx
index 83723c5a..296b7ed4 100644
--- a/benchmarks/dvmp/analysis/vm_invar.cxx
+++ b/benchmarks/dvmp/analysis/vm_invar.cxx
@@ -74,6 +74,10 @@ int vm_invar(const std::string& config_name)
     return util::momenta_from_tracking(tracks, decay_mass);
   };
 
+  auto calc_inv_quant_rec = [vm_mass](const std::vector<ROOT::Math::PxPyPzMVector>& parts) {
+    return util::calc_inv_quant_rec(parts, vm_mass);
+  };
+
   //====================================================================
 
   // Define analysis flow
diff --git a/include/util.h b/include/util.h
index 56fb1289..6a24b293 100644
--- a/include/util.h
+++ b/include/util.h
@@ -48,6 +48,8 @@ namespace util {
       return 3.0969;
     } else if (part == "upsilon") {
       return 9.49630;
+    } else if (part == "proton"){
+      return 0.938272;
     } else {
       throw unknown_particle_error{part};
     }
-- 
GitLab