Add Patio code gen for push-string

6 files changed, 94 insertions, 5 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
index 47e7c496..120a78af 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGen.java
@@ -1,6 +1,7 @@
 package org.shedlang.compiler.patio;
 
 import org.shedlang.compiler.patio.ast.PatioCompilationUnit;
+import org.shedlang.compiler.patio.ast.instructions.PatioDataPtrToInt;
 import org.shedlang.compiler.patio.ast.instructions.PatioIntAdd;
 import org.shedlang.compiler.patio.ast.instructions.PatioLocalStore;
 import org.shedlang.compiler.patio.ast.instructions.PatioReturn;
@@ -257,6 +258,34 @@ public class PatioCodeGen {
             case PushIntInstruction pushInt ->
                 pushValue(pushInt.value().intValueExact(), pushInt.source(), context);
 
+            case PushStringInstruction pushString -> {
+                var stringPtrOperandResult = PatioCodeGenStrings.define(
+                    pushString.value(),
+                    pushString.source(),
+                    context
+                );
+                var stringPtrOperand = stringPtrOperandResult.value();
+                context = stringPtrOperandResult.context();
+
+                var stringPtrResult = context.allocateLocal(
+                    PatioCodeGenLocalSpace.STRING,
+                    PatioCodeGenTypes.VALUE_TYPE
+                );
+                var stringPtr = stringPtrResult.value();
+                context = stringPtrResult.context();
+
+                context = context.addInstruction(new PatioDataPtrToInt(
+                    stringPtr,
+                    stringPtrOperand,
+                    pushString.source()
+                ));
+
+                context = context.temporaryPush(stringPtr);
+
+                yield context;
+            }
+
+
             case PushUnitInstruction pushUnit ->
                 pushValue(0, pushUnit.source(), context);
 
diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenCompilationUnitElementSpace.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenCompilationUnitElementSpace.java
index 0db4e4de..4d18223e 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenCompilationUnitElementSpace.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenCompilationUnitElementSpace.java
@@ -1,7 +1,8 @@
 package org.shedlang.compiler.patio;
 
 public enum PatioCodeGenCompilationUnitElementSpace {
-    FUNCTION("function");
+    FUNCTION("function"),
+    STRING("string");
 
     private final String name;
 
diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
index 4fe625f3..f19f92f2 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenContext.java
@@ -231,7 +231,18 @@ public record PatioCodeGenContext(
         );
     }
 
-    public PatioCodeGenContext addDataElement(String name, PatioDataLayout layout, Source source) {
+    public PatioCodeGenResult<String> allocateCompilationUnitElementName(
+        PatioCodeGenCompilationUnitElementSpace space
+    ) {
+        return withCompilationUnitContextResult(
+            this.compilationUnitContext.allocateCompilationUnitName(
+                space,
+                Optional.empty()
+            )
+        );
+    }
+
+    public PatioCodeGenContext addDataElementMutable(String name, PatioDataLayout layout, Source source) {
         var dataElement = new PatioDataElementMutable(name, layout, source);
         return addCompilationUnitElement(dataElement);
     }
@@ -355,7 +366,7 @@ public record PatioCodeGenContext(
         return new PatioCodeGenContext(this.compilationUnitContext, result.context).toResult(result.value);
     }
 
-    private <T> PatioCodeGenResult<T> toResult(T value) {
+    public <T> PatioCodeGenResult<T> toResult(T value) {
         return new PatioCodeGenResult<>(value, this);
     }
 
diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenLocalSpace.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenLocalSpace.java
index 0a8cac3e..1735c2a0 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenLocalSpace.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenLocalSpace.java
@@ -12,6 +12,7 @@ public enum PatioCodeGenLocalSpace {
     NAMESPACE_VALUE("namespaceValue"),
     LOCAL("local"),
     RECEIVER("receiver"),
+    STRING("string"),
     TEMPORARY("tmp");
 
     private final String name;
diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenNamespaces.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenNamespaces.java
index fbbd5c79..123bbf4a 100644
--- a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenNamespaces.java
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenNamespaces.java
@@ -37,13 +37,13 @@ public class PatioCodeGenNamespaces {
         var namespaceTypeInfo = context.namespaceTypeInfo(namespaceName);
         var layout = namespaceDataLayout(namespaceTypeInfo);
 
-        context = context.addDataElement(
+        context = context.addDataElementMutable(
             globalNameNamespaceValue(namespaceName),
             layout,
             source
         );
 
-        context = context.addDataElement(
+        context = context.addDataElementMutable(
             globalNameNamespaceIsInited(namespaceName),
             PatioDataLayouts.scalar(PatioTypes.BOOL),
             source
diff --git a/src/main/java/org/shedlang/compiler/patio/PatioCodeGenStrings.java b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenStrings.java
new file mode 100644
index 00000000..a9b7c753
--- /dev/null
+++ b/src/main/java/org/shedlang/compiler/patio/PatioCodeGenStrings.java
@@ -0,0 +1,47 @@
+package org.shedlang.compiler.patio;
+
+import org.shedlang.compiler.patio.ast.PatioDataElementImmutable;
+import org.shedlang.compiler.patio.ast.operands.PatioDataPtr;
+import org.shedlang.compiler.patio.ast.types.PatioDataValue;
+import org.shedlang.compiler.patio.ast.types.PatioDataValues;
+import org.shedlang.compiler.sources.Source;
+
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.List;
+
+public class PatioCodeGenStrings {
+    private PatioCodeGenStrings() {}
+
+    public static PatioCodeGenResult<PatioDataPtr> define(
+        String value,
+        Source source,
+        PatioCodeGenContext context
+    ) {
+        var codeUnits = value.getBytes(StandardCharsets.UTF_8);
+
+        var codeUnitsDataValues = new ArrayList<PatioDataValue>();
+        for (var codeUnit : codeUnits) {
+            codeUnitsDataValues.add(PatioDataValues.byte_(codeUnit));
+        }
+
+        var dataValue = PatioDataValues.struct(List.of(
+            PatioDataValues.int_(codeUnits.length),
+            PatioDataValues.array(codeUnitsDataValues)
+        ));
+
+        var elementNameResult = context.allocateCompilationUnitElementName(
+            PatioCodeGenCompilationUnitElementSpace.STRING
+        );
+        var elementName = elementNameResult.value();
+        context = elementNameResult.context();
+
+        context = context.addCompilationUnitElement(new PatioDataElementImmutable(
+            elementName,
+            dataValue,
+            source
+        ));
+
+        return context.toResult(new PatioDataPtr(elementName));
+    }
+}