Adds support for different video extensions to the h264_to_mp4 script

Adds option to overwrite existing MP4s
Makes sure that files are merged in the right order

Author: ggilestro

accessories/h264_to_mp4.py

@@ -50,19 +50,22 @@ def get_video_fps(video_file):
         print(f"Error getting FPS: {e}")
         return None
 
-def process_video(folder, verbose=True):
+def process_video(folder, verbose=True, extension="h264"):
     """
     Process video in folder with detailed status updates and a progress bar.
     """
     os.chdir(folder)
     print(f"Processing folder: {folder}")
 
-    video_files = glob("*.h264")
+    video_files = glob(f"*.{extension}")
     if not video_files:
-        print("No .h264 files found in the folder.")
+        print(f"No .{extension} files found in the folder.")
         return
 
-    print(f"Number of .h264 files: {len(video_files)}")
+    # Sort files based on the last 5 digits in the filename
+    video_files.sort(key=lambda f: int(f.split('_')[-1].split('.')[0]))
+
+    print(f"Number of .{extension} files: {len(video_files)}")
 
     # Get the first video file to process
     video_file = video_files[0]
@@ -76,10 +79,10 @@ def process_video(folder, verbose=True):
     tmp_file = f"{prefix}.tmp"
     filename = f"{prefix}.mp4"
 
-    # Calculate total size of all .h264 files for progress reporting
+    # Calculate total size of all video files for progress reporting
     total_size = sum(os.path.getsize(f) for f in video_files)
     readable_size = sizeof_fmt(total_size)
-    print(f"Total size of all .h264 files: {readable_size}")
+    print(f"Total size of all .{extension} files: {readable_size}")
 
     # Merge files into one big chunk
     with open(tmp_file, 'wb') as wfd:
@@ -113,16 +116,13 @@ def process_video(folder, verbose=True):
     # Cleanup temporary files
     os.system(f"rm {tmp_file}")
 
-    if verbose:
-        print(f"\nSuccessfully processed files in folder {folder}")
-
 def print_progress(current_file, total_files, total_size, current_size):
     percent = (current_size / total_size) * 100
     bar_length = 40
     filled_length = int(round(bar_length * current_file / float(total_files)))
     bar = '#' * filled_length + '-' * (bar_length - filled_length)
-    sys.stdout.write(f"\rMerging file {current_file}/{total_files} |{bar}| {percent:.2f}% Completed")
-    sys.stdout.flush()
+    os.sys.stdout.write(f"\rMerging file {current_file}/{total_files} |{bar}| {percent:.2f}% Completed")
+    os.sys.stdout.flush()
 
 
 def sizeof_fmt(num, suffix='B'):
@@ -147,26 +147,34 @@ def list_mp4s (root_path):
     return have_mp4s
 
 
-def crawl (root_path):
+def crawl (root_path, extension="h264", force=False):
     '''
     crawl all terminal folders in root_path
     '''
 
     all_folders = [ x[0] for x in os.walk(root_path) ]
 
     have_mp4s = [p for p in all_folders if glob(os.path.join(p, "*.mp4"))]
-    terminal_folders = [p for p in all_folders if glob(os.path.join(p, "*.h264"))]
+    terminal_folders = [p for p in all_folders if glob(os.path.join(p, f"*.{extension}"))]
+
 
-    for folder in terminal_folders:
-        if folder not in have_mp4s:
-            process_video (folder)
+    # Choose folders based on the `force` condition
+    folders_to_process = terminal_folders if force else [folder for folder in terminal_folders if folder not in have_mp4s]
+
+    # Print the number of new folders to process
+    print(f"We have {len(folders_to_process)} new folders to process")
+
+    for folder in folders_to_process:
+        process_video (folder, extension)
 
 
 if __name__ == '__main__':
 
     parser = OptionParser()
     parser.add_option("-p", "--path", dest="path", default="/ethoscope_data/videos", help="The root path containing the videos to process")
     parser.add_option("-l", "--list", dest="list", default=False, help="Returns a list of folders containing mp4 files", action="store_true")
+    parser.add_option("-e", "--extension", dest="extension", default="h264", help="The extension of the video file chunks generated by the ethoscope")
+    parser.add_option("--force", dest="force", default=False, help="Force recreating videos even when MP4s are present", action="store_true")
 
     (options, args) = parser.parse_args()
     option_dict = vars(options)
@@ -178,4 +186,8 @@ def crawl (root_path):
         print ("Found %s folders with mp4 files" % len(l))
         os.sys.exit()
 
-    crawl( option_dict['path'] )
+    crawl(
+           option_dict['path'],
+           extension=option_dict["extension"],
+           force=option_dict["force"] 
+         )