trans.rs 93.6 KB
Newer Older
1
import std._str;
2
import std._uint;
3 4 5
import std._vec;
import std._str.rustrt.sbuf;
import std._vec.rustrt.vbuf;
6
import std.map;
7
import std.map.hashmap;
8 9 10
import std.option;
import std.option.some;
import std.option.none;
11

12
import front.ast;
13
import driver.session;
14
import middle.typeck;
15
import back.x86;
16 17
import back.abi;

P
Patrick Walton 已提交
18 19
import middle.typeck.pat_ty;

20
import util.common;
21
import util.common.istr;
22
import util.common.new_def_hash;
23
import util.common.new_str_hash;
24 25 26

import lib.llvm.llvm;
import lib.llvm.builder;
27
import lib.llvm.target_data;
28
import lib.llvm.type_handle;
29
import lib.llvm.mk_pass_manager;
30
import lib.llvm.mk_target_data;
31
import lib.llvm.mk_type_handle;
32 33 34
import lib.llvm.llvm.ModuleRef;
import lib.llvm.llvm.ValueRef;
import lib.llvm.llvm.TypeRef;
35
import lib.llvm.llvm.TypeHandleRef;
36 37
import lib.llvm.llvm.BuilderRef;
import lib.llvm.llvm.BasicBlockRef;
38

39 40
import lib.llvm.False;
import lib.llvm.True;
41

42 43 44 45 46 47 48
state obj namegen(mutable int i) {
    fn next(str prefix) -> str {
        i += 1;
        ret prefix + istr(i);
    }
}

49 50
type glue_fns = rec(ValueRef activate_glue,
                    ValueRef yield_glue,
51
                    ValueRef exit_task_glue,
52 53
                    vec[ValueRef] upcall_glues);

54
tag arity { nullary; n_ary; }
55
type tag_info = rec(type_handle th,
56 57
                    mutable vec[tup(ast.def_id,arity)] variants,
                    mutable uint size);
58

59
type ty_info = rec(ValueRef take_glue, ValueRef drop_glue);
60

61
state type crate_ctxt = rec(session.session sess,
62
                            ModuleRef llmod,
63
                            target_data td,
64
                            hashmap[str, ValueRef] upcalls,
65
                            hashmap[str, ValueRef] intrinsics,
66 67
                            hashmap[str, ValueRef] item_names,
                            hashmap[ast.def_id, ValueRef] item_ids,
68
                            hashmap[ast.def_id, @ast.item] items,
69
                            hashmap[ast.def_id, @tag_info] tags,
70
                            hashmap[@typeck.ty, @ty_info] types,
71 72 73
                            @glue_fns glues,
                            namegen names,
                            str path);
74

75 76
state type fn_ctxt = rec(ValueRef llfn,
                         ValueRef lltaskptr,
77
                         hashmap[ast.def_id, ValueRef] llargs,
78
                         hashmap[ast.def_id, ValueRef] lllocals,
79
                         @crate_ctxt ccx);
80

81
tag cleanup {
82
    clean(fn(@block_ctxt cx) -> result);
83 84 85 86
}

state type block_ctxt = rec(BasicBlockRef llbb,
                            builder build,
87
                            block_parent parent,
88
                            bool is_scope,
89 90 91
                            mutable vec[cleanup] cleanups,
                            @fn_ctxt fcx);

92 93 94 95 96 97 98 99
// FIXME: we should be able to use option.t[@block_parent] here but
// the infinite-tag check in rustboot gets upset.

tag block_parent {
    parent_none;
    parent_some(@block_ctxt);
}

100

101 102 103 104 105 106 107 108
state type result = rec(mutable @block_ctxt bcx,
                        mutable ValueRef val);

fn res(@block_ctxt bcx, ValueRef val) -> result {
    ret rec(mutable bcx = bcx,
            mutable val = val);
}

109 110 111 112 113 114 115 116 117 118 119
fn ty_str(TypeRef t) -> str {
    ret lib.llvm.type_to_str(t);
}

fn val_ty(ValueRef v) -> TypeRef {
    ret llvm.LLVMTypeOf(v);
}

fn val_str(ValueRef v) -> str {
    ret ty_str(val_ty(v));
}
120 121 122 123


// LLVM type constructors.

124 125 126 127 128 129 130 131 132 133 134
fn T_void() -> TypeRef {
    // Note: For the time being llvm is kinda busted here, it has the notion
    // of a 'void' type that can only occur as part of the signature of a
    // function, but no general unit type of 0-sized value. This is, afaict,
    // vestigial from its C heritage, and we'll be attempting to submit a
    // patch upstream to fix it. In the mean time we only model function
    // outputs (Rust functions and C functions) using T_void, and model the
    // Rust general purpose nil type you can construct as 1-bit (always
    // zero). This makes the result incorrect for now -- things like a tuple
    // of 10 nil values will have 10-bit size -- but it doesn't seem like we
    // have any other options until it's fixed upstream.
135 136 137
    ret llvm.LLVMVoidType();
}

138 139 140 141 142
fn T_nil() -> TypeRef {
    // NB: See above in T_void().
    ret llvm.LLVMInt1Type();
}

143 144 145 146
fn T_i1() -> TypeRef {
    ret llvm.LLVMInt1Type();
}

147 148 149 150 151 152 153 154 155
fn T_i8() -> TypeRef {
    ret llvm.LLVMInt8Type();
}

fn T_i16() -> TypeRef {
    ret llvm.LLVMInt16Type();
}

fn T_i32() -> TypeRef {
156 157 158
    ret llvm.LLVMInt32Type();
}

159 160 161 162
fn T_i64() -> TypeRef {
    ret llvm.LLVMInt64Type();
}

163 164 165 166 167 168 169 170
fn T_f32() -> TypeRef {
    ret llvm.LLVMFloatType();
}

fn T_f64() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

171 172 173 174
fn T_bool() -> TypeRef {
    ret T_i1();
}

175 176 177 178 179
fn T_int() -> TypeRef {
    // FIXME: switch on target type.
    ret T_i32();
}

180 181 182 183
fn T_char() -> TypeRef {
    ret T_i32();
}

184 185 186 187
fn T_fn(vec[TypeRef] inputs, TypeRef output) -> TypeRef {
    ret llvm.LLVMFunctionType(output,
                              _vec.buf[TypeRef](inputs),
                              _vec.len[TypeRef](inputs),
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206
                              False);
}

fn T_ptr(TypeRef t) -> TypeRef {
    ret llvm.LLVMPointerType(t, 0u);
}

fn T_struct(vec[TypeRef] elts) -> TypeRef {
    ret llvm.LLVMStructType(_vec.buf[TypeRef](elts),
                            _vec.len[TypeRef](elts),
                            False);
}

fn T_opaque() -> TypeRef {
    ret llvm.LLVMOpaqueType();
}

fn T_task() -> TypeRef {
    ret T_struct(vec(T_int(),      // Refcount
207 208 209 210 211 212 213 214 215 216
                     T_int(),      // Delegate pointer
                     T_int(),      // Stack segment pointer
                     T_int(),      // Runtime SP
                     T_int(),      // Rust SP
                     T_int(),      // GC chain
                     T_int(),      // Domain pointer
                     T_int()       // Crate cache pointer
                     ));
}

217 218 219 220 221 222 223 224 225 226 227 228 229 230 231
fn T_tydesc() -> TypeRef {
    auto pvoid = T_ptr(T_i8());
    auto glue_fn_ty = T_ptr(T_fn(vec(T_taskptr(), pvoid), T_void()));
    ret T_struct(vec(pvoid,             // first_param
                     T_int(),           // size
                     T_int(),           // align
                     glue_fn_ty,        // copy_glue_off
                     glue_fn_ty,        // drop_glue_off
                     glue_fn_ty,        // free_glue_off
                     glue_fn_ty,        // sever_glue_off
                     glue_fn_ty,        // mark_glue_off
                     glue_fn_ty,        // obj_drop_glue_off
                     glue_fn_ty));      // is_stateful
}

232 233 234 235
fn T_array(TypeRef t, uint n) -> TypeRef {
    ret llvm.LLVMArrayType(t, n);
}

236 237 238 239 240
fn T_vec(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(),       // Refcount
                     T_int(),       // Alloc
                     T_int(),       // Fill
                     T_array(t, 0u) // Body elements
241 242 243
                     ));
}

244 245
fn T_str() -> TypeRef {
    ret T_vec(T_i8());
246 247
}

248 249 250 251
fn T_box(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(), t));
}

252 253 254 255 256 257 258 259 260 261 262 263 264 265
fn T_crate() -> TypeRef {
    ret T_struct(vec(T_int(),      // ptrdiff_t image_base_off
                     T_int(),      // uintptr_t self_addr
                     T_int(),      // ptrdiff_t debug_abbrev_off
                     T_int(),      // size_t debug_abbrev_sz
                     T_int(),      // ptrdiff_t debug_info_off
                     T_int(),      // size_t debug_info_sz
                     T_int(),      // size_t activate_glue_off
                     T_int(),      // size_t yield_glue_off
                     T_int(),      // size_t unwind_glue_off
                     T_int(),      // size_t gc_glue_off
                     T_int(),      // size_t main_exit_task_glue_off
                     T_int(),      // int n_rust_syms
                     T_int(),      // int n_c_syms
266
                     T_int()       // int n_libs
267 268 269
                     ));
}

270 271 272 273 274 275
fn T_double() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

fn T_taskptr() -> TypeRef {
    ret T_ptr(T_task());
276 277
}

278
fn type_of(@crate_ctxt cx, @typeck.ty t) -> TypeRef {
279 280
    let TypeRef llty = type_of_inner(cx, t);
    check (llty as int != 0);
281
    llvm.LLVMAddTypeName(cx.llmod, _str.buf(typeck.ty_to_str(t)), llty);
282 283 284
    ret llty;
}

285 286 287 288
fn type_of_fn(@crate_ctxt cx,
              vec[typeck.arg] inputs,
              @typeck.ty output) -> TypeRef {
    let vec[TypeRef] atys = vec(T_taskptr());
289 290 291 292 293 294 295 296 297

    auto fn_ty = typeck.plain_ty(typeck.ty_fn(inputs, output));
    auto ty_param_count = typeck.count_ty_params(fn_ty);
    auto i = 0u;
    while (i < ty_param_count) {
        atys += T_tydesc();
        i += 1u;
    }

298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318
    for (typeck.arg arg in inputs) {
        let TypeRef t = type_of(cx, arg.ty);
        alt (arg.mode) {
            case (ast.alias) {
                t = T_ptr(t);
            }
            case (_) { /* fall through */  }
        }
        atys += t;
    }

    auto ret_ty;
    if (typeck.type_is_nil(output)) {
        ret_ty = llvm.LLVMVoidType();
    } else {
        ret_ty = type_of(cx, output);
    }

    ret T_fn(atys, ret_ty);
}

319
fn type_of_inner(@crate_ctxt cx, @typeck.ty t) -> TypeRef {
320 321 322 323 324 325
    alt (t.struct) {
        case (typeck.ty_nil) { ret T_nil(); }
        case (typeck.ty_bool) { ret T_bool(); }
        case (typeck.ty_int) { ret T_int(); }
        case (typeck.ty_uint) { ret T_int(); }
        case (typeck.ty_machine(?tm)) {
326 327 328 329 330 331 332 333 334 335 336 337 338
            alt (tm) {
                case (common.ty_i8) { ret T_i8(); }
                case (common.ty_u8) { ret T_i8(); }
                case (common.ty_i16) { ret T_i16(); }
                case (common.ty_u16) { ret T_i16(); }
                case (common.ty_i32) { ret T_i32(); }
                case (common.ty_u32) { ret T_i32(); }
                case (common.ty_i64) { ret T_i64(); }
                case (common.ty_u64) { ret T_i64(); }
                case (common.ty_f32) { ret T_f32(); }
                case (common.ty_f64) { ret T_f64(); }
            }
        }
339
        case (typeck.ty_char) { ret T_char(); }
340
        case (typeck.ty_str) { ret T_ptr(T_str()); }
341 342 343
        case (typeck.ty_tag(?tag_id)) {
            ret llvm.LLVMResolveTypeHandle(cx.tags.get(tag_id).th.llth);
        }
344
        case (typeck.ty_box(?t)) {
345 346
            ret T_ptr(T_box(type_of(cx, t)));
        }
347
        case (typeck.ty_vec(?t)) {
348
            ret T_ptr(T_vec(type_of(cx, t)));
349
        }
350
        case (typeck.ty_tup(?elts)) {
351
            let vec[TypeRef] tys = vec();
352 353
            for (@typeck.ty elt in elts) {
                tys += type_of(cx, elt);
354 355 356
            }
            ret T_struct(tys);
        }
357 358 359 360 361 362 363
        case (typeck.ty_rec(?fields)) {
            let vec[TypeRef] tys = vec();
            for (typeck.field f in fields) {
                tys += type_of(cx, f.ty);
            }
            ret T_struct(tys);
        }
364
        case (typeck.ty_fn(?args, ?out)) {
365 366 367 368 369 370 371
            ret type_of_fn(cx, args, out);
        }
        case (typeck.ty_obj(?meths)) {
            let vec[TypeRef] mtys = vec();
            for (typeck.method m in meths) {
                let TypeRef mty = type_of_fn(cx, m.inputs, m.output);
                mtys += T_ptr(mty);
372
            }
373 374 375 376 377
            let TypeRef vtbl = T_struct(mtys);
            let TypeRef pair =
                T_struct(vec(T_ptr(vtbl),
                             T_ptr(T_box(T_opaque()))));
            ret pair;
378
        }
379 380
        case (typeck.ty_var(_)) {
            log "ty_var in trans.type_of";
381 382 383 384
            fail;
        }
        case (typeck.ty_param(_)) {
            ret T_ptr(T_i8());
385 386 387 388 389
        }
    }
    fail;
}

