mirror of https://github.com/irmen/prog8.git
58 lines
1.9 KiB
Kotlin
58 lines
1.9 KiB
Kotlin
package prog8tests.ast
|
|
|
|
import io.kotest.assertions.fail
|
|
import io.kotest.core.spec.style.FunSpec
|
|
import io.kotest.matchers.ints.shouldBeGreaterThan
|
|
import io.kotest.matchers.shouldBe
|
|
import io.kotest.matchers.shouldNotBe
|
|
import prog8.code.ast.PtVariable
|
|
import prog8.code.core.CbmPrgLauncherType
|
|
import prog8.code.core.CompilationOptions
|
|
import prog8.code.core.OutputType
|
|
import prog8.code.core.ZeropageType
|
|
import prog8.code.target.C64Target
|
|
import prog8.compiler.IntermediateAstMaker
|
|
import prog8tests.helpers.compileText
|
|
|
|
class TestIntermediateAst: FunSpec({
|
|
|
|
test("creation") {
|
|
val text="""
|
|
%import textio
|
|
%import graphics
|
|
main {
|
|
sub start() {
|
|
ubyte cc
|
|
ubyte[] array = [1,2,3]
|
|
cc = 11 in array
|
|
cc = cc |> sin8u() |> cos8u()
|
|
}
|
|
}
|
|
"""
|
|
val result = compileText(C64Target(), false, text, writeAssembly = false)!!
|
|
val options = CompilationOptions(
|
|
OutputType.PRG,
|
|
CbmPrgLauncherType.BASIC,
|
|
ZeropageType.BASICSAFE,
|
|
emptyList(),
|
|
floats = true,
|
|
noSysInit = false,
|
|
compTarget = C64Target()
|
|
)
|
|
val ast = IntermediateAstMaker(result.program, options).transform()
|
|
ast.name shouldBe result.program.name
|
|
ast.allBlocks().any() shouldBe true
|
|
val entry = ast.entrypoint() ?: fail("no main.start() found")
|
|
entry.name shouldBe "start"
|
|
entry.scopedName shouldBe listOf("main", "start")
|
|
val blocks = ast.allBlocks().toList()
|
|
blocks.size shouldBeGreaterThan 1
|
|
blocks[0].name shouldBe "main"
|
|
blocks[0].scopedName shouldBe listOf("main")
|
|
val ccdecl = entry.children[0] as PtVariable
|
|
ccdecl.name shouldBe "cc"
|
|
ccdecl.scopedName shouldBe listOf("main", "start", "cc")
|
|
ast.print()
|
|
}
|
|
|
|
}) |