#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.

References bounds_, Exception, cuy::ib, kMaxPt, LogTrace, Skims_PA_cff::name, name_, nbins_, safety_factor_h_, safety_factor_l_, sort_type_, AlCaHLTBitMon_QueryRunRegistry::string, cond::impl::to_string(), verbosity_, wdws_phi_, and wdws_theta_.

     : 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);
   }
 }

◆ ~L1TkMuMantra()

L1TkMuMantra::~L1TkMuMantra ( )

inline

Definition at line 54 of file L1TkMuMantra.h.

54 {};

Member Function Documentation

◆ eta_to_theta()

static double L1TkMuMantra::eta_to_theta ( double x )

inlinestatic

Definition at line 85 of file L1TkMuMantra.h.

References JetChargeProducer_cfi::exp, and x.

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

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

◆ 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.

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(), jetUpdater_cfi::sort, sort_type_, L1TkMuMantraDF::track_df::theta, tracks, parallelization::uint, wdws_phi_, and wdws_theta_.

                                                                                                           {
   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;
 }

◆ findBin()

int L1TkMuMantra::findBin ( double val )

private

Definition at line 82 of file L1TkMuMantra.cc.

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

Referenced by find_match(), and test().

                                     {
   // 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;
 }

◆ 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.

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

Referenced by L1TkMuonProducer::L1TkMuonProducer().

                                                                                     {
   // 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;
 }

◆ relax_windows()

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

Definition at line 180 of file L1TkMuMantra.cc.

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

Referenced by find_match().

                                                                        {
   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;
 }

◆ set_safety_factor()

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

inline

Definition at line 64 of file L1TkMuMantra.h.

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

                                                  {
     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;
   }

◆ setArbitrationType()

void L1TkMuMantra::setArbitrationType ( std::string type )

Definition at line 190 of file L1TkMuMantra.cc.

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

                                                     {
   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";
 }

◆ sign()

int L1TkMuMantra::sign ( double x )

inline

Definition at line 73 of file L1TkMuMantra.h.

References x.

Referenced by find_match().

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

◆ test()

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

Definition at line 99 of file L1TkMuMantra.cc.

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

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

                                              {
   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;
 }

◆ to_mpio2_pio2()

static double L1TkMuMantra::to_mpio2_pio2 ( double x )

inlinestatic

Definition at line 90 of file L1TkMuMantra.h.

References M_PI, and x.

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

                                         {
     //  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;
   }

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().

Public Member Functions

Static Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Enumeration Documentation

◆ sortParType

Constructor & Destructor Documentation

◆ L1TkMuMantra()

◆ ~L1TkMuMantra()

Member Function Documentation

◆ eta_to_theta()

◆ find_match()

◆ findBin()

◆ prepare_corr_bounds()

◆ relax_windows()

◆ set_safety_factor()

◆ setArbitrationType()

◆ sign()

◆ test()

◆ to_mpio2_pio2()

Member Data Documentation

◆ bounds_

◆ max_chi2

◆ min_nstubs

◆ name_

◆ nbins_

◆ safety_factor_h_

◆ safety_factor_l_

◆ sort_type_

◆ verbosity_

◆ wdws_phi_

◆ wdws_theta_