Inheritance diagram for ticl::TracksterP4FromEnergySum:

Public Member Functions
void	setP4 (const std::vector< const Trackster * > &tracksters, std::vector< TICLCandidate > &ticl_cands, edm::Event &event) const override

	TracksterP4FromEnergySum (const edm::ParameterSet &, edm::ConsumesCollector &&iC)

Public Member Functions inherited from ticl::TracksterMomentumPluginBase
	TracksterMomentumPluginBase (const edm::ParameterSet &, edm::ConsumesCollector &&iC)

virtual	~TracksterMomentumPluginBase ()

Private Member Functions
std::tuple< TracksterMomentumPluginBase::LorentzVector, float >	calcP4 (const ticl::Trackster &trackster, const reco::Vertex &vertex, const std::vector< reco::CaloCluster > &calo_clusters) const

Private Attributes
bool	energy_from_regression_

edm::EDGetTokenT< std::vector< reco::CaloCluster > >	layer_clusters_token_

edm::EDGetTokenT< std::vector< reco::Vertex > >	vertex_token_

Additional Inherited Members
Public Types inherited from ticl::TracksterMomentumPluginBase
typedef reco::Candidate::LorentzVector	LorentzVector

Detailed Description

Definition at line 11 of file TracksterP4FromEnergySumPlugin.cc.

Constructor & Destructor Documentation

◆ TracksterP4FromEnergySum()

ticl::TracksterP4FromEnergySum::TracksterP4FromEnergySum	(	const edm::ParameterSet &	ps,
		edm::ConsumesCollector &&	iC
	)

explicit

Definition at line 28 of file TracksterP4FromEnergySumPlugin.cc.

       : TracksterMomentumPluginBase(ps, std::move(ic)),
         energy_from_regression_(ps.getParameter<bool>("energyFromRegression")),
         vertex_token_(ic.consumes<std::vector<reco::Vertex>>(ps.getParameter<edm::InputTag>("vertices"))),
         layer_clusters_token_(
             ic.consumes<std::vector<reco::CaloCluster>>(ps.getParameter<edm::InputTag>("layerClusters"))) {}

Member Function Documentation

◆ calcP4()

std::tuple< TracksterMomentumPluginBase::LorentzVector, float > ticl::TracksterP4FromEnergySum::calcP4	(	const ticl::Trackster &	trackster,
		const reco::Vertex &	vertex,
		const std::vector< reco::CaloCluster > &	calo_clusters
	)		const

private

Definition at line 64 of file TracksterP4FromEnergySumPlugin.cc.

                                                              {
     std::array<double, 3> barycentre{{0., 0., 0.}};
     double energy = 0.;
     size_t counter = 0;
  
     for (auto idx : trackster.vertices()) {
       auto n_vertices = trackster.vertex_multiplicity(counter++);
       auto fraction = n_vertices ? 1.f / n_vertices : 1.f;
       auto weight = calo_clusters[idx].energy() * fraction;
       energy += weight;
       barycentre[0] += calo_clusters[idx].x() * weight;
       barycentre[1] += calo_clusters[idx].y() * weight;
       barycentre[2] += calo_clusters[idx].z() * weight;
     }
     std::transform(
         std::begin(barycentre), std::end(barycentre), std::begin(barycentre), [&energy](double val) -> double {
           return energy >= 0. ? val / energy : val;
         });
  
     math::XYZVector direction(barycentre[0] - vertex.x(), barycentre[1] - vertex.y(), barycentre[2] - vertex.z());
     direction = direction.Unit();
     auto raw_energy = energy;
     energy = energy_from_regression_ ? trackster.regressed_energy() : raw_energy;
     direction *= energy;
  
     math::XYZTLorentzVector cartesian(direction.X(), direction.Y(), direction.Z(), energy);
     // Convert px, py, pz, E vector to CMS standard pt/eta/phi/m vector
     TracksterP4FromEnergySum::LorentzVector p4(cartesian);
     return std::tuple(p4, raw_energy);
   }

References begin, end, HCALHighEnergyHPDFilter_cfi::energy, energy_from_regression_, HLT_2018_cff::fraction, training_settings::idx, p4, ticl::Trackster::regressed_energy(), HcalDetIdTransform::transform(), heppy_batch::val, bphysicsOniaDQM_cfi::vertex, ticl::Trackster::vertex_multiplicity(), ticl::Trackster::vertices(), and mps_merge::weight.

Referenced by setP4().

◆ setP4()

void ticl::TracksterP4FromEnergySum::setP4	(	const std::vector< const Trackster * > &	tracksters,
		std::vector< TICLCandidate > &	ticl_cands,
		edm::Event &	event
	)		const

overridevirtual

Implements ticl::TracksterMomentumPluginBase.

Definition at line 35 of file TracksterP4FromEnergySumPlugin.cc.

                                                               {
     // Find best vertex
     edm::Handle<std::vector<reco::Vertex>> vertex_h;
     event.getByToken(vertex_token_, vertex_h);
     auto vertex_coll = *vertex_h;
     reco::Vertex best_vertex;
     if (!vertex_coll.empty()) {
       const auto& vertex = vertex_coll[0];
       if (vertex.isValid() && !(vertex.isFake())) {
         best_vertex = vertex;
       }
     }
  
     edm::Handle<std::vector<reco::CaloCluster>> layer_clusters_h;
     event.getByToken(layer_clusters_token_, layer_clusters_h);
  
     auto size = std::min(tracksters.size(), ticl_cands.size());
     for (size_t i = 0; i < size; ++i) {
       const auto* trackster = tracksters[i];
       auto ret = calcP4(*trackster, best_vertex, *layer_clusters_h);
  
       auto& ticl_cand = ticl_cands[i];
       ticl_cand.setP4(std::get<0>(ret));
       ticl_cand.setRawEnergy(std::get<1>(ret));
     }
   }