CombinedMVAJetTagComputer.h

Go to the documentation of this file.

#ifndef RecoBTau_JetTagComputer_CombinedMVAJetTagComputer_h
#define RecoBTau_JetTagComputer_CombinedMVAJetTagComputer_h

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

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/BTauReco/interface/TaggingVariable.h"
#include "RecoBTau/JetTagComputer/interface/JetTagComputer.h"
#include "RecoBTau/JetTagComputer/interface/GenericMVAJetTagComputer.h"

class CombinedMVAJetTagComputer : public GenericMVAJetTagComputer {
    public:
    CombinedMVAJetTagComputer(const edm::ParameterSet &parameters);
    virtual ~CombinedMVAJetTagComputer();

    virtual void initialize(const JetTagComputerRecord & record);

    virtual reco::TaggingVariableList
    taggingVariables(const TagInfoHelper &info) const;

    private:
    struct Computer {
        std::string     name;
        bool            discriminator;
        bool            variables;

        const JetTagComputer    *computer;
        std::vector<int>    indices;
    };

    std::vector<Computer> computers;
};

#endif // RecoBTau_JetTagComputer_CombinedMVAJetTagComputer_h