trans.rs 95.7 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,
                    ValueRef no_op_type_glue);
54

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

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

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

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

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

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

93 94 95 96 97 98 99 100
// 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);
}

101

102 103 104 105 106 107 108 109
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);
}

110 111 112 113 114 115 116 117 118 119 120
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));
}
121 122 123 124


// LLVM type constructors.

125 126 127 128 129 130 131 132 133 134 135
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.
136 137 138
    ret llvm.LLVMVoidType();
}

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

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

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

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

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

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

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

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

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

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

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

185 186 187 188
fn T_fn(vec[TypeRef] inputs, TypeRef output) -> TypeRef {
    ret llvm.LLVMFunctionType(output,
                              _vec.buf[TypeRef](inputs),
                              _vec.len[TypeRef](inputs),
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
                              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
208 209 210 211 212 213 214 215 216 217
                     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
                     ));
}

218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
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
}

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

237 238 239 240 241
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
242 243 244
                     ));
}

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

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

253 254 255 256 257 258 259 260 261 262 263 264 265 266
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
267
                     T_int()       // int n_libs
268 269 270
                     ));
}

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

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

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

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

    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;
    }

299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319
    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);
}

320
fn type_of_inner(@crate_ctxt cx, @typeck.ty t) -> TypeRef {
321 322 323 324 325 326
    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)) {
327 328 329 330 331 332 333 334 335 336 337 338 339
            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(); }
            }
        }
340
        case (typeck.ty_char) { ret T_char(); }
341
        case (typeck.ty_str) { ret T_ptr(T_str()); }
342 343 344
        case (typeck.ty_tag(?tag_id)) {
            ret llvm.LLVMResolveTypeHandle(cx.tags.get(tag_id).th.llth);
        }
345
        case (typeck.ty_box(?t)) {
346 347
            ret T_ptr(T_box(type_of(cx, t)));
        }
348
        case (typeck.ty_vec(?t)) {
349
            ret T_ptr(T_vec(type_of(cx, t)));
350
        }
351
        case (typeck.ty_tup(?elts)) {
352
            let vec[TypeRef] tys = vec();
353 354
            for (@typeck.ty elt in elts) {
                tys += type_of(cx, elt);
355 356 357
            }
            ret T_struct(tys);
        }
358 359 360 361 362 363 364
        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);
        }
365
        case (typeck.ty_fn(?args, ?out)) {
366 367 368 369 370 371 372
            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);
373
            }
374 375 376 377 378
            let TypeRef vtbl = T_struct(mtys);
            let TypeRef pair =
                T_struct(vec(T_ptr(vtbl),
                             T_ptr(T_box(T_opaque()))));
            ret pair;
379
        }
380 381
        case (typeck.ty_var(_)) {
            log "ty_var in trans.type_of";
382 383 384 385
            fail;
        }
        case (typeck.ty_param(_)) {
            ret T_ptr(T_i8());
386 387 388 389 390
        }
    }
    fail;
}

391 392 393 394 395 396 397 398
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;
}

399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
// 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;
}

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

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

421
fn C_integral(int i, TypeRef t) -> ValueRef {
422 423 424 425 426 427
    // 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);
    //
428 429 430
    ret llvm.LLVMConstIntOfString(t, _str.buf(istr(i)), 10);
}

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

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

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

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

457 458 459 460 461 462 463 464 465 466 467
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));
}

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

474
fn decl_fn(ModuleRef llmod, str name, uint cc, TypeRef llty) -> ValueRef {
475 476
    let ValueRef llfn =
        llvm.LLVMAddFunction(llmod, _str.buf(name), llty);
477
    llvm.LLVMSetFunctionCallConv(llfn, cc);
478 479 480
    ret llfn;
}

481 482
fn decl_cdecl_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMCCallConv, llty);
483 484
}

485 486
fn decl_fastcall_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMFastCallConv, llty);
487 488
}

489
fn decl_glue(ModuleRef llmod, str s) -> ValueRef {
490
    ret decl_cdecl_fn(llmod, s, T_fn(vec(T_taskptr()), T_void()));
491 492
}

493
fn decl_upcall(ModuleRef llmod, uint _n) -> ValueRef {
494 495 496 497
    // 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.
498
    let int n = _n as int;
499
    let str s = abi.upcall_glue_name(n);
500
    let vec[TypeRef] args =
501 502
        vec(T_taskptr(), // taskptr
            T_int())     // callee
503 504
        + _vec.init_elt[TypeRef](T_int(), n as uint);

505
    ret decl_fastcall_fn(llmod, s, T_fn(args, T_int()));
506 507
}

508
fn get_upcall(@crate_ctxt cx, str name, int n_args) -> ValueRef {
509 510 511
    if (cx.upcalls.contains_key(name)) {
        ret cx.upcalls.get(name);
    }
512
    auto inputs = vec(T_taskptr());
513
    inputs += _vec.init_elt[TypeRef](T_int(), n_args as uint);
514
    auto output = T_int();
515
    auto f = decl_cdecl_fn(cx.llmod, name, T_fn(inputs, output));
516 517 518 519
    cx.upcalls.insert(name, f);
    ret f;
}

520
fn trans_upcall(@block_ctxt cx, str name, vec[ValueRef] args) -> result {
521
    let int n = _vec.len[ValueRef](args) as int;
522
    let ValueRef llupcall = get_upcall(cx.fcx.ccx, name, n);
523 524
    llupcall = llvm.LLVMConstPointerCast(llupcall, T_int());

525
    let ValueRef llglue = cx.fcx.ccx.glues.upcall_glues.(n);
526 527 528 529
    let vec[ValueRef] call_args = vec(cx.fcx.lltaskptr, llupcall);
    for (ValueRef a in args) {
        call_args += cx.build.ZExtOrBitCast(a, T_int());
    }
530
    ret res(cx, cx.build.FastCall(llglue, call_args));
531 532
}

533 534 535
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)));
536 537
}

538 539 540 541 542 543 544 545 546 547 548 549 550 551
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;
        }
    }
}

552
fn trans_malloc(@block_ctxt cx, @typeck.ty t) -> result {
553
    auto scope_cx = find_scope_cx(cx);
554 555 556 557 558 559 560
    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);
561
    scope_cx.cleanups += clean(bind drop_ty(_, sub.val, t));
562 563 564 565
    ret sub;
}


566 567 568 569 570 571 572 573 574
// Type descriptor and type glue stuff

// Given a type and a field index into its corresponding type descriptor,
// returns an LLVM ValueRef of that field from the tydesc, generating the
// tydesc if necessary.
fn field_of_tydesc(@block_ctxt cx, @typeck.ty ty, int field) -> ValueRef {
    auto tydesc = get_tydesc(cx.fcx.ccx, ty);
    ret cx.build.GEP(tydesc, vec(C_int(0), C_int(field)));
}
575

