ticl::TracksterP4FromTrackAndPCA Class Reference

Inheritance diagram for ticl::TracksterP4FromTrackAndPCA:

Public Member Functions
void	setP4 (const std::vector< const Trackster *> &tracksters, std::vector< TICLCandidate > &ticl_cands, edm::Event &event) const override
	TracksterP4FromTrackAndPCA (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

Private Attributes
bool	energy_from_regression_
edm::EDGetTokenT< reco::TrackCollection >	tracks_token_

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

Detailed Description

Definition at line 11 of file TracksterP4FromTrackAndPCA.cc.

Constructor & Destructor Documentation

TracksterP4FromTrackAndPCA()

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

explicit

Definition at line 24 of file TracksterP4FromTrackAndPCA.cc.

      : TracksterMomentumPluginBase(ps, std::move(ic)),
        energy_from_regression_(ps.getParameter<bool>("energyFromRegression")),
        tracks_token_(ic.consumes<reco::TrackCollection>(ps.getParameter<edm::InputTag>("tracks"))) {}

Member Function Documentation

calcP4()

std::tuple<TracksterMomentumPluginBase::LorentzVector, float> ticl::TracksterP4FromTrackAndPCA::calcP4 ( const ticl::Trackster &  trackster ) const

private

setP4()

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

overridevirtual

Implements ticl::TracksterMomentumPluginBase.

Definition at line 29 of file TracksterP4FromTrackAndPCA.cc.

References cms::cuda::assert(), ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), hcalRecHitTable_cff::energy, energy_from_regression_, mps_fire::i, edm::HandleBase::id(), SiStripPI::min, ticl::mpion2, edm::Handle< T >::product(), mathSSE::sqrt(), JetHT_cfg::trackCollection, and tracks_token_.

Referenced by Jet.Jet::setCorrP4().

                                                                {
    edm::Handle<reco::TrackCollection> tracks_h;
    event.getByToken(tracks_token_, tracks_h);
    edm::ProductID trkId = tracks_h.id();
    const reco::TrackCollection& trackCollection = *tracks_h.product();

    auto size = std::min(tracksters.size(), ticl_cands.size());
    for (size_t i = 0; i < size; ++i) {
      const auto* trackster = tracksters[i];
      // If there's a track, use it.
      if (trackster->seedIndex() != -1) {
        assert(trackster->seedID() == trkId);
        auto const& tkRef = trackCollection[trackster->seedIndex()];
        auto const& three_mom = tkRef.momentum();
        constexpr double mpion2 = 0.13957 * 0.13957;
        double energy = std::sqrt(tkRef.momentum().mag2() + mpion2);
        math::XYZTLorentzVector trk_p4(three_mom.x(), three_mom.y(), three_mom.z(), energy);
        auto& ticl_cand = ticl_cands[i];
        ticl_cand.setP4(trk_p4);
        ticl_cand.setRawEnergy(energy);
      } else {
        auto direction = trackster->eigenvectors(0).Unit();
        auto energy = energy_from_regression_ ? trackster->regressed_energy() : trackster->raw_energy();
        direction *= energy;
        math::XYZTLorentzVector cartesian(direction.X(), direction.Y(), direction.Z(), energy);
        auto& ticl_cand = ticl_cands[i];
        ticl_cand.setP4(cartesian);
        ticl_cand.setRawEnergy(energy);
      }
    }
  }

Member Data Documentation

energy_from_regression_

bool ticl::TracksterP4FromTrackAndPCA::energy_from_regression_

private

Definition at line 20 of file TracksterP4FromTrackAndPCA.cc.

Referenced by setP4().

tracks_token_

edm::EDGetTokenT<reco::TrackCollection> ticl::TracksterP4FromTrackAndPCA::tracks_token_

private

Definition at line 21 of file TracksterP4FromTrackAndPCA.cc.

Referenced by setP4().