TemplatedJetProbabilityComputer< Container, Base > Class Template Reference

#include <TemplatedJetProbabilityComputer.h>

Inheritance diagram for TemplatedJetProbabilityComputer< Container, Base >:

Public Types
typedef reco::IPTagInfo < Container, Base >	TagInfo
using	Tokens = void

Public Member Functions
float	discriminator (const TagInfoHelper &ti) const override
double	jetProbability (const std::vector< float > &v) const
	TemplatedJetProbabilityComputer (const edm::ParameterSet &parameters)
Public Member Functions inherited from JetTagComputer
const std::vector< std::string > &	getInputLabels () const
virtual void	initialize (const JetTagComputerRecord &)
	JetTagComputer ()
	JetTagComputer (const edm::ParameterSet &configuration)
float	operator() (const reco::BaseTagInfo &info) const
float	operator() (const TagInfoHelper &helper) const
void	setupDone ()
virtual	~JetTagComputer ()

Private Attributes
double	m_cutMaxDecayLen
double	m_cutMaxDistToAxis
double	m_deltaR
int	m_ipType
double	m_minTrackProb
reco::TrackBase::TrackQuality	m_trackQuality
int	m_trackSign
bool	m_useAllQualities
bool	useVariableJTA_
reco::btag::variableJTAParameters	varJTApars

Additional Inherited Members
Protected Member Functions inherited from JetTagComputer
virtual float	discriminator (const reco::BaseTagInfo &) const
void	uses (unsigned int id, const std::string &label)
void	uses (const std::string &label)

Detailed Description

template<class Container, class Base>
class TemplatedJetProbabilityComputer< Container, Base >

Definition at line 15 of file TemplatedJetProbabilityComputer.h.

Member Typedef Documentation

template<class Container , class Base >

typedef reco::IPTagInfo<Container, Base> TemplatedJetProbabilityComputer< Container, Base >::TagInfo

Definition at line 19 of file TemplatedJetProbabilityComputer.h.

template<class Container , class Base >

using TemplatedJetProbabilityComputer< Container, Base >::Tokens = void

Definition at line 17 of file TemplatedJetProbabilityComputer.h.

Constructor & Destructor Documentation

template<class Container , class Base >

TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer ( const edm::ParameterSet &  parameters )

inline

Definition at line 21 of file TemplatedJetProbabilityComputer.h.

References edm::ParameterSet::getParameter(), TemplatedJetProbabilityComputer< Container, Base >::m_cutMaxDecayLen, TemplatedJetProbabilityComputer< Container, Base >::m_cutMaxDistToAxis, TemplatedJetProbabilityComputer< Container, Base >::m_deltaR, TemplatedJetProbabilityComputer< Container, Base >::m_ipType, TemplatedJetProbabilityComputer< Container, Base >::m_minTrackProb, TemplatedJetProbabilityComputer< Container, Base >::m_trackQuality, TemplatedJetProbabilityComputer< Container, Base >::m_trackSign, TemplatedJetProbabilityComputer< Container, Base >::m_useAllQualities, reco::TrackBase::qualityByName(), AlCaHLTBitMon_QueryRunRegistry::string, JetTagComputer::uses(), TemplatedJetProbabilityComputer< Container, Base >::useVariableJTA_, and TemplatedJetProbabilityComputer< Container, Base >::varJTApars.

                                                                     {
    m_ipType = parameters.getParameter<int>("impactParameterType");
    m_minTrackProb = parameters.getParameter<double>("minimumProbability");
    m_deltaR = parameters.getParameter<double>("deltaR");
    m_trackSign = parameters.getParameter<int>("trackIpSign");
    m_cutMaxDecayLen = parameters.getParameter<double>("maximumDecayLength");
    m_cutMaxDistToAxis = parameters.getParameter<double>("maximumDistanceToJetAxis");
    //
    // access track quality class; "any" takes everything
    //
    std::string trackQualityType = parameters.getParameter<std::string>("trackQualityClass");  //used
    m_trackQuality = reco::TrackBase::qualityByName(trackQualityType);
    m_useAllQualities = false;
    if (trackQualityType == "any" || trackQualityType == "Any" || trackQualityType == "ANY")
      m_useAllQualities = true;

    useVariableJTA_ = parameters.getParameter<bool>("useVariableJTA");
    if (useVariableJTA_)
      varJTApars = {parameters.getParameter<double>("a_dR"),
                    parameters.getParameter<double>("b_dR"),
                    parameters.getParameter<double>("a_pT"),
                    parameters.getParameter<double>("b_pT"),
                    parameters.getParameter<double>("min_pT"),
                    parameters.getParameter<double>("max_pT"),
                    parameters.getParameter<double>("min_pT_dRcut"),
                    parameters.getParameter<double>("max_pT_dRcut"),
                    parameters.getParameter<double>("max_pT_trackPTcut")};

    uses("ipTagInfos");
  }

Member Function Documentation

template<class Container , class Base >

float TemplatedJetProbabilityComputer< Container, Base >::discriminator ( const TagInfoHelper &  ti ) const

inlineoverridevirtual

Reimplemented from JetTagComputer.

Definition at line 52 of file TemplatedJetProbabilityComputer.h.

