hgcal::ClusterTools Class Reference

#include <ClusterTools.h>

Public Member Functions
	ClusterTools ()
	ClusterTools (const edm::ParameterSet &, edm::ConsumesCollector &)
float	getClusterHadronFraction (const reco::CaloCluster &) const
void	getEvent (const edm::Event &)
void	getEventSetup (const edm::EventSetup &)
int	getLayer (const DetId) const
double	getMultiClusterEnergy (const reco::HGCalMultiCluster &) const
math::XYZPoint	getMultiClusterPosition (const reco::HGCalMultiCluster &) const
bool	getWidths (const reco::CaloCluster &clus, double &sigmaetaeta, double &sigmaphiphi, double &sigmaetaetalog, double &sigmaphiphilog) const
	~ClusterTools ()

Private Member Functions
std::vector< size_t >	sort_by_z (const reco::HGCalMultiCluster &v) const

Private Attributes
const HGCRecHitCollection *	bhrh_
const edm::EDGetTokenT< HGCRecHitCollection >	bhtok
const edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	caloGeometryToken_
const HGCRecHitCollection *	eerh_
const edm::EDGetTokenT< HGCRecHitCollection >	eetok
const HGCRecHitCollection *	fhrh_
const edm::EDGetTokenT< HGCRecHitCollection >	fhtok
const std::unordered_map< DetId, const unsigned int > *	hitMap_
const edm::EDGetTokenT< std::unordered_map< DetId, const unsigned int > >	hitMapToken_
std::unique_ptr< MultiVectorManager< HGCRecHit > >	rechitManager_
RecHitTools	rhtools_

Detailed Description

Definition at line 30 of file ClusterTools.h.

Constructor & Destructor Documentation

ClusterTools() [1/2]

ClusterTools::ClusterTools

(

)

Definition at line 19 of file ClusterTools.cc.

{}

ClusterTools() [2/2]

ClusterTools::ClusterTools	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector &	sumes
	)

Definition at line 21 of file ClusterTools.cc.

References edm::ConsumesCollector::esConsumes().

    : eetok(sumes.consumes<HGCRecHitCollection>(conf.getParameter<edm::InputTag>("HGCEEInput"))),
      fhtok(sumes.consumes<HGCRecHitCollection>(conf.getParameter<edm::InputTag>("HGCFHInput"))),
      bhtok(sumes.consumes<HGCRecHitCollection>(conf.getParameter<edm::InputTag>("HGCBHInput"))),
      hitMapToken_(sumes.consumes<std::unordered_map<DetId, const unsigned int>>(
          conf.getParameter<edm::InputTag>("hgcalHitMap"))),
      caloGeometryToken_{sumes.esConsumes()} {}

~ClusterTools()

hgcal::ClusterTools::~ClusterTools ( )

inline

Definition at line 34 of file ClusterTools.h.

{}

Member Function Documentation

getClusterHadronFraction()

float ClusterTools::getClusterHadronFraction

(

const reco::CaloCluster &

clus

)

const

Definition at line 42 of file ClusterTools.cc.

References HBHEDarkening_cff::energy, f, DetId::Forward, HLT_2024v14_cff::fraction, DetId::Hcal, HcalEndcap, DetId::HGCalHSc, DetId::HGCalHSi, HGCHEF, hitMap_, hfClusterShapes_cfi::hits, reco::CaloCluster::hitsAndFractions(), nano_mu_digi_cff::rawId, and rechitManager_.

                                                                              {
  float energy = 0.f, energyHad = 0.f;
  const auto& hits = clus.hitsAndFractions();
  const auto& rhmanager = *rechitManager_;
  for (const auto& hit : hits) {
    const auto& id = hit.first;
    const float fraction = hit.second;
    auto hitIter = hitMap_->find(id.rawId());
    if (hitIter == hitMap_->end()) {
      continue;
    }
    unsigned int rechitIndex = hitIter->second;
    float hitEnergy = rhmanager[rechitIndex].energy() * fraction;
    energy += hitEnergy;
    if (id.det() == DetId::HGCalHSi || id.det() == DetId::HGCalHSc ||
        (id.det() == DetId::Forward && id.subdetId() == HGCHEF) ||
        (id.det() == DetId::Hcal && id.subdetId() == HcalEndcap)) {
      energyHad += hitEnergy;
    }
  }
  float hadronicFraction = -1.f;
  if (energy > 0.f) {
    hadronicFraction = energyHad / energy;
  }
  return hadronicFraction;
}

getEvent()

void ClusterTools::getEvent

(

const edm::Event &

ev

)

Definition at line 29 of file ClusterTools.cc.

