deep_tau::TauWPThreshold Class Reference

#include <DeepTauBase.h>

Public Member Functions
double	operator() (const reco::BaseTau &tau, bool isPFTau) const
	TauWPThreshold (const std::string &cut_str)

Private Attributes
std::unique_ptr< TF1 >	fn_
double	value_

Detailed Description

Definition at line 38 of file DeepTauBase.h.

Constructor & Destructor Documentation

deep_tau::TauWPThreshold::TauWPThreshold ( const std::string &  cut_str )

explicit

Definition at line 17 of file DeepTauBase.cc.

References fn_, PostProcessorHGCAL_cfi::prefix, AlCaHLTBitMon_QueryRunRegistry::string, and value_.

                                                         {
    bool simple_value = false;
    try {
      size_t pos = 0;
      value_ = std::stod(cut_str, &pos);
      simple_value = (pos == cut_str.size());
    } catch (std::invalid_argument&) {
    } catch (std::out_of_range&) {
    }
    if (!simple_value) {
      static const std::string prefix =
          "[&](double *x, double *p) { const int decayMode = p[0];"
          "const double pt = p[1]; const double eta = p[2];";
      static const int n_params = 3;
      static const auto handler = [](int, Bool_t, const char*, const char*) -> void {};

      std::string fn_str = prefix;
      if (cut_str.find("return") == std::string::npos)
        fn_str += " return " + cut_str + ";}";
      else
        fn_str += cut_str + "}";
      auto old_handler = SetErrorHandler(handler);
      fn_ = std::make_unique<TF1>("fn_", fn_str.c_str(), 0, 1, n_params);
      SetErrorHandler(old_handler);
      if (!fn_->IsValid())
        throw cms::Exception("TauWPThreshold: invalid formula") << "Invalid WP cut formula = '" << cut_str << "'.";
    }
  }

Member Function Documentation

double deep_tau::TauWPThreshold::operator()	(	const reco::BaseTau &	tau,
		bool	isPFTau
	)		const

Definition at line 46 of file DeepTauBase.cc.

References reco::LeafCandidate::eta(), fn_, reco::LeafCandidate::pt(), and value_.

                                                                              {
    if (!fn_) {
      return value_;
    }

    if (isPFTau)
      fn_->SetParameter(0, dynamic_cast<const reco::PFTau&>(tau).decayMode());
    else
      fn_->SetParameter(0, dynamic_cast<const pat::Tau&>(tau).decayMode());
    fn_->SetParameter(1, tau.pt());
    fn_->SetParameter(2, tau.eta());
    return fn_->Eval(0);
  }

Member Data Documentation

std::unique_ptr<TF1> deep_tau::TauWPThreshold::fn_

private

Definition at line 44 of file DeepTauBase.h.

Referenced by operator()(), and TauWPThreshold().

double deep_tau::TauWPThreshold::value_

private

Definition at line 45 of file DeepTauBase.h.

Referenced by operator()(), and TauWPThreshold().