pat-e/encoding-scripts-old
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Return to working version

Browse Source
This commit is contained in:
pat-e
2025-11-06 13:10:16 +01:00
parent c12a85ceec
commit 500047f6b1
1 changed files with 6 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
av1_opus_encoder.py
View File

@@ -71,31 +71,14 @@ def convert_audio_track(index, ch, lang, audio_temp_dir, source_file, should_dow
# First pass: Analyze the audio to get loudnorm stats # First pass: Analyze the audio to get loudnorm stats
# The stats are printed to stderr, so we must use subprocess.run directly to capture it. # The stats are printed to stderr, so we must use subprocess.run directly to capture it.
print(" - Pass 1: Analyzing...") print(" - Pass 1: Analyzing...")
result = subprocess.run(
cmd = [ ["ffmpeg", "-v", "info", "-i", str(temp_extracted), "-af", "loudnorm=I=-18:LRA=7:tp=-1:print_format=json", "-f", "null", "-"],
"ffmpeg", "-v", "error", "-stats", "-i", str(temp_extracted), capture_output=True, text=True, check=True)
"-af", "loudnorm=I=-18:LRA=7:tp=-1:print_format=json", "-f", "null", "-"
]
# Use Popen to capture stderr in real-time and print it, while also buffering for JSON parsing.
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, encoding='utf-8')
# Read stderr and print progress updates.
stderr_output = ""
for line in iter(process.stderr.readline, ''):
stderr_output += line # Buffer the full output for parsing
# Write the line, clearing the rest of the line with an ANSI escape code,
# and end with a carriage return to reset the line position.
sys.stdout.write(line.strip() + '\x1B[K\r')
sys.stdout.flush() # Ensure it's written immediately
process.wait() # Wait for the process to terminate
if process.returncode != 0:
raise subprocess.CalledProcessError(process.returncode, cmd, output=process.stdout.read(), stderr="".join(stderr_buffer))
# Find the start of the JSON block in stderr and parse it. # Find the start of the JSON block in stderr and parse it.
# This is more robust than slicing the last N lines. # This is more robust than slicing the last N lines.
# We find the start and end of the JSON block to avoid parsing extra data. # We find the start and end of the JSON block to avoid parsing extra data.
stderr_output = result.stderr
json_start_index = stderr_output.find('{') json_start_index = stderr_output.find('{')
if json_start_index == -1: if json_start_index == -1:
raise ValueError("Could not find start of JSON block in ffmpeg output for loudnorm analysis.") raise ValueError("Could not find start of JSON block in ffmpeg output for loudnorm analysis.")
@@ -113,9 +96,8 @@ def convert_audio_track(index, ch, lang, audio_temp_dir, source_file, should_dow
stats = json.loads(stderr_output[json_start_index:json_end_index]) stats = json.loads(stderr_output[json_start_index:json_end_index])
# Second pass: Apply the normalization. A final newline is needed to clear the progress line. # Second pass: Apply the normalization using the stats from the first pass
# A print() is needed to move to the next line after the progress bar. print(" - Pass 2: Applying normalization...")
print("\n - Pass 2: Applying normalization...")
run_cmd([ run_cmd([
"ffmpeg", "-v", "quiet", "-stats", "-y", "-i", str(temp_extracted), "-af", "ffmpeg", "-v", "quiet", "-stats", "-y", "-i", str(temp_extracted), "-af",
f"loudnorm=I=-18:LRA=7:tp=-1:measured_i={stats['input_i']}:measured_lra={stats['input_lra']}:measured_tp={stats['input_tp']}:measured_thresh={stats['input_thresh']}:offset={stats['target_offset']}", f"loudnorm=I=-18:LRA=7:tp=-1:measured_i={stats['input_i']}:measured_lra={stats['input_lra']}:measured_tp={stats['input_tp']}:measured_thresh={stats['input_thresh']}:offset={stats['target_offset']}",