html_gen allow showing legend for explicit set of coroutines

1 files changed, 5 insertions, 2 deletions

diff --git a/nemesis/html_gen.py b/nemesis/html_gen.py
index 2a417fc..85d2e5f 100644
--- a/nemesis/html_gen.py
+++ b/nemesis/html_gen.py
@@ -53,9 +53,12 @@ def plot_results(results: "collections.defaultdict[str, collections.defaultdict[
         shared_yaxes=False,
     )
 
+    all_coros = set() # used to not add a coro to the legend twice
     for coro_name, x_values in results.items():
         for speedup, xx_values in x_values.items():
             for loop, experiments in xx_values.items():
+                show_legend = coro_name not in all_coros
+                all_coros.add(coro_name)
                 col = loop_to_col[loop]
 
                 x_list = []
@@ -109,7 +112,7 @@ def plot_results(results: "collections.defaultdict[str, collections.defaultdict[
                     mode='markers',
                     name=coro_name,
                     marker=dict(color=get_color(coro_name)),
-                    showlegend=True,
+                    showlegend=show_legend,
                 ), row=1, col=col)
 
                 fig.add_trace(go.Scatter(
@@ -143,7 +146,7 @@ def plot_results(results: "collections.defaultdict[str, collections.defaultdict[
         height=1080,
         width=1920 * num_loops,
         title_text=f"Coroutine Performance Metrics: {input_file}",
-        showlegend=False,
+        showlegend=True,
     )
     for col in range(1, num_loops + 1):
         fig.update_xaxes(title_text="speedup (% optimized away)", row=4, col=col)