390 391 392 393 394 395 396 397
fn type_of_arg(@crate_ctxt cx, &typeck.arg arg) -> TypeRef {
    auto ty = type_of(cx, arg.ty);
    if (arg.mode == ast.alias) {
        ty = T_ptr(ty);
    }
    ret ty;
}

398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413
// Name sanitation. LLVM will happily accept identifiers with weird names, but
// gas doesn't!

fn sanitize(str s) -> str {
    auto result = "";
    for (u8 c in s) {
        if (c == ('@' as u8)) {
            result += "boxed_";
        } else {
            auto v = vec(c);
            result += _str.from_bytes(v);
        }
    }
    ret result;
}

414 415 416 417 418 419
// LLVM constant constructors.

fn C_null(TypeRef t) -> ValueRef {
    ret llvm.LLVMConstNull(t);
}

420
fn C_integral(int i, TypeRef t) -> ValueRef {
421 422 423 424 425 426
    // FIXME. We can't use LLVM.ULongLong with our existing minimal native
    // API, which only knows word-sized args.  Lucky for us LLVM has a "take a
    // string encoding" version.  Hilarious. Please fix to handle:
    //
    // ret llvm.LLVMConstInt(T_int(), t as LLVM.ULongLong, False);
    //
427 428 429
    ret llvm.LLVMConstIntOfString(t, _str.buf(istr(i)), 10);
}

430 431 432 433 434
fn C_nil() -> ValueRef {
    // NB: See comment above in T_void().
    ret C_integral(0, T_i1());
}

435 436
fn C_bool(bool b) -> ValueRef {
    if (b) {
437
        ret C_integral(1, T_bool());
438
    } else {
439
        ret C_integral(0, T_bool());
440 441 442
    }
}

443 444
fn C_int(int i) -> ValueRef {
    ret C_integral(i, T_int());
445 446
}

447
fn C_str(@crate_ctxt cx, str s) -> ValueRef {
448
    auto sc = llvm.LLVMConstString(_str.buf(s), _str.byte_len(s), False);
449
    auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(sc),
450 451
                                _str.buf(cx.names.next("str")));
    llvm.LLVMSetInitializer(g, sc);
452
    llvm.LLVMSetGlobalConstant(g, True);
453
    ret g;
454 455
}

456 457 458 459 460 461 462 463 464 465 466
fn C_zero_byte_arr(uint size) -> ValueRef {
    auto i = 0u;
    let vec[ValueRef] elts = vec();
    while (i < size) {
        elts += vec(C_integral(0, T_i8()));
        i += 1u;
    }
    ret llvm.LLVMConstArray(T_i8(), _vec.buf[ValueRef](elts),
                            _vec.len[ValueRef](elts));
}

467 468 469 470 471 472
fn C_struct(vec[ValueRef] elts) -> ValueRef {
    ret llvm.LLVMConstStruct(_vec.buf[ValueRef](elts),
                             _vec.len[ValueRef](elts),
                             False);
}

473
fn C_tydesc(TypeRef t) -> ValueRef {
474 475 476 477 478 479 480 481 482 483
    ret C_struct(vec(C_null(T_ptr(T_opaque())),        // first_param
                     llvm.LLVMSizeOf(t),               // size
                     llvm.LLVMAlignOf(t),              // align
                     C_null(T_ptr(T_opaque())),        // copy_glue_off
                     C_null(T_ptr(T_opaque())),        // drop_glue_off
                     C_null(T_ptr(T_opaque())),        // free_glue_off
                     C_null(T_ptr(T_opaque())),        // sever_glue_off
                     C_null(T_ptr(T_opaque())),        // mark_glue_off
                     C_null(T_ptr(T_opaque())),        // obj_drop_glue_off
                     C_null(T_ptr(T_opaque()))));      // is_stateful
484 485
}

486
fn decl_fn(ModuleRef llmod, str name, uint cc, TypeRef llty) -> ValueRef {
487 488
    let ValueRef llfn =
        llvm.LLVMAddFunction(llmod, _str.buf(name), llty);
489
    llvm.LLVMSetFunctionCallConv(llfn, cc);
490 491 492
    ret llfn;
}

493 494
fn decl_cdecl_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMCCallConv, llty);
495 496
}

497 498
fn decl_fastcall_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMFastCallConv, llty);
499 500
}

501
fn decl_glue(ModuleRef llmod, str s) -> ValueRef {
502
    ret decl_cdecl_fn(llmod, s, T_fn(vec(T_taskptr()), T_void()));
503 504
}

505
fn decl_upcall(ModuleRef llmod, uint _n) -> ValueRef {
506 507 508 509
    // It doesn't actually matter what type we come up with here, at the
    // moment, as we cast the upcall function pointers to int before passing
    // them to the indirect upcall-invocation glue.  But eventually we'd like
    // to call them directly, once we have a calling convention worked out.
510
    let int n = _n as int;
511
    let str s = abi.upcall_glue_name(n);
512
    let vec[TypeRef] args =
513 514
        vec(T_taskptr(), // taskptr
            T_int())     // callee
515 516
        + _vec.init_elt[TypeRef](T_int(), n as uint);

517
    ret decl_fastcall_fn(llmod, s, T_fn(args, T_int()));
518 519
}

520
fn get_upcall(@crate_ctxt cx, str name, int n_args) -> ValueRef {
521 522 523
    if (cx.upcalls.contains_key(name)) {
        ret cx.upcalls.get(name);
    }
524
    auto inputs = vec(T_taskptr());
525
    inputs += _vec.init_elt[TypeRef](T_int(), n_args as uint);
526
    auto output = T_int();
527
    auto f = decl_cdecl_fn(cx.llmod, name, T_fn(inputs, output));
528 529 530 531
    cx.upcalls.insert(name, f);
    ret f;
}

532
fn trans_upcall(@block_ctxt cx, str name, vec[ValueRef] args) -> result {
533
    let int n = _vec.len[ValueRef](args) as int;
534
    let ValueRef llupcall = get_upcall(cx.fcx.ccx, name, n);
535 536
    llupcall = llvm.LLVMConstPointerCast(llupcall, T_int());

537
    let ValueRef llglue = cx.fcx.ccx.glues.upcall_glues.(n);
538 539 540 541
    let vec[ValueRef] call_args = vec(cx.fcx.lltaskptr, llupcall);
    for (ValueRef a in args) {
        call_args += cx.build.ZExtOrBitCast(a, T_int());
    }
542
    ret res(cx, cx.build.FastCall(llglue, call_args));
543 544
}

545 546 547
fn trans_non_gc_free(@block_ctxt cx, ValueRef v) -> result {
    ret trans_upcall(cx, "upcall_free", vec(cx.build.PtrToInt(v, T_int()),
                                            C_int(0)));
548 549
}

550 551 552 553 554 555 556 557 558 559 560 561 562 563
fn find_scope_cx(@block_ctxt cx) -> @block_ctxt {
    if (cx.is_scope) {
        ret cx;
    }
    alt (cx.parent) {
        case (parent_some(?b)) {
            be find_scope_cx(b);
        }
        case (parent_none) {
            fail;
        }
    }
}

564
fn trans_malloc(@block_ctxt cx, @typeck.ty t) -> result {
565
    auto scope_cx = find_scope_cx(cx);
566 567 568 569 570 571 572
    auto ptr_ty = type_of(cx.fcx.ccx, t);
    auto body_ty = lib.llvm.llvm.LLVMGetElementType(ptr_ty);
    // FIXME: need a table to collect tydesc globals.
    auto tydesc = C_int(0);
    auto sz = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(body_ty), T_int());
    auto sub = trans_upcall(cx, "upcall_malloc", vec(sz, tydesc));
    sub.val = sub.bcx.build.IntToPtr(sub.val, ptr_ty);
573
    scope_cx.cleanups += clean(bind drop_ty(_, sub.val, t));
574 575 576 577
    ret sub;
}


578 579 580 581 582 583 584 585 586 587
// Glue and referent count twiddling

fn get_ty_info(@crate_ctxt cx, @typeck.ty ty) -> @ty_info {
    if (!cx.types.contains_key(ty)) {
        make_ty_info(cx, ty);
    }
    ret cx.types.get(ty);
}

fn make_ty_info(@crate_ctxt cx, @typeck.ty ty) {
588 589 590 591 592
    auto tg = make_take_glue;
    auto take_glue = make_generic_glue(cx, ty, "take", tg);
    auto dg = make_drop_glue;
    auto drop_glue = make_generic_glue(cx, ty, "drop", dg);
    cx.types.insert(ty, @rec(take_glue=take_glue, drop_glue=drop_glue));
593 594
}

595 596
fn make_generic_glue(@crate_ctxt cx, @typeck.ty t, str name,
                     val_and_ty_fn helper) -> ValueRef {
597 598 599 600 601 602 603 604
    auto arg_t;
    if (typeck.type_is_structural(t)) {
        arg_t = T_ptr(type_of(cx, t));
    } else {
        arg_t = type_of(cx, t);
    }
    auto llfnty = T_fn(vec(T_taskptr(), arg_t), T_void());

605
    auto fn_name = cx.names.next("_rust_" + name) + "." + typeck.ty_to_str(t);
606 607 608 609 610 611 612 613
    fn_name = sanitize(fn_name);
    auto llfn = decl_fastcall_fn(cx.llmod, fn_name, llfnty);

    auto fcx = new_fn_ctxt(cx, fn_name, llfn);
    auto bcx = new_top_block_ctxt(fcx);

    auto llval = llvm.LLVMGetParam(llfn, 1u);

614
    auto res = helper(bcx, llval, t);
615

616
    res.bcx.build.RetVoid();
617 618 619
    ret llfn;
}

620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653
fn make_take_glue(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result {
    if (typeck.type_is_boxed(t)) {
        ret incr_refcnt_of_boxed(cx, v);

    } else if (typeck.type_is_binding(t)) {
        cx.fcx.ccx.sess.unimpl("binding type in trans.incr_all_refcnts");

    } else if (typeck.type_is_structural(t)) {
        ret iter_structural_ty(cx, v, t,
                               bind incr_all_refcnts(_, _, _));
    }
    ret res(cx, C_nil());
}

fn incr_refcnt_of_boxed(@block_ctxt cx, ValueRef box_ptr) -> result {
    auto rc_ptr = cx.build.GEP(box_ptr, vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
    auto rc = cx.build.Load(rc_ptr);

    auto rc_adj_cx = new_sub_block_ctxt(cx, "rc++");
    auto next_cx = new_sub_block_ctxt(cx, "next");

    auto const_test = cx.build.ICmp(lib.llvm.LLVMIntEQ,
                                    C_int(abi.const_refcount as int), rc);
    cx.build.CondBr(const_test, next_cx.llbb, rc_adj_cx.llbb);

    rc = rc_adj_cx.build.Add(rc, C_int(1));
    rc_adj_cx.build.Store(rc, rc_ptr);
    rc_adj_cx.build.Br(next_cx.llbb);

    ret res(next_cx, C_nil());
}

fn make_drop_glue(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result {
654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711
    alt (t.struct) {
        case (typeck.ty_str) {
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind trans_non_gc_free(_, v),
                                        "free string",
                                        T_int(), C_int(0));
        }

        case (typeck.ty_vec(_)) {
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @typeck.ty t) -> result {
                auto res = iter_sequence(cx, v, t, bind drop_ty(_,_,_));
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(res.bcx, v);
            }
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind hit_zero(_, v, t),
                                        "free vector",
                                        T_int(), C_int(0));
        }

        case (typeck.ty_box(?body_ty)) {
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @typeck.ty body_ty) -> result {
                auto body = cx.build.GEP(v,
                                         vec(C_int(0),
                                             C_int(abi.box_rc_field_body)));

                auto body_val = load_non_structural(cx, body, body_ty);
                auto res = drop_ty(cx, body_val, body_ty);
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(res.bcx, v);
            }
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind hit_zero(_, v, body_ty),
                                        "free box",
                                        T_int(), C_int(0));
        }

        case (_) {
            if (typeck.type_is_structural(t)) {
                ret iter_structural_ty(cx, v, t,
                                       bind drop_ty(_, _, _));

            } else if (typeck.type_is_binding(t)) {
                cx.fcx.ccx.sess.unimpl("binding type in " +
                                       "trans.make_drop_glue_inner");

            } else if (typeck.type_is_scalar(t) ||
                       typeck.type_is_nil(t)) {
                ret res(cx, C_nil());
            }
        }
    }
    cx.fcx.ccx.sess.bug("bad type in trans.make_drop_glue_inner");
    fail;
}

712 713
fn decr_refcnt_and_if_zero(@block_ctxt cx,
                           ValueRef box_ptr,
714
                           fn(@block_ctxt cx) -> result inner,
715
                           str inner_name,
716
                           TypeRef t_else, ValueRef v_else) -> result {
717

718
    auto load_rc_cx = new_sub_block_ctxt(cx, "load rc");
719 720 721 722
    auto rc_adj_cx = new_sub_block_ctxt(cx, "rc--");
    auto inner_cx = new_sub_block_ctxt(cx, inner_name);
    auto next_cx = new_sub_block_ctxt(cx, "next");

723 724
    auto null_test = cx.build.IsNull(box_ptr);
    cx.build.CondBr(null_test, next_cx.llbb, load_rc_cx.llbb);
725

726 727 728 729 730 731 732 733 734 735

    auto rc_ptr = load_rc_cx.build.GEP(box_ptr,
                                       vec(C_int(0),
                                           C_int(abi.box_rc_field_refcnt)));

    auto rc = load_rc_cx.build.Load(rc_ptr);
    auto const_test =
        load_rc_cx.build.ICmp(lib.llvm.LLVMIntEQ,
                              C_int(abi.const_refcount as int), rc);
    load_rc_cx.build.CondBr(const_test, next_cx.llbb, rc_adj_cx.llbb);
736 737 738 739

    rc = rc_adj_cx.build.Sub(rc, C_int(1));
    rc_adj_cx.build.Store(rc, rc_ptr);
    auto zero_test = rc_adj_cx.build.ICmp(lib.llvm.LLVMIntEQ, C_int(0), rc);
740 741 742 743
    rc_adj_cx.build.CondBr(zero_test, inner_cx.llbb, next_cx.llbb);

    auto inner_res = inner(inner_cx);
    inner_res.bcx.build.Br(next_cx.llbb);
744

745
    auto phi = next_cx.build.Phi(t_else,
746
                                 vec(v_else, v_else, v_else, inner_res.val),
747
                                 vec(cx.llbb,
748
                                     load_rc_cx.llbb,
749
                                     rc_adj_cx.llbb,
750 751
                                     inner_res.bcx.llbb));

752
    ret res(next_cx, phi);
753 754
}

