From 0ffa8c8e37569ca34850576e01c11bcc79e7ea57 Mon Sep 17 00:00:00 2001
From: TheZoq2 <frans.skarman@protonmail.com>
Date: Tue, 14 Feb 2023 14:10:31 +0100
Subject: [PATCH] Add critical path computation

---
 b_asic/signal_flow_graph.py | 6 ++++++
 test/test_sfg.py            | 8 ++++++++
 2 files changed, 14 insertions(+)

diff --git a/b_asic/signal_flow_graph.py b/b_asic/signal_flow_graph.py
index c4a6545d..bacdb629 100644
--- a/b_asic/signal_flow_graph.py
+++ b/b_asic/signal_flow_graph.py
@@ -1451,3 +1451,9 @@ class SFG(AbstractOperation):
         if format is not None:
             dg.format = format
         dg.view()
+
+    def critical_path(self):
+        # Import here needed to avoid circular imports
+        from b_asic.schedule import Schedule
+
+        return Schedule(self, scheduling_algorithm="ASAP").schedule_time
diff --git a/test/test_sfg.py b/test/test_sfg.py
index 00079b7d..57ddc39b 100644
--- a/test/test_sfg.py
+++ b/test/test_sfg.py
@@ -1559,3 +1559,11 @@ class TestSFGErrors:
             match=re.escape("Output index out of range (expected 0-0, got 1)"),
         ):
             sfg.inputs_required_for_output(1)
+
+
+class TestCriticalPath:
+    def test_single_accumulator(self, sfg_simple_accumulator: SFG):
+        sfg_simple_accumulator.set_latency_of_type(Addition.type_name(), 5)
+        sfg_simple_accumulator.inputs[0].latency_offset = 0
+        sfg_simple_accumulator.outputs[0].latency_offset = 0
+        assert sfg_simple_accumulator.critical_path() == 5
-- 
GitLab