nixfiles/musicutil/transcode_presets.py

from functools import reduce


def add_to_arr(
        arr: list[str],
        items: list[str]) -> list[str]:
    for item in items:
        arr.append(item)


# does not include copy
transcode_presets = {}

transcode_presets["mp3"] = []

# mp3 v0 -> v9
add_to_arr(transcode_presets["mp3"], [
    f"mp3-v{quality}" for quality in range(0, 10)
])
# mp3 bitrates
mp3_bitrates = [
    8,
    16,
    24,
    32,
    40,
    48,
    64,
    80,
    96,
    112,
    128,
    160,
    192,
    224,
    256,
    320]
add_to_arr(transcode_presets["mp3"], [
    f"mp3-{bitrate}k" for bitrate in mp3_bitrates
])

transcode_presets["opus"] = []

opus_bitrates = ["16", "24", "32", "64", "96", "128", "256"]
add_to_arr(transcode_presets["opus"], [
    f"opus-{bitrate}k" for bitrate in opus_bitrates
])

transcode_presets["vorbis"] = []

add_to_arr(transcode_presets["vorbis"], [
    f"vorbis-q{quality}" for quality in range(-2, 11)
])

transcode_presets["speex"] = []

add_to_arr(transcode_presets["speex"], [
    f"speex-q{quality}" for quality in range(0, 11)
])

transcode_presets["g726"] = []

g726_bitrates = ["16", "24", "32", "40"]
add_to_arr(transcode_presets["g726"], [
    f"g726-{bitrate}k" for bitrate in g726_bitrates
])

# Extra Default Mappings
preset_transcode_presets = {}

mp3_presets = {
    "mp3-low": "mp3-v4",
    "mp3-medium": "mp3-v2",
    "mp3-high": "mp3-v0",
}

preset_transcode_presets = preset_transcode_presets | mp3_presets
add_to_arr(transcode_presets["opus"], mp3_presets.keys())

opus_presets = {
    "opus-low": "opus-32k",
    "opus-medium": "opus-64k",
    "opus-high": "opus-96k",
    "opus-higher": "opus-128k",
    "opus-extreme": "opus-256k"
}

preset_transcode_presets = preset_transcode_presets | opus_presets
add_to_arr(transcode_presets["opus"], opus_presets.keys())

transcode_presets["mp3"].sort()
transcode_presets["opus"].sort()
transcode_presets["vorbis"].sort()
transcode_presets["speex"].sort()
transcode_presets["g726"].sort()


def print_transcode_presets():
    for category in transcode_presets.keys():
        print(f"Category {category}:")
        for preset in transcode_presets[category]:
            print(f"- {preset}")


transcode_presets_list = reduce(
    lambda a, b: a + b, transcode_presets.values())