From 504e8328724fb51a86ce0cae1bacd9f0b1b92dcf Mon Sep 17 00:00:00 2001
From: Graham <gpe@openrs2.dev>
Date: Sun, 26 Jul 2020 11:27:40 +0100
Subject: [PATCH] Improve AddSubTransformer

This commit improves support for simplifying a series of multiple
additions/subtractions and doing so in a single pass.

Signed-off-by: Graham <gpe@openrs2.dev>
---
 .../deob/ast/transform/AddSubTransformer.kt   | 77 ++++++++++++++-----
 1 file changed, 58 insertions(+), 19 deletions(-)

diff --git a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/AddSubTransformer.kt b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/AddSubTransformer.kt
index 5683b2cc..65dde0a6 100644
--- a/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/AddSubTransformer.kt
+++ b/deob-ast/src/main/java/dev/openrs2/deob/ast/transform/AddSubTransformer.kt
@@ -19,32 +19,71 @@ class AddSubTransformer : Transformer() {
     override fun transformUnit(group: LibraryGroup, library: Library, unit: CompilationUnit) {
         unit.walk { expr: BinaryExpr ->
             val op = expr.operator
-            val left = expr.left
-            val right = expr.right
-            val type = expr.calculateResolvedType()
+            if (op != BinaryExpr.Operator.PLUS && op != BinaryExpr.Operator.MINUS) {
+                return@walk
+            }
 
-            if (op == BinaryExpr.Operator.PLUS && type.isString()) {
+            val type = expr.calculateResolvedType()
+            if (type.isString()) {
                 return@walk
             }
 
-            if (op == BinaryExpr.Operator.PLUS && right.isNegative()) {
-                // x + -y => x - y
-                expr.operator = BinaryExpr.Operator.MINUS
-                expr.right = right.negate()
-            } else if (op == BinaryExpr.Operator.PLUS && left.isNegative()) {
-                if (expr.hasSideEffects()) {
-                    return@walk
+            val terms = mutableListOf<Expression>()
+            addTerms(terms, expr, negate = false)
+
+            terms.sortWith(Comparator { a, b ->
+                // preserve the order of adjacent expressions with side effects
+                val aHasSideEffects = a.hasSideEffects()
+                val bHasSideEffects = b.hasSideEffects()
+                if (aHasSideEffects && bHasSideEffects) {
+                    return@Comparator 0
                 }
 
-                // -x + y => y - x
-                expr.operator = BinaryExpr.Operator.MINUS
-                expr.left = right.clone()
-                expr.right = left.negate()
-            } else if (op == BinaryExpr.Operator.MINUS && right.isNegative()) {
-                // x - -y => x + y
-                expr.operator = BinaryExpr.Operator.PLUS
-                expr.right = right.negate()
+                // push negative expressions to the right so we can replace unary minus with binary minus
+                val aNegative = a.isNegative()
+                val bNegative = b.isNegative()
+                if (aNegative && !bNegative) {
+                    return@Comparator 1
+                } else if (!aNegative && bNegative) {
+                    return@Comparator -1
+                }
+
+                return@Comparator 0
+            })
+
+            val newExpr = terms.reduce { left, right ->
+                if (right.isNegative()) {
+                    BinaryExpr(left.clone(), right.negate(), BinaryExpr.Operator.MINUS)
+                } else {
+                    BinaryExpr(left.clone(), right.clone(), BinaryExpr.Operator.PLUS)
+                }
+            }
+
+            expr.replace(newExpr)
+        }
+    }
+
+    private fun addTerms(terms: MutableList<Expression>, expr: Expression, negate: Boolean) {
+        when {
+            expr is UnaryExpr -> when {
+                expr.operator == UnaryExpr.Operator.MINUS -> addTerms(terms, expr.expression, !negate)
+                negate -> terms += expr.negate()
+                else -> terms += expr
+            }
+            expr is BinaryExpr -> when {
+                expr.operator == BinaryExpr.Operator.PLUS -> {
+                    addTerms(terms, expr.left, negate)
+                    addTerms(terms, expr.right, negate)
+                }
+                expr.operator == BinaryExpr.Operator.MINUS -> {
+                    addTerms(terms, expr.left, negate)
+                    addTerms(terms, expr.right, !negate)
+                }
+                negate -> terms += expr.negate()
+                else -> terms += expr
             }
+            negate -> terms += expr.negate()
+            else -> terms += expr
         }
     }