2011-08-22 13:48:00 -05:00
|
|
|
import syntax::ast;
|
2012-01-30 23:00:57 -06:00
|
|
|
import driver::session::session;
|
2012-02-01 04:04:56 -06:00
|
|
|
import lib::llvm::{ValueRef, TypeRef};
|
2011-08-22 13:48:00 -05:00
|
|
|
import back::abi;
|
2012-05-25 19:22:14 -05:00
|
|
|
import base::{call_memmove,
|
2012-01-19 12:21:42 -06:00
|
|
|
INIT, copy_val, load_if_immediate, get_tydesc,
|
2012-02-17 06:17:40 -06:00
|
|
|
sub_block, do_spill_noroot,
|
2012-06-14 19:34:06 -05:00
|
|
|
dest, bcx_icx, non_gc_box_cast,
|
|
|
|
heap, heap_exchange, heap_shared};
|
2012-04-09 19:32:49 -05:00
|
|
|
import syntax::codemap::span;
|
2012-03-12 04:05:15 -05:00
|
|
|
import shape::llsize_of;
|
2012-01-27 06:17:06 -06:00
|
|
|
import build::*;
|
|
|
|
import common::*;
|
2012-06-12 18:41:20 -05:00
|
|
|
import util::ppaux::ty_to_str;
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-06-14 19:34:06 -05:00
|
|
|
// Boxed vector types are in some sense currently a "shorthand" for a box
|
|
|
|
// containing an unboxed vector. This expands a boxed vector type into such an
|
|
|
|
// expanded type. It doesn't respect mutability, but that doesn't matter at
|
|
|
|
// this point.
|
|
|
|
fn expand_boxed_vec_ty(tcx: ty::ctxt, t: ty::t) -> ty::t {
|
|
|
|
let unit_ty = ty::sequence_element_type(tcx, t);
|
|
|
|
let unboxed_vec_ty = ty::mk_mut_unboxed_vec(tcx, unit_ty);
|
|
|
|
alt ty::get(t).struct {
|
|
|
|
ty::ty_vec(_) | ty::ty_str |
|
|
|
|
ty::ty_estr(ty::vstore_uniq) | ty::ty_evec(_, ty::vstore_uniq) {
|
|
|
|
ty::mk_imm_uniq(tcx, unboxed_vec_ty)
|
|
|
|
}
|
|
|
|
ty::ty_estr(ty::vstore_box) | ty::ty_evec(_, ty::vstore_box) {
|
|
|
|
ty::mk_imm_box(tcx, unboxed_vec_ty)
|
|
|
|
}
|
|
|
|
_ { tcx.sess.bug("non boxed-vec type \
|
|
|
|
in tvec::expand_boxed_vec_ty");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
fn get_fill(bcx: block, vptr: ValueRef) -> ValueRef {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::get_fill");
|
2012-05-04 00:31:38 -05:00
|
|
|
Load(bcx, GEPi(bcx, vptr, [0u, abi::vec_elt_fill]))
|
2011-08-25 03:18:02 -05:00
|
|
|
}
|
2012-03-22 10:02:34 -05:00
|
|
|
fn set_fill(bcx: block, vptr: ValueRef, fill: ValueRef) {
|
2012-05-04 00:31:38 -05:00
|
|
|
Store(bcx, fill, GEPi(bcx, vptr, [0u, abi::vec_elt_fill]));
|
2012-03-22 10:02:34 -05:00
|
|
|
}
|
|
|
|
fn get_alloc(bcx: block, vptr: ValueRef) -> ValueRef {
|
2012-05-04 00:31:38 -05:00
|
|
|
Load(bcx, GEPi(bcx, vptr, [0u, abi::vec_elt_alloc]))
|
2012-03-22 10:02:34 -05:00
|
|
|
}
|
2012-05-21 20:36:52 -05:00
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
fn get_bodyptr(bcx: block, vptr: ValueRef) -> ValueRef {
|
|
|
|
non_gc_box_cast(bcx, GEPi(bcx, vptr, [0u, abi::box_field_body]))
|
2012-05-21 20:36:52 -05:00
|
|
|
}
|
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
fn get_dataptr(bcx: block, vptr: ValueRef)
|
2011-09-27 13:21:44 -05:00
|
|
|
-> ValueRef {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::get_dataptr");
|
2012-06-12 18:41:20 -05:00
|
|
|
GEPi(bcx, vptr, [0u, abi::vec_elt_elems, 0u])
|
2011-08-25 03:18:02 -05:00
|
|
|
}
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
fn pointer_add(bcx: block, ptr: ValueRef, bytes: ValueRef) -> ValueRef {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::pointer_add");
|
2011-08-25 03:18:02 -05:00
|
|
|
let old_ty = val_ty(ptr);
|
|
|
|
let bptr = PointerCast(bcx, ptr, T_ptr(T_i8()));
|
|
|
|
ret PointerCast(bcx, InBoundsGEP(bcx, bptr, [bytes]), old_ty);
|
|
|
|
}
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-06-14 19:34:06 -05:00
|
|
|
fn alloc_raw(bcx: block, unit_ty: ty::t,
|
|
|
|
fill: ValueRef, alloc: ValueRef, heap: heap) -> result {
|
|
|
|
let _icx = bcx.insn_ctxt("tvec::alloc_uniq");
|
2012-02-21 07:20:18 -06:00
|
|
|
let ccx = bcx.ccx();
|
2012-06-12 16:54:02 -05:00
|
|
|
|
|
|
|
let vecbodyty = ty::mk_mut_unboxed_vec(bcx.tcx(), unit_ty);
|
|
|
|
let vecsize = Add(bcx, alloc, llsize_of(ccx, ccx.opaque_vec_type));
|
|
|
|
|
2012-06-14 19:34:06 -05:00
|
|
|
let {box, body} = base::malloc_general_dyn(bcx, vecbodyty, heap, vecsize);
|
2012-06-12 16:54:02 -05:00
|
|
|
Store(bcx, fill, GEPi(bcx, body, [0u, abi::vec_elt_fill]));
|
|
|
|
Store(bcx, alloc, GEPi(bcx, body, [0u, abi::vec_elt_alloc]));
|
|
|
|
ret {bcx: bcx, val: box};
|
2011-08-29 15:30:18 -05:00
|
|
|
}
|
2012-06-14 19:34:06 -05:00
|
|
|
fn alloc_uniq_raw(bcx: block, unit_ty: ty::t,
|
|
|
|
fill: ValueRef, alloc: ValueRef) -> result {
|
|
|
|
alloc_raw(bcx, unit_ty, fill, alloc, heap_exchange)
|
|
|
|
}
|
2011-08-29 15:30:18 -05:00
|
|
|
|
2012-06-14 19:34:06 -05:00
|
|
|
fn alloc_vec(bcx: block, unit_ty: ty::t, elts: uint, heap: heap) -> result {
|
2012-04-19 17:42:02 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::alloc_uniq");
|
2012-02-21 07:20:18 -06:00
|
|
|
let ccx = bcx.ccx();
|
2012-05-21 20:36:52 -05:00
|
|
|
let llunitty = type_of::type_of(ccx, unit_ty);
|
2012-03-12 04:05:15 -05:00
|
|
|
let unit_sz = llsize_of(ccx, llunitty);
|
2011-08-25 03:18:02 -05:00
|
|
|
|
2011-10-14 22:38:24 -05:00
|
|
|
let fill = Mul(bcx, C_uint(ccx, elts), unit_sz);
|
2012-03-12 04:05:15 -05:00
|
|
|
let alloc = if elts < 4u { Mul(bcx, C_int(ccx, 4), unit_sz) }
|
|
|
|
else { fill };
|
2012-06-14 19:34:06 -05:00
|
|
|
let {bcx: bcx, val: vptr} = alloc_raw(bcx, unit_ty, fill, alloc, heap);
|
2012-04-19 17:42:02 -05:00
|
|
|
ret {bcx: bcx, val: vptr};
|
2011-08-25 03:18:02 -05:00
|
|
|
}
|
2011-08-29 15:30:18 -05:00
|
|
|
|
2012-04-19 17:42:02 -05:00
|
|
|
fn duplicate_uniq(bcx: block, vptr: ValueRef, vec_ty: ty::t) -> result {
|
|
|
|
let _icx = bcx.insn_ctxt("tvec::duplicate_uniq");
|
2012-05-21 20:36:52 -05:00
|
|
|
|
2012-06-13 12:46:33 -05:00
|
|
|
let fill = get_fill(bcx, get_bodyptr(bcx, vptr));
|
2012-02-21 07:20:18 -06:00
|
|
|
let unit_ty = ty::sequence_element_type(bcx.tcx(), vec_ty);
|
2012-06-13 12:46:33 -05:00
|
|
|
let {bcx, val: newptr} = alloc_uniq_raw(bcx, unit_ty, fill, fill);
|
|
|
|
|
|
|
|
let data_ptr = get_dataptr(bcx, get_bodyptr(bcx, vptr));
|
|
|
|
let new_data_ptr = get_dataptr(bcx, get_bodyptr(bcx, newptr));
|
|
|
|
call_memmove(bcx, new_data_ptr, data_ptr, fill);
|
2012-05-21 20:36:52 -05:00
|
|
|
|
2012-03-23 08:45:47 -05:00
|
|
|
let bcx = if ty::type_needs_drop(bcx.tcx(), unit_ty) {
|
2012-06-14 19:34:06 -05:00
|
|
|
iter_vec_raw(bcx, new_data_ptr, vec_ty, fill, base::take_ty)
|
2012-03-23 08:45:47 -05:00
|
|
|
} else { bcx };
|
2011-10-10 06:32:50 -05:00
|
|
|
ret rslt(bcx, newptr);
|
2011-08-25 03:18:02 -05:00
|
|
|
}
|
2012-06-14 19:34:06 -05:00
|
|
|
|
|
|
|
fn make_drop_glue_unboxed(bcx: block, vptr: ValueRef, vec_ty: ty::t) ->
|
2012-02-17 06:17:40 -06:00
|
|
|
block {
|
2012-06-14 19:34:06 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::make_drop_glue_unboxed");
|
2012-02-21 07:20:18 -06:00
|
|
|
let tcx = bcx.tcx(), unit_ty = ty::sequence_element_type(tcx, vec_ty);
|
2012-06-14 19:34:06 -05:00
|
|
|
if ty::type_needs_drop(tcx, unit_ty) {
|
|
|
|
iter_vec_unboxed(bcx, vptr, vec_ty, base::drop_ty)
|
|
|
|
} else { bcx }
|
2011-08-22 15:30:53 -05:00
|
|
|
}
|
|
|
|
|
2012-04-19 17:42:02 -05:00
|
|
|
fn trans_evec(bcx: block, args: [@ast::expr],
|
|
|
|
vst: ast::vstore, id: ast::node_id, dest: dest) -> block {
|
|
|
|
let _icx = bcx.insn_ctxt("tvec::trans_evec");
|
2012-03-15 08:47:03 -05:00
|
|
|
let ccx = bcx.ccx();
|
|
|
|
let mut bcx = bcx;
|
2012-01-27 06:17:06 -06:00
|
|
|
if dest == base::ignore {
|
2012-03-27 08:14:12 -05:00
|
|
|
for vec::each(args) {|arg|
|
2012-01-27 06:17:06 -06:00
|
|
|
bcx = base::trans_expr(bcx, arg, base::ignore);
|
2011-09-27 03:50:18 -05:00
|
|
|
}
|
|
|
|
ret bcx;
|
|
|
|
}
|
2012-04-19 17:42:02 -05:00
|
|
|
|
2012-02-02 05:37:17 -06:00
|
|
|
let vec_ty = node_id_type(bcx, id);
|
2012-04-19 17:42:02 -05:00
|
|
|
let unit_ty = ty::sequence_element_type(bcx.tcx(), vec_ty);
|
|
|
|
let llunitty = type_of::type_of(ccx, unit_ty);
|
2012-04-19 18:37:06 -05:00
|
|
|
let unit_sz = llsize_of(ccx, llunitty);
|
2012-04-19 17:42:02 -05:00
|
|
|
|
|
|
|
let mut {bcx, val, dataptr} =
|
|
|
|
alt vst {
|
|
|
|
ast::vstore_fixed(_) {
|
|
|
|
// Destination should be pre-allocated for us.
|
|
|
|
let v = alt dest {
|
|
|
|
base::save_in(v) {
|
|
|
|
PointerCast(bcx, v, T_ptr(llunitty))
|
|
|
|
}
|
|
|
|
_ {
|
|
|
|
bcx.ccx().sess.bug("bad dest for vstore_fixed \
|
|
|
|
in tvec::trans_evec");
|
|
|
|
}
|
|
|
|
};
|
|
|
|
{bcx: bcx, val: v, dataptr: v}
|
|
|
|
}
|
|
|
|
ast::vstore_slice(_) {
|
|
|
|
let n = vec::len(args);
|
|
|
|
let n = C_uint(ccx, n);
|
|
|
|
let vp = base::arrayalloca(bcx, llunitty, n);
|
2012-04-19 18:37:06 -05:00
|
|
|
let len = Mul(bcx, n, unit_sz);
|
|
|
|
|
|
|
|
let p = base::alloca(bcx, T_struct([T_ptr(llunitty),
|
|
|
|
ccx.int_type]));
|
2012-05-04 00:31:38 -05:00
|
|
|
Store(bcx, vp, GEPi(bcx, p, [0u, abi::slice_elt_base]));
|
|
|
|
Store(bcx, len, GEPi(bcx, p, [0u, abi::slice_elt_len]));
|
2012-04-19 18:37:06 -05:00
|
|
|
|
|
|
|
{bcx: bcx, val: p, dataptr: vp}
|
2012-04-19 17:42:02 -05:00
|
|
|
}
|
|
|
|
ast::vstore_uniq {
|
2012-06-14 19:34:06 -05:00
|
|
|
let {bcx, val} = alloc_vec(bcx, unit_ty, args.len(),
|
|
|
|
heap_exchange);
|
2012-04-19 17:42:02 -05:00
|
|
|
add_clean_free(bcx, val, true);
|
2012-06-12 18:41:20 -05:00
|
|
|
let dataptr = get_dataptr(bcx, get_bodyptr(bcx, val));
|
2012-04-19 17:42:02 -05:00
|
|
|
{bcx: bcx, val: val, dataptr: dataptr}
|
|
|
|
}
|
|
|
|
ast::vstore_box {
|
2012-06-14 19:34:06 -05:00
|
|
|
let {bcx, val} = alloc_vec(bcx, unit_ty, args.len(),
|
|
|
|
heap_shared);
|
|
|
|
add_clean_free(bcx, val, true);
|
|
|
|
let dataptr = get_dataptr(bcx, get_bodyptr(bcx, val));
|
|
|
|
{bcx: bcx, val: val, dataptr: dataptr}
|
2012-04-19 17:42:02 -05:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2011-08-22 15:30:53 -05:00
|
|
|
|
|
|
|
// Store the individual elements.
|
2012-04-19 17:42:02 -05:00
|
|
|
let mut i = 0u, temp_cleanups = [val];
|
|
|
|
#debug("trans_evec: v: %s, dataptr: %s",
|
|
|
|
val_str(ccx.tn, val),
|
|
|
|
val_str(ccx.tn, dataptr));
|
2012-03-27 08:14:12 -05:00
|
|
|
for vec::each(args) {|e|
|
2012-03-12 04:05:15 -05:00
|
|
|
let lleltptr = InBoundsGEP(bcx, dataptr, [C_uint(ccx, i)]);
|
2012-01-27 06:17:06 -06:00
|
|
|
bcx = base::trans_expr_save_in(bcx, e, lleltptr);
|
2011-09-27 03:50:18 -05:00
|
|
|
add_clean_temp_mem(bcx, lleltptr, unit_ty);
|
|
|
|
temp_cleanups += [lleltptr];
|
2011-08-22 13:48:00 -05:00
|
|
|
i += 1u;
|
|
|
|
}
|
2012-04-19 17:42:02 -05:00
|
|
|
|
2012-03-27 08:14:12 -05:00
|
|
|
for vec::each(temp_cleanups) {|cln| revoke_clean(bcx, cln); }
|
2012-04-19 17:42:02 -05:00
|
|
|
|
|
|
|
alt vst {
|
|
|
|
ast::vstore_fixed(_) {
|
|
|
|
// We wrote into the destination in the fixed case.
|
|
|
|
ret bcx;
|
|
|
|
}
|
2012-04-19 18:37:06 -05:00
|
|
|
ast::vstore_slice(_) {
|
|
|
|
ret base::store_in_dest(bcx, Load(bcx, val), dest);
|
|
|
|
}
|
2012-04-19 17:42:02 -05:00
|
|
|
_ {
|
|
|
|
ret base::store_in_dest(bcx, val, dest);
|
|
|
|
}
|
|
|
|
}
|
2011-08-22 13:48:00 -05:00
|
|
|
}
|
2011-09-27 13:21:44 -05:00
|
|
|
|
2012-04-09 19:32:49 -05:00
|
|
|
fn trans_vstore(bcx: block, e: @ast::expr,
|
|
|
|
v: ast::vstore, dest: dest) -> block {
|
|
|
|
alt e.node {
|
|
|
|
ast::expr_lit(@{node: ast::lit_str(s), span: _}) {
|
2012-04-16 18:17:51 -05:00
|
|
|
ret trans_estr(bcx, s, v, dest);
|
2012-04-09 19:32:49 -05:00
|
|
|
}
|
|
|
|
ast::expr_vec(es, mutbl) {
|
2012-04-19 17:42:02 -05:00
|
|
|
ret trans_evec(bcx, es, v, e.id, dest);
|
2012-04-09 19:32:49 -05:00
|
|
|
}
|
|
|
|
_ {
|
|
|
|
bcx.sess().span_bug(e.span, "vstore on non-sequence type");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-04-17 17:07:38 -05:00
|
|
|
fn get_base_and_len(cx: block, v: ValueRef, e_ty: ty::t)
|
|
|
|
-> (ValueRef, ValueRef) {
|
|
|
|
|
2012-04-19 17:42:02 -05:00
|
|
|
let ccx = cx.ccx();
|
|
|
|
let tcx = ccx.tcx;
|
2012-04-17 17:07:38 -05:00
|
|
|
let vec_ty = ty::type_autoderef(tcx, e_ty);
|
|
|
|
let unit_ty = ty::sequence_element_type(tcx, vec_ty);
|
2012-04-19 17:42:02 -05:00
|
|
|
let llunitty = type_of::type_of(ccx, unit_ty);
|
|
|
|
let unit_sz = llsize_of(ccx, llunitty);
|
|
|
|
|
2012-04-17 17:07:38 -05:00
|
|
|
let vstore = alt ty::get(vec_ty).struct {
|
|
|
|
ty::ty_estr(vst) | ty::ty_evec(_, vst) { vst }
|
|
|
|
_ { ty::vstore_uniq }
|
|
|
|
};
|
|
|
|
|
|
|
|
alt vstore {
|
|
|
|
ty::vstore_fixed(n) {
|
2012-05-04 00:31:38 -05:00
|
|
|
let base = GEPi(cx, v, [0u, 0u]);
|
2012-04-19 17:42:02 -05:00
|
|
|
let n = if ty::type_is_str(e_ty) { n + 1u } else { n };
|
|
|
|
let len = Mul(cx, C_uint(ccx, n), unit_sz);
|
2012-04-17 17:07:38 -05:00
|
|
|
(base, len)
|
|
|
|
}
|
|
|
|
ty::vstore_slice(_) {
|
2012-05-04 00:31:38 -05:00
|
|
|
let base = Load(cx, GEPi(cx, v, [0u, abi::slice_elt_base]));
|
|
|
|
let len = Load(cx, GEPi(cx, v, [0u, abi::slice_elt_len]));
|
2012-04-17 17:07:38 -05:00
|
|
|
(base, len)
|
|
|
|
}
|
2012-06-14 19:34:06 -05:00
|
|
|
ty::vstore_uniq | ty::vstore_box {
|
|
|
|
#debug["get_base_and_len: %s", val_str(ccx.tn, v)];
|
2012-06-12 18:41:20 -05:00
|
|
|
let body = tvec::get_bodyptr(cx, v);
|
|
|
|
(tvec::get_dataptr(cx, body), tvec::get_fill(cx, body))
|
2012-04-17 17:07:38 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-06-10 02:49:59 -05:00
|
|
|
fn trans_estr(bcx: block, s: @str, vstore: ast::vstore,
|
2012-04-16 18:17:51 -05:00
|
|
|
dest: dest) -> block {
|
2012-04-09 19:32:49 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::trans_estr");
|
2012-04-10 20:34:21 -05:00
|
|
|
let ccx = bcx.ccx();
|
|
|
|
|
|
|
|
let c = alt vstore {
|
2012-04-09 19:32:49 -05:00
|
|
|
ast::vstore_fixed(_)
|
|
|
|
{
|
2012-04-18 19:02:00 -05:00
|
|
|
// "hello"/_ => "hello"/5 => [i8 x 6] in llvm
|
2012-06-10 02:49:59 -05:00
|
|
|
#debug("trans_estr: fixed: %s", *s);
|
|
|
|
C_postr(*s)
|
2012-04-10 20:34:21 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
ast::vstore_slice(_) {
|
2012-04-18 19:02:00 -05:00
|
|
|
// "hello" => (*i8, 6u) in llvm
|
2012-06-10 02:49:59 -05:00
|
|
|
#debug("trans_estr: slice '%s'", *s);
|
|
|
|
C_estr_slice(ccx, *s)
|
2012-04-09 19:32:49 -05:00
|
|
|
}
|
2012-04-10 20:34:21 -05:00
|
|
|
|
2012-04-16 18:17:51 -05:00
|
|
|
ast::vstore_uniq {
|
2012-06-10 02:49:59 -05:00
|
|
|
let cs = PointerCast(bcx, C_cstr(ccx, *s), T_ptr(T_i8()));
|
|
|
|
let len = C_uint(ccx, str::len(*s));
|
2012-05-21 20:36:52 -05:00
|
|
|
let c = Call(bcx, ccx.upcalls.str_new_uniq, [cs, len]);
|
|
|
|
PointerCast(bcx, c,
|
|
|
|
T_unique_ptr(T_unique(ccx, T_vec(ccx, T_i8()))))
|
2012-04-16 18:17:51 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
ast::vstore_box {
|
2012-06-10 02:49:59 -05:00
|
|
|
let cs = PointerCast(bcx, C_cstr(ccx, *s), T_ptr(T_i8()));
|
|
|
|
let len = C_uint(ccx, str::len(*s));
|
2012-06-18 14:50:09 -05:00
|
|
|
let c = Call(bcx, ccx.upcalls.str_new_shared, [cs, len]);
|
|
|
|
PointerCast(bcx, c,
|
|
|
|
T_box_ptr(T_box(ccx, T_vec(ccx, T_i8()))))
|
2012-04-09 19:32:49 -05:00
|
|
|
}
|
2012-04-10 20:34:21 -05:00
|
|
|
};
|
|
|
|
|
|
|
|
#debug("trans_estr: type: %s", val_str(ccx.tn, c));
|
|
|
|
base::store_in_dest(bcx, c, dest)
|
2012-04-09 19:32:49 -05:00
|
|
|
}
|
|
|
|
|
2012-03-12 04:05:15 -05:00
|
|
|
fn trans_append(bcx: block, vec_ty: ty::t, lhsptr: ValueRef,
|
2012-02-17 06:17:40 -06:00
|
|
|
rhs: ValueRef) -> block {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::trans_append");
|
2011-08-22 13:48:00 -05:00
|
|
|
// Cast to opaque interior vector types if necessary.
|
2012-03-12 04:05:15 -05:00
|
|
|
let ccx = bcx.ccx();
|
|
|
|
let unit_ty = ty::sequence_element_type(ccx.tcx, vec_ty);
|
2012-04-19 17:42:02 -05:00
|
|
|
let strings = ty::type_is_str(vec_ty);
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2011-08-25 03:18:02 -05:00
|
|
|
let lhs = Load(bcx, lhsptr);
|
2012-02-01 04:04:56 -06:00
|
|
|
let self_append = ICmp(bcx, lib::llvm::IntEQ, lhs, rhs);
|
2012-06-12 18:41:20 -05:00
|
|
|
let lfill = get_fill(bcx, get_bodyptr(bcx, lhs));
|
|
|
|
let rfill = get_fill(bcx, get_bodyptr(bcx, rhs));
|
2012-03-15 08:47:03 -05:00
|
|
|
let mut new_fill = Add(bcx, lfill, rfill);
|
2011-10-14 22:38:24 -05:00
|
|
|
if strings { new_fill = Sub(bcx, new_fill, C_int(ccx, 1)); }
|
|
|
|
let opaque_lhs = PointerCast(bcx, lhsptr,
|
2012-05-21 20:36:52 -05:00
|
|
|
T_ptr(T_ptr(T_i8())));
|
2012-03-12 04:05:15 -05:00
|
|
|
Call(bcx, ccx.upcalls.vec_grow,
|
2011-10-20 04:42:40 -05:00
|
|
|
[opaque_lhs, new_fill]);
|
2011-08-25 03:18:02 -05:00
|
|
|
// Was overwritten if we resized
|
2011-10-10 06:32:50 -05:00
|
|
|
let lhs = Load(bcx, lhsptr);
|
2012-03-12 04:05:15 -05:00
|
|
|
let rhs = Select(bcx, self_append, lhs, rhs);
|
2011-08-25 03:18:02 -05:00
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
let lbody = get_bodyptr(bcx, lhs);
|
2012-05-21 20:36:52 -05:00
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
let lhs_data = get_dataptr(bcx, lbody);
|
2012-03-15 08:47:03 -05:00
|
|
|
let mut lhs_off = lfill;
|
2011-10-14 22:38:24 -05:00
|
|
|
if strings { lhs_off = Sub(bcx, lhs_off, C_int(ccx, 1)); }
|
2011-08-25 03:18:02 -05:00
|
|
|
let write_ptr = pointer_add(bcx, lhs_data, lhs_off);
|
2011-09-02 17:12:27 -05:00
|
|
|
let write_ptr_ptr = do_spill_noroot(bcx, write_ptr);
|
2012-04-19 17:42:02 -05:00
|
|
|
iter_vec_uniq(bcx, rhs, vec_ty, rfill, {|bcx, addr, _ty|
|
2012-03-12 04:05:15 -05:00
|
|
|
let write_ptr = Load(bcx, write_ptr_ptr);
|
|
|
|
let bcx = copy_val(bcx, INIT, write_ptr,
|
|
|
|
load_if_immediate(bcx, addr, unit_ty), unit_ty);
|
|
|
|
Store(bcx, InBoundsGEP(bcx, write_ptr, [C_int(ccx, 1)]),
|
|
|
|
write_ptr_ptr);
|
2012-03-26 05:39:20 -05:00
|
|
|
bcx
|
2012-03-12 04:05:15 -05:00
|
|
|
})
|
2011-08-22 13:48:00 -05:00
|
|
|
}
|
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
fn trans_append_literal(bcx: block, vptrptr: ValueRef, vec_ty: ty::t,
|
|
|
|
vals: [@ast::expr]) -> block {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::trans_append_literal");
|
2012-03-21 09:42:20 -05:00
|
|
|
let mut bcx = bcx, ccx = bcx.ccx();
|
2012-02-21 07:20:18 -06:00
|
|
|
let elt_ty = ty::sequence_element_type(bcx.tcx(), vec_ty);
|
2012-03-22 10:02:34 -05:00
|
|
|
let elt_llty = type_of::type_of(ccx, elt_ty);
|
|
|
|
let elt_sz = shape::llsize_of(ccx, elt_llty);
|
|
|
|
let scratch = base::alloca(bcx, elt_llty);
|
2012-03-27 08:14:12 -05:00
|
|
|
for vec::each(vals) {|val|
|
2012-03-22 10:02:34 -05:00
|
|
|
bcx = base::trans_expr_save_in(bcx, val, scratch);
|
2012-06-12 18:41:20 -05:00
|
|
|
let vptr = get_bodyptr(bcx, Load(bcx, vptrptr));
|
2012-03-22 10:02:34 -05:00
|
|
|
let old_fill = get_fill(bcx, vptr);
|
|
|
|
let new_fill = Add(bcx, old_fill, elt_sz);
|
|
|
|
let do_grow = ICmp(bcx, lib::llvm::IntUGT, new_fill,
|
|
|
|
get_alloc(bcx, vptr));
|
|
|
|
bcx = base::with_cond(bcx, do_grow) {|bcx|
|
|
|
|
let pt = PointerCast(bcx, vptrptr,
|
2012-05-21 20:36:52 -05:00
|
|
|
T_ptr(T_ptr(T_i8())));
|
2012-03-22 10:02:34 -05:00
|
|
|
Call(bcx, ccx.upcalls.vec_grow, [pt, new_fill]);
|
|
|
|
bcx
|
|
|
|
};
|
2012-06-12 18:41:20 -05:00
|
|
|
let vptr = get_bodyptr(bcx, Load(bcx, vptrptr));
|
2012-03-22 10:02:34 -05:00
|
|
|
set_fill(bcx, vptr, new_fill);
|
2012-06-12 18:41:20 -05:00
|
|
|
let targetptr = pointer_add(bcx, get_dataptr(bcx, vptr),
|
2012-03-22 10:02:34 -05:00
|
|
|
old_fill);
|
2012-03-23 08:45:47 -05:00
|
|
|
call_memmove(bcx, targetptr, scratch, elt_sz);
|
2011-08-24 06:53:34 -05:00
|
|
|
}
|
2012-03-22 10:02:34 -05:00
|
|
|
bcx
|
2011-08-24 06:53:34 -05:00
|
|
|
}
|
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
fn trans_add(bcx: block, vec_ty: ty::t, lhs: ValueRef,
|
|
|
|
rhs: ValueRef, dest: dest) -> block {
|
2012-03-22 15:44:20 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::trans_add");
|
2012-02-21 07:20:18 -06:00
|
|
|
let ccx = bcx.ccx();
|
2012-03-19 16:06:59 -05:00
|
|
|
|
2012-05-21 20:36:52 -05:00
|
|
|
let unit_ty = ty::sequence_element_type(bcx.tcx(), vec_ty);
|
|
|
|
let llunitty = type_of::type_of(ccx, unit_ty);
|
|
|
|
|
2012-03-19 16:06:59 -05:00
|
|
|
if ty::get(vec_ty).struct == ty::ty_str {
|
2012-05-21 20:36:52 -05:00
|
|
|
let lhs = PointerCast(bcx, lhs, T_ptr(T_i8()));
|
|
|
|
let rhs = PointerCast(bcx, rhs, T_ptr(T_i8()));
|
2012-03-19 16:06:59 -05:00
|
|
|
let n = Call(bcx, ccx.upcalls.str_concat, [lhs, rhs]);
|
2012-05-21 20:36:52 -05:00
|
|
|
let n = PointerCast(
|
|
|
|
bcx, n, T_unique_ptr(T_unique(ccx, T_vec(ccx, llunitty))));
|
2012-03-19 16:06:59 -05:00
|
|
|
ret base::store_in_dest(bcx, n, dest);
|
|
|
|
}
|
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
let lhs_fill = get_fill(bcx, get_bodyptr(bcx, lhs));
|
|
|
|
let rhs_fill = get_fill(bcx, get_bodyptr(bcx, rhs));
|
2011-08-25 03:18:02 -05:00
|
|
|
let new_fill = Add(bcx, lhs_fill, rhs_fill);
|
2012-04-19 17:42:02 -05:00
|
|
|
let mut {bcx: bcx, val: new_vec_ptr} =
|
2012-05-21 20:36:52 -05:00
|
|
|
alloc_uniq_raw(bcx, unit_ty, new_fill, new_fill);
|
2011-08-25 03:18:02 -05:00
|
|
|
|
2012-06-12 18:41:20 -05:00
|
|
|
let new_vec_body_ptr = get_bodyptr(bcx, new_vec_ptr);
|
2011-09-27 13:21:44 -05:00
|
|
|
let write_ptr_ptr = do_spill_noroot
|
2012-06-12 18:41:20 -05:00
|
|
|
(bcx, get_dataptr(bcx, new_vec_body_ptr));
|
2012-02-17 06:17:40 -06:00
|
|
|
let copy_fn = fn@(bcx: block, addr: ValueRef,
|
|
|
|
_ty: ty::t) -> block {
|
2012-02-21 07:20:18 -06:00
|
|
|
let ccx = bcx.ccx();
|
2012-01-11 11:58:05 -06:00
|
|
|
let write_ptr = Load(bcx, write_ptr_ptr);
|
|
|
|
let bcx = copy_val(bcx, INIT, write_ptr,
|
|
|
|
load_if_immediate(bcx, addr, unit_ty), unit_ty);
|
2012-03-12 04:05:15 -05:00
|
|
|
Store(bcx, InBoundsGEP(bcx, write_ptr, [C_int(ccx, 1)]),
|
2012-01-11 11:58:05 -06:00
|
|
|
write_ptr_ptr);
|
|
|
|
ret bcx;
|
|
|
|
};
|
2011-08-22 18:39:18 -05:00
|
|
|
|
2012-04-19 17:42:02 -05:00
|
|
|
let bcx = iter_vec_uniq(bcx, lhs, vec_ty, lhs_fill, copy_fn);
|
|
|
|
let bcx = iter_vec_uniq(bcx, rhs, vec_ty, rhs_fill, copy_fn);
|
2012-01-27 06:17:06 -06:00
|
|
|
ret base::store_in_dest(bcx, new_vec_ptr, dest);
|
2011-08-29 09:14:24 -05:00
|
|
|
}
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
type val_and_ty_fn = fn@(block, ValueRef, ty::t) -> result;
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-02-17 06:17:40 -06:00
|
|
|
type iter_vec_block = fn(block, ValueRef, ty::t) -> block;
|
2011-08-22 13:48:00 -05:00
|
|
|
|
2012-04-19 17:42:02 -05:00
|
|
|
fn iter_vec_raw(bcx: block, data_ptr: ValueRef, vec_ty: ty::t,
|
2012-02-17 06:17:40 -06:00
|
|
|
fill: ValueRef, f: iter_vec_block) -> block {
|
2012-06-14 19:34:06 -05:00
|
|
|
let _icx = bcx.insn_ctxt("tvec::iter_vec_raw");
|
2012-04-19 17:42:02 -05:00
|
|
|
|
2012-02-21 07:20:18 -06:00
|
|
|
let unit_ty = ty::sequence_element_type(bcx.tcx(), vec_ty);
|
2011-08-25 03:18:02 -05:00
|
|
|
|
|
|
|
// Calculate the last pointer address we want to handle.
|
2012-02-21 07:20:18 -06:00
|
|
|
// FIXME: Optimize this when the size of the unit type is statically
|
2012-06-07 15:49:01 -05:00
|
|
|
// known to not use pointer casts, which tend to confuse LLVM. (#2536)
|
2011-08-25 03:18:02 -05:00
|
|
|
let data_end_ptr = pointer_add(bcx, data_ptr, fill);
|
|
|
|
|
|
|
|
// Now perform the iteration.
|
2012-02-17 06:17:40 -06:00
|
|
|
let header_cx = sub_block(bcx, "iter_vec_loop_header");
|
2011-08-25 03:18:02 -05:00
|
|
|
Br(bcx, header_cx.llbb);
|
2011-09-09 07:20:15 -05:00
|
|
|
let data_ptr = Phi(header_cx, val_ty(data_ptr), [data_ptr], [bcx.llbb]);
|
2011-09-02 17:34:58 -05:00
|
|
|
let not_yet_at_end =
|
2012-02-01 04:04:56 -06:00
|
|
|
ICmp(header_cx, lib::llvm::IntULT, data_ptr, data_end_ptr);
|
2012-02-17 06:17:40 -06:00
|
|
|
let body_cx = sub_block(header_cx, "iter_vec_loop_body");
|
|
|
|
let next_cx = sub_block(header_cx, "iter_vec_next");
|
2011-08-25 03:18:02 -05:00
|
|
|
CondBr(header_cx, not_yet_at_end, body_cx.llbb, next_cx.llbb);
|
2012-03-15 08:47:03 -05:00
|
|
|
let body_cx = f(body_cx, data_ptr, unit_ty);
|
2012-03-12 04:05:15 -05:00
|
|
|
AddIncomingToPhi(data_ptr, InBoundsGEP(body_cx, data_ptr,
|
2012-04-19 17:42:02 -05:00
|
|
|
[C_int(bcx.ccx(), 1)]),
|
|
|
|
body_cx.llbb);
|
2011-08-25 03:18:02 -05:00
|
|
|
Br(body_cx, header_cx.llbb);
|
2011-08-30 06:50:58 -05:00
|
|
|
ret next_cx;
|
2012-04-19 17:42:02 -05:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
fn iter_vec_uniq(bcx: block, vptr: ValueRef, vec_ty: ty::t,
|
|
|
|
fill: ValueRef, f: iter_vec_block) -> block {
|
|
|
|
let _icx = bcx.insn_ctxt("tvec::iter_vec_uniq");
|
2012-06-12 18:41:20 -05:00
|
|
|
let data_ptr = get_dataptr(bcx, get_bodyptr(bcx, vptr));
|
2012-04-19 17:42:02 -05:00
|
|
|
iter_vec_raw(bcx, data_ptr, vec_ty, fill, f)
|
2011-08-22 13:48:00 -05:00
|
|
|
}
|
2011-08-25 03:18:02 -05:00
|
|
|
|
2012-06-14 19:34:06 -05:00
|
|
|
fn iter_vec_unboxed(bcx: block, body_ptr: ValueRef, vec_ty: ty::t,
|
|
|
|
f: iter_vec_block) -> block {
|
|
|
|
let _icx = bcx.insn_ctxt("tvec::iter_vec_unboxed");
|
|
|
|
let fill = get_fill(bcx, body_ptr);
|
|
|
|
let dataptr = get_dataptr(bcx, body_ptr);
|
|
|
|
ret iter_vec_raw(bcx, dataptr, vec_ty, fill, f);
|
2011-08-29 09:14:24 -05:00
|
|
|
}
|
|
|
|
|
2011-08-24 06:53:34 -05:00
|
|
|
//
|
|
|
|
// Local Variables:
|
|
|
|
// mode: rust
|
|
|
|
// fill-column: 78;
|
|
|
|
// indent-tabs-mode: nil
|
|
|
|
// c-basic-offset: 4
|
|
|
|
// buffer-file-coding-system: utf-8-unix
|
|
|
|
// End:
|
|
|
|
//
|