TrackerTraitsGPU.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#ifndef ITSTRACKINGGPU_TRACKERTRAITSGPU_H_
#define ITSTRACKINGGPU_TRACKERTRAITSGPU_H_
 
#include "ITStracking/TrackerTraits.h"
#include "ITStrackingGPU/TimeFrameGPU.h"
 
namespace o2
{
namespace its
{
 
template <int nLayers = 7>
class TrackerTraitsGPU final : public TrackerTraits
{
 public:
  TrackerTraitsGPU() = default;
  ~TrackerTraitsGPU() override = default;
 
  void adoptTimeFrame(TimeFrame* tf) final;
  void initialiseTimeFrame(const int iteration) final;
 
  void computeLayerTracklets(const int iteration, int, int) final;
  void computeLayerCells(const int iteration) final;
  void findCellsNeighbours(const int iteration) final;
  void findRoads(const int iteration) final;
 
  bool supportsExtendTracks() const noexcept final { return false; }
  bool supportsFindShortPrimaries() const noexcept final { return false; }
 
  void setBz(float) final;
 
  const char* getName() const noexcept final { return "GPU"; }
  bool isGPU() const noexcept final { return true; }
 
  // TimeFrameGPU information forwarding
  int getTFNumberOfClusters() const override;
  int getTFNumberOfTracklets() const override;
  int getTFNumberOfCells() const override;
 
 private:
  IndexTableUtils* mDeviceIndexTableUtils;
  gpu::TimeFrameGPU<7>* mTimeFrameGPU;
};
 
template <int nLayers>
inline void TrackerTraitsGPU<nLayers>::adoptTimeFrame(TimeFrame* tf)
{
  mTimeFrameGPU = static_cast<gpu::TimeFrameGPU<nLayers>*>(tf);
  mTimeFrame = static_cast<TimeFrame*>(tf);
}
} // namespace its
} // namespace o2
 
#endif