Feat/audio metadata (#1019)

* audio metadata

* audio metadata

* audio metadata

* audio metadata

* audio metadata

* audio metadata

* audio metadata

tests/test_ffprobe.py

@@ -0,0 +1,62 @@
+import subprocess
+
+import pytest
+
+from facefusion import process_manager
+from facefusion.download import conditional_download
+from facefusion.ffprobe import detect_audio_channel_total, detect_audio_format, detect_audio_frame_total, detect_audio_sample_rate
+from .helper import get_test_example_file, get_test_examples_directory
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	process_manager.start()
+	conditional_download(get_test_examples_directory(),
+	[
+		'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3'
+	])
+	subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), get_test_example_file('source.wav') ])
+
+
+def test_detect_audio_sample_rate() -> None:
+	audio_sample_rate = detect_audio_sample_rate(get_test_example_file('source.mp3'))
+	assert audio_sample_rate == 44100
+
+	audio_sample_rate = detect_audio_sample_rate(get_test_example_file('source.wav'))
+	assert audio_sample_rate == 44100
+
+	audio_sample_rate = detect_audio_sample_rate(get_test_example_file('invalid.mp3'))
+	assert audio_sample_rate is None
+
+
+def test_detect_audio_channel_total() -> None:
+	audio_channel_total = detect_audio_channel_total(get_test_example_file('source.mp3'))
+	assert audio_channel_total == 1
+
+	audio_channel_total = detect_audio_channel_total(get_test_example_file('source.wav'))
+	assert audio_channel_total == 1
+
+	audio_channel_total = detect_audio_channel_total(get_test_example_file('invalid.mp3'))
+	assert audio_channel_total is None
+
+
+def test_detect_audio_frame_total() -> None:
+	audio_frame_total = detect_audio_frame_total(get_test_example_file('source.mp3'))
+	assert audio_frame_total == 167039
+
+	audio_frame_total = detect_audio_frame_total(get_test_example_file('source.wav'))
+	assert audio_frame_total == 167039
+
+	audio_frame_total = detect_audio_frame_total(get_test_example_file('invalid.mp3'))
+	assert audio_frame_total is None
+
+
+def test_detect_audio_format() -> None:
+	audio_format = detect_audio_format(get_test_example_file('source.mp3'))
+	assert audio_format == 'mp3'
+
+	audio_format = detect_audio_format(get_test_example_file('source.wav'))
+	assert audio_format == 'pcm_s16le'
+
+	audio_format = detect_audio_format(get_test_example_file('invalid.mp3'))
+	assert audio_format is None

tests/test_ffprobe_builder.py

@@ -0,0 +1,43 @@
+from shutil import which
+
+from facefusion import ffprobe_builder
+from facefusion.ffprobe_builder import chain, format_to_key_value, format_to_value, run, select_audio_stream, set_input, show_stream_entries
+
+
+def test_run() -> None:
+	assert run([ '-v', 'error' ]) == [ which('ffprobe'), '-loglevel', 'error', '-v', 'error' ]
+
+
+def test_chain() -> None:
+	assert chain(
+		ffprobe_builder.select_audio_stream(0)
+	) == [ '-select_streams', 'a:0' ]
+	assert chain(
+		ffprobe_builder.select_audio_stream(0),
+		ffprobe_builder.show_stream_entries([ 'sample_rate' ]),
+		ffprobe_builder.format_to_value(),
+		ffprobe_builder.set_input('audio.mp3')
+	) == [ '-select_streams', 'a:0', '-show_entries', 'stream=sample_rate', '-of', 'default=noprint_wrappers=1:nokey=1', 'audio.mp3' ]
+
+
+def test_select_audio_stream() -> None:
+	assert select_audio_stream(0) == [ '-select_streams', 'a:0' ]
+
+
+def test_show_entries() -> None:
+	assert show_stream_entries([ 'sample_rate' ]) == [ '-show_entries', 'stream=sample_rate' ]
+	assert show_stream_entries([ 'channels' ]) == [ '-show_entries', 'stream=channels' ]
+	assert show_stream_entries([ 'duration', 'sample_rate' ]) == [ '-show_entries', 'stream=duration,sample_rate' ]
+
+
+def test_format_to_value() -> None:
+	assert format_to_value() == [ '-of', 'default=noprint_wrappers=1:nokey=1' ]
+
+
+def test_format_to_key_value() -> None:
+	assert format_to_key_value() == [ '-of', 'default=noprint_wrappers=1' ]
+
+
+def test_set_input() -> None:
+	assert set_input('input.mp3') == [ 'input.mp3' ]
+	assert set_input('input.wav') == [ 'input.wav' ]