/data/doxygen/doxygen-1.7.3/gen/CMSSW_4_2_8/src/RecoBTau/JetTagComputer/src/TagInfoMVACategorySelector.cc

Go to the documentation of this file.

#include <ext/functional>
#include <string>
#include <vector>

#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Utilities/interface/EDMException.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/BTauReco/interface/TaggingVariable.h"
#include "PhysicsTools/MVAComputer/interface/Calibration.h"
#include "RecoBTau/JetTagComputer/interface/TagInfoMVACategorySelector.h"

using namespace reco;

TagInfoMVACategorySelector::TagInfoMVACategorySelector(
                                        const edm::ParameterSet &params)
{
        std::string variableName =
                params.getParameter<std::string>("categoryVariableName");

        categoryVariable = getTaggingVariableName(variableName);
        if (categoryVariable >= btau::lastTaggingVariable)
                throw cms::Exception("TagInfoMVACategorySelector")
                        << "No such tagging variable \""
                        << categoryVariable << "\"." << std::endl;

        categoryLabels = params.getParameter<std::vector<std::string> >(
                                                        "calibrationRecords");
        for(std::vector<std::string>::iterator iter = categoryLabels.begin();
            iter != categoryLabels.end(); iter++)
                if (*iter == " " || *iter == "-" || *iter == "*")
                        *iter = "";
}

TagInfoMVACategorySelector::~TagInfoMVACategorySelector()
{
}

int TagInfoMVACategorySelector::findCategory(
                        const TaggingVariableList &taggingVariables) const
{
        int index = (int)taggingVariables.get(categoryVariable, -1);

        if (index < 0 || (unsigned int)index >= categoryLabels.size())
                return -1;

        return index;
}