From a22aa808e01d34de9a5ade59b00d9cd4d3351e2b Mon Sep 17 00:00:00 2001
From: CPunch <sethtstubbs@gmail.com>
Date: Fri, 26 Aug 2022 01:18:24 -0500
Subject: [PATCH] lp: added support for OP_TEST

---
 lparser.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/lparser.py b/lparser.py
index d0ed1b8..2c0d4df 100644
--- a/lparser.py
+++ b/lparser.py
@@ -231,7 +231,8 @@ class LuaDecomp:
     def __emitOperand(self, a: int, b: str, c: str, op: str) -> None:
         self.__setReg(a, "(" + b + op + c + ")")
 
-    def __compJmp(self, op: str):
+    # handles conditional jumps
+    def __condJmp(self, op: str, rkBC: bool = True):
         instr = self.__getCurrInstr()
         jmpType = "if"
         scopeStart = "then"
@@ -254,7 +255,13 @@ class LuaDecomp:
             self.__addExpr("%s not " % jmpType)
         else:
             self.__addExpr("%s " % jmpType)
-        self.__addExpr(self.__readRK(instr.B) + op + self.__readRK(instr.C) + " ")
+
+        # write actual comparison
+        if rkBC:
+            self.__addExpr(self.__readRK(instr.B) + op + self.__readRK(instr.C) + " ")
+        else: # just testing rkB
+            self.__addExpr(op + self.__readRK(instr.B))
+
         self.pc += 1 # skip next instr
         if scopeStart:
             self.__startScope("%s " % scopeStart, self.pc - 1, jmp)
@@ -368,11 +375,16 @@ class LuaDecomp:
         elif instr.opcode == Opcodes.JMP:
             pass
         elif instr.opcode == Opcodes.EQ:
-            self.__compJmp(" == ")
+            self.__condJmp(" == ")
         elif instr.opcode == Opcodes.LT:
-            self.__compJmp(" < ")
+            self.__condJmp(" < ")
         elif instr.opcode == Opcodes.LE:
-            self.__compJmp(" <= ")
+            self.__condJmp(" <= ")
+        elif instr.opcode == Opcodes.TEST:
+            if instr.C == 0:
+                self.__condJmp("", False)
+            else:
+                self.__condJmp("not ", False)
         elif instr.opcode == Opcodes.CALL:
             preStr = ""
             callStr = ""