Added low to string translation

Author: RedDocMD

src/main/kotlin/deep/decaf/low/amd64/LowIRClasses.kt

@@ -6,43 +6,94 @@ import java.util.*
 fun getUUID(): String = UUID.randomUUID().toString().replace("-", "")
 
 sealed class Offset
-data class NumberOffset(val offset: Int) : Offset()
-data class StringOffset(val name: String) : Offset()
+data class NumberOffset(val offset: Int) : Offset() {
+    override fun toString(): String {
+        return "$offset"
+    }
+}
+
+data class StringOffset(val name: String) : Offset() {
+    override fun toString(): String {
+        return name
+    }
+}
 
 
 sealed class Location
-data class ImmediateVal(val num: Int) : Location()
-data class Register(val name: String, val offset: Offset?) : Location() {
+data class ImmediateVal(val num: Int) : Location() {
+    override fun toString(): String {
+        return "\$$num"
+    }
+}
+
+data class Register(val name: String) : Location() {
     companion object {
-        fun basePointer() = Register("rbp", null)
-        fun stackPointer() = Register("rsp", null)
-        fun instructionPointer() = Register("rip", null)
-        fun returnRegister() = Register("rax", null)
-        fun r10() = Register("r10", null)
-        fun r10b() = Register("r10b", null)
-        fun r11() = Register("r11", null)
-        fun rax() = Register("rax", null)
-        fun rdx() = Register("rdx", null)
+        fun basePointer() = Register("rbp")
+        fun stackPointer() = Register("rsp")
+        fun instructionPointer() = Register("rip")
+        fun returnRegister() = Register("rax")
+        fun r10() = Register("r10")
+        fun r10b() = Register("r10b")
+        fun r11() = Register("r11")
+        fun rax() = Register("rax")
+        fun rdx() = Register("rdx")
+    }
+
+    override fun toString(): String {
+        return "%$name"
     }
 }
 
-data class Label(val label: String) : Location()
-data class MemLoc(val reg: Register, val offset: NumberOffset) : Location()
-data class ArrayAsm(val name: String, val offsetRegister: Register) : Location()
+data class Label(val label: String) : Location() {
+    override fun toString(): String {
+        return label
+    }
+}
+
+data class MemLoc(val reg: Register, val offset: NumberOffset) : Location() {
+    override fun toString(): String {
+        return "$offset($reg)"
+    }
+}
+
+data class ArrayAsm(val name: String, val offsetRegister: Register) : Location() {
+    override fun toString(): String {
+        return "$name(, $offsetRegister, 8)"
+    }
+}
 
 enum class AsmCMoveOp {
     CMOVE,
     CMOVNE,
     CMOVG,
     CMOVL,
     CMOVGE,
-    CMOVLE
+    CMOVLE;
+
+    override fun toString(): String {
+        return when (this) {
+            CMOVE -> "cmove"
+            CMOVNE -> "cmovne"
+            CMOVG -> "cmovg"
+            CMOVL -> "cmovl"
+            CMOVGE -> "cmovge"
+            CMOVLE -> "cmovle"
+        }
+    }
 }
 
 enum class AsmJumpOp {
     JMP,
     JE,
-    JNE
+    JNE;
+
+    override fun toString(): String {
+        return when (this) {
+            JMP -> "jmp"
+            JE -> "je"
+            JNE -> "jne"
+        }
+    }
 }
 
 enum class SetType {
@@ -51,30 +102,126 @@ enum class SetType {
     SETG,
     SETL,
     SETGE,
-    SETLE
+    SETLE;
+
+    override fun toString(): String {
+        return when (this) {
+            SETE -> "sete"
+            SETNE -> "setne"
+            SETG -> "setg"
+            SETL -> "setl"
+            SETGE -> "setge"
+            SETLE -> "setle"
+        }
+    }
 }
 
 sealed class Statement
-data class AddStatement(val src: Location, val dest: Location) : Statement()
-data class SubStatement(val src: Location, val dest: Location) : Statement()
-data class IMulStatement(val src: Location, val dest: Location) : Statement()
-data class IDivStatement(val src: Location) : Statement()
-data class CmpStatement(val src: Location, val dest: Location) : Statement()
-data class AndStatement(val src: Location, val dest: Location): Statement()
-data class OrStatement(val src: Location, val dest: Location): Statement()
-data class NotStatement(val src: Location) : Statement()
-data class NegStatement(val src: Location) : Statement()
-data class JumpStatement(val type: AsmJumpOp, val target: String) : Statement()
-data class MoveStatement(val src: Location, val dest: Location) : Statement()
-data class CMoveStatement(val type: AsmCMoveOp, val src: Register, val dest: Register) : Statement()
-data class SetStatement(val type: SetType, val reg: Register) : Statement()
-object SignedExtendStatement : Statement()
-object ReturnStatement : Statement()
-data class CallStatement(val label: String) : Statement()
-data class PushStatement(val src: Location) : Statement()
-data class PopStatement(val src: Location?) : Statement()
-object LeaveStatement : Statement()
-data class EnterStatement(val size: ImmediateVal) : Statement()
+data class AddStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "add $src, $dest"
+    }
+}
+
+data class SubStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "sub $src, $dest"
+    }
+}
+
+data class IMulStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "imul $src, $dest"
+    }
+}
+
+data class IDivStatement(val src: Location) : Statement() {
+    override fun toString(): String {
+        return "idivq $src"
+    }
+}
+
+data class CmpStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "cmp $src, $dest"
+    }
+}
+
+data class AndStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "and $src, $dest"
+    }
+}
+
+data class OrStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "or $src, $dest"
+    }
+}
+
+data class NotStatement(val src: Location) : Statement() {
+    override fun toString(): String {
+        return "not $src"
+    }
+}
+
+data class NegStatement(val src: Location) : Statement() {
+    override fun toString(): String {
+        return "nge $src"
+    }
+}
+
+data class JumpStatement(val type: AsmJumpOp, val target: String) : Statement() {
+    override fun toString(): String {
+        return "$type $target"
+    }
+}
+
+data class MoveStatement(val src: Location, val dest: Location) : Statement() {
+    override fun toString(): String {
+        return "mov $src, $dest"
+    }
+}
+
+data class CMoveStatement(val type: AsmCMoveOp, val src: Register, val dest: Register) : Statement() {
+    override fun toString(): String {
+        return "$type $src $dest"
+    }
+}
+
+data class SetStatement(val type: SetType, val reg: Register) : Statement() {
+    override fun toString(): String {
+        return "$type $reg"
+    }
+}
+
+object SignedExtendStatement : Statement() {
+    override fun toString() = "cqto"
+}
+
+object ReturnStatement : Statement() {
+    override fun toString() = "ret"
+}
+
+data class CallStatement(val label: String) : Statement() {
+    override fun toString() = "call $label"
+}
+
+data class PushStatement(val src: Location) : Statement() {
+    override fun toString() = "push $src"
+}
+
+data class PopStatement(val src: Location?) : Statement() {
+    override fun toString() = "pop ${src ?: ""}"
+}
+
+object LeaveStatement : Statement() {
+    override fun toString() = "leave"
+}
+
+data class EnterStatement(val size: ImmediateVal) : Statement() {
+    override fun toString() = "enter \$($size*8), $0"
+}
 
 data class Block(val label: String?, val statements: List<Statement>)