HGCalIsoCalculator.h

Go to the documentation of this file.

/*
 * PhotonIsoProducer.h
 *
 *  Created on: 13 Oct 2017
 *      Author: jkiesele, ncsmith
 */

#ifndef RecoEgamma_EgammaTools_HGCalIsoCalculator_h
#define RecoEgamma_EgammaTools_HGCalIsoCalculator_h

#include "DataFormats/HGCRecHit/interface/HGCRecHitCollections.h"
#include "RecoLocalCalo/HGCalRecAlgos/interface/RecHitTools.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"

/*
 *
 * This class calculates the energy around the photon/electron in DR=0.15 that is
 * not associated to the seed cluster.
 * The energy is summed in 5 rings (default) between HGCal layer 1 and 30
 * This gives back 5 calorimeter isolation values.
 * Only the first values should be significantly affected by pileup.
 *
 * Usage:
 *
 * PhotonHGCalIsoCalculator prod;
 * prod.setRecHitTools(rechittools)
 * prod.setRecHits(recHitsEEHandle,recHitsFHHandle,recHitsBHHandle)
 *
 * <optional>
 * prod.setDeltaR(0.15)
 * <optional>
 * prod.setNRings(5)
 * <optional>
 * prod.setMinDeltaR(0)
 *
 * for p in photons
 *   prod.produceHGCalIso(p.superCluster()->seed())
 *   a=prod.getIso(0)
 *   b=prod.getIso(1)
 *   c=prod.getIso(2)
 *   d=prod.getIso(3)
 *   e=prod.getIso(4)
 *
 *
 */
class HGCalIsoCalculator{
public:
    HGCalIsoCalculator();

    ~HGCalIsoCalculator();

    void setDeltaR(const float dr){dr2_=dr*dr;}

    void setMinDeltaR(const float dr){mindr2_=dr*dr;}

    void setRecHitTools(const hgcal::RecHitTools * recHitTools){rechittools_ = recHitTools;}

    void setNRings(const size_t nrings);

    void setNLayers(unsigned int nLayers){nlayers_=nLayers;}

    void setRecHits(
        edm::Handle<HGCRecHitCollection> hitsEE,
        edm::Handle<HGCRecHitCollection> hitsFH,
        edm::Handle<HGCRecHitCollection> hitsBH
      );

    void produceHGCalIso(const reco::CaloClusterPtr & seedCluster);

    const float getIso(const unsigned int ring) const;

private:
    std::vector<float> isoringdeposits_;
    std::vector<unsigned int> ringasso_;

    float dr2_,mindr2_;

    const hgcal::RecHitTools* rechittools_;
    edm::Handle<HGCRecHitCollection> recHitsEE_;
    edm::Handle<HGCRecHitCollection> recHitsFH_;
    edm::Handle<HGCRecHitCollection> recHitsBH_;
    std::vector<std::pair<float,float>> hitEtaPhiCache_;
    bool debug_;
    unsigned int nlayers_;
};


#endif /* RecoEgamma_EgammaTools_HGCalIsoCalculator_h */