rustc: Start work on lvals

This commit is contained in:
Patrick Walton 2010-10-13 12:43:35 -07:00
parent c7ab80f743
commit 699ef987d7
3 changed files with 140 additions and 62 deletions

View file

@ -13,6 +13,7 @@ import front.ast.ident;
import front.ast.name;
import front.ast.ty;
import front.ast.expr;
import front.ast.lval;
import front.ast.stmt;
import front.ast.block;
import front.ast.item;
@ -65,16 +66,6 @@ type ast_fold[ENV] =
(fn(&ENV e, &span sp,
@ast.lit) -> @expr) fold_expr_lit,
(fn(&ENV e, &span sp,
&name n,
&option[referent] r) -> @expr) fold_expr_name,
(fn(&ENV e, &span sp,
@expr e, ident i) -> @expr) fold_expr_field,
(fn(&ENV e, &span sp,
@expr e, @expr ix) -> @expr) fold_expr_index,
(fn(&ENV e, &span sp,
@expr cond, block thn,
&option[block] els) -> @expr) fold_expr_if,
@ -82,6 +73,22 @@ type ast_fold[ENV] =
(fn(&ENV e, &span sp,
block blk) -> @expr) fold_expr_block,
(fn(&ENV e, &span sp,
@lval lhs, @expr rhs) -> @expr) fold_expr_assign,
(fn(&ENV e, &span sp,
@lval lv) -> @expr) fold_expr_lval,
// Lvalue folds.
(fn(&ENV e, &span sp,
@expr e, ident i) -> @lval) fold_lval_field,
(fn(&ENV e, &span sp,
@expr e, @expr ix) -> @lval) fold_lval_index,
(fn(&ENV e, &span sp,
&name n,
&option[referent] r) -> @lval) fold_lval_name,
// Decl folds.
(fn(&ENV e, &span sp,
@ -131,6 +138,7 @@ type ast_fold[ENV] =
(fn(&ENV e, @ast.crate c) -> ENV) update_env_for_crate,
(fn(&ENV e, @item i) -> ENV) update_env_for_item,
(fn(&ENV e, @stmt s) -> ENV) update_env_for_stmt,
(fn(&ENV e, @lval l) -> ENV) update_env_for_lval,
(fn(&ENV e, @expr x) -> ENV) update_env_for_expr,
(fn(&ENV e, @ty t) -> ENV) update_env_for_ty,
@ -156,6 +164,33 @@ fn fold_decl[ENV](&ENV env, ast_fold[ENV] fld, @decl d) -> @decl {
ret d;
}
fn fold_lval[ENV](&ENV env, ast_fold[ENV] fld, @lval lv) -> @lval {
let ENV env_ = fld.update_env_for_lval(env, lv);
if (!fld.keep_going(env_)) {
ret lv;
}
alt (lv.node) {
case (ast.lval_field(?e, ?i)) {
auto ee = fold_expr(env_, fld, e);
ret fld.fold_lval_field(env_, lv.span, ee, i);
}
case (ast.lval_index(?e, ?ix)) {
auto ee = fold_expr(env_, fld, e);
auto iix = fold_expr(env_, fld, ix);
ret fld.fold_lval_index(env_, lv.span, ee, iix);
}
case (ast.lval_name(?n, ?r)) {
ret fld.fold_lval_name(env_, lv.span, n, r);
}
}
fail; // shoudn't be reached
}
fn fold_exprs[ENV](&ENV env, ast_fold[ENV] fld, vec[@expr] e) -> vec[@expr] {
let operator[@expr, @expr] fe = bind fold_expr[ENV](env, fld, _);
ret _vec.map[@expr, @expr](fe, e);
@ -220,22 +255,6 @@ fn fold_expr[ENV](&ENV env, ast_fold[ENV] fld, &@expr e) -> @expr {
ret fld.fold_expr_lit(env_, e.span, lit);
}
case (ast.expr_name(?n, ?r)) {
auto nn = fold_expr_name(env_, fld, n, r);
ret fld.fold_expr_name(env_, e.span, nn._0, nn._1);
}
case (ast.expr_field(?e, ?i)) {
auto ee = fold_expr(env_, fld, e);
ret fld.fold_expr_field(env_, e.span, ee, i);
}
case (ast.expr_index(?e, ?i)) {
auto ee = fold_expr(env_, fld, e);
auto ii = fold_expr(env_, fld, i);
ret fld.fold_expr_index(env_, e.span, ee, ii);
}
case (ast.expr_if(?cnd, ?thn, ?els)) {
auto ccnd = fold_expr(env_, fld, cnd);
auto tthn = fold_block(env_, fld, thn);
@ -252,6 +271,17 @@ fn fold_expr[ENV](&ENV env, ast_fold[ENV] fld, &@expr e) -> @expr {
auto bb = fold_block(env_, fld, b);
ret fld.fold_expr_block(env_, e.span, bb);
}
case (ast.expr_assign(?lhs, ?rhs)) {
auto llhs = fold_lval(env_, fld, lhs);
auto rrhs = fold_expr(env_, fld, rhs);
ret fld.fold_expr_assign(env_, e.span, llhs, rrhs);
}
case (ast.expr_lval(?lv)) {
auto llv = fold_lval(env_, fld, lv);
ret fld.fold_expr_lval(env_, e.span, llv);
}
}
ret e;
@ -462,21 +492,6 @@ fn identity_fold_expr_lit[ENV](&ENV env, &span sp, @ast.lit lit) -> @expr {
ret @respan(sp, ast.expr_lit(lit));
}
fn identity_fold_expr_name[ENV](&ENV env, &span sp, &name n,
&option[referent] r) -> @expr {
ret @respan(sp, ast.expr_name(n, r));
}
fn identity_fold_expr_field[ENV](&ENV env, &span sp,
@expr e, ident i) -> @expr {
ret @respan(sp, ast.expr_field(e, i));
}
fn identity_fold_expr_index[ENV](&ENV env, &span sp,
@expr e, @expr ix) -> @expr {
ret @respan(sp, ast.expr_index(e, ix));
}
fn identity_fold_expr_if[ENV](&ENV env, &span sp,
@expr cond, block thn,
&option[block] els) -> @expr {
@ -487,6 +502,33 @@ fn identity_fold_expr_block[ENV](&ENV env, &span sp, block blk) -> @expr {
ret @respan(sp, ast.expr_block(blk));
}
fn identity_fold_expr_assign[ENV](&ENV env, &span sp,
@lval lhs, @expr rhs) -> @expr {
ret @respan(sp, ast.expr_assign(lhs, rhs));
}
fn identity_fold_expr_lval[ENV](&ENV env, &span sp, @lval lv) -> @expr {
ret @respan(sp, ast.expr_lval(lv));
}
// Lvalue identities.
fn identity_fold_lval_field[ENV](&ENV env, &span sp,
@expr e, ident i) -> @lval {
ret @respan(sp, ast.lval_field(e, i));
}
fn identity_fold_lval_index[ENV](&ENV env, &span sp,
@expr e, @expr ix) -> @lval {
ret @respan(sp, ast.lval_index(e, ix));
}
fn identity_fold_lval_name[ENV](&ENV env, &span sp,
&name n, &option[referent] r) -> @lval {
ret @respan(sp, ast.lval_name(n, r));
}
// Decl identities.
@ -575,6 +617,10 @@ fn identity_update_env_for_stmt[ENV](&ENV e, @stmt s) -> ENV {
ret e;
}
fn identity_update_env_for_lval[ENV](&ENV e, @lval l) -> ENV {
ret e;
}
fn identity_update_env_for_expr[ENV](&ENV e, @expr x) -> ENV {
ret e;
}
@ -613,11 +659,14 @@ fn new_identity_fold[ENV]() -> ast_fold[ENV] {
fold_expr_binary = bind identity_fold_expr_binary[ENV](_,_,_,_,_),
fold_expr_unary = bind identity_fold_expr_unary[ENV](_,_,_,_),
fold_expr_lit = bind identity_fold_expr_lit[ENV](_,_,_),
fold_expr_name = bind identity_fold_expr_name[ENV](_,_,_,_),
fold_expr_field = bind identity_fold_expr_field[ENV](_,_,_,_),
fold_expr_index = bind identity_fold_expr_index[ENV](_,_,_,_),
fold_expr_if = bind identity_fold_expr_if[ENV](_,_,_,_,_),
fold_expr_block = bind identity_fold_expr_block[ENV](_,_,_),
fold_expr_assign = bind identity_fold_expr_assign[ENV](_,_,_,_),
fold_expr_lval = bind identity_fold_expr_lval[ENV](_,_,_),
fold_lval_field = bind identity_fold_lval_field[ENV](_,_,_,_),
fold_lval_index = bind identity_fold_lval_index[ENV](_,_,_,_),
fold_lval_name = bind identity_fold_lval_name[ENV](_,_,_,_),
fold_decl_local = bind identity_fold_decl_local[ENV](_,_,_,_,_),
fold_decl_item = bind identity_fold_decl_item[ENV](_,_,_,_),
@ -639,6 +688,7 @@ fn new_identity_fold[ENV]() -> ast_fold[ENV] {
update_env_for_crate = bind identity_update_env_for_crate[ENV](_,_),
update_env_for_item = bind identity_update_env_for_item[ENV](_,_),
update_env_for_stmt = bind identity_update_env_for_stmt[ENV](_,_),
update_env_for_lval = bind identity_update_env_for_lval[ENV](_,_),
update_env_for_expr = bind identity_update_env_for_expr[ENV](_,_),
update_env_for_ty = bind identity_update_env_for_ty[ENV](_,_),