#!/usr/bin/python3
import numpy as np
import cv2
from math import sqrt, atan2
def vector_visualizer(vidIn,vidOut,frameSampleRate,frameSampleSize):
	cap = cv2.VideoCapture(vidIn)

	# Define the codec and create VideoWriter object
	fourcc = cv2.VideoWriter_fourcc(*'XVID')
	frameRate = cap.get(cv2.CAP_PROP_FPS)
	framesize=(round(cap.get(cv2.CAP_PROP_FRAME_WIDTH))//frameSampleSize,round(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))//frameSampleSize)
	out = cv2.VideoWriter(vidOut,fourcc, frameRate/frameSampleRate, framesize)
	count=0
	frameCount = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

	averages={}
	counts={}
	lastDir=0
	while(cap.isOpened() and count<frameCount-1 ):
		ret, frame = cap.read()
		if ret==True:
			if(count%frameSampleRate==0):
				newFrame= cv2.resize(frame,framesize,fx=0,fy=0, interpolation = cv2.INTER_CUBIC)

				out.write(newFrame)
			count+=1
			if(count%100==0):
				print(count)
		else:
			break
	print('\a')
	# Release everything if job is finished
	cap.release()
	out.release()
	cv2.destroyAllWindows()

if __name__ == '__main__':

	import os
	from argparse import *
	cwd=os.getcwd()
	parser=ArgumentParser(description="Tool to down sample footage")
	parser.add_argument('VIDEO',type=str, help="Start", default="",nargs='?')
	parser.add_argument('OUTPUT',type=str, help="output video (avi)", default="",nargs='?')
	parser.add_argument('-f',dest="frame",type=int, help="framerate downscale multiplier",default=1)
	parser.add_argument('-s',dest="scale",type=int, help="frame size downscale multiplier",default=1)
	args = parser.parse_args()
	
	videoInputPath = cwd+r'/DJI_0001.MOV'
	videoOutputPath = cwd+r'/output.avi'#video must be avi

	
	if args.VIDEO=="":
		inputV=input("Video to use (leave blank for default): ")
		if not inputV=="":
			videoInputPath = inputV
	else:
		videoInputPath=args.VIDEO


	if args.OUTPUT=="":
		inputV=input("file to export (must be .avi)(leave blank for default): ")
		if not inputV=="":
			videoOutputPath = inputV
	else:
		videoOutputPath=args.OUTPUT	

	vector_visualizer(videoInputPath,videoOutputPath,args.frame,args.scale)