L1TkMuMantra Class Reference

#include <L1TkMuMantra.h>

Public Member Functions
std::vector< int >	find_match (const std::vector< L1TkMuMantraDF::track_df > &tracks, const std::vector< L1TkMuMantraDF::muon_df > &muons)
	L1TkMuMantra (const std::vector< double > &bounds, TFile *fIn_theta, TFile *fIn_phi, std::string name)
void	relax_windows (double &low, double cent, double &high)
void	set_safety_factor (float sf_l, float sf_h)
void	setArbitrationType (std::string type)
int	sign (double x)
void	test (double eta, double pt)
	~L1TkMuMantra ()

Static Public Member Functions
static double	eta_to_theta (double x)
static std::vector< double >	prepare_corr_bounds (std::string fname, std::string hname)
static double	to_mpio2_pio2 (double x)

Private Types
enum	sortParType { kMaxPt, kMinDeltaPt }

Private Member Functions
int	findBin (double val)

Private Attributes
std::vector< double >	bounds_
double	max_chi2 = 100
int	min_nstubs = 4
std::string	name_
int	nbins_
float	safety_factor_h_
float	safety_factor_l_
sortParType	sort_type_
int	verbosity_ = 0
std::vector< MuMatchWindow >	wdws_phi_
std::vector< MuMatchWindow >	wdws_theta_

Detailed Description

Definition at line 51 of file L1TkMuMantra.h.

Member Enumeration Documentation

sortParType

enum L1TkMuMantra::sortParType

private

Enumerator
kMaxPt
kMinDeltaPt

Definition at line 115 of file L1TkMuMantra.h.

                   {
    kMaxPt,      // pick the highest pt track matched
    kMinDeltaPt  // pick the track with the smallest pt difference w.r.t the muon
  };

Constructor & Destructor Documentation

L1TkMuMantra()

L1TkMuMantra::L1TkMuMantra	(	const std::vector< double > &	bounds,
		TFile *	fIn_theta,
		TFile *	fIn_phi,
		std::string	name = "mantra"
	)

Definition at line 10 of file L1TkMuMantra.cc.

    : wdws_theta_(bounds.size() - 1, MuMatchWindow()), wdws_phi_(bounds.size() - 1, MuMatchWindow()) {
  name_ = name;
 
  safety_factor_l_ = 0.0;
  safety_factor_h_ = 0.0;
 
  sort_type_ = kMaxPt;
 
  // copy boundaries
  nbins_ = bounds.size() - 1;
  bounds_ = bounds;
 
  // now load in memory the TF1 fits
 
  for (int ib = 0; ib < nbins_; ++ib) {
    std::string wdn;
    std::string nml;
    std::string nmc;
    std::string nmh;
    TF1* fl;
    TF1* fc;
    TF1* fh;
 
    wdn = name_ + std::string("_wdw_theta_") + std::to_string(ib + 1);
    nml = std::string("fit_low_") + std::to_string(ib + 1);
    nmc = std::string("fit_cent_") + std::to_string(ib + 1);
    nmh = std::string("fit_high_") + std::to_string(ib + 1);
 
    fl = (TF1*)fIn_theta->Get(nml.c_str());
    fc = (TF1*)fIn_theta->Get(nmc.c_str());
    fh = (TF1*)fIn_theta->Get(nmh.c_str());
    if (fl == nullptr || fc == nullptr || fh == nullptr) {
      if (verbosity_ > 0) {
        LogTrace("L1TkMuMantra") << "... fit theta low  : " << fl << std::endl;
        LogTrace("L1TkMuMantra") << "... fit theta cent : " << fc << std::endl;
        LogTrace("L1TkMuMantra") << "... fit theta high : " << fh << std::endl;
      }
      throw cms::Exception("L1TkMuMantra") << "TF1 named " << nml << " or " << nmc << " or " << nmh
                                           << " not found in file " << fIn_theta->GetName() << ".\n";
    }
    wdws_theta_.at(ib).SetName(wdn);
    wdws_theta_.at(ib).SetLower(fl);
    wdws_theta_.at(ib).SetCentral(fc);
    wdws_theta_.at(ib).SetUpper(fh);
 
    wdn = name_ + std::string("_wdw_phi_") + std::to_string(ib + 1);
    nml = std::string("fit_low_") + std::to_string(ib + 1);
    nmc = std::string("fit_cent_") + std::to_string(ib + 1);
    nmh = std::string("fit_high_") + std::to_string(ib + 1);
    fl = (TF1*)fIn_phi->Get(nml.c_str());
    fc = (TF1*)fIn_phi->Get(nmc.c_str());
    fh = (TF1*)fIn_phi->Get(nmh.c_str());
    if (fl == nullptr || fc == nullptr || fh == nullptr) {
      if (verbosity_ > 0) {
        LogTrace("L1TkMuMantra") << "... fit phi low  : " << fl << std::endl;
        LogTrace("L1TkMuMantra") << "... fit phi cent : " << fc << std::endl;
        LogTrace("L1TkMuMantra") << "... fit phi high : " << fh << std::endl;
      }
      throw cms::Exception("L1TkMuMantra") << "TF1 named " << nml << " or " << nmc << " or " << nmh
                                           << " not found in file " << fIn_theta->GetName() << ".\n";
    }
    wdws_phi_.at(ib).SetName(wdn);
    wdws_phi_.at(ib).SetLower(fl);
    wdws_phi_.at(ib).SetCentral(fc);
    wdws_phi_.at(ib).SetUpper(fh);
  }
}