755 756 757 758 759 760 761 762 763 764 765 766 767
fn type_of_variant(@crate_ctxt cx, &ast.variant v) -> TypeRef {
    let vec[TypeRef] lltys = vec();
    alt (typeck.ann_to_type(v.ann).struct) {
        case (typeck.ty_fn(?args, _)) {
            for (typeck.arg arg in args) {
                lltys += vec(type_of(cx, arg.ty));
            }
        }
        case (_) { fail; }
    }
    ret T_struct(lltys);
}

768 769 770 771 772 773 774 775 776 777 778 779 780
type val_and_ty_fn =
    fn(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result;

// Iterates through the elements of a tup, rec or tag.
fn iter_structural_ty(@block_ctxt cx,
                      ValueRef v,
                      @typeck.ty t,
                      val_and_ty_fn f)
    -> result {
    let result r = res(cx, C_nil());
    alt (t.struct) {
        case (typeck.ty_tup(?args)) {
            let int i = 0;
781
            for (@typeck.ty arg in args) {
782
                auto elt = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
783 784 785
                r = f(r.bcx,
                      load_non_structural(r.bcx, elt, arg),
                      arg);
786 787 788
                i += 1;
            }
        }
789 790 791 792
        case (typeck.ty_rec(?fields)) {
            let int i = 0;
            for (typeck.field fld in fields) {
                auto llfld = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
793 794 795
                r = f(r.bcx,
                      load_non_structural(r.bcx, llfld, fld.ty),
                      fld.ty);
796 797 798
                i += 1;
            }
        }
799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820
        case (typeck.ty_tag(?tid)) {
            check (cx.fcx.ccx.tags.contains_key(tid));
            auto info = cx.fcx.ccx.tags.get(tid);
            auto n_variants = _vec.len[tup(ast.def_id,arity)](info.variants);

            // Look up the tag in the typechecked AST.
            check (cx.fcx.ccx.items.contains_key(tid));
            auto tag_item = cx.fcx.ccx.items.get(tid);
            let vec[ast.variant] variants = vec();  // FIXME: typestate bug
            alt (tag_item.node) {
                case (ast.item_tag(_, ?vs, _, _)) {
                    variants = vs;
                }
                case (_) {
                    log "trans: ty_tag doesn't actually refer to a tag";
                    fail;
                }
            }

            auto lldiscrim_ptr = cx.build.GEP(v, vec(C_int(0), C_int(0)));
            auto llunion_ptr = cx.build.GEP(v, vec(C_int(0), C_int(1)));
            auto lldiscrim = cx.build.Load(lldiscrim_ptr);
G
Graydon Hoare 已提交
821

822 823 824 825 826
            auto unr_cx = new_sub_block_ctxt(cx, "tag-iter-unr");
            unr_cx.build.Unreachable();

            auto llswitch = cx.build.Switch(lldiscrim, unr_cx.llbb,
                                            n_variants);
G
Graydon Hoare 已提交
827

828 829 830 831 832 833 834 835 836 837 838 839
            auto next_cx = new_sub_block_ctxt(cx, "tag-iter-next");

            auto i = 0u;
            for (tup(ast.def_id,arity) variant in info.variants) {
                auto variant_cx = new_sub_block_ctxt(cx, "tag-iter-variant-" +
                                                     _uint.to_str(i, 10u));
                llvm.LLVMAddCase(llswitch, C_int(i as int), variant_cx.llbb);

                alt (variant._1) {
                    case (n_ary) {
                        let vec[ValueRef] vals = vec(C_int(0), C_int(1),
                                                     C_int(i as int));
840
                        auto llvar = variant_cx.build.GEP(v, vals);
841 842
                        auto llvarty = type_of_variant(cx.fcx.ccx,
                                                       variants.(i));
843

844
                        auto fn_ty = typeck.ann_to_type(variants.(i).ann);
845 846
                        alt (fn_ty.struct) {
                            case (typeck.ty_fn(?args, _)) {
847 848 849 850
                                auto llvarp = variant_cx.build.
                                    TruncOrBitCast(llunion_ptr,
                                                   T_ptr(llvarty));

851 852
                                auto j = 0u;
                                for (typeck.arg a in args) {
853 854
                                    auto llfldp = variant_cx.build.GEP(llvarp,
                                        vec(C_int(0), C_int(j as int)));
855 856
                                    auto llfld =
                                        load_non_structural(variant_cx,
857
                                                            llfldp, a.ty);
858

859 860 861 862 863 864 865 866 867
                                    auto res = f(variant_cx, llfld, a.ty);
                                    variant_cx = res.bcx;
                                    j += 1u;
                                }
                            }
                            case (_) { fail; }
                        }

                        variant_cx.build.Br(next_cx.llbb);
868 869 870 871 872 873 874 875 876 877 878 879
                    }
                    case (nullary) {
                        // Nothing to do.
                        variant_cx.build.Br(next_cx.llbb);
                    }
                }

                i += 1u;
            }

            ret res(next_cx, C_nil());
        }
880 881 882
        case (_) {
            cx.fcx.ccx.sess.unimpl("type in iter_structural_ty");
        }
883
    }
884
    ret r;
885 886
}

887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902
// Iterates through the elements of a vec or str.
fn iter_sequence(@block_ctxt cx,
                 ValueRef v,
                 @typeck.ty ty,
                 val_and_ty_fn f) -> result {

    fn iter_sequence_body(@block_ctxt cx,
                          ValueRef v,
                          @typeck.ty elt_ty,
                          val_and_ty_fn f,
                          bool trailing_null) -> result {

        auto p0 = cx.build.GEP(v, vec(C_int(0),
                                      C_int(abi.vec_elt_data)));
        auto lenptr = cx.build.GEP(v, vec(C_int(0),
                                          C_int(abi.vec_elt_fill)));
903 904 905 906 907

        auto llunit_ty = type_of(cx.fcx.ccx, elt_ty);
        auto unit_sz = llvm.LLVMConstIntCast(llvm.LLVMSizeOf(llunit_ty),
                                             T_int(), False);

908 909
        auto len = cx.build.Load(lenptr);
        if (trailing_null) {
910
            len = cx.build.Sub(len, unit_sz);
911 912 913 914
        }

        auto r = res(cx, C_nil());

915 916
        auto cond_cx = new_scope_block_ctxt(cx, "sequence-iter cond");
        auto body_cx = new_scope_block_ctxt(cx, "sequence-iter body");
917 918
        auto next_cx = new_sub_block_ctxt(cx, "next");

919 920
        cx.build.Br(cond_cx.llbb);

921
        auto ix = cond_cx.build.Phi(T_int(), vec(C_int(0)), vec(cx.llbb));
922 923 924 925 926
        auto scaled_ix = cond_cx.build.Phi(T_int(),
                                           vec(C_int(0)), vec(cx.llbb));

        auto end_test = cond_cx.build.ICmp(lib.llvm.LLVMIntNE,
                                           scaled_ix, len);
927 928
        cond_cx.build.CondBr(end_test, body_cx.llbb, next_cx.llbb);

929
        auto elt = body_cx.build.GEP(p0, vec(C_int(0), ix));
930 931 932
        auto body_res = f(body_cx,
                          load_non_structural(body_cx, elt, elt_ty),
                          elt_ty);
933
        auto next_ix = body_res.bcx.build.Add(ix, C_int(1));
934 935
        auto next_scaled_ix = body_res.bcx.build.Add(scaled_ix, unit_sz);

936 937 938
        cond_cx.build.AddIncomingToPhi(ix, vec(next_ix),
                                       vec(body_res.bcx.llbb));

939 940 941
        cond_cx.build.AddIncomingToPhi(scaled_ix, vec(next_scaled_ix),
                                       vec(body_res.bcx.llbb));

942 943 944 945 946 947 948 949 950 951
        body_res.bcx.build.Br(cond_cx.llbb);
        ret res(next_cx, C_nil());
    }

    alt (ty.struct) {
        case (typeck.ty_vec(?et)) {
            ret iter_sequence_body(cx, v, et, f, false);
        }
        case (typeck.ty_str) {
            auto et = typeck.plain_ty(typeck.ty_machine(common.ty_u8));
952
            ret iter_sequence_body(cx, v, et, f, true);
953
        }
954
        case (_) { fail; }
955
    }
956 957 958 959 960 961 962
    cx.fcx.ccx.sess.bug("bad type in trans.iter_sequence");
    fail;
}

fn incr_all_refcnts(@block_ctxt cx,
                    ValueRef v,
                    @typeck.ty t) -> result {
963 964
    cx.build.FastCall(get_ty_info(cx.fcx.ccx, t).take_glue,
                      vec(cx.fcx.lltaskptr, v));
965
    ret res(cx, C_nil());
966 967
}

968 969 970
fn drop_slot(@block_ctxt cx,
             ValueRef slot,
             @typeck.ty t) -> result {
971
    be drop_ty(cx, load_non_structural(cx, slot, t), t);
972 973
}

974 975 976
fn drop_ty(@block_ctxt cx,
           ValueRef v,
           @typeck.ty t) -> result {
977 978 979
    cx.build.FastCall(get_ty_info(cx.fcx.ccx, t).drop_glue,
                      vec(cx.fcx.lltaskptr, v));
    ret res(cx, C_nil());
980 981 982 983 984 985
}

fn build_memcpy(@block_ctxt cx,
                ValueRef dst,
                ValueRef src,
                TypeRef llty) -> result {
986 987 988
    // FIXME: switch to the 64-bit variant when on such a platform.
    check (cx.fcx.ccx.intrinsics.contains_key("llvm.memcpy.p0i8.p0i8.i32"));
    auto memcpy = cx.fcx.ccx.intrinsics.get("llvm.memcpy.p0i8.p0i8.i32");
989 990
    auto src_ptr = cx.build.PointerCast(src, T_ptr(T_i8()));
    auto dst_ptr = cx.build.PointerCast(dst, T_ptr(T_i8()));
G
Graydon Hoare 已提交
991 992
    auto size = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llty),
                                 T_i32());
993 994 995 996 997 998
    auto align = cx.build.IntCast(C_int(1), T_i32());

    // FIXME: align seems like it should be
    //   lib.llvm.llvm.LLVMAlignOf(llty);
    // but this makes it upset because it's not a constant.

P
Patrick Walton 已提交
999
    log "building memcpy";
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
    auto volatile = C_integral(0, T_i1());
    ret res(cx, cx.build.Call(memcpy,
                              vec(dst_ptr, src_ptr,
                                  size, align, volatile)));
}

fn copy_ty(@block_ctxt cx,
           bool is_init,
           ValueRef dst,
           ValueRef src,
           @typeck.ty t) -> result {
    if (typeck.type_is_scalar(t)) {
        ret res(cx, cx.build.Store(src, dst));

    } else if (typeck.type_is_nil(t)) {
        ret res(cx, C_nil());

    } else if (typeck.type_is_binding(t)) {
        cx.fcx.ccx.sess.unimpl("binding type in trans.copy_ty");

    } else if (typeck.type_is_boxed(t)) {
1021
        auto r = incr_all_refcnts(cx, src, t);
1022
        if (! is_init) {
1023
            r = drop_ty(r.bcx, r.bcx.build.Load(dst), t);
1024 1025 1026 1027 1028 1029 1030 1031
        }
        ret res(r.bcx, r.bcx.build.Store(src, dst));

    } else if (typeck.type_is_structural(t)) {
        auto r = incr_all_refcnts(cx, src, t);
        if (! is_init) {
            r = drop_ty(r.bcx, dst, t);
        }
1032 1033 1034 1035
        // In this one surprising case, we do a load/store on
        // structure types. This results in a memcpy. Usually
        // we talk about structures by pointers in this file.
        ret res(r.bcx, r.bcx.build.Store(r.bcx.build.Load(src), dst));
1036 1037 1038 1039
    }

    cx.fcx.ccx.sess.bug("unexpected type in trans.copy_ty: " +
                        typeck.ty_to_str(t));
1040 1041 1042
    fail;
}

