import os
from tracker import performMarkerTracking


ROOT_DATA_DIR = '/home/mmbrian/HiWi/etra2016_mohsen/code/recording/data/participants'
SQUARE_SIZE = '0.16'

def main(force = False):
	'''
	Processes all the participants recordings and performs marker tracking on each video and stores marker data in a npy file near the video
	'''
	c = 0
	for d1 in os.listdir(ROOT_DATA_DIR):
		if d1.startswith('p'): # every participant
			d2 = os.path.join(ROOT_DATA_DIR, d1) # .../pi/
			d2 = os.path.join(d2, os.listdir(d2)[0]) # .../pi/../
			for d3 in os.listdir(d2): # every recording
				d4 = os.path.join(d2, d3) # .../pi/../00X/
				print '> Processing', d4
				frames_dir = os.path.join(d4, '_aruco_frames.npy')
				if os.path.isfile(frames_dir):
					if not force: # already processed this recording
						print '> Recording already processed...'
						continue
					else: # video processed, but forced to process again
						# remove old file
						print '> Reprocessing file:', frames_dir
						print '> Removing old marker file...'
						os.remove(frames_dir)

				world_path = os.path.join(d4, 'world.avi')

				log_dir = os.path.join(d4, '_ArUco.log')
				read_log = False
				if os.path.isfile(log_dir):
					read_log = True
				else: # no log available > process video
					if not os.path.isfile(world_path): # .avi file exists
						# Creating avi
						print '> AVI video does not exists, generating it from mp4 source file...'
						os.popen('avconv -i ' + os.path.join(d4, 'world.mp4') + ' -c:a copy -c:v copy ' + world_path + ' -v quiet')
				
				# Perform tracking...
				performMarkerTracking(read_from_log = read_log, log = True, log_dir = log_dir, 
										recording_video = world_path + ' ', frames_dir = frames_dir, 
										square_size = SQUARE_SIZE) # in case of relocating camera.yml input the new path as cam_math
				if not read_log:
					# Removing avi file
					print '> Removing avi file:', world_path
					os.remove(world_path)
				c += 1
				print '> Processed %s recordings so far...' % c
				# print '> Halting...'
				# return

if __name__ == '__main__':
	main(force=False)
	print 'Finished.'