fix: 🐛 Able to handle OPTOS PSEUDOCOLOUR_ULTRAWIDEFIELD modality

pontikos-lab · Nov 12, 2024 · 728cc86 · 728cc86
1 parent 96f89bd
commit 728cc86
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 1 deletion.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -8,7 +8,7 @@ repos:
       - id: check-shebang-scripts-are-executable
       - id: check-added-large-files
         args:
-          - "--maxkb=10000"
+          - "--maxkb=12000"
       # - id: debug-statements # not for process_dcm
       - id: check-yaml
         exclude: mkdocs.yml

diff --git a/process_dcm/const.py b/process_dcm/const.py
@@ -70,6 +70,12 @@ class ImageModality(Enum):
 
     FACE_PHOTO = ("FACE", "Face photo", ModalityFlag.IS_2D_IMAGE | ModalityFlag.IS_ANTERIOR | ModalityFlag.SENSITIVE)
 
+    PSEUDOCOLOUR_ULTRAWIDEFIELD = (
+        "PCUWF",
+        "Pseudocolour Ultra-widefield",
+        ModalityFlag.IS_2D_IMAGE | ModalityFlag.IS_INTERIOR | ModalityFlag.IS_COLOUR,
+    )
+
     # VF-related modalities
     FDF = ("FDF", "Flicker Defined Form Perimetry")
     SAP = ("SAP", "Standard Automated Perimetry")

diff --git a/process_dcm/utils.py b/process_dcm/utils.py
@@ -249,6 +249,8 @@ def update_modality(dcm: FileDataset) -> bool:
     elif dcm.Modality == "OP":
         if dcm.Manufacturer.upper() == "TOPCON":
             dcm.Modality = ImageModality.COLOUR_PHOTO
+        elif dcm.Manufacturer.upper() == "OPTOS" and dcm.HorizontalFieldOfView == 200:
+            dcm.Modality = ImageModality.PSEUDOCOLOUR_ULTRAWIDEFIELD
         elif " IR" in dcm.get("SeriesDescription", ""):
             dcm.Modality = ImageModality.SLO_INFRARED
         elif " BAF " in dcm.get("SeriesDescription", ""):

diff --git a/tests/example-optos/fundus.dcm b/tests/example-optos/fundus.dcm
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -192,6 +192,21 @@ def test_process_task():
         assert result == ("0780320450", "bbff7a25-d32c-4192-9330-0bb01d49f746")
 
 
+def test_process_task_optos():
+    with TemporaryDirectory() as tmpdirname:
+        output_dir = Path(tmpdirname)
+        task_data = ("tests/example-optos/", str(output_dir))
+        image_format = "png"
+        overwrite = True
+        verbose = True
+        keep = ""
+        mapping = ""
+        group = True
+        tol = 2
+        result = process_task(task_data, image_format, overwrite, verbose, keep, mapping, group, tol)
+        assert result == ("0570586923", "BEH002")
+
+
 # def test_process_taskL():
 #     with TemporaryDirectory() as tmpdirname:
 #         output_dir = Path(tmpdirname)