OverlapTest.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_interface_OverlapTest_h
#define PhysicsTools_PatAlgos_interface_OverlapTest_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "PhysicsTools/PatUtils/interface/StringParserTools.h"
#include "PhysicsTools/PatUtils/interface/PATDiObjectProxy.h"

namespace pat { namespace helper {

// Base class for a test for overlaps
class OverlapTest {
    public:
        OverlapTest(const std::string &name, const edm::ParameterSet &iConfig, edm::ConsumesCollector & iC) :
            srcToken_(iC.consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("src"))),
            name_(name),
            requireNoOverlaps_(iConfig.getParameter<bool>("requireNoOverlaps")) {}
        virtual ~OverlapTest() {}
        virtual void readInput(const edm::Event & iEvent, const edm::EventSetup &iSetup) = 0;
        virtual bool fillOverlapsForItem(const reco::Candidate &item, reco::CandidatePtrVector &overlapsToFill) const = 0;
        virtual void done() {}
        // -- basic getters ---

        const std::string & name() const { return name_; }
        bool requireNoOverlaps() const { return requireNoOverlaps_; }
    protected:
        edm::EDGetTokenT<reco::CandidateView> srcToken_;
        std::string   name_;
        bool          requireNoOverlaps_;
};

class BasicOverlapTest : public OverlapTest {
    public:
        BasicOverlapTest(const std::string &name, const edm::ParameterSet &iConfig, edm::ConsumesCollector && iC) :
            OverlapTest(name, iConfig, iC),
            presel_(iConfig.getParameter<std::string>("preselection")),
            deltaR_(iConfig.getParameter<double>("deltaR")),
            checkRecoComponents_(iConfig.getParameter<bool>("checkRecoComponents")),
            pairCut_(iConfig.getParameter<std::string>("pairCut")) {}
        // implementation of mother methods
        virtual void readInput(const edm::Event & iEvent, const edm::EventSetup &iSetup) ;
        virtual bool fillOverlapsForItem(const reco::Candidate &item, reco::CandidatePtrVector &overlapsToFill) const ;
    protected:
        // ---- configurables ----
        PATStringCutObjectSelector presel_;
        double deltaR_;
        bool checkRecoComponents_;
        StringCutObjectSelector<pat::DiObjectProxy> pairCut_;
        // ---- working variables ----
        edm::Handle<reco::CandidateView> candidates_;
        std::vector<bool> isPreselected_;
};

class OverlapBySuperClusterSeed : public OverlapTest {
    public:
        // constructor: nothing except initialize the base class
        OverlapBySuperClusterSeed(const std::string &name, const edm::ParameterSet &iConfig, edm::ConsumesCollector && iC) : OverlapTest(name, iConfig, iC) {}
        // every event: nothing except read the input list
        virtual void readInput(const edm::Event & iEvent, const edm::EventSetup &iSetup) {
            iEvent.getByToken(srcToken_, others_);
        }
        virtual bool fillOverlapsForItem(const reco::Candidate &item, reco::CandidatePtrVector &overlapsToFill) const ;
    protected:
//         edm::Handle<edm::View<reco::RecoCandidate> > others_;
        edm::Handle<reco::CandidateView> others_;
};



} } // namespaces

#endif