Merge branch 'main' into scripts-updates-to-basop-ci-branch (eb672b93) · Commits · IVAS Codec Public Collaboration / IVAS Codec

tests/split_rendering/utils.py

+88 −62

Original line number	Original line	Diff line number	Diff line
	@@ -38,6 +38,7 @@ from typing import Tuple

	import numpy as np		import numpy as np
	import pytest		import pytest
			import logging

	from tests.renderer.utils import (		from tests.renderer.utils import (
	check_BE,		check_BE,
	@@ -178,6 +179,7 @@ def run_full_chain_split_rendering(
	get_ssnr=False,		get_ssnr=False,
	get_odg=False,		get_odg=False,
	get_odg_bin=False,		get_odg_bin=False,
			delay_profile: Optional[Path] = None,
	) -> str:		) -> str:
	"""		"""
	Runs the full split rendering chain consisting of		Runs the full split rendering chain consisting of
	@@ -228,63 +230,80 @@ def run_full_chain_split_rendering(
	in_meta_files = None		in_meta_files = None

	# run encoder		# run encoder
	cmd = SPLIT_PRE_COD_CMD[:]		enc_cmd = SPLIT_PRE_COD_CMD[:]

	if test_info.config.option.create_ref:		if test_info.config.option.create_ref:
	cmd[0] += BIN_SUFFIX_MERGETARGET		enc_cmd[0] += BIN_SUFFIX_MERGETARGET
	cmd[0] += binary_suffix		enc_cmd[0] += binary_suffix
	cmd[1] = bitrate		enc_cmd[1] = bitrate
	# if in REF or CUT creation mode use the comparetestv		# if in REF or CUT creation mode use the comparetestv
	if test_info.config.option.create_ref or test_info.config.option.create_cut:		if test_info.config.option.create_ref or test_info.config.option.create_cut:
	in_file = FORMAT_TO_FILE_COMPARETEST[in_fmt]		in_file = FORMAT_TO_FILE_COMPARETEST[in_fmt]
	truncate_signal(in_file, cut_in_file)		truncate_signal(in_file, cut_in_file)

	cmd[3] = str(cut_in_file)		enc_cmd[3] = str(cut_in_file)
	else:		else:
	cmd[3] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])		enc_cmd[3] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])
	cmd[4] = str(ivas_bitstream)		enc_cmd[4] = str(ivas_bitstream)

	if in_meta_files:		if in_meta_files:
	cmd[1:1] = in_meta_files		enc_cmd[1:1] = in_meta_files

	cmd[1:1] = FORMAT_TO_IVAS_COD_FORMAT[in_fmt]		enc_cmd[1:1] = FORMAT_TO_IVAS_COD_FORMAT[in_fmt]

	run_ivas_isar_enc_cmd(cmd)		run_ivas_isar_enc_cmd(enc_cmd)

			if delay_profile:
			rtp_bitstream = ivas_bitstream.with_suffix(".rtpg192")
			netsim_cmd = [
			str(TOOLS_DIR / f"networkSimulator_g192{EXE_SUFFIX}"),
			str(delay_profile),
			str(ivas_bitstream),
			str(rtp_bitstream),
			str(tmp_dir / "tmp.netsimtrace"),
			"1",
			"0",
			]
			run_cmd(netsim_cmd, test_info=test_info)
			ivas_bitstream = rtp_bitstream

	# decode to split-rendering bitstream		# decode to split-rendering bitstream
	cmd = SPLIT_PRE_DEC_CMD[:]		dec_cmd = SPLIT_PRE_DEC_CMD[:]

	if test_info.config.option.create_ref:		if test_info.config.option.create_ref:
	cmd[0] += BIN_SUFFIX_MERGETARGET		dec_cmd[0] += BIN_SUFFIX_MERGETARGET
	cmd[0] += binary_suffix		dec_cmd[0] += binary_suffix
	cmd[2] = str(pre_trajectory)		dec_cmd[2] = str(pre_trajectory)
	cmd[4] = str(render_config)		dec_cmd[4] = str(render_config)
	cmd[6] = str(pre_rend_fr)		dec_cmd[6] = str(pre_rend_fr)
	cmd[7] = renderer_fmt		dec_cmd[7] = renderer_fmt
	cmd[9] = str(ivas_bitstream)		dec_cmd[9] = str(ivas_bitstream)
	cmd[10] = str(split_bitstream)		dec_cmd[10] = str(split_bitstream)

	if renderer_fmt == "BINAURAL_SPLIT_PCM":		if renderer_fmt == "BINAURAL_SPLIT_PCM":
	cmd[5:5] = ["-om", str(split_md_file)]		dec_cmd[5:5] = ["-om", str(split_md_file)]

			if delay_profile:
			dec_cmd[5:5] = ["-voip"]

	run_ivas_isar_dec_cmd(cmd)		run_ivas_isar_dec_cmd(dec_cmd)

	# run split renderer		# run split renderer
	cmd = SPLIT_POST_REND_CMD[:]		post_rend_cmd = SPLIT_POST_REND_CMD[:]

	if test_info.config.option.create_ref:		if test_info.config.option.create_ref:
	cmd[0] += BIN_SUFFIX_MERGETARGET		post_rend_cmd[0] += BIN_SUFFIX_MERGETARGET
	cmd[0] += binary_suffix		post_rend_cmd[0] += binary_suffix
	cmd[4] = str(split_bitstream)		post_rend_cmd[4] = str(split_bitstream)
	cmd[6] = renderer_fmt		post_rend_cmd[6] = renderer_fmt
	cmd[8] = str(out_file)		post_rend_cmd[8] = str(out_file)
	cmd[10] = str(post_trajectory)		post_rend_cmd[10] = str(post_trajectory)
	cmd[12] = post_rend_fr		post_rend_cmd[12] = post_rend_fr

	if renderer_fmt == "BINAURAL_SPLIT_PCM":		if renderer_fmt == "BINAURAL_SPLIT_PCM":
	cmd[7:7] = ["-im", str(split_md_file)]		post_rend_cmd[7:7] = ["-im", str(split_md_file)]

	run_isar_post_rend_cmd(cmd)		run_isar_post_rend_cmd(post_rend_cmd)

	if test_info.config.option.create_cut:		if test_info.config.option.create_cut:
	# CUT creation mode will run a comparison with REF		# CUT creation mode will run a comparison with REF
	@@ -329,6 +348,11 @@ def run_full_chain_split_rendering(
	record_property(k, v)		record_property(k, v)

	if output_differs:		if output_differs:
			logging.error(f"Encoder command line was: {' '.join(enc_cmd)}")
			if delay_profile:
			logging.error(f"Netsim command line was: {' '.join(netsim_cmd)}")
			logging.error(f"Decoder command line was: {' '.join(dec_cmd)}")
			logging.error(f"Post renderer command line was: {' '.join(post_rend_cmd)}")
	pytest.fail(f"Output differs: ({reason})")		pytest.fail(f"Output differs: ({reason})")

	return out_file		return out_file
	@@ -390,52 +414,52 @@ def run_external_split_rendering(
	in_meta_files = None		in_meta_files = None

	# generate split-rendering bitstream		# generate split-rendering bitstream
	cmd = SPLIT_PRE_REND_CMD[:]		split_pre_cmd = SPLIT_PRE_REND_CMD[:]

	if test_info.config.option.create_ref:		if test_info.config.option.create_ref:
	cmd[0] += BIN_SUFFIX_MERGETARGET		split_pre_cmd[0] += BIN_SUFFIX_MERGETARGET
	cmd[0] += binary_suffix		split_pre_cmd[0] += binary_suffix
	cmd[4] = str(render_config)		split_pre_cmd[4] = str(render_config)
	if is_comparetest:		if is_comparetest:
	in_file = FORMAT_TO_FILE_COMPARETEST[in_fmt]		in_file = FORMAT_TO_FILE_COMPARETEST[in_fmt]
	truncate_signal(in_file, cut_in_file)		truncate_signal(in_file, cut_in_file)

	cmd[6] = str(cut_in_file)		split_pre_cmd[6] = str(cut_in_file)
	else:		else:
	cmd[6] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])		split_pre_cmd[6] = str(FORMAT_TO_FILE_SMOKETEST[in_fmt])
	cmd[8] = in_fmt		split_pre_cmd[8] = in_fmt
	cmd[10] = str(split_bitstream)		split_pre_cmd[10] = str(split_bitstream)
	cmd[12] = renderer_fmt		split_pre_cmd[12] = renderer_fmt
	cmd[14] = str(pre_trajectory)		split_pre_cmd[14] = str(pre_trajectory)
	cmd[16] = pre_rend_fr		split_pre_cmd[16] = pre_rend_fr

	if renderer_fmt == "BINAURAL_SPLIT_PCM":		if renderer_fmt == "BINAURAL_SPLIT_PCM":
	cmd[13:13] = ["-om", str(split_md_file)]		split_pre_cmd[13:13] = ["-om", str(split_md_file)]

	if in_meta_files:		if in_meta_files:
	cmd[9:9] = ["-im", *in_meta_files]		split_pre_cmd[9:9] = ["-im", *in_meta_files]

	run_isar_ext_rend_cmd(cmd)		run_isar_ext_rend_cmd(split_pre_cmd)

	# run split renderer		# run split renderer
	cmd = SPLIT_POST_REND_CMD[:]		split_post_cmd = SPLIT_POST_REND_CMD[:]

	if test_info.config.option.create_ref:		if test_info.config.option.create_ref:
	cmd[0] += BIN_SUFFIX_MERGETARGET		split_post_cmd[0] += BIN_SUFFIX_MERGETARGET
	cmd[0] += binary_suffix		split_post_cmd[0] += binary_suffix
	cmd[4] = str(split_bitstream)		split_post_cmd[4] = str(split_bitstream)
	cmd[6] = renderer_fmt		split_post_cmd[6] = renderer_fmt
	cmd[8] = str(out_file)		split_post_cmd[8] = str(out_file)
	cmd[10] = str(post_trajectory)		split_post_cmd[10] = str(post_trajectory)
	cmd[12] = post_rend_fr		split_post_cmd[12] = post_rend_fr

	if renderer_fmt == "BINAURAL_SPLIT_PCM":		if renderer_fmt == "BINAURAL_SPLIT_PCM":
	cmd[7:7] = ["-im", str(split_md_file)]		split_post_cmd[7:7] = ["-im", str(split_md_file)]

	if plc_error_pattern:		if plc_error_pattern:
	cmd[1:1] = ["-prbfi", str(plc_error_pattern)]		split_post_cmd[1:1] = ["-prbfi", str(plc_error_pattern)]

	run_isar_ext_rend_cmd(cmd)		run_isar_ext_rend_cmd(split_post_cmd)

	if test_info.config.option.create_cut:		if test_info.config.option.create_cut:
	# CUT creation mode will run a comparison with REF		# CUT creation mode will run a comparison with REF
	@@ -484,6 +508,8 @@ def run_external_split_rendering(
	record_property(k, v)		record_property(k, v)

	if output_differs:		if output_differs:
			logging.error(f"Split Pre command line was: {' '.join(split_pre_cmd)}")
			logging.error(f"Split Post command line was: {' '.join(split_post_cmd)}")
	pytest.fail(f"Output differs: ({reason})")		pytest.fail(f"Output differs: ({reason})")

	return out_file		return out_file