Add support for wrap/wrap_target

.. This also needs support in the core for specifying them in the StateMachine constructor.

Author: jepler

adafruit_pioasm.py

@@ -48,7 +48,9 @@ def __init__(self, text_program: str) -> None:
         labels = {}
         instructions = []
         sideset_count = 0
-        sideset_enable = 0
+        sideset_enable = False
+        wrap = None
+        wrap_target = 0
         for line in text_program.split("\n"):
             line = line.strip()
             if not line:
@@ -60,13 +62,14 @@ def __init__(self, text_program: str) -> None:
                     raise RuntimeError("Multiple programs not supported")
                 program_name = line.split()[1]
             elif line.startswith(".wrap_target"):
-                if len(instructions) > 0:
-                    raise RuntimeError("wrap_target not supported")
+                wrap_target = len(instructions)
             elif line.startswith(".wrap"):
-                pass
+                if len(instructions) == 0:
+                    raise RuntimeError("Cannot have .wrap as first instruction")
+                wrap = len(instructions) - 1
             elif line.startswith(".side_set"):
                 sideset_count = int(line.split()[1])
-                sideset_enable = 1 if "opt" in line else 0
+                sideset_enable = "opt" in line
             elif line.endswith(":"):
                 label = line[:-1]
                 if label in labels:
@@ -223,6 +226,10 @@ def __init__(self, text_program: str) -> None:
             "sideset_enable": sideset_enable,
         }
 
+        if wrap is not None:
+            self.pio_kwargs["wrap"] = wrap
+            self.pio_kwargs["wrap_target"] = wrap_target
+
         self.assembled = array.array("H", assembled)
 
 

tests/testpioasm.py

@@ -135,8 +135,14 @@ def testMovReverse(self):
         self.assertAssemblesTo("mov x, :: x", [0b101_00000_001_10_001])
         self.assertAssemblesTo("mov x, ::x", [0b101_00000_001_10_001])
 
+
 class TestWrap(AssembleChecks):
     def testWrap(self):
         self.assertAssemblyFails(".wrap")
-        self.assertPioKwargs("nop\n.wrap_target\nnop\nnop\n.wrap", 
-            sideset_count=0, sideset_enable=False, wrap=2, wrap_target=1)
+        self.assertPioKwargs(
+            "nop\n.wrap_target\nnop\nnop\n.wrap",
+            sideset_count=0,
+            sideset_enable=False,
+            wrap=2,
+            wrap_target=1,
+        )