576 577 578
fn get_tydesc(@crate_ctxt cx, @typeck.ty ty) -> ValueRef {
    if (!cx.tydescs.contains_key(ty)) {
        make_tydesc(cx, ty);
579
    }
580
    ret cx.tydescs.get(ty);
581 582
}

583
fn make_tydesc(@crate_ctxt cx, @typeck.ty ty) {
584 585 586 587
    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);
588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607

    auto llty = type_of(cx, ty);
    auto pvoid = T_ptr(T_i8());
    auto glue_fn_ty = T_ptr(T_fn(vec(T_taskptr(), pvoid), T_void()));
    auto tydesc = C_struct(vec(C_null(pvoid),
                               llvm.LLVMSizeOf(llty),
                               llvm.LLVMAlignOf(llty),
                               take_glue,             // copy_glue_off
                               drop_glue,             // drop_glue_off
                               C_null(glue_fn_ty),    // free_glue_off
                               C_null(glue_fn_ty),    // sever_glue_off
                               C_null(glue_fn_ty),    // mark_glue_off
                               C_null(glue_fn_ty),    // obj_drop_glue_off
                               C_null(glue_fn_ty)));  // is_stateful

    auto name = sanitize(cx.names.next("tydesc_" + typeck.ty_to_str(ty)));
    auto gvar = llvm.LLVMAddGlobal(cx.llmod, val_ty(tydesc), _str.buf(name));
    llvm.LLVMSetInitializer(gvar, tydesc);
    llvm.LLVMSetGlobalConstant(gvar, True);
    cx.tydescs.insert(ty, gvar);
608 609
}

610 611
fn make_generic_glue(@crate_ctxt cx, @typeck.ty t, str name,
                     val_and_ty_fn helper) -> ValueRef {
612
    auto llfnty = T_fn(vec(T_taskptr(), T_ptr(T_i8())), T_void());
613

614
    auto fn_name = cx.names.next("_rust_" + name) + "." + typeck.ty_to_str(t);
615 616 617 618 619 620
    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);

621 622 623 624 625 626 627 628
    auto re;
    if (!typeck.type_is_scalar(t)) {
        auto llty;
        if (typeck.type_is_structural(t)) {
            llty = T_ptr(type_of(cx, t));
        } else {
            llty = type_of(cx, t);
        }
629

630 631 632 633 634 635 636
        auto llrawptr = llvm.LLVMGetParam(llfn, 1u);
        auto llval = bcx.build.BitCast(llrawptr, llty);
        
        re = helper(bcx, llval, t);
    } else {
        re = res(bcx, C_nil());
    }
637

638
    re.bcx.build.RetVoid();
639 640 641
    ret llfn;
}

642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
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 {
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 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733
    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;
}

734 735
fn decr_refcnt_and_if_zero(@block_ctxt cx,
                           ValueRef box_ptr,
736
                           fn(@block_ctxt cx) -> result inner,
737
                           str inner_name,
738
                           TypeRef t_else, ValueRef v_else) -> result {
739

740
    auto load_rc_cx = new_sub_block_ctxt(cx, "load rc");
741 742 743 744
    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");

745 746
    auto null_test = cx.build.IsNull(box_ptr);
    cx.build.CondBr(null_test, next_cx.llbb, load_rc_cx.llbb);
747

748 749 750 751 752 753 754 755 756 757

    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);
758 759 760 761

    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);
762 763 764 765
    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);
766

767
    auto phi = next_cx.build.Phi(t_else,
768
                                 vec(v_else, v_else, v_else, inner_res.val),
769
                                 vec(cx.llbb,
770
                                     load_rc_cx.llbb,
771
                                     rc_adj_cx.llbb,
772 773
                                     inner_res.bcx.llbb));

774
    ret res(next_cx, phi);
775 776
}

777 778 779 780 781 782 783 784 785 786 787 788 789
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);
}

790 791 792
type val_and_ty_fn =
    fn(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result;

793
// Iterates through the elements of a box, tup, rec or tag.
794 795 796 797 798 799 800 801 802
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;
803
            for (@typeck.ty arg in args) {
804
                auto elt = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
805 806 807
                r = f(r.bcx,
                      load_non_structural(r.bcx, elt, arg),
                      arg);
808 809 810
                i += 1;
            }
        }
811 812 813 814
        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)));
815 816 817
                r = f(r.bcx,
                      load_non_structural(r.bcx, llfld, fld.ty),
                      fld.ty);
818 819 820
                i += 1;
            }
        }
821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842
        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 已提交
843

844 845 846 847 848
            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 已提交
849

850 851 852 853 854 855 856 857 858 859 860 861
            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));
862
                        auto llvar = variant_cx.build.GEP(v, vals);
863 864
                        auto llvarty = type_of_variant(cx.fcx.ccx,
                                                       variants.(i));
865

866
                        auto fn_ty = typeck.ann_to_type(variants.(i).ann);
867 868
                        alt (fn_ty.struct) {
                            case (typeck.ty_fn(?args, _)) {
869 870 871 872
                                auto llvarp = variant_cx.build.
                                    TruncOrBitCast(llunion_ptr,
                                                   T_ptr(llvarty));

873 874
                                auto j = 0u;
                                for (typeck.arg a in args) {
875 876
                                    auto llfldp = variant_cx.build.GEP(llvarp,
                                        vec(C_int(0), C_int(j as int)));
877 878
                                    auto llfld =
                                        load_non_structural(variant_cx,
879
                                                            llfldp, a.ty);
880

881 882 883 884 885 886 887 888 889
                                    auto res = f(variant_cx, llfld, a.ty);
                                    variant_cx = res.bcx;
                                    j += 1u;
                                }
                            }
                            case (_) { fail; }
                        }

                        variant_cx.build.Br(next_cx.llbb);
890 891 892 893 894 895 896 897 898 899 900 901
                    }
                    case (nullary) {
                        // Nothing to do.
                        variant_cx.build.Br(next_cx.llbb);
                    }
                }

                i += 1u;
            }

            ret res(next_cx, C_nil());
        }
902 903 904
        case (_) {
            cx.fcx.ccx.sess.unimpl("type in iter_structural_ty");
        }
905
    }
906
    ret r;
907 908
}

909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
// 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)));
925 926 927 928 929

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

930 931
        auto len = cx.build.Load(lenptr);
        if (trailing_null) {
932
            len = cx.build.Sub(len, unit_sz);
933 934 935 936
        }

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

937 938
        auto cond_cx = new_scope_block_ctxt(cx, "sequence-iter cond");
        auto body_cx = new_scope_block_ctxt(cx, "sequence-iter body");
939 940
        auto next_cx = new_sub_block_ctxt(cx, "next");

941 942
        cx.build.Br(cond_cx.llbb);

