Translate put_z verilog case into nmigen

diff --git a/src/add/nmigen_add_experiment.py b/src/add/nmigen_add_experiment.py
index 24e87428e559177cdb1727d8fa7a5ed44d013daf..a69b77b202e1bbcca50eb0da536820f9272ee43c 100644 (file)
--- a/src/add/nmigen_add_experiment.py
+++ b/src/add/nmigen_add_experiment.py
@@ -330,17 +330,16 @@ class FPADD:
             # ******
             # put_z stage
 
-            """
-              put_z:
-              begin
-                s_out_z_stb <= 1;
-                s_out_z <= z;
-                if (s_out_z_stb && out_z_ack) begin
-                  s_out_z_stb <= 0;
-                  state <= get_a;
-                end
-              end
-            """
+            with m.State("put_z"):
+              m.next = "get_a"
+              m.d.sync += [
+                  s_out_z_stb.eq(1),
+                  s_out_z.eq(z)
+              ]
+              with m.If(s_out_z_stb & out_z_ack):
+                  m.d.sync += [
+                  s_out_z_stb.eq(0)
+                  ]
 
         return m