IsolatorByDepositCount.h

Go to the documentation of this file.

#ifndef MuonIsolation_IsolatorByDepositCount_H
#define MuonIsolation_IsolatorByDepositCount_H
 
#include "RecoMuon/MuonIsolation/interface/MuIsoBaseIsolator.h"
#include <vector>
 
namespace muonisolation {
  class IsolatorByDepositCount : public MuIsoBaseIsolator {
  public:
    typedef MuIsoBaseIsolator::DepositContainer DepositContainer;
 
    struct ConeSizeFunction {
      virtual ~ConeSizeFunction() = default;
      virtual float coneSize(float eta, float pt) const = 0;
    };
 
    IsolatorByDepositCount(float conesize, const std::vector<double>& thresh);
    IsolatorByDepositCount(const ConeSizeFunction* conesize, const std::vector<double>& thresh);
 
    ~IsolatorByDepositCount() override = default;
 
    Result result(const DepositContainer& deposits, const edm::Event* = nullptr) const override;
 
    void setConeSize(float conesize) {
      theConeSize = conesize;
      theConeSizeFunction = nullptr;
    }
 
    void setConeSize(ConeSizeFunction* conesize) { theConeSizeFunction = conesize; }
 
    virtual float coneSize(float eta, float pT) const {
      return theConeSizeFunction ? theConeSizeFunction->coneSize(eta, pT) : theConeSize;
    }
 
    ResultType resultType() const override { return ISOL_INT_TYPE; }
 
  private:
    const ConeSizeFunction* theConeSizeFunction;
    float theConeSize;
    std::vector<double> theDepThresholds;
  };
}  // namespace muonisolation
 
#endif