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;
};
}

#endif