BackProjectorByBinNiftyPET.h

Go to the documentation of this file.

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

    SPDX-License-Identifier: Apache-2.0

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

#include "stir/RegisteredParsingObject.h"
#include "stir/recon_buildblock/BackProjectorByBin.h"
#include "stir/recon_buildblock/NiftyPET_projector/NiftyPETHelper.h"

START_NAMESPACE_STIR

class DataSymmetriesForViewSegmentNumbers;

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

  BackProjectorByBinNiftyPET();

  virtual ~BackProjectorByBinNiftyPET();

  virtual void initialise_keymap();


  virtual 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
  );

  virtual const DataSymmetriesForViewSegmentNumbers* get_symmetries_used() const;

  void back_project(const ProjData&, int subset_num = 0, int num_subsets = 1);

  virtual void get_output(DiscretisedDensity<3, float>&) const;

  virtual void start_accumulating_in_new_target();

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

  void set_use_truncation(const bool use_truncation) { _use_truncation = use_truncation; }

protected:
  virtual 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);

private:
  shared_ptr<DataSymmetriesForViewSegmentNumbers> _symmetries_sptr;
  NiftyPETHelper _helper;
  int _cuda_device;
  bool _cuda_verbosity;
  std::vector<float> _np_sino_w_gaps;
  bool _use_truncation;
};

END_NAMESPACE_STIR

#endif // __stir_gpu_BackProjectorByBinNiftyPET_h__