943
        auto ix = cond_cx.build.Phi(T_int(), vec(C_int(0)), vec(cx.llbb));
944 945 946 947 948
        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);
949 950
        cond_cx.build.CondBr(end_test, body_cx.llbb, next_cx.llbb);

951
        auto elt = body_cx.build.GEP(p0, vec(C_int(0), ix));
952 953 954
        auto body_res = f(body_cx,
                          load_non_structural(body_cx, elt, elt_ty),
                          elt_ty);
955
        auto next_ix = body_res.bcx.build.Add(ix, C_int(1));
956 957
        auto next_scaled_ix = body_res.bcx.build.Add(scaled_ix, unit_sz);

958 959 960
        cond_cx.build.AddIncomingToPhi(ix, vec(next_ix),
                                       vec(body_res.bcx.llbb));

961 962 963
        cond_cx.build.AddIncomingToPhi(scaled_ix, vec(next_scaled_ix),
                                       vec(body_res.bcx.llbb));

964 965 966 967 968 969 970 971 972 973
        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));
974
            ret iter_sequence_body(cx, v, et, f, true);
975
        }
976
        case (_) { fail; }
977
    }
978 979 980 981 982 983 984
    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 {
985 986
    if (!typeck.type_is_scalar(t)) {
        auto llrawptr = cx.build.BitCast(v, T_ptr(T_i8()));
987 988 989
        auto llfnptr = field_of_tydesc(cx, t, abi.tydesc_field_copy_glue_off);
        auto llfn = cx.build.Load(llfnptr);
        cx.build.FastCall(llfn, vec(cx.fcx.lltaskptr, llrawptr));
990
    }
991
    ret res(cx, C_nil());
992 993
}

994 995 996
fn drop_slot(@block_ctxt cx,
             ValueRef slot,
             @typeck.ty t) -> result {
997 998 999 1000 1001 1002 1003
    auto llptr = load_non_structural(cx, slot, t);
    auto re = drop_ty(cx, llptr, t);

    auto llty = val_ty(slot);
    auto llelemty = lib.llvm.llvm.LLVMGetElementType(llty);
    re.bcx.build.Store(C_null(llelemty), slot);
    ret re;
1004 1005
}

1006 1007 1008
fn drop_ty(@block_ctxt cx,
           ValueRef v,
           @typeck.ty t) -> result {
1009 1010
    if (!typeck.type_is_scalar(t)) {
        auto llrawptr = cx.build.BitCast(v, T_ptr(T_i8()));
1011 1012 1013
        auto llfnptr = field_of_tydesc(cx, t, abi.tydesc_field_drop_glue_off);
        auto llfn = cx.build.Load(llfnptr);
        cx.build.FastCall(llfn, vec(cx.fcx.lltaskptr, llrawptr));
1014
    }
1015
    ret res(cx, C_nil());
1016 1017 1018 1019 1020 1021
}

fn build_memcpy(@block_ctxt cx,
                ValueRef dst,
                ValueRef src,
                TypeRef llty) -> result {
1022 1023 1024
    // 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");
1025 1026
    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 已提交
1027 1028
    auto size = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llty),
                                 T_i32());
1029 1030 1031 1032 1033 1034
    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 已提交
1035
    log "building memcpy";
1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
    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)) {
1057
        auto r = incr_all_refcnts(cx, src, t);
1058
        if (! is_init) {
1059
            r = drop_ty(r.bcx, r.bcx.build.Load(dst), t);
1060 1061 1062 1063 1064 1065 1066 1067
        }
        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);
        }
1068 1069 1070 1071
        // 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));
1072 1073 1074 1075
    }

    cx.fcx.ccx.sess.bug("unexpected type in trans.copy_ty: " +
                        typeck.ty_to_str(t));
1076 1077 1078
    fail;
}

1079
impure fn trans_lit(@block_ctxt cx, &ast.lit lit, &ast.ann ann) -> result {
1080
    alt (lit.node) {
1081
        case (ast.lit_int(?i)) {
1082
            ret res(cx, C_int(i));
1083 1084
        }
        case (ast.lit_uint(?u)) {
1085
            ret res(cx, C_int(u as int));
1086
        }
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
        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));
        }
1108
        case (ast.lit_char(?c)) {
1109
            ret res(cx, C_integral(c as int, T_char()));
1110 1111
        }
        case (ast.lit_bool(?b)) {
1112 1113 1114 1115
            ret res(cx, C_bool(b));
        }
        case (ast.lit_nil) {
            ret res(cx, C_nil());
1116 1117 1118
        }
        case (ast.lit_str(?s)) {
            auto len = (_str.byte_len(s) as int) + 1;
1119
            auto sub = trans_upcall(cx, "upcall_new_str",
1120
                                    vec(p2i(C_str(cx.fcx.ccx, s)),
1121 1122
                                        C_int(len)));
            sub.val = sub.bcx.build.IntToPtr(sub.val,
1123
                                             T_ptr(T_str()));
1124
            auto t = node_ann_type(cx.fcx.ccx, ann);
1125
            find_scope_cx(cx).cleanups +=
1126
                clean(bind drop_ty(_, sub.val, t));
1127
            ret sub;
1128 1129 1130 1131
        }
    }
}

1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
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);
        }
1142
        case (_) { /* fall through */ }
1143 1144 1145 1146 1147
    }
    ret t;
}

fn node_ann_type(@crate_ctxt cx, &ast.ann a) -> @typeck.ty {
1148 1149
    alt (a) {
        case (ast.ann_none) {
1150
            cx.sess.bug("missing type annotation");
1151 1152
        }
        case (ast.ann_type(?t)) {
1153
            ret target_type(cx, t);
1154 1155 1156 1157
        }
    }
}

1158 1159 1160 1161
fn node_type(@crate_ctxt cx, &ast.ann a) -> TypeRef {
    ret type_of(cx, node_ann_type(cx, a));
}

1162
impure fn trans_unary(@block_ctxt cx, ast.unop op,
1163
                      @ast.expr e, &ast.ann a) -> result {
1164 1165 1166

    auto sub = trans_expr(cx, e);

1167 1168
    alt (op) {
        case (ast.bitnot) {
1169 1170
            sub.val = cx.build.Not(sub.val);
            ret sub;
1171 1172
        }
        case (ast.not) {
1173 1174
            sub.val = cx.build.Not(sub.val);
            ret sub;
1175 1176 1177
        }
        case (ast.neg) {
            // FIXME: switch by signedness.
1178 1179
            sub.val = cx.build.Neg(sub.val);
            ret sub;
1180
        }
1181
        case (ast.box) {
1182 1183 1184 1185 1186
            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,
1187 1188
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
1189 1190 1191 1192 1193 1194
            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);
1195
        }
1196 1197 1198 1199 1200 1201 1202 1203 1204 1205
        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;
1206
        }
