alt.rs 25.6 KB
Newer Older
1
import driver::session::session;
2
import lib::llvm::llvm;
3
import lib::llvm::{ValueRef, BasicBlockRef};
4
import pat_util::*;
5
import build::*;
6
import base::*;
7
import syntax::ast;
8
import syntax::ast_util;
9
import syntax::ast_util::{dummy_sp, path_to_ident};
10 11
import syntax::ast::def_id;
import syntax::codemap::span;
12
import syntax::print::pprust::pat_to_str;
13
import back::abi;
M
Marijn Haverbeke 已提交
14
import resolve::def_map;
15
import std::map::hashmap;
N
Niko Matsakis 已提交
16
import dvec::{dvec, extensions};
17

18
import common::*;
19

P
Patrick Walton 已提交
20 21
// An option identifying a branch (either a literal, a enum variant or a
// range)
P
Patrick Walton 已提交
22
enum opt {
P
Patrick Walton 已提交
23
    lit(@ast::expr),
T
Tim Chevalier 已提交
24
    var(/* disr val */int, /* variant dids */{enm: def_id, var: def_id}),
25
    range(@ast::expr, @ast::expr)
26
}
27
fn opt_eq(tcx: ty::ctxt, a: opt, b: opt) -> bool {
28
    alt (a, b) {
29
      (lit(a), lit(b)) { const_eval::compare_lit_exprs(tcx, a, b) == 0 }
30
      (range(a1, a2), range(b1, b2)) {
31 32
        const_eval::compare_lit_exprs(tcx, a1, b1) == 0 &&
        const_eval::compare_lit_exprs(tcx, a2, b2) == 0
M
Marijn Haverbeke 已提交
33
      }
34 35
      (var(a, _), var(b, _)) { a == b }
      _ { false }
36 37
    }
}
38

P
Patrick Walton 已提交
39
enum opt_result {
P
Patrick Walton 已提交
40 41
    single_result(result),
    range_result(result, result),
42
}
43
fn trans_opt(bcx: block, o: opt) -> opt_result {
44
    let _icx = bcx.insn_ctxt("alt::trans_opt");
45 46
    let ccx = bcx.ccx();
    let mut bcx = bcx;
M
Marijn Haverbeke 已提交
47
    alt o {
48 49
      lit(l) {
        alt l.node {
50
          ast::expr_lit(@{node: ast::lit_str(s), _}) {
51
            let strty = ty::mk_str(bcx.tcx());
52
            let cell = empty_dest_cell();
53
            bcx = tvec::trans_estr(bcx, s, ast::vstore_uniq, by_val(cell));
54 55
            add_clean_temp(bcx, *cell, strty);
            ret single_result(rslt(bcx, *cell));
56 57
          }
          _ {
58
            ret single_result(
59
                rslt(bcx, trans_const_expr(ccx, l)));
60 61 62
          }
        }
      }
63
      var(disr_val, _) { ret single_result(rslt(bcx, C_int(ccx, disr_val))); }
64
      range(l1, l2) {
65 66
        ret range_result(rslt(bcx, trans_const_expr(ccx, l1)),
                         rslt(bcx, trans_const_expr(ccx, l2)));
67
      }
68 69 70
    }
}

M
Marijn Haverbeke 已提交
71 72 73
fn variant_opt(tcx: ty::ctxt, pat_id: ast::node_id) -> opt {
    let vdef = ast_util::variant_def_ids(tcx.def_map.get(pat_id));
    let variants = ty::enum_variants(tcx, vdef.enm);
74
    for vec::each(*variants) {|v|
75
        if vdef.var == v.id { ret var(v.disr_val, vdef); }
76
    }
77
    core::unreachable();
78 79
}

80
type bind_map = [{ident: ast::ident, val: ValueRef}]/~;
B
Brian Anderson 已提交
81
fn assoc(key: ast::ident, list: bind_map) -> option<ValueRef> {
82
    for vec::each(list) {|elt|
B
Brian Anderson 已提交
83
        if str::eq(*elt.ident, *key) { ret some(elt.val); }
M
Marijn Haverbeke 已提交
84 85 86 87
    }
    ret none;
}

M
Marijn Haverbeke 已提交
88
type match_branch =
89
    @{pats: [@ast::pat]/~,
M
Marijn Haverbeke 已提交
90
      bound: bind_map,
91
      data: @{bodycx: block,
T
Tim Chevalier 已提交
92
              guard: option<@ast::expr>,
93
              id_map: pat_id_map}};
94
type match = [match_branch]/~;
95

96
fn has_nested_bindings(m: match, col: uint) -> bool {
97
    for vec::each(m) {|br|
98
        alt br.pats[col].node {
99
          ast::pat_ident(_, some(_)) { ret true; }
100 101 102 103 104 105 106
          _ {}
        }
    }
    ret false;
}

