IR WIP

compiler/build.gradle.kts

@@ -4,6 +4,7 @@ plugins {
 
 dependencies {
   api(project(":ast"))
+  api(project(":bir"))
   api(project(":bytecode"))
   api(project(":parser"))
   api(project(":frontend"))

compiler/src/main/kotlin/gay/pizza/pork/compiler/CompilableSlab.kt

@@ -1,9 +1,14 @@
 package gay.pizza.pork.compiler
 
 import gay.pizza.pork.ast.gen.Symbol
+import gay.pizza.pork.bir.IrDefinition
+import gay.pizza.pork.bir.IrSlab
+import gay.pizza.pork.bir.IrSlabLocation
 import gay.pizza.pork.frontend.Slab
 
 class CompilableSlab(val compiler: Compiler, val slab: Slab) {
+  val compiledIrSlab: IrSlab by lazy { compileIrSlab() }
+
   val compilableSymbols: List<CompilableSymbol> by lazy {
     slab.scope.internalSymbols.map { symbol ->
       CompilableSymbol(this, symbol)
@@ -18,4 +23,13 @@ class CompilableSlab(val compiler: Compiler, val slab: Slab) {
     val scopeSymbol = slab.scope.resolve(symbol) ?: return null
     return compiler.resolveOrNull(scopeSymbol)
   }
+
+  private fun compileIrSlab(): IrSlab {
+    val definitions = mutableListOf<IrDefinition>()
+    for (compilableSymbol in compilableSymbols) {
+      definitions.add(compilableSymbol.compiledIrDefinition)
+    }
+    val irSlabLocation = IrSlabLocation(slab.location.form, slab.location.filePath)
+    return IrSlab(irSlabLocation, definitions)
+  }
 }

compiler/src/main/kotlin/gay/pizza/pork/compiler/CompilableSymbol.kt

@@ -3,9 +3,14 @@ package gay.pizza.pork.compiler
 import gay.pizza.pork.ast.gen.FunctionDefinition
 import gay.pizza.pork.ast.gen.LetDefinition
 import gay.pizza.pork.ast.gen.visit
+import gay.pizza.pork.bir.IrCodeBlock
+import gay.pizza.pork.bir.IrDefinition
+import gay.pizza.pork.bir.IrDefinitionType
+import gay.pizza.pork.bir.IrSymbolTag
 import gay.pizza.pork.frontend.scope.ScopeSymbol
 
 class CompilableSymbol(val compilableSlab: CompilableSlab, val scopeSymbol: ScopeSymbol) {
+  val compiledIrDefinition: IrDefinition by lazy { compileIrDefinition() }
   val compiledStubOps: CompiledSymbolResult by lazy { compile() }
 
   val usedSymbols: List<ScopeSymbol>
@@ -27,6 +32,29 @@ class CompilableSymbol(val compilableSlab: CompilableSlab, val scopeSymbol: Scop
     return emitter.code.build()
   }
 
+  private fun compileIrDefinition(): IrDefinition {
+    val compiler = compilableSlab.compiler
+    val functionSymbol = compiler.irSymbolWorld.lookup(scopeSymbol, IrSymbolTag.Function)
+    val irCodeEmitter = IrCodeEmitter(
+      self = functionSymbol,
+      irSymbolWorld = compiler.irSymbolWorld,
+      irSymbolAssignment = compiler.irSymbolAssignment,
+      scope = compilableSlab.slab.scope
+    )
+    val what = if (scopeSymbol.definition is FunctionDefinition) {
+      val functionDefinition = scopeSymbol.definition as FunctionDefinition
+      functionDefinition.block ?: functionDefinition.nativeFunctionDescriptor!!
+    } else {
+      val letDefinition = scopeSymbol.definition as LetDefinition
+      letDefinition.value
+    }
+    val irCodeElement = irCodeEmitter.visit(what)
+    val irCodeBlock = if (irCodeElement is IrCodeBlock) {
+      irCodeElement
+    } else IrCodeBlock(listOf(irCodeElement))
+    return IrDefinition(functionSymbol, IrDefinitionType.Function, irCodeBlock)
+  }
+
   val id: String
     get() = "${compilableSlab.slab.location.commonLocationIdentity} ${scopeSymbol.symbol.id}"
 

compiler/src/main/kotlin/gay/pizza/pork/compiler/CompileError.kt

@@ -0,0 +1,5 @@
+package gay.pizza.pork.compiler
+
+import gay.pizza.pork.ast.gen.Node
+
+class CompileError(message: String, val node: Node? = null) : RuntimeException(message)

compiler/src/main/kotlin/gay/pizza/pork/compiler/Compiler.kt

@@ -1,5 +1,6 @@
 package gay.pizza.pork.compiler
 
+import gay.pizza.pork.bir.IrSymbolAssignment
 import gay.pizza.pork.bytecode.CompiledWorld
 import gay.pizza.pork.bytecode.MutableConstantPool
 import gay.pizza.pork.frontend.Slab
@@ -11,6 +12,9 @@ class Compiler {
     CompilableSlab(this, slab)
   }
 
+  val irSymbolAssignment: IrSymbolAssignment = IrSymbolAssignment()
+  val irSymbolWorld: IrSymbolWorld = IrSymbolWorld(irSymbolAssignment)
+
   fun resolveOrNull(scopeSymbol: ScopeSymbol): CompilableSymbol? {
     val compiledSlab = compilableSlabs.of(scopeSymbol.slabScope.slab)
     return compiledSlab.resolve(scopeSymbol.symbol)

compiler/src/main/kotlin/gay/pizza/pork/compiler/IrCodeEmitter.kt

@@ -0,0 +1,268 @@
+package gay.pizza.pork.compiler
+
+import gay.pizza.pork.ast.FunctionLevelVisitor
+import gay.pizza.pork.ast.gen.*
+import gay.pizza.pork.bir.*
+import gay.pizza.pork.frontend.scope.ScopeSymbol
+import gay.pizza.pork.frontend.scope.SlabScope
+
+class IrCodeEmitter(
+  val self: IrSymbol,
+  val irSymbolWorld: IrSymbolWorld,
+  val irSymbolAssignment: IrSymbolAssignment,
+  val scope: SlabScope
+) : FunctionLevelVisitor<IrCodeElement>() {
+  private val loopSymbols = mutableListOf<IrSymbol>()
+  private val localVariables = mutableListOf<MutableList<LocalVariable>>()
+
+  private fun startLoop(): IrSymbol {
+    val symbol = irSymbolAssignment.next(IrSymbolTag.Loop)
+    loopSymbols.add(symbol)
+    return symbol
+  }
+
+  private fun endLoop() {
+    loopSymbols.removeLast()
+  }
+
+  private fun <T> loop(block: (IrSymbol) -> T): T {
+    val symbol = startLoop()
+    return try {
+      block(symbol)
+    } finally {
+      endLoop()
+    }
+  }
+
+  private fun enterBlockScope() {
+    val locals = mutableListOf<LocalVariable>()
+    localVariables.add(locals)
+  }
+
+  private fun exitBlockScope() {
+    localVariables.removeLast()
+  }
+
+  private fun createLocalVariable(name: Symbol): IrSymbol {
+    val symbol = irSymbolAssignment.next(IrSymbolTag.Local)
+    val variable = LocalVariable(symbol, name)
+    localVariables.last().add(variable)
+    return symbol
+  }
+
+  private fun scopeSymbolToTag(scopeSymbol: ScopeSymbol): IrSymbolTag =
+    if (scopeSymbol.definition is FunctionDefinition) {
+      IrSymbolTag.Function
+    } else {
+      IrSymbolTag.Variable
+    }
+
+  private fun lookupLocalVariable(name: Symbol): IrSymbol? {
+    for (i in 0..localVariables.size) {
+      val b = localVariables.size - i - 1
+      val scope = localVariables[b]
+      val found = scope.firstOrNull { it.name == name }
+      if (found != null) {
+        return found.symbol
+      }
+    }
+    return null
+  }
+
+  private fun lookup(name: Symbol): IrSymbol? {
+    val local = lookupLocalVariable(name)
+    if (local != null) {
+      return local
+    }
+    val scoped = scope.resolve(name)
+    if (scoped != null) {
+      return irSymbolWorld.lookup(scoped, scopeSymbolToTag(scoped))
+    }
+    return null
+  }
+
+  private fun lookupFunction(name: Symbol): Pair<ScopeSymbol, IrSymbol>? {
+    val scoped = scope.resolve(name) ?: return null
+    return scoped to irSymbolWorld.lookup(scoped, scopeSymbolToTag(scoped))
+  }
+
+  override fun visitBlock(node: Block): IrCodeBlock {
+    enterBlockScope()
+    val block = IrCodeBlock(node.expressions.map { it.visit(this) })
+    exitBlockScope()
+    return block
+  }
+
+  override fun visitBooleanLiteral(node: BooleanLiteral): IrCodeElement =
+    IrBooleanConstant(node.value)
+
+  override fun visitBreak(node: Break): IrCodeElement {
+    val currentLoopSymbol = loopSymbols.lastOrNull() ?:
+      throw CompileError("break does not have a target loop", node)
+    return IrBreak(currentLoopSymbol)
+  }
+
+  override fun visitContinue(node: Continue): IrCodeElement {
+    val currentLoopSymbol = loopSymbols.lastOrNull() ?:
+      throw CompileError("continue does not have a target loop", node)
+    return IrBreak(currentLoopSymbol)
+  }
+
+  override fun visitDoubleLiteral(node: DoubleLiteral): IrCodeElement =
+    IrDoubleConstant(node.value)
+
+  override fun visitForIn(node: ForIn): IrCodeElement {
+    return IrNoneConstant
+  }
+
+  override fun visitFunctionCall(node: FunctionCall): IrCodeElement {
+    val (scopeSymbol, symbol) = lookupFunction(node.symbol) ?:
+      throw CompileError("Failed to resolve function call target '${node.symbol.id}'", node)
+    if (symbol.tag != IrSymbolTag.Function) {
+      throw CompileError("Failed to resolve function call target '${node.symbol.id}', it is not a function", node)
+    }
+
+    val functionDefinition = scopeSymbol.definition as FunctionDefinition
+    val arguments = mutableListOf<IrCodeElement>()
+    var variableArguments: List<IrCodeElement>? = null
+
+    val inputs = node.arguments
+
+    for ((index, spec) in functionDefinition.arguments.withIndex()) {
+      if (variableArguments != null) {
+        throw CompileError(
+          "Failed to build function call, '${node.symbol.id}', illegal function definition",
+          node
+        )
+      }
+
+      if (spec.multiple) {
+        variableArguments = inputs.drop(index).map { it.visit(this) }
+      } else {
+        if (index > inputs.size - 1) {
+          throw CompileError(
+            "Failed to build function call, '${node.symbol.id}', no matching argument for '${spec.symbol.id}'",
+            node
+          )
+        }
+        arguments.add(inputs[index].visit(this))
+      }
+    }
+
+    if (functionDefinition.arguments.any { it.multiple }) {
+      variableArguments = mutableListOf()
+    }
+
+    return IrCall(symbol, arguments, variableArguments)
+  }
+
+  override fun visitIf(node: If): IrCodeElement =
+    IrConditional(
+      node.condition.visit(this),
+      node.thenBlock.visit(this),
+      node.elseBlock?.visit(this) ?: IrNoneConstant
+    )
+
+  override fun visitIndexedBy(node: IndexedBy): IrCodeElement {
+    TODO("Not yet implemented")
+  }
+
+  override fun visitInfixOperation(node: InfixOperation): IrCodeElement {
+    val op = when (node.op) {
+      InfixOperator.Plus -> IrInfixOp.Add
+      InfixOperator.Minus -> IrInfixOp.Subtract
+      InfixOperator.Multiply -> IrInfixOp.Multiply
+      InfixOperator.Divide -> IrInfixOp.Divide
+      InfixOperator.Equals -> IrInfixOp.Equals
+      InfixOperator.NotEquals -> IrInfixOp.NotEquals
+      InfixOperator.EuclideanModulo -> IrInfixOp.EuclideanModulo
+      InfixOperator.Remainder -> IrInfixOp.Remainder
+      InfixOperator.Lesser -> IrInfixOp.Lesser
+      InfixOperator.Greater -> IrInfixOp.Greater
+      InfixOperator.GreaterEqual -> IrInfixOp.GreaterEqual
+      InfixOperator.LesserEqual -> IrInfixOp.LesserEqual
+      InfixOperator.BooleanAnd -> IrInfixOp.BooleanAnd
+      InfixOperator.BooleanOr -> IrInfixOp.BooleanOr
+      InfixOperator.BinaryAnd -> IrInfixOp.BinaryAnd
+      InfixOperator.BinaryOr -> IrInfixOp.BinaryOr
+      InfixOperator.BinaryExclusiveOr -> IrInfixOp.BinaryExclusiveOr
+    }
+
+    return IrInfix(op, node.left.visit(this), node.right.visit(this))
+  }
+
+  override fun visitIntegerLiteral(node: IntegerLiteral): IrCodeElement =
+    IrIntegerConstant(node.value)
+
+  override fun visitLetAssignment(node: LetAssignment): IrCodeElement {
+    val symbol = createLocalVariable(node.symbol)
+    return IrStore(symbol, node.value.visit(this))
+  }
+
+  override fun visitListLiteral(node: ListLiteral): IrCodeElement =
+    IrList(node.items.map { it.visit(this) })
+
+  override fun visitLongLiteral(node: LongLiteral): IrCodeElement =
+    IrLongConstant(node.value)
+
+  override fun visitNoneLiteral(node: NoneLiteral): IrCodeElement =
+    IrNoneConstant
+
+  override fun visitParentheses(node: Parentheses): IrCodeElement =
+    node.expression.visit(this)
+
+  override fun visitPrefixOperation(node: PrefixOperation): IrCodeElement {
+    val op = when (node.op) {
+      PrefixOperator.BooleanNot -> IrPrefixOp.BooleanNot
+      PrefixOperator.UnaryPlus -> IrPrefixOp.UnaryPlus
+      PrefixOperator.UnaryMinus -> IrPrefixOp.UnaryMinus
+      PrefixOperator.BinaryNot -> IrPrefixOp.BinaryNot
+    }
+    return IrPrefix(op, node.expression.visit(this))
+  }
+
+  override fun visitReturn(node: Return): IrCodeElement =
+    IrReturn(from = self, value = node.value.visit(this))
+
+  override fun visitSetAssignment(node: SetAssignment): IrCodeElement {
+    val symbol = lookupLocalVariable(node.symbol) ?:
+      throw CompileError("Unable to find local variable target '${node.symbol.id}'", node)
+    return IrStore(symbol, node.value.visit(this))
+  }
+
+  override fun visitStringLiteral(node: StringLiteral): IrCodeElement =
+    IrStringConstant(node.text)
+
+  override fun visitSuffixOperation(node: SuffixOperation): IrCodeElement {
+    val op = when (node.op) {
+      SuffixOperator.Increment -> IrSuffixOp.Increment
+      SuffixOperator.Decrement -> IrSuffixOp.Decrement
+    }
+    val symbol = lookup(node.reference.symbol) ?: throw CompileError(
+      "Unable to find symbol for suffix operation '${node.reference.symbol.id}'", node)
+    return IrSuffix(op, symbol)
+  }
+
+  override fun visitSymbolReference(node: SymbolReference): IrCodeElement {
+    val symbol = lookup(node.symbol) ?:
+      throw CompileError("Unable to resolve symbol reference '${node.symbol.id}'", node)
+    return IrLoad(symbol)
+  }
+
+  override fun visitVarAssignment(node: VarAssignment): IrCodeElement {
+    val local = createLocalVariable(node.symbol)
+    return IrStore(local, node.value.visit(this))
+  }
+
+  override fun visitWhile(node: While): IrCodeElement = loop { symbol ->
+    IrLoop(
+      symbol = symbol,
+      condition = node.condition.visit(this),
+      inner = node.block.visit(this)
+    )
+  }
+
+  override fun visitNativeFunctionDescriptor(node: NativeFunctionDescriptor): IrCodeElement {
+    return IrNoneConstant
+  }
+}

compiler/src/main/kotlin/gay/pizza/pork/compiler/IrSymbolWorld.kt

@@ -0,0 +1,13 @@
+package gay.pizza.pork.compiler
+
+import gay.pizza.pork.bir.IrSymbol
+import gay.pizza.pork.bir.IrSymbolAssignment
+import gay.pizza.pork.bir.IrSymbolTag
+
+class IrSymbolWorld(val irSymbolAssignment: IrSymbolAssignment) {
+  private val symbols = mutableMapOf<Any, IrSymbol>()
+
+  fun lookup(value: Any, tag: IrSymbolTag): IrSymbol = symbols.getOrPut(value) {
+    irSymbolAssignment.next(tag)
+  }
+}

compiler/src/main/kotlin/gay/pizza/pork/compiler/LocalVariable.kt

@@ -0,0 +1,6 @@
+package gay.pizza.pork.compiler
+
+import gay.pizza.pork.ast.gen.Symbol
+import gay.pizza.pork.bir.IrSymbol
+
+data class LocalVariable(val symbol: IrSymbol, val name: Symbol)