ticl::TracksterP4FromEnergySum Class Reference

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.

References mps_fire::end, HCALHighEnergyHPDFilter_cfi::energy, energy_from_regression_, HLT_2022v12_cff::fraction, heavyIonCSV_trainingSettings::idx, 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().

                                                             {
    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);
  }

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.

References calcP4(), mps_fire::i, layer_clusters_token_, SiStripPI::min, runTheMatrix::ret, findQualityFiles::size, bphysicsOniaDQM_cfi::vertex, and vertex_token_.

Referenced by Jet.Jet::setCorrP4().

                                                              {
    // 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));
    }
  }

Member Data Documentation

energy_from_regression_

bool ticl::TracksterP4FromEnergySum::energy_from_regression_

private

Definition at line 23 of file TracksterP4FromEnergySumPlugin.cc.

Referenced by calcP4().

layer_clusters_token_

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

private

Definition at line 25 of file TracksterP4FromEnergySumPlugin.cc.

Referenced by setP4().

vertex_token_

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

private

Definition at line 24 of file TracksterP4FromEnergySumPlugin.cc.

Referenced by setP4().