Apple-2-SW/prog8
1
0
Fork 0
mirror of https://github.com/irmen/prog8.git synced 2024-10-19 07:23:56 +00:00
Code Issues Projects Releases Wiki Activity

loop todos

Browse Source
This commit is contained in:
Irmen de Jong 2020-07-30 02:54:37 +02:00
parent fbe3ce008b
commit bd7f75c130
2 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
compiler/src/prog8/compiler/target/c64/codegen/AsmGen.kt
View File

@ -740,38 +740,40 @@ internal class AsmGen(private val program: Program,
loopContinueLabels.pop() loopContinueLabels.pop()
} }
private fun repeatWordCountInAY(counterLabel: String, repeatLabel: String, endLabel: String, body: AnonymousScope) { private fun repeatWordCountInAY(counterVar: String, repeatLabel: String, endLabel: String, body: AnonymousScope) {
// note: A/Y must have been loaded with the number of iterations already! // note: A/Y must have been loaded with the number of iterations already!
// TODO allocate word counterVar on zeropage preferrably
out(""" out("""
sta $counterLabel sta $counterVar
sty $counterLabel+1 sty $counterVar+1
$repeatLabel lda $counterLabel $repeatLabel lda $counterVar
bne + bne +
lda $counterLabel+1 lda $counterVar+1
beq $endLabel beq $endLabel
+ lda $counterLabel + lda $counterVar
bne + bne +
dec $counterLabel+1 dec $counterVar+1
+ dec $counterLabel + dec $counterVar
""") """)
translate(body) translate(body)
out(""" out("""
jmp $repeatLabel jmp $repeatLabel
$counterLabel .word 0 $counterVar .word 0
$endLabel""") $endLabel""")
} }
private fun repeatByteCountInA(counterLabel: String, repeatLabel: String, endLabel: String, body: AnonymousScope) { private fun repeatByteCountInA(counterVar: String, repeatLabel: String, endLabel: String, body: AnonymousScope) {
// note: A must have been loaded with the number of iterations already! // note: A must have been loaded with the number of iterations already!
// TODO allocate word counterVar on zeropage preferrably
out(""" out("""
sta $counterLabel sta $counterVar
$repeatLabel lda $counterLabel $repeatLabel lda $counterVar
beq $endLabel beq $endLabel
dec $counterLabel""") dec $counterVar""")
translate(body) translate(body)
out(""" out("""
jmp $repeatLabel jmp $repeatLabel
$counterLabel .byte 0 $counterVar .byte 0
$endLabel""") $endLabel""")
} }

9
compiler/src/prog8/compiler/target/c64/codegen/ForLoopsAsmGen.kt
View File

@ -14,9 +14,10 @@ import prog8.compiler.target.c64.C64MachineDefinition.ESTACK_LO_PLUS1_HEX
import prog8.compiler.toHex import prog8.compiler.toHex
import kotlin.math.absoluteValue import kotlin.math.absoluteValue
// todo optimized loop code for common simple cases 0..N, N..0, >=1..N , N..>=1 where N<=255
// todo choose more efficient comparisons to avoid needless lda's // todo choose more efficient comparisons to avoid needless lda's
// todo optimize common case when step == 2 or -2 // todo optimize common case when step == 2 or -2
// todo allocate loop counter variable dynamically, preferrably on zeropage
internal class ForLoopsAsmGen(private val program: Program, private val asmgen: AsmGen) { internal class ForLoopsAsmGen(private val program: Program, private val asmgen: AsmGen) {
@ -277,7 +278,7 @@ $continueLabel inc $loopLabel+1
$endLabel""") $endLabel""")
} }
DataType.ARRAY_UB, DataType.ARRAY_B -> { DataType.ARRAY_UB, DataType.ARRAY_B -> {
// TODO: optimize loop code when the length of the array is < 256, don't need a separate counter var in such cases // TODO: optimize loop code when the length of the array is < 256
val length = decl.arraysize!!.size()!! val length = decl.arraysize!!.size()!!
val counterLabel = asmgen.makeLabel("for_counter") val counterLabel = asmgen.makeLabel("for_counter")
val modifiedLabel = asmgen.makeLabel("for_modified") val modifiedLabel = asmgen.makeLabel("for_modified")
@ -301,7 +302,7 @@ $counterLabel .byte 0
$endLabel""") $endLabel""")
} }
DataType.ARRAY_W, DataType.ARRAY_UW -> { DataType.ARRAY_W, DataType.ARRAY_UW -> {
// TODO: optimize loop code when the length of the array is < 256, don't need a separate counter var in such cases // TODO: optimize loop code when the length of the array is < 256
val length = decl.arraysize!!.size()!! * 2 val length = decl.arraysize!!.size()!! * 2
val counterLabel = asmgen.makeLabel("for_counter") val counterLabel = asmgen.makeLabel("for_counter")
val modifiedLabel = asmgen.makeLabel("for_modified") val modifiedLabel = asmgen.makeLabel("for_modified")
@ -343,7 +344,7 @@ $endLabel""")
} }
private fun translateForOverConstRange(stmt: ForLoop, iterableDt: DataType, range: IntProgression) { private fun translateForOverConstRange(stmt: ForLoop, iterableDt: DataType, range: IntProgression) {
// TODO: optimize loop code when the range is < 256 iterations, don't need a separate counter var in such cases // TODO: optimize loop code when the range is < 256 iterations
if (range.isEmpty()) if (range.isEmpty())
throw AssemblyError("empty range") throw AssemblyError("empty range")
val loopLabel = asmgen.makeLabel("for_loop") val loopLabel = asmgen.makeLabel("for_loop")