BackProjectorByBinParallelproj.h

Go to the documentation of this file.

//
//
/*
    Copyright (C) 2019, 2021 University College London
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/
#ifndef __stir_gpu_BackProjectorByBinParallelproj_h__
#define __stir_gpu_BackProjectorByBinParallelproj_h__

#include "stir/RegisteredParsingObject.h"
#include "stir/recon_buildblock/BackProjectorByBin.h"
//#include "stir/recon_buildblock/Parallelproj_projector/ParallelprojHelper.h"

START_NAMESPACE_STIR

class DataSymmetriesForViewSegmentNumbers;
class ProjDataInMemory;
namespace detail
{
class ParallelprojHelper;
}

class BackProjectorByBinParallelproj : public RegisteredParsingObject<BackProjectorByBinParallelproj, BackProjectorByBin>
{
public:
  static const char* const registered_name;

  BackProjectorByBinParallelproj();

  ~BackProjectorByBinParallelproj() override;

  void initialise_keymap() override;


  void set_up(const shared_ptr<const ProjDataInfo>& proj_data_info_ptr,
              const shared_ptr<const DiscretisedDensity<3, float>>& density_info_sptr // TODO should be Info only
              ) override;

  const DataSymmetriesForViewSegmentNumbers* get_symmetries_used() const override;

#if 0
  void back_project(const ProjData&, int subset_num = 0, int num_subsets = 1);
#endif
  void get_output(DiscretisedDensity<3, float>&) const override;

  void start_accumulating_in_new_target() override;

  void set_defaults() override;

  void set_verbosity(const bool verbosity)
  {
    _cuda_verbosity = verbosity;
  }

  // set/get number of gpu chunks to use
  void set_num_gpu_chunks(int num_gpu_chunks)
  {
    _num_gpu_chunks = num_gpu_chunks;
  }
  int get_num_gpu_chunks()
  {
    return _num_gpu_chunks;
  }

  BackProjectorByBinParallelproj* clone() const override;

protected:
  void actual_back_project(const RelatedViewgrams<float>&,
                           const int min_axial_pos_num,
                           const int max_axial_pos_num,
                           const int min_tangential_pos_num,
                           const int max_tangential_pos_num) override;

private:
  shared_ptr<DataSymmetriesForViewSegmentNumbers> _symmetries_sptr;
  shared_ptr<ProjDataInMemory> _proj_data_to_backproject_sptr;
  shared_ptr<detail::ParallelprojHelper> _helper;
  bool _do_not_setup_helper;
  friend class ProjectorByBinPairUsingParallelproj;
  void set_helper(shared_ptr<detail::ParallelprojHelper>);
  bool _cuda_verbosity;
  int _num_gpu_chunks;
};

END_NAMESPACE_STIR

#endif // __stir_gpu_BackProjectorByBinParallelproj_h__