1207 1208 1209 1210
    }
    fail;
}

1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
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;
}

1247
impure fn trans_binary(@block_ctxt cx, ast.binop op,
1248
                       @ast.expr a, @ast.expr b) -> result {
1249

1250 1251 1252 1253 1254 1255 1256
    // First couple cases are lazy:

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

1257
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1258 1259
            auto rhs_res = trans_expr(rhs_cx, b);

1260
            auto lhs_false_cx = new_scope_block_ctxt(cx, "lhs false");
1261
            auto lhs_false_res = res(lhs_false_cx, C_bool(false));
1262 1263 1264

            lhs_res.bcx.build.CondBr(lhs_res.val,
                                     rhs_cx.llbb,
1265 1266 1267 1268
                                     lhs_false_cx.llbb);

            ret join_results(cx, T_bool(),
                             vec(lhs_false_res, rhs_res));
1269 1270 1271 1272 1273 1274
        }

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

1275
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1276 1277
            auto rhs_res = trans_expr(rhs_cx, b);

1278
            auto lhs_true_cx = new_scope_block_ctxt(cx, "lhs true");
1279
            auto lhs_true_res = res(lhs_true_cx, C_bool(true));
1280 1281

            lhs_res.bcx.build.CondBr(lhs_res.val,
1282
                                     lhs_true_cx.llbb,
1283
                                     rhs_cx.llbb);
1284 1285 1286

            ret join_results(cx, T_bool(),
                             vec(lhs_true_res, rhs_res));
1287
        }
1288 1289

        case (_) {
1290 1291 1292 1293 1294
            // 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));
1295
        }
1296 1297 1298 1299
    }
    fail;
}

1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330
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);
        }
1331 1332

        case (_) { /* fall through */ }
1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343
    }

    // 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);
}

1344
impure fn trans_if(@block_ctxt cx, @ast.expr cond,
1345
                   &ast.block thn, &option.t[ast.block] els) -> result {
1346 1347 1348

    auto cond_res = trans_expr(cx, cond);

1349
    auto then_cx = new_scope_block_ctxt(cx, "then");
1350 1351
    auto then_res = trans_block(then_cx, thn);

1352
    auto else_cx = new_scope_block_ctxt(cx, "else");
1353
    auto else_res = res(else_cx, C_nil());
1354 1355 1356

    alt (els) {
        case (some[ast.block](?eblk)) {
1357
            else_res = trans_block(else_cx, eblk);
1358
        }
1359
        case (_) { /* fall through */ }
1360 1361
    }

1362
    cond_res.bcx.build.CondBr(cond_res.val,
1363 1364
                              then_cx.llbb,
                              else_cx.llbb);
1365 1366 1367 1368

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

1371
impure fn trans_while(@block_ctxt cx, @ast.expr cond,
1372 1373
                      &ast.block body) -> result {

1374 1375
    auto cond_cx = new_scope_block_ctxt(cx, "while cond");
    auto body_cx = new_scope_block_ctxt(cx, "while loop body");
1376
    auto next_cx = new_sub_block_ctxt(cx, "next");
1377 1378

    auto body_res = trans_block(body_cx, body);
1379 1380 1381 1382 1383 1384 1385 1386
    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);
1387 1388 1389 1390
    ret res(next_cx, C_nil());
}

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

1393
    auto body_cx = new_scope_block_ctxt(cx, "do-while loop body");
1394
    auto next_cx = new_sub_block_ctxt(cx, "next");
1395 1396

    auto body_res = trans_block(body_cx, body);
1397 1398 1399 1400 1401 1402
    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);
1403 1404 1405
    ret res(next_cx, body_res.val);
}

P
Patrick Walton 已提交
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 1472 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
// 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;
1508
                i += 1;
P
Patrick Walton 已提交
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550
            }

            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());
}

1551 1552 1553
// 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).
1554

1555 1556 1557 1558 1559 1560 1561 1562
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)),
1563
                            true);
1564 1565 1566 1567 1568
                }
                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 已提交
1569
                }
P
Patrick Walton 已提交
1570 1571 1572 1573
                case (ast.def_binding(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
                    ret tup(res(cx, cx.fcx.lllocals.get(did)), true);
                }
1574
                case (ast.def_fn(?did)) {
1575 1576
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
1577 1578
                            false);
                }
1579 1580 1581 1582 1583
                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);
                }
1584 1585
                case (ast.def_variant(?tid, ?vid)) {
                    check (cx.fcx.ccx.tags.contains_key(tid));
1586 1587 1588
                    check (cx.fcx.ccx.item_ids.contains_key(vid));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(vid)),
                            false);
1589
                }
1590 1591
                case (_) {
                    cx.fcx.ccx.sess.unimpl("def variant in trans");
G
Graydon Hoare 已提交
1592 1593 1594
                }
            }
        }
1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
        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);
        }
1613 1614 1615 1616 1617 1618
        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);
        }
1619
        case (_) { cx.fcx.ccx.sess.unimpl("field variant in trans_field"); }
1620 1621 1622 1623
    }
    fail;
}

1624 1625 1626
fn trans_index(@block_ctxt cx, &ast.span sp, @ast.expr base,
               @ast.expr idx, &ast.ann ann) -> tup(result, bool) {

G
Graydon Hoare 已提交
1627 1628 1629
    auto lv = trans_expr(cx, base);
    auto ix = trans_expr(lv.bcx, idx);
    auto v = lv.val;
1630 1631 1632 1633 1634 1635 1636 1637

    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);
1638

1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
    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 已提交
1656
    ret tup(res(next_cx, elt), true);
1657 1658
}

1659 1660 1661 1662
// 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).

1663 1664 1665 1666 1667 1668 1669 1670
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);
        }
1671 1672 1673
        case (ast.expr_index(?base, ?idx, ?ann)) {
            ret trans_index(cx, e.span, base, idx, ann);
        }
1674
        case (_) { cx.fcx.ccx.sess.unimpl("expr variant in trans_lval"); }
G
Graydon Hoare 已提交
1675 1676 1677 1678
    }
    fail;
}

1679
impure fn trans_cast(@block_ctxt cx, @ast.expr e, &ast.ann ann) -> result {
1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709
    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;
}

1710

1711
impure fn trans_args(@block_ctxt cx, &vec[@ast.expr] es, @typeck.ty fn_ty)
1712 1713 1714 1715
    -> tup(@block_ctxt, vec[ValueRef]) {
    let vec[ValueRef] vs = vec(cx.fcx.lltaskptr);
    let @block_ctxt bcx = cx;

1716 1717 1718 1719 1720 1721 1722
    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;
1723
    for (@ast.expr e in es) {
1724 1725 1726
        auto mode = args.(i).mode;

        auto re;
1727
        if (typeck.type_is_structural(typeck.expr_ty(e))) {
1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754
            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);
            }
1755
        }
