GhostTrackComputer Class Reference

#include <GhostTrackComputer.h>

Public Member Functions
	GhostTrackComputer (const edm::ParameterSet &params)
reco::TaggingVariableList	operator() (const reco::TrackIPTagInfo &ipInfo, const reco::SecondaryVertexTagInfo &svInfo) const

Private Member Functions
const reco::btag::TrackIPData &	threshTrack (const reco::TrackIPTagInfo &trackIPTagInfo, const reco::btag::SortCriteria sort, const reco::Jet &jet, const GlobalPoint &pv) const

Private Attributes
double	charmCut
double	minTrackWeight
reco::btag::SortCriteria	sortCriterium
reco::TrackSelector	trackNoDeltaRSelector
reco::V0Filter	trackPairV0Filter
reco::TrackSelector	trackSelector
bool	vertexMassCorrection

Detailed Description

Definition at line 14 of file GhostTrackComputer.h.

Constructor & Destructor Documentation

GhostTrackComputer::GhostTrackComputer

(

const edm::ParameterSet &

params

)

Definition at line 42 of file GhostTrackComputer.cc.

                                                                    :
    charmCut(params.getParameter<double>("charmCut")),
    sortCriterium(TrackSorting::getCriterium(params.getParameter<std::string>("trackSort"))),
    trackSelector(params.getParameter<edm::ParameterSet>("trackSelection")),
    trackNoDeltaRSelector(dropDeltaR(params.getParameter<edm::ParameterSet>("trackSelection"))),
    minTrackWeight(params.getParameter<double>("minimumTrackWeight")),
    trackPairV0Filter(params.getParameter<edm::ParameterSet>("trackPairV0Filter"))
{
}

Member Function Documentation

TaggingVariableList GhostTrackComputer::operator()	(	const reco::TrackIPTagInfo &	ipInfo,
		const reco::SecondaryVertexTagInfo &	svInfo
	)		const

Definition at line 99 of file GhostTrackComputer.cc.

References addMeas(), edm::RefVector< C, T, F >::begin(), data, deltaR(), HLT_25ns14e33_v1_cff::DeltaR, edm::RefVector< C, T, F >::end(), reco::btau::etaRel(), reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::flightDirection(), reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::flightDistance(), reco::btau::flightDistance2dSig, reco::btau::flightDistance2dVal, reco::btau::flightDistance3dSig, reco::btau::flightDistance3dVal, reco::Vertex::hasRefittedTracks(), reco::TrackBase::hitPattern(), i, customizeTrackingMonitorSeedNumber::idx, reco::IPTagInfo< Container, Base >::impactParameterData(), reco::TaggingVariableList::insert(), reco::btag::IP2DSig, reco::btag::IP3DSig, edm::Ref< C, T, F >::isNonnull(), j, metsig::jet, reco::btau::jetEta, reco::btau::jetNSecondaryVertices, reco::btau::jetNSingleTrackVertices, reco::btau::jetPt, minTrackWeight, reco::TrackBase::momentum(), gen::n, reco::TrackBase::normalizedChi2(), reco::btag::Vertices::NoVertex, reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::nVertices(), convertSQLiteXML::ok, reco::HitPattern::pixelLayersWithMeasurement(), reco::IPTagInfo< Container, Base >::primaryVertex(), reco::btag::Vertices::PseudoVertex, MetAnalyzer::pv(), reco::btag::Vertices::RecoVertex, reco::Vertex::refittedTrack(), reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::secondaryVertex(), reco::IPTagInfo< Container, Base >::selectedTracks(), sortCriterium, reco::IPTagInfo< Container, Base >::sortedIndexes(), mathSSE::sqrt(), threshTrack(), reco::btau::trackChi2, reco::btau::trackDecayLenVal, reco::btau::trackDeltaR, reco::HitPattern::trackerLayersWithMeasurement(), reco::btau::trackEta, reco::btau::trackEtaRel, reco::btau::trackGhostTrackDistSig, reco::btau::trackGhostTrackDistVal, reco::btau::trackGhostTrackWeight, reco::btau::trackJetDistVal, reco::btau::trackMomentum, reco::btau::trackNPixelHits, reco::btau::trackNTotalHits, trackPairV0Filter, reco::btau::trackPtRel, testEve_cfg::tracks, trackSelector, reco::btau::trackSip2dSig, reco::btau::trackSip2dSigAboveCharm, reco::btau::trackSip2dVal, reco::btau::trackSip3dSig, reco::btau::trackSip3dSigAboveCharm, reco::btau::trackSip3dVal, reco::btau::trackSumJetDeltaR, reco::btau::trackSumJetEtRatio, reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::trackWeight(), reco::btau::vertexCategory, reco::btau::vertexEnergyRatio, reco::btau::vertexJetDeltaR, reco::btau::vertexMass, reco::btau::vertexNTracks, reco::TemplatedSecondaryVertexTagInfo< IPTI, VTX >::vertexTracks(), and w.

