Added boolean data type.

1 files changed, 20 insertions, 0 deletions

diff --git a/nodes.py b/nodes.py
index d58729a..4bbaa5d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -27,6 +27,15 @@ class Expression(Node):
 class Statement(Node):
     pass
 
+class BoolExpression(Expression):
+    def __init__(self, value, lineno):
+        Expression.__init__(self, lineno)
+        self.value = value
+        assert(isinstance(value, bool))
+    
+    def get_value(self):
+        return self.value
+
 class AtomExpression(Expression):
     def __init__(self, value, lineno):
         Expression.__init__(self, lineno)
@@ -49,6 +58,15 @@ class AtomStatement(Statement):
     def get_value(self):
         return self.value
 
+class BoolStatement(Statement):
+    def __init__(self, value, lineno):
+        Statement.__init__(self, lineno)
+        assert(isinstance(value, bool))
+        self.value = value
+
+    def get_value(self):
+        return self.value
+
 class StringStatement(Statement):
     def __init__(self, value, lineno):
         assert(type(value) == type(''))
@@ -107,4 +125,6 @@ def statement_from_expression(expr):
         return AtomStatement(expr.value, expr.lineno())
     if isinstance(expr, StringExpression):
         return StringStatement(expr.value, expr.lineno())
+    if isinstance(expr, BoolExpression):
+        return BoolStatement(expr.value, expr.lineno())
     raise RuntimeError('Can not convert unknown expression to a statement.')