SimulatedCamera: new sensor_shape argument instead of hardcoding (512, 512).

Author: carandraug

NEWS.rst

@@ -28,6 +28,9 @@ Version 0.7.0 (upcoming)
     accidentally changed from "strobed" to "trig. first".  This has
     now been reversed.
 
+  * `SimulatedCamera` now takes an optional `sensor_shape` constructor
+    argument instead of always being 512x512 pixels.
+
 * The device server logging was broken in version 0.6.0 for Windows
   and macOS (systems not using fork for multiprocessing).  This
   version fixes that issue.

microscope/simulators/__init__.py

@@ -29,6 +29,7 @@
 import random
 import time
 import typing
+from typing import Tuple
 
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
@@ -154,10 +155,11 @@ def sawtooth(self, w, h, dark, light):
 class SimulatedCamera(
     microscope._utils.OnlyTriggersOnceOnSoftwareMixin, microscope.abc.Camera
 ):
-    def __init__(self, **kwargs):
+    def __init__(self, sensor_shape: Tuple[int, int] = (512, 512), **kwargs):
         super().__init__(**kwargs)
         # Binning and ROI
-        self._roi = microscope.ROI(0, 0, 512, 512)
+        self._sensor_shape = sensor_shape
+        self._roi = microscope.ROI(0, 0, *sensor_shape)
         self._binning = microscope.Binning(1, 1)
         # Function used to generate test image
         self._image_generator = _ImageGenerator()
@@ -278,7 +280,7 @@ def get_cycle_time(self):
         return self._exposure_time
 
     def _get_sensor_shape(self):
-        return (512, 512)
+        return self._sensor_shape
 
     def soft_trigger(self):
         # deprecated, use self.trigger()