Added comparison.

1 files changed, 18 insertions, 2 deletions

diff --git a/nodes.py b/nodes.py
index b87022e..da2ab6c 100644
--- a/nodes.py
+++ b/nodes.py
@@ -83,13 +83,29 @@ class IfStatement(Statement):
     def get_falseblock(self):
         return self.falseblock
 
+class Comparison(Statement):
+    def __init__(self, first, ctype, second, lineno):
+        Statement.__init__(self, lineno)
+        self.first = first
+        self.ctype = ctype
+        self.second = second
+    
+    def get_first(self):
+        return self.first
+
+    def get_ctype(self):
+        return self.ctype
+
+    def get_second(self):
+        return self.second
+
 class StringStatement(Statement):
     def __init__(self, value, lineno):
         assert(type(value) == type(''))
         Statement.__init__(self, lineno)
         self.value = value
-        
-    def get_string(self):
+
+    def get_value(self):
         return self.value
 
 class FunctionCall(Statement):