References bhrh_, bhtok, eerh_, eetok, makeMEIFBenchmarkPlots::ev, fhrh_, fhtok, hitMap_, hitMapToken_, and rechitManager_.

                                              {
  eerh_ = &ev.get(eetok);
  fhrh_ = &ev.get(fhtok);
  bhrh_ = &ev.get(bhtok);
  hitMap_ = &ev.get(hitMapToken_);
  rechitManager_ = std::make_unique<MultiVectorManager<HGCRecHit>>();
  rechitManager_->addVector(*eerh_);
  rechitManager_->addVector(*fhrh_);
  rechitManager_->addVector(*bhrh_);
}

getEventSetup()

void ClusterTools::getEventSetup

(

const edm::EventSetup &

es

)

Definition at line 40 of file ClusterTools.cc.

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

{ rhtools_.setGeometry(es.getData(caloGeometryToken_)); }

getLayer()

int ClusterTools::getLayer

(

const DetId

detid

)

const

Definition at line 98 of file ClusterTools.cc.

References ALCARECOPPSCalTrackBasedSel_cff::detid, hgcal::RecHitTools::getLayerWithOffset(), and rhtools_.

Referenced by getWidths().

{ return rhtools_.getLayerWithOffset(detid); }

getMultiClusterEnergy()

double ClusterTools::getMultiClusterEnergy

(

const reco::HGCalMultiCluster &

clu

)

const

Definition at line 100 of file ClusterTools.cc.

References reco::HGCalMultiCluster::clusters().

Referenced by getMultiClusterPosition().

                                                                                 {
  double acc = 0.0;
  for (const auto& ptr : clu.clusters()) {
    acc += ptr->energy();
  }
  return acc;
}

getMultiClusterPosition()

math::XYZPoint ClusterTools::getMultiClusterPosition

(

const reco::HGCalMultiCluster &

clu

)

const

Definition at line 69 of file ClusterTools.cc.

References reco::HGCalMultiCluster::clusters(), getMultiClusterEnergy(), and mps_merge::weight.

                                                                                         {
  if (clu.clusters().empty())
    return math::XYZPoint();

  double acc_x = 0.0;
  double acc_y = 0.0;
  double acc_z = 0.0;
  double totweight = 0.;
  double mcenergy = getMultiClusterEnergy(clu);
  for (const auto& ptr : clu.clusters()) {
    if (mcenergy != 0) {
      if (ptr->energy() < .01 * mcenergy)
        continue;  //cutoff < 1% layer contribution
    }
    const double weight = ptr->energy();  // weigth each corrdinate only by the total energy of the layer cluster
    acc_x += ptr->x() * weight;
    acc_y += ptr->y() * weight;
    acc_z += ptr->z() * weight;
    totweight += weight;
  }
  if (totweight != 0) {
    acc_x /= totweight;
    acc_y /= totweight;
    acc_z /= totweight;
  }
  // return x/y/z in absolute coordinates
  return math::XYZPoint(acc_x, acc_y, acc_z);
}

getWidths()

bool ClusterTools::getWidths	(	const reco::CaloCluster &	clus,
		double &	sigmaetaeta,
		double &	sigmaphiphi,
		double &	sigmaetaetalog,
		double &	sigmaphiphilog
	)		const

Definition at line 108 of file ClusterTools.cc.

