From: whitequark <cz@m-labs.hk>
Date: Thu, 13 Dec 2018 11:50:56 +0000 (+0000)
Subject: fhdl.ir: a subfragment's input that we don't drive is also our input.
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=1143878667505a7ead4cbb9e853391ec95129e67;p=nmigen.git

fhdl.ir: a subfragment's input that we don't drive is also our input.
---

diff --git a/nmigen/fhdl/ir.py b/nmigen/fhdl/ir.py
index 4676c48..a3d1d1d 100644
--- a/nmigen/fhdl/ir.py
+++ b/nmigen/fhdl/ir.py
@@ -152,8 +152,10 @@ class Fragment:
             # to provide. If the subfragment is not driving it, it will silently ignore it.
             sub_ins, sub_outs = subfrag._propagate_ports(ports=self_used | ports)
             # Refine the input port approximation: if a subfragment is driving a signal,
-            # it is definitely not our input.
+            # it is definitely not our input. But, if a subfragment requires a signal as an input,
+            # and we aren't driving it, it has to be our input as well.
             ins  -= sub_outs
+            ins  |= sub_ins - self_driven
             # Refine the output port approximation: if a subfragment is driving a signal,
             # and we're asked to provide it, we can provide it now.
             outs |= ports & sub_outs
diff --git a/nmigen/test/test_fhdl_ir.py b/nmigen/test/test_fhdl_ir.py
index 2bf592b..35fdbf2 100644
--- a/nmigen/test/test_fhdl_ir.py
+++ b/nmigen/test/test_fhdl_ir.py
@@ -71,6 +71,19 @@ class FragmentPortsTestCase(FHDLTestCase):
         self.assertEqual(f1.ports, ValueSet())
         self.assertEqual(f2.ports, ValueSet((self.s1,)))
 
+    def test_input_only_in_subfragment(self):
+        f1 = Fragment()
+        f2 = Fragment()
+        f2.add_statements(
+            self.c1.eq(self.s1)
+        )
+        f1.add_subfragment(f2)
+        ins, outs = f1._propagate_ports(ports=())
+        self.assertEqual(ins, ValueSet((self.s1,)))
+        self.assertEqual(outs, ValueSet())
+        self.assertEqual(f1.ports, ValueSet((self.s1,)))
+        self.assertEqual(f2.ports, ValueSet((self.s1,)))
+
     def test_output_from_subfragment(self):
         f1 = Fragment()
         f1.add_statements(