IIsolationSelectionTool.h

Go to the documentation of this file.

/*
 Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef ISOLATIONSELECTION_IISOLATIONSELECTIONTOOL_H
#define ISOLATIONSELECTION_IISOLATIONSELECTIONTOOL_H
 
// Framework include(s):
#include "AsgTools/IAsgTool.h"
#include "PATCore/AcceptData.h"
#include "PATCore/AcceptInfo.h"
 
// EDM include(s):
#include "xAODEgamma/Electron.h"
#include "xAODEgamma/Photon.h"
#include "xAODMuon/Muon.h"
#include "xAODPrimitives/IsolationType.h"
// #include "IsolationSelection/strMuon.h"
 
// std include(s):
#include <map>
#include <memory>

 
 
namespace CP {
    struct strObj;
    class IsolationWP;
    class IIsolationSelectionTool : virtual public asg::IAsgTool {
        ASG_TOOL_INTERFACE(CP::IIsolationSelectionTool)
 
    public:
        // Cut on object
        virtual asg::AcceptData accept(const xAOD::Photon& x) const = 0;
        virtual asg::AcceptData accept(const xAOD::Electron& x) const = 0;
        virtual asg::AcceptData accept(const xAOD::Muon& x) const = 0;
        virtual asg::AcceptData accept(const strObj& x) const = 0;
        virtual asg::AcceptData accept(const xAOD::IParticle& x) const = 0;
        virtual StatusCode setIParticleCutsFrom(xAOD::Type::ObjectType ObjType) = 0;
 
        virtual const asg::AcceptInfo& getPhotonAcceptInfo() const = 0;
        virtual const asg::AcceptInfo& getElectronAcceptInfo() const = 0;
        virtual const asg::AcceptInfo& getMuonAcceptInfo() const = 0;
        virtual const asg::AcceptInfo& getObjAcceptInfo() const = 0;
 
        virtual const std::vector<std::unique_ptr<IsolationWP>>& getMuonWPs() const = 0;
        virtual const std::vector<std::unique_ptr<IsolationWP>>& getElectronWPs() const = 0;
        virtual const std::vector<std::unique_ptr<IsolationWP>>& getPhotonWPs() const = 0;
        virtual const std::vector<std::unique_ptr<IsolationWP>>& getObjWPs() const = 0;
    };
}  // namespace CP
#endif