References reco::CaloCluster::energy(), PV3DBase< T, PVType, FrameType >::eta(), dqmdumpme::first, DetId::Forward, getLayer(), hgcal::RecHitTools::getPosition(), DetId::HGCalEE, HGCEE, hitMap_, reco::CaloCluster::hitsAndFractions(), hgcal::RecHitTools::lastLayerEE(), CrabHelper::log, WZElectronSkims53X_cff::max, PV3DBase< T, PVType, FrameType >::phi(), reco::CaloCluster::position(), position, nano_mu_digi_cff::rawId, rechitManager_, rhtools_, edm::second(), mathSSE::sqrt(), and mps_merge::weight.

                                                         {
  if (getLayer(clus.hitsAndFractions()[0].first) > (int)rhtools_.lastLayerEE())
    return false;
  const math::XYZPoint& position(clus.position());
  unsigned nhit = clus.hitsAndFractions().size();

  sigmaetaeta = 0.;
  sigmaphiphi = 0.;
  sigmaetaetal = 0.;
  sigmaphiphil = 0.;

  double sumw = 0.;
  double sumlogw = 0.;
  const auto& rhmanager = *rechitManager_;

  for (unsigned int ih = 0; ih < nhit; ++ih) {
    const DetId& id = (clus.hitsAndFractions())[ih].first;
    if ((clus.hitsAndFractions())[ih].second == 0.)
      continue;

    if ((id.det() == DetId::HGCalEE) || (id.det() == DetId::Forward && id.subdetId() == HGCEE)) {
      auto hitIter = hitMap_->find(id.rawId());
      if (hitIter == hitMap_->end()) {
        continue;
      }
      unsigned int rechitIndex = hitIter->second;
      float hitEnergy = rhmanager[rechitIndex].energy();

      GlobalPoint cellPos = rhtools_.getPosition(id);
      double weight = hitEnergy;
      // take w0=2 To be optimized
      double logweight = 0;
      if (clus.energy() != 0) {
        logweight = std::max(0., 2 + std::log(hitEnergy / clus.energy()));
      }
      double deltaetaeta2 = (cellPos.eta() - position.eta()) * (cellPos.eta() - position.eta());
      double deltaphiphi2 = (cellPos.phi() - position.phi()) * (cellPos.phi() - position.phi());
      sigmaetaeta += deltaetaeta2 * weight;
      sigmaphiphi += deltaphiphi2 * weight;
      sigmaetaetal += deltaetaeta2 * logweight;
      sigmaphiphil += deltaphiphi2 * logweight;
      sumw += weight;
      sumlogw += logweight;
    }
  }

  if (sumw <= 0.)
    return false;

  sigmaetaeta /= sumw;
  sigmaetaeta = std::sqrt(sigmaetaeta);
  sigmaphiphi /= sumw;
  sigmaphiphi = std::sqrt(sigmaphiphi);

  if (sumlogw != 0) {
    sigmaetaetal /= sumlogw;
    sigmaetaetal = std::sqrt(sigmaetaetal);
    sigmaphiphil /= sumlogw;
    sigmaphiphil = std::sqrt(sigmaphiphil);
  }

  return true;
}

sort_by_z()

std::vector<size_t> hgcal::ClusterTools::sort_by_z ( const reco::HGCalMultiCluster &  v ) const

inlineprivate

Definition at line 55 of file ClusterTools.h.

References testProducerWithPsetDescEmpty_cfi::i1, testProducerWithPsetDescEmpty_cfi::i2, heavyIonCSV_trainingSettings::idx, jetUpdater_cfi::sort, and findQualityFiles::v.

                                                                      {
      std::vector<size_t> idx(v.size());
      std::iota(std::begin(idx), std::end(idx), 0);
      sort(
          idx.begin(), idx.end(), [&v](size_t i1, size_t i2) { return v.clusters()[i1]->z() < v.clusters()[i2]->z(); });
      return idx;
    }

Member Data Documentation

bhrh_

const HGCRecHitCollection * hgcal::ClusterTools::bhrh_

private

Definition at line 68 of file ClusterTools.h.

Referenced by getEvent().

bhtok

const edm::EDGetTokenT<HGCRecHitCollection> hgcal::ClusterTools::bhtok

private

Definition at line 64 of file ClusterTools.h.

Referenced by getEvent().

caloGeometryToken_

const edm::ESGetToken<CaloGeometry, CaloGeometryRecord> hgcal::ClusterTools::caloGeometryToken_

private

Definition at line 66 of file ClusterTools.h.

Referenced by getEventSetup().

eerh_

const HGCRecHitCollection* hgcal::ClusterTools::eerh_

private

Definition at line 68 of file ClusterTools.h.

Referenced by getEvent().

eetok

const edm::EDGetTokenT<HGCRecHitCollection> hgcal::ClusterTools::eetok

private

Definition at line 64 of file ClusterTools.h.

Referenced by getEvent().

fhrh_

const HGCRecHitCollection * hgcal::ClusterTools::fhrh_

private

Definition at line 68 of file ClusterTools.h.

Referenced by getEvent().

fhtok

const edm::EDGetTokenT<HGCRecHitCollection> hgcal::ClusterTools::fhtok

private

Definition at line 64 of file ClusterTools.h.

Referenced by getEvent().

hitMap_

const std::unordered_map<DetId, const unsigned int>* hgcal::ClusterTools::hitMap_

private

Definition at line 69 of file ClusterTools.h.

Referenced by getClusterHadronFraction(), getEvent(), and getWidths().

hitMapToken_

const edm::EDGetTokenT<std::unordered_map<DetId, const unsigned int> > hgcal::ClusterTools::hitMapToken_

private

Definition at line 65 of file ClusterTools.h.

Referenced by getEvent().

rechitManager_

std::unique_ptr<MultiVectorManager<HGCRecHit> > hgcal::ClusterTools::rechitManager_

private

Definition at line 70 of file ClusterTools.h.

Referenced by getClusterHadronFraction(), getEvent(), and getWidths().

rhtools_

RecHitTools hgcal::ClusterTools::rhtools_

private

Definition at line 63 of file ClusterTools.h.

Referenced by getEventSetup(), getLayer(), and getWidths().