WIP: Test on PID instead of pmass

benchmarks/dvmp/analysis/dvmp.h

@@ -54,12 +54,12 @@ namespace util {
     }
     return momenta;
   }
-  /*inline double test_jpsi_pt(const std::vector<ROOT::Math::PxPyPzMVector>& momenta){
+  inline double test_jpsi_pt(const std::vector<ROOT::Math::PxPyPzMVector>& momenta){
     double px = momenta[4].px;
     double py = momenta[4].py;
     double pt = sqrt(px*px + py*py);
     return pt;
-  }*/
+  }
   
   
  //for Dummy rc

benchmarks/dvmp/analysis/vm_invar.cxx

@@ -94,8 +94,8 @@ int vm_invar(const std::string& config_name)
                   .Define("nu_sim", util::get_nu, {"invariant_quantities_sim"})
                   .Define("Q2_sim", util::get_Q2, {"invariant_quantities_sim"})
                   .Define("x_sim", util::get_x, {"invariant_quantities_sim"})
-                  .Define("t_sim", util::get_t, {"invariant_quantities_sim"});
-                  //.Define("test_pt", util::test_jpsi_pt, {""});
+                  .Define("t_sim", util::get_t, {"invariant_quantities_sim"})
+                  .Define("test_pt", util::test_jpsi_pt, {"test_sim_ordered"});
   //================================================================
 
   // Define output histograms
@@ -110,6 +110,8 @@ int vm_invar(const std::string& config_name)
   auto h_x_rec  = d_im.Histo1D({"h_x_rec", ";x;#", 100, 0., 0.1}, "x_rec");
   auto h_t_rec  = d_im.Histo1D({"h_t_rec", ";t;#", 100, -1., 0.}, "t_rec");
   
+  auto h_test_pt = d_im.Histo1D({"h_test_pt", "h_test_pt", 50, 0., 10.}, "test_pt");
+  
   // Plot our histograms.
   // TODO: to start I'm explicitly plotting the histograms, but want to
   // factorize out the plotting code moving forward.
@@ -202,7 +204,7 @@ int vm_invar(const std::string& config_name)
     
     // pad 3 x
     c.cd(4);
-    auto& ht_rec = *h_t_rec;
+    /*auto& ht_rec = *h_t_rec;
     auto& ht_sim = *h_t_sim;
     // histogram style
     ht_rec.SetLineColor(plot::kMpOrange);
@@ -225,7 +227,27 @@ int vm_invar(const std::string& config_name)
     tptr4->SetTextColor(plot::kMpBlue);
     tptr4 = t4->AddText("rec(PlaceHolder)");
     tptr4->SetTextColor(plot::kMpOrange);
-    t4->Draw();
+    t4->Draw();*/
+    auto& htest = *h_test_pt;
+    htest.SetLineColor(plot::kMpBlue);
+    htest.SetLineWidth(2);
+    htest.GetXaxis()->CenterTitle();
+    htest.GetYaxis()->CenterTitle();
+    htest.DrawClone("hist");
+    plot::draw_label(10, 100, detector);
+    TText* tptrtest;
+    auto   ttest = new TPaveText(.6, .8417, .9, .925, "NB NDC");
+    ttest->SetFillColorAlpha(kWhite, 0);
+    ttest->SetTextFont(43);
+    ttest->SetTextSize(25);
+    tptrtest = ttest->AddText("simulated");
+    tptrtest->SetTextColor(plot::kMpBlue);
+    //tptrtest = ttest->AddText("reconstructed");
+    //tptrtest->SetTextColor(plot::kMpOrange);
+    ttest->Draw();
+    
+    
+    
     
     c.Print(fmt::format("{}InvariantQuantities.png", output_prefix).c_str());