From f9a097961336998c044e4cc13b64ce9af89f5326 Mon Sep 17 00:00:00 2001 From: Noemi Calace <58694235+noemina@users.noreply.github.com> Date: Wed, 8 Jun 2022 09:35:18 +0200 Subject: [PATCH] feat: Allow passing seed quality to seeds (#1268) * passing quality to seeds * remove what not needed * remove useless include * fixing format * re-trigger CI Co-authored-by: robertlangenberg <56069170+robertlangenberg@users.noreply.github.com> Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com> --- Core/include/Acts/Seeding/Seed.hpp | 9 +++++++-- Core/include/Acts/Seeding/SeedFilter.ipp | 2 +- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/Core/include/Acts/Seeding/Seed.hpp b/Core/include/Acts/Seeding/Seed.hpp index a00abefed00..46be4589a20 100644 --- a/Core/include/Acts/Seeding/Seed.hpp +++ b/Core/include/Acts/Seeding/Seed.hpp @@ -8,6 +8,7 @@ #pragma once +#include #include #include @@ -21,16 +22,19 @@ class Seed { public: Seed(const SpacePoint& b, const SpacePoint& m, const SpacePoint& u, - float vertex); + float vertex, + float seedQuality = -std::numeric_limits::infinity()); Seed(const Seed&) = default; Seed& operator=(const Seed&) = default; const auto& sp() const { return m_spacepoints; } double z() const { return m_zvertex; } + float seedQuality() const { return m_seedQuality; } private: boost::container::small_vector m_spacepoints; float m_zvertex; + float m_seedQuality; }; /////////////////////////////////////////////////////////////////////////////// @@ -39,11 +43,12 @@ class Seed { template Seed::Seed(const SpacePoint& b, const SpacePoint& m, - const SpacePoint& u, float vertex) { + const SpacePoint& u, float vertex, float seedQuality) { m_zvertex = vertex; m_spacepoints.push_back(&b); m_spacepoints.push_back(&m); m_spacepoints.push_back(&u); + m_seedQuality = seedQuality; } } // namespace Acts diff --git a/Core/include/Acts/Seeding/SeedFilter.ipp b/Core/include/Acts/Seeding/SeedFilter.ipp index 797d20eb60c..6225aac0d74 100644 --- a/Core/include/Acts/Seeding/SeedFilter.ipp +++ b/Core/include/Acts/Seeding/SeedFilter.ipp @@ -298,7 +298,7 @@ void SeedFilter::filterSeeds_1SpFixed( outIt = Seed{ (*it).second->sp[0]->sp(), (*it).second->sp[1]->sp(), - (*it).second->sp[2]->sp(), (*it).second->z()}; + (*it).second->sp[2]->sp(), (*it).second->z(), bestSeedQuality}; numTotalSeeds += 1; } }