#include <HGCalShowerShapeHelper.h>

Classes
struct	ShowerWidths

Public Member Functions
const double	getCellSize (DetId detId)

const ShowerWidths	getPCAWidths (double cylinderR, bool useFractions=false)

const double	getRvar (double cylinderR, double energyNorm, bool useFractions=true, bool useCellSize=true)

	HGCalShowerShapeHelper (edm::ConsumesCollector &&sumes)

void	initPerEvent (const edm::EventSetup &iSetup, const std::vector< reco::PFRecHit > &recHits)

void	initPerObject (const std::vector< std::pair< DetId, float > > &hitsAndFracs, double minHitE=0, double minHitET=0, int minLayer=1, int maxLayer=-1, DetId::Detector subDet=DetId::HGCalEE)

void	setLayerWiseInfo ()

Static Public Attributes
static const double	kHDWaferCellSize_ = 0.465

static const double	kLDWaferCellSize_ = 0.698

Private Member Functions
void	setFilteredHitsAndFractions (const std::vector< std::pair< DetId, float > > &hitsAndFracs)

void	setPFRecHitPtrMap (const std::vector< reco::PFRecHit > &recHits)

Private Attributes
edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	caloGeometryToken_

ROOT::Math::XYZVector	centroid_

std::vector< double >	hitEnergies_

std::vector< double >	hitEnergiesWithFracs_

std::vector< std::pair< DetId, float > >	hitsAndFracs_

std::vector< ROOT::Math::XYZVector >	layerCentroids_

std::vector< double >	layerEnergies_

int	maxLayer_

double	minHitE_

double	minHitET2_

double	minHitET_

int	minLayer_

int	nLayer_

std::unordered_map< uint32_t, const reco::PFRecHit * >	pfRecHitPtrMap_

hgcal::RecHitTools	recHitTools_

DetId::Detector	subDet_

Detailed Description

Definition at line 46 of file HGCalShowerShapeHelper.h.

Constructor & Destructor Documentation

◆ HGCalShowerShapeHelper()

HGCalShowerShapeHelper::HGCalShowerShapeHelper ( edm::ConsumesCollector && sumes )

Definition at line 6 of file HGCalShowerShapeHelper.cc.

7 : caloGeometryToken_{sumes.esConsumes<CaloGeometry, CaloGeometryRecord>()} {}

Member Function Documentation

◆ getCellSize()

const double HGCalShowerShapeHelper::getCellSize ( DetId detId )

Definition at line 133 of file HGCalShowerShapeHelper.cc.

                                                             {
   return recHitTools_.getSiThickIndex(detId) == 0 ? kHDWaferCellSize_ : kLDWaferCellSize_;
 }

References hgcal::RecHitTools::getSiThickIndex(), kHDWaferCellSize_, kLDWaferCellSize_, and recHitTools_.

Referenced by getRvar().

◆ getPCAWidths()

const HGCalShowerShapeHelper::ShowerWidths HGCalShowerShapeHelper::getPCAWidths	(	double	cylinderR,
		bool	useFractions = `false`
	)

