From 835706272f3cf14f5021188ff9468ebeb1dea098 Mon Sep 17 00:00:00 2001 From: Erik Norvell Date: Tue, 17 Sep 2024 17:02:14 +0200 Subject: [PATCH] Add ODG analysis to create_histogram_summary.py, and improve newline handling in figures --- scripts/create_histogram_summary.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/scripts/create_histogram_summary.py b/scripts/create_histogram_summary.py index 6e940c883a..ae5ead4260 100644 --- a/scripts/create_histogram_summary.py +++ b/scripts/create_histogram_summary.py @@ -43,7 +43,7 @@ if __name__ == "__main__": "--measure", type=str, nargs=1, - help="Measure, any of: MLD, DIFF, SSNR, default: MLD", + help="Measure, any of: MLD, DIFF, SSNR, ODG, default: MLD", default=["MLD"], ) parser.add_argument( @@ -67,6 +67,7 @@ if __name__ == "__main__": "MLD": ("MLD", [0, 5, 10, math.inf]), "DIFF": ("MAXIMUM ABS DIFF", [0, 1024, 16384, 32769]), "SSNR": ("MIN_SSNR", [-math.inf, 0, 20, 40, 60, 100]), + "ODG": ("MIN_ODG", [-5, -4, -3, -2, -1, 0]), } (measure_label, limits) = limits_per_measure[measure] @@ -85,7 +86,7 @@ if __name__ == "__main__": # Output CSV file with open(csv_summary, "w") as fp: limits_labels = [ - f"{str(a)} --\n {str(b)}" for (a, b) in zip(limits[0:-1], limits[1:]) + f"{str(a)} -- {str(b)}" for (a, b) in zip(limits[0:-1], limits[1:]) ] + ["None"] # Zero difference is treated as a special category for MLD and MAXIMUM ABS DIFF if measure_label == "MLD" or measure_label == "MAXIMUM ABS DIFF": @@ -124,7 +125,8 @@ if __name__ == "__main__": fp.write(line) # Matplotlib histogram - ax.bar(limits_labels, data, 0.5, label=cat, bottom=bottom) + labels = [x if len(x) < 10 else x.replace('--','--\n') for x in limits_labels] + ax.bar(labels, data, 0.5, label=cat, bottom=bottom) bottom += data # Histogram layout -- GitLab