1756 1757 1758 1759 1760

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

        i += 1u;
1761 1762 1763 1764 1765
    }

    ret tup(bcx, vs);
}

1766
impure fn trans_call(@block_ctxt cx, @ast.expr f,
G
Graydon Hoare 已提交
1767
                     vec[@ast.expr] args, &ast.ann ann) -> result {
1768 1769
    auto f_res = trans_lval(cx, f);
    check (! f_res._1);
1770
    auto fn_ty = typeck.expr_ty(f);
G
Graydon Hoare 已提交
1771
    auto ret_ty = typeck.ann_to_type(ann);
1772
    auto args_res = trans_args(f_res._0.bcx, args, fn_ty);
1773 1774 1775 1776 1777 1778 1779 1780
    
    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 已提交
1781

G
Graydon Hoare 已提交
1782 1783 1784 1785 1786 1787 1788 1789 1790
    // 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 已提交
1791 1792 1793 1794 1795 1796
    // 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);
1797 1798
}

1799
impure fn trans_tup(@block_ctxt cx, vec[ast.elt] elts,
G
Graydon Hoare 已提交
1800
                    &ast.ann ann) -> result {
1801 1802 1803 1804
    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 已提交
1805 1806
    let int i = 0;
    auto r = res(cx, C_nil());
1807 1808 1809
    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 已提交
1810 1811 1812 1813 1814 1815 1816
        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 已提交
1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851
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;
    }
1852 1853 1854 1855
    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 已提交
1856 1857 1858
    ret res(sub.bcx, vec_val);
}

1859
impure fn trans_rec(@block_ctxt cx, vec[ast.field] fields,
1860
                    &ast.ann ann) -> result {
1861 1862 1863 1864
    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));
1865 1866
    let int i = 0;
    auto r = res(cx, C_nil());
1867 1868 1869 1870
    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)));
1871 1872 1873 1874
        // FIXME: calculate copy init-ness in typestate.
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
1875
    ret res(r.bcx, rec_val);
1876 1877
}

G
Graydon Hoare 已提交
1878

G
Graydon Hoare 已提交
1879

1880
impure fn trans_expr(@block_ctxt cx, @ast.expr e) -> result {
1881
    alt (e.node) {
1882 1883
        case (ast.expr_lit(?lit, ?ann)) {
            ret trans_lit(cx, *lit, ann);
1884 1885
        }

1886
        case (ast.expr_unary(?op, ?x, ?ann)) {
1887
            ret trans_unary(cx, op, x, ann);
1888 1889
        }

P
Patrick Walton 已提交
1890
        case (ast.expr_binary(?op, ?x, ?y, _)) {
1891
            ret trans_binary(cx, op, x, y);
1892
        }
1893

P
Patrick Walton 已提交
1894
        case (ast.expr_if(?cond, ?thn, ?els, _)) {
1895
            ret trans_if(cx, cond, thn, els);
1896 1897
        }

1898
        case (ast.expr_while(?cond, ?body, _)) {
1899
            ret trans_while(cx, cond, body);
1900 1901
        }

1902
        case (ast.expr_do_while(?body, ?cond, _)) {
1903
            ret trans_do_while(cx, body, cond);
1904 1905
        }

P
Patrick Walton 已提交
1906 1907 1908 1909
        case (ast.expr_alt(?expr, ?arms, _)) {
            ret trans_alt(cx, expr, arms);
        }

P
Patrick Walton 已提交
1910
        case (ast.expr_block(?blk, _)) {
1911
            auto sub_cx = new_scope_block_ctxt(cx, "block-expr body");
1912
            auto next_cx = new_sub_block_ctxt(cx, "next");
1913 1914 1915 1916 1917 1918 1919
            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);
        }
1920

1921
        case (ast.expr_assign(?dst, ?src, ?ann)) {
1922
            auto lhs_res = trans_lval(cx, dst);
1923
            check (lhs_res._1);
1924
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
1925
            auto t = node_ann_type(cx.fcx.ccx, ann);
G
Graydon Hoare 已提交
1926
            // FIXME: calculate copy init-ness in typestate.
1927
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, rhs_res.val, t);
1928
        }
G
Graydon Hoare 已提交
1929

1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941
        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 已提交
1942 1943
        case (ast.expr_call(?f, ?args, ?ann)) {
            ret trans_call(cx, f, args, ann);
1944 1945
        }

1946
        case (ast.expr_cast(?e, _, ?ann)) {
1947
            ret trans_cast(cx, e, ann);
1948
        }
G
Graydon Hoare 已提交
1949

G
Graydon Hoare 已提交
1950 1951 1952 1953
        case (ast.expr_vec(?args, ?ann)) {
            ret trans_vec(cx, args, ann);
        }

G
Graydon Hoare 已提交
1954 1955 1956
        case (ast.expr_tup(?args, ?ann)) {
            ret trans_tup(cx, args, ann);
        }
G
Graydon Hoare 已提交
1957

1958 1959 1960 1961
        case (ast.expr_rec(?args, ?ann)) {
            ret trans_rec(cx, args, ann);
        }

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

1965 1966 1967
        case (_) {
            auto t = typeck.expr_ty(e);
            auto sub = trans_lval(cx, e);
1968 1969
            ret res(sub._0.bcx,
                    load_non_structural(sub._0.bcx, sub._0.val, t));
1970
        }
1971
    }
1972
    cx.fcx.ccx.sess.unimpl("expr variant in trans_expr");
1973 1974 1975
    fail;
}

1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989
// 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);
    }
}

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

1992 1993 1994 1995 1996 1997 1998 1999
    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));
2000 2001
        }
        case (_) {
2002 2003 2004
            ret trans_upcall(sub.bcx,
                             "upcall_log_int",
                             vec(sub.val));
2005 2006
        }
    }
2007
    fail;
2008 2009
}

2010
impure fn trans_check_expr(@block_ctxt cx, @ast.expr e) -> result {
2011 2012 2013
    auto cond_res = trans_expr(cx, e);

    // FIXME: need pretty-printer.
2014 2015
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "<expr>"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, e.span.filename));
2016 2017 2018
    auto V_line = e.span.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));

2019
    auto fail_cx = new_sub_block_ctxt(cx, "fail");
2020 2021
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);

2022
    auto next_cx = new_sub_block_ctxt(cx, "next");
2023 2024 2025 2026 2027 2028 2029
    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());
}

2030
impure fn trans_ret(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
2031 2032 2033
    auto r = res(cx, C_nil());
    alt (e) {
        case (some[@ast.expr](?x)) {
G
Graydon Hoare 已提交
2034
            auto t = typeck.expr_ty(x);
2035
            r = trans_expr(cx, x);
2036 2037 2038 2039 2040

            // 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 已提交
2041 2042 2043 2044 2045 2046 2047
            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);
            }
2048
        }