{
    using namespace ROOT::Math;

    edm::RefToBase<Jet> jet = ipInfo.jet();
    math::XYZVector jetDir = jet->momentum().Unit();
    bool havePv = ipInfo.primaryVertex().isNonnull();
    GlobalPoint pv;
    if (havePv)
        pv = GlobalPoint(ipInfo.primaryVertex()->x(),
                         ipInfo.primaryVertex()->y(),
                         ipInfo.primaryVertex()->z());

    btag::Vertices::VertexType vtxType = btag::Vertices::NoVertex;

    TaggingVariableList vars;

    vars.insert(btau::jetPt, jet->pt(), true);
    vars.insert(btau::jetEta, jet->eta(), true);

    TrackKinematics allKinematics;
    TrackKinematics vertexKinematics;
    TrackKinematics trackKinematics;

    std::pair<double, double> vertexDist2D, vertexDist3D;
    std::pair<double, double> tracksDist2D, tracksDist3D;

    unsigned int nVertices = 0;
    unsigned int nVertexTracks = 0;
    unsigned int nTracks = 0;
    for(unsigned int i = 0; i < svInfo.nVertices(); i++) {
        const Vertex &vertex = svInfo.secondaryVertex(i);
        bool hasRefittedTracks = vertex.hasRefittedTracks();
        TrackRefVector tracks = svInfo.vertexTracks(i);
        unsigned int n = 0;
        for(TrackRefVector::const_iterator track = tracks.begin();
            track != tracks.end(); track++)
            if (svInfo.trackWeight(i, *track) >= minTrackWeight)
                n++;

        if (n < 1)
            continue;
        bool isTrackVertex = (n == 1);
        ++*(isTrackVertex ? &nTracks : &nVertices);

        addMeas(*(isTrackVertex ? &tracksDist2D : &vertexDist2D),
                    svInfo.flightDistance(i, true));
        addMeas(*(isTrackVertex ? &tracksDist3D : &vertexDist3D),
                    svInfo.flightDistance(i, false));

        TrackKinematics &kin = isTrackVertex ? trackKinematics
                                             : vertexKinematics;
        for(TrackRefVector::const_iterator track = tracks.begin();
            track != tracks.end(); track++) {
            float w = svInfo.trackWeight(i, *track);
            if (w < minTrackWeight)
                continue;
            if (hasRefittedTracks) {
                Track actualTrack =
                        vertex.refittedTrack(*track);
                kin.add(actualTrack, w);
                vars.insert(btau::trackEtaRel, reco::btau::etaRel(jetDir,
                        actualTrack.momentum()), true);
            } else {
                kin.add(**track, w);
                vars.insert(btau::trackEtaRel, reco::btau::etaRel(jetDir,
                        (*track)->momentum()), true);
            }
            if (!isTrackVertex)
                nVertexTracks++;
        }
    }

    Measurement1D dist2D, dist3D;
    if (nVertices) {
        vtxType = btag::Vertices::RecoVertex;

        if (nVertices == 1 && nTracks) {
            vertexDist2D.first += tracksDist2D.first;
            vertexDist2D.second += tracksDist2D.second;
            vertexDist3D.first += tracksDist3D.first;
            vertexDist3D.second += tracksDist3D.second;
            vertexKinematics += trackKinematics;
        }

        dist2D = Measurement1D(
                vertexDist2D.first / vertexDist2D.second,
                std::sqrt(1. / vertexDist2D.second));
        dist3D = Measurement1D(
                vertexDist3D.first / vertexDist3D.second,
                std::sqrt(1. / vertexDist3D.second));

        vars.insert(btau::jetNSecondaryVertices, nVertices, true);
        vars.insert(btau::vertexNTracks, nVertexTracks, true);
    } else if (nTracks) {
        vtxType = btag::Vertices::PseudoVertex;
        vertexKinematics = trackKinematics;

        dist2D = Measurement1D(
                tracksDist2D.first / tracksDist2D.second,
                std::sqrt(1. / tracksDist2D.second));
        dist3D = Measurement1D(
                tracksDist3D.first / tracksDist3D.second,
                std::sqrt(1. / tracksDist3D.second));
    }

    if (nVertices || nTracks) {
        vars.insert(btau::flightDistance2dVal, dist2D.value(), true);
        vars.insert(btau::flightDistance2dSig, dist2D.significance(), true);
        vars.insert(btau::flightDistance3dVal, dist3D.value(), true);
        vars.insert(btau::flightDistance3dSig, dist3D.significance(), true);
        vars.insert(btau::vertexJetDeltaR,
                    Geom::deltaR(svInfo.flightDirection(0), jetDir),true);
        vars.insert(btau::jetNSingleTrackVertices, nTracks, true);
    }

    std::vector<std::size_t> indices = ipInfo.sortedIndexes(sortCriterium);
    const std::vector<btag::TrackIPData> &ipData =
                        ipInfo.impactParameterData();
    const edm::RefVector<TrackCollection> &tracks =
                        ipInfo.selectedTracks();

    TrackRef trackPairV0Test[2];
    for(unsigned int i = 0; i < indices.size(); i++) {
        std::size_t idx = indices[i];
        const btag::TrackIPData &data = ipData[idx];
        const TrackRef &trackRef = tracks[idx];
        const Track &track = *trackRef;

        // filter track

        if (!trackSelector(track, data, *jet, pv))
            continue;

        // add track to kinematics for all tracks in jet

        allKinematics.add(track);

        // check against all other tracks for V0 track pairs

        trackPairV0Test[0] = tracks[idx];
        bool ok = true;
        for(unsigned int j = 0; j < indices.size(); j++) {
            if (i == j)
                continue;

            std::size_t pairIdx = indices[j];
            const btag::TrackIPData &pairTrackData =
                            ipData[pairIdx];
            const TrackRef &pairTrackRef = tracks[pairIdx];
            const Track &pairTrack = *pairTrackRef;

            if (!trackSelector(pairTrack, pairTrackData, *jet, pv))
                continue;

            trackPairV0Test[1] = pairTrackRef;
            if (!trackPairV0Filter(trackPairV0Test, 2)) {
                ok = false;
                break;
            }
        }
        if (!ok)
            continue;

        // add track variables

        math::XYZVector trackMom = track.momentum();
        double trackMag = std::sqrt(trackMom.Mag2());

        vars.insert(btau::trackSip3dVal, data.ip3d.value(), true);
        vars.insert(btau::trackSip3dSig, data.ip3d.significance(), true);
        vars.insert(btau::trackSip2dVal, data.ip2d.value(), true);
        vars.insert(btau::trackSip2dSig, data.ip2d.significance(), true);
        vars.insert(btau::trackJetDistVal, data.distanceToJetAxis.value(), true);
        vars.insert(btau::trackGhostTrackDistVal, data.distanceToGhostTrack.value(), true);
        vars.insert(btau::trackGhostTrackDistSig, data.distanceToGhostTrack.significance(), true);
        vars.insert(btau::trackGhostTrackWeight, data.ghostTrackWeight, true);
        vars.insert(btau::trackDecayLenVal, havePv ? (data.closestToGhostTrack - pv).mag() : -1.0, true);

        vars.insert(btau::trackMomentum, trackMag, true);
        vars.insert(btau::trackEta, trackMom.Eta(), true);

        vars.insert(btau::trackChi2, track.normalizedChi2(), true);
        vars.insert(btau::trackNPixelHits, track.hitPattern().pixelLayersWithMeasurement(), true);
        vars.insert(btau::trackNTotalHits, track.hitPattern().trackerLayersWithMeasurement(), true);
        vars.insert(btau::trackPtRel, VectorUtil::Perp(trackMom, jetDir), true);
        vars.insert(btau::trackDeltaR, VectorUtil::DeltaR(trackMom, jetDir), true);
    } 

    vars.insert(btau::vertexCategory, vtxType, true);
    vars.insert(btau::trackSumJetDeltaR,
                VectorUtil::DeltaR(allKinematics.vectorSum(), jetDir), true);
    vars.insert(btau::trackSumJetEtRatio,
                allKinematics.vectorSum().Et() / ipInfo.jet()->et(), true);
    vars.insert(btau::trackSip3dSigAboveCharm,
                threshTrack(ipInfo, reco::btag::IP3DSig, *jet, pv)
                                    .ip3d.significance(),
                true);
    vars.insert(btau::trackSip2dSigAboveCharm,
                threshTrack(ipInfo, reco::btag::IP2DSig, *jet, pv)
                                    .ip2d.significance(),
                true);

    if (vtxType != btag::Vertices::NoVertex) {
        math::XYZTLorentzVector allSum = allKinematics.vectorSum();
        math::XYZTLorentzVector vertexSum = vertexKinematics.vectorSum();

        vars.insert(btau::vertexMass, vertexSum.M(), true);
        if (allKinematics.numberOfTracks())
            vars.insert(btau::vertexEnergyRatio,
                        vertexSum.E() / allSum.E(), true);
        else
            vars.insert(btau::vertexEnergyRatio, 1, true);
    }

    vars.finalize();

    return vars;
}

