/data/doxygen/doxygen-1.7.3/gen/CMSSW_4_2_8/src/RecoBTag/SecondaryVertex/src/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"))
{
        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);
        }
}

bool
TrackSelector::operator () (const Track &track,
                            const TrackIPTagInfo::TrackIPData &ipData,
                            const Jet &jet,
                            const GlobalPoint &pv) const
{
        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 &&
               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;
}