Add comparison methods

gte/turmit.py

@@ -238,7 +238,49 @@ class Turmit(object):
         pass
     
     @RpnOp
-    def jmp(self, a):
+    def jmp(self, offset):
         ''' @brief Increments IP
+            @param offset int : jump offset symbols
         '''
-        self.__ip += a
+        self.__ip += offset
+
+    @RpnOp
+    def jz(self, a, offset):
+        ''' @brief Increments IP if a == 0
+            @param a int : operand
+            @param offset int : jump offset symbols
+        '''
+        if a == 0:
+            self.__ip += offset
+
+    @RpnOp
+    def jnz(self, a, offset):
+        ''' @brief Increments IP if a != 0
+            @param a int : operand
+            @param offset int : jump offset symbols
+        '''
+        if a != 0:
+            self.__ip += offset
+
+    @RpnOp
+    def jcmp(self, a, b, cnd, offset):
+        ''' @brief Increments IP if a == b
+
+            Condition type is encoded on an int : 
+            - 0b0000 : je
+            - 0b0001 : jne
+            - 0b0010 : jle
+            - 0b0011 : jl
+            - 0b0100 : jg
+            - 0b0101 : jge
+            - 0b0110 : jne
+            - 0b0111 : jmp
+
+            @param a int : operand
+            @param b int : operand
+            @param cnd int : condition type
+            @param offset int : jump offset symbols
+        '''
+        jmp = False
+        if cnd & 1 and a == b or cnd & 0b10 and a < b or cnd & 0b100 and a > b:
+            self.__ip += offset