From 7acebc63b0a641b8470795aa359f2f3dd0814836 Mon Sep 17 00:00:00 2001
From: Jeff Epler <jepler@gmail.com>
Date: Sat, 7 Sep 2024 08:50:25 -0500
Subject: [PATCH] Add mov pindirs

---
 adafruit_pioasm.py    | 10 ++++++++--
 tests/test_version.py |  3 +++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/adafruit_pioasm.py b/adafruit_pioasm.py
index 3c6c446..8662e04 100644
--- a/adafruit_pioasm.py
+++ b/adafruit_pioasm.py
@@ -29,7 +29,8 @@ CONDITIONS = ["", "!x", "x--", "!y", "y--", "x!=y", "pin", "!osre"]
 IN_SOURCES = ["pins", "x", "y", "null", None, None, "isr", "osr"]
 OUT_DESTINATIONS = ["pins", "x", "y", "null", "pindirs", "pc", "isr", "exec"]
 WAIT_SOURCES = ["gpio", "pin", "irq", None]
-MOV_DESTINATIONS = ["pins", "x", "y", None, "exec", "pc", "isr", "osr"]
+MOV_DESTINATIONS_V0 = ["pins", "x", "y", None, "exec", "pc", "isr", "osr"]
+MOV_DESTINATIONS_V1 = ["pins", "x", "y", "pindirs", "exec", "pc", "isr", "osr"]
 MOV_SOURCES = ["pins", "x", "y", "null", None, "status", "isr", "osr"]
 MOV_OPS = [None, "~", "::", None]
 SET_DESTINATIONS = ["pins", "x", "y", None, "pindirs", None, None, None]
@@ -219,6 +220,11 @@ class Program:  # pylint: disable=too-few-public-methods
                 instructions.append(line)
                 linemap.append(i)
 
+        if pio_version >= 1:
+            mov_destinations = MOV_DESTINATIONS_V1
+        else:
+            mov_destinations = MOV_DESTINATIONS_V0
+
         max_delay = 2 ** (5 - sideset_count - sideset_enable) - 1
         assembled = []
         for line in instructions:
@@ -331,7 +337,7 @@ class Program:  # pylint: disable=too-few-public-methods
                     assembled[-1] |= parse_rxfifo_brackets(instruction[2], "get")
                 else:
                     assembled.append(0b101_00000_000_00_000)
-                    assembled[-1] |= MOV_DESTINATIONS.index(instruction[1]) << 5
+                    assembled[-1] |= mov_destinations.index(instruction[1]) << 5
                     source = instruction[-1]
                     source_split = mov_splitter(source)
                     if len(source_split) == 1:
diff --git a/tests/test_version.py b/tests/test_version.py
index ff5840a..2999caf 100644
--- a/tests/test_version.py
+++ b/tests/test_version.py
@@ -122,3 +122,6 @@ def test_mov_v1() -> None:
     assert_assembly_fails(prefix + "mov osr, rxfifo[8]")
     assert_assembles_to(prefix + "mov rxfifo[y], isr", [0b100_00000_0001_1_000])
     assert_assembles_to(prefix + "mov osr, rxfifo[1]", [0b100_00000_1001_0_001])
+
+    assert_assembly_fails("mov pindirs, null", errtype=ValueError)
+    assert_assembles_to(prefix + "mov pindirs, null", [0b101_00000_01100011])