fn expand_nested_bindings(m: match, col: uint, val: ValueRef) -> match {
107
    let mut result = []/~;
108
    for vec::each(m) {|br|
109 110
      alt br.pats[col].node {
          ast::pat_ident(name, some(inner)) {
E
Eric Holk 已提交
111 112 113 114 115 116 117 118 119 120
            let pats = vec::append(
                vec::slice(br.pats, 0u, col),
                vec::append([inner]/~,
                            vec::view(br.pats, col + 1u, br.pats.len())));
            vec::push(result,
                      @{pats: pats,
                        bound: vec::append(
                            br.bound, [{ident: path_to_ident(name),
                                        val: val}]/~)
                                with *br});
121
          }
122
          _ { vec::push(result, br); }
123 124 125
        }
    }
    result
126 127
}

128
type enter_pat = fn(@ast::pat) -> option<[@ast::pat]/~>;
M
Marijn Haverbeke 已提交
129

M
Marijn Haverbeke 已提交
130 131
fn enter_match(dm: def_map, m: match, col: uint, val: ValueRef,
               e: enter_pat) -> match {
132
    let mut result = []/~;
133
    for vec::each(m) {|br|
B
Brian Anderson 已提交
134
        alt e(br.pats[col]) {
M
Marijn Haverbeke 已提交
135
          some(sub) {
E
Eric Holk 已提交
136 137 138
            let pats = vec::append(
                vec::append(sub, vec::view(br.pats, 0u, col)),
                vec::view(br.pats, col + 1u, br.pats.len()));
M
Marijn Haverbeke 已提交
139 140 141
            let self = br.pats[col];
            let bound = alt self.node {
              ast::pat_ident(name, none) if !pat_is_variant(dm, self) {
E
Eric Holk 已提交
142 143
                vec::append(br.bound,
                            [{ident: path_to_ident(name), val: val}]/~)
M
Marijn Haverbeke 已提交
144 145 146
              }
              _ { br.bound }
            };
147
            vec::push(result, @{pats: pats, bound: bound with *br});
M
Marijn Haverbeke 已提交
148
          }
149
          none { }
150 151 152 153 154
        }
    }
    ret result;
}

M
Marijn Haverbeke 已提交
155 156
fn enter_default(dm: def_map, m: match, col: uint, val: ValueRef) -> match {
    enter_match(dm, m, col, val) {|p|
157
        alt p.node {
158
          ast::pat_wild | ast::pat_rec(_, _) | ast::pat_tup(_) { some([]/~) }
M
Marijn Haverbeke 已提交
159
          ast::pat_ident(_, none) if !pat_is_variant(dm, p) {
160
            some([]/~)
M
Marijn Haverbeke 已提交
161 162
          }
          _ { none }
163
        }
164
    }
165 166
}

M
Marijn Haverbeke 已提交
167 168
fn enter_opt(tcx: ty::ctxt, m: match, opt: opt, col: uint,
             variant_size: uint, val: ValueRef) -> match {
169
    let dummy = @{id: 0, node: ast::pat_wild, span: dummy_sp()};
M
Marijn Haverbeke 已提交
170
    enter_match(tcx.def_map, m, col, val) {|p|
M
Marijn Haverbeke 已提交
171
        alt p.node {
M
Marijn Haverbeke 已提交
172
          ast::pat_enum(_, subpats) {
173
            if opt_eq(tcx, variant_opt(tcx, p.id), opt) {
174
              some(option::get_default(subpats,
175
                     vec::from_elem(variant_size, dummy))) }
M
Marijn Haverbeke 已提交
176 177 178
            else { none }
          }
          ast::pat_ident(_, none) if pat_is_variant(tcx.def_map, p) {
179
            if opt_eq(tcx, variant_opt(tcx, p.id), opt) { some([]/~) }
M
Marijn Haverbeke 已提交
180
            else { none }
M
Marijn Haverbeke 已提交
181 182
          }
          ast::pat_lit(l) {
183
            if opt_eq(tcx, lit(l), opt) { some([]/~) } else { none }
M
Marijn Haverbeke 已提交
184
          }
185
          ast::pat_range(l1, l2) {
186
            if opt_eq(tcx, range(l1, l2), opt) { some([]/~) } else { none }
187
          }
188
          _ { some(vec::from_elem(variant_size, dummy)) }
189 190 191 192
        }
    }
}