Definition at line 188 of file HGCalShowerShapeHelper.cc.

                                                                                                                  {
   if (hitsAndFracs_.empty()) {
     return ShowerWidths();
   }
  
   double cylinderR2 = cylinderR * cylinderR;
  
   TMatrixD covMat(3, 3);
  
   double dxdx = 0.0;
   double dydy = 0.0;
   double dzdz = 0.0;
  
   double dxdy = 0.0;
   double dydz = 0.0;
   double dzdx = 0.0;
  
   double totalW = 0.0;
  
   auto hitEnergyIter = useFractions ? hitEnergiesWithFracs_.begin() : hitEnergies_.begin();
  
   int nHit = 0;
   hitEnergyIter--;
  
   for (const auto &hnf : hitsAndFracs_) {
     hitEnergyIter++;
  
     DetId hitId = hnf.first;
  
     const auto &hitPos = recHitTools_.getPosition(hitId);
     ROOT::Math::XYZVector hitXYZ(hitPos.x(), hitPos.y(), hitPos.z());
  
     int hitLayer = recHitTools_.getLayer(hitId) - 1;
  
     ROOT::Math::XYZVector radXYZ = hitXYZ - layerCentroids_[hitLayer];
  
     double r2 = radXYZ.x() * radXYZ.x() + radXYZ.y() * radXYZ.y();
  
     if (r2 > cylinderR2) {
       continue;
     }
  
     ROOT::Math::XYZVector dXYZ = hitXYZ - centroid_;
  
     double weight = *hitEnergyIter;
     totalW += weight;
  
     dxdx += weight * dXYZ.x() * dXYZ.x();
     dydy += weight * dXYZ.y() * dXYZ.y();
     dzdz += weight * dXYZ.z() * dXYZ.z();
  
     dxdy += weight * dXYZ.x() * dXYZ.y();
     dydz += weight * dXYZ.y() * dXYZ.z();
     dzdx += weight * dXYZ.z() * dXYZ.x();
  
     nHit++;
   }
  
   if (!totalW || nHit < 2) {
     return ShowerWidths();
   }
  
   dxdx /= totalW;
   dydy /= totalW;
   dzdz /= totalW;
  
   dxdy /= totalW;
   dydz /= totalW;
   dzdx /= totalW;
  
   covMat(0, 0) = dxdx;
   covMat(1, 1) = dydy;
   covMat(2, 2) = dzdz;
  
   covMat(0, 1) = covMat(1, 0) = dxdy;
   covMat(0, 2) = covMat(2, 0) = dzdx;
   covMat(1, 2) = covMat(2, 1) = dydz;
  
   if (!covMat.Sum()) {
     return ShowerWidths();
   }
  
   // Get eigen values and vectors
   TVectorD eigVals(3);
   TMatrixD eigVecMat(3, 3);
  
   eigVecMat = covMat.EigenVectors(eigVals);
  
   ShowerWidths returnWidths;
  
   returnWidths.sigma2xx = dxdx;
   returnWidths.sigma2yy = dydy;
   returnWidths.sigma2zz = dzdz;
  
   returnWidths.sigma2xy = dxdy;
   returnWidths.sigma2yz = dydz;
   returnWidths.sigma2zx = dzdx;
  
   returnWidths.sigma2uu = eigVals(1);
   returnWidths.sigma2vv = eigVals(2);
   returnWidths.sigma2ww = eigVals(0);
  
   return returnWidths;
 }

Referenced by EgammaHLTHGCalIDVarProducer::produce().

◆ getRvar()

const double HGCalShowerShapeHelper::getRvar	(	double	cylinderR,
		double	energyNorm,
		bool	useFractions = `true`,
		bool	useCellSize = `true`
	)

Definition at line 137 of file HGCalShowerShapeHelper.cc.

                                                                                                                      {
   if (hitsAndFracs_.empty()) {
     return 0.0;
   }
  
   if (energyNorm <= 0.0) {
     edm::LogWarning("HGCalShowerShapeHelper")
         << "Encountered negative or zero energy for HGCal R-variable denominator: " << energyNorm << std::endl;
   }
  
   double cylinderR2 = cylinderR * cylinderR;
  
   double rVar = 0.0;
  
   auto hitEnergyIter = useFractions ? hitEnergiesWithFracs_.begin() : hitEnergies_.begin();
  
   hitEnergyIter--;
  
   for (const auto &hnf : hitsAndFracs_) {
     hitEnergyIter++;
  
     DetId hitId = hnf.first;
  
     int hitLayer = recHitTools_.getLayer(hitId) - 1;
  
     const auto &hitPos = recHitTools_.getPosition(hitId);
     ROOT::Math::XYZVector hitXYZ(hitPos.x(), hitPos.y(), hitPos.z());
  
     auto distXYZ = hitXYZ - layerCentroids_[hitLayer];
  
     double r2 = distXYZ.x() * distXYZ.x() + distXYZ.y() * distXYZ.y();
  
     // Including the cell size seems to make the variable less sensitive to the HD/LD transition region
     if (useCellSize) {
       if (std::sqrt(r2) > cylinderR + getCellSize(hitId)) {
         continue;
       }
     }
  
     else if (r2 > cylinderR2) {
       continue;
     }
  
     rVar += *hitEnergyIter;
   }
  
   rVar /= energyNorm;
  
   return rVar;
 }

