Cuda/Seeding2/TestHostCuts.hpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #pragma once
0010
0011 // Local include(s).
0012 #include "TestSpacePoint.hpp"
0013
0014 // Acts include(s).
0015 #include "Acts/Seeding/IExperimentCuts.hpp"
0016
0017 /// Custom selection cuts for the test, used on the host
0018 class TestHostCuts : public Acts::IExperimentCuts<TestSpacePoint> {
0019  public:
0020   /// Returns seed weight bonus/malus depending on detector considerations.
0021   /// @param bottom bottom space point of the current seed
0022   /// @param middle middle space point of the current seed
0023   /// @param top top space point of the current seed
0024   /// @return seed weight to be added to the seed's weight
0025   float seedWeight(
0026       const Acts::InternalSpacePoint<TestSpacePoint>& bottom,
0027       const Acts::InternalSpacePoint<TestSpacePoint>& middle,
0028       const Acts::InternalSpacePoint<TestSpacePoint>& top) const final;
0029
0030   /// @param weight the current seed weight
0031   /// @param bottom bottom space point of the current seed
0032   /// @param middle middle space point of the current seed
0033   /// @param top top space point of the current seed
0034   /// @return true if the seed should be kept, false if the seed should be
0035   /// discarded
0036   bool singleSeedCut(
0037       float weight, const Acts::InternalSpacePoint<TestSpacePoint>& bottom,
0038       const Acts::InternalSpacePoint<TestSpacePoint>&,
0039       const Acts::InternalSpacePoint<TestSpacePoint>&) const final;
0040
0041   /// @param seedCandidates contains collection of seed candidates created for one middle
0042   /// space point in a std::tuple format
0043   /// @return vector of seed candidates that pass the cut
0044   std::vector<typename Acts::CandidatesForMiddleSp<
0045       const Acts::InternalSpacePoint<TestSpacePoint>>::value_type>
0046   cutPerMiddleSP(
0047       std::vector<typename Acts::CandidatesForMiddleSp<
0048           const Acts::InternalSpacePoint<TestSpacePoint>>::value_type>
0049           seedCandidates) const final;
0050
0051 };  // struct TestHostCuts