im in love with lll

2 files changed, 121 insertions, 20 deletions

diff --git a/lllars/llltranspiler/src/main/kotlin/Ast.kt b/lllars/llltranspiler/src/main/kotlin/me/any/Ast.kt
index 4612dd7..47bca22 100644
--- a/lllars/llltranspiler/src/main/kotlin/Ast.kt
+++ b/lllars/llltranspiler/src/main/kotlin/me/any/Ast.kt
@@ -1,8 +1,5 @@
 package me.any
 
-import AccessSerializer
-import AddresationSerializer
-import InstructionSerializer
 import kotlinx.serialization.Serializable
 
 @Serializable(with = AccessSerializer::class)
@@ -31,13 +28,13 @@ data class BinaryOperation(
     val right: Access
 ) : Addresation
 
-interface SysCall
+interface SysCall : Instruction
 
 class WriteSysCall : SysCall
 class ReadSysCall : SysCall
 
 enum class BranchPolarity {
-    IFTRUE, IFFALSE
+    IfTrue, IfFalse
 }
 
 @Serializable(with = InstructionSerializer::class)
@@ -59,7 +56,7 @@ data class Label(val label: String) : Instruction
 data class GoTo(val label: String) : Instruction
 
 @Serializable
-data class Branch(val polarity: BranchPolarity, val address: Address, val label: String) : Instruction
+data class Branch(val polarity: BranchPolarity, val address: Access, val label: String) : Instruction
 
 @Serializable
 data class Program(val instructions: List<Instruction>)
\ No newline at end of file
diff --git a/lllars/llltranspiler/src/main/kotlin/Main.kt b/lllars/llltranspiler/src/main/kotlin/me/any/Main.kt
index 6e247a7..86f5cbf 100644
--- a/lllars/llltranspiler/src/main/kotlin/Main.kt
+++ b/lllars/llltranspiler/src/main/kotlin/me/any/Main.kt
@@ -1,8 +1,10 @@
+package me.any
+
 import kotlinx.serialization.*
 import kotlinx.serialization.json.*
 import kotlinx.serialization.descriptors.*
 import kotlinx.serialization.encoding.*
