summaryrefslogtreecommitdiffstats
path: root/tests/roi.py
blob: f429740dbaf3f8c3996ec01482eb8fd057f4f373 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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, help="TANGO device path")

    args = parser.parse_args()

    camera = PyTango.DeviceProxy(args.device)

    grab(camera)
    grab(camera, height=3640)
    grab(camera, height=3640, offset=200)