LmToProjDataNiftyPET.h

Go to the documentation of this file.

#pragma once
//
//
/*
    Copyright (C) 2020, University College of London
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/

#include "stir/listmode/LmToProjDataAbstract.h"
#include "stir/error.h"

START_NAMESPACE_STIR

class ProjData;

class LmToProjDataNiftyPET : public LmToProjDataAbstract
{
public:
  LmToProjDataNiftyPET();

  virtual ~LmToProjDataNiftyPET() {}

  void set_span(const int span) { _span = span; }

  void set_cuda_device(const char cuda_device) { _cuda_device = cuda_device; }

  void set_cuda_verbosity(const bool cuda_verbosity) { _cuda_verbosity = cuda_verbosity; }

  void set_start_time(const int start_time) { _start_time = start_time; }

  void set_stop_time(const int stop_time) { _stop_time = stop_time; }

  void set_listmode_binary_file(const std::string& listmode_binary_file) { _listmode_binary_file = listmode_binary_file; }

  void set_norm_binary_file(const std::string& norm_binary_file) { _norm_binary_file = norm_binary_file; }

  virtual void process_data();

  shared_ptr<const ProjData> get_prompts_sptr() const { return _prompts_sptr; }

  shared_ptr<const ProjData> get_delayeds_sptr() const { return _delayeds_sptr; }

  shared_ptr<const ProjData> get_randoms_sptr() const { return _randoms_sptr; }

  shared_ptr<const ProjData> get_norm_sptr() const
  {
    if (_norm_binary_file.empty())
      error("Set norm binary filename before extracting listmode.");
    return _norm_sptr;
  }

private:
  void check_input() const;

  int _span;
  char _cuda_device;
  bool _cuda_verbosity;
  int _start_time, _stop_time;
  std::string _listmode_binary_file;
  std::string _norm_binary_file;
  shared_ptr<ProjData> _prompts_sptr;
  shared_ptr<ProjData> _delayeds_sptr;
  shared_ptr<ProjData> _randoms_sptr;
  shared_ptr<ProjData> _norm_sptr;
};

END_NAMESPACE_STIR