add MR as an SOP Class that can be used for doing a Force Link

sjswerdloff · Apr 20, 2024 · e88aba1 · e88aba1
1 parent efb8ac6
commit e88aba1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/Model/ForceLink.py b/src/Model/ForceLink.py
@@ -60,8 +60,14 @@ def force_link(frame_overwrite, file_path, dicom_array_in):
     new_id = ""
     new_study_id = ""
     for dicom_file in dicom:
-        if dicom_file.FrameOfReferenceUID == frame_overwrite and \
-                dicom_file.SOPClassUID == src.dicom_constants.CT_IMAGE:
+        if (
+            (dicom_file.FrameOfReferenceUID == frame_overwrite)
+            and
+            (dicom_file.SOPClassUID == src.dicom_constants.CT_IMAGE
+            or
+            dicom_file.SOPClassUID == src.dicom_constants.MR_IMAGE)
+            ):
+
             new_id = dicom_file.FrameOfReferenceUID
             new_study_id = dicom_file.StudyInstanceUID
             break

diff --git a/src/dicom_constants.py b/src/dicom_constants.py
@@ -1,6 +1,7 @@
 """Contains DICOM standard SOP class UID"""
 
 CT_IMAGE = "1.2.840.10008.5.1.4.1.1.2"
+MR_IMAGE = "1.2.840.10008.5.1.4.1.1.4"
 RT_STRUCTURE_SET = "1.2.840.10008.5.1.4.1.1.481.3"
 RT_DOSE = "1.2.840.10008.5.1.4.1.1.481.2"
 RT_PLAN = "1.2.840.10008.5.1.4.1.1.481.5"