References bounds_, Exception, benchmark_cfg::fc, cuy::ib, kMaxPt, LogTrace, Skims_PA_cff::name, name_, nbins_, safety_factor_h_, safety_factor_l_, sort_type_, AlCaHLTBitMon_QueryRunRegistry::string, tools::TF1, verbosity_, wdws_phi_, and wdws_theta_.

~L1TkMuMantra()

L1TkMuMantra::~L1TkMuMantra ( )

inline

Definition at line 54 of file L1TkMuMantra.h.

{};

Member Function Documentation

eta_to_theta()

static double L1TkMuMantra::eta_to_theta ( double  x )

inlinestatic

Definition at line 85 of file L1TkMuMantra.h.

                                       {
    //  give theta in rad
    return (2. * atan(exp(-1. * x)));
  }

References JetChargeProducer_cfi::exp, and x.

Referenced by L1TkMuonProducer::product_to_muvec(), and L1TkMuonProducer::product_to_trkvec().

find_match()

std::vector< int > L1TkMuMantra::find_match	(	const std::vector< L1TkMuMantraDF::track_df > &	tracks,
		const std::vector< L1TkMuMantraDF::muon_df > &	muons
	)

Definition at line 117 of file L1TkMuMantra.cc.

                                                                                                          {
  std::vector<int> result(muons.size(), -1);  // init all TkMu to index -1
  for (uint imu = 0; imu < muons.size(); ++imu) {
    muon_df mu = muons.at(imu);
    std::vector<std::pair<double, int>> matched_trks;  // sort_par, idx
    for (uint itrk = 0; itrk < tracks.size(); ++itrk) {
      // preselection of tracks
      track_df trk = tracks.at(itrk);
      if (trk.chi2 >= max_chi2)
        continue;  // require trk.chi2 < max_chi2
      if (trk.nstubs < min_nstubs)
        continue;  // require trk.nstubs >= min_nstubs
 
      double dphi_charge = reco::deltaPhi(trk.phi, mu.phi) * trk.charge;
      // sign from theta, to avoid division by 0
      double dtheta_endc = (mu.theta - trk.theta) * sign(mu.theta);
      if (sign(mu.theta) != sign(trk.theta)) {
        // crossing the barrel -> remove 180 deg to the theta of the neg candidate to avoid jumps at eta = 0
        dtheta_endc -= TMath::Pi();
      }
 
      // lookup the values
      int ibin = findBin(std::abs(trk.eta));
 
      double phi_low = wdws_phi_.at(ibin).bound_low(trk.pt);
      double phi_cent = wdws_phi_.at(ibin).bound_cent(trk.pt);
      double phi_high = wdws_phi_.at(ibin).bound_high(trk.pt);
      relax_windows(phi_low, phi_cent, phi_high);  // apply the safety factor
      bool in_phi = (dphi_charge > phi_low && dphi_charge < phi_high);
 
      double theta_low = wdws_theta_.at(ibin).bound_low(trk.pt);
      double theta_cent = wdws_theta_.at(ibin).bound_cent(trk.pt);
      double theta_high = wdws_theta_.at(ibin).bound_high(trk.pt);
      relax_windows(theta_low, theta_cent, theta_high);  // apply the safety factor
      bool in_theta = (dtheta_endc > theta_low && dtheta_endc < theta_high);
 
      if (in_phi && in_theta) {
        double sort_par = 99999;
        if (sort_type_ == kMaxPt)
          sort_par = trk.pt;
        else if (sort_type_ == kMinDeltaPt) {
          // trk.pt should always be > 0, but put this protection just in case
          sort_par = (trk.pt > 0 ? std::abs(1. - (mu.pt / trk.pt)) : 0);
        }
        matched_trks.emplace_back(sort_par, itrk);
      }
    }
 
    // choose out of the matched tracks the best one
    if (!matched_trks.empty()) {
      sort(matched_trks.begin(), matched_trks.end());
      int ibest = 99999;
      if (sort_type_ == kMaxPt)
        ibest = matched_trks.rbegin()->second;  // sorted low to high -> take last for highest pT (rbegin)
      else if (sort_type_ == kMinDeltaPt)
        ibest = matched_trks.begin()->second;  // sorted low to high -> take first for min pT distance (begin)
      result.at(imu) = ibest;
    }
  }
 
  return result;
}

