#!/usr/bin/env python3 """ Command Line Arguments for tools """ from lib.cli import FaceSwapArgs from lib.cli import ContextFullPaths, DirFullPaths, FileFullPaths, SaveFileFullPaths from lib.utils import _image_extensions class EffmpegArgs(FaceSwapArgs): """ Class to parse the command line arguments for EFFMPEG tool """ @staticmethod def __parse_transpose(value): index = 0 opts = ["(0, 90CounterClockwise&VerticalFlip)", "(1, 90Clockwise)", "(2, 90CounterClockwise)", "(3, 90Clockwise&VerticalFlip)"] if len(value) == 1: index = int(value) else: for i in range(5): if value in opts[i]: index = i break return opts[index] def get_argument_list(self): argument_list = list() argument_list.append({"opts": ('-a', '--action'), "dest": "action", "choices": ("extract", "gen-vid", "get-fps", "get-info", "mux-audio", "rescale", "rotate", "slice"), "default": "extract", "help": "Choose which action you want ffmpeg " "ffmpeg to do.\n" "'slice' cuts a portion of the video " "into a separate video file.\n" "'get-fps' returns the chosen video's " "fps."}) argument_list.append({"opts": ('-i', '--input'), "action": ContextFullPaths, "dest": "input", "default": "input", "help": "Input file.", "required": True, "action_option": "-a", "filetypes": "video"}) argument_list.append({"opts": ('-o', '--output'), "action": ContextFullPaths, "dest": "output", "default": "", "help": "Output file. If no output is " "specified then: if the output is " "meant to be a video then a video " "called 'out.mkv' will be created in " "the input directory; if the output is " "meant to be a directory then a " "directory called 'out' will be " "created inside the input " "directory.\n" "Note: the chosen output file " "extension will determine the file " "encoding.", "action_option": "-a", "filetypes": "video"}) argument_list.append({"opts": ('-r', '--reference-video'), "action": FileFullPaths, "dest": "ref_vid", "default": None, "help": "Path to reference video if 'input' " "was not a video.", "filetypes": "video"}) argument_list.append({"opts": ('-fps', '--fps'), "type": str, "dest": "fps", "default": "-1.0", "help": "Provide video fps. Can be an integer, " "float or fraction. Negative values " "will make the program try to get the " "fps from the input or reference " "videos."}) argument_list.append({"opts": ("-ef", "--extract-filetype"), "choices": _image_extensions, "dest": "extract_ext", "default": ".png", "help": "Image format that extracted images " "should be saved as. '.bmp' will offer " "the fastest extraction speed, but " "will take the most storage space. " "'.png' will be slower but will take " "less storage."}) argument_list.append({"opts": ('-s', '--start'), "type": str, "dest": "start", "default": "00:00:00", "help": "Enter the start time from which an " "action is to be applied.\n" "Default: 00:00:00, in HH:MM:SS " "format. You can also enter the time " "with or without the colons, e.g. " "00:0000 or 026010."}) argument_list.append({"opts": ('-e', '--end'), "type": str, "dest": "end", "default": "00:00:00", "help": "Enter the end time to which an action " "is to be applied. If both an end time " "and duration are set, then the end " "time will be used and the duration " "will be ignored.\n" "Default: 00:00:00, in HH:MM:SS."}) argument_list.append({"opts": ('-d', '--duration'), "type": str, "dest": "duration", "default": "00:00:00", "help": "Enter the duration of the chosen " "action, for example if you enter " "00:00:10 for slice, then the first 10 " "seconds after and including the start " "time will be cut out into a new " "video.\n" "Default: 00:00:00, in HH:MM:SS " "format. You can also enter the time " "with or without the colons, e.g. " "00:0000 or 026010."}) argument_list.append({"opts": ('-m', '--mux-audio'), "action": "store_true", "dest": "mux_audio", "default": False, "help": "Mux the audio from the reference " "video into the input video. This " "option is only used for the 'gen-vid' " "action. 'mux-audio' action has this " "turned on implicitly."}) argument_list.append({"opts": ('-tr', '--transpose'), "choices": ("(0, 90CounterClockwise&VerticalFlip)", "(1, 90Clockwise)", "(2, 90CounterClockwise)", "(3, 90Clockwise&VerticalFlip)"), "type": lambda v: self.__parse_transpose(v), "dest": "transpose", "default": None, "help": "Transpose the video. If transpose is " "set, then degrees will be ignored. For " "cli you can enter either the number " "or the long command name, " "e.g. to use (1, 90Clockwise) " "-tr 1 or -tr 90Clockwise"}) argument_list.append({"opts": ('-de', '--degrees'), "type": str, "dest": "degrees", "default": None, "help": "Rotate the video clockwise by the " "given number of degrees."}) argument_list.append({"opts": ('-sc', '--scale'), "type": str, "dest": "scale", "default": "1920x1080", "help": "Set the new resolution scale if the " "chosen action is 'rescale'."}) argument_list.append({"opts": ('-pr', '--preview'), "action": "store_true", "dest": "preview", "default": False, "help": "Uses ffplay to preview the effects of " "actions that have a video output. " "Currently preview does not work when " "muxing audio."}) argument_list.append({"opts": ('-q', '--quiet'), "action": "store_true", "dest": "quiet", "default": False, "help": "Reduces output verbosity so that only " "serious errors are printed. If both " "quiet and verbose are set, verbose " "will override quiet."}) argument_list.append({"opts": ('-v', '--verbose'), "action": "store_true", "dest": "verbose", "default": False, "help": "Increases output verbosity. If both " "quiet and verbose are set, verbose " "will override quiet."}) return argument_list class SortArgs(FaceSwapArgs): """ Class to parse the command line arguments for sort tool """ @staticmethod def get_argument_list(): """ Put the arguments in a list so that they are accessible from both argparse and gui """ argument_list = list() argument_list.append({"opts": ('-i', '--input'), "action": DirFullPaths, "dest": "input_dir", "default": "input_dir", "help": "Input directory of aligned faces.", "required": True}) argument_list.append({"opts": ('-o', '--output'), "action": DirFullPaths, "dest": "output_dir", "default": "_output_dir", "help": "Output directory for sorted aligned " "faces."}) argument_list.append({"opts": ('-fp', '--final-process'), "type": str, "choices": ("folders", "rename"), "dest": 'final_process', "default": "rename", "help": "'folders': files are sorted using the " "-s/--sort-by method, then they are " "organized into folders using the " "-g/--group-by grouping method. " "'rename': files are sorted using the " "-s/--sort-by then they are renamed. " "Default: rename"}) argument_list.append({"opts": ('-k', '--keep'), "action": 'store_true', "dest": 'keep_original', "default": False, "help": "Keeps the original files in the input " "directory. Be careful when using this " "with rename grouping and no specified " "output directory as this would keep " "the original and renamed files in the " "same directory."}) argument_list.append({"opts": ('-s', '--sort-by'), "type": str, "choices": ("blur", "face", "face-cnn", "face-cnn-dissim", "face-dissim", "face-yaw", "hist", "hist-dissim"), "dest": 'sort_method', "default": "hist", "help": "Sort by method. " "Choose how images are sorted. " "Default: hist"}) argument_list.append({"opts": ('-g', '--group-by'), "type": str, "choices": ("blur", "face", "face-cnn", "face-yaw", "hist"), "dest": 'group_method', "default": "hist", "help": "Group by method. " "When -fp/--final-processing by " "folders choose the how the images are " "grouped after sorting. " "Default: hist"}) argument_list.append({"opts": ('-t', '--ref_threshold'), "type": float, "dest": 'min_threshold', "default": -1.0, "help": "Float value. " "Minimum threshold to use for grouping " "comparison with 'face' and 'hist' " "methods. The lower the value the more " "discriminating the grouping is. " "Leaving -1.0 will make the program " "set the default value automatically. " "For face 0.6 should be enough, with " "0.5 being very discriminating. " "For face-cnn 7.2 should be enough, " "with 4 being very discriminating. " "For hist 0.3 should be enough, with " "0.2 being very discriminating. " "Be careful setting a value that's too " "low in a directory with many images, " "as this could result in a lot of " "directories being created. " "Defaults: face 0.6, face-cnn 7.2, " "hist 0.3"}) argument_list.append({"opts": ('-b', '--bins'), "type": int, "dest": 'num_bins', "default": 5, "help": "Integer value. " "Number of folders that will be used " "to group by blur and face-yaw. " "For blur folder 0 will be the least " "blurry, while the last folder will be " "the blurriest. " "For face-yaw the number of bins is by " "how much 180 degrees is divided. So " "if you use 18, then each folder will " "be a 10 degree increment. Folder 0 " "will contain faces looking the most " "to the left whereas the last folder " "will contain the faces looking the " "most to the right. " "If the number of images doesn't " "divide evenly into the number of " "bins, the remaining images get put in " "the last bin." "Default value: 5"}) argument_list.append({"opts": ('-l', '--log-changes'), "action": 'store_true', "dest": 'log_changes', "default": False, "help": "Logs file renaming changes if " "grouping by renaming, or it logs the " "file copying/movement if grouping by " "folders. If no log file is specified " "with '--log-file', then a " "'sort_log.json' file will be created " "in the input directory."}) argument_list.append({"opts": ('-lf', '--log-file'), "action": SaveFileFullPaths, "filetypes": "alignments", "dest": 'log_file_path', "default": 'sort_log.json', "help": "Specify a log file to use for saving " "the renaming or grouping information. " "If specified extension isn't 'json' " "or 'yaml', then json will be used as " "the serializer, with the supplied " "filename. " "Default: sort_log.json"}) return argument_list