SeedFinder.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file SeedFinder.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

// CUDA plugin include(s).
#include "Acts/Plugins/Cuda/Seeding2/TripletFilterConfig.hpp"

// Acts include(s).
#include "Acts/Seeding/Seed.hpp"
#include "Acts/Seeding/SeedFilterConfig.hpp"
#include "Acts/Seeding/SeedfinderConfig.hpp"

namespace Acts {
namespace Cuda {

template <typename external_spacepoint_t>
class SeedFinder {
  // Public methods:

 public:
  SeedFinder(SeedfinderConfig<external_spacepoint_t> commonConfig,
             const SeedFilterConfig& seedFilterConfig,
             const TripletFilterConfig& tripletFilterConfig);

  template <typename sp_range_t>
  std::vector<Seed<external_spacepoint_t> > createSeedsForGroup(
      sp_range_t bottomSPs, sp_range_t middleSPs, sp_range_t topSPs) const;

 private:
  SeedfinderConfig<external_spacepoint_t> m_commonConfig;
  SeedFilterConfig m_seedFilterConfig;
  TripletFilterConfig m_tripletFilterConfig;
};

}  // namespace Cuda
}  // namespace Acts

// Include the template implementation.
#include "Acts/Plugins/Cuda/Seeding2/SeedFinder.ipp"