Implement Patio byte-size-of instruction

1 files changed, 36 insertions, 1 deletions

diff --git a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
index 8582cfd0..ddc9c386 100644
--- a/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
+++ b/src/test/java/org/shedlang/compiler/patio/PatioExampleTests.java
@@ -819,7 +819,7 @@ public abstract class PatioExampleTests {
     }
 
     @Test
-    public void canGetByteOffset() throws IOException, InterruptedException {
+    public void canGetByteOffsetOf() throws IOException, InterruptedException {
         var patioFunction = PatioFunctionDefinition.builder()
             .withName(FUNCTION_NAME)
             .withReturnType(PatioTypes.INT)
@@ -854,6 +854,41 @@ public abstract class PatioExampleTests {
         assertThat(result, isSilentExit(4));
     }
 
+    @Test
+    public void canGetByteSizeOf() throws IOException, InterruptedException {
+        var patioFunction = PatioFunctionDefinition.builder()
+            .withName(FUNCTION_NAME)
+            .withReturnType(PatioTypes.INT)
+            .addLocal(Patio.val("x", PatioTypes.INT))
+            .addBodyInstruction(Patio.byteSizeOf(
+                Patio.localRef("x"),
+                PatioDataLayouts.struct(List.of(
+                    PatioDataLayouts.struct(List.of(
+                        PatioDataLayouts.BYTE,
+                        PatioDataLayouts.BYTE
+                    )),
+                    PatioDataLayouts.struct(List.of(
+                        PatioDataLayouts.BYTE,
+                        PatioDataLayouts.BYTE,
+                        PatioDataLayouts.BYTE,
+                        PatioDataLayouts.BYTE
+                    )),
+                    PatioDataLayouts.BYTE
+                ))
+            ))
+            .addBodyInstruction(Patio.return_(
+                Patio.localRef("x")
+            ))
+            .build();
+        var patio = Patio.compilationUnit(List.of(
+            patioFunction
+        ));
+
+        var result = run(patio);
+
+        assertThat(result, isSilentExit(7));
+    }
+
     private Matcher<ExecutionResult> isSilentExit(int exitCode) {
         return allOf(
             has("stderr", x -> x.stderr(), equalTo("")),