flir_capture_multiple_video.py

import sys
import time
import PySpin

import numpy as np
import cv2

import SaveToAvi

def set_trigger_mode(cam, triggerSource):
    cam.TriggerMode.SetValue(PySpin.TriggerMode_Off)
    cam.TriggerSource.SetValue(triggerSource)
    cam.TriggerMode.SetValue(PySpin.TriggerMode_On)

def reset_trigger_mode_software(cam):
    cam.TriggerMode.SetValue(PySpin.TriggerMode_Off)
    print("reset trigger mode")

def mode_status():
    if sw_vid:
        print("record video")
    elif not sw_vid:
        if sw_con:
            print("record continuous photo")
        else:
            print("record single photo")

#
#   setup
#

system = PySpin.System.GetInstance()
cam_list = system.GetCameras()

master_id = "18284509"
master = None

print("number of cameras {}".format(cam_list.GetSize()))


if cam_list.GetSize() == 0:
    print("no cameras found, aborting")
    system.ReleaseInstance()
    del system
    sys.exit()

cameras = []
for i in range(cam_list.GetSize()):
    try:
        cam = cam_list.GetByIndex(i)
        cam_id = cam.GetUniqueID()
        cam.Init()
    

        if cam_id == master_id:
            print("master: {}".format(cam_id))
            master = cam
            set_trigger_mode(cam, PySpin.TriggerSource_Software)
        else:
            print("follower: {}".format(cam_id))
            set_trigger_mode(cam, PySpin.TriggerSource_Line3)
        cam.AcquisitionMode.SetValue(PySpin.AcquisitionMode_Continuous)
        cam.BeginAcquisition()
        cameras.append(cam)
    except PySpin.SpinnakerException as ex:
        print("Error: {}".format(ex))


#
#   loop
#


count = 0
sw_rec = False
sw_con = False
sw_vid = False
size = 4
frame = {}

from path import Path
directory = Path("./")


camera_dict = { "18284509": "", "18284511": "_1", "18284512": "_2"}

print("start loop")

while 1:

    key = cv2.waitKey(1)

    if key == 27: # ESC
        cv2.destroyAllWindows()
        break
    elif key == 32: # SPACE
        if sw_con:
            sw_rec = not sw_rec
            if sw_rec:
                print("start recording")
            elif not sw_rec:
                print("finish recording")
        elif not sw_vid:
            print("take picture:{}".format(count))
            for key, value in frame.items():
                cv2.imwrite("frame_{:012}{}.jpg".format(count,camera_dict[key]), value)
            count = count + 1
    elif key == 99: # C
        sw_con = not sw_con
        mode_status()
    elif key == 114: # R
        count = 0
        print("reset count")
        files = directory.walkfiles("*.jpg")
        files = directory.walkfiles("*.avi")
        for file in files:
            file.remove()
        print("reset .jpg .avi")
    elif key == 118: # V
        sw_vid = not sw_vid
        mode_status()

    try:
        master.TriggerSoftware.Execute()
    except PySpin.SpinnakerException as ex:
        print("Error: {}".format(ex))

    for cam in cameras:
        try:
            i = cam.GetNextImage()
            #print(i.GetWidth(), i.GetHeight(), i.GetBitsPerPixel())

            if i.IsIncomplete():
                pass
            else:
                cam_id = cam.GetUniqueID()
                # see documentation: enum ColorProcessingAlgorithm 
                image_converted = i.Convert(PySpin.PixelFormat_BGR8, PySpin.DIRECTIONAL_FILTER)
                image_data = image_converted.GetData()
                cvi = np.frombuffer(image_data, dtype=np.uint8)
                cvi = cvi.reshape((i.GetHeight(),i.GetWidth(),3))
                frame[cam_id] = cvi

                if sw_vid:
                    print("recording video")
                    #..............................................

                if not sw_rec:
                    res = cv2.resize(cvi, (int(1280/size),int(1024/size)))
                    cv2.imshow("cam{}".format(cam_id),res)
                
                if sw_rec:
                    if not sw_vid:
                        print("take picture:{}".format(count))
                        for key, value in frame.items():
                            cv2.imwrite("frame_{:012}{}.jpg".format(count,camera_dict[key]), value)
                        count = count + 1
            i.Release()
            del i
        except PySpin.SpinnakerException as ex:
            print("Error: {}".format(ex))

#
#   cleanup
#

del master
for cam in cameras:
    cam.EndAcquisition()
    reset_trigger_mode_software(cam)
    cam.DeInit()
    del cam
del cameras
del cam_list

system.ReleaseInstance()
del system