References getCellSize(), hgcal::RecHitTools::getLayer(), hgcal::RecHitTools::getPosition(), hitEnergies_, hitEnergiesWithFracs_, hitsAndFracs_, layerCentroids_, diffTwoXMLs::r2, recHitTools_, and mathSSE::sqrt().

Referenced by EgammaHLTHGCalIDVarProducer::produce().

◆ initPerEvent()

void HGCalShowerShapeHelper::initPerEvent	(	const edm::EventSetup &	iSetup,
		const std::vector< reco::PFRecHit > &	recHits
	)

Definition at line 9 of file HGCalShowerShapeHelper.cc.

                                                                                                                {
   recHitTools_.setGeometry(iSetup.getData(caloGeometryToken_));
   setPFRecHitPtrMap(pfRecHits);
 }

References caloGeometryToken_, edm::EventSetup::getData(), recHitTools_, hgcal::RecHitTools::setGeometry(), and setPFRecHitPtrMap().

Referenced by EgammaHLTHGCalIDVarProducer::produce().

◆ initPerObject()

void HGCalShowerShapeHelper::initPerObject	(	const std::vector< std::pair< DetId, float > > &	hitsAndFracs,
		double	minHitE = `0`,
		double	minHitET = `0`,
		int	minLayer = `1`,
		int	maxLayer = `-1`,
		DetId::Detector	subDet = `DetId::HGCalEE`
	)

Definition at line 14 of file HGCalShowerShapeHelper.cc.

                                                                  {
   maxLayer = maxLayer <= 0 ? recHitTools_.lastLayerEE() : maxLayer;
  
   // Safety checks
   nLayer_ = maxLayer - minLayer + 1;
   assert(nLayer_ > 0);
  
   minHitE_ = minHitE;
   minHitET_ = minHitET;
   minHitET2_ = minHitET * minHitET;
   minLayer_ = minLayer;
   maxLayer_ = maxLayer;
   subDet_ = subDet;
  
   setFilteredHitsAndFractions(hitsAndFracs);
  
   setLayerWiseInfo();
 }

References cms::cuda::assert(), hgcal::RecHitTools::lastLayerEE(), maxLayer_, minHitE_, minHitET2_, minHitET_, HLT_2018_cff::minLayer, minLayer_, nLayer_, recHitTools_, setFilteredHitsAndFractions(), setLayerWiseInfo(), and subDet_.

Referenced by EgammaHLTHGCalIDVarProducer::produce().

◆ setFilteredHitsAndFractions()

void HGCalShowerShapeHelper::setFilteredHitsAndFractions ( const std::vector< std::pair< DetId, float > > & hitsAndFracs )

private

Definition at line 46 of file HGCalShowerShapeHelper.cc.

                                                                                                               {
   hitsAndFracs_.clear();
   hitEnergies_.clear();
   hitEnergiesWithFracs_.clear();
  
   for (const auto &hnf : hitsAndFracs) {
     DetId hitId = hnf.first;
     float hitEfrac = hnf.second;
  
     int hitLayer = recHitTools_.getLayer(hitId);
  
     if (hitLayer > nLayer_) {
       continue;
     }
  
     if (hitId.det() != subDet_) {
       continue;
     }
  
     if (pfRecHitPtrMap_.find(hitId.rawId()) == pfRecHitPtrMap_.end()) {
       continue;
     }
  
     const reco::PFRecHit &recHit = *pfRecHitPtrMap_[hitId.rawId()];
  
     if (recHit.energy() < minHitE_) {
       continue;
     }
  
     if (recHit.pt2() < minHitET2_) {
       continue;
     }
  
     // Fill the vectors
     hitsAndFracs_.push_back(hnf);
     hitEnergies_.push_back(recHit.energy());
     hitEnergiesWithFracs_.push_back(recHit.energy() * hitEfrac);
   }
 }

