#!/usr/bin/python3

import sys
import os

def _get_arg_value(args, key_str):
    for arg in sys.argv:
        parts = arg.split(":")
        if len(parts) > 1:
            if parts[0] == key_str:
                return parts[1]
    
    return None

modules_path = os.path.dirname(os.path.abspath(sys.argv[0])).rstrip("/launch")

sys.path.insert(0, modules_path)
import processutils
processutils.update_sys_path(modules_path)

try:
    import stabilizeheadless
    root_dir = modules_path.split("/")[1]

    session_id = _get_arg_value(sys.argv, "session_id")
    parent_folder = _get_arg_value(sys.argv, "parent_folder")
    write_file = _get_arg_value(sys.argv, "write_file")
    clip_path = _get_arg_value(sys.argv, "clip_path")
    accuracy = _get_arg_value(sys.argv, "accuracy")
    shakiness = _get_arg_value(sys.argv, "shakiness")
    profile_desc_under_score = _get_arg_value(sys.argv, "profile_desc")
    profile_desc = profile_desc_under_score.replace("_", " ") # We need to put underscores in profile names to get them here in one piece.
                                                              # Now we take underscores out to get correct MLT profile names.
except Exception as err:
    print ("Failed to import statbilizeheadless")
    print ("ERROR:", err)
    print ("Installation was assumed to be at:", modules_path)
    sys.exit(1)

stabilizeheadless.main(modules_path, session_id, parent_folder, profile_desc, write_file, clip_path, accuracy, shakiness)




