compiler: add support for not, and, or, xor

1 files changed, 32 insertions, 0 deletions

diff --git a/src/forth/compiler.rs b/src/forth/compiler.rs
index 5a6d502..2e301b3 100644
--- a/src/forth/compiler.rs
+++ b/src/forth/compiler.rs
@@ -229,6 +229,10 @@ impl<'a> Compiler<'a> {
                     ">=" => self.bc_push(OpCode::GTE, anno),
                     "<" => self.bc_push(OpCode::LT, anno),
                     "<=" => self.bc_push(OpCode::LTE, anno),
+                    "not" => self.bc_push(OpCode::Not, anno),
+                    "and" => self.bc_push(OpCode::And, anno),
+                    "or" => self.bc_push(OpCode::Or, anno),
+                    "xor" => self.bc_push(OpCode::Xor, anno),
                     other => return Err(CompileError::UnknownWord(String::from(other))),
                 }
             }
@@ -297,6 +301,34 @@ mod tests {
     }
 
     #[test]
+    fn not_opcode() {
+        let comp = compiler_for("not\n");
+        let main = &comp.wordlist.0[0];
+        assert_eq!(main, vec![OpCode::Not]);
+    }
+
+    #[test]
+    fn and_opcode() {
+        let comp = compiler_for("and\n");
+        let main = &comp.wordlist.0[0];
+        assert_eq!(main, vec![OpCode::And]);
+    }
+
+    #[test]
+    fn or_opcode() {
+        let comp = compiler_for("or\n");
+        let main = &comp.wordlist.0[0];
+        assert_eq!(main, vec![OpCode::Or]);
+    }
+
+    #[test]
+    fn xor_opcode() {
+        let comp = compiler_for("xor\n");
+        let main = &comp.wordlist.0[0];
+        assert_eq!(main, vec![OpCode::Xor]);
+    }
+
+    #[test]
     fn def_word() {
         let comp = compiler_for(": add2 2 + ; 3 add2\n");
         let main = &comp.wordlist.0[0];