1043
impure fn trans_lit(@block_ctxt cx, &ast.lit lit, &ast.ann ann) -> result {
1044
    alt (lit.node) {
1045
        case (ast.lit_int(?i)) {
1046
            ret res(cx, C_int(i));
1047 1048
        }
        case (ast.lit_uint(?u)) {
1049
            ret res(cx, C_int(u as int));
1050
        }
1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071
        case (ast.lit_mach_int(?tm, ?i)) {
            // FIXME: the entire handling of mach types falls apart
            // if target int width is larger than host, at the moment;
            // re-do the mach-int types using 'big' when that works.
            auto t = T_int();
            alt (tm) {
                case (common.ty_u8) { t =  T_i8(); }
                case (common.ty_u16) { t =  T_i16(); }
                case (common.ty_u32) { t =  T_i32(); }
                case (common.ty_u64) { t =  T_i64(); }

                case (common.ty_i8) { t =  T_i8(); }
                case (common.ty_i16) { t =  T_i16(); }
                case (common.ty_i32) { t =  T_i32(); }
                case (common.ty_i64) { t =  T_i64(); }
                case (_) {
                    cx.fcx.ccx.sess.bug("bad mach int literal type");
                }
            }
            ret res(cx, C_integral(i, t));
        }
1072
        case (ast.lit_char(?c)) {
1073
            ret res(cx, C_integral(c as int, T_char()));
1074 1075
        }
        case (ast.lit_bool(?b)) {
1076 1077 1078 1079
            ret res(cx, C_bool(b));
        }
        case (ast.lit_nil) {
            ret res(cx, C_nil());
1080 1081 1082
        }
        case (ast.lit_str(?s)) {
            auto len = (_str.byte_len(s) as int) + 1;
1083
            auto sub = trans_upcall(cx, "upcall_new_str",
1084
                                    vec(p2i(C_str(cx.fcx.ccx, s)),
1085 1086
                                        C_int(len)));
            sub.val = sub.bcx.build.IntToPtr(sub.val,
1087
                                             T_ptr(T_str()));
1088
            auto t = node_ann_type(cx.fcx.ccx, ann);
1089
            find_scope_cx(cx).cleanups +=
1090
                clean(bind drop_ty(_, sub.val, t));
1091
            ret sub;
1092 1093 1094 1095
        }
    }
}

1096 1097 1098 1099 1100 1101 1102 1103 1104 1105
fn target_type(@crate_ctxt cx, @typeck.ty t) -> @typeck.ty {
    alt (t.struct) {
        case (typeck.ty_int) {
            auto tm = typeck.ty_machine(cx.sess.get_targ_cfg().int_type);
            ret @rec(struct=tm with *t);
        }
        case (typeck.ty_uint) {
            auto tm = typeck.ty_machine(cx.sess.get_targ_cfg().uint_type);
            ret @rec(struct=tm with *t);
        }
1106
        case (_) { /* fall through */ }
1107 1108 1109 1110 1111
    }
    ret t;
}

fn node_ann_type(@crate_ctxt cx, &ast.ann a) -> @typeck.ty {
1112 1113
    alt (a) {
        case (ast.ann_none) {
1114
            cx.sess.bug("missing type annotation");
1115 1116
        }
        case (ast.ann_type(?t)) {
1117
            ret target_type(cx, t);
1118 1119 1120 1121
        }
    }
}

1122 1123 1124 1125
fn node_type(@crate_ctxt cx, &ast.ann a) -> TypeRef {
    ret type_of(cx, node_ann_type(cx, a));
}

1126
impure fn trans_unary(@block_ctxt cx, ast.unop op,
1127
                      @ast.expr e, &ast.ann a) -> result {
1128 1129 1130

    auto sub = trans_expr(cx, e);

1131 1132
    alt (op) {
        case (ast.bitnot) {
1133 1134
            sub.val = cx.build.Not(sub.val);
            ret sub;
1135 1136
        }
        case (ast.not) {
1137 1138
            sub.val = cx.build.Not(sub.val);
            ret sub;
1139 1140 1141
        }
        case (ast.neg) {
            // FIXME: switch by signedness.
1142 1143
            sub.val = cx.build.Neg(sub.val);
            ret sub;
1144
        }
1145
        case (ast.box) {
1146 1147 1148 1149 1150
            auto e_ty = typeck.expr_ty(e);
            auto e_val = sub.val;
            sub = trans_malloc(sub.bcx, node_ann_type(sub.bcx.fcx.ccx, a));
            auto box = sub.val;
            auto rc = sub.bcx.build.GEP(box,
1151 1152
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
1153 1154 1155 1156 1157 1158
            auto body = sub.bcx.build.GEP(box,
                                          vec(C_int(0),
                                              C_int(abi.box_rc_field_body)));
            sub.bcx.build.Store(C_int(1), rc);
            sub = copy_ty(sub.bcx, true, body, e_val, e_ty);
            ret res(sub.bcx, box);
1159
        }
1160 1161 1162 1163 1164 1165 1166 1167 1168 1169
        case (ast.deref) {
            sub.val = sub.bcx.build.GEP(sub.val,
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_body)));
            auto e_ty = node_ann_type(sub.bcx.fcx.ccx, a);
            if (typeck.type_is_scalar(e_ty) ||
                typeck.type_is_nil(e_ty)) {
                sub.val = sub.bcx.build.Load(sub.val);
            }
            ret sub;
1170
        }
1171 1172 1173 1174
    }
    fail;
}

1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
fn trans_eager_binop(@block_ctxt cx, ast.binop op,
                     ValueRef lhs, ValueRef rhs) -> ValueRef {

    alt (op) {
        case (ast.add) { ret cx.build.Add(lhs, rhs); }
        case (ast.sub) { ret cx.build.Sub(lhs, rhs); }

        // FIXME: switch by signedness.
        case (ast.mul) { ret cx.build.Mul(lhs, rhs); }
        case (ast.div) { ret cx.build.SDiv(lhs, rhs); }
        case (ast.rem) { ret cx.build.SRem(lhs, rhs); }

        case (ast.bitor) { ret cx.build.Or(lhs, rhs); }
        case (ast.bitand) { ret cx.build.And(lhs, rhs); }
        case (ast.bitxor) { ret cx.build.Xor(lhs, rhs); }
        case (ast.lsl) { ret cx.build.Shl(lhs, rhs); }
        case (ast.lsr) { ret cx.build.LShr(lhs, rhs); }
        case (ast.asr) { ret cx.build.AShr(lhs, rhs); }
        case (_) {
            auto cmp = lib.llvm.LLVMIntEQ;
            alt (op) {
                case (ast.eq) { cmp = lib.llvm.LLVMIntEQ; }
                case (ast.ne) { cmp = lib.llvm.LLVMIntNE; }

                // FIXME: switch by signedness.
                case (ast.lt) { cmp = lib.llvm.LLVMIntSLT; }
                case (ast.le) { cmp = lib.llvm.LLVMIntSLE; }
                case (ast.ge) { cmp = lib.llvm.LLVMIntSGE; }
                case (ast.gt) { cmp = lib.llvm.LLVMIntSGT; }
            }
            ret cx.build.ICmp(cmp, lhs, rhs);
        }
    }
    fail;
}

1211
impure fn trans_binary(@block_ctxt cx, ast.binop op,
1212
                       @ast.expr a, @ast.expr b) -> result {
1213

1214 1215 1216 1217 1218 1219 1220
    // First couple cases are lazy:

    alt (op) {
        case (ast.and) {
            // Lazy-eval and
            auto lhs_res = trans_expr(cx, a);

1221
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1222 1223
            auto rhs_res = trans_expr(rhs_cx, b);

1224
            auto lhs_false_cx = new_scope_block_ctxt(cx, "lhs false");
1225
            auto lhs_false_res = res(lhs_false_cx, C_bool(false));
1226 1227 1228

            lhs_res.bcx.build.CondBr(lhs_res.val,
                                     rhs_cx.llbb,
1229 1230 1231 1232
                                     lhs_false_cx.llbb);

            ret join_results(cx, T_bool(),
                             vec(lhs_false_res, rhs_res));
1233 1234 1235 1236 1237 1238
        }

        case (ast.or) {
            // Lazy-eval or
            auto lhs_res = trans_expr(cx, a);

1239
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1240 1241
            auto rhs_res = trans_expr(rhs_cx, b);

1242
            auto lhs_true_cx = new_scope_block_ctxt(cx, "lhs true");
1243
            auto lhs_true_res = res(lhs_true_cx, C_bool(true));
1244 1245

            lhs_res.bcx.build.CondBr(lhs_res.val,
1246
                                     lhs_true_cx.llbb,
1247
                                     rhs_cx.llbb);
1248 1249 1250

            ret join_results(cx, T_bool(),
                             vec(lhs_true_res, rhs_res));
1251
        }
1252 1253

        case (_) {
1254 1255 1256 1257 1258
            // Remaining cases are eager:
            auto lhs = trans_expr(cx, a);
            auto sub = trans_expr(lhs.bcx, b);
            ret res(sub.bcx, trans_eager_binop(sub.bcx, op,
                                               lhs.val, sub.val));
1259
        }
1260 1261 1262 1263
    }
    fail;
}

1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294
fn join_results(@block_ctxt parent_cx,
                TypeRef t,
                vec[result] ins)
    -> result {

    let vec[result] live = vec();
    let vec[ValueRef] vals = vec();
    let vec[BasicBlockRef] bbs = vec();

    for (result r in ins) {
        if (! is_terminated(r.bcx)) {
            live += r;
            vals += r.val;
            bbs += r.bcx.llbb;
        }
    }

    alt (_vec.len[result](live)) {
        case (0u) {
            // No incoming edges are live, so we're in dead-code-land.
            // Arbitrarily pick the first dead edge, since the caller
            // is just going to propagate it outward.
            check (_vec.len[result](ins) >= 1u);
            ret ins.(0);
        }

        case (1u) {
            // Only one incoming edge is live, so we just feed that block
            // onward.
            ret live.(0);
        }
1295 1296

        case (_) { /* fall through */ }
1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307
    }

    // We have >1 incoming edges. Make a join block and br+phi them into it.
    auto join_cx = new_sub_block_ctxt(parent_cx, "join");
    for (result r in live) {
        r.bcx.build.Br(join_cx.llbb);
    }
    auto phi = join_cx.build.Phi(t, vals, bbs);
    ret res(join_cx, phi);
}

1308
impure fn trans_if(@block_ctxt cx, @ast.expr cond,
1309
                   &ast.block thn, &option.t[ast.block] els) -> result {
1310 1311 1312

    auto cond_res = trans_expr(cx, cond);

1313
    auto then_cx = new_scope_block_ctxt(cx, "then");
1314 1315
    auto then_res = trans_block(then_cx, thn);

1316
    auto else_cx = new_scope_block_ctxt(cx, "else");
1317
    auto else_res = res(else_cx, C_nil());
1318 1319 1320

    alt (els) {
        case (some[ast.block](?eblk)) {
1321
            else_res = trans_block(else_cx, eblk);
1322
        }
1323
        case (_) { /* fall through */ }
1324 1325
    }

1326
    cond_res.bcx.build.CondBr(cond_res.val,
1327 1328
                              then_cx.llbb,
                              else_cx.llbb);
1329 1330 1331 1332

    // FIXME: use inferred type when available.
    ret join_results(cx, T_nil(),
                     vec(then_res, else_res));
1333 1334
}

1335
impure fn trans_while(@block_ctxt cx, @ast.expr cond,
1336 1337
                      &ast.block body) -> result {

1338 1339
    auto cond_cx = new_scope_block_ctxt(cx, "while cond");
    auto body_cx = new_scope_block_ctxt(cx, "while loop body");
1340
    auto next_cx = new_sub_block_ctxt(cx, "next");
1341 1342

    auto body_res = trans_block(body_cx, body);
1343 1344 1345 1346 1347 1348 1349 1350
    auto cond_res = trans_expr(cond_cx, cond);

    body_res.bcx.build.Br(cond_cx.llbb);
    cond_res.bcx.build.CondBr(cond_res.val,
                              body_cx.llbb,
                              next_cx.llbb);

    cx.build.Br(cond_cx.llbb);
1351 1352 1353 1354
    ret res(next_cx, C_nil());
}

impure fn trans_do_while(@block_ctxt cx, &ast.block body,
1355
                         @ast.expr cond) -> result {
1356

1357
    auto body_cx = new_scope_block_ctxt(cx, "do-while loop body");
1358
    auto next_cx = new_sub_block_ctxt(cx, "next");
1359 1360

    auto body_res = trans_block(body_cx, body);
1361 1362 1363 1364 1365 1366
    auto cond_res = trans_expr(body_res.bcx, cond);

    cond_res.bcx.build.CondBr(cond_res.val,
                              body_cx.llbb,
                              next_cx.llbb);
    cx.build.Br(body_cx.llbb);
1367 1368 1369
    ret res(next_cx, body_res.val);
}

P
Patrick Walton 已提交
1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471
// Pattern matching translation

// Returns a pointer to the union part of the LLVM representation of a tag
// type, cast to the appropriate type.
fn get_pat_union_ptr(@block_ctxt cx, vec[@ast.pat] subpats, ValueRef llval)
        -> ValueRef {
    auto llblobptr = cx.build.GEP(llval, vec(C_int(0), C_int(1)));

    // Generate the union type.
    let vec[TypeRef] llsubpattys = vec();
    for (@ast.pat subpat in subpats) {
        llsubpattys += vec(type_of(cx.fcx.ccx, pat_ty(subpat)));
    }

    // Recursively check subpatterns.
    auto llunionty = T_struct(llsubpattys);
    ret cx.build.TruncOrBitCast(llblobptr, T_ptr(llunionty));
}

impure fn trans_pat_match(@block_ctxt cx, @ast.pat pat, ValueRef llval,
                          @block_ctxt next_cx) -> result {
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
        case (ast.pat_bind(_, _, _)) { ret res(cx, llval); }
        case (ast.pat_tag(?id, ?subpats, ?vdef_opt, ?ann)) {
            auto lltagptr = cx.build.GEP(llval, vec(C_int(0), C_int(0)));
            auto lltag = cx.build.Load(lltagptr);
            
            auto vdef = option.get[ast.variant_def](vdef_opt);
            auto variant_id = vdef._1;
            auto tinfo = cx.fcx.ccx.tags.get(vdef._0);
            auto variant_tag = 0;
            auto i = 0;
            for (tup(ast.def_id,arity) vinfo in tinfo.variants) {
                auto this_variant_id = vinfo._0;
                if (variant_id._0 == this_variant_id._0 &&
                        variant_id._1 == this_variant_id._1) {
                    variant_tag = i;
                }
                i += 1;
            }

            auto matched_cx = new_sub_block_ctxt(cx, "matched_cx");

            auto lleq = cx.build.ICmp(lib.llvm.LLVMIntEQ, lltag,
                                      C_int(variant_tag));
            cx.build.CondBr(lleq, matched_cx.llbb, next_cx.llbb);

            if (_vec.len[@ast.pat](subpats) > 0u) {
                auto llunionptr = get_pat_union_ptr(matched_cx, subpats,
                                                    llval);
                auto i = 0;
                for (@ast.pat subpat in subpats) {
                    auto llsubvalptr = matched_cx.build.GEP(llunionptr,
                                                            vec(C_int(0),
                                                                C_int(i)));
                    auto llsubval = load_non_structural(matched_cx,
                                                        llsubvalptr,
                                                        pat_ty(subpat));
                    auto subpat_res = trans_pat_match(matched_cx, subpat,
                                                      llsubval, next_cx);
                    matched_cx = subpat_res.bcx;
                }
            }

            ret res(matched_cx, llval);
        }
    }

    fail;
}