References HLT_FULL_cff::DeltaR, JetTagComputer::TagInfoHelper::get(), mps_fire::i, reco::IPTagInfo< Container, Base >::impactParameterData(), PyquenDefaultSettings_cff::impactParameters, edm::Ref< C, T, F >::isNull(), TemplatedJetProbabilityComputer< Container, Base >::jetProbability(), TemplatedJetProbabilityComputer< Container, Base >::m_ipType, TemplatedJetProbabilityComputer< Container, Base >::m_useAllQualities, AlCaHLTBitMon_ParallelJobs::p, reco::IPTagInfo< Container, Base >::primaryVertex(), reco::IPTagInfo< Container, Base >::probabilities(), MetAnalyzer::pv(), reco::TrackBase::quality(), reco::IPTagInfo< Container, Base >::selectedTracks(), reco::btag::toTrack(), tracks, TemplatedJetProbabilityComputer< Container, Base >::useVariableJTA_, reco::IPTagInfo< Container, Base >::variableJTA(), and TemplatedJetProbabilityComputer< Container, Base >::varJTApars.

                                                              {
    const TagInfo& tkip = ti.get<TagInfo>();
    const Container& tracks(tkip.selectedTracks());
    const std::vector<float>& allProbabilities((tkip.probabilities(m_ipType)));
    const std::vector<reco::btag::TrackIPData>& impactParameters((tkip.impactParameterData()));

    if (tkip.primaryVertex().isNull())
      return 0;

    GlobalPoint pv(tkip.primaryVertex()->position().x(),
                   tkip.primaryVertex()->position().y(),
                   tkip.primaryVertex()->position().z());

    std::vector<float> probabilities;
    int i = 0;
    for (std::vector<float>::const_iterator it = allProbabilities.begin(); it != allProbabilities.end(); ++it, i++) {
      if (fabs(impactParameters[i].distanceToJetAxis.value()) < m_cutMaxDistToAxis &&  // distance to JetAxis
          (impactParameters[i].closestToJetAxis - pv).mag() < m_cutMaxDecayLen &&      // max decay len
          (m_useAllQualities == true ||
           reco::btag::toTrack(tracks[i])->quality(m_trackQuality))  // use selected track qualities
      ) {
        float p;
        if (m_trackSign == 0) {
          if (*it >= 0) {
            p = *it / 2.;
          } else {
            p = 1. + *it / 2.;
          }
        } else if (m_trackSign > 0) {
          if (*it >= 0)
            p = *it;
          else
            continue;
        } else {
          if (*it <= 0)
            p = -*it;
          else
            continue;
        }
        if (useVariableJTA_) {
          if (tkip.variableJTA(varJTApars)[i])
            probabilities.push_back(p);
        } else {
          if (m_deltaR <= 0 ||
              ROOT::Math::VectorUtil::DeltaR((*tkip.jet()).p4().Vect(), (*tracks[i]).momentum()) < m_deltaR)
            probabilities.push_back(p);
        }
      }
    }
    return jetProbability(probabilities);
  }

template<class Container , class Base >

double TemplatedJetProbabilityComputer< Container, Base >::jetProbability ( const std::vector< float > &  v ) const

inline

Definition at line 104 of file TemplatedJetProbabilityComputer.h.

References gather_cfg::cout, funct::exp(), cmsLHEtoEOSManager::l, log, TemplatedJetProbabilityComputer< Container, Base >::m_minTrackProb, SiStripPI::max, min(), and submitPVResolutionJobs::q.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::discriminator().

                                                         {
    int ngoodtracks = v.size();
    double SumJet = 0.;

    for (std::vector<float>::const_iterator q = v.begin(); q != v.end(); q++) {
      SumJet += (*q > m_minTrackProb) ? log(*q) : log(m_minTrackProb);
    }

    double ProbJet;
    double Loginvlog = 0;

    if (SumJet < 0.) {
      if (ngoodtracks >= 2) {
        Loginvlog = log(-SumJet);
      }
      double Prob = 1.;
      double lfact = 1.;
      for (int l = 1; l != ngoodtracks; l++) {
        lfact *= l;
        Prob += exp(l * Loginvlog - log(1. * lfact));
      }
      double LogProb = log(Prob);
      ProbJet = std::min(exp(std::max(LogProb + SumJet, -30.)), 1.);
    } else {
      ProbJet = 1.;
    }
    if (ProbJet > 1)
      std::cout << "ProbJet too high: " << ProbJet << std::endl;

    //double LogProbJet=-log(ProbJet);
    //  //return 1.-ProbJet;
    return -log10(ProbJet) / 4.;
  }

Member Data Documentation

template<class Container , class Base >

double TemplatedJetProbabilityComputer< Container, Base >::m_cutMaxDecayLen

private

Definition at line 145 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

double TemplatedJetProbabilityComputer< Container, Base >::m_cutMaxDistToAxis

private

Definition at line 146 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

double TemplatedJetProbabilityComputer< Container, Base >::m_deltaR

private

Definition at line 143 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

int TemplatedJetProbabilityComputer< Container, Base >::m_ipType

private

Definition at line 142 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::discriminator(), and TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

double TemplatedJetProbabilityComputer< Container, Base >::m_minTrackProb

private

Definition at line 141 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::jetProbability(), and TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

reco::TrackBase::TrackQuality TemplatedJetProbabilityComputer< Container, Base >::m_trackQuality

private

Definition at line 147 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

int TemplatedJetProbabilityComputer< Container, Base >::m_trackSign

private

Definition at line 144 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

bool TemplatedJetProbabilityComputer< Container, Base >::m_useAllQualities

private

Definition at line 148 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::discriminator(), and TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

bool TemplatedJetProbabilityComputer< Container, Base >::useVariableJTA_

private

Definition at line 139 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::discriminator(), and TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().

template<class Container , class Base >

reco::btag::variableJTAParameters TemplatedJetProbabilityComputer< Container, Base >::varJTApars

private

Definition at line 140 of file TemplatedJetProbabilityComputer.h.

Referenced by TemplatedJetProbabilityComputer< Container, Base >::discriminator(), and TemplatedJetProbabilityComputer< Container, Base >::TemplatedJetProbabilityComputer().