import cv2
import openpiv.tools
import openpiv.filters

videoInputPath = videoInputPath = r'~/Desktop/reu/DJI_0001.MOV'
pivOutputPath = r'/run/user/1000/gvfs/sftp:host=isoptera.lcsc.edu,user=jamcdonald/home/jamcdonald/reu/OpenPIV_Out'
maskRemoveOutputPath = r'/run/user/1000/gvfs/sftp:host=isoptera.lcsc.edu,user=jamcdonald/home/jamcdonald/reu/Mask_Out'

#Capture video and save frame count
cap = cv2.VideoCapture(videoInputPath)
frameCount = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
cap.release()

#Display vector fields for corresponding frames
for i in range(0, 99):
    openpiv.tools.display_vector_field(pivOutputPath + "\\piv%4d.txt" % i)
    openpiv.tools.display_vector_field(maskRemoveOutputPath + "\\maskRemoved%4d.txt" % i)