193
fn enter_rec(dm: def_map, m: match, col: uint, fields: [ast::ident]/~,
M
Marijn Haverbeke 已提交
194
             val: ValueRef) -> match {
195
    let dummy = @{id: 0, node: ast::pat_wild, span: dummy_sp()};
M
Marijn Haverbeke 已提交
196
    enter_match(dm, m, col, val) {|p|
M
Marijn Haverbeke 已提交
197 198
        alt p.node {
          ast::pat_rec(fpats, _) {
199
            let mut pats = []/~;
200
            for vec::each(fields) {|fname|
201
                let mut pat = dummy;
202
                for vec::each(fpats) {|fpat|
B
Brian Anderson 已提交
203
                    if str::eq(*fpat.ident, *fname) { pat = fpat.pat; break; }
204
                }
205
                vec::push(pats, pat);
206
            }
M
Marijn Haverbeke 已提交
207
            some(pats)
M
Marijn Haverbeke 已提交
208
          }
209
          _ { some(vec::from_elem(fields.len(), dummy)) }
210 211 212 213
        }
    }
}

M
Marijn Haverbeke 已提交
214 215
fn enter_tup(dm: def_map, m: match, col: uint, val: ValueRef,
             n_elts: uint) -> match {
216
    let dummy = @{id: 0, node: ast::pat_wild, span: dummy_sp()};
M
Marijn Haverbeke 已提交
217
    enter_match(dm, m, col, val) {|p|
M
Marijn Haverbeke 已提交
218
        alt p.node {
M
Marijn Haverbeke 已提交
219
          ast::pat_tup(elts) { some(elts) }
220
          _ { some(vec::from_elem(n_elts, dummy)) }
M
Marijn Haverbeke 已提交
221 222 223 224
        }
    }
}

M
Marijn Haverbeke 已提交
225
fn enter_box(dm: def_map, m: match, col: uint, val: ValueRef) -> match {
226
    let dummy = @{id: 0, node: ast::pat_wild, span: dummy_sp()};
M
Marijn Haverbeke 已提交
227
    enter_match(dm, m, col, val) {|p|
M
Marijn Haverbeke 已提交
228
        alt p.node {
229 230
          ast::pat_box(sub) { some([sub]/~) }
          _ { some([dummy]/~) }
M
Marijn Haverbeke 已提交
231 232 233 234
        }
    }
}

M
Marijn Haverbeke 已提交
235
fn enter_uniq(dm: def_map, m: match, col: uint, val: ValueRef) -> match {
236
    let dummy = @{id: 0, node: ast::pat_wild, span: dummy_sp()};
M
Marijn Haverbeke 已提交
237
    enter_match(dm, m, col, val) {|p|
238
        alt p.node {
239 240
          ast::pat_uniq(sub) { some([sub]/~) }
          _ { some([dummy]/~) }
241 242 243 244
        }
    }
}

245
fn get_options(ccx: @crate_ctxt, m: match, col: uint) -> [opt]/~ {
N
Niko Matsakis 已提交
246 247 248
    fn add_to_set(tcx: ty::ctxt, &&set: dvec<opt>, val: opt) {
        if set.any({|l| opt_eq(tcx, l, val)}) {ret;}
        set.push(val);
249 250
    }

N
Niko Matsakis 已提交
251
    let found = dvec();
252
    for vec::each(m) {|br|
M
Marijn Haverbeke 已提交
253 254
        let cur = br.pats[col];
        if pat_is_variant(ccx.tcx.def_map, cur) {
255
            add_to_set(ccx.tcx, found, variant_opt(ccx.tcx, br.pats[col].id));
M
Marijn Haverbeke 已提交
256 257
        } else {
            alt cur.node {
258
              ast::pat_lit(l) { add_to_set(ccx.tcx, found, lit(l)); }
M
Marijn Haverbeke 已提交
259
              ast::pat_range(l1, l2) {
260
                add_to_set(ccx.tcx, found, range(l1, l2));
M
Marijn Haverbeke 已提交
261 262 263
              }
              _ {}
            }
264 265
        }
    }
N
Niko Matsakis 已提交
266
    ret vec::from_mut(dvec::unwrap(found));
267 268
}

269
fn extract_variant_args(bcx: block, pat_id: ast::node_id,
T
Tim Chevalier 已提交
270
                        vdefs: {enm: def_id, var: def_id}, val: ValueRef) ->
271
   {vals: [ValueRef]/~, bcx: block} {
272
    let _icx = bcx.insn_ctxt("alt::extract_variant_args");
273
    let ccx = bcx.fcx.ccx;
M
Marijn Haverbeke 已提交
274
    let enum_ty_substs = alt check ty::get(node_id_type(bcx, pat_id)).struct {
275
      ty::ty_enum(id, substs) { assert id == vdefs.enm; substs.tps }
M
Marijn Haverbeke 已提交
276
    };
277
    let mut blobptr = val;
T
Tim Chevalier 已提交
278
    let variants = ty::enum_variants(ccx.tcx, vdefs.enm);
279 280 281
    let size = ty::enum_variant_with_id(ccx.tcx, vdefs.enm,
                                        vdefs.var).args.len();
    if size > 0u && (*variants).len() != 1u {
282
        let enumptr =
283
            PointerCast(bcx, val, T_opaque_enum_ptr(ccx));
284
        blobptr = GEPi(bcx, enumptr, [0u, 1u]/~);
285
    }
T
Tim Chevalier 已提交
286
    let vdefs_tg = vdefs.enm;
T
Tim Chevalier 已提交
287
    let vdefs_var = vdefs.var;
288 289 290 291
    let args = vec::from_fn(size) { |i|
        GEP_enum(bcx, blobptr, vdefs_tg, vdefs_var,
                 enum_ty_substs, i)
    };
M
Marijn Haverbeke 已提交
292
    ret {vals: args, bcx: bcx};
293 294
}

