codegen.c: add short circuit constant folding for `NODE_AND`, `NODE_OR`.

7c6c4ebb · Yukihiro "Matz" Matsumoto · 863671c8 · 7c6c4ebb
Unverified Commit 7c6c4ebb authored Sep 24, 2021 by Yukihiro "Matz" Matsumoto
Hide whitespace changes
Inline Side-by-side

Showing with 16 additions and 0 deletions

mrbgems/mruby-compiler/core/codegen.c mrbgems/mruby-compiler/core/codegen.c +16 -0

No files found.
--- a/mrbgems/mruby-compiler/core/codegen.c
+++ b/mrbgems/mruby-compiler/core/codegen.c
@@ -2224,6 +2224,14 @@ codegen(codegen_scope *s, node *tree, int val)
    {
      uint32_t pos;

+      if (true_always(tree->car)) {
+        codegen(s, tree->cdr, val);
+        goto exit;
+      }
+      if (false_always(tree->car)) {
+        codegen(s, tree->car, val);
+        goto exit;
+      }
      codegen(s, tree->car, VAL);
      pop();
      pos = genjmp2_0(s, OP_JMPNOT, cursp(), val);
@@ -2236,6 +2244,14 @@ codegen(codegen_scope *s, node *tree, int val)
    {
      uint32_t pos;

+      if (true_always(tree->car)) {
+        codegen(s, tree->car, val);
+        goto exit;
+      }
+      if (false_always(tree->car)) {
+        codegen(s, tree->cdr, val);
+        goto exit;
+      }
      codegen(s, tree->car, VAL);
      pop();
      pos = genjmp2_0(s, OP_JMPIF, cursp(), val);