impure fn trans_pat_binding(@block_ctxt cx, @ast.pat pat, ValueRef llval)
        -> result {
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
        case (ast.pat_bind(?id, ?def_id, ?ann)) {
            auto ty = node_ann_type(cx.fcx.ccx, ann);
            auto llty = type_of(cx.fcx.ccx, ty);

            auto dst = cx.build.Alloca(llty);
            llvm.LLVMSetValueName(dst, _str.buf(id));
            cx.fcx.lllocals.insert(def_id, dst);
            cx.cleanups += clean(bind drop_slot(_, dst, ty));

            ret copy_ty(cx, true, dst, llval, ty);
        }
        case (ast.pat_tag(_, ?subpats, _, _)) {
            if (_vec.len[@ast.pat](subpats) == 0u) { ret res(cx, llval); }

            auto llunionptr = get_pat_union_ptr(cx, subpats, llval);

            auto this_cx = cx;
            auto i = 0;
            for (@ast.pat subpat in subpats) {
                auto llsubvalptr = this_cx.build.GEP(llunionptr,
                                                     vec(C_int(0), C_int(i)));
                auto llsubval = load_non_structural(this_cx, llsubvalptr,
                                                    pat_ty(subpat));
                auto subpat_res = trans_pat_binding(this_cx, subpat,
                                                    llsubval);
                this_cx = subpat_res.bcx;
1472
                i += 1;
P
Patrick Walton 已提交
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
            }

            ret res(this_cx, llval);
        }
    }
}

impure fn trans_alt(@block_ctxt cx, @ast.expr expr, vec[ast.arm] arms)
        -> result {
    auto expr_res = trans_expr(cx, expr);

    auto last_cx = new_sub_block_ctxt(expr_res.bcx, "last");

    auto this_cx = expr_res.bcx;
    for (ast.arm arm in arms) {
        auto next_cx = new_sub_block_ctxt(expr_res.bcx, "next");
        auto match_res = trans_pat_match(this_cx, arm.pat, expr_res.val,
                                         next_cx);

        auto binding_cx = new_scope_block_ctxt(match_res.bcx, "binding");
        match_res.bcx.build.Br(binding_cx.llbb);

        auto binding_res = trans_pat_binding(binding_cx, arm.pat,
                                             expr_res.val);

        auto block_res = trans_block(binding_res.bcx, arm.block);
        if (!is_terminated(block_res.bcx)) {
            block_res.bcx.build.Br(last_cx.llbb);
        }

        this_cx = next_cx;
    }

    // FIXME: This is executed when none of the patterns match; it should fail
    // instead!
    this_cx.build.Br(last_cx.llbb);

    // FIXME: This is very wrong; we should phi together all the arm blocks,
    // since this is an expression.
    ret res(last_cx, C_nil());
}

1515 1516 1517
// The additional bool returned indicates whether it's mem (that is
// represented as an alloca or heap, hence needs a 'load' to be used as an
// immediate).
1518

1519 1520 1521 1522 1523 1524 1525 1526
fn trans_name(@block_ctxt cx, &ast.name n, &option.t[ast.def] dopt)
    -> tup(result, bool) {
    alt (dopt) {
        case (some[ast.def](?def)) {
            alt (def) {
                case (ast.def_arg(?did)) {
                    check (cx.fcx.llargs.contains_key(did));
                    ret tup(res(cx, cx.fcx.llargs.get(did)),
1527
                            true);
1528 1529 1530 1531 1532
                }
                case (ast.def_local(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
                    ret tup(res(cx, cx.fcx.lllocals.get(did)),
                            true);
G
Graydon Hoare 已提交
1533
                }
P
Patrick Walton 已提交
1534 1535 1536 1537
                case (ast.def_binding(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
                    ret tup(res(cx, cx.fcx.lllocals.get(did)), true);
                }
1538
                case (ast.def_fn(?did)) {
1539 1540
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
1541 1542
                            false);
                }
1543 1544 1545 1546 1547
                case (ast.def_obj(?did)) {
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
                            false);
                }
1548 1549
                case (ast.def_variant(?tid, ?vid)) {
                    check (cx.fcx.ccx.tags.contains_key(tid));
1550 1551 1552
                    check (cx.fcx.ccx.item_ids.contains_key(vid));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(vid)),
                            false);
1553
                }
1554 1555
                case (_) {
                    cx.fcx.ccx.sess.unimpl("def variant in trans");
G
Graydon Hoare 已提交
1556 1557 1558
                }
            }
        }
1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576
        case (none[ast.def]) {
            cx.fcx.ccx.sess.err("unresolved expr_name in trans");
        }
    }
    fail;
}

fn trans_field(@block_ctxt cx, &ast.span sp, @ast.expr base,
               &ast.ident field, &ast.ann ann) -> tup(result, bool) {
    auto lv = trans_lval(cx, base);
    auto r = lv._0;
    auto ty = typeck.expr_ty(base);
    alt (ty.struct) {
        case (typeck.ty_tup(?fields)) {
            let uint ix = typeck.field_num(cx.fcx.ccx.sess, sp, field);
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
            ret tup(res(r.bcx, v), lv._1);
        }
1577 1578 1579 1580 1581 1582
        case (typeck.ty_rec(?fields)) {
            let uint ix = typeck.field_idx(cx.fcx.ccx.sess, sp,
                                           field, fields);
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
            ret tup(res(r.bcx, v), lv._1);
        }
1583
        case (_) { cx.fcx.ccx.sess.unimpl("field variant in trans_field"); }
1584 1585 1586 1587
    }
    fail;
}

1588 1589 1590
fn trans_index(@block_ctxt cx, &ast.span sp, @ast.expr base,
               @ast.expr idx, &ast.ann ann) -> tup(result, bool) {

G
Graydon Hoare 已提交
1591 1592 1593
    auto lv = trans_expr(cx, base);
    auto ix = trans_expr(lv.bcx, idx);
    auto v = lv.val;
1594 1595 1596 1597 1598 1599 1600 1601

    auto llunit_ty = node_type(cx.fcx.ccx, ann);
    auto unit_sz = ix.bcx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llunit_ty),
                                      T_int());
    auto scaled_ix = ix.bcx.build.Mul(ix.val, unit_sz);

    auto lim = ix.bcx.build.GEP(v, vec(C_int(0), C_int(abi.vec_elt_fill)));
    lim = ix.bcx.build.Load(lim);
1602

1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
    auto bounds_check = ix.bcx.build.ICmp(lib.llvm.LLVMIntULT,
                                          scaled_ix, lim);

    auto fail_cx = new_sub_block_ctxt(ix.bcx, "fail");
    auto next_cx = new_sub_block_ctxt(ix.bcx, "next");
    ix.bcx.build.CondBr(bounds_check, next_cx.llbb, fail_cx.llbb);

    // fail: bad bounds check.
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "out-of-bounds access"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, sp.filename));
    auto V_line = sp.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);
    fail_res.bcx.build.Br(next_cx.llbb);

    auto body = next_cx.build.GEP(v, vec(C_int(0), C_int(abi.vec_elt_data)));
    auto elt = next_cx.build.GEP(body, vec(C_int(0), ix.val));
G
Graydon Hoare 已提交
1620
    ret tup(res(next_cx, elt), true);
1621 1622
}

1623 1624 1625 1626
// The additional bool returned indicates whether it's mem (that is
// represented as an alloca or heap, hence needs a 'load' to be used as an
// immediate).

1627 1628 1629 1630 1631 1632 1633 1634
fn trans_lval(@block_ctxt cx, @ast.expr e) -> tup(result, bool) {
    alt (e.node) {
        case (ast.expr_name(?n, ?dopt, _)) {
            ret trans_name(cx, n, dopt);
        }
        case (ast.expr_field(?base, ?ident, ?ann)) {
            ret trans_field(cx, e.span, base, ident, ann);
        }
1635 1636 1637
        case (ast.expr_index(?base, ?idx, ?ann)) {
            ret trans_index(cx, e.span, base, idx, ann);
        }
1638
        case (_) { cx.fcx.ccx.sess.unimpl("expr variant in trans_lval"); }
G
Graydon Hoare 已提交
1639 1640 1641 1642
    }
    fail;
}

1643
impure fn trans_cast(@block_ctxt cx, @ast.expr e, &ast.ann ann) -> result {
1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673
    auto e_res = trans_expr(cx, e);
    auto llsrctype = val_ty(e_res.val);
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto lldsttype = type_of(cx.fcx.ccx, t);
    if (!typeck.type_is_fp(t)) {
        if (llvm.LLVMGetIntTypeWidth(lldsttype) >
            llvm.LLVMGetIntTypeWidth(llsrctype)) {
            if (typeck.type_is_signed(t)) {
                // Widening signed cast.
                e_res.val =
                    e_res.bcx.build.SExtOrBitCast(e_res.val,
                                                  lldsttype);
            } else {
                // Widening unsigned cast.
                e_res.val =
                    e_res.bcx.build.ZExtOrBitCast(e_res.val,
                                                  lldsttype);
            }
        } else {
            // Narrowing cast.
            e_res.val =
                e_res.bcx.build.TruncOrBitCast(e_res.val,
                                               lldsttype);
        }
    } else {
        cx.fcx.ccx.sess.unimpl("fp cast");
    }
    ret e_res;
}

1674

1675
impure fn trans_args(@block_ctxt cx, &vec[@ast.expr] es, @typeck.ty fn_ty)
1676 1677 1678 1679
    -> tup(@block_ctxt, vec[ValueRef]) {
    let vec[ValueRef] vs = vec(cx.fcx.lltaskptr);
    let @block_ctxt bcx = cx;

1680 1681 1682 1683 1684 1685 1686
    let vec[typeck.arg] args = vec();   // FIXME: typestate bug
    alt (fn_ty.struct) {
        case (typeck.ty_fn(?a, _)) { args = a; }
        case (_) { fail; }
    }

    auto i = 0u;
1687
    for (@ast.expr e in es) {
1688 1689 1690
        auto mode = args.(i).mode;

        auto re;
1691
        if (typeck.type_is_structural(typeck.expr_ty(e))) {
1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718
            re = trans_expr(bcx, e);
            if (mode == ast.val) {
                // Until here we've been treating structures by pointer;
                // we are now passing it as an arg, so need to load it.
                re.val = re.bcx.build.Load(re.val);
            }
        } else {
            if (mode == ast.alias) {
                let tup(result, bool /* is a pointer? */) pair;
                if (typeck.is_lval(e)) {
                    pair = trans_lval(bcx, e);
                } else {
                    pair = tup(trans_expr(bcx, e), false);
                }

                if (!pair._1) {
                    // Have to synthesize a pointer here...
                    auto llty = val_ty(pair._0.val);
                    auto llptr = pair._0.bcx.build.Alloca(llty);
                    pair._0.bcx.build.Store(pair._0.val, llptr);
                    re = res(pair._0.bcx, llptr);
                } else {
                    re = pair._0;
                }
            } else {
                re = trans_expr(bcx, e);
            }
1719
        }
1720 1721 1722 1723 1724

        vs += re.val;
        bcx = re.bcx;

        i += 1u;
1725 1726 1727 1728 1729
    }

    ret tup(bcx, vs);
}

1730
impure fn trans_call(@block_ctxt cx, @ast.expr f,
G
Graydon Hoare 已提交
1731
                     vec[@ast.expr] args, &ast.ann ann) -> result {
1732 1733
    auto f_res = trans_lval(cx, f);
    check (! f_res._1);
1734
    auto fn_ty = typeck.expr_ty(f);
G
Graydon Hoare 已提交
1735
    auto ret_ty = typeck.ann_to_type(ann);
1736
    auto args_res = trans_args(f_res._0.bcx, args, fn_ty);
1737 1738 1739 1740 1741 1742 1743 1744
    
    auto real_retval = args_res._0.build.FastCall(f_res._0.val, args_res._1);
    auto retval;
    if (typeck.type_is_nil(ret_ty)) {
        retval = C_nil();
    } else {
        retval = real_retval;
    }
G
Graydon Hoare 已提交
1745

G
Graydon Hoare 已提交
1746 1747 1748 1749 1750 1751 1752 1753 1754
    // Structured returns come back as first-class values. This is nice for
    // LLVM but wrong for us; we treat structured values by pointer in
    // most of our code here. So spill it to an alloca.
    if (typeck.type_is_structural(ret_ty)) {
        auto local = args_res._0.build.Alloca(type_of(cx.fcx.ccx, ret_ty));
        args_res._0.build.Store(retval, local);
        retval = local;
    }

G
Graydon Hoare 已提交
1755 1756 1757 1758 1759 1760
    // Retval doesn't correspond to anything really tangible in the frame, but
    // it's a ref all the same, so we put a note here to drop it when we're
    // done in this scope.
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, retval, ret_ty));

    ret res(args_res._0, retval);
1761 1762
}

1763
impure fn trans_tup(@block_ctxt cx, vec[ast.elt] elts,
G
Graydon Hoare 已提交
1764
                    &ast.ann ann) -> result {
1765 1766 1767 1768
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, ty);
    auto tup_val = cx.build.Alloca(llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, tup_val, ty));
