Extract compile_super from iseq_compile_each0
This commit is contained in:
Nobuyoshi Nakada
parent
6cf9f17191
commit
996489d7e0
Notes:
git
2021-09-01 16:17:30 +09:00
272
compile.c
272
compile.c
@ -8479,6 +8479,144 @@ compile_op_log(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node,
|
|||||||
return COMPILE_OK;
|
return COMPILE_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static int
|
||||||
|
compile_super(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node, int popped, const enum node_type type)
|
||||||
|
{
|
||||||
|
struct rb_iseq_constant_body *const body = iseq->body;
|
||||||
|
DECL_ANCHOR(args);
|
||||||
|
int argc;
|
||||||
|
unsigned int flag = 0;
|
||||||
|
struct rb_callinfo_kwarg *keywords = NULL;
|
||||||
|
const rb_iseq_t *parent_block = ISEQ_COMPILE_DATA(iseq)->current_block;
|
||||||
|
|
||||||
|
INIT_ANCHOR(args);
|
||||||
|
ISEQ_COMPILE_DATA(iseq)->current_block = NULL;
|
||||||
|
if (type == NODE_SUPER) {
|
||||||
|
VALUE vargc = setup_args(iseq, args, node->nd_args, &flag, &keywords);
|
||||||
|
CHECK(!NIL_P(vargc));
|
||||||
|
argc = FIX2INT(vargc);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
/* NODE_ZSUPER */
|
||||||
|
int i;
|
||||||
|
const rb_iseq_t *liseq = body->local_iseq;
|
||||||
|
const struct rb_iseq_constant_body *const local_body = liseq->body;
|
||||||
|
const struct rb_iseq_param_keyword *const local_kwd = local_body->param.keyword;
|
||||||
|
int lvar_level = get_lvar_level(iseq);
|
||||||
|
|
||||||
|
argc = local_body->param.lead_num;
|
||||||
|
|
||||||
|
/* normal arguments */
|
||||||
|
for (i = 0; i < local_body->param.lead_num; i++) {
|
||||||
|
int idx = local_body->local_table_size - i;
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (local_body->param.flags.has_opt) {
|
||||||
|
/* optional arguments */
|
||||||
|
int j;
|
||||||
|
for (j = 0; j < local_body->param.opt_num; j++) {
|
||||||
|
int idx = local_body->local_table_size - (i + j);
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
}
|
||||||
|
i += j;
|
||||||
|
argc = i;
|
||||||
|
}
|
||||||
|
if (local_body->param.flags.has_rest) {
|
||||||
|
/* rest argument */
|
||||||
|
int idx = local_body->local_table_size - local_body->param.rest_start;
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
ADD_INSN1(args, node, splatarray, Qfalse);
|
||||||
|
|
||||||
|
argc = local_body->param.rest_start + 1;
|
||||||
|
flag |= VM_CALL_ARGS_SPLAT;
|
||||||
|
}
|
||||||
|
if (local_body->param.flags.has_post) {
|
||||||
|
/* post arguments */
|
||||||
|
int post_len = local_body->param.post_num;
|
||||||
|
int post_start = local_body->param.post_start;
|
||||||
|
|
||||||
|
if (local_body->param.flags.has_rest) {
|
||||||
|
int j;
|
||||||
|
for (j=0; j<post_len; j++) {
|
||||||
|
int idx = local_body->local_table_size - (post_start + j);
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
}
|
||||||
|
ADD_INSN1(args, node, newarray, INT2FIX(j));
|
||||||
|
ADD_INSN (args, node, concatarray);
|
||||||
|
/* argc is settled at above */
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
int j;
|
||||||
|
for (j=0; j<post_len; j++) {
|
||||||
|
int idx = local_body->local_table_size - (post_start + j);
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
}
|
||||||
|
argc = post_len + post_start;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (local_body->param.flags.has_kw) { /* TODO: support keywords */
|
||||||
|
int local_size = local_body->local_table_size;
|
||||||
|
argc++;
|
||||||
|
|
||||||
|
ADD_INSN1(args, node, putspecialobject, INT2FIX(VM_SPECIAL_OBJECT_VMCORE));
|
||||||
|
|
||||||
|
if (local_body->param.flags.has_kwrest) {
|
||||||
|
int idx = local_body->local_table_size - local_kwd->rest_start;
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
if (local_kwd->num > 0) {
|
||||||
|
ADD_SEND (args, node, rb_intern("dup"), INT2FIX(0));
|
||||||
|
flag |= VM_CALL_KW_SPLAT_MUT;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
ADD_INSN1(args, node, newhash, INT2FIX(0));
|
||||||
|
flag |= VM_CALL_KW_SPLAT_MUT;
|
||||||
|
}
|
||||||
|
for (i = 0; i < local_kwd->num; ++i) {
|
||||||
|
ID id = local_kwd->table[i];
|
||||||
|
int idx = local_size - get_local_var_idx(liseq, id);
|
||||||
|
ADD_INSN1(args, node, putobject, ID2SYM(id));
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
}
|
||||||
|
ADD_SEND(args, node, id_core_hash_merge_ptr, INT2FIX(i * 2 + 1));
|
||||||
|
if (local_body->param.flags.has_rest) {
|
||||||
|
ADD_INSN1(args, node, newarray, INT2FIX(1));
|
||||||
|
ADD_INSN (args, node, concatarray);
|
||||||
|
--argc;
|
||||||
|
}
|
||||||
|
flag |= VM_CALL_KW_SPLAT;
|
||||||
|
}
|
||||||
|
else if (local_body->param.flags.has_kwrest) {
|
||||||
|
int idx = local_body->local_table_size - local_kwd->rest_start;
|
||||||
|
ADD_GETLOCAL(args, node, idx, lvar_level);
|
||||||
|
|
||||||
|
if (local_body->param.flags.has_rest) {
|
||||||
|
ADD_INSN1(args, node, newarray, INT2FIX(1));
|
||||||
|
ADD_INSN (args, node, concatarray);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
argc++;
|
||||||
|
}
|
||||||
|
flag |= VM_CALL_KW_SPLAT;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
flag |= VM_CALL_SUPER | VM_CALL_FCALL;
|
||||||
|
if (type == NODE_ZSUPER) flag |= VM_CALL_ZSUPER;
|
||||||
|
ADD_INSN(ret, node, putself);
|
||||||
|
ADD_SEQ(ret, args);
|
||||||
|
ADD_INSN2(ret, node, invokesuper,
|
||||||
|
new_callinfo(iseq, 0, argc, flag, keywords, parent_block != NULL),
|
||||||
|
parent_block);
|
||||||
|
|
||||||
|
if (popped) {
|
||||||
|
ADD_INSN(ret, node, pop);
|
||||||
|
}
|
||||||
|
return COMPILE_OK;
|
||||||
|
}
|
||||||
|
|
||||||
static int iseq_compile_each0(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node, int popped);
|
static int iseq_compile_each0(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node, int popped);
|
||||||
/**
|
/**
|
||||||
compile each node
|
compile each node
|
||||||
@ -8728,139 +8866,9 @@ iseq_compile_each0(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const no
|
|||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case NODE_SUPER:
|
case NODE_SUPER:
|
||||||
case NODE_ZSUPER:{
|
case NODE_ZSUPER:
|
||||||
DECL_ANCHOR(args);
|
CHECK(compile_super(iseq, ret, node, popped, type));
|
||||||
int argc;
|
|
||||||
unsigned int flag = 0;
|
|
||||||
struct rb_callinfo_kwarg *keywords = NULL;
|
|
||||||
const rb_iseq_t *parent_block = ISEQ_COMPILE_DATA(iseq)->current_block;
|
|
||||||
|
|
||||||
INIT_ANCHOR(args);
|
|
||||||
ISEQ_COMPILE_DATA(iseq)->current_block = NULL;
|
|
||||||
if (type == NODE_SUPER) {
|
|
||||||
VALUE vargc = setup_args(iseq, args, node->nd_args, &flag, &keywords);
|
|
||||||
CHECK(!NIL_P(vargc));
|
|
||||||
argc = FIX2INT(vargc);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
/* NODE_ZSUPER */
|
|
||||||
int i;
|
|
||||||
const rb_iseq_t *liseq = body->local_iseq;
|
|
||||||
const struct rb_iseq_constant_body *const local_body = liseq->body;
|
|
||||||
const struct rb_iseq_param_keyword *const local_kwd = local_body->param.keyword;
|
|
||||||
int lvar_level = get_lvar_level(iseq);
|
|
||||||
|
|
||||||
argc = local_body->param.lead_num;
|
|
||||||
|
|
||||||
/* normal arguments */
|
|
||||||
for (i = 0; i < local_body->param.lead_num; i++) {
|
|
||||||
int idx = local_body->local_table_size - i;
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (local_body->param.flags.has_opt) {
|
|
||||||
/* optional arguments */
|
|
||||||
int j;
|
|
||||||
for (j = 0; j < local_body->param.opt_num; j++) {
|
|
||||||
int idx = local_body->local_table_size - (i + j);
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
}
|
|
||||||
i += j;
|
|
||||||
argc = i;
|
|
||||||
}
|
|
||||||
if (local_body->param.flags.has_rest) {
|
|
||||||
/* rest argument */
|
|
||||||
int idx = local_body->local_table_size - local_body->param.rest_start;
|
|
||||||
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
ADD_INSN1(args, node, splatarray, Qfalse);
|
|
||||||
|
|
||||||
argc = local_body->param.rest_start + 1;
|
|
||||||
flag |= VM_CALL_ARGS_SPLAT;
|
|
||||||
}
|
|
||||||
if (local_body->param.flags.has_post) {
|
|
||||||
/* post arguments */
|
|
||||||
int post_len = local_body->param.post_num;
|
|
||||||
int post_start = local_body->param.post_start;
|
|
||||||
|
|
||||||
if (local_body->param.flags.has_rest) {
|
|
||||||
int j;
|
|
||||||
for (j=0; j<post_len; j++) {
|
|
||||||
int idx = local_body->local_table_size - (post_start + j);
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
}
|
|
||||||
ADD_INSN1(args, node, newarray, INT2FIX(j));
|
|
||||||
ADD_INSN (args, node, concatarray);
|
|
||||||
/* argc is settled at above */
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
int j;
|
|
||||||
for (j=0; j<post_len; j++) {
|
|
||||||
int idx = local_body->local_table_size - (post_start + j);
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
}
|
|
||||||
argc = post_len + post_start;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (local_body->param.flags.has_kw) { /* TODO: support keywords */
|
|
||||||
int local_size = local_body->local_table_size;
|
|
||||||
argc++;
|
|
||||||
|
|
||||||
ADD_INSN1(args, node, putspecialobject, INT2FIX(VM_SPECIAL_OBJECT_VMCORE));
|
|
||||||
|
|
||||||
if (local_body->param.flags.has_kwrest) {
|
|
||||||
int idx = local_body->local_table_size - local_kwd->rest_start;
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
if (local_kwd->num > 0) {
|
|
||||||
ADD_SEND(args, node, rb_intern("dup"), INT2FIX(0));
|
|
||||||
flag |= VM_CALL_KW_SPLAT_MUT;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
ADD_INSN1(args, node, newhash, INT2FIX(0));
|
|
||||||
flag |= VM_CALL_KW_SPLAT_MUT;
|
|
||||||
}
|
|
||||||
for (i = 0; i < local_kwd->num; ++i) {
|
|
||||||
ID id = local_kwd->table[i];
|
|
||||||
int idx = local_size - get_local_var_idx(liseq, id);
|
|
||||||
ADD_INSN1(args, node, putobject, ID2SYM(id));
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
}
|
|
||||||
ADD_SEND(args, node, id_core_hash_merge_ptr, INT2FIX(i * 2 + 1));
|
|
||||||
if (local_body->param.flags.has_rest) {
|
|
||||||
ADD_INSN1(args, node, newarray, INT2FIX(1));
|
|
||||||
ADD_INSN (args, node, concatarray);
|
|
||||||
--argc;
|
|
||||||
}
|
|
||||||
flag |= VM_CALL_KW_SPLAT;
|
|
||||||
}
|
|
||||||
else if (local_body->param.flags.has_kwrest) {
|
|
||||||
int idx = local_body->local_table_size - local_kwd->rest_start;
|
|
||||||
ADD_GETLOCAL(args, node, idx, lvar_level);
|
|
||||||
|
|
||||||
if (local_body->param.flags.has_rest) {
|
|
||||||
ADD_INSN1(args, node, newarray, INT2FIX(1));
|
|
||||||
ADD_INSN (args, node, concatarray);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
argc++;
|
|
||||||
}
|
|
||||||
flag |= VM_CALL_KW_SPLAT;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ADD_INSN(ret, node, putself);
|
|
||||||
ADD_SEQ(ret, args);
|
|
||||||
ADD_INSN2(ret, node, invokesuper,
|
|
||||||
new_callinfo(iseq, 0, argc, flag | VM_CALL_SUPER | (type == NODE_ZSUPER ? VM_CALL_ZSUPER : 0) | VM_CALL_FCALL, keywords, parent_block != NULL),
|
|
||||||
parent_block);
|
|
||||||
|
|
||||||
if (popped) {
|
|
||||||
ADD_INSN(ret, node, pop);
|
|
||||||
}
|
|
||||||
break;
|
break;
|
||||||
}
|
|
||||||
case NODE_LIST:{
|
case NODE_LIST:{
|
||||||
CHECK(compile_array(iseq, ret, node, popped) >= 0);
|
CHECK(compile_array(iseq, ret, node, popped) >= 0);
|
||||||
break;
|
break;
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user