2049
        case (_) { /* fall through */  }
2050 2051
    }

2052 2053
    // Run all cleanups and back out.
    let bool more_cleanups = true;
2054
    auto cleanup_cx = cx;
2055
    while (more_cleanups) {
2056 2057
        r.bcx = trans_block_cleanups(r.bcx, cleanup_cx);
        alt (cleanup_cx.parent) {
2058
            case (parent_some(?b)) {
2059
                cleanup_cx = b;
2060 2061 2062 2063 2064 2065 2066
            }
            case (parent_none) {
                more_cleanups = false;
            }
        }
    }

2067
    alt (e) {
2068 2069 2070 2071 2072 2073 2074
        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);
            }
2075 2076
            ret r;
        }
2077
        case (_) { /* fall through */  }
2078 2079 2080 2081
    }

    // FIXME: until LLVM has a unit type, we are moving around
    // C_nil values rather than their void type.
2082 2083
    r.bcx.build.RetVoid();
    r.val = C_nil();
2084 2085 2086
    ret r;
}

2087
impure fn trans_stmt(@block_ctxt cx, &ast.stmt s) -> result {
2088
    auto sub = res(cx, C_nil());
2089
    alt (s.node) {
2090
        case (ast.stmt_log(?a)) {
2091
            sub.bcx = trans_log(cx, a).bcx;
2092 2093
        }

2094
        case (ast.stmt_check_expr(?a)) {
2095
            sub.bcx = trans_check_expr(cx, a).bcx;
2096 2097
        }

2098 2099 2100 2101
        case (ast.stmt_ret(?e)) {
            sub.bcx = trans_ret(cx, e).bcx;
        }

2102
        case (ast.stmt_expr(?e)) {
2103
            sub.bcx = trans_expr(cx, e).bcx;
2104
        }
2105

2106 2107 2108
        case (ast.stmt_decl(?d)) {
            alt (d.node) {
                case (ast.decl_local(?local)) {
2109 2110 2111 2112 2113

                    // 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);
2114
                    find_scope_cx(cx).cleanups +=
2115 2116
                        clean(bind drop_slot(_, llptr, ty));

2117 2118
                    alt (local.init) {
                        case (some[@ast.expr](?e)) {
2119
                            sub = trans_expr(cx, e);
2120
                            sub = copy_ty(sub.bcx, true, llptr, sub.val, ty);
2121
                        }
2122 2123 2124 2125 2126
                        case (_) {
                            auto llty = type_of(cx.fcx.ccx, ty);
                            auto null = lib.llvm.llvm.LLVMConstNull(llty);
                            sub = res(cx, cx.build.Store(null, llptr));
                        }
2127 2128 2129 2130
                    }
                }
            }
        }
2131
        case (_) {
2132
            cx.fcx.ccx.sess.unimpl("stmt variant");
2133 2134
        }
    }
2135
    ret sub;
2136 2137
}

2138
fn new_builder(BasicBlockRef llbb) -> builder {
2139 2140 2141 2142 2143
    let BuilderRef llbuild = llvm.LLVMCreateBuilder();
    llvm.LLVMPositionBuilderAtEnd(llbuild, llbb);
    ret builder(llbuild);
}

2144 2145
// You probably don't want to use this one. See the
// next three functions instead.
2146
fn new_block_ctxt(@fn_ctxt cx, block_parent parent,
2147
                  bool is_scope,
2148
                  str name) -> @block_ctxt {
2149
    let vec[cleanup] cleanups = vec();
2150
    let BasicBlockRef llbb =
2151
        llvm.LLVMAppendBasicBlock(cx.llfn,
2152
                                  _str.buf(cx.ccx.names.next(name)));
2153

2154
    ret @rec(llbb=llbb,
2155
             build=new_builder(llbb),
2156
             parent=parent,
2157
             is_scope=is_scope,
2158
             mutable cleanups=cleanups,
2159 2160 2161
             fcx=cx);
}

2162 2163
// 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 {
2164 2165
    ret new_block_ctxt(fcx, parent_none, true, "function top level");
}
2166

2167 2168 2169
// 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);
2170 2171
}

2172
// Use this when you're making a general CFG BB within a scope.
2173
fn new_sub_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
2174
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), false, n);
2175
}
2176

2177

2178 2179
fn trans_block_cleanups(@block_ctxt cx,
                        @block_ctxt cleanup_cx) -> @block_ctxt {
2180
    auto bcx = cx;
2181 2182 2183 2184 2185

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

2186
    for (cleanup c in cleanup_cx.cleanups) {
2187
        alt (c) {
2188
            case (clean(?cfn)) {
2189
                bcx = cfn(bcx).bcx;
2190 2191 2192
            }
        }
    }
2193 2194 2195
    ret bcx;
}

2196 2197 2198 2199 2200 2201 2202 2203 2204 2205
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;
                    }
2206
                    case (_) { /* fall through */ }
2207 2208
                }
            }
2209
            case (_) { /* fall through */ }
2210 2211 2212 2213
        }
    }
}

2214
impure fn trans_block(@block_ctxt cx, &ast.block b) -> result {
2215 2216
    auto bcx = cx;

2217
    for each (@ast.local local in block_locals(b)) {
2218
        auto ty = node_type(cx.fcx.ccx, local.ann);
2219 2220 2221
        auto val = bcx.build.Alloca(ty);
        cx.fcx.lllocals.insert(local.id, val);
    }
2222
    auto r = res(bcx, C_nil());
2223

2224
    for (@ast.stmt s in b.node.stmts) {
2225 2226
        r = trans_stmt(bcx, *s);
        bcx = r.bcx;
2227 2228 2229 2230 2231
        // 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;
        }
2232
    }
2233

2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246
    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());
        }
    }

2247
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
2248
    ret res(bcx, r.val);
2249 2250
}

2251
fn new_fn_ctxt(@crate_ctxt cx,
2252
               str name,
2253
               ValueRef llfndecl) -> @fn_ctxt {
2254

2255
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfndecl, 0u);
2256

2257
    let hashmap[ast.def_id, ValueRef] lllocals = new_def_hash[ValueRef]();
2258 2259
    let hashmap[ast.def_id, ValueRef] llargs = new_def_hash[ValueRef]();

2260
    ret @rec(llfn=llfndecl,
2261
             lltaskptr=lltaskptr,
2262
             llargs=llargs,
2263
             lllocals=lllocals,
2264
             ccx=cx);
2265 2266
}

2267

2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278
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;
    }
}


2279 2280 2281 2282
// 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.

2283 2284
fn copy_args_to_allocas(@block_ctxt cx, vec[ast.arg] args,
                        vec[typeck.arg] arg_tys) {
2285 2286 2287

    let uint arg_n = 0u;

2288
    for (ast.arg aarg in args) {
2289 2290 2291 2292 2293 2294 2295 2296 2297
        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);
        }

