Commit 98770454 authored by Archit Tamarapu's avatar Archit Tamarapu
Browse files

[fix] don't skip EXT format

parent 6641fc0e

scripts/config/ci_loudness_linux.json

+5 −5
Original line number Diff line number Diff line
@@ -10,11 +10,11 @@ 
        "SBA": "/usr/local/testv/pinknoise/HOA3.wav",

        "MASA1TC": "/usr/local/testv/pinknoise/MASA1DIR1.wav",

        "MASA2TC": "/usr/local/testv/pinknoise/MASA2DIR2.wav",

        "5_1": "/usr/local/testv/pinknoise/5_1.wav",

        "5_1_2": "/usr/local/testv/pinknoise/5_1_2.wav",

        "5_1_4": "/usr/local/testv/pinknoise/5_1_4.wav",

        "7_1": "/usr/local/testv/pinknoise/7_1.wav",

        "7_1_4": "/usr/local/testv/pinknoise/7_1_4.wav",

        "5_1": "/usr/local/testv/pinknoise/MC_5_1.wav",

        "5_1_2": "/usr/local/testv/pinknoise/MC_5_1_2.wav",

        "5_1_4": "/usr/local/testv/pinknoise/MC_5_1_4.wav",

        "7_1": "/usr/local/testv/pinknoise/MC_7_1.wav",

        "7_1_4": "/usr/local/testv/pinknoise/MC_7_1_4.wav",

        "ISM1": "/usr/local/testv/pinknoise/ISM1.wav",

        "ISM2": "/usr/local/testv/pinknoise/ISM2.wav",

        "ISM3": "/usr/local/testv/pinknoise/ISM3.wav",

scripts/get_loudness_data.py

+11 −3
Original line number Diff line number Diff line
@@ -93,9 +93,17 @@ def process_output_file(outfile, infile, format, input_loudness, input_loudness_ 
        )



        if outformat == "EXT":

            return None

            outformat = format



        metadata = []



        # runIvasCodec metadata outputs are .wav.met and .wav.0.csv

        if "MASA" in outformat:

            metadata.append(outfile.with_suffix(".wav.met"))

        if "ISM" in outformat:

            metadata.append(list(outfile.parent.glob(f"{outfile.name}*.csv")))



        output_audio = fromfile(outformat.upper(), outfile)

        output_audio = fromfile(outformat.upper(), outfile, in_meta=metadata)

        output_loudness, scale_factor, output_loudness_format = get_loudness(

            output_audio

        )

scripts/parse_loudness_data.py

+17 −4
Original line number Diff line number Diff line
@@ -257,17 +257,30 @@ def plot_loudness_by_bandwidth(df, in_fmt, out_fmt, out_dir): 
    plt.close()





out_dir = Path(__file__).parent.parent.joinpath("plots")

out_dir.mkdir(parents=True, exist_ok=True)



df = pd.read_csv(LOUDNESS_DATA_FILENAME)

df = df[df["bitrate"].isin(VALID_BITRATES_IVAS)]



# create directory tree for easier navigation

plots_dir = Path(__file__).parent.parent.joinpath("plots")

plots_dir.mkdir(parents=True, exist_ok=True)

passthru = plots_dir.joinpath("passthrough")



for in_fmt in df["format"].unique():

    print(f"Processing {in_fmt}: ")



    out_fmts = df[df["format"] == in_fmt]["outformat"].unique()



    print(f"Processing {in_fmt}: ")

    progressbar_update(0, len(out_fmts), width=50)



    for idx, out_fmt in enumerate(out_fmts):

        # create dir for out format

        out_dir = (

            passthru

            if in_fmt.casefold() == out_fmt.casefold()

            else plots_dir.joinpath(out_fmt.upper())

        )

        out_dir.mkdir(exist_ok=True)



        plot_loudness_by_bandwidth(df, in_fmt, out_fmt, out_dir)

        progressbar_update(idx + 1, len(out_fmts), width=50)

    print()