nobu	2017-12-21 00:45:01 +0900 (Thu, 21 Dec 2017)

  New Revision: 61371

  https://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=revision&revision=61371

  Log:
    compile.c: branch on block param
    
    * compile.c (compile_branch_condition): replace block param value
      in branch conditions with `defined(yield)`, to get rid of
      creating proc value.

  Modified files:
    trunk/compile.c
Index: compile.c
===================================================================
--- compile.c	(revision 61370)
+++ compile.c	(revision 61371)
@@ -3367,6 +3367,21 @@ compile_flip_flop(rb_iseq_t *iseq, LINK_ https://github.com/ruby/ruby/blob/trunk/compile.c#L3367
 }
 
 static int
+local_block_param_p(rb_iseq_t *iseq, ID vid)
+{
+    int idx = iseq->body->local_iseq->body->local_table_size - get_local_var_idx(iseq, vid);
+    int level = get_lvar_level(iseq);
+    return iseq_local_block_param_p(iseq, idx, level);
+}
+
+static void
+compile_defined_yield(rb_iseq_t *iseq, LINK_ANCHOR *const ret, int line, VALUE needstr)
+{
+    ADD_INSN(ret, line, putnil);
+    ADD_INSN3(ret, line, defined, INT2FIX(DEFINED_YIELD), 0, needstr);
+}
+
+static int
 compile_branch_condition(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *cond,
 			 LABEL *then_label, LABEL *else_label)
 {
@@ -3414,6 +3429,11 @@ compile_branch_condition(rb_iseq_t *iseq https://github.com/ruby/ruby/blob/trunk/compile.c#L3429
       case NODE_DEFINED:
 	CHECK(compile_defined_expr(iseq, ret, cond, Qfalse));
 	goto branch;
+      case NODE_LVAR:
+	if (local_block_param_p(iseq, cond->nd_vid)) {
+	    compile_defined_yield(iseq, ret, nd_line(cond), Qfalse);
+	    goto branch;
+	}
       default:
 	CHECK(COMPILE(ret, "branch condition", cond));
       branch:
@@ -4150,9 +4170,7 @@ defined_expr0(rb_iseq_t *iseq, LINK_ANCH https://github.com/ruby/ruby/blob/trunk/compile.c#L4170
       }
 
       case NODE_YIELD:
-	ADD_INSN(ret, nd_line(node), putnil);
-	ADD_INSN3(ret, nd_line(node), defined, INT2FIX(DEFINED_YIELD), 0,
-		  needstr);
+	compile_defined_yield(iseq, ret, nd_line(node), needstr);
 	return 1;
 
       case NODE_BACK_REF:

--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/