References DetId::det(), hgcal::RecHitTools::getLayer(), hitEnergies_, hitEnergiesWithFracs_, hitsAndFracs_, minHitE_, minHitET2_, nLayer_, pfRecHitPtrMap_, DetId::rawId(), rpcPointValidation_cfi::recHit, recHitTools_, and subDet_.

Referenced by initPerObject().

◆ setLayerWiseInfo()

void HGCalShowerShapeHelper::setLayerWiseInfo ( )

Definition at line 86 of file HGCalShowerShapeHelper.cc.

                                               {
   layerEnergies_.clear();
   layerEnergies_.resize(nLayer_);
  
   layerCentroids_.clear();
   layerCentroids_.resize(nLayer_);
  
   centroid_.SetXYZ(0, 0, 0);
  
   int iHit = -1;
   double totalW = 0.0;
  
   // Compute the centroid per layer
   for (const auto &hnf : hitsAndFracs_) {
     iHit++;
  
     DetId hitId = hnf.first;
  
     double weight = hitEnergies_[iHit];
     totalW += weight;
  
     const auto &hitPos = recHitTools_.getPosition(hitId);
     ROOT::Math::XYZVector hitXYZ(hitPos.x(), hitPos.y(), hitPos.z());
  
     centroid_ += weight * hitXYZ;
  
     int hitLayer = recHitTools_.getLayer(hitId) - 1;
  
     layerEnergies_[hitLayer] += weight;
     layerCentroids_[hitLayer] += weight * hitXYZ;
   }
  
   int iLayer = -1;
  
   for (auto &centroid : layerCentroids_) {
     iLayer++;
  
     if (layerEnergies_[iLayer]) {
       centroid /= layerEnergies_[iLayer];
     }
   }
  
   if (totalW) {
     centroid_ /= totalW;
   }
 }

References centroid_, hgcal::RecHitTools::getLayer(), hgcal::RecHitTools::getPosition(), hitEnergies_, hitsAndFracs_, layerCentroids_, layerEnergies_, nLayer_, recHitTools_, and mps_merge::weight.

Referenced by initPerObject().

◆ setPFRecHitPtrMap()

void HGCalShowerShapeHelper::setPFRecHitPtrMap ( const std::vector< reco::PFRecHit > & recHits )

private

Definition at line 38 of file HGCalShowerShapeHelper.cc.

                                                                                      {
   pfRecHitPtrMap_.clear();
  
   for (const auto &recHit : recHits) {
     pfRecHitPtrMap_[recHit.detId()] = &recHit;
   }
 }

References pfRecHitPtrMap_, rpcPointValidation_cfi::recHit, and FastTrackerRecHitMaskProducer_cfi::recHits.

Referenced by initPerEvent().

Member Data Documentation

◆ caloGeometryToken_

edm::ESGetToken<CaloGeometry, CaloGeometryRecord> HGCalShowerShapeHelper::caloGeometryToken_

private

Definition at line 118 of file HGCalShowerShapeHelper.h.

Referenced by initPerEvent().

◆ centroid_

ROOT::Math::XYZVector HGCalShowerShapeHelper::centroid_

private

Definition at line 126 of file HGCalShowerShapeHelper.h.

Referenced by getPCAWidths(), and setLayerWiseInfo().

◆ hitEnergies_

std::vector<double> HGCalShowerShapeHelper::hitEnergies_

private

Definition at line 123 of file HGCalShowerShapeHelper.h.

Referenced by getPCAWidths(), getRvar(), setFilteredHitsAndFractions(), and setLayerWiseInfo().

◆ hitEnergiesWithFracs_

std::vector<double> HGCalShowerShapeHelper::hitEnergiesWithFracs_

private

Definition at line 124 of file HGCalShowerShapeHelper.h.

