Add JPEG handler (used by webcams at NSLS-II BMM and elsewhere)

bluesky · May 29, 2024 · 3787364 · 3787364
1 parent f58ca85
commit 3787364
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/area_detector_handlers/handlers.py b/area_detector_handlers/handlers.py
@@ -8,6 +8,7 @@
 import numpy as np
 import pandas as pd
 import tifffile
+from PIL import Image
 
 from . import HandlerBase
 from .spe_reader import PrincetonSPEFile
@@ -654,3 +655,14 @@ def __init__(self, filename, frame_per_point=1):
         super(TimepixHDF5Handler, self).__init__(
                 filename=filename, key=self.hardcoded_key,
                 frame_per_point=frame_per_point)
+
+
+class JPEGHandler:
+    specs = {"JPEG"}
+
+    def __init__(self, resource_path):
+        self._resource_path = resource_path
+
+    def __call__(self):
+        filepath = self._resource_path
+        return np.asarray(Image.open(filepath))
diff --git a/setup.py b/setup.py
@@ -66,6 +66,7 @@
             "MERLIN_FLY_STREAM_V1 = area_detector_handlers.handlers:BulkXSPRESS",
             "MERLIN_FLY = area_detector_handlers.handlers:BulkXSPRESS",
             "IMM = area_detector_handlers.handlers:IMMHandler",
+            "JPEG = area_detector_handlers.handlers:JPEGHandler",
         ]
     },
     include_package_data=True,