#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 HGCRecHitCollection *	eerh_

const edm::EDGetTokenT< HGCRecHitCollection >	eetok

const HGCRecHitCollection *	fhrh_

const edm::EDGetTokenT< HGCRecHitCollection >	fhtok

RecHitTools	rhtools_

Static Private Attributes
static const int	lastLayerEE = 28

Detailed Description

Definition at line 27 of file ClusterTools.h.

Constructor & Destructor Documentation

ClusterTools::ClusterTools ( )

Definition at line 20 of file ClusterTools.cc.

20 {

21 }

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

Definition at line 23 of file ClusterTools.cc.

                                :
   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")) ) {
   }

hgcal::ClusterTools::~ClusterTools ( )

inline

Definition at line 31 of file ClusterTools.h.

31 {}

Member Function Documentation

float ClusterTools::getClusterHadronFraction ( const reco::CaloCluster & clus ) const

Definition at line 45 of file ClusterTools.cc.

References bhrh_, eerh_, Exception, f, fhrh_, edm::SortedCollection< T, SORT >::find(), DetId::Forward, dedxEstimators_cff::fraction, DetId::Hcal, HcalEndcap, DetId::HGCalEE, DetId::HGCalHSc, DetId::HGCalHSi, HGCEE, HGCHEF, hfClusterShapes_cfi::hits, 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 30 of file ClusterTools.cc.

References bhrh_, bhtok, eerh_, eetok, fhrh_, fhtok, edm::Event::getByToken(), hgcal::RecHitTools::getEvent(), edm::Handle< T >::product(), rhtools_, and groupFilesInBlocks::temp.

                                               {
   rhtools_.getEvent(ev);
   edm::Handle<HGCRecHitCollection> temp;
   ev.getByToken(eetok, temp);
   eerh_ = temp.product();
   ev.getByToken(fhtok, temp);
   fhrh_ = temp.product();
   ev.getByToken(bhtok, temp);
   bhrh_ = temp.product();
 }

void ClusterTools::getEventSetup ( const edm::EventSetup & es )

Definition at line 41 of file ClusterTools.cc.

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

                                                         {
   rhtools_.getEventSetup(es);
 }

int ClusterTools::getLayer ( const DetId detid ) const

Definition at line 121 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 125 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 94 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);
 }

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

Definition at line 133 of file ClusterTools.cc.

References eerh_, CaloRecHit::energy(), reco::CaloCluster::energy(), PV3DBase< T, PVType, FrameType >::eta(), edm::SortedCollection< T, SORT >::find(), plotBeamSpotDB::first, DetId::Forward, getLayer(), hgcal::RecHitTools::getPosition(), DetId::HGCalEE, HGCEE, reco::CaloCluster::hitsAndFractions(), lastLayerEE, cmsBatch::log, SiStripPI::max, PV3DBase< T, PVType, FrameType >::phi(), reco::CaloCluster::position(), position, rhtools_, edm::second(), mathSSE::sqrt(), and mps_merge::weight.

                                                                                                                                                          {
 
   if (getLayer(clus.hitsAndFractions()[0].first) > 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 48 of file ClusterTools.h.

References begin, reco::HGCalMultiCluster::clusters(), end, training_settings::idx, 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;
     }