From 59c779899fb21c7e1466b84fe235fbf25abddda1 Mon Sep 17 00:00:00 2001
From: swittl <simon.wittl@th-deg.de>
Date: Thu, 28 Nov 2024 15:34:50 +0100
Subject: [PATCH] added foc quat to get
---
src/thd_json/projection/__init__.py | 2 ++
src/thd_json/projection_geometry/__init__.py | 5 +++++
2 files changed, 7 insertions(+)
diff --git a/src/thd_json/projection/__init__.py b/src/thd_json/projection/__init__.py
index 903f659..06c9b53 100644
--- a/src/thd_json/projection/__init__.py
+++ b/src/thd_json/projection/__init__.py
@@ -49,6 +49,7 @@ def get_projection_dict(
image_dimensions_px: np.ndarray,
pixel_pitch_mm: np.ndarray,
header: JsonHeader | None = None,
+ focal_spot_orientation_quat: np.ndarray | None = None
):
if header is None:
header = generate_header()
@@ -59,6 +60,7 @@ def get_projection_dict(
detector_center_position_mm,
detector_center_orientation_quat,
header,
+ focal_spot_orientation_quat
)
projection['detector'] = get_detector_dict(image_dimensions_px, pixel_pitch_mm)
diff --git a/src/thd_json/projection_geometry/__init__.py b/src/thd_json/projection_geometry/__init__.py
index f2e0f9f..de1571f 100644
--- a/src/thd_json/projection_geometry/__init__.py
+++ b/src/thd_json/projection_geometry/__init__.py
@@ -45,6 +45,7 @@ def get_projection_geometry_dict(
detector_center_position_mm: np.ndarray,
detector_center_orientation_quat: np.ndarray,
header: JsonHeader | None = None,
+ focal_spot_orientation_quat: np.ndarray | None = None
):
if header is None:
header = generate_header()
@@ -59,6 +60,10 @@ def get_projection_geometry_dict(
projection_geometry["detector_center_orientation_quat"] = quaternion_to_dict(
detector_center_orientation_quat
)
+ if focal_spot_orientation_quat is not None:
+ projection_geometry["focal_spot_orientation_quat"] = quaternion_to_dict(
+ focal_spot_orientation_quat
+ )
projection_geometry["header"] = {"uuid": header.uuid}
--
GitLab