HGCalHistoClusteringImpl Class Reference

#include <HGCalHistoClusteringImpl.h>

Public Member Functions
void	clusterizeHisto (const std::vector< edm::Ptr< l1t::HGCalCluster >> &clustersPtr, const std::vector< std::pair< GlobalPoint, double >> &seedPositionsEnergy, const HGCalTriggerGeometryBase &triggerGeometry, l1t::HGCalMulticlusterBxCollection &multiclusters) const
float	dR (const l1t::HGCalCluster &clu, const GlobalPoint &seed) const
void	eventSetup (const edm::EventSetup &es)
	HGCalHistoClusteringImpl (const edm::ParameterSet &conf)

Private Types
enum	ClusterAssociationStrategy { NearestNeighbour, EnergySplit }

Private Member Functions
std::vector< l1t::HGCalMulticluster >	clusterSeedMulticluster (const std::vector< edm::Ptr< l1t::HGCalCluster >> &clustersPtrs, const std::vector< std::pair< GlobalPoint, double >> &seeds) const
void	finalizeClusters (std::vector< l1t::HGCalMulticluster > &, l1t::HGCalMulticlusterBxCollection &, const HGCalTriggerGeometryBase &) const

Private Attributes
std::string	cluster_association_input_
ClusterAssociationStrategy	cluster_association_strategy_
double	dr_
std::vector< double >	dr_byLayer_coefficientA_
std::vector< double >	dr_byLayer_coefficientB_
std::unique_ptr< HGCalTriggerClusterIdentificationBase >	id_
double	ptC3dThreshold_
HGCalShowerShape	shape_
HGCalTriggerTools	triggerTools_

Static Private Attributes
static double	kMidRadius_ = 2.3

Detailed Description

Definition at line 14 of file HGCalHistoClusteringImpl.h.

Member Enumeration Documentation

enum HGCalHistoClusteringImpl::ClusterAssociationStrategy

private

Enumerator
NearestNeighbour
EnergySplit

Definition at line 38 of file HGCalHistoClusteringImpl.h.

{ NearestNeighbour, EnergySplit };

Constructor & Destructor Documentation

HGCalHistoClusteringImpl::HGCalHistoClusteringImpl

(

const edm::ParameterSet &

conf

)

Definition at line 6 of file HGCalHistoClusteringImpl.cc.

References cluster_association_input_, cluster_association_strategy_, dr_, EnergySplit, Exception, reco::get(), edm::ParameterSet::getParameter(), id_, NearestNeighbour, and ptC3dThreshold_.

    : dr_(conf.getParameter<double>("dR_multicluster")),
      dr_byLayer_coefficientA_(conf.existsAs<std::vector<double>>("dR_multicluster_byLayer_coefficientA")
                                   ? conf.getParameter<std::vector<double>>("dR_multicluster_byLayer_coefficientA")
                                   : std::vector<double>()),
      dr_byLayer_coefficientB_(conf.existsAs<std::vector<double>>("dR_multicluster_byLayer_coefficientB")
                                   ? conf.getParameter<std::vector<double>>("dR_multicluster_byLayer_coefficientB")
                                   : std::vector<double>()),
      ptC3dThreshold_(conf.getParameter<double>("minPt_multicluster")),
      cluster_association_input_(conf.getParameter<string>("cluster_association")) {
  if (cluster_association_input_ == "NearestNeighbour") {
    cluster_association_strategy_ = NearestNeighbour;
  } else if (cluster_association_input_ == "EnergySplit") {
    cluster_association_strategy_ = EnergySplit;
  } else {
    throw cms::Exception("HGCTriggerParameterError")
        << "Unknown cluster association strategy'" << cluster_association_strategy_;
  }

  edm::LogInfo("HGCalMulticlusterParameters")
      << "Multicluster dR: " << dr_ << "\nMulticluster minimum transverse-momentum: " << ptC3dThreshold_;

  id_ = std::unique_ptr<HGCalTriggerClusterIdentificationBase>{
      HGCalTriggerClusterIdentificationFactory::get()->create("HGCalTriggerClusterIdentificationBDT")};
  id_->initialize(conf.getParameter<edm::ParameterSet>("EGIdentification"));
}

Member Function Documentation

void HGCalHistoClusteringImpl::clusterizeHisto	(	const std::vector< edm::Ptr< l1t::HGCalCluster >> &	clustersPtr,
		const std::vector< std::pair< GlobalPoint, double >> &	seedPositionsEnergy,
		const HGCalTriggerGeometryBase &	triggerGeometry,
		l1t::HGCalMulticlusterBxCollection &	multiclusters
	)		const

