JumpIfTrue

evilpie · evilpie · commit c4d443266666 · 2020-10-17T17:42:02.000+02:00
diff --git a/jit/src/instructions.rs b/jit/src/instructions.rs
@@ -180,6 +180,21 @@ impl<'a, 'b> FunctionCompiler<'a, 'b> {
 
                 Ok(())
             }
+            Instruction::JumpIfTrue { target } => {
+                let cond = self.stack.pop().ok_or(JitCompileError::BadBytecode)?;
+
+                let then_block = self.builder.create_block();
+                self.label_to_block.insert(*target, then_block);
+
+                let val = self.boolean_val(cond)?;
+                self.builder.ins().brnz(val, then_block, &[]);
+
+                let block = self.builder.create_block();
+                self.builder.ins().fallthrough(block, &[]);
+                self.builder.switch_to_block(block);
+
+                Ok(())
+            }
             Instruction::Jump { target } => {
                 let target_block = self.builder.create_block();
                 self.label_to_block.insert(*target, target_block);
diff --git a/jit/tests/bool_tests.rs b/jit/tests/bool_tests.rs
@@ -34,3 +34,20 @@ fn test_not() {
     assert_eq!(not_(true), Ok(false));
     assert_eq!(not_(false), Ok(true));
 }
+
+#[test]
+fn test_if_not() {
+    let if_not = jit_function! { if_not(a: bool) -> i64 => r##"
+        def if_not(a: bool):
+            if not a:
+                return 0
+            else:
+                return 1
+
+            return -1
+    "## };
+
+    assert_eq!(if_not(true), Ok(1));
+    assert_eq!(if_not(false), Ok(0));
+}
+
