You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
package dev.openrs2.deob.ast.transform
|
|
|
|
|
|
|
|
import com.github.javaparser.ast.CompilationUnit
|
|
|
|
import com.github.javaparser.ast.Node
|
|
|
|
import com.github.javaparser.ast.expr.ConditionalExpr
|
|
|
|
import dev.openrs2.deob.ast.util.countNots
|
|
|
|
import dev.openrs2.deob.ast.util.not
|
|
|
|
import dev.openrs2.deob.ast.util.walk
|
|
|
|
|
|
|
|
class TernaryTransformer : Transformer() {
|
|
|
|
override fun transform(unit: CompilationUnit) {
|
|
|
|
unit.walk(Node.TreeTraversal.POSTORDER) { expr: ConditionalExpr ->
|
|
|
|
val condition = expr.condition
|
|
|
|
val notCondition = condition.not()
|
|
|
|
if (notCondition.countNots() >= condition.countNots()) {
|
|
|
|
return@walk
|
|
|
|
}
|
|
|
|
|
|
|
|
val thenExpr = expr.thenExpr
|
|
|
|
val elseExpr = expr.elseExpr
|
|
|
|
|
|
|
|
expr.condition = notCondition
|
|
|
|
|
|
|
|
expr.thenExpr = elseExpr.clone()
|
|
|
|
expr.elseExpr = thenExpr.clone()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|