295 296
fn collect_record_fields(m: match, col: uint) -> [ast::ident]/~ {
    let mut fields: [ast::ident]/~ = []/~;
297
    for vec::each(m) {|br|
B
Brian Anderson 已提交
298
        alt br.pats[col].node {
M
Marijn Haverbeke 已提交
299
          ast::pat_rec(fs, _) {
300
            for vec::each(fs) {|f|
B
Brian Anderson 已提交
301
                if !vec::any(fields, {|x| str::eq(*f.ident, *x)}) {
302
                    vec::push(fields, f.ident);
M
Marijn Haverbeke 已提交
303 304
                }
            }
M
Marijn Haverbeke 已提交
305 306
          }
          _ { }
M
Marijn Haverbeke 已提交
307 308 309 310 311
        }
    }
    ret fields;
}

312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331
fn root_pats_as_necessary(bcx: block, m: match, col: uint, val: ValueRef) {
    for vec::each(m) {|br|
        let pat_id = br.pats[col].id;

        alt bcx.ccx().maps.root_map.find({id:pat_id, derefs:0u}) {
          none {}
          some(scope_id) {
            // Note: the scope_id will always be the id of the alt.  See the
            // extended comment in rustc::middle::borrowck::preserve() for
            // details (look for the case covering cat_discr).

            let ty = node_id_type(bcx, pat_id);
            let val = load_if_immediate(bcx, val, ty);
            root_value(bcx, val, ty, scope_id);
            ret; // if we kept going, we'd only be rooting same value again
          }
        }
    }
}

332
fn any_box_pat(m: match, col: uint) -> bool {
333
    for vec::each(m) {|br|
B
Brian Anderson 已提交
334
        alt br.pats[col].node { ast::pat_box(_) { ret true; } _ { } }
M
Marijn Haverbeke 已提交
335 336 337 338
    }
    ret false;
}

339
fn any_uniq_pat(m: match, col: uint) -> bool {
340
    for vec::each(m) {|br|
341 342 343 344 345
        alt br.pats[col].node { ast::pat_uniq(_) { ret true; } _ { } }
    }
    ret false;
}

346
fn any_tup_pat(m: match, col: uint) -> bool {
347
    for vec::each(m) {|br|
B
Brian Anderson 已提交
348
        alt br.pats[col].node { ast::pat_tup(_) { ret true; } _ { } }
M
Marijn Haverbeke 已提交
349 350 351 352
    }
    ret false;
}

M
Marijn Haverbeke 已提交
353
type exit_node = {bound: bind_map, from: BasicBlockRef, to: BasicBlockRef};
354
type mk_fail = fn@() -> BasicBlockRef;
355

356
fn pick_col(m: match) -> uint {
357 358
    fn score(p: @ast::pat) -> uint {
        alt p.node {
359
          ast::pat_lit(_) | ast::pat_enum(_, _) | ast::pat_range(_, _) { 1u }
360
          ast::pat_ident(_, some(p)) { score(p) }
361 362 363
          _ { 0u }
        }
    }
364
    let scores = vec::to_mut(vec::from_elem(m[0].pats.len(), 0u));
365
    for vec::each(m) {|br|
366
        let mut i = 0u;
367
        for vec::each(br.pats) {|p| scores[i] += score(p); i += 1u; }
368
    }
369 370 371
    let mut max_score = 0u;
    let mut best_col = 0u;
    let mut i = 0u;
372
    for vec::each(scores) {|score|
373 374 375 376 377
        // Irrefutable columns always go first, they'd only be duplicated in
        // the branches.
        if score == 0u { ret i; }
        // If no irrefutable ones are found, we pick the one with the biggest
        // branching factor.
B
Brian Anderson 已提交
378
        if score > max_score { max_score = score; best_col = i; }
379 380 381 382
        i += 1u;
    }
    ret best_col;
}
383