Referenced by getPCAWidths(), getRvar(), and setFilteredHitsAndFractions().

◆ hitsAndFracs_

std::vector<std::pair<DetId, float> > HGCalShowerShapeHelper::hitsAndFracs_

private

Definition at line 122 of file HGCalShowerShapeHelper.h.

Referenced by getPCAWidths(), getRvar(), setFilteredHitsAndFractions(), and setLayerWiseInfo().

◆ kHDWaferCellSize_

const double HGCalShowerShapeHelper::kHDWaferCellSize_ = 0.465

static

Definition at line 58 of file HGCalShowerShapeHelper.h.

Referenced by getCellSize().

◆ kLDWaferCellSize_

const double HGCalShowerShapeHelper::kLDWaferCellSize_ = 0.698

static

Definition at line 57 of file HGCalShowerShapeHelper.h.

Referenced by getCellSize().

◆ layerCentroids_

std::vector<ROOT::Math::XYZVector> HGCalShowerShapeHelper::layerCentroids_

private

Definition at line 128 of file HGCalShowerShapeHelper.h.

Referenced by getPCAWidths(), getRvar(), and setLayerWiseInfo().

◆ layerEnergies_

std::vector<double> HGCalShowerShapeHelper::layerEnergies_

private

Definition at line 127 of file HGCalShowerShapeHelper.h.

Referenced by setLayerWiseInfo().

◆ maxLayer_

int HGCalShowerShapeHelper::maxLayer_

private

Definition at line 114 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject().

◆ minHitE_

double HGCalShowerShapeHelper::minHitE_

private

Definition at line 110 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject(), and setFilteredHitsAndFractions().

◆ minHitET2_

double HGCalShowerShapeHelper::minHitET2_

private

Definition at line 112 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject(), and setFilteredHitsAndFractions().

◆ minHitET_

double HGCalShowerShapeHelper::minHitET_

private

Definition at line 111 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject().

◆ minLayer_

int HGCalShowerShapeHelper::minLayer_

private

Definition at line 113 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject().

◆ nLayer_

int HGCalShowerShapeHelper::nLayer_

private

Definition at line 115 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject(), setFilteredHitsAndFractions(), and setLayerWiseInfo().

◆ pfRecHitPtrMap_

std::unordered_map<uint32_t, const reco::PFRecHit *> HGCalShowerShapeHelper::pfRecHitPtrMap_

private

Definition at line 121 of file HGCalShowerShapeHelper.h.

Referenced by setFilteredHitsAndFractions(), and setPFRecHitPtrMap().

◆ recHitTools_

hgcal::RecHitTools HGCalShowerShapeHelper::recHitTools_

private

Definition at line 119 of file HGCalShowerShapeHelper.h.

Referenced by getCellSize(), getPCAWidths(), getRvar(), initPerEvent(), initPerObject(), setFilteredHitsAndFractions(), and setLayerWiseInfo().

◆ subDet_

DetId::Detector HGCalShowerShapeHelper::subDet_

private

Definition at line 116 of file HGCalShowerShapeHelper.h.

Referenced by initPerObject(), and setFilteredHitsAndFractions().

Classes

Public Member Functions

Static Public Attributes

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ HGCalShowerShapeHelper()

Member Function Documentation

◆ getCellSize()

◆ getPCAWidths()

◆ getRvar()

◆ initPerEvent()

◆ initPerObject()

◆ setFilteredHitsAndFractions()

◆ setLayerWiseInfo()

◆ setPFRecHitPtrMap()

Member Data Documentation

◆ caloGeometryToken_

◆ centroid_

◆ hitEnergies_

◆ hitEnergiesWithFracs_

◆ hitsAndFracs_

◆ kHDWaferCellSize_

◆ kLDWaferCellSize_

◆ layerCentroids_

◆ layerEnergies_

◆ maxLayer_

◆ minHitE_

◆ minHitET2_

◆ minHitET_

◆ minLayer_

◆ nLayer_

◆ pfRecHitPtrMap_

◆ recHitTools_

◆ subDet_