Working on axes order

benchmarks/scripts/analyze.py

@@ -197,16 +197,33 @@ def coverage(args):
 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)
-    num_rows = len(df)
+    num_records = len(df)
+    rt_axes = get_rt_axes(df)
+
+    vertical_axis_name = rt_axes[0]
+    if 'Elements{io}[pow2]' in rt_axes:
+        vertical_axis_name = 'Elements{io}[pow2]'
+    vertical_axis_values = extract_rt_axes_values(df)[vertical_axis_name] 
+    vertical_axis_ids = {}
+
+    for idx, val in enumerate(vertical_axis_values):
+        vertical_axis_ids[val] = idx
+
+    num_rows = len(vertical_axis_ids)
+    num_cols = num_records // num_rows
+    fig, axes = plt.subplots(nrows=num_rows, ncols=num_cols, gridspec_kw = {'wspace': 0, 'hspace': 0})
 
-    fig, axes = plt.subplots(nrows=num_rows, ncols=1, gridspec_kw = {'wspace': 0, 'hspace': 0})
+    col_id = 0
     for id, row in df.iterrows():
         description = row['variant']
-        for rt_axis in get_rt_axes(df):
+        for rt_axis in rt_axes:
             description += ' / ' + rt_axis + '=' + row[rt_axis]
         data = {description: row['samples'],
                 'base': row['base_samples']}
-        sns.histplot(data, ax=axes[id], kde=True)
+        sns.histplot(data, ax=axes[vertical_axis_ids[row[vertical_axis_name]], col_id], kde=True)
+        col_id = col_id + 1
+        if col_id >= num_cols:
+            col_id = 0
 
     for ax in axes.flat:
         ax.set_xticklabels([])