384 385
fn compile_submatch(bcx: block, m: match, vals: [ValueRef]/~,
                    chk: option<mk_fail>, &exits: [exit_node]/~) {
386
    let _icx = bcx.insn_ctxt("alt::compile_submatch");
387 388
    let mut bcx = bcx;
    let tcx = bcx.tcx(), dm = tcx.def_map;
389
    if m.len() == 0u { Br(bcx, option::get(chk)()); ret; }
390
    if m[0].pats.len() == 0u {
M
Marijn Haverbeke 已提交
391 392 393
        let data = m[0].data;
        alt data.guard {
          some(e) {
394 395
            // Temporarily set bindings. They'll be rewritten to PHI nodes
            // for the actual arm block.
396
            for data.id_map.each {|key, val|
397 398 399
                let loc = local_mem(option::get(assoc(key, m[0].bound)));
                bcx.fcx.lllocals.insert(val, loc);
            };
N
Niko Matsakis 已提交
400 401 402 403
            let {bcx: guard_cx, val} = {
                with_scope_result(bcx, e.info(), "guard") {|bcx|
                    trans_temp_expr(bcx, e)
                }
404 405
            };
            bcx = with_cond(guard_cx, Not(guard_cx, val)) {|bcx|
406
                compile_submatch(bcx, vec::tail(m), vals, chk, exits);
407
                bcx
408
            };
M
Marijn Haverbeke 已提交
409
          }
B
Brian Anderson 已提交
410
          _ { }
M
Marijn Haverbeke 已提交
411
        }
412
        if !bcx.unreachable {
413 414
            vec::push(exits, {bound: m[0].bound, from: bcx.llbb,
                       to: data.bodycx.llbb});
415
        }
416
        Br(bcx, data.bodycx.llbb);
417 418 419
        ret;
    }

420
    let col = pick_col(m);
B
Brian Anderson 已提交
421
    let val = vals[col];
422 423
    let m = if has_nested_bindings(m, col) {
                expand_nested_bindings(m, col, val)
424
            } else { m };
425

E
Eric Holk 已提交
426 427
    let vals_left = vec::append(vec::slice(vals, 0u, col),
                                vec::view(vals, col + 1u, vals.len()));
428
    let ccx = bcx.fcx.ccx;
429
    let mut pat_id = 0;
430
    for vec::each(m) {|br|
431 432
        // Find a real id (we're adding placeholder wildcard patterns, but
        // each column is guaranteed to have at least one real pattern)
B
Brian Anderson 已提交
433
        if pat_id == 0 { pat_id = br.pats[col].id; }
434
    }
M
Marijn Haverbeke 已提交
435

436 437
    root_pats_as_necessary(bcx, m, col, val);

M
Marijn Haverbeke 已提交
438
    let rec_fields = collect_record_fields(m, col);
439
    // Separate path for extracting and binding record fields
440
    if rec_fields.len() > 0u {
M
Marijn Haverbeke 已提交
441
        let fields = ty::get_fields(node_id_type(bcx, pat_id));
442
        let mut rec_vals = []/~;
443
        for vec::each(rec_fields) {|field_name|
444
            let ix = option::get(ty::field_idx(field_name, fields));
445
            vec::push(rec_vals, GEPi(bcx, val, [0u, ix]/~));
446
        }
M
Marijn Haverbeke 已提交
447
        compile_submatch(bcx, enter_rec(dm, m, col, rec_fields, val),
E
Eric Holk 已提交
448
                         vec::append(rec_vals, vals_left), chk, exits);
449 450 451
        ret;
    }

M
Marijn Haverbeke 已提交
452
    if any_tup_pat(m, col) {
453
        let tup_ty = node_id_type(bcx, pat_id);
454
        let n_tup_elts = alt ty::get(tup_ty).struct {
455
          ty::ty_tup(elts) { elts.len() }
456
          _ { ccx.sess.bug("non-tuple type in tuple pattern"); }
457
        };
458
        let mut tup_vals = []/~, i = 0u;
M
Marijn Haverbeke 已提交
459
        while i < n_tup_elts {
460
            vec::push(tup_vals, GEPi(bcx, val, [0u, i]/~));
M
Marijn Haverbeke 已提交
461 462
            i += 1u;
        }
M
Marijn Haverbeke 已提交
463
        compile_submatch(bcx, enter_tup(dm, m, col, val, n_tup_elts),
E
Eric Holk 已提交
464
                         vec::append(tup_vals, vals_left), chk, exits);
M
Marijn Haverbeke 已提交
465 466 467
        ret;
    }

M
Marijn Haverbeke 已提交
468
    // Unbox in case of a box field
M
Marijn Haverbeke 已提交
469
    if any_box_pat(m, col) {
470 471
        let llbox = Load(bcx, val);
        let box_no_addrspace = non_gc_box_cast(bcx, llbox);
472 473
        let unboxed =
            GEPi(bcx, box_no_addrspace, [0u, abi::box_field_body]/~);
E
Eric Holk 已提交
474 475
        compile_submatch(bcx, enter_box(dm, m, col, val),
                         vec::append([unboxed]/~, vals_left), chk, exits);
M
Marijn Haverbeke 已提交
476 477 478
        ret;
    }

479
    if any_uniq_pat(m, col) {
480 481
        let llbox = Load(bcx, val);
        let box_no_addrspace = non_gc_box_cast(bcx, llbox);
482 483
        let unboxed =
            GEPi(bcx, box_no_addrspace, [0u, abi::box_field_body]/~);
M
Marijn Haverbeke 已提交
484
        compile_submatch(bcx, enter_uniq(dm, m, col, val),
E
Eric Holk 已提交
485
                         vec::append([unboxed]/~, vals_left), chk, exits);
486 487 488
        ret;
    }

489
    // Decide what kind of branch we need
M
Marijn Haverbeke 已提交
490
    let opts = get_options(ccx, m, col);
P
Patrick Walton 已提交
491
    enum branch_kind { no_branch, single, switch, compare, }
492 493
    let mut kind = no_branch;
    let mut test_val = val;
494
    if opts.len() > 0u {
B
Brian Anderson 已提交
495
        alt opts[0] {
M
Marijn Haverbeke 已提交
496
          var(_, vdef) {
M
Marijn Haverbeke 已提交
497
            if (*ty::enum_variants(tcx, vdef.enm)).len() == 1u {
M
Marijn Haverbeke 已提交
498 499
                kind = single;
            } else {
500
                let enumptr =
501
                    PointerCast(bcx, val, T_opaque_enum_ptr(ccx));
502
                let discrimptr = GEPi(bcx, enumptr, [0u, 0u]/~);
503
                test_val = Load(bcx, discrimptr);
M
Marijn Haverbeke 已提交
504
                kind = switch;
505
            }
M
Marijn Haverbeke 已提交
506 507
          }
          lit(l) {
508
            test_val = Load(bcx, val);
509
            let pty = node_id_type(bcx, pat_id);
510 511
            kind = if ty::type_is_integral(pty) { switch }
                   else { compare };
M
Marijn Haverbeke 已提交
512
          }
513 514 515 516 517 518
          range(_, _) {
            test_val = Load(bcx, val);
            kind = compare;
          }
        }
    }
519
    for vec::each(opts) {|o|
520 521 522
        alt o {
          range(_, _) { kind = compare; break; }
          _ { }
523 524
        }
    }
525 526 527 528 529 530 531
    let else_cx = alt kind {
      no_branch | single { bcx }
      _ { sub_block(bcx, "match_else") }
    };
    let sw = if kind == switch {
        Switch(bcx, test_val, else_cx.llbb, opts.len())
    } else { C_int(ccx, 0) }; // Placeholder for when not using a switch
M
Marijn Haverbeke 已提交
532

533 534
    let defaults = enter_default(dm, m, col, val);
    let exhaustive = option::is_none(chk) && defaults.len() == 0u;
535 536
    let len = opts.len();
    let mut i = 0u;
537
    // Compile subtrees for each option
538
    for vec::each(opts) {|opt|
539
        i += 1u;
540
        let mut opt_cx = else_cx;
541 542 543 544 545 546 547 548 549
        if !exhaustive || i < len {
            opt_cx = sub_block(bcx, "match_case");
            alt kind {
              single { Br(bcx, opt_cx.llbb); }
              switch {
                alt check trans_opt(bcx, opt) {
                  single_result(r) {
                    llvm::LLVMAddCase(sw, r.val, opt_cx.llbb);
                    bcx = r.bcx;
550 551
                  }
                }
552 553 554
              }
              compare {
                let t = node_id_type(bcx, pat_id);
N
Niko Matsakis 已提交
555 556 557 558 559 560 561
                let {bcx: after_cx, val: matches} = {
                    with_scope_result(bcx, none, "compare_scope") {|bcx|
                        alt trans_opt(bcx, opt) {
                          single_result({bcx, val}) {
                            trans_compare(bcx, ast::eq, test_val, t, val, t)
                          }
                          range_result({val: vbegin, _}, {bcx, val: vend}) {
562
                            let {bcx, val: llge} = trans_compare(
N
Niko Matsakis 已提交
563
                                bcx, ast::ge, test_val, t, vbegin, t);
564
                            let {bcx, val: llle} = trans_compare(
N
Niko Matsakis 已提交
565
                                bcx, ast::le, test_val, t, vend, t);
566
                            {bcx: bcx, val: And(bcx, llge, llle)}
N
Niko Matsakis 已提交
567 568
                          }
                        }
569 570 571 572 573 574 575 576
                    }
                };
                bcx = sub_block(after_cx, "compare_next");
                CondBr(after_cx, matches, opt_cx.llbb, bcx.llbb);
              }
              _ { }
            }
        } else if kind == compare { Br(bcx, else_cx.llbb); }
577
        let mut size = 0u;
578
        let mut unpacked = []/~;
579
        alt opt {
M
Marijn Haverbeke 已提交
580 581
          var(_, vdef) {
            let args = extract_variant_args(opt_cx, pat_id, vdef, val);
582
            size = args.vals.len();
M
Marijn Haverbeke 已提交
583 584 585
            unpacked = args.vals;
            opt_cx = args.bcx;
          }
586
          lit(_) | range(_, _) { }
587
        }
M
Marijn Haverbeke 已提交
588
        compile_submatch(opt_cx, enter_opt(tcx, m, opt, col, size, val),
E
Eric Holk 已提交
589
                         vec::append(unpacked, vals_left), chk, exits);
590 591
    }

592 593 594 595 596 597
    // Compile the fall-through case, if any
    if !exhaustive {
        if kind == compare { Br(bcx, else_cx.llbb); }
        if kind != single {
            compile_submatch(else_cx, defaults, vals_left, chk, exits);
        }
598 599 600
    }
}

