collect output filename and add a scaling command for convenience (7616daec) · Commits · IVAS Codec Public Collaboration / IVAS Codec

scripts/get_loudness_data.py

+22 −1

Original line number	Diff line number	Diff line
		@@ -6,6 +6,7 @@ import os
		import re
		import sys
		import json
		import platform
		from pathlib import Path
		from concurrent.futures import ProcessPoolExecutor, as_completed

		@@ -93,18 +94,38 @@ def process_output_file(outfile, infile, format, input_loudness, input_loudness_
		output_audio
		)

		# add a scaling command with IVAS_rend for convenience
		rend = "./IVAS_rend"
		if platform.system() == "Windows":
		rend = ".\\IVAS_rend.exe"
		scale_cmd = [
		rend,
		"-i",
		str(outfile),
		"-if",
		format,
		"-of",
		format,
		"-g",
		str(scale_factor),
		"-o",
		str(outfile.with_suffix(f"scaled_{scale_factor:3.2f}_{outfile.suffix}")),
		]

		return {
		"infile": infile.name,
		"format": format,
		"input_loudness": input_loudness,
		"input_loudness_format": input_loudness_format,
		"outformat": outformat,
		"bitrate": bitrate,
		"bandwidth": bandwidth,
		"dtx": dtx,
		"outformat": outformat,
		"outfile": outfile.name,
		"output_loudness": output_loudness,
		"output_loudness_format": output_loudness_format,
		"scale_factor": scale_factor,
		"scale_cmd": scale_cmd,
		}
		except (AssertionError, ValueError) as e:
		print(f"\n⚠️ Skipping {outfile.stem}: {e}", file=sys.stderr)