From 1eb0c49222334c1d1cbf58874a5af57106a334e2 Mon Sep 17 00:00:00 2001
From: Georgy Evtushenko <evtushenko.georgy@gmail.com>
Date: Thu, 22 Dec 2022 17:12:22 +0400
Subject: [PATCH] Adopt histogram script to new format

---
 scripts/nvbench_histogram.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)
 mode change 100644 => 100755 scripts/nvbench_histogram.py

diff --git a/scripts/nvbench_histogram.py b/scripts/nvbench_histogram.py
old mode 100644
new mode 100755
index 1df17cc..5c37d29
--- a/scripts/nvbench_histogram.py
+++ b/scripts/nvbench_histogram.py
@@ -38,6 +38,20 @@ def parse_files():
     return filenames
 
 
+def extract_filename(summary):
+    summary_data = summary["data"]
+    value_data = next(filter(lambda v: v["name"] == "filename", summary_data))
+    assert(value_data["type"] == "string")
+    return value_data["value"]
+
+
+def extract_size(summary):
+    summary_data = summary["data"]
+    value_data = next(filter(lambda v: v["name"] == "size", summary_data))
+    assert(value_data["type"] == "int64")
+    return int(value_data["value"])
+
+
 def parse_samples_meta(filename, state):
     summaries = state["summaries"]
     if not summaries:
@@ -49,13 +63,13 @@ def parse_samples_meta(filename, state):
     if not summary:
         return None, None
 
-    sample_filename = summary["filename"]["value"]
+    sample_filename = extract_filename(summary)
 
     # If not absolute, the path is relative to the associated .json file:
     if not os.path.isabs(sample_filename):
         sample_filename = os.path.join(os.path.dirname(filename), sample_filename)
 
-    sample_count = int(summary["size"]["value"])
+    sample_count = extract_size(summary)
     return sample_count, sample_filename