TrackSelector.cc

Go to the documentation of this file.

#include <Math/GenVector/VectorUtil.h>

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/BTauReco/interface/TrackIPTagInfo.h"

#include "RecoBTag/SecondaryVertex/interface/TrackSelector.h"

using namespace reco; 
using namespace ROOT::Math;

TrackSelector::TrackSelector(const edm::ParameterSet &params) :
    minPixelHits(params.getParameter<unsigned int>("pixelHitsMin")),
    minTotalHits(params.getParameter<unsigned int>("totalHitsMin")),
    minPt(params.getParameter<double>("ptMin")),
    maxNormChi2(params.getParameter<double>("normChi2Max")),
    maxJetDeltaR(params.getParameter<double>("jetDeltaRMax")),
    maxDistToAxis(params.getParameter<double>("maxDistToAxis")),
    maxDecayLen(params.getParameter<double>("maxDecayLen")),
    sip2dValMin(params.getParameter<double>("sip2dValMin")),
    sip2dValMax(params.getParameter<double>("sip2dValMax")),
    sip2dSigMin(params.getParameter<double>("sip2dSigMin")),
    sip2dSigMax(params.getParameter<double>("sip2dSigMax")),
    sip3dValMin(params.getParameter<double>("sip3dValMin")),
    sip3dValMax(params.getParameter<double>("sip3dValMax")),
    sip3dSigMin(params.getParameter<double>("sip3dSigMin")),
    sip3dSigMax(params.getParameter<double>("sip3dSigMax")),
    useVariableJTA_(params.existsAs<bool>("useVariableJTA") ?  params.getParameter<bool>("useVariableJTA") : false)
{
    std::string qualityClass =
            params.getParameter<std::string>("qualityClass");
    if (qualityClass == "any" || qualityClass == "Any" ||
        qualityClass == "ANY" || qualityClass == "") {
        selectQuality = false;
        quality = reco::TrackBase::undefQuality;
    } else {
        selectQuality = true;
        quality = reco::TrackBase::qualityByName(qualityClass);
    }
    if (useVariableJTA_){
      varJTApars = {
        params.getParameter<double>("a_dR"),
        params.getParameter<double>("b_dR"),
        params.getParameter<double>("a_pT"),
        params.getParameter<double>("b_pT"),
        params.getParameter<double>("min_pT"),  
        params.getParameter<double>("max_pT"),
        params.getParameter<double>("min_pT_dRcut"),  
        params.getParameter<double>("max_pT_dRcut"),
        params.getParameter<double>("max_pT_trackPTcut") };
    }
}

bool
TrackSelector::operator () (const Track &track,
                            const btag::TrackIPData &ipData,
                            const Jet &jet,
                            const GlobalPoint &pv) const
{

 
  bool jtaPassed = false;
  if (useVariableJTA_) {
    jtaPassed = TrackIPTagInfo::passVariableJTA( varJTApars,
                    jet.pt(),track.pt(),
                    VectorUtil::DeltaR(jet.momentum(),track.momentum()));
  }
  else  jtaPassed = true;

  return (!selectQuality || track.quality(quality)) &&
    (minPixelHits <= 0 ||
     track.hitPattern().numberOfValidPixelHits() >= (int)minPixelHits) &&
    (minTotalHits <= 0 ||
     track.hitPattern().numberOfValidHits() >= (int)minTotalHits) &&
    track.pt() >= minPt &&
    track.normalizedChi2() < maxNormChi2 &&
    VectorUtil::DeltaR(jet.momentum(),
               track.momentum()) < maxJetDeltaR &&
    jtaPassed &&
    std::abs(ipData.distanceToJetAxis.value()) <= maxDistToAxis &&
    (ipData.closestToJetAxis - pv).mag() <= maxDecayLen &&
    ipData.ip2d.value()        >= sip2dValMin &&
    ipData.ip2d.value()        <= sip2dValMax &&
    ipData.ip2d.significance() >= sip2dSigMin &&
    ipData.ip2d.significance() <= sip2dSigMax &&
    ipData.ip3d.value()        >= sip3dValMin &&
    ipData.ip3d.value()        <= sip3dValMax &&
    ipData.ip3d.significance() >= sip3dSigMin &&
    ipData.ip3d.significance() <= sip3dSigMax;
}