Definition at line 107 of file HGCalHistoClusteringImpl.cc.

References clusterSeedMulticluster(), and finalizeClusters().

Referenced by eventSetup().

                                                                                                      {
  /* clusterize clusters around seeds */
  std::vector<l1t::HGCalMulticluster> multiclustersTmp = clusterSeedMulticluster(clustersPtrs, seedPositionsEnergy);
  /* making the collection of multiclusters */
  finalizeClusters(multiclustersTmp, multiclusters, triggerGeometry);
}

std::vector< l1t::HGCalMulticluster > HGCalHistoClusteringImpl::clusterSeedMulticluster ( const std::vector< edm::Ptr< l1t::HGCalCluster >> &  clustersPtrs, const std::vector< std::pair< GlobalPoint, double >> &  seeds  ) const

private

Definition at line 39 of file HGCalHistoClusteringImpl.cc.

References funct::abs(), cluster_association_strategy_, edmIntegrityCheck::d, dR(), dr_, dr_byLayer_coefficientA_, dr_byLayer_coefficientB_, EnergySplit, plotBeamSpotDB::first, iseed, kMidRadius_, HGCalTriggerTools::layerWithOffset(), NearestNeighbour, triggerTools_, PV3DBase< T, PVType, FrameType >::z(), and HGCalTriggerTools::zside().

Referenced by clusterizeHisto().

                                                                {
  std::map<int, l1t::HGCalMulticluster> mapSeedMulticluster;
  std::vector<l1t::HGCalMulticluster> multiclustersTmp;

  for (const auto& clu : clustersPtrs) {
    int z_side = triggerTools_.zside(clu->detId());

    double radiusCoefficientA =
        dr_byLayer_coefficientA_.empty() ? dr_ : dr_byLayer_coefficientA_[triggerTools_.layerWithOffset(clu->detId())];
    double radiusCoefficientB =
        dr_byLayer_coefficientB_.empty() ? 0 : dr_byLayer_coefficientB_[triggerTools_.layerWithOffset(clu->detId())];

    double minDist = radiusCoefficientA + radiusCoefficientB * (kMidRadius_ - std::abs(clu->eta()));

    std::vector<pair<int, double>> targetSeedsEnergy;

    for (unsigned int iseed = 0; iseed < seeds.size(); iseed++) {
      GlobalPoint seedPosition = seeds[iseed].first;
      double seedEnergy = seeds[iseed].second;

      if (z_side * seedPosition.z() < 0)
        continue;
      double d = this->dR(*clu, seeds[iseed].first);

      if (d < minDist) {
        if (cluster_association_strategy_ == EnergySplit) {
          targetSeedsEnergy.emplace_back(iseed, seedEnergy);
        } else if (cluster_association_strategy_ == NearestNeighbour) {
          minDist = d;

          if (targetSeedsEnergy.empty()) {
            targetSeedsEnergy.emplace_back(iseed, seedEnergy);
          } else {
            targetSeedsEnergy.at(0).first = iseed;
            targetSeedsEnergy.at(0).second = seedEnergy;
          }
        }
      }
    }

    if (targetSeedsEnergy.empty())
      continue;
    //Loop over target seeds and divide up the clusters energy
    double totalTargetSeedEnergy = 0;
    for (const auto& energy : targetSeedsEnergy) {
      totalTargetSeedEnergy += energy.second;
    }

    for (const auto& energy : targetSeedsEnergy) {
      double seedWeight = 1;
      if (cluster_association_strategy_ == EnergySplit)
        seedWeight = energy.second / totalTargetSeedEnergy;
      if (mapSeedMulticluster[energy.first].size() == 0) {
        mapSeedMulticluster[energy.first] = l1t::HGCalMulticluster(clu, seedWeight);
      } else {
        mapSeedMulticluster[energy.first].addConstituent(clu, true, seedWeight);
      }
    }
  }

  for (const auto& mclu : mapSeedMulticluster)
    multiclustersTmp.emplace_back(mclu.second);

  return multiclustersTmp;
}

float HGCalHistoClusteringImpl::dR	(	const l1t::HGCalCluster &	clu,
		const GlobalPoint &	seed
	)		const

Definition at line 33 of file HGCalHistoClusteringImpl.cc.

