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

#endif