Ratio plot

benchmarks/scripts/analyze.py

@@ -359,6 +359,37 @@ def displot(data, ax):
         hd_displot(data[variant], variant, ax)
 
 
+def variant_ratio(data, variant, ax):
+    if variant not in color_map:
+        color_map[variant] = next(color_cycle)
+    color = color_map[variant]
+
+    variant_samples = data[variant]
+    base_samples = data['base']
+
+    variant_widths, variant_heights = qrde_hd(variant_samples)
+    base_widths, base_heights = qrde_hd(base_samples)
+
+    quantiles = []
+    ratios = []
+
+    for i in range(1, len(variant_heights) - 1):
+        variant_height = variant_heights[i]
+        base_height = base_heights[i]
+        quantiles.append(i * precision)
+        ratios.append(variant_height / base_height)
+
+    ax.plot(quantiles, ratios, label=variant, color=color)
+    ax.axhline(1, color='red', alpha=0.7)
+    ax.legend()
+
+
+def ratio(data, ax):
+    for variant in data:
+        if variant != 'base':
+            variant_ratio(data, variant, ax)
+
+
 def case_variants(pattern, algname, ct_point_name, case_df):
     title = "{}[{}]:".format(algname, ct_point_name)
     df = case_df[case_df['variant'].str.contains(pattern, regex=True)].reset_index(drop=True)
@@ -432,7 +463,8 @@ def extract_horizontal_space(df):
                 data[variant_name] = horizontal_df[horizontal_df['variant'] == variant_name].iloc[0]['samples']
 
             # sns.histplot(data=data, ax=ax, kde=True)
-            displot(data, ax)
+            # displot(data, ax)
+            ratio(data, ax)
 
             if len(horizontal_axes) > 0:
                 ax=axes[vertical_id, horizontal_id]