migrated code to public repository

code/recording/data_grabber.py

@@ -0,0 +1,61 @@
+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.'
+
+
+