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

try to get printouts to show up in CI

parent bfa1b0b0

scripts/get_loudness_data.py

+7 −1
Original line number Diff line number Diff line 
#!/usr/bin/env python3

import pandas as pd

import logging

import os

import re

import sys

import time
@@ -14,6 +15,9 @@ from ivas_processing_scripts.utils import progressbar_update, spinner 


logging.basicConfig(level=logging.WARNING)



if os.environ.get("CI") or not sys.stdout.isatty():

    sys.stdout.reconfigure(line_buffering=True)



INPUT_FOLDER_BASE = Path(__file__).parent.joinpath("testv", "pinknoise")

OUTPUT_FOLDER = Path(__file__).parent.parent.joinpath("out/dec")

FORMATS = [
@@ -133,7 +137,9 @@ def main(): 
        # Load and cache input audio once per format

        if format not in input_audio_cache:

            if format.startswith("MASA"):

                input_audio = fromfile(format, infile, in_meta=infile.with_suffix(".met"))

                input_audio = fromfile(

                    format, infile, in_meta=infile.with_suffix(".met")

                )

            else:

                input_audio = fromfile(format, infile)

            input_loudness, _, input_loudness_format = get_loudness(input_audio)

scripts/parse_loudness_data.py

+13 −1
Original line number Diff line number Diff line
@@ -2,8 +2,16 @@ 
import matplotlib.pyplot as plt

import pandas as pd

import numpy as np

import os

import sys

from pathlib import Path



sys.path.append("./ivas-processing-scripts")

from ivas_processing_scripts.utils import progressbar_update



if os.environ.get("CI") or not sys.stdout.isatty():

    sys.stdout.reconfigure(line_buffering=True)



LOUDNESS_DATA_FILENAME = "loudness.csv"
@@ -101,5 +109,9 @@ out_dir.mkdir(parents=True, exist_ok=True) 
df = pd.read_csv(LOUDNESS_DATA_FILENAME)



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

    for out_fmt in df["outformat"].unique():

    print(f"Processing {in_fmt}: ")

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

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

    for idx, out_fmt in enumerate(out_fmts):

        plot_loudness_by_bandwidth(df, in_fmt, out_fmt, out_dir)

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