mirror of
https://github.com/librempeg/librempeg
synced 2024-11-22 00:51:37 +00:00
74 lines
2.2 KiB
Python
Executable File
74 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import logging
|
|
import shlex
|
|
import subprocess
|
|
|
|
HELP = '''
|
|
Normalize audio input.
|
|
|
|
The command uses ffprobe to analyze an input file with the ebur128
|
|
filter, and finally run ffmpeg to normalize the input depending on the
|
|
computed adjustment.
|
|
|
|
ffmpeg encoding arguments can be passed through the extra arguments
|
|
after options, for example as in:
|
|
normalize.py --input input.mp3 --output output.mp3 -- -loglevel debug -y
|
|
'''
|
|
|
|
logging.basicConfig(format='normalize|%(levelname)s> %(message)s', level=logging.INFO)
|
|
log = logging.getLogger()
|
|
|
|
|
|
class Formatter(
|
|
argparse.ArgumentDefaultsHelpFormatter, argparse.RawDescriptionHelpFormatter
|
|
):
|
|
pass
|
|
|
|
|
|
def normalize():
|
|
parser = argparse.ArgumentParser(description=HELP, formatter_class=Formatter)
|
|
parser.add_argument('--input', '-i', required=True, help='specify input file')
|
|
parser.add_argument('--output', '-o', required=True, help='specify output file')
|
|
parser.add_argument('--dry-run', '-n', help='simulate commands', action='store_true')
|
|
parser.add_argument('encode_arguments', nargs='*', help='specify encode options used for the actual encoding')
|
|
|
|
args = parser.parse_args()
|
|
|
|
analysis_cmd = [
|
|
'ffprobe', '-v', 'error', '-of', 'compact=p=0:nk=1',
|
|
'-show_entries', 'frame_tags=lavfi.r128.I', '-f', 'lavfi',
|
|
f"amovie='{args.input}',ebur128=metadata=1"
|
|
]
|
|
|
|
def _run_command(cmd, dry_run=False):
|
|
log.info(f"Running command:\n$ {shlex.join(cmd)}")
|
|
if not dry_run:
|
|
result = subprocess.run(cmd, check=True, stdout=subprocess.PIPE)
|
|
return result
|
|
|
|
result = _run_command(analysis_cmd)
|
|
|
|
loudness = ref = -23
|
|
for line in result.stdout.splitlines():
|
|
sline = line.rstrip()
|
|
if sline:
|
|
loudness = sline
|
|
|
|
adjust = ref - float(loudness)
|
|
if abs(adjust) < 0.0001:
|
|
logging.info(f"No normalization needed for '{args.input}'")
|
|
return
|
|
|
|
logging.info(f"Adjusting '{args.input}' by {adjust:.2f}dB...")
|
|
normalize_cmd = [
|
|
'ffmpeg', '-i', args.input, '-af', f'volume={adjust:.2f}dB'
|
|
] + args.encode_arguments + [args.output]
|
|
|
|
_run_command(normalize_cmd, args.dry_run)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
normalize()
|