VertexSorting.cc

Go to the documentation of this file.

#include <string>
#include <vector>
#include <map>

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

#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1D.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

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

using namespace reco; 

VertexSorting::SortCriterium
VertexSorting::getSortCriterium(const std::string &criterium)
{
    if (criterium == "dist3dError")
        return sortDist3dErr;
    if (criterium == "dist3dValue")
        return sortDist3dVal;
    if (criterium == "dist3dSignificance")
        return sortDist3dSig;
    if (criterium == "dist2dError")
        return sortDist2dErr;
    if (criterium == "dist2dValue")
        return sortDist2dVal;
    if (criterium == "dist2dSignificance")
        return sortDist2dSig;

    throw cms::Exception("InvalidArgument")
        << "Vertex sort criterium \"" << criterium << "\" is invalid."
        << std::endl;
}

VertexSorting::VertexSorting(const edm::ParameterSet &params) :
    sortCriterium(getSortCriterium(params.getParameter<std::string>("sortCriterium")))
{
}

// identify most probable SV (closest to interaction point, significance-wise)
// FIXME: identify if this is the best strategy!

std::vector<unsigned int> VertexSorting::operator () (
        const std::vector<SecondaryVertex> &svCandidates) const
{
    Measurement1D (SecondaryVertex::*measurementFn)() const = 0;
    switch(sortCriterium) {
        case sortDist3dErr:
        case sortDist3dVal:
        case sortDist3dSig:
        measurementFn = &SecondaryVertex::dist3d;
        break;
        case sortDist2dErr:
        case sortDist2dVal:
        case sortDist2dSig:
        measurementFn = &SecondaryVertex::dist2d;
        break;
    }

    double (Measurement1D::*valueFn)() const = 0;
    switch(sortCriterium) {
        case sortDist3dErr:
        case sortDist2dErr:
        valueFn = &Measurement1D::error;
        break;
        case sortDist3dVal:
        case sortDist2dVal:
        valueFn = &Measurement1D::value;
        break;
        case sortDist3dSig:
        case sortDist2dSig:
        valueFn = &Measurement1D::significance;
        break;
    }

    std::multimap<double, unsigned int> sortedMap;
    unsigned int i = 0;
    for(std::vector<SecondaryVertex>::const_iterator iter =
        svCandidates.begin(); iter != svCandidates.end(); iter++) {

        double value = std::abs((((*iter).*measurementFn)().*valueFn)());
        sortedMap.insert(std::make_pair(value, i++));
    }

    std::vector<unsigned int> result;
    for(std::multimap<double, unsigned int>::const_iterator iter =
        sortedMap.begin(); iter != sortedMap.end(); iter++)
        result.push_back(iter->second);

    return result;
}