References funct::abs(), L1TkMuMantraDF::track_df::charge, L1TkMuMantraDF::track_df::chi2, reco::deltaPhi(), L1TkMuMantraDF::track_df::eta, findBin(), kMaxPt, kMinDeltaPt, max_chi2, min_nstubs, amptDefaultParameters_cff::mu, PDWG_BPHSkim_cff::muons, L1TkMuMantraDF::track_df::nstubs, L1TkMuMantraDF::track_df::phi, Pi, L1TkMuMantraDF::track_df::pt, relax_windows(), mps_fire::result, sign(), sort_type_, L1TkMuMantraDF::track_df::theta, PDWG_EXOHSCP_cff::tracks, parallelization::uint, wdws_phi_, and wdws_theta_.

findBin()

int L1TkMuMantra::findBin ( double  val )

private

Definition at line 82 of file L1TkMuMantra.cc.

                                    {
  // FIXME: not the most efficient, nor the most elegant implementation for now
  if (val < bounds_.at(0))
    return 0;
  if (val >= bounds_.back())
    return (nbins_ - 1);  // i.e. bounds_size() -2
 
  for (uint ib = 0; ib < bounds_.size() - 1; ++ib) {
    if (val >= bounds_.at(ib) && val < bounds_.at(ib + 1))
      return ib;
  }
 
  if (verbosity_ > 0)
    LogTrace("L1TkMuMantra") << "Something strange happened at val " << val << std::endl;
  return 0;
}

References bounds_, cuy::ib, LogTrace, nbins_, parallelization::uint, heppy_batch::val, and verbosity_.

Referenced by find_match(), and test().

prepare_corr_bounds()

std::vector< double > L1TkMuMantra::prepare_corr_bounds ( std::string  fname, std::string  hname  )

static

Definition at line 201 of file L1TkMuMantra.cc.

                                                                                    {
  // find the boundaries of the match windoww
  TFile* fIn = TFile::Open(fname.c_str());
  if (fIn == nullptr) {
    throw cms::Exception("L1TkMuMantra") << "Can't find file " << fname << " to derive bounds.\n";
  }
  TH2* h_test = (TH2*)fIn->Get(hname.c_str());
  if (h_test == nullptr) {
    throw cms::Exception("L1TkMuMantra") << "Can't find histo " << hname << " in file " << fname
                                         << " to derive bounds.\n";
  }
 
  int nbds = h_test->GetNbinsY() + 1;
  std::vector<double> bounds(nbds);
  for (int ib = 0; ib < nbds; ++ib) {
    bounds.at(ib) = h_test->GetYaxis()->GetBinLowEdge(ib + 1);
  }
  fIn->Close();
  return bounds;
}

References Exception, alignmentValidation::fname, and cuy::ib.

Referenced by L1TkMuonProducer::L1TkMuonProducer().

relax_windows()

void L1TkMuMantra::relax_windows	(	double &	low,
		double	cent,
		double &	high
	)

Definition at line 180 of file L1TkMuMantra.cc.

                                                                       {
  double delta_high = high - cent;
  double delta_low = cent - low;
 
  high = high + safety_factor_h_ * delta_high;
  low = low - safety_factor_l_ * delta_low;
 
  return;
}

References LaserClient_cfi::high, LaserClient_cfi::low, safety_factor_h_, and safety_factor_l_.

Referenced by find_match().

set_safety_factor()

void L1TkMuMantra::set_safety_factor ( float  sf_l, float  sf_h  )

inline

Definition at line 64 of file L1TkMuMantra.h.

                                                 {
    safety_factor_l_ = sf_l;
    safety_factor_h_ = sf_h;
    if (verbosity_ > 0)
      LogTrace("L1TkMuMantra") << name_ << " safety factor LOW is " << safety_factor_l_ << std::endl;
    if (verbosity_ > 0)
      LogTrace("L1TkMuMantra") << name_ << " safety factor HIGH is " << safety_factor_h_ << std::endl;
  }

References LogTrace, name_, safety_factor_h_, safety_factor_l_, and verbosity_.

setArbitrationType()

void L1TkMuMantra::setArbitrationType

(

std::string

type

)

Definition at line 190 of file L1TkMuMantra.cc.

                                                    {
  if (verbosity_ > 0)
    LogTrace("L1TkMuMantra") << "L1TkMuMantra : setting arbitration type to " << type << std::endl;
  if (type == "MaxPt")
    sort_type_ = kMaxPt;
  else if (type == "MinDeltaPt")
    sort_type_ = kMinDeltaPt;
  else
    throw cms::Exception("L1TkMuMantra") << "setArbitrationType : cannot understand the arbitration type passed.\n";
}

