Flir camera adaptor

copylot/gui/viewer/demo/demo_flir_camera.py

@@ -0,0 +1,46 @@
+from copylot.gui.viewer.viewer import Viewer
+from copylot.hardware.cameras.flir.flir_camera import FlirCamera
+import numpy as np
+
+# create camera object
+cam = FlirCamera()
+# open the system
+cam.open()
+
+
+# 2 ways to stream in viewer
+# FIRST OPTION: cycle one by one
+
+# function to take and stack an image
+def snap_mono(c: FlirCamera):
+    snap = c.snap()
+    return np.stack((snap, snap, snap), axis=2)
+
+
+# create initial blank (placeholder) image
+init_im = snap_mono(cam)
+
+# create vispy canvas with initial image
+view1 = Viewer(init_im)
+
+for i in range(5):
+    im = snap_mono(cam)
+    view1.update(im)
+    view1.process()  # introduces a delay between snapping the images
+
+# run the vispy event loop
+view1.run()
+
+# SECOND OPTION: get all images and then cycle them through the viewer
+# create vispy canvas with initial image
+view2 = Viewer(init_im)
+
+ims = cam.snap(10)
+for i in range(10):
+    ima = np.stack((ims[i], ims[i], ims[i]), axis=2)
+    view2.update(ima)
+    view2.process()
+
+view2.run()
+
+cam.close()

copylot/gui/viewer/viewer.py

@@ -1,3 +1,4 @@
+import numpy as np
 from vispy import scene
 from vispy import app
 
@@ -28,8 +29,43 @@ def __init__(self, img_data=None):
         self.view.camera.set_range()
         self.view.camera.zoom(1, (250, 200))
 
-    def run(self):
+        # set parameters for contrast
+        self.img_data = img_data
+        self._contrast_limits = None
+        self.update_freq = 0
+
+    @staticmethod
+    def run():
+        """
+        Start the vispy event loop
+        """
         app.run()
 
+    @staticmethod
+    def process():
+        """
+        Update the vispy event loop
+        """
+        app.process_events()
+
     def update(self, data):
+        """
+        Update the data layer of the image in the current viewbox
+
+        Parameters
+        ----------
+        data: ndarray of the new image to be shown
+        """
+        # input new image data
+        self.image.set_data(data)
+        # update the SceneCanvas object
+        self.canvas.update()
+
+    @property
+    def contrast_limits(self):
+        return self._contrast_limits
+
+    @contrast_limits.setter
+    def contrast_limits(self, factor):
+        # TODO: get tuple as argument, use vispy clim
         pass

copylot/hardware/cameras/flir/demo/demo.py

@@ -0,0 +1,66 @@
+from copylot.hardware.cameras.flir.flir_camera import FlirCamera
+
+if __name__ == '__main__':
+    cam = FlirCamera()
+
+    # open the system
+    cam.open()
+
+    # serial number
+    print(cam.device_id)
+    # list of cameras
+    print(cam.list_available_cameras())
+
+    # Return 10 frames and save output arrays as .csv files (this can be changed)
+
+    # Option 1: take multiple frames in a single acquisition
+    # Can control timeout (wait_time) for grabbing images from the camera buffer
+    snap1 = cam.snap(n_images=5, wait_time=1000)
+    cam.save_image(snap1)
+    # Option 2: iterate over snap()
+    # Saving in each iteration causes a delay between beginning/ending camera acquisition
+    # Could also collect the snap2 outputs and then save outside the loop to avoid delays
+    # (which is just what Option 1 does implicitly)
+    for i in range(0, 5):
+        snap2 = cam.snap()
+        cam.save_image(snap2)
+
+    # open again
+    cam.open()
+
+    # cam bit depth
+    print('Default bit depth', cam.bitdepth)
+    cam.bitdepth = 3
+    assert cam.bitdepth == 3
+
+    # cam shutter mode
+    print('Default shutter mode', cam.shutter_mode)
+    cam.shutter_mode = 'rolling'
+    assert cam.shutter_mode == 'rolling'
+
+    # cam binning - THERE ARE SOME SPECIFIC MULTIPLES
+    print('Default binsize', cam.binning)
+    cam.binning = (2, 2)
+    assert cam.binning == (2, 2)
+
+    # cam image size
+    print('Default image size', cam.image_size)  # set to the max by default
+    print('Limit image size', cam.image_size_limits)
+    cam.image_size = (1000, 2000)
+    assert cam.image_size == (1000, 2000)
+
+    # cam gain methods
+    print('Default AutoGain', cam.gain, cam.gain_limits)
+    cam.gain = 0.8
+    assert cam.gain == 0.8
+
+    # cam exposure methods
+    print('Default AutoExposure', cam.exposure, cam.exposure_limits)
+    cam.exposure = 100.0
+    assert cam.exposure == 100.0
+
+    # cam frame rate methods
+    print('Default frame rate', cam.framerate)
+    # set method might be removed
+
+    cam.close()