MVAComputerCache.h

Go to the documentation of this file.

#ifndef PhysicsTools_MVAComputer_MVAComputerCache_h
#define PhysicsTools_MVAComputer_MVAComputerCache_h
// -*- C++ -*-
//
// Package:     MVAComputerCache
// Class  :     MVAComputerCache
//

//
// Author:  Christophe Saout <christophe.saout@cern.ch>
// Created:     Sat Feb 23 15:38 CEST 2007
//

#include <memory>

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "PhysicsTools/MVAComputer/interface/Calibration.h"
#include "PhysicsTools/MVAComputer/interface/MVAComputer.h"

namespace PhysicsTools {

class MVAComputerCache {
    public:
    MVAComputerCache();
    ~MVAComputerCache();

    bool update(const Calibration::MVAComputer *computer);
    bool update(const Calibration::MVAComputerContainer *container,
                const char *calib);

    template<class T>
    bool update(const edm::EventSetup &es)
    {
        edm::ESHandle<Calibration::MVAComputer> handle;
        es.get<T>().get(handle);
        return update(handle.product());
    }

    template<class T>
    bool update(const edm::EventSetup &es, const char *calib)
    {
        edm::ESHandle<Calibration::MVAComputerContainer> handle;
        es.get<T>().get(handle);
        return update(handle.product(), calib);
    }

    template<class T>
    bool update(const char *label, const edm::EventSetup &es)
    {
        edm::ESHandle<Calibration::MVAComputer> handle;
        es.get<T>().get(label, handle);
        return update(handle.product());
    }

    template<class T>
    bool update(const char *label, const edm::EventSetup &es,
                const char *calib)
    {
        edm::ESHandle<Calibration::MVAComputerContainer> handle;
        es.get<T>().get(label, handle);
        return update(handle.product(), calib);
    }

    operator bool() const { return computer.get(); }

    MVAComputer &operator * () { return *computer; }
    const MVAComputer &operator * () const { return *computer; }

    MVAComputer *operator -> () { return computer.get(); }
    const MVAComputer *operator -> () const { return computer.get(); }

    MVAComputer *get() { return computer.get(); }
    const MVAComputer *get() const { return computer.get(); }

    std::unique_ptr<MVAComputer> release();

    void reset() { computer.reset(); }

    private:
    Calibration::MVAComputerContainer::CacheId  containerCacheId;
    Calibration::MVAComputer::CacheId       computerCacheId;
    std::unique_ptr<MVAComputer>            computer;
};

} // namespace PhysicsTools

#endif // PhysicsTools_MVAComputer_MVAComputerCache_h