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