mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
node.c: predicates for special NODEs
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@62861 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
798316eac2
commit
b4b3de3e7c
5 changed files with 13 additions and 12 deletions
|
@ -4088,7 +4088,7 @@ compile_massign(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node,
|
|||
const NODE *rhsn = node->nd_value;
|
||||
const NODE *splatn = node->nd_args;
|
||||
const NODE *lhsn = node->nd_head;
|
||||
int lhs_splat = (splatn && splatn != NODE_SPECIAL_NO_NAME_REST) ? 1 : 0;
|
||||
int lhs_splat = (splatn && NODE_NAMED_REST_P(splatn)) ? 1 : 0;
|
||||
|
||||
if (!popped || splatn || !compile_massign_opt(iseq, ret, rhsn, lhsn)) {
|
||||
int llen = 0;
|
||||
|
@ -4145,12 +4145,12 @@ compile_massign(rb_iseq_t *iseq, LINK_ANCHOR *const ret, const NODE *const node,
|
|||
const NODE *postn = splatn->nd_2nd;
|
||||
const NODE *restn = splatn->nd_1st;
|
||||
int num = (int)postn->nd_alen;
|
||||
int flag = 0x02 | ((restn == NODE_SPECIAL_NO_NAME_REST) ? 0x00 : 0x01);
|
||||
int flag = 0x02 | (NODE_NAMED_REST_P(restn) ? 0x01 : 0x00);
|
||||
|
||||
ADD_INSN2(ret, nd_line(splatn), expandarray,
|
||||
INT2FIX(num), INT2FIX(flag));
|
||||
|
||||
if (restn != NODE_SPECIAL_NO_NAME_REST) {
|
||||
if (NODE_NAMED_REST_P(restn)) {
|
||||
CHECK(compile_massign_lhs(iseq, ret, restn));
|
||||
}
|
||||
while (postn) {
|
||||
|
|
|
@ -239,7 +239,7 @@ node_children(rb_ast_t *ast, NODE *node)
|
|||
return ary;
|
||||
}
|
||||
case NODE_MASGN:
|
||||
if (node->nd_args != NODE_SPECIAL_NO_NAME_REST) {
|
||||
if (NODE_NAMED_REST_P(node->nd_args)) {
|
||||
return rb_ary_new_from_node_args(ast, 3, node->nd_value, node->nd_head, node->nd_args);
|
||||
}
|
||||
return rb_ary_new_from_node_args(ast, 2, node->nd_value, node->nd_head);
|
||||
|
@ -253,7 +253,7 @@ node_children(rb_ast_t *ast, NODE *node)
|
|||
goto asgn;
|
||||
case NODE_CVASGN:
|
||||
asgn:
|
||||
if (node->nd_value == NODE_SPECIAL_REQUIRED_KEYWORD) {
|
||||
if (NODE_REQUIRED_KEYWORD_P(node)) {
|
||||
return rb_ary_new_from_node_args(ast, 0);
|
||||
}
|
||||
return rb_ary_new_from_node_args(ast, 1, node->nd_value);
|
||||
|
@ -404,7 +404,7 @@ node_children(rb_ast_t *ast, NODE *node)
|
|||
case NODE_KW_ARG:
|
||||
return rb_ary_new_from_node_args(ast, 2, node->nd_body, node->nd_next);
|
||||
case NODE_POSTARG:
|
||||
if (node->nd_1st != NODE_SPECIAL_NO_NAME_REST) {
|
||||
if (NODE_NAMED_REST_P(node->nd_1st)) {
|
||||
return rb_ary_new_from_node_args(ast, 2, node->nd_1st, node->nd_2nd);
|
||||
}
|
||||
return rb_ary_new_from_node_args(ast, 1, node->nd_2nd);
|
||||
|
|
8
node.c
8
node.c
|
@ -342,7 +342,7 @@ dump_node(VALUE buf, VALUE indent, int comment, const NODE * node)
|
|||
ANN("example: a, b = foo");
|
||||
F_NODE(nd_value, "rhsn");
|
||||
F_NODE(nd_head, "lhsn");
|
||||
if (node->nd_args != NODE_SPECIAL_NO_NAME_REST) {
|
||||
if (NODE_NAMED_REST_P(node->nd_args)) {
|
||||
LAST_NODE;
|
||||
F_NODE(nd_args, "splatn");
|
||||
}
|
||||
|
@ -356,7 +356,7 @@ dump_node(VALUE buf, VALUE indent, int comment, const NODE * node)
|
|||
ANN("format: [nd_vid](lvar) = [nd_value]");
|
||||
ANN("example: x = foo");
|
||||
F_ID(nd_vid, "local variable");
|
||||
if (node->nd_value == NODE_SPECIAL_REQUIRED_KEYWORD) {
|
||||
if (NODE_REQUIRED_KEYWORD_P(node)) {
|
||||
F_MSG(nd_value, "rvalue", "NODE_SPECIAL_REQUIRED_KEYWORD (required keyword argument)");
|
||||
}
|
||||
else {
|
||||
|
@ -377,7 +377,7 @@ dump_node(VALUE buf, VALUE indent, int comment, const NODE * node)
|
|||
ANN("format: [nd_vid](current dvar) = [nd_value]");
|
||||
ANN("example: 1.times { x = foo }");
|
||||
F_ID(nd_vid, "local variable");
|
||||
if (node->nd_value == NODE_SPECIAL_REQUIRED_KEYWORD) {
|
||||
if (NODE_REQUIRED_KEYWORD_P(node)) {
|
||||
F_MSG(nd_value, "rvalue", "NODE_SPECIAL_REQUIRED_KEYWORD (required keyword argument)");
|
||||
}
|
||||
else {
|
||||
|
@ -955,7 +955,7 @@ dump_node(VALUE buf, VALUE indent, int comment, const NODE * node)
|
|||
ANN("post arguments");
|
||||
ANN("format: *[nd_1st], [nd_2nd..] = ..");
|
||||
ANN("example: a, *rest, z = foo");
|
||||
if (node->nd_1st != NODE_SPECIAL_NO_NAME_REST) {
|
||||
if (NODE_NAMED_REST_P(node->nd_1st)) {
|
||||
F_NODE(nd_1st, "rest argument");
|
||||
}
|
||||
else {
|
||||
|
|
2
node.h
2
node.h
|
@ -469,7 +469,9 @@ typedef struct RNode {
|
|||
#define NEW_ATTRASGN(r,m,a,loc) NEW_NODE(NODE_ATTRASGN,r,m,a,loc)
|
||||
|
||||
#define NODE_SPECIAL_REQUIRED_KEYWORD ((NODE *)-1)
|
||||
#define NODE_REQUIRED_KEYWORD_P(node) ((node)->nd_value == NODE_SPECIAL_REQUIRED_KEYWORD)
|
||||
#define NODE_SPECIAL_NO_NAME_REST ((NODE *)-1)
|
||||
#define NODE_NAMED_REST_P(node) ((node) != NODE_SPECIAL_NO_NAME_REST)
|
||||
|
||||
RUBY_SYMBOL_EXPORT_BEGIN
|
||||
|
||||
|
|
3
parse.y
3
parse.y
|
@ -9955,10 +9955,9 @@ new_args_tail(struct parser_params *p, NODE *kw_args, ID kw_rest_arg, ID block,
|
|||
int i;
|
||||
|
||||
while (kwn) {
|
||||
NODE *val_node = kwn->nd_body->nd_value;
|
||||
ID vid = kwn->nd_body->nd_vid;
|
||||
|
||||
if (val_node == NODE_SPECIAL_REQUIRED_KEYWORD) {
|
||||
if (NODE_REQUIRED_KEYWORD_P(kwn->nd_body)) {
|
||||
vtable_add(required_kw_vars, vid);
|
||||
}
|
||||
else {
|
||||
|
|
Loading…
Reference in a new issue