ticl::SuperclusteringDNNInputV2 Class Reference

#include <SuperclusteringDNNInputs.h>

Inheritance diagram for ticl::SuperclusteringDNNInputV2:

Public Member Functions
std::vector< float >	computeVector (ticl::Trackster const &ts_base, ticl::Trackster const &ts_toCluster) override
unsigned int	featureCount () const override
std::vector< std::string >	featureNames () const override
Public Member Functions inherited from ticl::AbstractSuperclusteringDNNInput
virtual	~AbstractSuperclusteringDNNInput ()=default

Detailed Description

Definition at line 63 of file SuperclusteringDNNInputs.h.

Member Function Documentation

computeVector()

std::vector< float > ticl::SuperclusteringDNNInputV2::computeVector ( ticl::Trackster const &  ts_base, ticl::Trackster const &  ts_toCluster  )

overridevirtual

Compute feature for seed and candidate pair

Implements ticl::AbstractSuperclusteringDNNInput.

Definition at line 63 of file SuperclusteringDNNInputs.cc.

References funct::abs(), ticl::Angle2D(), ticl::Trackster::barycenter(), ticl::Trackster::eigenvalues(), ticl::Trackster::eigenvectors(), f, ticl::Trackster::raw_energy(), makeMuonMisalignmentScenario::rot, and funct::sin().

                                                                                                                   {
    using ROOT::Math::XYVectorF;
    using ROOT::Math::XYZVectorF;
    using ROOT::Math::VectorUtil::Angle;
    XYZVectorF const& pca_seed_cmsFrame(ts_base.eigenvectors(0));
    XYZVectorF const& pca_cand_cmsFrame(ts_toCluster.eigenvectors(0));
    XYZVectorF xs(pca_seed_cmsFrame.Cross(XYZVectorF(0, 0, 1)).Unit());
    ROOT::Math::Rotation3D rot(xs, xs.Cross(pca_seed_cmsFrame).Unit(), pca_seed_cmsFrame);

    XYZVectorF pca_cand_seedFrame = rot(pca_cand_cmsFrame);  // seed coordinates

    float explVar_denominator = std::accumulate(
        std::begin(ts_toCluster.eigenvalues()), std::end(ts_toCluster.eigenvalues()), 0.f, std::plus<float>());
    float explVarRatio = 0.;
    if (explVar_denominator != 0.) {
      explVarRatio = ts_toCluster.eigenvalues()[0] / explVar_denominator;
    } else {
      edm::LogWarning("HGCalTICLSuperclustering")
          << "Sum of eigenvalues was zero for trackster. Could not compute explained variance ratio.";
    }

    return {{
        std::abs(ts_toCluster.barycenter().Eta()) - std::abs(ts_base.barycenter().Eta()),  //DeltaEtaBaryc
        ts_toCluster.barycenter().Phi() - ts_base.barycenter().phi(),                      //DeltaPhiBaryc
        ts_toCluster.raw_energy(),                                                         //multi_en
        ts_toCluster.barycenter().Eta(),                                                   //multi_eta
        (ts_toCluster.raw_energy() * std::sin(ts_toCluster.barycenter().Theta())),         //multi_pt
        ts_base.barycenter().Eta(),                                                        //seedEta
        ts_base.barycenter().Phi(),                                                        //seedPhi
        ts_base.raw_energy(),                                                              //seedEn
        (ts_base.raw_energy() * std::sin(ts_toCluster.barycenter().Theta())),              //seedPt
        static_cast<float>(Angle(pca_cand_cmsFrame, pca_seed_cmsFrame)),  // theta : angle between seed and candidate
        Angle2D(XYVectorF(pca_cand_seedFrame.x(), pca_cand_seedFrame.z()), XYVectorF(0, 1)),  // theta_xz_seedFrame
        Angle2D(XYVectorF(pca_cand_seedFrame.y(), pca_cand_seedFrame.z()), XYVectorF(0, 1)),  // theta_yz_seedFrame
        Angle2D(XYVectorF(pca_cand_cmsFrame.x(), pca_cand_cmsFrame.y()),
                XYVectorF(pca_seed_cmsFrame.x(), pca_seed_cmsFrame.y())),  // theta_xy_cmsFrame
        Angle2D(XYVectorF(pca_cand_cmsFrame.y(), pca_cand_cmsFrame.z()),
                XYVectorF(pca_seed_cmsFrame.y(), pca_seed_cmsFrame.z())),  // theta_yz_cmsFrame
        Angle2D(XYVectorF(pca_cand_cmsFrame.x(), pca_cand_cmsFrame.z()),
                XYVectorF(pca_seed_cmsFrame.x(), pca_seed_cmsFrame.z())),  // theta_xz_cmsFrame
        ts_toCluster.eigenvalues()[0],                                     // explVar
        explVarRatio                                                       // explVarRatio
    }};
  }

featureCount()

unsigned int ticl::SuperclusteringDNNInputV2::featureCount ( ) const

inlineoverridevirtual

Reimplemented from ticl::AbstractSuperclusteringDNNInput.

Definition at line 65 of file SuperclusteringDNNInputs.h.

{ return 17; }

featureNames()

std::vector<std::string> ticl::SuperclusteringDNNInputV2::featureNames ( ) const

inlineoverridevirtual

Get name of features. Used for SuperclusteringSampleDumper branch names (inference does not use the names, only the indices) The default implementation is meant to be overriden by inheriting classes

Reimplemented from ticl::AbstractSuperclusteringDNNInput.

Definition at line 69 of file SuperclusteringDNNInputs.h.

                                                       {
      return {"DeltaEtaBaryc",
              "DeltaPhiBaryc",
              "multi_en",
              "multi_eta",
              "multi_pt",
              "seedEta",
              "seedPhi",
              "seedEn",
              "seedPt",
              "theta",
              "theta_xz_seedFrame",
              "theta_yz_seedFrame",
              "theta_xy_cmsFrame",
              "theta_yz_cmsFrame",
              "theta_xz_cmsFrame",
              "explVar",
              "explVarRatio"};
    }