GenericMVAComputerCache.h

Go to the documentation of this file.

#ifndef RecoBTau_JetTagComputer_GenericMVAComputerCache_h
#define RecoBTau_JetTagComputer_GenericMVAComputerCache_h

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

#include "DataFormats/BTauReco/interface/BaseTagInfo.h"
#include "CondFormats/PhysicsToolsObjects/interface/MVAComputer.h"
#include "RecoBTau/JetTagComputer/interface/GenericMVAComputer.h"

class GenericMVAComputerCache {
    public:
    GenericMVAComputerCache(const std::vector<std::string> &labels);
    ~GenericMVAComputerCache();

    bool
    update(const PhysicsTools::Calibration::MVAComputerContainer *calib);

    inline GenericMVAComputer *getComputer(int index) const
    { return index >= 0 ? computers[index].computer.get() : 0; }

    inline bool isEmpty() const { return empty; }

    private:
    struct IndividualComputer {
        IndividualComputer();
        IndividualComputer(const IndividualComputer &orig);
        ~IndividualComputer();

        std::string                     label;
        std::auto_ptr<GenericMVAComputer>           computer;
        PhysicsTools::Calibration::MVAComputer::CacheId     cacheId;
    };

    std::vector<IndividualComputer>                 computers;
    PhysicsTools::Calibration::MVAComputerContainer::CacheId    cacheId;
    bool                                initialized;
    bool                                empty;
};

#endif // RecoBTau_JetTagComputer_GenericMVAComputerCache_h