601
// Returns false for unreachable blocks
602
fn make_phi_bindings(bcx: block, map: [exit_node]/~,
603
                     ids: pat_util::pat_id_map) -> bool {
604
    let _icx = bcx.insn_ctxt("alt::make_phi_bindings");
M
Marijn Haverbeke 已提交
605
    let our_block = bcx.llbb as uint;
606
    let mut success = true, bcx = bcx;
607
    for ids.each {|name, node_id|
608 609
        let mut llbbs = []/~;
        let mut vals = []/~;
610
        for vec::each(map) {|ex|
M
Marijn Haverbeke 已提交
611
            if ex.to as uint == our_block {
612
                alt assoc(name, ex.bound) {
613 614 615 616
                  some(val) {
                    vec::push(llbbs, ex.from);
                    vec::push(vals, val);
                  }
617
                  none { }
618
                }
619 620
            }
        }
621
        if vals.len() > 0u {
622
            let local = Phi(bcx, val_ty(vals[0]), vals, llbbs);
623
            bcx.fcx.lllocals.insert(node_id, local_mem(local));
624
        } else { success = false; }
625
    };
626
    if !success {
627
        Unreachable(bcx);
628
    }
629
    ret success;
630 631
}

N
Niko Matsakis 已提交
632 633 634
fn trans_alt(bcx: block,
             alt_expr: @ast::expr,
             expr: @ast::expr,
635
             arms: [ast::arm]/~,
N
Niko Matsakis 已提交
636 637
             mode: ast::alt_mode,
             dest: dest) -> block {
638
    let _icx = bcx.insn_ctxt("alt::trans_alt");
N
Niko Matsakis 已提交
639
    with_scope(bcx, alt_expr.info(), "alt") {|bcx|
640 641
        trans_alt_inner(bcx, expr, arms, mode, dest)
    }
642 643
}

