mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
97711e799c
DASH reads the AudioChannelConfiguration scheme (Dolby mask, CICP index incl. immersive layouts, channel count); HLS and ISM counts map 6/7/8 to 5.1/6.1/7.1.
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
import xml.etree.ElementTree as ET
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.manifests.dash import DASH
|
|
|
|
COUNT = "urn:mpeg:dash:23003:3:audio_channel_configuration:2011"
|
|
CICP = "urn:mpeg:mpegB:cicp:ChannelConfiguration"
|
|
DOLBY = "tag:dolby.com,2014:dash:audio_channel_configuration:2011"
|
|
DOLBY_LEGACY = "urn:dolby:dash:audio_channel_configuration:2011"
|
|
|
|
|
|
def config(*descriptors: tuple[str, str], tag: str = "AdaptationSet") -> ET.Element:
|
|
children = "".join(f'<AudioChannelConfiguration schemeIdUri="{s}" value="{v}"/>' for s, v in descriptors)
|
|
return ET.fromstring(f"<{tag}>{children}</{tag}>")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("scheme", "value", "expected"),
|
|
[
|
|
(COUNT, "2", 2.0),
|
|
(COUNT, "6", 5.1),
|
|
(COUNT, "7", 6.1),
|
|
(COUNT, "8", 7.1),
|
|
("urn:dts:dash:audio_channel_configuration:2012", "6", 5.1),
|
|
(CICP, "2", 2.0),
|
|
(CICP, "6", 5.1),
|
|
(CICP, "7", 7.1),
|
|
(CICP, "8", 7.1),
|
|
(CICP, "12", 7.1),
|
|
(CICP, "16", "5.1.4"),
|
|
(CICP, "17", "5.1.6"),
|
|
(CICP, "18", "7.1.6"),
|
|
(CICP, "19", "7.1.4"),
|
|
(CICP, "20", "9.1.4"),
|
|
(DOLBY, "F801", 5.1),
|
|
(DOLBY_LEGACY, "FA01", 7.1),
|
|
# an all-digit mask is still a mask under a Dolby scheme
|
|
(DOLBY, "4000", 1.0),
|
|
],
|
|
)
|
|
def test_a_channel_value_is_decoded_by_its_scheme(scheme: str, value: str, expected: object) -> None:
|
|
assert DASH.get_audio_channels(config((scheme, value)), None) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("scheme", "value"),
|
|
[
|
|
("tag:dolby.com,2015:dash:audio_channel_configuration:2015", "000047"),
|
|
("urn:mpeg:dash:outputChannelPositionList:2012", "L R C LFE"),
|
|
(CICP, "13"),
|
|
(COUNT, "x"),
|
|
(DOLBY, "zz"),
|
|
],
|
|
)
|
|
def test_an_unknown_scheme_or_value_gives_no_channels(scheme: str, value: str) -> None:
|
|
assert DASH.get_audio_channels(config((scheme, value)), None) is None
|
|
|
|
|
|
def test_the_first_decodable_descriptor_wins() -> None:
|
|
element = config(("urn:example:unknown", "9"), (DOLBY, "F801"), (CICP, "2"))
|
|
assert DASH.get_audio_channels(element, None) == 5.1
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("representation", "expected"), [(config((COUNT, "6"), tag="Representation"), 5.1), (None, 2.0)]
|
|
)
|
|
def test_the_representation_overrides_the_adaptation_set(representation: Optional[ET.Element], expected: float) -> None:
|
|
assert DASH.get_audio_channels(config((COUNT, "2")), representation) == expected
|