References Exception, kMaxPt, kMinDeltaPt, LogTrace, sort_type_, and verbosity_.

sign()

int L1TkMuMantra::sign ( double  x )

inline

Definition at line 73 of file L1TkMuMantra.h.

                     {
    if (x == 0)
      return 1;
    return (0 < x) - (x < 0);
  }

References x.

Referenced by find_match().

test()

void L1TkMuMantra::test	(	double	eta,
		double	pt
	)

Definition at line 99 of file L1TkMuMantra.cc.

                                             {
  int ibin = findBin(eta);
 
  LogTrace("L1TkMuMantra") << " ---- eta : " << eta << " pt: " << pt << std::endl;
  LogTrace("L1TkMuMantra") << " ---- bin " << ibin << std::endl;
  LogTrace("L1TkMuMantra") << " ---- "
                           << "- low_phi  : " << wdws_phi_.at(ibin).bound_low(pt)
                           << "- cent_phi : " << wdws_phi_.at(ibin).bound_cent(pt)
                           << "- high_phi : " << wdws_phi_.at(ibin).bound_high(pt) << std::endl;
 
  LogTrace("L1TkMuMantra") << " ---- "
                           << "- low_theta  : " << wdws_theta_.at(ibin).bound_low(pt)
                           << "- cent_theta : " << wdws_theta_.at(ibin).bound_cent(pt)
                           << "- high_theta : " << wdws_theta_.at(ibin).bound_high(pt) << std::endl;
 
  return;
}

References PVValHelper::eta, findBin(), LogTrace, DiDispStaMuonMonitor_cfi::pt, wdws_phi_, and wdws_theta_.

Referenced by edmIntegrityCheck.IntegrityCheck::report(), and edmIntegrityCheck.IntegrityCheck::structured().

to_mpio2_pio2()

static double L1TkMuMantra::to_mpio2_pio2 ( double  x )

inlinestatic

Definition at line 90 of file L1TkMuMantra.h.

                                        {
    //  put the angle in radians between -pi/2 and pi/2
    while (x >= 0.5 * M_PI)
      x -= M_PI;
    while (x < -0.5 * M_PI)
      x += M_PI;
    return x;
  }

References M_PI, and x.

Referenced by L1TkMuonProducer::product_to_muvec(), and L1TkMuonProducer::product_to_trkvec().

Member Data Documentation

bounds_

std::vector<double> L1TkMuMantra::bounds_

private

Definition at line 105 of file L1TkMuMantra.h.

Referenced by findBin(), and L1TkMuMantra().

max_chi2

double L1TkMuMantra::max_chi2 = 100

private

Definition at line 110 of file L1TkMuMantra.h.

Referenced by find_match().

min_nstubs

int L1TkMuMantra::min_nstubs = 4

private

Definition at line 109 of file L1TkMuMantra.h.

Referenced by find_match().

name_

std::string L1TkMuMantra::name_

private

Definition at line 102 of file L1TkMuMantra.h.

Referenced by L1TkMuMantra(), and set_safety_factor().

nbins_

int L1TkMuMantra::nbins_

private

Definition at line 104 of file L1TkMuMantra.h.

Referenced by findBin(), and L1TkMuMantra().

safety_factor_h_

float L1TkMuMantra::safety_factor_h_

private

Definition at line 113 of file L1TkMuMantra.h.

Referenced by L1TkMuMantra(), relax_windows(), and set_safety_factor().

safety_factor_l_

float L1TkMuMantra::safety_factor_l_

private

Definition at line 112 of file L1TkMuMantra.h.

Referenced by L1TkMuMantra(), relax_windows(), and set_safety_factor().

sort_type_

sortParType L1TkMuMantra::sort_type_

private

Definition at line 120 of file L1TkMuMantra.h.

Referenced by find_match(), L1TkMuMantra(), and setArbitrationType().

verbosity_

int L1TkMuMantra::verbosity_ = 0

private

Definition at line 122 of file L1TkMuMantra.h.

Referenced by findBin(), L1TkMuMantra(), set_safety_factor(), and setArbitrationType().

wdws_phi_

std::vector<MuMatchWindow> L1TkMuMantra::wdws_phi_

private

Definition at line 107 of file L1TkMuMantra.h.

Referenced by find_match(), L1TkMuMantra(), and test().

wdws_theta_

std::vector<MuMatchWindow> L1TkMuMantra::wdws_theta_

private

Definition at line 106 of file L1TkMuMantra.h.

Referenced by find_match(), L1TkMuMantra(), and test().