Added type argument to print_slacks_type and added tie-breaker for sort-y-times

• Added optional type argument to Schedule.print_slack_type().

• Added tie-breaker for Schedule.sort_y_times_on_start_times().

Closes: #334 (closed) #331 (closed)

b_asic/schedule.py

@@ -410,7 +410,7 @@ class Schedule:
             raise ValueError(f"No operation with graph_id {graph_id!r} in schedule")
         return self.backward_slack(graph_id), self.forward_slack(graph_id)
 
-    def print_slacks(self, order: int = 0) -> None:
+    def print_slacks(self, order: int = 0, type_name: TypeName | None = None) -> None:
         """
         Print the slack times for all operations in the schedule.
 
@@ -422,14 +422,24 @@ class Schedule:
             * 0: alphabetical on Graph ID
             * 1: backward slack
             * 2: forward slack
+
+        type_name : TypeName, optional
+            If given, only the slack times for operations of this type will be printed.
         """
+        if type_name is None:
+            operations = self._sfg.operations
+        else:
+            operations = [
+                op for op in self._sfg.operations if isinstance(op, type_name)
+            ]
+
         res: list[tuple[GraphID, int, int]] = [
             (
                 op.graph_id,
                 cast(int, self.backward_slack(op.graph_id)),
                 self.forward_slack(op.graph_id),
             )
-            for op in self._sfg.operations
+            for op in operations
         ]
         res.sort(key=lambda tup: tup[order])
         res_str = [
@@ -1084,10 +1094,17 @@ class Schedule:
         move_y_location
         set_y_location
         """
-        for i, graph_id in enumerate(
-            sorted(self._start_times, key=self._start_times.get)
-        ):
+
+        def sort_key(graph_id):
+            op = self._sfg.find_by_id(graph_id)
+            return (
+                self._start_times[op.graph_id],
+                -self._sfg.find_by_id(graph_id).latency,
+            )
+
+        for i, graph_id in enumerate(sorted(self._start_times, key=sort_key)):
             self.set_y_location(graph_id, i)
+
         for graph_id in self._start_times:
             op = cast(Operation, self._sfg.find_by_id(graph_id))
             # Position Outputs and Sinks adjacent to the operation generating them