mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 10:43:27 -04:00
235 lines
9.7 KiB
Python
235 lines
9.7 KiB
Python
#!/usr/bin/env python3
|
|
""" Command Line Arguments for tools """
|
|
import argparse
|
|
import gettext
|
|
|
|
from lib.cli.args import FaceSwapArgs
|
|
from lib.cli.actions import ContextFullPaths, FileFullPaths, Radio
|
|
from lib.utils import IMAGE_EXTENSIONS
|
|
|
|
|
|
# LOCALES
|
|
_LANG = gettext.translation("tools.effmpeg.cli", localedir="locales", fallback=True)
|
|
_ = _LANG.gettext
|
|
|
|
_HELPTEXT = _("This command allows you to easily execute common ffmpeg tasks.")
|
|
|
|
|
|
def __parse_transpose(value: str) -> str:
|
|
""" Parse transpose option
|
|
|
|
Parameters
|
|
----------
|
|
value: str
|
|
The value to parse
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
The option item for the given 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]
|
|
|
|
|
|
class EffmpegArgs(FaceSwapArgs):
|
|
""" Class to parse the command line arguments for EFFMPEG tool """
|
|
|
|
@staticmethod
|
|
def get_info():
|
|
""" Return command information """
|
|
return _("A wrapper for ffmpeg for performing image <> video converting.")
|
|
|
|
@staticmethod
|
|
def get_argument_list():
|
|
argument_list = []
|
|
argument_list.append({
|
|
"opts": ('-a', '--action'),
|
|
"action": Radio,
|
|
"dest": "action",
|
|
"choices": ("extract", "gen-vid", "get-fps", "get-info", "mux-audio", "rescale",
|
|
"rotate", "slice"),
|
|
"default": "extract",
|
|
"help": _("R|Choose which action you want ffmpeg ffmpeg to do."
|
|
"\nL|'extract': turns videos into images "
|
|
"\nL|'gen-vid': turns images into videos "
|
|
"\nL|'get-fps' returns the chosen video's fps."
|
|
"\nL|'get-info' returns information about a video."
|
|
"\nL|'mux-audio' add audio from one video to another."
|
|
"\nL|'rescale' resize video."
|
|
"\nL|'rotate' rotate video."
|
|
"\nL|'slice' cuts a portion of the video into a separate video file.")})
|
|
argument_list.append({
|
|
"opts": ('-i', '--input'),
|
|
"action": ContextFullPaths,
|
|
"dest": "input",
|
|
"default": "input",
|
|
"help": _("Input file."),
|
|
"group": _("data"),
|
|
"required": True,
|
|
"action_option": "-a",
|
|
"filetypes": "video"})
|
|
argument_list.append({
|
|
"opts": ('-o', '--output'),
|
|
"action": ContextFullPaths,
|
|
"group": _("data"),
|
|
"default": "",
|
|
"dest": "output",
|
|
"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. 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",
|
|
"group": _("data"),
|
|
"default": None,
|
|
"help": _("Path to reference video if 'input' was not a video."),
|
|
"filetypes": "video"})
|
|
argument_list.append({
|
|
"opts": ('-R', '--fps'),
|
|
"type": str,
|
|
"dest": "fps",
|
|
"group": _("output"),
|
|
"default": "-1.0",
|
|
"help": _("Provide video fps. Can be an integer, float or fraction. Negative values "
|
|
"will will make the program try to get the fps from the input or reference "
|
|
"videos.")})
|
|
argument_list.append({
|
|
"opts": ("-E", "--extract-filetype"),
|
|
"action": Radio,
|
|
"choices": IMAGE_EXTENSIONS,
|
|
"dest": "extract_ext",
|
|
"group": _("output"),
|
|
"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",
|
|
"group": _("clip"),
|
|
"default": "00:00:00",
|
|
"help": _("Enter the start time from which an action is to be applied. 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",
|
|
"group": _("clip"),
|
|
"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. Default: 00:00:00, in HH:MM:SS.")})
|
|
argument_list.append({
|
|
"opts": ('-d', '--duration'),
|
|
"type": str,
|
|
"dest": "duration",
|
|
"group": _("clip"),
|
|
"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. 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",
|
|
"group": _("output"),
|
|
"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": ('-T', '--transpose'),
|
|
"choices": ("(0, 90CounterClockwise&VerticalFlip)",
|
|
"(1, 90Clockwise)",
|
|
"(2, 90CounterClockwise)",
|
|
"(3, 90Clockwise&VerticalFlip)"),
|
|
"type": lambda v: __parse_transpose(v), # pylint:disable=unnecessary-lambda
|
|
"dest": "transpose",
|
|
"group": _("rotate"),
|
|
"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": ('-D', '--degrees'),
|
|
"type": str,
|
|
"dest": "degrees",
|
|
"default": None,
|
|
"group": _("rotate"),
|
|
"help": _("Rotate the video clockwise by the given number of degrees.")})
|
|
argument_list.append({
|
|
"opts": ('-S', '--scale'),
|
|
"type": str,
|
|
"dest": "scale",
|
|
"group": _("output"),
|
|
"default": "1920x1080",
|
|
"help": _("Set the new resolution scale if the chosen action is 'rescale'.")})
|
|
argument_list.append({
|
|
"opts": ('-q', '--quiet'),
|
|
"action": "store_true",
|
|
"dest": "quiet",
|
|
"group": _("settings"),
|
|
"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",
|
|
"group": _("settings"),
|
|
"default": False,
|
|
"help": _("Increases output verbosity. If both quiet and verbose are set, verbose "
|
|
"will override quiet.")})
|
|
# Deprecated multi-character switches
|
|
argument_list.append({
|
|
"opts": ('-fps', ),
|
|
"type": str,
|
|
"dest": "depr_fps_fps_R",
|
|
"help": argparse.SUPPRESS})
|
|
argument_list.append({
|
|
"opts": ("-ef", ),
|
|
"type": str,
|
|
"choices": IMAGE_EXTENSIONS,
|
|
"dest": "depr_extract_ext_et_E",
|
|
"help": argparse.SUPPRESS})
|
|
argument_list.append({
|
|
"opts": ('-tr', ),
|
|
"choices": ("(0, 90CounterClockwise&VerticalFlip)",
|
|
"(1, 90Clockwise)",
|
|
"(2, 90CounterClockwise)",
|
|
"(3, 90Clockwise&VerticalFlip)"),
|
|
"type": lambda v: __parse_transpose(v), # pylint:disable=unnecessary-lambda
|
|
"dest": "depr_transpose_tr_T",
|
|
"help": argparse.SUPPRESS})
|
|
argument_list.append({
|
|
"opts": ('-de', ),
|
|
"type": str,
|
|
"dest": "depr_degrees_de_D",
|
|
"help": argparse.SUPPRESS})
|
|
argument_list.append({
|
|
"opts": ('-sc', ),
|
|
"type": str,
|
|
"dest": "depr_scale_sc_S",
|
|
"help": argparse.SUPPRESS})
|
|
return argument_list
|