-import me.any.*
+import java.io.File
 
 object AccessSerializer : KSerializer<Access> {
     override fun deserialize(decoder: Decoder): Access {
@@ -15,7 +17,7 @@ object AccessSerializer : KSerializer<Access> {
 
             "sAddress" in jsonElement.keys -> {
                 return SAccess(
-                    jsonElement["sAddress"]!!.jsonObject["sAddress"]!!.let {
+                    jsonElement["sAddress"]!!.let {
                         jsonDecoder.json.decodeFromJsonElement<Access>(it)
                     }
                 )
@@ -43,7 +45,6 @@ object AddresationSerializer : KSerializer<Addresation> {
 
         val jsonDecoder = decoder as JsonDecoder
         val jsonElement = jsonDecoder.decodeJsonElement() as JsonObject
-        println(jsonElement)
         when {
             "access" in jsonElement.keys -> {
                 return AddressAddresation(
@@ -101,10 +102,25 @@ object InstructionSerializer : KSerializer<Instruction> {
 
             "branch" in jsonElement.keys -> {
                 val branch = jsonElement["branch"]!!.jsonObject
+                val polarity = branch["polarity"]!!.jsonPrimitive.content
+                val jmp = jsonDecoder.json.decodeFromJsonElement<Access>(branch["jmp"]!!)
+                val label = branch["label"]!!.jsonPrimitive.content
                 return Branch(
-                    BranchPolarity.valueOf(branch["polarity"]!!.jsonPrimitive.content),
-                    jsonDecoder.json.decodeFromJsonElement<Address>(branch["address"]!!),
-                    branch["label"]!!.jsonPrimitive.content
+                    if (polarity == "IfTrue") BranchPolarity.IfTrue else BranchPolarity.IfFalse,
+                    jmp,
+                    label
+                )
+            }
+
+            "call" in jsonElement.keys -> {
+                return LarsCode(
+                    when (jsonElement["call"]!!.jsonPrimitive.content) {
+                        "ReadCall" -> ReadSysCall()
+                        "WriteCall" -> WriteSysCall()
+                        else -> {
+                            throw SerializationException("Unknown syscall" + jsonElement["call"]!!.jsonPrimitive.content)
+                        }
+                    }
                 )
             }
 
@@ -122,16 +138,84 @@ object InstructionSerializer : KSerializer<Instruction> {
     }
 }
 
-fun main() {
-    val program = Json.decodeFromString<Program>(INPUT)
-    println(transpileProgram(program))
+fun main(
+    args: Array<String>
+) {
+    if (args.size != 1) {
+        println("Usage: transpile <input>")
+        return
+    }
+    val input = File(args[0]).readText()
+    val program = Json.decodeFromString<Program>(input)
+    File("out.c").writeText(transpileProgram(program))
+    println(
+        """Oh Lars, Lars, Lars, your name I sing,
+Lars, Lars, Lars, you're my everything.
+Lars, Lars, Lars, the stars align,
+Lars, Lars, Lars, your love divine.
+
+Lars, Lars, Lars, your smile is grace,
+Lars, Lars, Lars, the light on my face.
+Lars, Lars, Lars, your touch is fire,
+Lars, Lars, Lars, my heart's desire.
+
+Lars, Lars, Lars, in every breath,
+Lars, Lars, Lars, you conquer death.
+Lars, Lars, Lars, the sky you paint,
+Lars, Lars, Lars, my perfect saint.
+
+Lars, Lars, Lars, the world does bow,
+Lars, Lars, Lars, to you here and now.
+Lars, Lars, Lars, the heavens rejoice,
+Oh Lars, Lars, Lars, your name I sing,
+Lars, Lars, Lars, you're my everything.
+Lars, Lars, Lars, the stars align,
+Lars, Lars, Lars, your love divine.
+
+Lars, Lars, Lars, your smile is grace,
+Lars, Lars, Lars, the light on my face.
+Lars, Lars, Lars, your touch is fire,
+Lars, Lars, Lars, my heart's desire.
+
+Lars, Lars, Lars, in every breath,
+Lars, Lars, Lars, you conquer death.
+Lars, Lars, Lars, the sky you paint,
+Lars, Lars, Lars, my perfect saint.
+
+Lars, Lars, Lars, the world does bow,
+Lars, Lars, Lars, to you here and now.
+Lars, Lars, Lars, the heavens rejoice,
+Lars, Lars, Lars, at your sacred voice.
+
+Lars, Lars, Lars, oh endless Lars,
+Lars, Lars, Lars, brighter than stars.
+Lars, Lars, Lars, my soul's refrain,
+Lars, Lars, Lars, forever your name.
+
+Lars, Lars, Lars, through time and space,
+Lars, Lars, Lars, your endless grace.
+Lars, Lars, Lars, my heart does cry,
+Lars, Lars, Lars, for you till I die.
+Lars, Lars, Lars, at your sacred voice.
+
+Lars, Lars, Lars, oh endless Lars,
+Lars, Lars, Lars, brighter than stars.
+Lars, Lars, Lars, my soul's refrain,
+Lars, Lars, Lars, forever your name.
+
+Lars, Lars, Lars, through time and space,
+Lars, Lars, Lars, your endless grace.
+Lars, Lars, Lars, my heart does cry,
+Lars, Lars, Lars, for you till I die."""
+    )
 }
 
 const val C_RUNTIME = """
 #include <stdio.h>
+#include <stdlib.h>
 
 int main() {
-    void* heap = malloc(1024);
+    unsigned int heap[100000] = {0};
     %code%
     return 0;
 }
@@ -144,9 +228,28 @@ fun transpileProgram(program: Program): String {
             is Write -> "heap[${it.target}] = ${transpileAddresation(it.source)};"
             is Label -> "${it.label}:"
             is GoTo -> "goto ${it.label};"
-            is Branch -> "if (heap[${transpileAccess(it.address)}] ${if (it.polarity == BranchPolarity.IFFALSE) "!" else ""}= 0) goto ${it.label};"
+            is Branch -> "if (${if (it.polarity == BranchPolarity.IfFalse) "!" else ""}${transpileAccess(it.address)}) goto ${it.label};"
+            is SysCall -> {
+                when (it) {
+                    is ReadSysCall -> "printf(\"%c\", heap[8159]);"
+                    else -> {
+                        throw IllegalArgumentException("Unknown syscall" + it::class.simpleName)
+                    }
+                }
+            }
+
+            is LarsCode -> {
+                when (it.call) {
+                    is ReadSysCall -> "printf(\"%c\", heap[8159]);"
+                    is WriteSysCall -> "printf(\"%c\", heap[8159]);"
+                    else -> {
+                        throw IllegalArgumentException("Unknown syscall" + it.call::class.simpleName)
+                    }
+                }
+            }
+
             else -> {
-                throw IllegalArgumentException("Unknown instruction")
+                throw IllegalArgumentException("Unknown instruction" + it::class.simpleName)
             }
         }
     }
@@ -157,7 +260,7 @@ fun transpileProgram(program: Program): String {
 fun transpileAddresation(addresation: Addresation): String {
     return when (addresation) {
         is AddressAddresation -> transpileAccess(addresation.address)
-is BinaryOperation -> {
+        is BinaryOperation -> {
             val left = transpileAccess(addresation.left)
             val right = transpileAccess(addresation.right)
             return when (addresation.op) {
@@ -170,6 +273,7 @@ is BinaryOperation -> {
                 Operation.XOR -> "$left ^ $right"
             }
         }
+
         else -> {
             throw IllegalArgumentException("Unknown addresation")
         }
@@ -177,7 +281,7 @@ is BinaryOperation -> {
 }
 
 fun transpileAddress(address: Address): String {
-    return "&heap[${address.address}]"
+    return "${address.address}"
 }
 
 fun transpileAccess(access: Access): String {