644
fn trans_alt_inner(scope_cx: block, expr: @ast::expr, arms: [ast::arm]/~,
645
                   mode: ast::alt_mode, dest: dest) -> block {
646
    let _icx = scope_cx.insn_ctxt("alt::trans_alt_inner");
647
    let bcx = scope_cx, tcx = bcx.tcx();
648
    let mut bodies = []/~, match = []/~;
649 650 651 652

    let {bcx, val, _} = trans_temp_expr(bcx, expr);
    if bcx.unreachable { ret bcx; }

653
    for vec::each(arms) {|a|
N
Niko Matsakis 已提交
654
        let body = scope_block(bcx, a.body.info(), "case_body");
M
Marijn Haverbeke 已提交
655
        let id_map = pat_util::pat_id_map(tcx.def_map, a.pats[0]);
656
        vec::push(bodies, body);
657
        for vec::each(a.pats) {|p|
658
            vec::push(match, @{pats: [p]/~,
659
                        bound: []/~,
660
                        data: @{bodycx: body, guard: a.guard,
661
                                id_map: id_map}});
662 663 664
        }
    }

665 666 667
    let mk_fail = alt mode {
      ast::alt_check {
        // Cached fail-on-fallthrough block
G
Graydon Hoare 已提交
668
        let fail_cx = @mut none;
669
        fn mk_fail(bcx: block, sp: span,
G
Graydon Hoare 已提交
670
                   done: @mut option<BasicBlockRef>) -> BasicBlockRef {
671 672 673 674 675 676
            alt *done { some(bb) { ret bb; } _ { } }
            let fail_cx = sub_block(bcx, "case_fallthrough");
            trans_fail(fail_cx, some(sp), "non-exhaustive match failure");;
            *done = some(fail_cx.llbb);
            ret fail_cx.llbb;
        }
B
Brian Anderson 已提交
677
        some({||mk_fail(scope_cx, expr.span, fail_cx)})
678 679 680
      }
      ast::alt_exhaustive { none }
    };
681
    let mut exit_map = []/~;
682
    let t = node_id_type(bcx, expr.id);
683
    let spilled = spill_if_immediate(bcx, val, t);
684
    compile_submatch(bcx, match, [spilled]/~, mk_fail, exit_map);
685

686
    let mut arm_cxs = []/~, arm_dests = []/~, i = 0u;
687
    for vec::each(arms) {|a|
B
Brian Anderson 已提交
688
        let body_cx = bodies[i];
M
Marijn Haverbeke 已提交
689 690
        let id_map = pat_util::pat_id_map(tcx.def_map, a.pats[0]);
        if make_phi_bindings(body_cx, exit_map, id_map) {
691
            let arm_dest = dup_for_join(dest);
692
            vec::push(arm_dests, arm_dest);
693
            let mut arm_cx = trans_block(body_cx, a.body, arm_dest);
694
            arm_cx = trans_block_cleanups(arm_cx, body_cx);
695
            vec::push(arm_cxs, arm_cx);
696
        }
697 698
        i += 1u;
    }
699
    join_returns(scope_cx, arm_cxs, arm_dests, dest)
700 701
}

