File size: 4,313 Bytes
1a07caf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
from shutil import which
from facefusion import ffmpeg_builder
from facefusion.ffmpeg_builder import chain, run, select_frame_range, set_audio_quality, set_audio_sample_size, set_stream_mode, set_video_quality
def test_run() -> None:
assert run([]) == [ which('ffmpeg'), '-loglevel', 'error' ]
def test_chain() -> None:
assert chain(ffmpeg_builder.set_progress()) == [ '-progress' ]
def test_set_stream_mode() -> None:
assert set_stream_mode('udp') == [ '-f', 'mpegts' ]
assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ]
def test_select_frame_range() -> None:
assert select_frame_range(0, None, 30) == [ '-vf', 'trim=start_frame=0,fps=30' ]
assert select_frame_range(None, 100, 30) == [ '-vf', 'trim=end_frame=100,fps=30' ]
assert select_frame_range(0, 100, 30) == [ '-vf', 'trim=start_frame=0:end_frame=100,fps=30' ]
assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ]
def test_set_audio_sample_size() -> None:
assert set_audio_sample_size(16) == [ '-f', 's16le' ]
assert set_audio_sample_size(32) == [ '-f', 's32le' ]
def test_set_audio_quality() -> None:
assert set_audio_quality('aac', 0) == [ '-q:a', '0.1' ]
assert set_audio_quality('aac', 50) == [ '-q:a', '1.0' ]
assert set_audio_quality('aac', 100) == [ '-q:a', '2.0' ]
assert set_audio_quality('libmp3lame', 0) == [ '-q:a', '9' ]
assert set_audio_quality('libmp3lame', 50) == [ '-q:a', '4' ]
assert set_audio_quality('libmp3lame', 100) == [ '-q:a', '0' ]
assert set_audio_quality('libopus', 0) == [ '-b:a', '64k' ]
assert set_audio_quality('libopus', 50) == [ '-b:a', '192k' ]
assert set_audio_quality('libopus', 100) == [ '-b:a', '320k' ]
assert set_audio_quality('libvorbis', 0) == [ '-q:a', '-1.0' ]
assert set_audio_quality('libvorbis', 50) == [ '-q:a', '4.5' ]
assert set_audio_quality('libvorbis', 100) == [ '-q:a', '10.0' ]
assert set_audio_quality('flac', 0) == []
assert set_audio_quality('flac', 50) == []
assert set_audio_quality('flac', 100) == []
def test_set_video_quality() -> None:
assert set_video_quality('libx264', 0) == [ '-crf', '51' ]
assert set_video_quality('libx264', 50) == [ '-crf', '26' ]
assert set_video_quality('libx264', 100) == [ '-crf', '0' ]
assert set_video_quality('libx265', 0) == [ '-crf', '51' ]
assert set_video_quality('libx265', 50) == [ '-crf', '26' ]
assert set_video_quality('libx265', 100) == [ '-crf', '0' ]
assert set_video_quality('libvpx-vp9', 0) == [ '-crf', '63' ]
assert set_video_quality('libvpx-vp9', 50) == [ '-crf', '32' ]
assert set_video_quality('libvpx-vp9', 100) == [ '-crf', '0' ]
assert set_video_quality('h264_nvenc', 0) == [ '-cq' , '51' ]
assert set_video_quality('h264_nvenc', 50) == [ '-cq' , '26' ]
assert set_video_quality('h264_nvenc', 100) == [ '-cq' , '0' ]
assert set_video_quality('hevc_nvenc', 0) == [ '-cq' , '51' ]
assert set_video_quality('hevc_nvenc', 50) == [ '-cq' , '26' ]
assert set_video_quality('hevc_nvenc', 100) == [ '-cq' , '0' ]
assert set_video_quality('h264_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
assert set_video_quality('h264_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
assert set_video_quality('h264_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
assert set_video_quality('hevc_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
assert set_video_quality('hevc_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
assert set_video_quality('hevc_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
assert set_video_quality('h264_qsv', 0) == [ '-qp', '51' ]
assert set_video_quality('h264_qsv', 50) == [ '-qp', '26' ]
assert set_video_quality('h264_qsv', 100) == [ '-qp', '0' ]
assert set_video_quality('hevc_qsv', 0) == [ '-qp', '51' ]
assert set_video_quality('hevc_qsv', 50) == [ '-qp', '26' ]
assert set_video_quality('hevc_qsv', 100) == [ '-qp', '0' ]
assert set_video_quality('h264_videotoolbox', 0) == [ '-b:v', '1024k' ]
assert set_video_quality('h264_videotoolbox', 50) == [ '-b:v', '25768k' ]
assert set_video_quality('h264_videotoolbox', 100) == [ '-b:v', '50512k' ]
assert set_video_quality('hevc_videotoolbox', 0) == [ '-b:v', '1024k' ]
assert set_video_quality('hevc_videotoolbox', 50) == [ '-b:v', '25768k' ]
assert set_video_quality('hevc_videotoolbox', 100) == [ '-b:v', '50512k' ]
|