Merge branch 'master' of https://github.com/ufo-kit/uca-ufo

1 files changed, 37 insertions, 0 deletions

diff --git a/tests/roi.py b/tests/roi.py
new file mode 100644
index 0000000..fad0717
--- /dev/null
+++ b/tests/roi.py
@@ -0,0 +1,37 @@
+import argparse
+import PyTango
+import tifffile
+
+
+def grab(camera, height=3840, offset=0):
+    if height < 3840:
+        camera.roi_y0 = offset
+        camera.roi_height = height
+    else:
+        camera.roi_height = height
+        camera.roi_y0 = offset
+
+    try:
+        camera.Start()
+    except:
+        camera.Stop()
+        camera.Start()
+
+    frame = camera.image
+    tifffile.imsave('frame-{}-{}.tif'.format(offset, height), frame)
+    camera.Stop()
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--device', '-d', type=str, required=True,
+                        help="TANGO device path")
+
+    args = parser.parse_args()
+
+    camera = PyTango.DeviceProxy(args.device)
+    camera.trigger_source = 0
+
+    grab(camera)
+    grab(camera, height=3640)
+    grab(camera, height=3640, offset=200)