702
// Not alt-related, but similar to the pattern-munging code above
703 704
fn bind_irrefutable_pat(bcx: block, pat: @ast::pat, val: ValueRef,
                        make_copy: bool) -> block {
705
    let _icx = bcx.insn_ctxt("alt::bind_irrefutable_pat");
706 707
    let ccx = bcx.fcx.ccx;
    let mut bcx = bcx;
708 709

    // Necessary since bind_irrefutable_pat is called outside trans_alt
M
Marijn Haverbeke 已提交
710
    alt pat.node {
711
      ast::pat_ident(_,inner) {
M
Marijn Haverbeke 已提交
712
        if pat_is_variant(bcx.tcx().def_map, pat) { ret bcx; }
713
        if make_copy {
714
            let ty = node_id_type(bcx, pat.id);
715
            let llty = type_of::type_of(ccx, ty);
716 717 718
            let alloc = alloca(bcx, llty);
            bcx = copy_val(bcx, INIT, alloc,
                                  load_if_immediate(bcx, val, ty), ty);
719
            bcx.fcx.lllocals.insert(pat.id, local_mem(alloc));
720
            add_clean(bcx, alloc, ty);
721
        } else { bcx.fcx.lllocals.insert(pat.id, local_mem(val)); }
722 723 724 725
        alt inner {
          some(pat) { bcx = bind_irrefutable_pat(bcx, pat, val, true); }
          _ {}
        }
726
      }
727
      ast::pat_enum(_, sub) {
728
        let vdefs = ast_util::variant_def_ids(ccx.tcx.def_map.get(pat.id));
729
        let args = extract_variant_args(bcx, pat.id, vdefs, val);
730
        let mut i = 0;
731
        option::iter(sub) {|sub| for vec::each(args.vals) {|argval|
732
            bcx = bind_irrefutable_pat(bcx, sub[i], argval, make_copy);
733
            i += 1;
734
        }}
735 736
      }
      ast::pat_rec(fields, _) {
M
Marijn Haverbeke 已提交
737
        let rec_fields = ty::get_fields(node_id_type(bcx, pat.id));
738
        for vec::each(fields) {|f|
739
            let ix = option::get(ty::field_idx(f.ident, rec_fields));
740
            let fldptr = GEPi(bcx, val, [0u, ix]/~);
M
Marijn Haverbeke 已提交
741
            bcx = bind_irrefutable_pat(bcx, f.pat, fldptr, make_copy);
742 743
        }
      }
M
Marijn Haverbeke 已提交
744
      ast::pat_tup(elems) {
745
        let mut i = 0u;
746
        for vec::each(elems) {|elem|
747
            let fldptr = GEPi(bcx, val, [0u, i]/~);
M
Marijn Haverbeke 已提交
748
            bcx = bind_irrefutable_pat(bcx, elem, fldptr, make_copy);
M
Marijn Haverbeke 已提交
749 750 751
            i += 1u;
        }
      }
752
      ast::pat_box(inner) {
753
        let llbox = Load(bcx, val);
B
Brian Anderson 已提交
754
        let unboxed =
755
            GEPi(bcx, llbox, [0u, abi::box_field_body]/~);
756
        bcx = bind_irrefutable_pat(bcx, inner, unboxed, true);
757
      }
758
      ast::pat_uniq(inner) {
759
        let llbox = Load(bcx, val);
760
        let unboxed =
761
            GEPi(bcx, llbox, [0u, abi::box_field_body]/~);
762
        bcx = bind_irrefutable_pat(bcx, inner, unboxed, true);
763
      }
764
      ast::pat_wild | ast::pat_lit(_) | ast::pat_range(_, _) { }
765 766 767 768
    }
    ret bcx;
}

769
// Local Variables:
770
// mode: rust
771 772 773 774 775
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End: