CListRecordPETSIRD.h

Go to the documentation of this file.

/*
    Copyright 2025, 2026 University Medical Center Groningen
    Copyright 2025 National Physical Laboratory
 
    SPDX-License-Identifier: Apache-2.0
    See STIR/LICENSE.txt for details.
*/

 
#ifndef __stir_listmode_CListRecordPETSIRD_H__
#define __stir_listmode_CListRecordPETSIRD_H__
 
#include "stir/listmode/CListEventScannerWithDiscreteDetectors.h"
#include "stir/listmode/CListRecord.h"
#include "stir/DetectionPositionPair.h"
#include "stir/PETSIRDInfo.h"
#include "stir/Succeeded.h"
 
START_NAMESPACE_STIR
 
template <class ProjDataInfoT>
class CListEventPETSIRD : public CListEventScannerWithDiscreteDetectors<ProjDataInfoT>
{
public:
  inline CListEventPETSIRD(shared_ptr<const ProjDataInfo> proj_data_info_sptr)
      : CListEventScannerWithDiscreteDetectors<ProjDataInfoT>(proj_data_info_sptr)
  {}
 
  inline CListEventPETSIRD(shared_ptr<const ProjDataInfo> proj_data_info_sptr,
                           const DetectionPositionPair<>& det_pos_pair,
                           bool is_prompt_v)
      : CListEventScannerWithDiscreteDetectors<ProjDataInfoT>(proj_data_info_sptr),
        _det_pos_pair(det_pos_pair),
        _is_prompt(is_prompt_v)
  {}
 
  inline bool is_prompt() const override { return this->_is_prompt; }
 
  bool operator==(const CListEventPETSIRD& other) const
  {
    if (this == &other)
      return true;
 
    return is_prompt() == other.is_prompt() && get_detection_position() == other.get_detection_position();
  }
 
  inline Succeeded set_prompt(const bool prompt) override
  {
    this->_is_prompt = prompt;
    return Succeeded::yes;
  }
 
  virtual void get_detection_position(DetectionPositionPair<>& det_pos_pair_arg) const override
  {
    det_pos_pair_arg = this->_det_pos_pair;
  }
 
  virtual void set_detection_position(const DetectionPositionPair<>& det_pos_pair_arg) override
  {
    this->_det_pos_pair = det_pos_pair_arg;
  }
 
private:
  DetectionPositionPair<> _det_pos_pair;
  bool _is_prompt;
};
 
class CListTimePETSIRD : public ListTime
{
public:
  inline unsigned long get_time_in_millisecs() const { return static_cast<unsigned long>(time); }
  inline Succeeded set_time_in_millisecs(const unsigned long time_in_millisecs)
  {
    time = time_in_millisecs;
    return Succeeded::yes;
  }
  bool operator==(const CListTimePETSIRD& other) const { return time == other.time; }
  inline bool is_time() const { return true; }
  uint32_t time;
};

class CListRecordPETSIRD : public CListRecord
{
public:
  CListRecordPETSIRD(shared_ptr<const PETSIRDInfo> petsird_info_sptr, shared_ptr<const ProjDataInfo> proj_data_info_sptr)
      : event_data(make_event_data(proj_data_info_sptr)),
        petsird_info_sptr(std::move(petsird_info_sptr)),
        proj_data_info_sptr(std::move(proj_data_info_sptr))
  {}

  bool is_time() const override { return true; }
  bool is_event() const override { return true; }
 
  CListEvent& event() override { return *event_data; }
  const CListEvent& event() const override { return *event_data; }
 
  CListTimePETSIRD& time() override { return time_data; }
  const CListTimePETSIRD& time() const override { return time_data; }
 
  bool operator==(const CListRecordPETSIRD& e2) const { return event_data == e2.event_data && time_data == e2.time_data; }
 
  Succeeded init_from_data(const petsird::CoincidenceEvent& event, const bool is_prompt);
 
private:
  static std::unique_ptr<CListEventScannerWithDiscreteDetectorsBase>
  make_event_data(shared_ptr<const ProjDataInfo> proj_data_info);
 
  std::unique_ptr<CListEventScannerWithDiscreteDetectorsBase> event_data;
  CListTimePETSIRD time_data;
 
  shared_ptr<const PETSIRDInfo> petsird_info_sptr;
  shared_ptr<const ProjDataInfo> proj_data_info_sptr;
 
  bool is_prompt_event = true;
};
 
END_NAMESPACE_STIR
 
#endif