TrkDeDxMassEstFeatures.cxx

 
#include "DQM/TrkDeDxMassEstFeatures.h"
 
#include "Recon/Event/TrackDeDxMassEstimate.h"
 
namespace dqm {
 
void TrkDeDxMassEstFeatures::configure(framework::config::Parameters& ps) {
  mass_estimate_name_ = ps.get<std::string>("mass_estimate_name");
  mass_estimate_pass_ = ps.get<std::string>("mass_estimate_pass");
 
  return;
}
 
void TrkDeDxMassEstFeatures::analyze(const framework::Event& event) {
  auto mass_estimates{event.getCollection<ldmx::TrackDeDxMassEstimate>(
      mass_estimate_name_, mass_estimate_pass_)};
 
  for (const auto& mass_est : mass_estimates) {
    auto momentum = mass_est.getMomentum();
    histograms_.fill("momentum:harmonic_mean_dedx", momentum, mass_est.getIh());
    if (momentum < 2000.)
      histograms_.fill("momentum_low:harmonic_mean_dedx", momentum,
                       mass_est.getIh());
    histograms_.fill("harmonic_mean_dedx", mass_est.getIh());
    histograms_.fill("mass_estimate", mass_est.getMass());
    histograms_.fill("mass_squared", mass_est.getMassSquared());
    if (momentum < 1000.) {
      histograms_.fill("mass_estimate_low_p", mass_est.getMass());
      histograms_.fill("mass_squared_low_p", mass_est.getMassSquared());
    }
    if (momentum < 500.) {
      histograms_.fill("mass_estimate_very_low_p", mass_est.getMass());
      histograms_.fill("mass_squared_very_low_p", mass_est.getMassSquared());
      if ((mass_est.getPdgId() == 11) || (mass_est.getPdgId() == -11)) {
        histograms_.fill("mass_estimate_very_low_p_electron",
                         mass_est.getMass());
        histograms_.fill("mass_squared_very_low_p_electron",
                         mass_est.getMassSquared());
      }
      if ((mass_est.getPdgId() == 211) || (mass_est.getPdgId() == -211)) {
        histograms_.fill("mass_estimate_very_low_p_pion", mass_est.getMass());
        histograms_.fill("mass_squared_very_low_p_pion",
                         mass_est.getMassSquared());
      }
      if ((mass_est.getPdgId() == 321) || (mass_est.getPdgId() == -321)) {
        histograms_.fill("mass_estimate_very_low_p_kaon", mass_est.getMass());
        histograms_.fill("mass_squared_very_low_p_kaon",
                         mass_est.getMassSquared());
      }
      if ((mass_est.getPdgId() == 2212) || (mass_est.getPdgId() == -2212)) {
        histograms_.fill("mass_estimate_very_low_p_proton", mass_est.getMass());
        histograms_.fill("mass_squared_very_low_p_proton",
                         mass_est.getMassSquared());
      }
    }
    histograms_.fill("track_type", mass_est.getTrackType() + 0.5);
  }
 
  return;
}
 
}  // namespace dqm
 
DECLARE_ANALYZER(dqm::TrkDeDxMassEstFeatures);