BaseIsolator.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_interface_BaseIsolator_h
#define PhysicsTools_PatAlgos_interface_BaseIsolator_h

#include "DataFormats/Common/interface/ValueMap.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include <atomic>

namespace pat { namespace helper {
class BaseIsolator {
    public:
        typedef edm::ValueMap<float> Isolation;
        BaseIsolator() {}
        BaseIsolator(const edm::ParameterSet &conf, edm::ConsumesCollector & iC, bool withCut) ;
        virtual ~BaseIsolator() {}
        virtual void beginEvent(const edm::Event &event, const edm::EventSetup &eventSetup) = 0;
        virtual void endEvent() = 0;

        template<typename AnyRef> bool test(const AnyRef &ref) const {
            bool ok = (getValue(ref.id(), ref.key()) < cut_);
            try_++; if (!ok) fail_++;
            return ok;
        }
        template<typename AnyRef> float getValue(const AnyRef &ref) const {
            return getValue(ref.id(), ref.key());
        }

        virtual std::string description() const = 0;
        void print(std::ostream &out) const ;
    protected:
        virtual float getValue(const edm::ProductID &id, size_t index) const = 0;
        edm::InputTag input_;
        edm::EDGetTokenT<Isolation> inputToken_;
        float cut_;
        mutable std::atomic<uint64_t> try_, fail_;
}; // class BaseIsolator
} } // namespaces

inline std::ostream & operator<<(std::ostream &stream, const pat::helper::BaseIsolator &iso) {
    iso.print(stream);
    return stream;
}
#endif