PETSIRDInfo.h

Go to the documentation of this file.

/*
   Copyright 2025, University Medical Center Groningen
 
    SPDX-License-Identifier: Apache-2.0
    See STIR/LICENSE.txt for detail
*/
#ifndef __stir_listmode_PETSIRDInfo_H__
#define __stir_listmode_PETSIRDInfo_H__
 
#include "stir/DetectionPosition.h"
#include "stir/DetectionPositionPair.h"
#include "petsird/protocols.h"
#include "stir/Scanner.h"
#include "stir/DetectorCoordinateMap.h"
#include <set>
#include <map>
 
#include "petsird_helpers.h"
#include "petsird_helpers/create.h"   // for make_detection_bin
#include "petsird_helpers/geometry.h" // depending on where get_detection_efficiency lives
 
START_NAMESPACE_STIR

struct ExpandedDetectionBinLess
{
 
  bool operator()(const petsird::ExpandedDetectionBin& a, const petsird::ExpandedDetectionBin& b) const
  {
    // Adjust field names if needed (I assume: module, element, energy_bin)
    if (a.module_index < b.module_index)
      return true;
    if (a.module_index > b.module_index)
      return false;
 
    if (a.element_index < b.element_index)
      return true;
    if (a.element_index > b.element_index)
      return false;
    return a.energy_index < b.energy_index;
  }
};

using PETSIRDToSTIRDetectorIndexMap
    = std::map<petsird::ExpandedDetectionBin, stir::DetectionPosition<>, ExpandedDetectionBinLess>;
 
using STIRToPETSIRDDetectorIndexMap = std::map<stir::DetectionPosition<>, petsird::ExpandedDetectionBin>;

*/
class PETSIRDInfo
{
public:
  explicit PETSIRDInfo(shared_ptr<const petsird::Header> header_sptr, std::string scanner_geometry = "");
 
  // void initialize();
 
  inline std::shared_ptr<Scanner> get_scanner_sptr() const { return stir_scanner_sptr; }
 
  inline const petsird::ScannerInformation& get_petsird_scanner_info() const { return petsird_header_sptr->scanner; }
 
  inline shared_ptr<PETSIRDToSTIRDetectorIndexMap> get_petsird_to_stir_map() const { return petsird_to_stir; }
 
  inline shared_ptr<STIRToPETSIRDDetectorIndexMap> get_stir_to_petsird_map() const { return stir_to_petsird; }
 
  inline shared_ptr<DetectorCoordinateMap::det_pos_to_coord_type> get_petsird_map_sptr() const { return petsird_map_sptr; }
 
  inline bool is_generic_geometry_used() const { return is_generic_geometry; }
 
  inline bool is_block_configuration_used() const { return is_block_configuration; }
 
  inline bool is_cylindrical_configuration_used() const { return is_cylindrical; };
 
  float get_detection_efficiency_for_bin(const stir::DetectionPositionPair<>& dp) const;
 
  float get_lower_energy_threshold() const;
 
  float get_upper_energy_threshold() const;
 
private:
  int figure_out_scanner_blocks_and_rotation_axis(std::set<float>& unique_dim1_values,
                                                  std::set<float>& unique_dim2_values,
                                                  std::set<float>& unique_dim3_values);
  void figure_out_block_angles(std::set<float>& unique_angle_modules, const int rot_axis);
  void figure_out_block_element_transformations(std::set<float>& unique_dim1_values,
                                                std::set<float>& unique_dim2_values,
                                                std::set<float>& unique_dim3_values,
                                                float& radius,
                                                int& radius_index,
                                                const int rotation_axis);


  shared_ptr<const petsird::Header> petsird_header_sptr;
  shared_ptr<Scanner> stir_scanner_sptr;
  uint32_t numberOfModules;
  uint32_t numberOfElementsIndices;
  uint32_t blocks_per_bucket_transaxial;
  uint32_t blocks_per_bucket_axial;
  uint32_t num_axial_crystals_per_block;
  uint32_t num_trans_crystals_per_block;
  bool is_cylindrical = true;
 
  bool is_generic_geometry = false;
 
  bool is_block_configuration = false;
 
  petsird::TypeOfModule type_of_module;
 
  petsird::TypeOfModulePair module_pair;
 
  std::string forced_geometry = "";
  shared_ptr<PETSIRDToSTIRDetectorIndexMap> petsird_to_stir;
 
  shared_ptr<STIRToPETSIRDDetectorIndexMap> stir_to_petsird;
  shared_ptr<DetectorCoordinateMap::det_pos_to_coord_type> petsird_map_sptr;
};
 
END_NAMESPACE_STIR
 
#endif