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