Visualize: fix highlights

schedgen/schedgen-visualize.py

@@ -69,7 +69,7 @@ def name_strategies(schedule, shorthands, max_id, names):
     for (time, s) in schedule:
         if s not in names:
             if s in shorthands:
-                names[s] = shorthands[s][0]
+                names[s] = (shorthands[s][0], 1)
             else:
                 names[s] = (max_id, 1)
                 max_id = max_id + 1
@@ -111,7 +111,7 @@ def print_strat_table(
             )
         else:
             output_file.write(
-                f"\\node[dup,minimum width={width}cm] at ({x},{y}) {{{node_str}}};\n"
+                f"\\node[highlight,minimum width={width}cm] at ({x},{y}) {{{node_str}}};\n"
             )
         x = x + width
 
@@ -221,7 +221,7 @@ if __name__ == "__main__":
             for row in reader:
                 strategy = row["strategy"]
                 shorthand = row["shorthand"]
-                highlight = bool(row["highlight"])
+                highlight = row["highlight"] in ["yes", "true", "1"]
                 shorthands[strategy] = (shorthand, highlight)
 
     map(lambda s: cut_off(total_time, s), schedules)