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
RecHitTools	rhtools_

Detailed Description

Definition at line 28 of file ClusterTools.h.

Constructor & Destructor Documentation

ClusterTools::ClusterTools

(

)

Definition at line 19 of file ClusterTools.cc.

{}

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"))),
      caloGeometryToken_{sumes.esConsumes()} {}

hgcal::ClusterTools::~ClusterTools ( )

inline

Definition at line 32 of file ClusterTools.h.

{}

Member Function Documentation

float ClusterTools::getClusterHadronFraction

(

const reco::CaloCluster &

clus

)

const

Definition at line 35 of file ClusterTools.cc.

References bhrh_, eerh_, relval_parameters_module::energy, Exception, validate-o2o-wbm::f, fhrh_, edm::SortedCollection< T, SORT >::find(), DetId::Forward, HLT_FULL_cff::fraction, DetId::Hcal, HcalEndcap, DetId::HGCalEE, DetId::HGCalHSc, DetId::HGCalHSi, HGCEE, HGCHEF, reco::CaloCluster::hitsAndFractions(), and groupFilesInBlocks::temp.

                                                                              {
  float energy = 0.f, energyHad = 0.f;
  const auto& hits = clus.hitsAndFractions();
  for (const auto& hit : hits) {
    const auto& id = hit.first;
    const float fraction = hit.second;
    if (id.det() == DetId::HGCalEE) {
      energy += eerh_->find(id)->energy() * fraction;
    } else if (id.det() == DetId::HGCalHSi) {
      const float temp = fhrh_->find(id)->energy();
      energy += temp * fraction;
      energyHad += temp * fraction;
    } else if (id.det() == DetId::HGCalHSc) {
      const float temp = bhrh_->find(id)->energy();
      energy += temp * fraction;
      energyHad += temp * fraction;
    } else if (id.det() == DetId::Forward) {
      switch (id.subdetId()) {
        case HGCEE:
          energy += eerh_->find(id)->energy() * fraction;
          break;
        case HGCHEF: {
          const float temp = fhrh_->find(id)->energy();
          energy += temp * fraction;
          energyHad += temp * fraction;
        } break;
        default:
          throw cms::Exception("HGCalClusterTools") << " Cluster contains hits that are not from HGCal! " << std::endl;
      }
    } else if (id.det() == DetId::Hcal && id.subdetId() == HcalEndcap) {
      const float temp = bhrh_->find(id)->energy();
      energy += temp * fraction;
      energyHad += temp * fraction;
    } else {
      throw cms::Exception("HGCalClusterTools") << " Cluster contains hits that are not from HGCal! " << std::endl;
    }
  }
  float fraction = -1.f;
  if (energy > 0.f) {
    fraction = energyHad / energy;
  }
  return fraction;
}

void ClusterTools::getEvent

(

const edm::Event &

ev

)

Definition at line 27 of file ClusterTools.cc.

References bhrh_, bhtok, eerh_, eetok, fhrh_, fhtok, and edm::Event::get().

                                              {
  eerh_ = &ev.get(eetok);
  fhrh_ = &ev.get(fhtok);
  bhrh_ = &ev.get(bhtok);
}

void ClusterTools::getEventSetup

(

const edm::EventSetup &

es

)

Definition at line 33 of file ClusterTools.cc.

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

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

int ClusterTools::getLayer

(

const DetId

detid

)

const

Definition at line 108 of file ClusterTools.cc.

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

Referenced by getWidths().

{ return rhtools_.getLayerWithOffset(detid); }

double ClusterTools::getMultiClusterEnergy

(

const reco::HGCalMultiCluster &

clu

)

const

Definition at line 110 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;
}

math::XYZPoint ClusterTools::getMultiClusterPosition

(

const reco::HGCalMultiCluster &

clu

)

const

Definition at line 79 of file ClusterTools.cc.

References reco::HGCalMultiCluster::clusters(), getMultiClusterEnergy(), and histoStyle::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);
}

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

Definition at line 118 of file ClusterTools.cc.

References eerh_, CaloRecHit::energy(), reco::CaloCluster::energy(), PV3DBase< T, PVType, FrameType >::eta(), edm::SortedCollection< T, SORT >::find(), first, DetId::Forward, getLayer(), hgcal::RecHitTools::getPosition(), DetId::HGCalEE, HGCEE, reco::CaloCluster::hitsAndFractions(), hgcal::RecHitTools::lastLayerEE(), log, SiStripPI::max, PV3DBase< T, PVType, FrameType >::phi(), reco::CaloCluster::position(), position, rhtools_, edm::second(), mathSSE::sqrt(), and histoStyle::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.;

  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)) {
      const HGCRecHit* theHit = &(*eerh_->find(id));

      GlobalPoint cellPos = rhtools_.getPosition(id);
      double weight = theHit->energy();
      // take w0=2 To be optimized
      double logweight = 0;
      if (clus.energy() != 0) {
        logweight = std::max(0., 2 + log(theHit->energy() / 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;
}

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

inlineprivate

Definition at line 53 of file ClusterTools.h.

References SplitLinear::begin, reco::HGCalMultiCluster::clusters(), dataset::end, reco::HGCalMultiCluster::size(), 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

const HGCRecHitCollection * hgcal::ClusterTools::bhrh_

private

Definition at line 65 of file ClusterTools.h.

Referenced by getClusterHadronFraction(), and getEvent().

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

private

Definition at line 62 of file ClusterTools.h.

Referenced by getEvent().

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

private

Definition at line 63 of file ClusterTools.h.

Referenced by getEventSetup().

const HGCRecHitCollection* hgcal::ClusterTools::eerh_

private

Definition at line 65 of file ClusterTools.h.

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

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

private

Definition at line 62 of file ClusterTools.h.

Referenced by getEvent().

const HGCRecHitCollection * hgcal::ClusterTools::fhrh_

private

Definition at line 65 of file ClusterTools.h.

Referenced by getClusterHadronFraction(), and getEvent().

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

private

Definition at line 62 of file ClusterTools.h.

Referenced by getEvent().

RecHitTools hgcal::ClusterTools::rhtools_

private

Definition at line 61 of file ClusterTools.h.

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