Add a random symbol generator + add some variables

gte/turmit.py

@@ -1,9 +1,16 @@
 import collections
 import inspect
+import random
 
 _op_list = collections.OrderedDict()
 _var_list = collections.OrderedDict()
 
+_var_list['x'] = 0
+_var_list['y'] = 0
+_var_list['r'] = 0
+_var_list['g'] = 0
+_var_list['b'] = 0
+
 def RpnOp(method):
     ''' @brief Decorator for RPN operation that autodetect argument count
         
@@ -48,7 +55,7 @@ class RpnSymbol(object):
         if optype == self.OPERATION:
             self.value = list(_op_list.keys())[value % len(_op_list)]
         elif optype == self.VARIABLE:
-            self.value %= len(_var_list)
+            self.value = list(_var_list.keys())[value % len(_var_list)]
 
     def __str__(self):
         if self.optype == self.OPERATION:
@@ -56,7 +63,15 @@ class RpnSymbol(object):
         elif self.optype == self.VALUE:
             return '0x%04X' % self.value
         else:
-            return _var_list[self.value]
+            return self.value.upper()
+
+    @classmethod
+    def random(cls):
+        optype = [cls.OPERATION, cls.VALUE, cls.VARIABLE]
+        optype = optype[random.randint(0,2)]
+        return cls(random.randint(0, 0xFFFF), optype)
+
+
 
 class Turmit(object):
     ''' @brief Represent a turmit that act given an RPN expression with an