From dde9a8755fd40c3e7053da21de68a241bba4ef54 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Thu, 9 Nov 2017 23:08:01 +0000
Subject: iseq.c: operand lvar

* iseq.c (rb_insn_operand_intern): show local variable operand
  name in unified instructions.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@60732 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 iseq.c                    | 31 +++++++++++++++++++++----------
 template/optinsn.inc.tmpl | 24 ++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 10 deletions(-)

diff --git a/iseq.c b/iseq.c
index ac926b7a18..6f16be8525 100644
--- a/iseq.c
+++ b/iseq.c
@@ -1289,6 +1289,22 @@ id_to_name(ID id, VALUE default_value)
     return str;
 }
 
+static VALUE
+local_var_name(const rb_iseq_t *diseq, VALUE level, VALUE op)
+{
+    VALUE i;
+    ID lid;
+
+    for (i = 0; i < level; i++) {
+	diseq = diseq->body->parent_iseq;
+    }
+    lid = diseq->body->local_table[diseq->body->local_table_size +
+				   VM_ENV_DATA_SIZE - 1 - op];
+    return id_to_name(lid, INT2FIX('*'));
+}
+
+int rb_insn_unified_local_var_level(VALUE);
+
 VALUE
 rb_insn_operand_intern(const rb_iseq_t *iseq,
 		       VALUE insn, int op_no, VALUE op,
@@ -1308,23 +1324,18 @@ rb_insn_operand_intern(const rb_iseq_t *iseq,
 	break;
 
       case TS_LINDEX:{
+	int level;
 	if (insn == BIN(getlocal) || insn == BIN(setlocal)) {
 	    if (pnop) {
-		const rb_iseq_t *diseq = iseq;
-		VALUE level = *pnop, i;
-		ID lid;
-
-		for (i = 0; i < level; i++) {
-		    diseq = diseq->body->parent_iseq;
-		}
-		lid = diseq->body->local_table[diseq->body->local_table_size +
-					       VM_ENV_DATA_SIZE - 1 - op];
-		ret = id_to_name(lid, INT2FIX('*'));
+		ret = local_var_name(iseq, *pnop, op);
 	    }
 	    else {
 		ret = rb_sprintf("%"PRIuVALUE, op);
 	    }
 	}
+	else if ((level = rb_insn_unified_local_var_level(insn)) >= 0) {
+	    ret = local_var_name(iseq, (VALUE)level, op);
+	}
 	else {
 	    ret = rb_inspect(INT2FIX(op));
 	}
diff --git a/template/optinsn.inc.tmpl b/template/optinsn.inc.tmpl
index 186ec46982..b1fba6dea3 100644
--- a/template/optinsn.inc.tmpl
+++ b/template/optinsn.inc.tmpl
@@ -52,3 +52,27 @@ insn_operands_unification(INSN *insnobj)
     return insnobj;
 }
 
+int
+rb_insn_unified_local_var_level(VALUE insn)
+{
+#ifdef OPT_OPERANDS_UNIFICATION
+    /* optimize rule */
+    switch (insn) {
+% opt_insns_map.each do |originsn, optinsns|
+%   optinsns.each {|opti|
+      case BIN(<%=opti.name%>):
+%     opti.defopes.each {|opinfo|
+%       next if opinfo[1] == '*'
+	return <%=opinfo[1]%>;
+%       break
+%     }
+%   }
+% end
+
+      default:
+	/* do nothing */;
+	break;
+    }
+#endif
+    return -1;
+}
-- 
cgit v1.2.3