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

  virtual ~IsolatorByDepositCount() {}

  virtual Result result(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_INT_TYPE;}


private:
  const ConeSizeFunction * theConeSizeFunction;
  float theConeSize;
  std::vector<double> theDepThresholds;
};
}

#endif