2298 2299 2300 2301
        arg_n += 1u;
    }
}

2302 2303 2304 2305 2306
fn is_terminated(@block_ctxt cx) -> bool {
    auto inst = llvm.LLVMGetLastInstruction(cx.llbb);
    ret llvm.LLVMIsATerminatorInst(inst) as int != 0;
}

2307 2308 2309 2310 2311 2312 2313 2314 2315
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;
}

2316 2317 2318 2319 2320 2321 2322 2323 2324
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;
}

2325 2326
impure fn trans_fn(@crate_ctxt cx, &ast._fn f, ast.def_id fid,
                   &ast.ann ann) {
2327

2328 2329 2330
    auto llfndecl = cx.item_ids.get(fid);
    cx.item_names.insert(cx.path, llfndecl);

2331 2332 2333
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, f.inputs);

2334
    auto bcx = new_top_block_ctxt(fcx);
2335

2336
    copy_args_to_allocas(bcx, f.inputs, arg_tys_of_fn(ann));
2337

2338 2339
    auto res = trans_block(bcx, f.body);
    if (!is_terminated(res.bcx)) {
2340 2341
        // FIXME: until LLVM has a unit type, we are moving around
        // C_nil values rather than their void type.
2342
        res.bcx.build.RetVoid();
2343
    }
2344 2345
}

G
Graydon Hoare 已提交
2346 2347 2348
impure fn trans_vtbl(@crate_ctxt cx, &ast._obj ob) -> ValueRef {
    let vec[ValueRef] methods = vec();
    for (@ast.method m in ob.methods) {
2349 2350 2351 2352 2353 2354

        auto llfnty = node_type(cx, m.node.ann);
        let str s = cx.names.next("_rust_method") + "." + cx.path;
        let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfnty);
        cx.item_ids.insert(m.node.id, llfn);

G
Graydon Hoare 已提交
2355
        trans_fn(cx, m.node.meth, m.node.id, m.node.ann);
2356
        methods += llfn;
G
Graydon Hoare 已提交
2357 2358 2359 2360
    }
    ret C_struct(methods);
}

2361 2362
impure fn trans_obj(@crate_ctxt cx, &ast._obj ob, ast.def_id oid,
                    &ast.ann ann) {
2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383

    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)));
2384 2385 2386 2387 2388
    auto vtbl = trans_vtbl(cx, ob);
    auto pair_vtbl = bcx.build.GEP(pair,
                                   vec(C_int(0),
                                       C_int(abi.obj_field_vtbl)));
    bcx.build.Store(vtbl, pair_vtbl);
2389 2390 2391
    bcx.build.Ret(pair);
}

2392
fn trans_tag_variant(@crate_ctxt cx, ast.def_id tag_id,
2393
                     &ast.variant variant, int index) {
2394
    if (_vec.len[ast.variant_arg](variant.args) == 0u) {
2395 2396 2397
        ret;    // nullary constructors are just constants
    }

2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414
    // 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);

2415 2416 2417
    let ValueRef llfndecl = cx.item_ids.get(variant.id);
    cx.item_names.insert(cx.path, llfndecl);

2418 2419 2420
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, fn_args);

2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449
    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) {
2450
        auto llargval = bcx.build.Load(fcx.llargs.get(va.id));
2451 2452 2453 2454 2455 2456 2457 2458 2459 2460
        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);
2461 2462
}

2463
impure fn trans_item(@crate_ctxt cx, &ast.item item) {
2464
    alt (item.node) {
2465
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
2466
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2467
            trans_fn(sub_cx, f, fid, ann);
2468
        }
2469 2470 2471 2472
        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);
        }
2473 2474
        case (ast.item_mod(?name, ?m, _)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2475
            trans_mod(sub_cx, m);
2476
        }
2477 2478
        case (ast.item_tag(?name, ?variants, _, ?tag_id)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2479
            auto i = 0;
2480
            for (ast.variant variant in variants) {
2481 2482
                trans_tag_variant(sub_cx, tag_id, variant, i);
                i += 1;
2483 2484
            }
        }
2485
        case (_) { /* fall through */ }
2486 2487 2488
    }
}

2489
impure fn trans_mod(@crate_ctxt cx, &ast._mod m) {
2490 2491
    for (@ast.item item in m.items) {
        trans_item(cx, *item);
2492 2493 2494
    }
}

2495

2496
fn collect_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
2497
    alt (i.node) {
2498 2499
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
            // TODO: type-params
2500 2501
            cx.items.insert(fid, i);
            auto llty = node_type(cx, ann);
2502
            let str s = cx.names.next("_rust_fn") + "." + name;
2503
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
2504
            cx.item_ids.insert(fid, llfn);
2505 2506
        }

2507 2508 2509 2510 2511 2512 2513 2514 2515
        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);
        }

2516 2517 2518 2519
        case (ast.item_const(?name, _, _, ?cid, _)) {
            cx.items.insert(cid, i);
        }

2520 2521 2522
        case (ast.item_mod(?name, ?m, ?mid)) {
            cx.items.insert(mid, i);
        }
2523 2524 2525 2526

        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
            auto vi = new_def_hash[uint]();
            auto navi = new_def_hash[uint]();
2527 2528
            let vec[tup(ast.def_id,arity)] variant_info = vec();
            cx.tags.insert(tag_id, @rec(th=mk_type_handle(),
2529 2530
                                        mutable variants=variant_info,
                                        mutable size=0u));
2531
            cx.items.insert(tag_id, i);
2532 2533
        }

2534
        case (_) { /* fall through */ }
2535 2536 2537 2538 2539
    }
    ret cx;
}


2540
fn collect_items(@crate_ctxt cx, @ast.crate crate) {
2541

2542 2543
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();
2544 2545 2546 2547

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

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

2551 2552 2553 2554 2555 2556
// 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)) {
2557 2558
            auto max_align = 0u;
            auto max_size = 0u;
2559 2560 2561 2562 2563 2564

            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;
2565
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
2566 2567 2568 2569 2570 2571 2572
                    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; }
2573 2574 2575 2576 2577 2578 2579 2580 2581 2582

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

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

            info.variants = variant_info;
2583
            info.size = max_size;
2584

2585 2586 2587
            // FIXME: alignment is wrong here, manually insert padding I
            // guess :(
            auto tag_ty = T_struct(vec(T_int(), T_array(T_i8(), max_size)));
2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598
            auto th = cx.tags.get(tag_id).th.llth;
            llvm.LLVMRefineType(llvm.LLVMResolveTypeHandle(th), tag_ty);
        }
        case (_) {
            // fall through
        }
    }

    ret cx;
}

2599 2600 2601 2602 2603 2604 2605 2606 2607 2608
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);
}

