bir: mutable values and inspection

bir/src/main/kotlin/gay/pizza/pork/bir/IrBreak.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrBreak(val target: IrSymbol) : IrCodeElement {
+data class IrBreak(var target: IrSymbol) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(target)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrCall.kt

@@ -4,10 +4,10 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrCall(
-  override val target: IrSymbol,
-  val arguments: List<IrCodeElement>,
-  val variableArguments: List<IrCodeElement>?
-) : IrCodeElement, IrSymbolUser {
+  override var target: IrSymbol,
+  var arguments: List<IrCodeElement>,
+  var variableArguments: List<IrCodeElement>?
+) : IrCodeElement(), IrSymbolUser {
   override fun crawl(block: (IrElement) -> Unit) {
     block(target)
     arguments.forEach(block)

bir/src/main/kotlin/gay/pizza/pork/bir/IrCodeBlock.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrCodeBlock(val items: List<IrCodeElement>) : IrCodeElement {
+data class IrCodeBlock(var items: List<IrCodeElement>) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     items.forEach(block)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrCodeElement.kt

@@ -3,4 +3,4 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-sealed interface IrCodeElement : IrElement
+sealed class IrCodeElement : IrElement()

bir/src/main/kotlin/gay/pizza/pork/bir/IrConditional.kt

@@ -4,10 +4,10 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrConditional(
-  val conditional: IrCodeElement,
-  val ifTrue: IrCodeElement,
-  val ifFalse: IrCodeElement
-) : IrCodeElement {
+  var conditional: IrCodeElement,
+  var ifTrue: IrCodeElement,
+  var ifFalse: IrCodeElement
+) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(conditional)
     block(ifTrue)

bir/src/main/kotlin/gay/pizza/pork/bir/IrConstant.kt

@@ -3,24 +3,24 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-sealed interface IrConstant : IrCodeElement {
+sealed class IrConstant : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {}
 }
 
 @Serializable
-data class IrIntegerConstant(val value: Int) : IrConstant
+data class IrIntegerConstant(var value: Int) : IrConstant()
 
 @Serializable
-data class IrLongConstant(val value: Long) : IrConstant
+data class IrLongConstant(var value: Long) : IrConstant()
 
 @Serializable
-data class IrDoubleConstant(val value: Double) : IrConstant
+data class IrDoubleConstant(var value: Double) : IrConstant()
 
 @Serializable
-data class IrStringConstant(val value: String) : IrConstant
+data class IrStringConstant(var value: String) : IrConstant()
 
 @Serializable
-data class IrBooleanConstant(val value: Boolean) : IrConstant
+data class IrBooleanConstant(var value: Boolean) : IrConstant()
 
 @Serializable
-data object IrNoneConstant : IrConstant
+data object IrNoneConstant : IrConstant()

bir/src/main/kotlin/gay/pizza/pork/bir/IrContinue.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrContinue(override val target: IrSymbol) : IrCodeElement, IrSymbolUser {
+data class IrContinue(override var target: IrSymbol) : IrCodeElement(), IrSymbolUser {
   override fun crawl(block: (IrElement) -> Unit) {
     block(target)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrDeclare.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrDeclare(override val symbol: IrSymbol, val value: IrCodeElement) : IrCodeElement, IrSymbolOwner {
+data class IrDeclare(override var symbol: IrSymbol, var value: IrCodeElement) : IrCodeElement(), IrSymbolOwner {
   override fun crawl(block: (IrElement) -> Unit) {
     value.crawl(block)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrDefinition.kt

@@ -4,11 +4,11 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrDefinition(
-  override val symbol: IrSymbol,
-  val type: IrDefinitionType,
-  val arguments: List<IrFunctionArgument>,
-  val code: IrCodeBlock
-) : IrElement, IrSymbolOwner {
+  override var symbol: IrSymbol,
+  var type: IrDefinitionType,
+  var arguments: List<IrFunctionArgument>,
+  var code: IrCodeBlock
+) : IrElement(), IrSymbolOwner {
   override fun crawl(block: (IrElement) -> Unit) {
     block(symbol)
     arguments.forEach(block)

bir/src/main/kotlin/gay/pizza/pork/bir/IrElement.kt

@@ -3,6 +3,6 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-sealed interface IrElement {
-  fun crawl(block: (IrElement) -> Unit)
+sealed class IrElement {
+  abstract fun crawl(block: (IrElement) -> Unit)
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrFunctionArgument.kt

@@ -3,6 +3,6 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrFunctionArgument(override val symbol: IrSymbol) : IrSymbolOwner {
+data class IrFunctionArgument(override var symbol: IrSymbol) : IrElement(), IrSymbolOwner {
   override fun crawl(block: (IrElement) -> Unit) {}
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrIndex.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrIndex(val data: IrCodeElement, val index: IrCodeElement) : IrCodeElement {
+data class IrIndex(var data: IrCodeElement, var index: IrCodeElement) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(data)
     block(index)

bir/src/main/kotlin/gay/pizza/pork/bir/IrInfix.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrInfix(val op: IrInfixOp, val left: IrCodeElement, val right: IrCodeElement) : IrCodeElement {
+data class IrInfix(var op: IrInfixOp, var left: IrCodeElement, var right: IrCodeElement) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(left)
     block(right)

bir/src/main/kotlin/gay/pizza/pork/bir/IrList.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrList(val items: List<IrCodeElement>) : IrCodeElement {
+data class IrList(var items: List<IrCodeElement>) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     items.forEach(block)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrListSize.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrListSize(val list: IrCodeElement) : IrCodeElement {
+data class IrListSize(var list: IrCodeElement) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(list)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrLoad.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrLoad(override val target: IrSymbol) : IrCodeElement, IrSymbolUser {
+data class IrLoad(override var target: IrSymbol) : IrCodeElement(), IrSymbolUser {
   override fun crawl(block: (IrElement) -> Unit) {
     block(target)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrLoop.kt

@@ -4,10 +4,10 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrLoop(
-  override val symbol: IrSymbol,
-  val condition: IrCodeElement,
-  val inner: IrCodeElement
-) : IrCodeElement, IrSymbolOwner {
+  override var symbol: IrSymbol,
+  var condition: IrCodeElement,
+  var inner: IrCodeElement
+) : IrCodeElement(), IrSymbolOwner {
   override fun crawl(block: (IrElement) -> Unit) {
     block(symbol)
     block(condition)

bir/src/main/kotlin/gay/pizza/pork/bir/IrNativeDefinition.kt

@@ -3,6 +3,6 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrNativeDefinition(val form: String, val definitions: List<String>) : IrCodeElement {
+data class IrNativeDefinition(var form: String, var definitions: List<String>) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {}
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrPrefix.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrPrefix(val op: IrPrefixOp, val value: IrCodeElement) : IrCodeElement {
+data class IrPrefix(var op: IrPrefixOp, var value: IrCodeElement) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(value)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrReturn.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrReturn(val from: IrSymbol, val value: IrCodeElement) : IrCodeElement {
+data class IrReturn(var from: IrSymbol, var value: IrCodeElement) : IrCodeElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     block(from)
     block(value)

bir/src/main/kotlin/gay/pizza/pork/bir/IrSlab.kt

@@ -4,9 +4,10 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrSlab(
-  val location: IrSlabLocation,
-  val definitions: List<IrDefinition>
-) : IrElement {
+  override var symbol: IrSymbol,
+  var location: IrSlabLocation,
+  var definitions: List<IrDefinition>
+) : IrElement(), IrSymbolOwner {
   override fun crawl(block: (IrElement) -> Unit) {
     block(location)
     definitions.forEach(block)

bir/src/main/kotlin/gay/pizza/pork/bir/IrSlabLocation.kt

@@ -4,8 +4,8 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class IrSlabLocation(
-  val form: String,
-  val path: String
-) : IrElement {
+  var form: String,
+  var path: String
+) : IrElement() {
   override fun crawl(block: (IrElement) -> Unit) {}
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrStore.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrStore(override val target: IrSymbol, val value: IrCodeElement) : IrCodeElement, IrSymbolUser {
+data class IrStore(override var target: IrSymbol, var value: IrCodeElement) : IrCodeElement(), IrSymbolUser {
   override fun crawl(block: (IrElement) -> Unit) {
     value.crawl(block)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrSuffix.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrSuffix(val op: IrSuffixOp, override val target: IrSymbol) : IrCodeElement, IrSymbolUser {
+data class IrSuffix(var op: IrSuffixOp, override var target: IrSymbol) : IrCodeElement(), IrSymbolUser {
   override fun crawl(block: (IrElement) -> Unit) {
     block(target)
   }

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbol.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrSymbol(val id: UInt, val tag: IrSymbolTag, val name: String? = null) : IrElement {
+data class IrSymbol(val id: UInt, val tag: IrSymbolTag, val name: String? = null) : IrElement() {
   override fun crawl(block: (IrElement) -> Unit) {}
 
   override fun equals(other: Any?): Boolean {

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolGraph.kt

@@ -1,7 +1,10 @@
 package gay.pizza.pork.bir
 
+import kotlinx.serialization.Serializable
+
+@Serializable(with = IrSymbolGraphSerializer::class)
 class IrSymbolGraph {
-  private val edges = mutableSetOf<Pair<IrSymbolUser, IrSymbolOwner>>()
+  private val edges = mutableSetOf<IrSymbolGraphEdge>()
 
   private fun crawlForKnown(known: MutableMap<IrSymbol, IrSymbolOwner>, root: IrElement) {
     if (root is IrSymbolOwner) {
@@ -17,7 +20,7 @@ class IrSymbolGraph {
     if (root is IrSymbolUser) {
       val what = known[root.target]
       if (what != null) {
-        edges.add(root to what)
+        edges.add(IrSymbolGraphEdge(root, what))
       }
     }
 
@@ -26,12 +29,16 @@ class IrSymbolGraph {
     }
   }
 
-  fun crawl(root: IrElement) {
+  fun buildFromRoot(root: IrElement) {
     val known = mutableMapOf<IrSymbol, IrSymbolOwner>()
     crawlForKnown(known, root)
     crawlForAssociations(known, root)
   }
 
+  fun buildFromEdges(edges: Collection<IrSymbolGraphEdge>) {
+    this.edges.addAll(edges)
+  }
+
   fun forEachEdge(block: (IrSymbolUser, IrSymbolOwner) -> Unit) {
     for ((from, to) in edges) {
       block(from, to)

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolGraphEdge.kt

@@ -0,0 +1,6 @@
+package gay.pizza.pork.bir
+
+import kotlinx.serialization.Serializable
+
+@Serializable
+data class IrSymbolGraphEdge(val user: IrSymbolUser, val owner: IrSymbolOwner)

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolGraphSerializer.kt

@@ -0,0 +1,26 @@
+package gay.pizza.pork.bir
+
+import kotlinx.serialization.KSerializer
+import kotlinx.serialization.builtins.ListSerializer
+import kotlinx.serialization.descriptors.SerialDescriptor
+import kotlinx.serialization.encoding.Decoder
+import kotlinx.serialization.encoding.Encoder
+
+class IrSymbolGraphSerializer : KSerializer<IrSymbolGraph> {
+  private val serializer = ListSerializer(IrSymbolGraphEdge.serializer())
+  override val descriptor: SerialDescriptor = serializer.descriptor
+
+  override fun deserialize(decoder: Decoder): IrSymbolGraph {
+    val graph = IrSymbolGraph()
+    graph.buildFromEdges(serializer.deserialize(decoder))
+    return graph
+  }
+
+  override fun serialize(encoder: Encoder, value: IrSymbolGraph) {
+    val edges = mutableListOf<IrSymbolGraphEdge>()
+    value.forEachEdge { user, owner ->
+      edges.add(IrSymbolGraphEdge(user, owner))
+    }
+    serializer.serialize(encoder, edges)
+  }
+}

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolOwner.kt

@@ -3,6 +3,6 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-sealed interface IrSymbolOwner : IrElement {
-  val symbol: IrSymbol
+sealed interface IrSymbolOwner {
+  var symbol: IrSymbol
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolTag.kt

@@ -4,6 +4,7 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 enum class IrSymbolTag {
+  Slab,
   Function,
   Variable,
   Local,

bir/src/main/kotlin/gay/pizza/pork/bir/IrSymbolUser.kt

@@ -3,6 +3,6 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-sealed interface IrSymbolUser : IrElement {
-  val target: IrSymbol
+sealed interface IrSymbolUser {
+  var target: IrSymbol
 }

bir/src/main/kotlin/gay/pizza/pork/bir/IrWorld.kt

@@ -3,7 +3,7 @@ package gay.pizza.pork.bir
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class IrWorld(val slabs: List<IrSlab>) : IrElement {
+data class IrWorld(var slabs: List<IrSlab>) : IrElement() {
   override fun crawl(block: (IrElement) -> Unit) {
     slabs.forEach(block)
   }

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

@@ -6,7 +6,7 @@ import gay.pizza.pork.bir.*
 import gay.pizza.pork.frontend.scope.ScopeSymbol
 import gay.pizza.pork.frontend.scope.SlabScope
 
-class IrCodeEmitter(
+class AstIrEmitter(
   val self: IrSymbol,
   val irSymbolWorld: IrSymbolWorld<Any>,
   val irSymbolAssignment: IrSymbolAssignment,

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

@@ -4,6 +4,7 @@ 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.bir.IrSymbolTag
 import gay.pizza.pork.frontend.Slab
 
 class CompilableSlab(val compiler: Compiler, val slab: Slab) {
@@ -20,11 +21,12 @@ class CompilableSlab(val compiler: Compiler, val slab: Slab) {
   }
 
   private fun compileIrSlab(): IrSlab {
+    val slabSymbol = compiler.irSymbolAssignment.next(IrSymbolTag.Slab, slab.location.commonLocationIdentity)
     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)
+    return IrSlab(slabSymbol, irSlabLocation, definitions)
   }
 }

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

@@ -29,7 +29,7 @@ class CompilableSymbol(val compilableSlab: CompilableSlab, val scopeSymbol: Scop
   private fun compileIrDefinition(): IrDefinition {
     val compiler = compilableSlab.compiler
     val functionSymbol = compiler.irSymbolWorld.create(scopeSymbol, IrSymbolTag.Function, scopeSymbol.symbol.id)
-    val irCodeEmitter = IrCodeEmitter(
+    val irCodeEmitter = AstIrEmitter(
       self = functionSymbol,
       irSymbolWorld = compiler.irSymbolWorld,
       irSymbolAssignment = compiler.irSymbolAssignment,

tool/src/main/kotlin/gay/pizza/pork/tool/CompileCommand.kt

@@ -4,16 +4,26 @@ import com.charleskorn.kaml.PolymorphismStyle
 import com.charleskorn.kaml.Yaml
 import com.github.ajalt.clikt.core.CliktCommand
 import com.github.ajalt.clikt.parameters.arguments.argument
+import com.github.ajalt.clikt.parameters.options.flag
+import com.github.ajalt.clikt.parameters.options.option
 import gay.pizza.dough.fs.PlatformFsProvider
 import gay.pizza.pork.ast.gen.Symbol
+import gay.pizza.pork.bir.IrSymbolGraph
 import gay.pizza.pork.bir.IrWorld
 import gay.pizza.pork.bytecode.CompiledWorld
 import gay.pizza.pork.compiler.Compiler
 import gay.pizza.pork.minimal.FileTool
 
-class CompileCommand : CliktCommand(help = "Compile Pork to Bytecode", name = "compile") {
+class CompileCommand : CliktCommand(help = "Compile Pork", name = "compile") {
+  val showIrCode by option("--show-ir-code").flag(default = false)
+  val showIrSymbolGraph by option("--show-ir-symbol-graph").flag(default = false)
+
   val path by argument("file")
 
+  private val yaml = Yaml(configuration = Yaml.default.configuration.copy(
+    polymorphismStyle = PolymorphismStyle.Property
+  ))
+
   override fun run() {
     val tool = FileTool(PlatformFsProvider.resolve(path))
     val world = tool.buildWorld()
@@ -22,16 +32,22 @@ class CompileCommand : CliktCommand(help = "Compile Pork to Bytecode", name = "c
     val compiledSlab = compiler.compilableSlabs.of(slab)
     val compiledMain = compiledSlab.resolve(Symbol("main"))
       ?: throw RuntimeException("'main' function not found.")
-    val compiledWorld = compiler.compile(compiledMain)
     val irWorld = compiler.compileIrWorld()
-    printCompiledIr(irWorld)
+    val compiledWorld = compiler.compile(compiledMain)
+    if (showIrCode) {
+      printCompiledIr(irWorld)
+    }
+
+    if (showIrSymbolGraph) {
+      val irSymbolGraph = IrSymbolGraph()
+      irSymbolGraph.buildFromRoot(irWorld)
+      println(yaml.encodeToString(IrSymbolGraph.serializer(), irSymbolGraph))
+    }
+
     printCompiledWorld(compiledWorld)
   }
 
   private fun printCompiledIr(irWorld: IrWorld) {
-    val yaml = Yaml(configuration = Yaml.default.configuration.copy(
-      polymorphismStyle = PolymorphismStyle.Property
-    ))
     println(yaml.encodeToString(IrWorld.serializer(), irWorld))
   }