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-2021 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

// Acts include(s).
#include "Acts/Seeding/SeedFilter.hpp"
#include "Acts/Seeding/SeedFinderConfig.hpp"
#include "Acts/Seeding/SpacePointGrid.hpp"

// SYCL plugin include(s).
#include "Acts/Plugins/Sycl/Seeding/DeviceExperimentCuts.hpp"
#include "Acts/Plugins/Sycl/Seeding/detail/Types.hpp"
#include "Acts/Plugins/Sycl/Utilities/QueueWrapper.hpp"

// VecMem include(s).
#include "vecmem/memory/memory_resource.hpp"

namespace Acts::Sycl {

template <typename external_spacepoint_t>
class SeedFinder {
 public:
  SeedFinder(Acts::SeedFinderConfig<external_spacepoint_t> config,
             const Acts::SeedFinderOptions& options,
             const Acts::Sycl::DeviceExperimentCuts& cuts,
             Acts::Sycl::QueueWrapper wrappedQueue,
             vecmem::memory_resource& resource,
             vecmem::memory_resource* device_resource = nullptr);

  ~SeedFinder() = default;
  SeedFinder() = delete;
  SeedFinder(const SeedFinder<external_spacepoint_t>&) = delete;
  SeedFinder<external_spacepoint_t>& operator=(
      const SeedFinder<external_spacepoint_t>&) = delete;

  template <typename sp_range_t>
  std::vector<Seed<external_spacepoint_t>> createSeedsForGroup(
      Acts::SpacePointData& spacePointData,
      Acts::SpacePointGrid<external_spacepoint_t>& grid,
      const sp_range_t& bottomSPs, const std::size_t middleSPs,
      const sp_range_t& topSPs) const;

 private:
  Acts::SeedFinderConfig<external_spacepoint_t> m_config;
  Acts::SeedFinderOptions m_options;

  Acts::Sycl::DeviceExperimentCuts m_deviceCuts;

  Acts::Sycl::detail::DeviceSeedFinderConfig m_deviceConfig;

  QueueWrapper m_wrappedQueue;

  vecmem::memory_resource* m_resource;

  vecmem::memory_resource* m_device_resource;
};

}  // namespace Acts::Sycl

// Include the template implementation.
#include "Acts/Plugins/Sycl/Seeding/SeedFinder.ipp"