rust/src/comp/syntax/fold.rs

738 lines
28 KiB
Rust
Raw Normal View History

import syntax::codemap::span;
2011-06-20 19:25:49 -05:00
import ast::*;
import core::{vec, option};
2011-06-20 19:25:49 -05:00
export ast_fold_precursor;
export ast_fold;
export default_ast_fold;
export make_fold;
export noop_fold_crate;
export noop_fold_item;
export noop_fold_expr;
export noop_fold_pat;
export noop_fold_mod;
export noop_fold_ty;
export wrap;
2011-06-20 19:25:49 -05:00
type ast_fold = @mutable a_f;
// We may eventually want to be able to fold over type parameters, too
type ast_fold_precursor =
2011-07-27 07:19:39 -05:00
//unlike the others, item_ is non-trivial
{fold_crate: fn@(crate_, span, ast_fold) -> (crate_, span),
fold_crate_directive: fn@(crate_directive_, span,
ast_fold) -> (crate_directive_, span),
fold_view_item: fn@(view_item_, ast_fold) -> view_item_,
fold_native_item: fn@(&&@native_item, ast_fold) -> @native_item,
fold_item: fn@(&&@item, ast_fold) -> @item,
fold_class_item: fn@(&&@class_item, ast_fold) -> @class_item,
fold_item_underscore: fn@(item_, ast_fold) -> item_,
fold_method: fn@(&&@method, ast_fold) -> @method,
fold_block: fn@(blk_, span, ast_fold) -> (blk_, span),
fold_stmt: fn@(stmt_, span, ast_fold) -> (stmt_, span),
fold_arm: fn@(arm, ast_fold) -> arm,
fold_pat: fn@(pat_, span, ast_fold) -> (pat_, span),
fold_decl: fn@(decl_, span, ast_fold) -> (decl_, span),
fold_expr: fn@(expr_, span, ast_fold) -> (expr_, span),
fold_ty: fn@(ty_, span, ast_fold) -> (ty_, span),
fold_constr: fn@(ast::constr_, span, ast_fold) -> (constr_, span),
fold_mod: fn@(_mod, ast_fold) -> _mod,
fold_native_mod: fn@(native_mod, ast_fold) -> native_mod,
fold_variant: fn@(variant_, span, ast_fold) -> (variant_, span),
fold_ident: fn@(&&ident, ast_fold) -> ident,
fold_path: fn@(path_, span, ast_fold) -> (path_, span),
fold_local: fn@(local_, span, ast_fold) -> (local_, span),
map_exprs: fn@(fn@(&&@expr) -> @expr, [@expr]) -> [@expr],
new_id: fn@(node_id) -> node_id,
new_span: fn@(span) -> span};
2011-06-20 19:25:49 -05:00
type a_f =
{fold_crate: fn@(crate) -> crate,
fold_crate_directive: fn@(&&@crate_directive) -> @crate_directive,
fold_view_item: fn@(&&@view_item) -> @view_item,
fold_native_item: fn@(&&@native_item) -> @native_item,
fold_item: fn@(&&@item) -> @item,
fold_class_item: fn@(&&@class_item) -> @class_item,
fold_item_underscore: fn@(item_) -> item_,
fold_method: fn@(&&@method) -> @method,
fold_block: fn@(blk) -> blk,
fold_stmt: fn@(&&@stmt) -> @stmt,
fold_arm: fn@(arm) -> arm,
fold_pat: fn@(&&@pat) -> @pat,
fold_decl: fn@(&&@decl) -> @decl,
fold_expr: fn@(&&@expr) -> @expr,
fold_ty: fn@(&&@ty) -> @ty,
fold_constr: fn@(&&@constr) -> @constr,
fold_mod: fn@(_mod) -> _mod,
fold_native_mod: fn@(native_mod) -> native_mod,
fold_variant: fn@(variant) -> variant,
fold_ident: fn@(&&ident) -> ident,
fold_path: fn@(&&@path) -> @path,
fold_local: fn@(&&@local) -> @local,
map_exprs: fn@(fn@(&&@expr) -> @expr, [@expr]) -> [@expr],
new_id: fn@(node_id) -> node_id,
new_span: fn@(span) -> span};
2011-06-20 19:25:49 -05:00
//fn nf_dummy<T>(&T node) -> T { fail; }
fn nf_crate_dummy(_c: crate) -> crate { fail; }
fn nf_crate_directive_dummy(&&_c: @crate_directive) -> @crate_directive {
2011-07-27 07:19:39 -05:00
fail;
}
fn nf_view_item_dummy(&&_v: @view_item) -> @view_item { fail; }
fn nf_native_item_dummy(&&_n: @native_item) -> @native_item { fail; }
fn nf_item_dummy(&&_i: @item) -> @item { fail; }
fn nf_class_item_dummy(&&_ci: @class_item) -> @class_item { fail; }
fn nf_item_underscore_dummy(_i: item_) -> item_ { fail; }
fn nf_method_dummy(&&_m: @method) -> @method { fail; }
fn nf_blk_dummy(_b: blk) -> blk { fail; }
fn nf_stmt_dummy(&&_s: @stmt) -> @stmt { fail; }
fn nf_arm_dummy(_a: arm) -> arm { fail; }
fn nf_pat_dummy(&&_p: @pat) -> @pat { fail; }
fn nf_decl_dummy(&&_d: @decl) -> @decl { fail; }
fn nf_expr_dummy(&&_e: @expr) -> @expr { fail; }
fn nf_ty_dummy(&&_t: @ty) -> @ty { fail; }
fn nf_constr_dummy(&&_c: @constr) -> @constr { fail; }
fn nf_mod_dummy(_m: _mod) -> _mod { fail; }
fn nf_native_mod_dummy(_n: native_mod) -> native_mod { fail; }
fn nf_variant_dummy(_v: variant) -> variant { fail; }
fn nf_ident_dummy(&&_i: ident) -> ident { fail; }
fn nf_path_dummy(&&_p: @path) -> @path { fail; }
fn nf_local_dummy(&&_o: @local) -> @local { fail; }
2011-06-20 19:25:49 -05:00
/* some little folds that probably aren't useful to have in ast_fold itself*/
//used in noop_fold_item and noop_fold_crate and noop_fold_crate_directive
fn fold_meta_item_(&&mi: @meta_item, fld: ast_fold) -> @meta_item {
2011-07-27 07:19:39 -05:00
ret @{node:
alt mi.node {
meta_word(id) { meta_word(fld.fold_ident(id)) }
meta_list(id, mis) {
let fold_meta_item = bind fold_meta_item_(_, fld);
meta_list(id, vec::map(mis, fold_meta_item))
2011-07-27 07:19:39 -05:00
}
meta_name_value(id, s) {
meta_name_value(fld.fold_ident(id), s)
}
},
span: mi.span};
2011-06-20 19:25:49 -05:00
}
//used in noop_fold_item and noop_fold_crate
fn fold_attribute_(at: attribute, fmi: fn@(&&@meta_item) -> @meta_item) ->
2011-07-27 07:19:39 -05:00
attribute {
ret {node: {style: at.node.style, value: *fmi(@at.node.value)},
span: at.span};
2011-06-20 19:25:49 -05:00
}
//used in noop_fold_native_item and noop_fold_fn_decl
fn fold_arg_(a: arg, fld: ast_fold) -> arg {
ret {ty: fld.fold_ty(a.ty),
ident: fld.fold_ident(a.ident)
with a};
2011-06-20 19:25:49 -05:00
}
//used in noop_fold_expr, and possibly elsewhere in the future
fn fold_mac_(m: mac, fld: ast_fold) -> mac {
2011-07-27 07:19:39 -05:00
ret {node:
alt m.node {
mac_invoc(pth, arg, body) {
mac_invoc(fld.fold_path(pth),
// FIXME: bind should work, but causes a crash
option::map(arg) {|arg| fld.fold_expr(arg)},
body)
}
mac_embed_type(ty) { mac_embed_type(fld.fold_ty(ty)) }
mac_embed_block(blk) { mac_embed_block(fld.fold_block(blk)) }
mac_ellipsis { mac_ellipsis }
mac_aq(_,_) { /* fixme */ m.node }
mac_var(_) { /* fixme */ m.node }
},
2011-07-27 07:19:39 -05:00
span: m.span};
}
fn fold_fn_decl(decl: ast::fn_decl, fld: ast_fold) -> ast::fn_decl {
ret {inputs: vec::map(decl.inputs, bind fold_arg_(_, fld)),
output: fld.fold_ty(decl.output),
purity: decl.purity,
cf: decl.cf,
constraints: vec::map(decl.constraints, fld.fold_constr)}
}
2011-06-20 19:25:49 -05:00
fn noop_fold_crate(c: crate_, fld: ast_fold) -> crate_ {
2011-07-27 07:19:39 -05:00
let fold_meta_item = bind fold_meta_item_(_, fld);
let fold_attribute = bind fold_attribute_(_, fold_meta_item);
2011-06-20 19:25:49 -05:00
ret {directives: vec::map(c.directives, fld.fold_crate_directive),
2011-07-27 07:19:39 -05:00
module: fld.fold_mod(c.module),
attrs: vec::map(c.attrs, fold_attribute),
config: vec::map(c.config, fold_meta_item)};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_crate_directive(cd: crate_directive_, fld: ast_fold) ->
2011-07-27 07:19:39 -05:00
crate_directive_ {
ret alt cd {
cdir_src_mod(id, attrs) {
cdir_src_mod(fld.fold_ident(id), attrs)
2011-07-27 07:19:39 -05:00
}
cdir_dir_mod(id, cds, attrs) {
cdir_dir_mod(fld.fold_ident(id),
vec::map(cds, fld.fold_crate_directive), attrs)
2011-07-27 07:19:39 -05:00
}
cdir_view_item(vi) { cdir_view_item(fld.fold_view_item(vi)) }
cdir_syntax(_) { cd }
}
2011-06-20 19:25:49 -05:00
}
fn noop_fold_view_item(vi: view_item_, _fld: ast_fold) -> view_item_ {
2011-06-20 19:25:49 -05:00
ret vi;
}
fn noop_fold_native_item(&&ni: @native_item, fld: ast_fold) -> @native_item {
2011-07-27 07:19:39 -05:00
let fold_arg = bind fold_arg_(_, fld);
let fold_meta_item = bind fold_meta_item_(_, fld);
let fold_attribute = bind fold_attribute_(_, fold_meta_item);
ret @{ident: fld.fold_ident(ni.ident),
attrs: vec::map(ni.attrs, fold_attribute),
2011-07-27 07:19:39 -05:00
node:
alt ni.node {
native_item_fn(fdec, typms) {
native_item_fn({inputs: vec::map(fdec.inputs, fold_arg),
2011-07-27 07:19:39 -05:00
output: fld.fold_ty(fdec.output),
purity: fdec.purity,
cf: fdec.cf,
constraints:
vec::map(fdec.constraints,
fld.fold_constr)}, typms)
2011-07-27 07:19:39 -05:00
}
},
id: ni.id,
span: ni.span};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_item(&&i: @item, fld: ast_fold) -> @item {
2011-07-27 07:19:39 -05:00
let fold_meta_item = bind fold_meta_item_(_, fld);
let fold_attribute = bind fold_attribute_(_, fold_meta_item);
2011-06-20 19:25:49 -05:00
2011-07-27 07:19:39 -05:00
ret @{ident: fld.fold_ident(i.ident),
attrs: vec::map(i.attrs, fold_attribute),
2011-07-27 07:19:39 -05:00
id: i.id,
node: fld.fold_item_underscore(i.node),
span: i.span};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_class_item(&&ci: @class_item, fld: ast_fold)
-> @class_item {
@{node: {
privacy:ci.node.privacy,
decl:
@alt *ci.node.decl {
instance_var(ident, t, cm, id) {
instance_var(ident, fld.fold_ty(t), cm, id)
}
class_method(i) { class_method(fld.fold_item(i)) }
}},
span: ci.span}
}
fn noop_fold_item_underscore(i: item_, fld: ast_fold) -> item_ {
2011-07-27 07:19:39 -05:00
ret alt i {
item_const(t, e) { item_const(fld.fold_ty(t), fld.fold_expr(e)) }
item_fn(decl, typms, body) {
let body = fld.fold_block(body);
item_fn(fold_fn_decl(decl, fld), typms, body)
}
2011-07-27 07:19:39 -05:00
item_mod(m) { item_mod(fld.fold_mod(m)) }
item_native_mod(nm) { item_native_mod(fld.fold_native_mod(nm)) }
item_ty(t, typms) { item_ty(fld.fold_ty(t), typms) }
item_enum(variants, typms) {
item_enum(vec::map(variants, fld.fold_variant), typms)
2011-07-27 07:19:39 -05:00
}
item_class(typms, items, ctor_decl, ctor_body) {
item_class(typms,
vec::map(items, fld.fold_class_item),
fold_fn_decl(ctor_decl, fld),
fld.fold_block(ctor_body))
}
item_impl(tps, ifce, ty, methods) {
item_impl(tps, option::map(ifce, fld.fold_ty), fld.fold_ty(ty),
vec::map(methods, fld.fold_method))
}
item_iface(tps, methods) { item_iface(tps, methods) }
item_res(decl, typms, body, did, cid) {
item_res(fold_fn_decl(decl, fld), typms, fld.fold_block(body),
did, cid)
2011-07-27 07:19:39 -05:00
}
};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_method(&&m: @method, fld: ast_fold) -> @method {
ret @{ident: fld.fold_ident(m.ident),
decl: fold_fn_decl(m.decl, fld),
body: fld.fold_block(m.body) with *m};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_block(b: blk_, fld: ast_fold) -> blk_ {
ret {view_items: vec::map(b.view_items, fld.fold_view_item),
stmts: vec::map(b.stmts, fld.fold_stmt),
expr: option::map(b.expr, fld.fold_expr),
id: b.id,
rules: b.rules};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_stmt(s: stmt_, fld: ast_fold) -> stmt_ {
2011-07-27 07:19:39 -05:00
ret alt s {
stmt_decl(d, nid) { stmt_decl(fld.fold_decl(d), nid) }
stmt_expr(e, nid) { stmt_expr(fld.fold_expr(e), nid) }
stmt_semi(e, nid) { stmt_semi(fld.fold_expr(e), nid) }
};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_arm(a: arm, fld: ast_fold) -> arm {
ret {pats: vec::map(a.pats, fld.fold_pat),
guard: option::map(a.guard, fld.fold_expr),
body: fld.fold_block(a.body)};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_pat(p: pat_, fld: ast_fold) -> pat_ {
2011-07-27 07:19:39 -05:00
ret alt p {
pat_wild { p }
pat_ident(pth, sub) {
pat_ident(fld.fold_path(pth), option::map(sub, fld.fold_pat))
}
2011-07-27 07:19:39 -05:00
pat_lit(_) { p }
pat_enum(pth, pats) {
pat_enum(fld.fold_path(pth), vec::map(pats, fld.fold_pat))
2011-07-27 07:19:39 -05:00
}
pat_rec(fields, etc) {
let fs = [];
for f: ast::field_pat in fields {
fs += [{ident: f.ident, pat: fld.fold_pat(f.pat)}];
2011-07-11 07:13:20 -05:00
}
pat_rec(fs, etc)
2011-07-27 07:19:39 -05:00
}
pat_tup(elts) { pat_tup(vec::map(elts, fld.fold_pat)) }
2011-07-27 07:19:39 -05:00
pat_box(inner) { pat_box(fld.fold_pat(inner)) }
pat_uniq(inner) { pat_uniq(fld.fold_pat(inner)) }
pat_range(_, _) { p }
2011-07-27 07:19:39 -05:00
};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_decl(d: decl_, fld: ast_fold) -> decl_ {
2011-07-27 07:19:39 -05:00
ret alt d {
decl_local(ls) {
decl_local(vec::map(ls, {|l| let (st, lc) = l;
(st, fld.fold_local(lc))}))
}
2011-07-27 07:19:39 -05:00
decl_item(it) { decl_item(fld.fold_item(it)) }
2011-06-20 19:25:49 -05:00
}
}
fn wrap<T>(f: fn@(T, ast_fold) -> T)
-> fn@(T, span, ast_fold) -> (T, span)
{
ret fn@(x: T, s: span, fld: ast_fold) -> (T, span) {
(f(x, fld), s)
}
}
fn noop_fold_expr(e: expr_, fld: ast_fold) -> expr_ {
fn fold_field_(field: field, fld: ast_fold) -> field {
2011-07-27 07:19:39 -05:00
ret {node:
{mut: field.node.mut,
ident: fld.fold_ident(field.node.ident),
expr: fld.fold_expr(field.node.expr)},
span: field.span};
2011-06-20 19:25:49 -05:00
}
2011-07-27 07:19:39 -05:00
let fold_field = bind fold_field_(_, fld);
2011-07-27 07:19:39 -05:00
let fold_mac = bind fold_mac_(_, fld);
2011-06-20 19:25:49 -05:00
2011-07-27 07:19:39 -05:00
ret alt e {
expr_vec(exprs, mutt) {
expr_vec(fld.map_exprs(fld.fold_expr, exprs), mutt)
2011-07-27 07:19:39 -05:00
}
expr_rec(fields, maybe_expr) {
expr_rec(vec::map(fields, fold_field),
option::map(maybe_expr, fld.fold_expr))
2011-07-27 07:19:39 -05:00
}
expr_tup(elts) { expr_tup(vec::map(elts, fld.fold_expr)) }
expr_call(f, args, blk) {
expr_call(fld.fold_expr(f), fld.map_exprs(fld.fold_expr, args),
blk)
2011-07-27 07:19:39 -05:00
}
expr_bind(f, args) {
let opt_map_se = bind option::map(_, fld.fold_expr);
expr_bind(fld.fold_expr(f), vec::map(args, opt_map_se))
2011-07-27 07:19:39 -05:00
}
expr_binary(binop, lhs, rhs) {
2011-06-20 19:25:49 -05:00
expr_binary(binop, fld.fold_expr(lhs), fld.fold_expr(rhs))
2011-07-27 07:19:39 -05:00
}
expr_unary(binop, ohs) { expr_unary(binop, fld.fold_expr(ohs)) }
expr_lit(_) { e }
expr_cast(expr, ty) { expr_cast(fld.fold_expr(expr), ty) }
expr_if(cond, tr, fl) {
expr_if(fld.fold_expr(cond), fld.fold_block(tr),
option::map(fl, fld.fold_expr))
2011-07-27 07:19:39 -05:00
}
expr_while(cond, body) {
2011-06-20 19:25:49 -05:00
expr_while(fld.fold_expr(cond), fld.fold_block(body))
2011-07-27 07:19:39 -05:00
}
expr_for(decl, expr, blk) {
expr_for(fld.fold_local(decl), fld.fold_expr(expr),
fld.fold_block(blk))
2011-07-27 07:19:39 -05:00
}
expr_do_while(blk, expr) {
expr_do_while(fld.fold_block(blk), fld.fold_expr(expr))
2011-07-27 07:19:39 -05:00
}
expr_alt(expr, arms) {
expr_alt(fld.fold_expr(expr), vec::map(arms, fld.fold_arm))
2011-07-27 07:19:39 -05:00
}
expr_fn(proto, decl, body, captures) {
expr_fn(proto, fold_fn_decl(decl, fld),
fld.fold_block(body), captures)
}
expr_fn_block(decl, body) {
expr_fn_block(fold_fn_decl(decl, fld), fld.fold_block(body))
}
2011-07-27 07:19:39 -05:00
expr_block(blk) { expr_block(fld.fold_block(blk)) }
expr_move(el, er) {
2011-06-20 19:25:49 -05:00
expr_move(fld.fold_expr(el), fld.fold_expr(er))
2011-07-27 07:19:39 -05:00
}
expr_copy(e) { expr_copy(fld.fold_expr(e)) }
2011-07-27 07:19:39 -05:00
expr_assign(el, er) {
2011-06-20 19:25:49 -05:00
expr_assign(fld.fold_expr(el), fld.fold_expr(er))
2011-07-27 07:19:39 -05:00
}
expr_swap(el, er) {
expr_swap(fld.fold_expr(el), fld.fold_expr(er))
2011-07-27 07:19:39 -05:00
}
expr_assign_op(op, el, er) {
2011-06-20 19:25:49 -05:00
expr_assign_op(op, fld.fold_expr(el), fld.fold_expr(er))
2011-07-27 07:19:39 -05:00
}
expr_field(el, id, tys) {
expr_field(fld.fold_expr(el), fld.fold_ident(id),
vec::map(tys, fld.fold_ty))
2011-07-27 07:19:39 -05:00
}
expr_index(el, er) {
2011-06-20 19:25:49 -05:00
expr_index(fld.fold_expr(el), fld.fold_expr(er))
2011-07-27 07:19:39 -05:00
}
expr_path(pth) { expr_path(fld.fold_path(pth)) }
expr_fail(e) { expr_fail(option::map(e, fld.fold_expr)) }
expr_break | expr_cont { e }
expr_ret(e) { expr_ret(option::map(e, fld.fold_expr)) }
2011-07-27 07:19:39 -05:00
expr_be(e) { expr_be(fld.fold_expr(e)) }
expr_log(i, lv, e) { expr_log(i, fld.fold_expr(lv),
fld.fold_expr(e)) }
2011-07-27 07:19:39 -05:00
expr_assert(e) { expr_assert(fld.fold_expr(e)) }
expr_check(m, e) { expr_check(m, fld.fold_expr(e)) }
expr_if_check(cond, tr, fl) {
expr_if_check(fld.fold_expr(cond), fld.fold_block(tr),
option::map(fl, fld.fold_expr))
2011-07-27 07:19:39 -05:00
}
expr_mac(mac) { expr_mac(fold_mac(mac)) }
}
2011-06-20 19:25:49 -05:00
}
2012-02-01 22:21:01 -06:00
fn noop_fold_ty(t: ty_, fld: ast_fold) -> ty_ {
let fold_mac = bind fold_mac_(_, fld);
fn fold_mt(mt: mt, fld: ast_fold) -> mt {
{ty: fld.fold_ty(mt.ty), mut: mt.mut}
}
fn fold_field(f: ty_field, fld: ast_fold) -> ty_field {
{node: {ident: fld.fold_ident(f.node.ident),
mt: fold_mt(f.node.mt, fld)},
span: fld.new_span(f.span)}
}
alt t {
ty_nil | ty_bot | ty_bool | ty_str {t}
ty_int(_) | ty_uint(_) | ty_float(_) {t}
ty_box(mt) {ty_box(fold_mt(mt, fld))}
ty_uniq(mt) {ty_uniq(fold_mt(mt, fld))}
ty_vec(mt) {ty_vec(fold_mt(mt, fld))}
ty_ptr(mt) {ty_ptr(fold_mt(mt, fld))}
ty_rec(fields) {ty_rec(vec::map(fields) {|f| fold_field(f, fld)})}
ty_fn(proto, decl) {ty_fn(proto, fold_fn_decl(decl, fld))}
ty_tup(tys) {ty_tup(vec::map(tys) {|ty| fld.fold_ty(ty)})}
ty_path(path, id) {ty_path(fld.fold_path(path), fld.new_id(id))}
// FIXME: constrs likely needs to be folded...
ty_constr(ty, constrs) {ty_constr(fld.fold_ty(ty), constrs)}
ty_mac(mac) {ty_mac(fold_mac(mac))}
ty_infer {t}
}
2011-06-20 19:25:49 -05:00
}
fn noop_fold_constr(c: constr_, fld: ast_fold) -> constr_ {
2011-07-27 07:19:39 -05:00
{path: fld.fold_path(c.path), args: c.args, id: c.id}
2011-06-20 19:25:49 -05:00
}
// ...nor do modules
fn noop_fold_mod(m: _mod, fld: ast_fold) -> _mod {
ret {view_items: vec::map(m.view_items, fld.fold_view_item),
items: vec::map(m.items, fld.fold_item)};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_native_mod(nm: native_mod, fld: ast_fold) -> native_mod {
ret {view_items: vec::map(nm.view_items, fld.fold_view_item),
items: vec::map(nm.items, fld.fold_native_item)}
2011-06-20 19:25:49 -05:00
}
fn noop_fold_variant(v: variant_, fld: ast_fold) -> variant_ {
fn fold_variant_arg_(va: variant_arg, fld: ast_fold) -> variant_arg {
2011-07-27 07:19:39 -05:00
ret {ty: fld.fold_ty(va.ty), id: va.id};
2011-06-20 19:25:49 -05:00
}
2011-07-27 07:19:39 -05:00
let fold_variant_arg = bind fold_variant_arg_(_, fld);
let args = vec::map(v.args, fold_variant_arg);
let fold_meta_item = bind fold_meta_item_(_, fld);
let fold_attribute = bind fold_attribute_(_, fold_meta_item);
let attrs = vec::map(v.attrs, fold_attribute);
let de = alt v.disr_expr {
some(e) {some(fld.fold_expr(e))}
none {none}
};
ret {name: v.name,
attrs: attrs,
args: args, id: v.id,
disr_expr: de};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_ident(&&i: ident, _fld: ast_fold) -> ident { ret i; }
2011-06-20 19:25:49 -05:00
fn noop_fold_path(&&p: path_, fld: ast_fold) -> path_ {
2011-07-27 07:19:39 -05:00
ret {global: p.global,
idents: vec::map(p.idents, fld.fold_ident),
types: vec::map(p.types, fld.fold_ty)};
2011-06-20 19:25:49 -05:00
}
fn noop_fold_local(l: local_, fld: ast_fold) -> local_ {
ret {ty: fld.fold_ty(l.ty),
pat: fld.fold_pat(l.pat),
init:
alt l.init {
option::none::<initializer> { l.init }
option::some::<initializer>(init) {
option::some::<initializer>({op: init.op,
expr: fld.fold_expr(init.expr)})
}
},
2011-07-27 07:19:39 -05:00
id: l.id};
2011-06-20 19:25:49 -05:00
}
/* temporarily eta-expand because of a compiler bug with using `fn<T>` as a
2011-07-21 18:47:47 -05:00
value */
fn noop_map_exprs(f: fn@(&&@expr) -> @expr, es: [@expr]) -> [@expr] {
ret vec::map(es, f);
2011-07-21 18:47:47 -05:00
}
fn noop_id(i: node_id) -> node_id { ret i; }
fn noop_span(sp: span) -> span { ret sp; }
2011-06-20 19:25:49 -05:00
fn default_ast_fold() -> @ast_fold_precursor {
ret @{fold_crate: wrap(noop_fold_crate),
fold_crate_directive: wrap(noop_fold_crate_directive),
2011-07-27 07:19:39 -05:00
fold_view_item: noop_fold_view_item,
fold_native_item: noop_fold_native_item,
fold_item: noop_fold_item,
fold_class_item: noop_fold_class_item,
2011-07-27 07:19:39 -05:00
fold_item_underscore: noop_fold_item_underscore,
fold_method: noop_fold_method,
fold_block: wrap(noop_fold_block),
fold_stmt: wrap(noop_fold_stmt),
2011-07-27 07:19:39 -05:00
fold_arm: noop_fold_arm,
fold_pat: wrap(noop_fold_pat),
fold_decl: wrap(noop_fold_decl),
fold_expr: wrap(noop_fold_expr),
fold_ty: wrap(noop_fold_ty),
fold_constr: wrap(noop_fold_constr),
2011-07-27 07:19:39 -05:00
fold_mod: noop_fold_mod,
fold_native_mod: noop_fold_native_mod,
fold_variant: wrap(noop_fold_variant),
2011-07-27 07:19:39 -05:00
fold_ident: noop_fold_ident,
fold_path: wrap(noop_fold_path),
fold_local: wrap(noop_fold_local),
map_exprs: noop_map_exprs,
new_id: noop_id,
new_span: noop_span};
2011-06-20 19:25:49 -05:00
}
fn make_fold(afp: ast_fold_precursor) -> ast_fold {
// FIXME: Have to bind all the bare functions into shared functions
// because @mutable is invariant with respect to its contents
2011-07-27 07:19:39 -05:00
let result: ast_fold =
@mutable {fold_crate: bind nf_crate_dummy(_),
fold_crate_directive: bind nf_crate_directive_dummy(_),
fold_view_item: bind nf_view_item_dummy(_),
fold_native_item: bind nf_native_item_dummy(_),
fold_item: bind nf_item_dummy(_),
fold_class_item: bind nf_class_item_dummy(_),
fold_item_underscore: bind nf_item_underscore_dummy(_),
fold_method: bind nf_method_dummy(_),
fold_block: bind nf_blk_dummy(_),
fold_stmt: bind nf_stmt_dummy(_),
fold_arm: bind nf_arm_dummy(_),
fold_pat: bind nf_pat_dummy(_),
fold_decl: bind nf_decl_dummy(_),
fold_expr: bind nf_expr_dummy(_),
fold_ty: bind nf_ty_dummy(_),
fold_constr: bind nf_constr_dummy(_),
fold_mod: bind nf_mod_dummy(_),
fold_native_mod: bind nf_native_mod_dummy(_),
fold_variant: bind nf_variant_dummy(_),
fold_ident: bind nf_ident_dummy(_),
fold_path: bind nf_path_dummy(_),
fold_local: bind nf_local_dummy(_),
map_exprs: bind noop_map_exprs(_, _),
new_id: bind noop_id(_),
new_span: bind noop_span(_)};
2011-06-20 19:25:49 -05:00
/* naturally, a macro to write these would be nice */
fn f_crate(afp: ast_fold_precursor, f: ast_fold, c: crate) -> crate {
let (n, s) = afp.fold_crate(c.node, c.span, f);
ret {node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_crate_directive(afp: ast_fold_precursor, f: ast_fold,
&&c: @crate_directive) -> @crate_directive {
let (n, s) = afp.fold_crate_directive(c.node, c.span, f);
ret @{node: n,
span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_view_item(afp: ast_fold_precursor, f: ast_fold, &&x: @view_item) ->
2011-07-27 07:19:39 -05:00
@view_item {
ret @{node: afp.fold_view_item(x.node, f),
span: afp.new_span(x.span)};
2011-06-20 19:25:49 -05:00
}
fn f_native_item(afp: ast_fold_precursor, f: ast_fold, &&x: @native_item)
-> @native_item {
2011-06-20 19:25:49 -05:00
ret afp.fold_native_item(x, f);
}
fn f_item(afp: ast_fold_precursor, f: ast_fold, &&i: @item) -> @item {
2011-06-20 19:25:49 -05:00
ret afp.fold_item(i, f);
}
fn f_class_item(afp: ast_fold_precursor, f: ast_fold,
&&ci: @class_item) -> @class_item {
@{node:
{privacy:ci.node.privacy,
decl:
@alt *ci.node.decl {
instance_var(nm, t, mt, id) {
instance_var(nm, f_ty(afp, f, t),
mt, id)
}
class_method(i) {
class_method(afp.fold_item(i, f))
}
}}, span: ci.span}
}
fn f_item_underscore(afp: ast_fold_precursor, f: ast_fold, i: item_) ->
2011-07-27 07:19:39 -05:00
item_ {
2011-06-20 19:25:49 -05:00
ret afp.fold_item_underscore(i, f);
}
fn f_method(afp: ast_fold_precursor, f: ast_fold, &&x: @method)
-> @method {
ret afp.fold_method(x, f);
2011-06-20 19:25:49 -05:00
}
fn f_block(afp: ast_fold_precursor, f: ast_fold, x: blk) -> blk {
let (n, s) = afp.fold_block(x.node, x.span, f);
ret {node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_stmt(afp: ast_fold_precursor, f: ast_fold, &&x: @stmt) -> @stmt {
let (n, s) = afp.fold_stmt(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_arm(afp: ast_fold_precursor, f: ast_fold, x: arm) -> arm {
2011-06-20 19:25:49 -05:00
ret afp.fold_arm(x, f);
}
fn f_pat(afp: ast_fold_precursor, f: ast_fold, &&x: @pat) -> @pat {
let (n, s) = afp.fold_pat(x.node, x.span, f);
ret @{id: afp.new_id(x.id),
node: n,
span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_decl(afp: ast_fold_precursor, f: ast_fold, &&x: @decl) -> @decl {
let (n, s) = afp.fold_decl(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_expr(afp: ast_fold_precursor, f: ast_fold, &&x: @expr) -> @expr {
let (n, s) = afp.fold_expr(x.node, x.span, f);
ret @{id: afp.new_id(x.id),
node: n,
span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_ty(afp: ast_fold_precursor, f: ast_fold, &&x: @ty) -> @ty {
let (n, s) = afp.fold_ty(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_constr(afp: ast_fold_precursor, f: ast_fold, &&x: @ast::constr) ->
2011-07-27 07:19:39 -05:00
@ast::constr {
let (n, s) = afp.fold_constr(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_mod(afp: ast_fold_precursor, f: ast_fold, x: _mod) -> _mod {
2011-06-20 19:25:49 -05:00
ret afp.fold_mod(x, f);
}
fn f_native_mod(afp: ast_fold_precursor, f: ast_fold, x: native_mod) ->
2011-07-27 07:19:39 -05:00
native_mod {
2011-06-20 19:25:49 -05:00
ret afp.fold_native_mod(x, f);
}
fn f_variant(afp: ast_fold_precursor, f: ast_fold, x: variant) ->
2011-07-27 07:19:39 -05:00
variant {
let (n, s) = afp.fold_variant(x.node, x.span, f);
ret {node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_ident(afp: ast_fold_precursor, f: ast_fold, &&x: ident) -> ident {
2011-06-20 19:25:49 -05:00
ret afp.fold_ident(x, f);
}
fn f_path(afp: ast_fold_precursor, f: ast_fold, &&x: @path) -> @path {
let (n, s) = afp.fold_path(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
fn f_local(afp: ast_fold_precursor, f: ast_fold, &&x: @local) -> @local {
let (n, s) = afp.fold_local(x.node, x.span, f);
ret @{node: n, span: afp.new_span(s)};
2011-06-20 19:25:49 -05:00
}
2011-07-27 07:19:39 -05:00
*result =
{fold_crate: bind f_crate(afp, result, _),
fold_crate_directive: bind f_crate_directive(afp, result, _),
fold_view_item: bind f_view_item(afp, result, _),
fold_native_item: bind f_native_item(afp, result, _),
fold_item: bind f_item(afp, result, _),
fold_class_item: bind f_class_item(afp, result, _),
2011-07-27 07:19:39 -05:00
fold_item_underscore: bind f_item_underscore(afp, result, _),
fold_method: bind f_method(afp, result, _),
fold_block: bind f_block(afp, result, _),
fold_stmt: bind f_stmt(afp, result, _),
fold_arm: bind f_arm(afp, result, _),
fold_pat: bind f_pat(afp, result, _),
fold_decl: bind f_decl(afp, result, _),
fold_expr: bind f_expr(afp, result, _),
fold_ty: bind f_ty(afp, result, _),
fold_constr: bind f_constr(afp, result, _),
fold_mod: bind f_mod(afp, result, _),
fold_native_mod: bind f_native_mod(afp, result, _),
fold_variant: bind f_variant(afp, result, _),
fold_ident: bind f_ident(afp, result, _),
fold_path: bind f_path(afp, result, _),
fold_local: bind f_local(afp, result, _),
map_exprs: afp.map_exprs,
new_id: afp.new_id,
new_span: afp.new_span};
ret result;
2011-06-20 19:25:49 -05:00
}
//
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End:
//