2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627
// 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.
2628
                        auto union_val = C_zero_byte_arr(info.size as uint);
2629
                        auto val = C_struct(vec(C_int(i as int), union_val));
G
Graydon Hoare 已提交
2630

2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646
                        // 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;
            }
        }
2647 2648 2649 2650 2651 2652 2653 2654

        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);
        }

2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671
        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);
}

2672 2673 2674 2675
fn p2i(ValueRef v) -> ValueRef {
    ret llvm.LLVMConstPtrToInt(v, T_int());
}

2676
fn trans_exit_task_glue(@crate_ctxt cx) {
2677 2678 2679 2680 2681 2682 2683
    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,
2684
                    llargs=new_def_hash[ValueRef](),
2685
                    lllocals=new_def_hash[ValueRef](),
2686
                    ccx=cx);
2687

2688
    auto bcx = new_top_block_ctxt(fcx);
2689
    trans_upcall(bcx, "upcall_exit", V_args);
2690
    bcx.build.RetVoid();
2691 2692
}

2693 2694 2695
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());
2696
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_tydesc"), T_tydesc());
2697 2698
}

2699
fn crate_constant(@crate_ctxt cx) -> ValueRef {
2700 2701

    let ValueRef crate_ptr =
2702
        llvm.LLVMAddGlobal(cx.llmod, T_crate(),
2703 2704 2705 2706 2707 2708 2709 2710 2711 2712
                           _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);

2713 2714
    let ValueRef exit_task_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.exit_task_glue), crate_addr);
2715 2716 2717

    let ValueRef crate_val =
        C_struct(vec(C_null(T_int()),     // ptrdiff_t image_base_off
2718
                     p2i(crate_ptr),      // uintptr_t self_addr
2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736
                     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;
}

2737
fn trans_main_fn(@crate_ctxt cx, ValueRef llcrate) {
2738 2739 2740
    auto T_main_args = vec(T_int(), T_int());
    auto T_rust_start_args = vec(T_int(), T_int(), T_int(), T_int());

2741 2742 2743 2744 2745 2746 2747
    auto main_name;
    if (_str.eq(std.os.target_os(), "win32")) {
        main_name = "WinMain@16";
    } else {
        main_name = "main";
    }

2748
    auto llmain =
2749
        decl_cdecl_fn(cx.llmod, main_name, T_fn(T_main_args, T_int()));
2750

2751 2752
    auto llrust_start = decl_cdecl_fn(cx.llmod, "rust_start",
                                      T_fn(T_rust_start_args, T_int()));
2753 2754 2755

    auto llargc = llvm.LLVMGetParam(llmain, 0u);
    auto llargv = llvm.LLVMGetParam(llmain, 1u);
2756 2757
    check (cx.item_names.contains_key("_rust.main"));
    auto llrust_main = cx.item_names.get("_rust.main");
2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768

    //
    // 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(""));
2769
    auto b = new_builder(llbb);
2770 2771 2772 2773 2774 2775

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

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

2776 2777
fn declare_intrinsics(ModuleRef llmod) -> hashmap[str,ValueRef] {

2778
    let vec[TypeRef] T_trap_args = vec();
2779 2780 2781 2782
    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());
2783 2784
    auto trap = decl_cdecl_fn(llmod, "llvm.trap",
                              T_fn(T_trap_args, T_void()));
2785 2786 2787 2788
    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()));
2789 2790 2791

    auto intrinsics = new_str_hash[ValueRef]();
    intrinsics.insert("llvm.trap", trap);
2792 2793
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i32", memcpy32);
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i64", memcpy64);
2794
    ret intrinsics;
2795 2796
}

2797 2798 2799 2800 2801 2802 2803 2804
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.
}

2805 2806 2807 2808 2809
fn make_no_op_type_glue(ModuleRef llmod) -> ValueRef {
    auto ty = T_fn(vec(T_taskptr(), T_ptr(T_i8())), T_void());
    auto fun = decl_fastcall_fn(llmod, "_rust_no_op_type_glue", ty);
    auto bb_name = _str.buf("_rust_no_op_type_glue_bb");
    auto llbb = llvm.LLVMAppendBasicBlock(fun, bb_name);
2810
    new_builder(llbb).RetVoid();
2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835
    ret fun;
}

fn make_glues(ModuleRef llmod) -> @glue_fns {
    ret @rec(activate_glue = decl_glue(llmod, abi.activate_glue_name()),
             yield_glue = decl_glue(llmod, abi.yield_glue_name()),
             /*
              * 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(),
                                            T_fn(vec(T_taskptr()), T_void())),

             upcall_glues =
              _vec.init_fn[ValueRef](bind decl_upcall(llmod, _),
                                     abi.n_upcall_glues as uint),
             no_op_type_glue = make_no_op_type_glue(llmod));
}

2836
fn trans_crate(session.session sess, @ast.crate crate, str output) {
2837 2838 2839 2840
    auto llmod =
        llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
                                               llvm.LLVMGetGlobalContext());

2841 2842
    llvm.LLVMSetDataLayout(llmod, _str.buf(x86.get_data_layout()));
    llvm.LLVMSetTarget(llmod, _str.buf(x86.get_target_triple()));
2843
    auto td = mk_target_data(x86.get_data_layout());
2844

2845 2846
    llvm.LLVMSetModuleInlineAsm(llmod, _str.buf(x86.get_module_asm()));

2847
    auto intrinsics = declare_intrinsics(llmod);
2848

2849
    auto glues = make_glues(llmod);
2850 2851
    auto hasher = typeck.hash_ty;
    auto eqer = typeck.eq_ty;
2852
    auto tydescs = map.mk_hashmap[@typeck.ty,ValueRef](hasher, eqer);
2853

2854 2855
    auto cx = @rec(sess = sess,
                   llmod = llmod,
2856
                   td = td,
2857
                   upcalls = new_str_hash[ValueRef](),
2858
                   intrinsics = intrinsics,
2859 2860
                   item_names = new_str_hash[ValueRef](),
                   item_ids = new_def_hash[ValueRef](),
2861
                   items = new_def_hash[@ast.item](),
2862
                   tags = new_def_hash[@tag_info](),
2863
                   tydescs = tydescs,
2864
                   glues = glues,
2865
                   names = namegen(0),
2866
                   path = "_rust");
2867

2868 2869
    create_typedefs(cx);

2870
    collect_items(cx, crate);
2871
    resolve_tag_types(cx, crate);
2872
    trans_constants(cx, crate);
2873

2874
    trans_mod(cx, crate.node.module);
2875
    trans_exit_task_glue(cx);
2876 2877
    trans_main_fn(cx, crate_constant(cx));

2878 2879
    check_module(llmod);

2880
    llvm.LLVMWriteBitcodeToFile(llmod, _str.buf(output));
2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893
    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:
//