G
Graydon Hoare 已提交
1769 1770
    let int i = 0;
    auto r = res(cx, C_nil());
1771 1772 1773
    for (ast.elt e in elts) {
        auto t = typeck.expr_ty(e.expr);
        auto src_res = trans_expr(r.bcx, e.expr);
G
Graydon Hoare 已提交
1774 1775 1776 1777 1778 1779 1780
        auto dst_elt = r.bcx.build.GEP(tup_val, vec(C_int(0), C_int(i)));
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
    ret res(r.bcx, tup_val);
}

G
Graydon Hoare 已提交
1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
impure fn trans_vec(@block_ctxt cx, vec[@ast.expr] args,
                    &ast.ann ann) -> result {
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto unit_ty = ty;
    alt (ty.struct) {
        case (typeck.ty_vec(?t)) {
            unit_ty = t;
        }
        case (_) {
            cx.fcx.ccx.sess.bug("non-vec type in trans_vec");
        }
    }

    auto llunit_ty = type_of(cx.fcx.ccx, unit_ty);
    auto unit_sz = llvm.LLVMConstIntCast(llvm.LLVMSizeOf(llunit_ty),
                                         T_int(), False);
    auto data_sz = llvm.LLVMConstMul(C_int(_vec.len[@ast.expr](args) as int),
                                     unit_sz);

    // FIXME: pass tydesc properly.
    auto sub = trans_upcall(cx, "upcall_new_vec", vec(data_sz, C_int(0)));

    auto llty = type_of(cx.fcx.ccx, ty);
    auto vec_val = sub.bcx.build.IntToPtr(sub.val, llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, vec_val, ty));

    auto body = sub.bcx.build.GEP(vec_val, vec(C_int(0),
                                               C_int(abi.vec_elt_data)));
    let int i = 0;
    for (@ast.expr e in args) {
        auto src_res = trans_expr(sub.bcx, e);
        auto dst_elt = sub.bcx.build.GEP(body, vec(C_int(0), C_int(i)));
        sub = copy_ty(src_res.bcx, true, dst_elt, src_res.val, unit_ty);
        i += 1;
    }
1816 1817 1818 1819
    auto fill = sub.bcx.build.GEP(vec_val,
                                  vec(C_int(0), C_int(abi.vec_elt_fill)));
    sub.bcx.build.Store(data_sz, fill);

G
Graydon Hoare 已提交
1820 1821 1822
    ret res(sub.bcx, vec_val);
}

1823
impure fn trans_rec(@block_ctxt cx, vec[ast.field] fields,
1824
                    &ast.ann ann) -> result {
1825 1826 1827 1828
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, ty);
    auto rec_val = cx.build.Alloca(llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, rec_val, ty));
1829 1830
    let int i = 0;
    auto r = res(cx, C_nil());
1831 1832 1833 1834
    for (ast.field f in fields) {
        auto t = typeck.expr_ty(f.expr);
        auto src_res = trans_expr(r.bcx, f.expr);
        auto dst_elt = r.bcx.build.GEP(rec_val, vec(C_int(0), C_int(i)));
1835 1836 1837 1838
        // FIXME: calculate copy init-ness in typestate.
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
1839
    ret res(r.bcx, rec_val);
1840 1841
}

G
Graydon Hoare 已提交
1842

G
Graydon Hoare 已提交
1843

1844
impure fn trans_expr(@block_ctxt cx, @ast.expr e) -> result {
1845
    alt (e.node) {
1846 1847
        case (ast.expr_lit(?lit, ?ann)) {
            ret trans_lit(cx, *lit, ann);
1848 1849
        }

1850
        case (ast.expr_unary(?op, ?x, ?ann)) {
1851
            ret trans_unary(cx, op, x, ann);
1852 1853
        }

P
Patrick Walton 已提交
1854
        case (ast.expr_binary(?op, ?x, ?y, _)) {
1855
            ret trans_binary(cx, op, x, y);
1856
        }
1857

P
Patrick Walton 已提交
1858
        case (ast.expr_if(?cond, ?thn, ?els, _)) {
1859
            ret trans_if(cx, cond, thn, els);
1860 1861
        }

1862
        case (ast.expr_while(?cond, ?body, _)) {
1863
            ret trans_while(cx, cond, body);
1864 1865
        }

1866
        case (ast.expr_do_while(?body, ?cond, _)) {
1867
            ret trans_do_while(cx, body, cond);
1868 1869
        }

P
Patrick Walton 已提交
1870 1871 1872 1873
        case (ast.expr_alt(?expr, ?arms, _)) {
            ret trans_alt(cx, expr, arms);
        }

P
Patrick Walton 已提交
1874
        case (ast.expr_block(?blk, _)) {
1875
            auto sub_cx = new_scope_block_ctxt(cx, "block-expr body");
1876
            auto next_cx = new_sub_block_ctxt(cx, "next");
1877 1878 1879 1880 1881 1882 1883
            auto sub = trans_block(sub_cx, blk);

            cx.build.Br(sub_cx.llbb);
            sub.bcx.build.Br(next_cx.llbb);

            ret res(next_cx, sub.val);
        }
1884

1885
        case (ast.expr_assign(?dst, ?src, ?ann)) {
1886
            auto lhs_res = trans_lval(cx, dst);
1887
            check (lhs_res._1);
1888
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
1889
            auto t = node_ann_type(cx.fcx.ccx, ann);
G
Graydon Hoare 已提交
1890
            // FIXME: calculate copy init-ness in typestate.
1891
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, rhs_res.val, t);
1892
        }
G
Graydon Hoare 已提交
1893

1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905
        case (ast.expr_assign_op(?op, ?dst, ?src, ?ann)) {
            auto t = node_ann_type(cx.fcx.ccx, ann);
            auto lhs_res = trans_lval(cx, dst);
            check (lhs_res._1);
            auto lhs_val = load_non_structural(lhs_res._0.bcx,
                                               lhs_res._0.val, t);
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
            auto v = trans_eager_binop(rhs_res.bcx, op, lhs_val, rhs_res.val);
            // FIXME: calculate copy init-ness in typestate.
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, v, t);
        }

G
Graydon Hoare 已提交
1906 1907
        case (ast.expr_call(?f, ?args, ?ann)) {
            ret trans_call(cx, f, args, ann);
1908 1909
        }

1910
        case (ast.expr_cast(?e, _, ?ann)) {
1911
            ret trans_cast(cx, e, ann);
1912
        }
G
Graydon Hoare 已提交
1913

G
Graydon Hoare 已提交
1914 1915 1916 1917
        case (ast.expr_vec(?args, ?ann)) {
            ret trans_vec(cx, args, ann);
        }

G
Graydon Hoare 已提交
1918 1919 1920
        case (ast.expr_tup(?args, ?ann)) {
            ret trans_tup(cx, args, ann);
        }
G
Graydon Hoare 已提交
1921

1922 1923 1924 1925
        case (ast.expr_rec(?args, ?ann)) {
            ret trans_rec(cx, args, ann);
        }

1926 1927
        // lval cases fall through to trans_lval and then
        // possibly load the result (if it's non-structural).
G
Graydon Hoare 已提交
1928

1929 1930 1931
        case (_) {
            auto t = typeck.expr_ty(e);
            auto sub = trans_lval(cx, e);
1932 1933
            ret res(sub._0.bcx,
                    load_non_structural(sub._0.bcx, sub._0.val, t));
1934
        }
1935
    }
1936
    cx.fcx.ccx.sess.unimpl("expr variant in trans_expr");
1937 1938 1939
    fail;
}

1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953
// We pass structural values around the compiler "by pointer" and
// non-structural values "by value". This function selects whether
// to load a pointer or pass it.

fn load_non_structural(@block_ctxt cx,
                       ValueRef v,
                       @typeck.ty t) -> ValueRef {
    if (typeck.type_is_structural(t)) {
        ret v;
    } else {
        ret cx.build.Load(v);
    }
}

1954
impure fn trans_log(@block_ctxt cx, @ast.expr e) -> result {
1955

1956 1957 1958 1959 1960 1961 1962 1963
    auto sub = trans_expr(cx, e);
    auto e_ty = typeck.expr_ty(e);
    alt (e_ty.struct) {
        case (typeck.ty_str) {
            auto v = sub.bcx.build.PtrToInt(sub.val, T_int());
            ret trans_upcall(sub.bcx,
                             "upcall_log_str",
                             vec(v));
1964 1965
        }
        case (_) {
1966 1967 1968
            ret trans_upcall(sub.bcx,
                             "upcall_log_int",
                             vec(sub.val));
1969 1970
        }
    }
1971
    fail;
1972 1973
}

1974
impure fn trans_check_expr(@block_ctxt cx, @ast.expr e) -> result {
1975 1976 1977
    auto cond_res = trans_expr(cx, e);

    // FIXME: need pretty-printer.
1978 1979
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "<expr>"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, e.span.filename));
1980 1981 1982
    auto V_line = e.span.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));

1983
    auto fail_cx = new_sub_block_ctxt(cx, "fail");
1984 1985
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);

1986
    auto next_cx = new_sub_block_ctxt(cx, "next");
1987 1988 1989 1990 1991 1992 1993
    fail_res.bcx.build.Br(next_cx.llbb);
    cond_res.bcx.build.CondBr(cond_res.val,
                              next_cx.llbb,
                              fail_cx.llbb);
    ret res(next_cx, C_nil());
}

1994
impure fn trans_ret(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
1995 1996 1997
    auto r = res(cx, C_nil());
    alt (e) {
        case (some[@ast.expr](?x)) {
G
Graydon Hoare 已提交
1998
            auto t = typeck.expr_ty(x);
1999
            r = trans_expr(cx, x);
2000 2001 2002 2003 2004

            // A return is an implicit copy into a newborn anonymous
            // 'return value' in the caller frame.
            r.bcx = incr_all_refcnts(r.bcx, r.val, t).bcx;

G
Graydon Hoare 已提交
2005 2006 2007 2008 2009 2010 2011
            if (typeck.type_is_structural(t)) {
                // We usually treat structurals by-pointer; in particular,
                // trans_expr will have given us a structure pointer. But in
                // this case we're about to return. LLVM wants a first-class
                // value here (which makes sense; the frame is going away!)
                r.val = r.bcx.build.Load(r.val);
            }
2012
        }
2013
        case (_) { /* fall through */  }
2014 2015
    }

2016 2017
    // Run all cleanups and back out.
    let bool more_cleanups = true;
2018
    auto cleanup_cx = cx;
2019
    while (more_cleanups) {
2020 2021
        r.bcx = trans_block_cleanups(r.bcx, cleanup_cx);
        alt (cleanup_cx.parent) {
2022
            case (parent_some(?b)) {
2023
                cleanup_cx = b;
2024 2025 2026 2027 2028 2029 2030
            }
            case (parent_none) {
                more_cleanups = false;
            }
        }
    }

2031
    alt (e) {
2032 2033 2034 2035 2036 2037 2038
        case (some[@ast.expr](?ex)) {
            if (typeck.type_is_nil(typeck.expr_ty(ex))) {
                r.bcx.build.RetVoid();
                r.val = C_nil();
            } else {
                r.val = r.bcx.build.Ret(r.val);
            }
2039 2040
            ret r;
        }
2041
        case (_) { /* fall through */  }
2042 2043 2044 2045
    }

    // FIXME: until LLVM has a unit type, we are moving around
    // C_nil values rather than their void type.
2046 2047
    r.bcx.build.RetVoid();
    r.val = C_nil();
2048 2049 2050
    ret r;
}

2051
impure fn trans_stmt(@block_ctxt cx, &ast.stmt s) -> result {
2052
    auto sub = res(cx, C_nil());
2053
    alt (s.node) {
2054
        case (ast.stmt_log(?a)) {
2055
            sub.bcx = trans_log(cx, a).bcx;
2056 2057
        }

2058
        case (ast.stmt_check_expr(?a)) {
2059
            sub.bcx = trans_check_expr(cx, a).bcx;
2060 2061
        }

2062 2063 2064 2065
        case (ast.stmt_ret(?e)) {
            sub.bcx = trans_ret(cx, e).bcx;
        }

2066
        case (ast.stmt_expr(?e)) {
2067
            sub.bcx = trans_expr(cx, e).bcx;
2068
        }
2069

2070 2071 2072
        case (ast.stmt_decl(?d)) {
            alt (d.node) {
                case (ast.decl_local(?local)) {
2073 2074 2075 2076 2077

                    // Make a note to drop this slot on the way out.
                    check (cx.fcx.lllocals.contains_key(local.id));
                    auto llptr = cx.fcx.lllocals.get(local.id);
                    auto ty = node_ann_type(cx.fcx.ccx, local.ann);
2078
                    find_scope_cx(cx).cleanups +=
2079 2080
                        clean(bind drop_slot(_, llptr, ty));

2081 2082
                    alt (local.init) {
                        case (some[@ast.expr](?e)) {
2083
                            sub = trans_expr(cx, e);
2084
                            sub = copy_ty(sub.bcx, true, llptr, sub.val, ty);
2085
                        }
2086 2087 2088 2089 2090
                        case (_) {
                            auto llty = type_of(cx.fcx.ccx, ty);
                            auto null = lib.llvm.llvm.LLVMConstNull(llty);
                            sub = res(cx, cx.build.Store(null, llptr));
                        }
2091 2092 2093 2094
                    }
                }
            }
        }
2095
        case (_) {
2096
            cx.fcx.ccx.sess.unimpl("stmt variant");
2097 2098
        }
    }
2099
    ret sub;
2100 2101
}

2102
fn new_builder(BasicBlockRef llbb, str name) -> builder {
2103 2104 2105 2106 2107
    let BuilderRef llbuild = llvm.LLVMCreateBuilder();
    llvm.LLVMPositionBuilderAtEnd(llbuild, llbb);
    ret builder(llbuild);
}

