Apple-2-SW/prog8
1
0
Fork 0
mirror of https://github.com/irmen/prog8.git synced 2025-02-26 11:29:24 +00:00
Code Issues Projects Releases Wiki Activity

fix the "x<2" optimization made a few commits ago to only work on unsigned

Browse Source
This commit is contained in:
Irmen de Jong 2024-09-09 23:06:36 +02:00
parent 4e98fb75d6
commit 4a2dcd20d1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
compiler/src/prog8/compiler/astprocessing/VariousCleanups.kt
View File

@ -136,15 +136,15 @@ internal class VariousCleanups(val program: Program, val errors: IErrorReporter,
if(isMultiComparisonRecurse(leftBinExpr1)) {
val elementType = needle.inferType(program).getOrElse { throw FatalAstException("invalid needle dt") }
if(values.size==2 || values.size==3 && elementType in IntegerDatatypes) {
if(values.size==2 || values.size==3 && (elementType==DataType.UBYTE || elementType==DataType.UWORD)) {
val numbers = values.map{it.number}.toSet()
if(numbers == setOf(0.0, 1.0)) {
// we can replace x==0 or x==1 with x<2
// we can replace unsigned x==0 or x==1 with x<2
val compare = BinaryExpression(needle, "<", NumericLiteral(elementType, 2.0, expr.position), expr.position)
return listOf(IAstModification.ReplaceNode(expr, compare, parent))
}
if(numbers == setOf(0.0, 1.0, 2.0)) {
// we can replace x==0 or x==1 or x==2 with x<3
// we can replace unsigned x==0 or x==1 or x==2 with x<3
val compare = BinaryExpression(needle, "<", NumericLiteral(elementType, 3.0, expr.position), expr.position)
return listOf(IAstModification.ReplaceNode(expr, compare, parent))
}