79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import os
|
|
import subprocess
|
|
import time
|
|
from . import res
|
|
|
|
DEFAULT_PARAMS_AV1 = "tune=0"
|
|
|
|
PRESETS_AV1 = [str(i) for i in range(6, 3, -1)]
|
|
CRFS_AV1 = [str(i) for i in range(30, 9, -1)]
|
|
|
|
def get_encoding_command_av1(f, preset, crf):
|
|
filenameSample = os.path.splitext(os.path.basename(f))[0]
|
|
filenameEncode = ".".join(
|
|
[
|
|
filenameSample,
|
|
'AV1',
|
|
'10Bit',
|
|
'Preset',
|
|
f'{preset}',
|
|
'CRF',
|
|
f'{crf}',
|
|
'mkv'
|
|
]
|
|
)
|
|
filepathEncode = os.path.join(res.get_path_data_encodes_av1(), filenameEncode)
|
|
|
|
cmd_parts = [
|
|
'ffmpeg',
|
|
f'-i "{f}"',
|
|
'-an',
|
|
'-c:v libsvtav1',
|
|
f'-preset {preset}',
|
|
f'-crf {crf}',
|
|
f'-svtav1-params "{DEFAULT_PARAMS_AV1}"',
|
|
f'-pix_fmt yuv420p10le',
|
|
'-g 245',
|
|
'-y',
|
|
f'"{filepathEncode}"'
|
|
]
|
|
|
|
return " ".join(cmd_parts)
|
|
|
|
def start_workflow():
|
|
res.bootstrap_folder_structure()
|
|
|
|
path_results_encoding_time_av1_json = os.path.join(
|
|
res.get_path_results_encoding_time(),
|
|
res.get_filename_results_encoding_time_av1()
|
|
)
|
|
results_encoding_time_av1 = res.read_dict_from_json_file(
|
|
path_results_encoding_time_av1_json
|
|
)
|
|
|
|
for preset in PRESETS_AV1:
|
|
if not preset in results_encoding_time_av1.keys():
|
|
results_encoding_time_av1[preset] = {}
|
|
for crf in CRFS_AV1:
|
|
if not crf in results_encoding_time_av1[preset].keys():
|
|
results_encoding_time_av1[preset][crf] = {}
|
|
for f in res.get_all_sample_files():
|
|
filename = os.path.splitext(os.path.basename(f))[0]
|
|
if filename in results_encoding_time_av1[preset][crf].keys():
|
|
continue
|
|
|
|
time_start = time.time()
|
|
subprocess.run(
|
|
get_encoding_command_av1(f, preset, crf),
|
|
shell=True,
|
|
check=True
|
|
)
|
|
time_encoding = round(time.time() - time_start)
|
|
|
|
results_encoding_time_av1[preset][crf][filename] = time_encoding
|
|
|
|
res.write_dict_to_json_file(
|
|
path_results_encoding_time_av1_json,
|
|
results_encoding_time_av1
|
|
)
|
|
|