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

  virtual ~IsolatorByDeposit() {}

  virtual void setWeights(const std::vector<double>& weights) {theWeights=weights;}

  virtual Result result(const DepositContainer& deposits, const edm::Event* = 0) const;

/*   virtual int resultInt(DepositContainer deposits) const; */



  void setConeSize(float conesize) { theConeSize = conesize; theConeSizeFunction = 0;} 

  void setConeSize(ConeSizeFunction * conesize) { theConeSizeFunction = conesize; }


  virtual float coneSize(float eta, float pT) const {
    return theConeSizeFunction ? theConeSizeFunction->coneSize(eta,pT) : theConeSize;
  }

  virtual ResultType resultType() const { 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