IsolatorByDeposit.h

Go to the documentation of this file.

#ifndef MuonIsolation_IsolatorByDeposit_H
#define MuonIsolation_IsolatorByDeposit_H
 
#include "RecoMuon/MuonIsolation/interface/MuIsoBaseIsolator.h"
#include <vector>
 
namespace muonisolation {
  class IsolatorByDeposit : public MuIsoBaseIsolator {
  public:
    typedef MuIsoBaseIsolator::DepositContainer DepositContainer;
 
    struct ConeSizeFunction {
      virtual ~ConeSizeFunction() = default;
      virtual float coneSize(float eta, float pt) const = 0;
    };
 
    IsolatorByDeposit(float conesize, const std::vector<double>& weights);
    IsolatorByDeposit(const ConeSizeFunction* conesize, const std::vector<double>& weights);
 
    IsolatorByDeposit(float conesize, const std::vector<double>& weights, const std::vector<double>& thresh);
    IsolatorByDeposit(const ConeSizeFunction* conesize,
                      const std::vector<double>& weights,
                      const std::vector<double>& thresh);
 
    ~IsolatorByDeposit() override {}
 
    virtual void setWeights(const std::vector<double>& weights) { theWeights = weights; }
 
    Result result(const DepositContainer& deposits, const edm::Event* = nullptr) const override;
 
    /*   virtual int resultInt(DepositContainer deposits) const; */
 
    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_FLOAT_TYPE; }
 
  private:
    // Compute the weighted sum of deposits of different type within dRcone
    double weightedSum(const DepositContainer& deposits, float dRcone) const;
 
  private:
    const ConeSizeFunction* theConeSizeFunction;
    float theConeSize;
    std::vector<double> theWeights;
    std::vector<double> theDepThresholds;
  };
}  // namespace muonisolation
 
#endif