2108 2109
// You probably don't want to use this one. See the
// next three functions instead.
2110
fn new_block_ctxt(@fn_ctxt cx, block_parent parent,
2111
                  bool is_scope,
2112
                  str name) -> @block_ctxt {
2113
    let vec[cleanup] cleanups = vec();
2114
    let BasicBlockRef llbb =
2115
        llvm.LLVMAppendBasicBlock(cx.llfn,
2116
                                  _str.buf(cx.ccx.names.next(name)));
2117

2118
    ret @rec(llbb=llbb,
2119 2120
             build=new_builder(llbb, name),
             parent=parent,
2121
             is_scope=is_scope,
2122
             mutable cleanups=cleanups,
2123 2124 2125
             fcx=cx);
}

2126 2127
// Use this when you're at the top block of a function or the like.
fn new_top_block_ctxt(@fn_ctxt fcx) -> @block_ctxt {
2128 2129
    ret new_block_ctxt(fcx, parent_none, true, "function top level");
}
2130

2131 2132 2133
// Use this when you're at a curly-brace or similar lexical scope.
fn new_scope_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), true, n);
2134 2135
}

2136
// Use this when you're making a general CFG BB within a scope.
2137
fn new_sub_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
2138
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), false, n);
2139
}
2140

2141

2142 2143
fn trans_block_cleanups(@block_ctxt cx,
                        @block_ctxt cleanup_cx) -> @block_ctxt {
2144
    auto bcx = cx;
2145 2146 2147 2148 2149

    if (!cleanup_cx.is_scope) {
        check (_vec.len[cleanup](cleanup_cx.cleanups) == 0u);
    }

2150
    for (cleanup c in cleanup_cx.cleanups) {
2151
        alt (c) {
2152
            case (clean(?cfn)) {
2153
                bcx = cfn(bcx).bcx;
2154 2155 2156
            }
        }
    }
2157 2158 2159
    ret bcx;
}

2160 2161 2162 2163 2164 2165 2166 2167 2168 2169
iter block_locals(&ast.block b) -> @ast.local {
    // FIXME: putting from inside an iter block doesn't work, so we can't
    // use the index here.
    for (@ast.stmt s in b.node.stmts) {
        alt (s.node) {
            case (ast.stmt_decl(?d)) {
                alt (d.node) {
                    case (ast.decl_local(?local)) {
                        put local;
                    }
2170
                    case (_) { /* fall through */ }
2171 2172
                }
            }
2173
            case (_) { /* fall through */ }
2174 2175 2176 2177
        }
    }
}

2178
impure fn trans_block(@block_ctxt cx, &ast.block b) -> result {
2179 2180
    auto bcx = cx;

2181
    for each (@ast.local local in block_locals(b)) {
2182
        auto ty = node_type(cx.fcx.ccx, local.ann);
2183 2184 2185
        auto val = bcx.build.Alloca(ty);
        cx.fcx.lllocals.insert(local.id, val);
    }
2186
    auto r = res(bcx, C_nil());
2187

2188
    for (@ast.stmt s in b.node.stmts) {
2189 2190
        r = trans_stmt(bcx, *s);
        bcx = r.bcx;
2191 2192 2193 2194 2195
        // If we hit a terminator, control won't go any further so
        // we're in dead-code land. Stop here.
        if (is_terminated(bcx)) {
            ret r;
        }
2196
    }
2197

2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210
    alt (b.node.expr) {
        case (some[@ast.expr](?e)) {
            r = trans_expr(bcx, e);
            bcx = r.bcx;
            if (is_terminated(bcx)) {
                ret r;
            }
        }
        case (none[@ast.expr]) {
            r = res(bcx, C_nil());
        }
    }

2211
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
2212
    ret res(bcx, r.val);
2213 2214
}

2215
fn new_fn_ctxt(@crate_ctxt cx,
2216
               str name,
2217
               ValueRef llfndecl) -> @fn_ctxt {
2218

2219
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfndecl, 0u);
2220

2221
    let hashmap[ast.def_id, ValueRef] lllocals = new_def_hash[ValueRef]();
2222 2223
    let hashmap[ast.def_id, ValueRef] llargs = new_def_hash[ValueRef]();

2224
    ret @rec(llfn=llfndecl,
2225
             lltaskptr=lltaskptr,
2226
             llargs=llargs,
2227
             lllocals=lllocals,
2228
             ccx=cx);
2229 2230
}

2231

2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242
fn create_llargs_for_fn_args(@fn_ctxt cx, vec[ast.arg] args) {
    let uint arg_n = 1u;
    for (ast.arg arg in args) {
        auto llarg = llvm.LLVMGetParam(cx.llfn, arg_n);
        check (llarg as int != 0);
        cx.llargs.insert(arg.id, llarg);
        arg_n += 1u;
    }
}


2243 2244 2245 2246
// Recommended LLVM style, strange though this is, is to copy from args to
// allocas immediately upon entry; this permits us to GEP into structures we
// were passed and whatnot. Apparently mem2reg will mop up.

2247 2248
fn copy_args_to_allocas(@block_ctxt cx, vec[ast.arg] args,
                        vec[typeck.arg] arg_tys) {
2249 2250 2251

    let uint arg_n = 0u;

2252
    for (ast.arg aarg in args) {
2253 2254 2255 2256 2257 2258 2259 2260 2261
        if (aarg.mode != ast.alias) {
            auto arg_t = type_of_arg(cx.fcx.ccx, arg_tys.(arg_n));
            auto alloca = cx.build.Alloca(arg_t);
            auto argval = cx.fcx.llargs.get(aarg.id);
            cx.build.Store(argval, alloca);
            // Overwrite the llargs entry for this arg with its alloca.
            cx.fcx.llargs.insert(aarg.id, alloca);
        }

2262 2263 2264 2265
        arg_n += 1u;
    }
}

2266 2267 2268 2269 2270
fn is_terminated(@block_ctxt cx) -> bool {
    auto inst = llvm.LLVMGetLastInstruction(cx.llbb);
    ret llvm.LLVMIsATerminatorInst(inst) as int != 0;
}

2271 2272 2273 2274 2275 2276 2277 2278 2279
fn arg_tys_of_fn(ast.ann ann) -> vec[typeck.arg] {
    alt (typeck.ann_to_type(ann).struct) {
        case (typeck.ty_fn(?arg_tys, _)) {
            ret arg_tys;
        }
    }
    fail;
}

2280 2281 2282 2283 2284 2285 2286 2287 2288
fn ret_ty_of_fn(ast.ann ann) -> @typeck.ty {
    alt (typeck.ann_to_type(ann).struct) {
        case (typeck.ty_fn(_, ?ret_ty)) {
            ret ret_ty;
        }
    }
    fail;
}

2289 2290
impure fn trans_fn(@crate_ctxt cx, &ast._fn f, ast.def_id fid,
                   &ast.ann ann) {
2291

2292 2293 2294
    auto llfndecl = cx.item_ids.get(fid);
    cx.item_names.insert(cx.path, llfndecl);

2295 2296 2297
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, f.inputs);

2298
    auto bcx = new_top_block_ctxt(fcx);
2299

2300
    copy_args_to_allocas(bcx, f.inputs, arg_tys_of_fn(ann));
2301

2302 2303
    auto res = trans_block(bcx, f.body);
    if (!is_terminated(res.bcx)) {
2304 2305
        // FIXME: until LLVM has a unit type, we are moving around
        // C_nil values rather than their void type.
2306
        res.bcx.build.RetVoid();
2307
    }
2308 2309
}

G
Graydon Hoare 已提交
2310 2311 2312 2313 2314 2315 2316 2317 2318
impure fn trans_vtbl(@crate_ctxt cx, &ast._obj ob) -> ValueRef {
    let vec[ValueRef] methods = vec();
    for (@ast.method m in ob.methods) {
        trans_fn(cx, m.node.meth, m.node.id, m.node.ann);
        methods += cx.item_ids.get(m.node.id);
    }
    ret C_struct(methods);
}

2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345
fn trans_obj(@crate_ctxt cx, &ast._obj ob, ast.def_id oid,
             &ast.ann ann) {

    auto llctor_decl = cx.item_ids.get(oid);
    cx.item_names.insert(cx.path, llctor_decl);

    // Translate obj ctor fields to function arguments.
    let vec[ast.arg] fn_args = vec();
    for (ast.obj_field f in ob.fields) {
        fn_args += vec(rec(mode=ast.alias,
                           ty=f.ty,
                           ident=f.ident,
                           id=f.id));
    }

    auto fcx = new_fn_ctxt(cx, cx.path, llctor_decl);
    create_llargs_for_fn_args(fcx, fn_args);

    auto bcx = new_top_block_ctxt(fcx);

    copy_args_to_allocas(bcx, fn_args, arg_tys_of_fn(ann));

    auto pair = bcx.build.Alloca(type_of(cx, ret_ty_of_fn(ann)));

    bcx.build.Ret(pair);
}

2346
fn trans_tag_variant(@crate_ctxt cx, ast.def_id tag_id,
2347
                     &ast.variant variant, int index) {
2348
    if (_vec.len[ast.variant_arg](variant.args) == 0u) {
2349 2350 2351
        ret;    // nullary constructors are just constants
    }

2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368
    // Translate variant arguments to function arguments.
    let vec[ast.arg] fn_args = vec();
    auto i = 0u;
    for (ast.variant_arg varg in variant.args) {
        fn_args += vec(rec(mode=ast.alias,
                           ty=varg.ty,
                           ident="arg" + _uint.to_str(i, 10u),
                           id=varg.id));
    }

    auto var_ty = typeck.ann_to_type(variant.ann);
    auto llfnty = type_of(cx, var_ty);

    let str s = cx.names.next("_rust_tag") + "." + cx.path;
    let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfnty);
    cx.item_ids.insert(variant.id, llfn);

2369 2370 2371
    let ValueRef llfndecl = cx.item_ids.get(variant.id);
    cx.item_names.insert(cx.path, llfndecl);

2372 2373 2374
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, fn_args);

2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403
    auto bcx = new_top_block_ctxt(fcx);

    auto arg_tys = arg_tys_of_fn(variant.ann);
    copy_args_to_allocas(bcx, fn_args, arg_tys);

    auto info = cx.tags.get(tag_id);

    auto lltagty = T_struct(vec(T_int(), T_array(T_i8(), info.size)));

    // FIXME: better name.
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("tag"), lltagty);

    auto lltagptr = bcx.build.Alloca(lltagty);
    auto lldiscrimptr = bcx.build.GEP(lltagptr, vec(C_int(0), C_int(0)));
    bcx.build.Store(C_int(index), lldiscrimptr);

    auto llblobptr = bcx.build.GEP(lltagptr, vec(C_int(0), C_int(1)));

    // First, generate the union type.
    let vec[TypeRef] llargtys = vec();
    for (typeck.arg arg in arg_tys) {
        llargtys += vec(type_of(cx, arg.ty));
    }

    auto llunionty = T_struct(llargtys);
    auto llunionptr = bcx.build.TruncOrBitCast(llblobptr, T_ptr(llunionty));

    i = 0u;
    for (ast.variant_arg va in variant.args) {
2404
        auto llargval = bcx.build.Load(fcx.llargs.get(va.id));
2405 2406 2407 2408 2409 2410 2411 2412 2413 2414
        auto lldestptr = bcx.build.GEP(llunionptr,
                                       vec(C_int(0), C_int(i as int)));

        bcx.build.Store(llargval, lldestptr);
        i += 1u;
    }

    auto lltagval = bcx.build.Load(lltagptr);
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
    bcx.build.Ret(lltagval);
2415 2416
}

2417
impure fn trans_item(@crate_ctxt cx, &ast.item item) {
2418
    alt (item.node) {
2419
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
2420
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2421
            trans_fn(sub_cx, f, fid, ann);
2422
        }
2423 2424 2425 2426
        case (ast.item_obj(?name, ?ob, _, ?oid, ?ann)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
            trans_obj(sub_cx, ob, oid, ann);
        }
2427 2428
        case (ast.item_mod(?name, ?m, _)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2429
            trans_mod(sub_cx, m);
2430
        }
2431 2432
        case (ast.item_tag(?name, ?variants, _, ?tag_id)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2433
            auto i = 0;
2434
            for (ast.variant variant in variants) {
2435 2436
                trans_tag_variant(sub_cx, tag_id, variant, i);
                i += 1;
2437 2438
            }
        }
2439
        case (_) { /* fall through */ }
2440 2441 2442
    }
}

2443
impure fn trans_mod(@crate_ctxt cx, &ast._mod m) {
2444 2445
    for (@ast.item item in m.items) {
        trans_item(cx, *item);
2446 2447 2448
    }
}

2449

2450
fn collect_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
2451
    alt (i.node) {
2452 2453
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
            // TODO: type-params
2454 2455
            cx.items.insert(fid, i);
            auto llty = node_type(cx, ann);
2456
            let str s = cx.names.next("_rust_fn") + "." + name;
2457
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
2458
            cx.item_ids.insert(fid, llfn);
2459 2460
        }

2461 2462 2463 2464 2465 2466 2467 2468 2469
        case (ast.item_obj(?name, ?ob, _, ?oid, ?ann)) {
            // TODO: type-params
            cx.items.insert(oid, i);
            auto llty = node_type(cx, ann);
            let str s = cx.names.next("_rust_obj_ctor") + "." + name;
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
            cx.item_ids.insert(oid, llfn);
        }

2470 2471 2472 2473
        case (ast.item_const(?name, _, _, ?cid, _)) {
            cx.items.insert(cid, i);
        }

2474 2475 2476
        case (ast.item_mod(?name, ?m, ?mid)) {
            cx.items.insert(mid, i);
        }
2477 2478 2479 2480

        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
            auto vi = new_def_hash[uint]();
            auto navi = new_def_hash[uint]();