References l1t::HGCalClusterT< C >::centreProj(), mag(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by clusterSeedMulticluster(), and eventSetup().

                                                                                            {
  Basic3DVector<float> seed_3dv(seed);
  GlobalPoint seed_proj(seed_3dv / seed.z());
  return (seed_proj - clu.centreProj()).mag();
}

void HGCalHistoClusteringImpl::eventSetup ( const edm::EventSetup &  es )

inline

Definition at line 18 of file HGCalHistoClusteringImpl.h.

References clusterizeHisto(), dR(), dr_byLayer_coefficientA_, dr_byLayer_coefficientB_, HGCalShowerShape::eventSetup(), HGCalTriggerTools::eventSetup(), Exception, HGCalTriggerTools::lastLayerBH(), SurveyInfoScenario_cff::seed, shape_, and triggerTools_.

                                           {
    triggerTools_.eventSetup(es);
    shape_.eventSetup(es);
    if ((!dr_byLayer_coefficientA_.empty() && (dr_byLayer_coefficientA_.size() - 1) < triggerTools_.lastLayerBH()) ||
        (!dr_byLayer_coefficientB_.empty() && (dr_byLayer_coefficientB_.size() - 1) < triggerTools_.lastLayerBH())) {
      throw cms::Exception("Configuration")
          << "The per-layer dR values go up to " << (dr_byLayer_coefficientA_.size() - 1) << "(A) and "
          << (dr_byLayer_coefficientB_.size() - 1) << "(B), while layers go up to " << triggerTools_.lastLayerBH()
          << "\n";
    }
  }

void HGCalHistoClusteringImpl::finalizeClusters ( std::vector< l1t::HGCalMulticluster > &  multiclusters_in, l1t::HGCalMulticlusterBxCollection &  multiclusters_out, const HGCalTriggerGeometryBase &  triggerGeometry  ) const

private

Definition at line 117 of file HGCalHistoClusteringImpl.cc.

References HGCalShowerShape::fillShapes(), id_, ptC3dThreshold_, BXVector< T >::push_back(), shape_, and TtFullHadEvtBuilder_cfi::sumPt.

Referenced by clusterizeHisto().

                                                                                                       {
  for (auto& multicluster : multiclusters_in) {
    // compute the eta, phi from its barycenter
    // + pT as scalar sum of pT of constituents
    double sumPt = multicluster.sumPt();

    math::PtEtaPhiMLorentzVector multiclusterP4(sumPt, multicluster.centre().eta(), multicluster.centre().phi(), 0.);
    multicluster.setP4(multiclusterP4);

    if (multicluster.pt() > ptC3dThreshold_) {
      //compute shower shapes
      shape_.fillShapes(multicluster, triggerGeometry);
      // fill quality flag
      multicluster.setHwQual(id_->decision(multicluster));
      // fill H/E
      multicluster.saveHOverE();

      multiclusters_out.push_back(0, multicluster);
    }
  }
}

Member Data Documentation

std::string HGCalHistoClusteringImpl::cluster_association_input_

private

Definition at line 53 of file HGCalHistoClusteringImpl.h.

Referenced by HGCalHistoClusteringImpl().

ClusterAssociationStrategy HGCalHistoClusteringImpl::cluster_association_strategy_

private

Definition at line 54 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and HGCalHistoClusteringImpl().

double HGCalHistoClusteringImpl::dr_

private

Definition at line 48 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and HGCalHistoClusteringImpl().

std::vector<double> HGCalHistoClusteringImpl::dr_byLayer_coefficientA_

private

Definition at line 49 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and eventSetup().

std::vector<double> HGCalHistoClusteringImpl::dr_byLayer_coefficientB_

private

Definition at line 50 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and eventSetup().

std::unique_ptr<HGCalTriggerClusterIdentificationBase> HGCalHistoClusteringImpl::id_

private

Definition at line 58 of file HGCalHistoClusteringImpl.h.

Referenced by finalizeClusters(), and HGCalHistoClusteringImpl().

double HGCalHistoClusteringImpl::kMidRadius_ = 2.3

staticprivate

Definition at line 60 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster().

double HGCalHistoClusteringImpl::ptC3dThreshold_

private

Definition at line 51 of file HGCalHistoClusteringImpl.h.

Referenced by finalizeClusters(), and HGCalHistoClusteringImpl().

HGCalShowerShape HGCalHistoClusteringImpl::shape_

private

Definition at line 56 of file HGCalHistoClusteringImpl.h.

Referenced by eventSetup(), and finalizeClusters().

HGCalTriggerTools HGCalHistoClusteringImpl::triggerTools_

private

Definition at line 57 of file HGCalHistoClusteringImpl.h.

Referenced by clusterSeedMulticluster(), and eventSetup().