const btag::TrackIPData & GhostTrackComputer::threshTrack ( const reco::TrackIPTagInfo &  trackIPTagInfo, const reco::btag::SortCriteria  sort, const reco::Jet &  jet, const GlobalPoint &  pv  ) const

private

Definition at line 53 of file GhostTrackComputer.cc.

References reco::TrackKinematics::add(), charmCut, data, reco::IPTagInfo< Container, Base >::impactParameterData(), reco::IPTagInfo< Container, Base >::selectedTracks(), reco::IPTagInfo< Container, Base >::sortedIndexes(), trackNoDeltaRSelector, testEve_cfg::tracks, and reco::TrackKinematics::vectorSum().

Referenced by operator()().

{
    const edm::RefVector<TrackCollection> &tracks =
                    trackIPTagInfo.selectedTracks();
    const std::vector<btag::TrackIPData> &ipData =
                    trackIPTagInfo.impactParameterData();
    std::vector<std::size_t> indices = trackIPTagInfo.sortedIndexes(sort);

    TrackKinematics kin;
    for(std::vector<std::size_t>::const_iterator iter = indices.begin();
        iter != indices.end(); ++iter) {
        const btag::TrackIPData &data = ipData[*iter];
        const Track &track = *tracks[*iter];

        if (!trackNoDeltaRSelector(track, data, jet, pv))
            continue;

        kin.add(track);
        if (kin.vectorSum().M() > charmCut) 
            return data;
    }

    static const btag::TrackIPData dummy = {
        GlobalPoint(),
        GlobalPoint(),
        Measurement1D(-1.0, 1.0),
        Measurement1D(-1.0, 1.0),
        Measurement1D(-1.0, 1.0),
        Measurement1D(-1.0, 1.0),
        0.
    };
    return dummy;
}

Member Data Documentation

double GhostTrackComputer::charmCut

private

Definition at line 29 of file GhostTrackComputer.h.

Referenced by threshTrack().

double GhostTrackComputer::minTrackWeight

private

Definition at line 33 of file GhostTrackComputer.h.

Referenced by operator()().

reco::btag::SortCriteria GhostTrackComputer::sortCriterium

private

Definition at line 30 of file GhostTrackComputer.h.

Referenced by operator()().

reco::TrackSelector GhostTrackComputer::trackNoDeltaRSelector

private

Definition at line 32 of file GhostTrackComputer.h.

Referenced by threshTrack().

reco::V0Filter GhostTrackComputer::trackPairV0Filter

private

Definition at line 35 of file GhostTrackComputer.h.

Referenced by operator()().

reco::TrackSelector GhostTrackComputer::trackSelector

private

Definition at line 31 of file GhostTrackComputer.h.

Referenced by operator()().

bool GhostTrackComputer::vertexMassCorrection

private

Definition at line 34 of file GhostTrackComputer.h.