2481 2482
            let vec[tup(ast.def_id,arity)] variant_info = vec();
            cx.tags.insert(tag_id, @rec(th=mk_type_handle(),
2483 2484
                                        mutable variants=variant_info,
                                        mutable size=0u));
2485
            cx.items.insert(tag_id, i);
2486 2487
        }

2488
        case (_) { /* fall through */ }
2489 2490 2491 2492 2493
    }
    ret cx;
}


2494
fn collect_items(@crate_ctxt cx, @ast.crate crate) {
2495

2496 2497
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();
2498 2499 2500 2501

    fld = @rec( update_env_for_item = bind collect_item(_,_)
                with *fld );

2502
    fold.fold_crate[@crate_ctxt](cx, fld, crate);
2503 2504
}

2505 2506 2507 2508 2509 2510
// The tag type resolution pass, which determines all the LLVM types that
// correspond to each tag type in the crate.

fn resolve_tag_types_for_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
    alt (i.node) {
        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
2511 2512
            auto max_align = 0u;
            auto max_size = 0u;
2513 2514 2515 2516 2517 2518

            auto info = cx.tags.get(tag_id);
            let vec[tup(ast.def_id,arity)] variant_info = vec();

            for (ast.variant variant in variants) {
                auto arity_info;
2519
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
2520 2521 2522 2523 2524 2525 2526
                    auto llvariantty = type_of_variant(cx, variant);
                    auto align = llvm.LLVMPreferredAlignmentOfType(cx.td.lltd,
                                                                 llvariantty);
                    auto size = llvm.LLVMStoreSizeOfType(cx.td.lltd,
                                                         llvariantty) as uint;
                    if (max_align < align) { max_align = align; }
                    if (max_size < size) { max_size = size; }
2527 2528 2529 2530 2531 2532 2533 2534 2535 2536

                    arity_info = n_ary;
                } else {
                    arity_info = nullary;
                }

                variant_info += vec(tup(variant.id, arity_info));
            }

            info.variants = variant_info;
2537
            info.size = max_size;
2538

2539 2540 2541
            // FIXME: alignment is wrong here, manually insert padding I
            // guess :(
            auto tag_ty = T_struct(vec(T_int(), T_array(T_i8(), max_size)));
2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552
            auto th = cx.tags.get(tag_id).th.llth;
            llvm.LLVMRefineType(llvm.LLVMResolveTypeHandle(th), tag_ty);
        }
        case (_) {
            // fall through
        }
    }

    ret cx;
}

2553 2554 2555 2556 2557 2558 2559 2560 2561 2562
fn resolve_tag_types(@crate_ctxt cx, @ast.crate crate) {
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();

    fld = @rec( update_env_for_item = bind resolve_tag_types_for_item(_,_)
                with *fld );

    fold.fold_crate[@crate_ctxt](cx, fld, crate);
}

2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581
// The constant translation pass.

fn trans_constant(&@crate_ctxt cx, @ast.item it) -> @crate_ctxt {
    alt (it.node) {
        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
            auto info = cx.tags.get(tag_id);

            auto tag_ty = llvm.LLVMResolveTypeHandle(info.th.llth);
            check (llvm.LLVMCountStructElementTypes(tag_ty) == 2u);
            auto elts = vec(0 as TypeRef, 0 as TypeRef);
            llvm.LLVMGetStructElementTypes(tag_ty, _vec.buf[TypeRef](elts));
            auto union_ty = elts.(1);

            auto i = 0u;
            while (i < _vec.len[tup(ast.def_id,arity)](info.variants)) {
                auto variant_info = info.variants.(i);
                alt (variant_info._1) {
                    case (nullary) {
                        // Nullary tags become constants.
2582
                        auto union_val = C_zero_byte_arr(info.size as uint);
2583
                        auto val = C_struct(vec(C_int(i as int), union_val));
G
Graydon Hoare 已提交
2584

2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600
                        // FIXME: better name
                        auto gvar = llvm.LLVMAddGlobal(cx.llmod, val_ty(val),
                                                       _str.buf("tag"));
                        llvm.LLVMSetInitializer(gvar, val);
                        llvm.LLVMSetGlobalConstant(gvar, True);
                        cx.item_ids.insert(variant_info._0, gvar);
                    }
                    case (n_ary) {
                        // N-ary tags are treated as functions and generated
                        // later.
                    }
                }

                i += 1u;
            }
        }
2601 2602 2603 2604 2605 2606 2607 2608

        case (ast.item_const(?name, _, ?expr, ?cid, ?ann)) {
            // FIXME: The whole expr-translation system needs cloning to deal
            // with consts.
            auto v = C_int(1);
            cx.item_ids.insert(cid, v);
        }

2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625
        case (_) {
            // empty
        }
    }

    ret cx;
}

fn trans_constants(@crate_ctxt cx, @ast.crate crate) {
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();

    fld = @rec(update_env_for_item = bind trans_constant(_,_) with *fld);

    fold.fold_crate[@crate_ctxt](cx, fld, crate);
}

2626 2627 2628 2629
fn p2i(ValueRef v) -> ValueRef {
    ret llvm.LLVMConstPtrToInt(v, T_int());
}

2630
fn trans_exit_task_glue(@crate_ctxt cx) {
2631 2632 2633 2634 2635 2636 2637
    let vec[TypeRef] T_args = vec();
    let vec[ValueRef] V_args = vec();

    auto llfn = cx.glues.exit_task_glue;
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfn, 0u);
    auto fcx = @rec(llfn=llfn,
                    lltaskptr=lltaskptr,
2638
                    llargs=new_def_hash[ValueRef](),
2639
                    lllocals=new_def_hash[ValueRef](),
2640
                    ccx=cx);
2641

2642
    auto bcx = new_top_block_ctxt(fcx);
2643
    trans_upcall(bcx, "upcall_exit", V_args);
2644
    bcx.build.RetVoid();
2645 2646
}

2647 2648 2649
fn create_typedefs(@crate_ctxt cx) {
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_crate"), T_crate());
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_task"), T_task());
2650
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_tydesc"), T_tydesc());
2651 2652
}

2653
fn crate_constant(@crate_ctxt cx) -> ValueRef {
2654 2655

    let ValueRef crate_ptr =
2656
        llvm.LLVMAddGlobal(cx.llmod, T_crate(),
2657 2658 2659 2660 2661 2662 2663 2664 2665 2666
                           _str.buf("rust_crate"));

    let ValueRef crate_addr = p2i(crate_ptr);

    let ValueRef activate_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.activate_glue), crate_addr);

    let ValueRef yield_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.yield_glue), crate_addr);

2667 2668
    let ValueRef exit_task_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.exit_task_glue), crate_addr);
2669 2670 2671

    let ValueRef crate_val =
        C_struct(vec(C_null(T_int()),     // ptrdiff_t image_base_off
2672
                     p2i(crate_ptr),      // uintptr_t self_addr
2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690
                     C_null(T_int()),     // ptrdiff_t debug_abbrev_off
                     C_null(T_int()),     // size_t debug_abbrev_sz
                     C_null(T_int()),     // ptrdiff_t debug_info_off
                     C_null(T_int()),     // size_t debug_info_sz
                     activate_glue_off,   // size_t activate_glue_off
                     yield_glue_off,      // size_t yield_glue_off
                     C_null(T_int()),     // size_t unwind_glue_off
                     C_null(T_int()),     // size_t gc_glue_off
                     exit_task_glue_off,  // size_t main_exit_task_glue_off
                     C_null(T_int()),     // int n_rust_syms
                     C_null(T_int()),     // int n_c_syms
                     C_null(T_int())      // int n_libs
                     ));

    llvm.LLVMSetInitializer(crate_ptr, crate_val);
    ret crate_ptr;
}

2691
fn trans_main_fn(@crate_ctxt cx, ValueRef llcrate) {
2692 2693 2694
    auto T_main_args = vec(T_int(), T_int());
    auto T_rust_start_args = vec(T_int(), T_int(), T_int(), T_int());

2695 2696 2697 2698 2699 2700 2701
    auto main_name;
    if (_str.eq(std.os.target_os(), "win32")) {
        main_name = "WinMain@16";
    } else {
        main_name = "main";
    }

2702
    auto llmain =
2703
        decl_cdecl_fn(cx.llmod, main_name, T_fn(T_main_args, T_int()));
2704

2705 2706
    auto llrust_start = decl_cdecl_fn(cx.llmod, "rust_start",
                                      T_fn(T_rust_start_args, T_int()));
2707 2708 2709

    auto llargc = llvm.LLVMGetParam(llmain, 0u);
    auto llargv = llvm.LLVMGetParam(llmain, 1u);
2710 2711
    check (cx.item_names.contains_key("_rust.main"));
    auto llrust_main = cx.item_names.get("_rust.main");
2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722

    //
    // Emit the moral equivalent of:
    //
    // main(int argc, char **argv) {
    //     rust_start(&_rust.main, &crate, argc, argv);
    // }
    //

    let BasicBlockRef llbb =
        llvm.LLVMAppendBasicBlock(llmain, _str.buf(""));
2723
    auto b = new_builder(llbb, "");
2724 2725 2726 2727 2728 2729

    auto start_args = vec(p2i(llrust_main), p2i(llcrate), llargc, llargv);

    b.Ret(b.Call(llrust_start, start_args));
}

2730 2731
fn declare_intrinsics(ModuleRef llmod) -> hashmap[str,ValueRef] {

2732
    let vec[TypeRef] T_trap_args = vec();
2733 2734 2735 2736
    let vec[TypeRef] T_memcpy32_args = vec(T_ptr(T_i8()), T_ptr(T_i8()),
                                           T_i32(), T_i32(), T_i1());
    let vec[TypeRef] T_memcpy64_args = vec(T_ptr(T_i8()), T_ptr(T_i8()),
                                           T_i32(), T_i32(), T_i1());
2737 2738
    auto trap = decl_cdecl_fn(llmod, "llvm.trap",
                              T_fn(T_trap_args, T_void()));
2739 2740 2741 2742
    auto memcpy32 = decl_cdecl_fn(llmod, "llvm.memcpy.p0i8.p0i8.i32",
                                  T_fn(T_memcpy32_args, T_void()));
    auto memcpy64 = decl_cdecl_fn(llmod, "llvm.memcpy.p0i8.p0i8.i64",
                                  T_fn(T_memcpy64_args, T_void()));
2743 2744 2745

    auto intrinsics = new_str_hash[ValueRef]();
    intrinsics.insert("llvm.trap", trap);
2746 2747
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i32", memcpy32);
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i64", memcpy64);
2748
    ret intrinsics;
2749 2750
}

2751 2752 2753 2754 2755 2756 2757 2758
fn check_module(ModuleRef llmod) {
    auto pm = mk_pass_manager();
    llvm.LLVMAddVerifierPass(pm.llpm);
    llvm.LLVMRunPassManager(pm.llpm, llmod);

    // TODO: run the linter here also, once there are llvm-c bindings for it.
}

2759
fn trans_crate(session.session sess, @ast.crate crate, str output) {
2760 2761 2762 2763
    auto llmod =
        llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
                                               llvm.LLVMGetGlobalContext());

2764 2765
    llvm.LLVMSetDataLayout(llmod, _str.buf(x86.get_data_layout()));
    llvm.LLVMSetTarget(llmod, _str.buf(x86.get_target_triple()));
2766
    auto td = mk_target_data(x86.get_data_layout());
2767

2768 2769
    llvm.LLVMSetModuleInlineAsm(llmod, _str.buf(x86.get_module_asm()));

2770
    auto intrinsics = declare_intrinsics(llmod);
2771

2772 2773 2774
    auto glues = @rec(activate_glue = decl_glue(llmod,
                                                abi.activate_glue_name()),
                      yield_glue = decl_glue(llmod, abi.yield_glue_name()),
2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788
                      /*
                       * Note: the signature passed to decl_cdecl_fn here
                       * looks unusual because it is. It corresponds neither
                       * to an upcall signature nor a normal rust-ABI
                       * signature. In fact it is a fake signature, that
                       * exists solely to acquire the task pointer as an
                       * argument to the upcall. It so happens that the
                       * runtime sets up the task pointer as the sole incoming
                       * argument to the frame that we return into when
                       * returning to the exit task glue. So this is the
                       * signature required to retrieve it.
                       */
                      exit_task_glue =
                      decl_cdecl_fn(llmod, abi.exit_task_glue_name(),
2789
                                    T_fn(vec(T_taskptr()), T_void())),
2790

2791 2792 2793 2794
                      upcall_glues =
                      _vec.init_fn[ValueRef](bind decl_upcall(llmod, _),
                                             abi.n_upcall_glues as uint));

2795 2796 2797 2798
    auto hasher = typeck.hash_ty;
    auto eqer = typeck.eq_ty;
    auto types = map.mk_hashmap[@typeck.ty,@ty_info](hasher, eqer);

2799 2800
    auto cx = @rec(sess = sess,
                   llmod = llmod,
2801
                   td = td,
2802
                   upcalls = new_str_hash[ValueRef](),
2803
                   intrinsics = intrinsics,
2804 2805
                   item_names = new_str_hash[ValueRef](),
                   item_ids = new_def_hash[ValueRef](),
2806
                   items = new_def_hash[@ast.item](),
2807
                   tags = new_def_hash[@tag_info](),
2808
                   types = types,
2809
                   glues = glues,
2810
                   names = namegen(0),
2811
                   path = "_rust");
2812

2813 2814
    create_typedefs(cx);

2815
    collect_items(cx, crate);
2816
    resolve_tag_types(cx, crate);
2817
    trans_constants(cx, crate);
2818

2819
    trans_mod(cx, crate.node.module);
2820
    trans_exit_task_glue(cx);
2821 2822
    trans_main_fn(cx, crate_constant(cx));

2823 2824
    check_module(llmod);

2825
    llvm.LLVMWriteBitcodeToFile(llmod, _str.buf(output));
2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838
    llvm.LLVMDisposeModule(llmod);
}

//
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C ../.. 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End:
//