rust/src/librustc/middle/trans/common.rs

1459 lines
43 KiB
Rust
Raw Normal View History

// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
/**
Code that is useful in various trans modules.
*/
use core::prelude::*;
use back::{link, abi, upcall};
use driver::session;
use driver::session::Session;
2012-09-04 13:54:36 -05:00
use lib::llvm::{ModuleRef, ValueRef, TypeRef, BasicBlockRef, BuilderRef};
use lib::llvm::{True, False, Bool};
use lib::llvm::{llvm, TargetData, TypeNames, associate_type, name_has_type};
use lib;
use metadata::common::LinkMeta;
use middle::astencode;
use middle::resolve;
use middle::trans::base;
use middle::trans::build;
use middle::trans::callee;
use middle::trans::datum;
use middle::trans::debuginfo;
use middle::trans::expr;
use middle::trans::glue;
use middle::trans::reachable;
use middle::trans::shape;
use middle::trans::type_of;
use middle::trans::type_use;
use middle::ty::substs;
use middle::ty;
use middle::typeck;
use util::ppaux::{expr_repr, ty_to_str};
use core::cast;
use core::hash;
use core::libc::c_uint;
use core::ptr;
use core::str;
use core::to_bytes;
use core::vec::raw::to_ptr;
use core::vec;
use std::oldmap::{HashMap, Set};
use syntax::ast::ident;
2012-09-04 13:54:36 -05:00
use syntax::ast_map::path;
use syntax::codemap::span;
2012-09-04 13:54:36 -05:00
use syntax::parse::token::ident_interner;
use syntax::{ast, ast_map};
2013-03-05 19:36:39 -06:00
pub type namegen = @fn(+s: ~str) -> ident;
pub fn new_namegen(intr: @ident_interner) -> namegen {
2013-03-05 19:36:39 -06:00
let f: @fn(+s: ~str) -> ident = |prefix| {
intr.gensym(@fmt!("%s_%u",
prefix,
2013-03-05 19:36:39 -06:00
intr.gensym(@prefix).repr))
2012-07-18 18:18:02 -05:00
};
f
}
pub type addrspace = c_uint;
2012-08-06 12:53:38 -05:00
// Address spaces communicate to LLVM which destructors need to run for
// specific types.
2012-08-06 12:53:38 -05:00
// 0 is ignored by the GC, and is used for all non-GC'd pointers.
// 1 is for opaque GC'd boxes.
// >= 2 are for specific types (e.g. resources).
pub const default_addrspace: addrspace = 0;
pub const gc_box_addrspace: addrspace = 1;
2012-08-06 12:53:38 -05:00
pub type addrspace_gen = @fn() -> addrspace;
pub fn new_addrspace_gen() -> addrspace_gen {
2012-08-06 12:53:38 -05:00
let i = @mut 1;
let result: addrspace_gen = || { *i += 1; *i };
result
2012-08-06 12:53:38 -05:00
}
pub struct tydesc_info {
ty: ty::t,
tydesc: ValueRef,
size: ValueRef,
align: ValueRef,
addrspace: addrspace,
take_glue: Option<ValueRef>,
drop_glue: Option<ValueRef>,
free_glue: Option<ValueRef>,
visit_glue: Option<ValueRef>
}
/*
* A note on nomenclature of linking: "extern", "foreign", and "upcall".
*
* An "extern" is an LLVM symbol we wind up emitting an undefined external
* reference to. This means "we don't have the thing in this compilation unit,
* please make sure you link it in at runtime". This could be a reference to
* C code found in a C library, or rust code found in a rust crate.
*
* Most "externs" are implicitly declared (automatically) as a result of a
* user declaring an extern _module_ dependency; this causes the rust driver
* to locate an extern crate, scan its compilation metadata, and emit extern
* declarations for any symbols used by the declaring crate.
*
* A "foreign" is an extern that references C (or other non-rust ABI) code.
* There is no metadata to scan for extern references so in these cases either
* a header-digester like bindgen, or manual function prototypes, have to
* serve as declarators. So these are usually given explicitly as prototype
* declarations, in rust code, with ABI attributes on them noting which ABI to
* link via.
*
* An "upcall" is a foreign call generated by the compiler (not corresponding
* to any user-written call in the code) into the runtime library, to perform
* some helper task such as bringing a task to life, allocating memory, etc.
*
*/
pub struct Stats {
n_static_tydescs: uint,
n_glues_created: uint,
n_null_glues: uint,
n_real_glues: uint,
n_fns: uint,
n_monos: uint,
n_inlines: uint,
n_closures: uint,
llvm_insn_ctxt: @mut ~[~str],
llvm_insns: HashMap<~str, uint>,
fn_times: @mut ~[(~str, int)] // (ident, time)
}
pub struct BuilderRef_res {
2012-09-06 21:40:15 -05:00
B: BuilderRef,
}
impl Drop for BuilderRef_res {
fn finalize(&self) {
unsafe {
llvm::LLVMDisposeBuilder(self.B);
}
}
}
pub fn BuilderRef_res(B: BuilderRef) -> BuilderRef_res {
2012-09-05 17:58:43 -05:00
BuilderRef_res {
B: B
}
}
pub type ExternMap = HashMap<@str, ValueRef>;
2013-02-04 17:30:32 -06:00
// Crate context. Every crate we compile has one of these.
pub struct CrateContext {
sess: session::Session,
llmod: ModuleRef,
td: TargetData,
tn: @TypeNames,
2013-02-04 17:30:32 -06:00
externs: ExternMap,
2012-09-10 17:38:28 -05:00
intrinsics: HashMap<~str, ValueRef>,
item_vals: HashMap<ast::node_id, ValueRef>,
2012-08-29 15:26:26 -05:00
exp_map2: resolve::ExportMap2,
reachable: reachable::map,
2012-09-10 17:38:28 -05:00
item_symbols: HashMap<ast::node_id, ~str>,
link_meta: LinkMeta,
2012-09-10 17:38:28 -05:00
enum_sizes: HashMap<ty::t, uint>,
discrims: HashMap<ast::def_id, ValueRef>,
discrim_symbols: HashMap<ast::node_id, ~str>,
tydescs: HashMap<ty::t, @mut tydesc_info>,
// Set when running emit_tydescs to enforce that no more tydescs are
// created.
2013-02-21 17:30:24 -06:00
finished_tydescs: @mut bool,
// Track mapping of external ids to local items imported for inlining
2012-09-10 17:38:28 -05:00
external: HashMap<ast::def_id, Option<ast::node_id>>,
// Cache instances of monomorphized functions
2012-09-10 17:38:28 -05:00
monomorphized: HashMap<mono_id, ValueRef>,
monomorphizing: HashMap<ast::def_id, uint>,
// Cache computed type parameter uses (see type_use.rs)
2012-09-10 17:38:28 -05:00
type_use_cache: HashMap<ast::def_id, ~[type_use::type_uses]>,
// Cache generated vtables
2012-09-10 17:38:28 -05:00
vtables: HashMap<mono_id, ValueRef>,
// Cache of constant strings,
const_cstr_cache: HashMap<@~str, ValueRef>,
// Reverse-direction for const ptrs cast from globals.
// Key is an int, cast from a ValueRef holding a *T,
// Val is a ValueRef holding a *[T].
//
// Needed because LLVM loses pointer->pointee association
// when we ptrcast, and we have to ptrcast during translation
// of a [T] const because we form a slice, a [*T,int] pair, not
// a pointer to an LLVM array type.
2012-09-10 17:38:28 -05:00
const_globals: HashMap<int, ValueRef>,
// Cache of emitted const values
const_values: HashMap<ast::node_id, ValueRef>,
2012-09-10 17:38:28 -05:00
module_data: HashMap<~str, ValueRef>,
lltypes: HashMap<ty::t, TypeRef>,
llsizingtypes: HashMap<ty::t, TypeRef>,
names: namegen,
2012-08-06 12:53:38 -05:00
next_addrspace: addrspace_gen,
symbol_hasher: @hash::State,
type_hashcodes: HashMap<ty::t, @str>,
2012-09-10 17:38:28 -05:00
type_short_names: HashMap<ty::t, ~str>,
all_llvm_symbols: Set<~str>,
tcx: ty::ctxt,
maps: astencode::Maps,
stats: @mut Stats,
upcalls: @upcall::Upcalls,
tydesc_type: TypeRef,
int_type: TypeRef,
float_type: TypeRef,
task_type: TypeRef,
opaque_vec_type: TypeRef,
builder: BuilderRef_res,
shape_cx: shape::Ctxt,
crate_map: ValueRef,
// Set when at least one function uses GC. Needed so that
// decl_gc_metadata knows whether to link to the module metadata, which
// is not emitted by LLVM's GC pass when no functions use GC.
2013-02-21 17:30:24 -06:00
uses_gc: @mut bool,
dbg_cx: Option<debuginfo::DebugContext>,
2013-02-21 17:30:24 -06:00
do_not_commit_warning_issued: @mut bool
}
// Types used for llself.
pub struct ValSelfData {
v: ValueRef,
t: ty::t,
is_owned: bool
}
pub enum local_val { local_mem(ValueRef), local_imm(ValueRef), }
// Here `self_ty` is the real type of the self parameter to this method. It
// will only be set in the case of default methods.
pub struct param_substs {
tys: ~[ty::t],
vtables: Option<typeck::vtable_res>,
bounds: @~[ty::param_bounds],
self_ty: Option<ty::t>
}
pub fn param_substs_to_str(tcx: ty::ctxt, substs: &param_substs) -> ~str {
fmt!("param_substs {tys:%?, vtables:%?, bounds:%?}",
substs.tys.map(|t| ty_to_str(tcx, *t)),
substs.vtables.map(|vs| vs.map(|v| v.to_str(tcx))),
substs.bounds.map(|b| ty::param_bounds_to_str(tcx, *b)))
}
// Function context. Every LLVM function we create will have one of
// these.
pub struct fn_ctxt_ {
2011-07-27 07:19:39 -05:00
// The ValueRef returned from a call to llvm::LLVMAddFunction; the
// address of the first instruction in the sequence of
// instructions for this function that will go in the .text
// section of the executable we're generating.
llfn: ValueRef,
2011-07-27 07:19:39 -05:00
// The two implicit arguments that arrive in the function we're creating.
// For instance, foo(int, int) is really foo(ret*, env*, int, int).
llenv: ValueRef,
llretptr: ValueRef,
2011-07-27 07:19:39 -05:00
// These elements: "hoisted basic blocks" containing
2011-07-27 07:19:39 -05:00
// administrative activities that have to happen in only one place in
// the function, due to LLVM's quirks.
// A block for all the function's static allocas, so that LLVM
// will coalesce them into a single alloca call.
2013-02-21 17:30:24 -06:00
llstaticallocas: BasicBlockRef,
2011-07-27 07:19:39 -05:00
// A block containing code that copies incoming arguments to space
// already allocated by code in one of the llallocas blocks.
// (LLVM requires that arguments be copied to local allocas before
// allowing most any operation to be performed on them.)
2013-02-21 17:30:24 -06:00
llloadenv: Option<BasicBlockRef>,
llreturn: BasicBlockRef,
// The 'self' value currently in use in this function, if there
// is one.
//
// NB: This is the type of the self *variable*, not the self *type*. The
// self type is set only for default methods, while the self variable is
// set for all methods.
2013-02-21 17:30:24 -06:00
llself: Option<ValSelfData>,
// The a value alloca'd for calls to upcalls.rust_personality. Used when
// outputting the resume instruction.
2013-02-21 17:30:24 -06:00
personality: Option<ValueRef>,
// If this is a for-loop body that returns, this holds the pointers needed
// for that (flagptr, retptr)
2013-02-21 17:30:24 -06:00
loop_ret: Option<(ValueRef, ValueRef)>,
2011-07-27 07:19:39 -05:00
// Maps arguments to allocas created for them in llallocas.
2013-02-21 17:30:24 -06:00
llargs: @HashMap<ast::node_id, local_val>,
2011-07-27 07:19:39 -05:00
// Maps the def_ids for local variables to the allocas created for
// them in llallocas.
2013-02-21 17:30:24 -06:00
lllocals: @HashMap<ast::node_id, local_val>,
// Same as above, but for closure upvars
2013-02-21 17:30:24 -06:00
llupvars: @HashMap<ast::node_id, ValueRef>,
2011-07-27 07:19:39 -05:00
// The node_id of the function, or -1 if it doesn't correspond to
// a user-defined function.
id: ast::node_id,
// The def_id of the impl we're inside, or None if we aren't inside one.
impl_id: Option<ast::def_id>,
// If this function is being monomorphized, this contains the type
// substitutions used.
param_substs: Option<@param_substs>,
// The source span and nesting context where this function comes from, for
// error reporting and symbol generation.
2012-08-20 14:23:37 -05:00
span: Option<span>,
path: path,
// This function's enclosing crate context.
2013-02-21 17:30:24 -06:00
ccx: @@CrateContext
}
2013-02-21 17:30:24 -06:00
pub type fn_ctxt = @mut fn_ctxt_;
2013-03-05 19:36:39 -06:00
pub fn warn_not_to_commit(ccx: @CrateContext, msg: &str) {
2013-02-21 17:30:24 -06:00
if !*ccx.do_not_commit_warning_issued {
*ccx.do_not_commit_warning_issued = true;
2013-03-05 19:36:39 -06:00
ccx.sess.warn(msg.to_str() + ~" -- do not commit like this!");
}
}
// Heap selectors. Indicate which heap something should go on.
#[deriving_eq]
pub enum heap {
heap_managed,
heap_managed_unique,
heap_exchange,
}
#[deriving_eq]
pub enum cleantype {
normal_exit_only,
normal_exit_and_unwind
}
pub enum cleanup {
clean(@fn(block) -> block, cleantype),
clean_temp(ValueRef, @fn(block) -> block, cleantype),
}
// Used to remember and reuse existing cleanup paths
// target: none means the path ends in an resume instruction
pub struct cleanup_path {
target: Option<BasicBlockRef>,
dest: BasicBlockRef
}
2013-03-05 19:36:39 -06:00
pub fn scope_clean_changed(+scope_info: &mut scope_info) {
if scope_info.cleanup_paths.len() > 0u { scope_info.cleanup_paths = ~[]; }
scope_info.landing_pad = None;
}
pub fn cleanup_type(cx: ty::ctxt, ty: ty::t) -> cleantype {
if ty::type_needs_unwind_cleanup(cx, ty) {
normal_exit_and_unwind
} else {
normal_exit_only
}
}
// This is not the same as datum::Datum::root(), which is used to keep copies
// of @ values live for as long as a borrowed pointer to the interior exists.
// In the new GC, we can identify immediates on the stack without difficulty,
// but have trouble knowing where non-immediates are on the stack. For
// non-immediates, we must add an additional level of indirection, which
// allows us to alloca a pointer with the right addrspace.
pub fn root_for_cleanup(bcx: block, v: ValueRef, t: ty::t)
-> (ValueRef, bool) {
2012-08-06 12:53:38 -05:00
let ccx = bcx.ccx();
let addrspace = base::get_tydesc(ccx, t).addrspace;
if addrspace > gc_box_addrspace {
let llty = type_of::type_of_rooted(ccx, t);
let root = base::alloca(bcx, llty);
build::Store(bcx, build::PointerCast(bcx, v, llty), root);
(root, true)
2012-08-06 12:53:38 -05:00
} else {
(v, false)
2012-08-06 12:53:38 -05:00
}
}
pub fn add_clean(bcx: block, val: ValueRef, t: ty::t) {
2012-08-06 12:53:38 -05:00
if !ty::type_needs_drop(bcx.tcx(), t) { return; }
2012-08-22 19:24:52 -05:00
debug!("add_clean(%s, %s, %s)",
2012-08-06 12:53:38 -05:00
bcx.to_str(), val_str(bcx.ccx().tn, val),
ty_to_str(bcx.ccx().tcx, t));
let (root, rooted) = root_for_cleanup(bcx, val, t);
2012-08-06 12:53:38 -05:00
let cleanup_type = cleanup_type(bcx.tcx(), t);
do in_scope_cx(bcx) |scope_info| {
scope_info.cleanups.push(
clean(|a| glue::drop_ty_root(a, root, rooted, t),
cleanup_type));
scope_clean_changed(scope_info);
2012-02-17 04:18:14 -06:00
}
}
pub fn add_clean_temp_immediate(cx: block, val: ValueRef, ty: ty::t) {
2012-08-01 19:30:05 -05:00
if !ty::type_needs_drop(cx.tcx(), ty) { return; }
2012-08-06 12:53:38 -05:00
debug!("add_clean_temp_immediate(%s, %s, %s)",
cx.to_str(), val_str(cx.ccx().tn, val),
2012-08-22 19:24:52 -05:00
ty_to_str(cx.ccx().tcx, ty));
let cleanup_type = cleanup_type(cx.tcx(), ty);
do in_scope_cx(cx) |scope_info| {
scope_info.cleanups.push(
clean_temp(val, |a| glue::drop_ty_immediate(a, val, ty),
cleanup_type));
scope_clean_changed(scope_info);
2012-02-17 04:18:14 -06:00
}
}
pub fn add_clean_temp_mem(bcx: block, val: ValueRef, t: ty::t) {
2012-08-06 12:53:38 -05:00
if !ty::type_needs_drop(bcx.tcx(), t) { return; }
2012-08-22 19:24:52 -05:00
debug!("add_clean_temp_mem(%s, %s, %s)",
2012-08-06 12:53:38 -05:00
bcx.to_str(), val_str(bcx.ccx().tn, val),
ty_to_str(bcx.ccx().tcx, t));
let (root, rooted) = root_for_cleanup(bcx, val, t);
2012-08-06 12:53:38 -05:00
let cleanup_type = cleanup_type(bcx.tcx(), t);
do in_scope_cx(bcx) |scope_info| {
scope_info.cleanups.push(
clean_temp(val, |a| glue::drop_ty_root(a, root, rooted, t),
cleanup_type));
scope_clean_changed(scope_info);
}
}
pub fn add_clean_frozen_root(bcx: block, val: ValueRef, t: ty::t) {
debug!("add_clean_frozen_root(%s, %s, %s)",
bcx.to_str(), val_str(bcx.ccx().tn, val),
ty_to_str(bcx.ccx().tcx, t));
let (root, rooted) = root_for_cleanup(bcx, val, t);
let cleanup_type = cleanup_type(bcx.tcx(), t);
do in_scope_cx(bcx) |scope_info| {
scope_info.cleanups.push(
clean_temp(val, |bcx| {
2013-02-27 20:34:04 -06:00
let bcx = callee::trans_lang_call(
bcx,
bcx.tcx().lang_items.return_to_mut_fn(),
~[
build::Load(bcx,
build::PointerCast(bcx,
root,
T_ptr(T_ptr(T_i8()))))
],
expr::Ignore
);
glue::drop_ty_root(bcx, root, rooted, t)
}, cleanup_type));
scope_clean_changed(scope_info);
2012-02-17 04:18:14 -06:00
}
}
pub fn add_clean_free(cx: block, ptr: ValueRef, heap: heap) {
2012-08-06 14:34:08 -05:00
let free_fn = match heap {
heap_managed | heap_managed_unique => {
let f: @fn(block) -> block = |a| glue::trans_free(a, ptr);
f
}
heap_exchange => {
let f: @fn(block) -> block = |a| glue::trans_exchange_free(a, ptr);
f
}
};
do in_scope_cx(cx) |scope_info| {
scope_info.cleanups.push(clean_temp(ptr, free_fn,
normal_exit_and_unwind));
scope_clean_changed(scope_info);
2012-02-17 04:18:14 -06:00
}
}
// Note that this only works for temporaries. We should, at some point, move
// to a system where we can also cancel the cleanup on local variables, but
// this will be more involved. For now, we simply zero out the local, and the
// drop glue checks whether it is zero.
pub fn revoke_clean(cx: block, val: ValueRef) {
do in_scope_cx(cx) |scope_info| {
let cleanup_pos = vec::position(
scope_info.cleanups,
2012-09-28 00:20:47 -05:00
|cu| match *cu {
clean_temp(v, _, _) if v == val => true,
_ => false
});
for cleanup_pos.each |i| {
scope_info.cleanups =
vec::append(vec::slice(scope_info.cleanups, 0u, *i).to_vec(),
vec::slice(scope_info.cleanups,
*i + 1u,
scope_info.cleanups.len()));
scope_clean_changed(scope_info);
}
}
}
pub fn block_cleanups(bcx: block) -> ~[cleanup] {
2013-02-21 13:57:20 -06:00
match *bcx.kind {
block_non_scope => ~[],
2013-02-21 13:57:20 -06:00
block_scope(ref mut inf) => /*bad*/copy inf.cleanups
}
}
pub enum block_kind {
// A scope at the end of which temporary values created inside of it are
// cleaned up. May correspond to an actual block in the language, but also
// to an implicit scope, for example, calls introduce an implicit scope in
// which the arguments are evaluated and cleaned up.
block_scope(scope_info),
// A non-scope block is a basic block created as a translation artifact
// from translating code that expresses conditional logic rather than by
// explicit { ... } block structure in the source language. It's called a
// non-scope block because it doesn't introduce a new variable scope.
block_non_scope,
}
pub struct scope_info {
2012-08-20 14:23:37 -05:00
loop_break: Option<block>,
loop_label: Option<ident>,
2012-02-17 04:18:14 -06:00
// A list of functions that must be run at when leaving this
// block, cleaning up any variables that were introduced in the
// block.
2013-02-21 13:57:20 -06:00
cleanups: ~[cleanup],
2012-02-17 04:18:14 -06:00
// Existing cleanup paths that may be reused, indexed by destination and
// cleared when the set of cleanups changes.
2013-02-21 13:57:20 -06:00
cleanup_paths: ~[cleanup_path],
2012-02-17 04:18:14 -06:00
// Unwinding landing pad. Also cleared when cleanups change.
2013-02-21 13:57:20 -06:00
landing_pad: Option<BasicBlockRef>,
}
2012-02-17 04:18:14 -06:00
pub trait get_node_info {
2013-02-22 00:41:37 -06:00
fn info(&self) -> Option<NodeInfo>;
}
impl get_node_info for @ast::expr {
2013-02-22 00:41:37 -06:00
fn info(&self) -> Option<NodeInfo> {
Some(NodeInfo { id: self.id, span: self.span })
2012-05-14 16:24:16 -05:00
}
}
impl get_node_info for ast::blk {
2013-02-22 00:41:37 -06:00
fn info(&self) -> Option<NodeInfo> {
Some(NodeInfo { id: self.node.id, span: self.span })
2012-05-14 16:24:16 -05:00
}
}
impl get_node_info for Option<@ast::expr> {
2013-02-22 00:41:37 -06:00
fn info(&self) -> Option<NodeInfo> {
2012-09-27 19:01:28 -05:00
self.chain_ref(|s| s.info())
2012-05-14 16:24:16 -05:00
}
}
pub struct NodeInfo {
2012-05-14 16:24:16 -05:00
id: ast::node_id,
span: span
}
2012-05-14 16:24:16 -05:00
// Basic block context. We create a block context for each basic block
// (single-entry, single-exit sequence of instructions) we generate from Rust
// code. Each basic block we generate is attached to a function, typically
// with many basic blocks per function. All the basic blocks attached to a
// function are organized as a directed graph.
pub struct block_ {
2011-07-27 07:19:39 -05:00
// The BasicBlockRef returned from a call to
// llvm::LLVMAppendBasicBlock(llfn, name), which adds a basic
// block to the function pointed to by llfn. We insert
// instructions into that block by way of this block context.
2011-09-02 17:34:58 -05:00
// The block pointing to this one in the function's digraph.
2012-09-06 21:40:15 -05:00
llbb: BasicBlockRef,
2013-02-21 13:57:20 -06:00
terminated: bool,
unreachable: bool,
2012-09-06 21:40:15 -05:00
parent: Option<block>,
2011-09-02 17:34:58 -05:00
// The 'kind' of basic block this is.
2013-02-21 13:57:20 -06:00
kind: @mut block_kind,
// Is this block part of a landing pad?
2012-09-06 21:40:15 -05:00
is_lpad: bool,
2012-05-14 16:24:16 -05:00
// info about the AST node this block originated from, if any
node_info: Option<NodeInfo>,
2011-09-02 17:34:58 -05:00
// The function context for the function to which this block is
// attached.
2012-09-06 21:40:15 -05:00
fcx: fn_ctxt
2012-09-05 17:58:43 -05:00
}
pub fn block_(llbb: BasicBlockRef, parent: Option<block>, -kind: block_kind,
is_lpad: bool, node_info: Option<NodeInfo>, fcx: fn_ctxt)
2012-09-05 17:58:43 -05:00
-> block_ {
block_ {
llbb: llbb,
terminated: false,
unreachable: false,
parent: parent,
2013-02-21 13:57:20 -06:00
kind: @mut kind,
2012-09-05 17:58:43 -05:00
is_lpad: is_lpad,
node_info: node_info,
fcx: fcx
}
}
2013-02-21 13:57:20 -06:00
pub type block = @mut block_;
pub fn mk_block(llbb: BasicBlockRef, parent: Option<block>, -kind: block_kind,
is_lpad: bool, node_info: Option<NodeInfo>, fcx: fn_ctxt)
-> block {
2013-02-21 13:57:20 -06:00
@mut block_(llbb, parent, kind, is_lpad, node_info, fcx)
}
// First two args are retptr, env
pub const first_real_arg: uint = 2u;
pub struct Result {
bcx: block,
val: ValueRef
}
pub fn rslt(bcx: block, val: ValueRef) -> Result {
Result {bcx: bcx, val: val}
}
pub impl Result {
2013-03-05 19:36:39 -06:00
fn unpack(&self, +bcx: &mut block) -> ValueRef {
*bcx = self.bcx;
return self.val;
}
}
pub fn ty_str(tn: @TypeNames, t: TypeRef) -> @str {
2012-08-01 19:30:05 -05:00
return lib::llvm::type_to_str(tn, t);
}
pub fn val_ty(v: ValueRef) -> TypeRef {
unsafe {
return llvm::LLVMTypeOf(v);
}
}
pub fn val_str(tn: @TypeNames, v: ValueRef) -> @str {
2012-08-01 19:30:05 -05:00
return ty_str(tn, val_ty(v));
}
2013-03-05 19:36:39 -06:00
pub fn in_scope_cx(cx: block, f: &fn(+si: &mut scope_info)) {
let mut cur = cx;
loop {
2013-02-21 13:57:20 -06:00
{
// XXX: Borrow check bug workaround.
let kind: &mut block_kind = &mut *cur.kind;
match *kind {
block_scope(ref mut inf) => {
debug!("in_scope_cx: selected cur=%s (cx=%s)",
cur.to_str(), cx.to_str());
f(inf);
return;
}
_ => ()
}
2012-02-17 04:18:14 -06:00
}
cur = block_parent(cur);
}
}
pub fn block_parent(cx: block) -> block {
2012-08-06 14:34:08 -05:00
match cx.parent {
2012-08-20 14:23:37 -05:00
Some(b) => b,
None => cx.sess().bug(fmt!("block_parent called on root block %?",
2012-08-22 19:24:52 -05:00
cx))
}
}
// Accessors
pub impl block {
2013-02-22 00:41:37 -06:00
pure fn ccx(&self) -> @CrateContext { *self.fcx.ccx }
pure fn tcx(&self) -> ty::ctxt { self.fcx.ccx.tcx }
pure fn sess(&self) -> Session { self.fcx.ccx.sess }
2013-02-22 00:41:37 -06:00
fn node_id_to_str(&self, id: ast::node_id) -> ~str {
ast_map::node_id_to_str(self.tcx().items, id, self.sess().intr())
}
2013-02-22 00:41:37 -06:00
fn expr_to_str(&self, e: @ast::expr) -> ~str {
expr_repr(self.tcx(), e)
}
2013-02-22 00:41:37 -06:00
fn expr_is_lval(&self, e: @ast::expr) -> bool {
ty::expr_is_lval(self.tcx(), self.ccx().maps.method_map, e)
}
2013-02-22 00:41:37 -06:00
fn expr_kind(&self, e: @ast::expr) -> ty::ExprKind {
ty::expr_kind(self.tcx(), self.ccx().maps.method_map, e)
}
2013-02-22 00:41:37 -06:00
fn def(&self, nid: ast::node_id) -> ast::def {
match self.tcx().def_map.find(&nid) {
Some(v) => v,
None => {
self.tcx().sess.bug(fmt!(
"No def associated with node id %?", nid));
}
}
}
2013-02-22 00:41:37 -06:00
fn val_str(&self, val: ValueRef) -> @str {
val_str(self.ccx().tn, val)
}
2013-02-22 00:41:37 -06:00
fn llty_str(&self, llty: TypeRef) -> @str {
ty_str(self.ccx().tn, llty)
}
2013-02-22 00:41:37 -06:00
fn ty_to_str(&self, t: ty::t) -> ~str {
ty_to_str(self.tcx(), t)
}
fn to_str(&self) -> ~str {
2012-08-06 14:34:08 -05:00
match self.node_info {
2012-08-20 14:23:37 -05:00
Some(node_info) => {
2012-08-22 19:24:52 -05:00
fmt!("[block %d]", node_info.id)
}
2012-08-20 14:23:37 -05:00
None => {
fmt!("[block %x]", ptr::addr_of(&(*self)) as uint)
}
}
}
}
// LLVM type constructors.
pub fn T_void() -> TypeRef {
unsafe {
return llvm::LLVMVoidType();
}
}
pub fn T_nil() -> TypeRef {
2013-01-07 21:10:16 -06:00
return T_struct(~[])
}
pub fn T_metadata() -> TypeRef { unsafe { return llvm::LLVMMetadataType(); } }
pub fn T_i1() -> TypeRef { unsafe { return llvm::LLVMInt1Type(); } }
pub fn T_i8() -> TypeRef { unsafe { return llvm::LLVMInt8Type(); } }
pub fn T_i16() -> TypeRef { unsafe { return llvm::LLVMInt16Type(); } }
pub fn T_i32() -> TypeRef { unsafe { return llvm::LLVMInt32Type(); } }
pub fn T_i64() -> TypeRef { unsafe { return llvm::LLVMInt64Type(); } }
pub fn T_f32() -> TypeRef { unsafe { return llvm::LLVMFloatType(); } }
pub fn T_f64() -> TypeRef { unsafe { return llvm::LLVMDoubleType(); } }
pub fn T_bool() -> TypeRef { return T_i8(); }
pub fn T_int(targ_cfg: @session::config) -> TypeRef {
2012-08-06 14:34:08 -05:00
return match targ_cfg.arch {
2012-08-03 21:59:04 -05:00
session::arch_x86 => T_i32(),
session::arch_x86_64 => T_i64(),
2013-01-29 08:28:08 -06:00
session::arch_arm => T_i32(),
session::arch_mips => T_i32()
};
}
pub fn T_int_ty(cx: @CrateContext, t: ast::int_ty) -> TypeRef {
2012-08-06 14:34:08 -05:00
match t {
2012-08-03 21:59:04 -05:00
ast::ty_i => cx.int_type,
ast::ty_char => T_char(),
ast::ty_i8 => T_i8(),
ast::ty_i16 => T_i16(),
ast::ty_i32 => T_i32(),
ast::ty_i64 => T_i64()
}
}
pub fn T_uint_ty(cx: @CrateContext, t: ast::uint_ty) -> TypeRef {
2012-08-06 14:34:08 -05:00
match t {
2012-08-03 21:59:04 -05:00
ast::ty_u => cx.int_type,
ast::ty_u8 => T_i8(),
ast::ty_u16 => T_i16(),
ast::ty_u32 => T_i32(),
ast::ty_u64 => T_i64()
}
}
pub fn T_float_ty(cx: @CrateContext, t: ast::float_ty) -> TypeRef {
2012-08-06 14:34:08 -05:00
match t {
2012-08-03 21:59:04 -05:00
ast::ty_f => cx.float_type,
ast::ty_f32 => T_f32(),
ast::ty_f64 => T_f64()
}
}
pub fn T_float(targ_cfg: @session::config) -> TypeRef {
2012-08-06 14:34:08 -05:00
return match targ_cfg.arch {
2012-08-03 21:59:04 -05:00
session::arch_x86 => T_f64(),
session::arch_x86_64 => T_f64(),
2013-01-29 08:28:08 -06:00
session::arch_arm => T_f64(),
session::arch_mips => T_f64()
};
}
pub fn T_char() -> TypeRef { return T_i32(); }
pub fn T_size_t(targ_cfg: @session::config) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_int(targ_cfg);
}
2013-03-05 19:36:39 -06:00
pub fn T_fn(inputs: &[TypeRef], output: TypeRef) -> TypeRef {
unsafe {
return llvm::LLVMFunctionType(output, to_ptr(inputs),
inputs.len() as c_uint,
False);
}
}
pub fn T_fn_pair(cx: @CrateContext, tfn: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(~[T_ptr(tfn), T_opaque_cbox_ptr(cx)]);
}
pub fn T_ptr(t: TypeRef) -> TypeRef {
unsafe {
return llvm::LLVMPointerType(t, default_addrspace);
}
2012-08-06 12:53:38 -05:00
}
pub fn T_root(t: TypeRef, addrspace: addrspace) -> TypeRef {
unsafe {
return llvm::LLVMPointerType(t, addrspace);
}
}
2013-03-05 19:36:39 -06:00
pub fn T_struct(elts: &[TypeRef]) -> TypeRef {
unsafe {
return llvm::LLVMStructType(to_ptr(elts),
elts.len() as c_uint,
False);
}
}
2013-03-05 19:36:39 -06:00
pub fn T_named_struct(name: &str) -> TypeRef {
unsafe {
let c = llvm::LLVMGetGlobalContext();
return str::as_c_str(name, |buf| llvm::LLVMStructCreateNamed(c, buf));
}
}
2013-03-05 19:36:39 -06:00
pub fn set_struct_body(t: TypeRef, elts: &[TypeRef]) {
unsafe {
llvm::LLVMStructSetBody(t,
to_ptr(elts),
elts.len() as c_uint,
False);
}
}
pub fn T_empty_struct() -> TypeRef { return T_struct(~[]); }
// A vtable is, in reality, a vtable pointer followed by zero or more pointers
// to tydescs and other vtables that it closes over. But the types and number
// of those are rarely known to the code that needs to manipulate them, so
// they are described by this opaque type.
pub fn T_vtable() -> TypeRef { T_array(T_ptr(T_i8()), 1u) }
pub fn T_task(targ_cfg: @session::config) -> TypeRef {
let t = T_named_struct(~"task");
2011-07-27 07:19:39 -05:00
2011-09-02 17:34:58 -05:00
// Refcount
// Delegate pointer
// Stack segment pointer
// Runtime SP
// Rust SP
// GC chain
2011-09-02 17:34:58 -05:00
// Domain pointer
// Crate cache pointer
let t_int = T_int(targ_cfg);
let elems =
~[t_int, t_int, t_int, t_int,
t_int, t_int, t_int, t_int];
set_struct_body(t, elems);
2012-08-01 19:30:05 -05:00
return t;
}
pub fn T_tydesc_field(cx: @CrateContext, field: uint) -> TypeRef {
// Bit of a kludge: pick the fn typeref out of the tydesc..
unsafe {
let mut tydesc_elts: ~[TypeRef] =
vec::from_elem::<TypeRef>(abi::n_tydesc_fields,
T_nil());
llvm::LLVMGetStructElementTypes(
cx.tydesc_type,
ptr::to_mut_unsafe_ptr(&mut tydesc_elts[0]));
let t = llvm::LLVMGetElementType(tydesc_elts[field]);
return t;
}
}
pub fn T_generic_glue_fn(cx: @CrateContext) -> TypeRef {
let s = @"glue_fn";
2012-08-06 14:34:08 -05:00
match name_has_type(cx.tn, s) {
2012-08-20 14:23:37 -05:00
Some(t) => return t,
2012-08-03 21:59:04 -05:00
_ => ()
}
2011-07-27 07:19:39 -05:00
let t = T_tydesc_field(cx, abi::tydesc_field_drop_glue);
associate_type(cx.tn, s, t);
2012-08-01 19:30:05 -05:00
return t;
}
pub fn T_tydesc(targ_cfg: @session::config) -> TypeRef {
let tydesc = T_named_struct(~"tydesc");
2011-07-27 07:19:39 -05:00
let tydescpp = T_ptr(T_ptr(tydesc));
let pvoid = T_ptr(T_i8());
let glue_fn_ty =
T_ptr(T_fn(~[T_ptr(T_nil()), T_ptr(T_nil()), tydescpp,
pvoid], T_void()));
let int_type = T_int(targ_cfg);
2011-08-04 13:25:09 -05:00
let elems =
~[int_type, int_type,
glue_fn_ty, glue_fn_ty, glue_fn_ty, glue_fn_ty,
T_ptr(T_i8()), T_ptr(T_i8())];
set_struct_body(tydesc, elems);
2012-08-01 19:30:05 -05:00
return tydesc;
}
pub fn T_array(t: TypeRef, n: uint) -> TypeRef {
unsafe {
return llvm::LLVMArrayType(t, n as c_uint);
}
}
// Interior vector.
pub fn T_vec2(targ_cfg: @session::config, t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(~[T_int(targ_cfg), // fill
T_int(targ_cfg), // alloc
T_array(t, 0u)]); // elements
}
pub fn T_vec(ccx: @CrateContext, t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_vec2(ccx.sess.targ_cfg, t);
}
// Note that the size of this one is in bytes.
pub fn T_opaque_vec(targ_cfg: @session::config) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_vec2(targ_cfg, T_i8());
}
// Let T be the content of a box @T. tuplify_box_ty(t) returns the
// representation of @T as a tuple (i.e., the ty::t version of what T_box()
// returns).
pub fn tuplify_box_ty(tcx: ty::ctxt, t: ty::t) -> ty::t {
let ptr = ty::mk_ptr(
tcx,
ty::mt {ty: ty::mk_nil(tcx), mutbl: ast::m_imm}
);
2012-08-01 19:30:05 -05:00
return ty::mk_tup(tcx, ~[ty::mk_uint(tcx), ty::mk_type(tcx),
ptr, ptr,
t]);
}
pub fn T_box_header_fields(cx: @CrateContext) -> ~[TypeRef] {
let ptr = T_ptr(T_i8());
2012-08-01 19:30:05 -05:00
return ~[cx.int_type, T_ptr(cx.tydesc_type), ptr, ptr];
}
pub fn T_box_header(cx: @CrateContext) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(T_box_header_fields(cx));
}
pub fn T_box(cx: @CrateContext, t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(vec::append(T_box_header_fields(cx), ~[t]));
}
pub fn T_box_ptr(t: TypeRef) -> TypeRef {
unsafe {
return llvm::LLVMPointerType(t, gc_box_addrspace);
}
2012-05-07 16:27:43 -05:00
}
pub fn T_opaque_box(cx: @CrateContext) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_box(cx, T_i8());
}
pub fn T_opaque_box_ptr(cx: @CrateContext) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_box_ptr(T_opaque_box(cx));
}
pub fn T_unique(cx: @CrateContext, t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(vec::append(T_box_header_fields(cx), ~[t]));
}
pub fn T_unique_ptr(t: TypeRef) -> TypeRef {
unsafe {
return llvm::LLVMPointerType(t, gc_box_addrspace);
}
}
pub fn T_port(cx: @CrateContext, _t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(~[cx.int_type]); // Refcount
}
pub fn T_chan(cx: @CrateContext, _t: TypeRef) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(~[cx.int_type]); // Refcount
}
pub fn T_taskptr(cx: @CrateContext) -> TypeRef { return T_ptr(cx.task_type); }
// This type must never be used directly; it must always be cast away.
pub fn T_typaram(tn: @TypeNames) -> TypeRef {
let s = @"typaram";
2012-08-06 14:34:08 -05:00
match name_has_type(tn, s) {
2012-08-20 14:23:37 -05:00
Some(t) => return t,
2012-08-03 21:59:04 -05:00
_ => ()
}
2011-07-27 07:19:39 -05:00
let t = T_i8();
associate_type(tn, s, t);
2012-08-01 19:30:05 -05:00
return t;
}
pub fn T_typaram_ptr(tn: @TypeNames) -> TypeRef {
return T_ptr(T_typaram(tn));
}
pub fn T_opaque_cbox_ptr(cx: @CrateContext) -> TypeRef {
// closures look like boxes (even when they are ~fn or &fn)
// see trans_closure.rs
2012-08-01 19:30:05 -05:00
return T_opaque_box_ptr(cx);
}
pub fn T_enum_discrim(cx: @CrateContext) -> TypeRef {
2012-08-01 19:30:05 -05:00
return cx.int_type;
}
pub fn T_opaque_enum(cx: @CrateContext) -> TypeRef {
let s = @"opaque_enum";
2012-08-06 14:34:08 -05:00
match name_has_type(cx.tn, s) {
2012-08-20 14:23:37 -05:00
Some(t) => return t,
2012-08-03 21:59:04 -05:00
_ => ()
}
let t = T_struct(~[T_enum_discrim(cx), T_i8()]);
associate_type(cx.tn, s, t);
2012-08-01 19:30:05 -05:00
return t;
}
pub fn T_opaque_enum_ptr(cx: @CrateContext) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_ptr(T_opaque_enum(cx));
}
pub fn T_captured_tydescs(cx: @CrateContext, n: uint) -> TypeRef {
2012-08-01 19:30:05 -05:00
return T_struct(vec::from_elem::<TypeRef>(n, T_ptr(cx.tydesc_type)));
}
pub fn T_opaque_trait(cx: @CrateContext, vstore: ty::vstore) -> TypeRef {
2012-10-05 18:55:42 -05:00
match vstore {
2012-10-31 17:09:26 -05:00
ty::vstore_box => {
T_struct(~[T_ptr(cx.tydesc_type), T_opaque_box_ptr(cx)])
}
ty::vstore_uniq => {
T_struct(~[T_ptr(cx.tydesc_type),
T_unique_ptr(T_unique(cx, T_i8())),
T_ptr(cx.tydesc_type)])
}
_ => T_struct(~[T_ptr(cx.tydesc_type), T_ptr(T_i8())])
2012-10-05 18:55:42 -05:00
}
}
pub fn T_opaque_port_ptr() -> TypeRef { return T_ptr(T_i8()); }
pub fn T_opaque_chan_ptr() -> TypeRef { return T_ptr(T_i8()); }
// LLVM constant constructors.
pub fn C_null(t: TypeRef) -> ValueRef {
unsafe {
return llvm::LLVMConstNull(t);
}
}
pub fn C_integral(t: TypeRef, u: u64, sign_extend: Bool) -> ValueRef {
unsafe {
return llvm::LLVMConstInt(t, u, sign_extend);
}
}
2013-03-05 19:36:39 -06:00
pub fn C_floating(s: &str, t: TypeRef) -> ValueRef {
unsafe {
return str::as_c_str(s, |buf| llvm::LLVMConstRealOfString(t, buf));
}
}
pub fn C_nil() -> ValueRef {
2013-01-07 21:10:16 -06:00
return C_struct(~[]);
}
pub fn C_bool(b: bool) -> ValueRef {
C_integral(T_bool(), if b { 1u64 } else { 0u64 }, False)
}
pub fn C_i1(b: bool) -> ValueRef {
return C_integral(T_i1(), if b { 1 } else { 0 }, False);
}
pub fn C_i32(i: i32) -> ValueRef {
2012-08-01 19:30:05 -05:00
return C_integral(T_i32(), i as u64, True);
2011-10-26 00:23:28 -05:00
}
pub fn C_i64(i: i64) -> ValueRef {
2012-08-01 19:30:05 -05:00
return C_integral(T_i64(), i as u64, True);
}
pub fn C_int(cx: @CrateContext, i: int) -> ValueRef {
2012-08-01 19:30:05 -05:00
return C_integral(cx.int_type, i as u64, True);
}
pub fn C_uint(cx: @CrateContext, i: uint) -> ValueRef {
2012-08-01 19:30:05 -05:00
return C_integral(cx.int_type, i as u64, False);
}
pub fn C_u8(i: uint) -> ValueRef {
return C_integral(T_i8(), i as u64, False);
}
// This is a 'c-like' raw string, which differs from
// our boxed-and-length-annotated strings.
pub fn C_cstr(cx: @CrateContext, s: @~str) -> ValueRef {
unsafe {
match cx.const_cstr_cache.find(&s) {
2013-02-16 12:36:09 -06:00
Some(llval) => return llval,
None => ()
}
let sc = do str::as_c_str(*s) |buf| {
2013-02-16 12:36:09 -06:00
llvm::LLVMConstString(buf, s.len() as c_uint, False)
};
let g =
str::as_c_str(fmt!("str%u", (cx.names)(~"str").repr),
|buf| llvm::LLVMAddGlobal(cx.llmod, val_ty(sc), buf));
llvm::LLVMSetInitializer(g, sc);
llvm::LLVMSetGlobalConstant(g, True);
lib::llvm::SetLinkage(g, lib::llvm::InternalLinkage);
cx.const_cstr_cache.insert(s, g);
return g;
}
}
// NB: Do not use `do_spill_noroot` to make this into a constant string, or
// you will be kicked off fast isel. See issue #4352 for an example of this.
pub fn C_estr_slice(cx: @CrateContext, s: @~str) -> ValueRef {
unsafe {
let len = s.len();
let cs = llvm::LLVMConstPointerCast(C_cstr(cx, s), T_ptr(T_i8()));
C_struct(~[cs, C_uint(cx, len + 1u /* +1 for null */)])
}
}
// Returns a Plain Old LLVM String:
2013-03-05 19:36:39 -06:00
pub fn C_postr(s: &str) -> ValueRef {
unsafe {
return do str::as_c_str(s) |buf| {
llvm::LLVMConstString(buf, str::len(s) as c_uint, False)
};
}
}
pub fn C_zero_byte_arr(size: uint) -> ValueRef {
unsafe {
let mut i = 0u;
let mut elts: ~[ValueRef] = ~[];
while i < size { elts.push(C_u8(0u)); i += 1u; }
return llvm::LLVMConstArray(T_i8(),
vec::raw::to_ptr(elts),
elts.len() as c_uint);
}
}
pub fn C_struct(elts: &[ValueRef]) -> ValueRef {
unsafe {
do vec::as_imm_buf(elts) |ptr, len| {
llvm::LLVMConstStruct(ptr, len as c_uint, False)
}
}
}
pub fn C_packed_struct(elts: &[ValueRef]) -> ValueRef {
unsafe {
do vec::as_imm_buf(elts) |ptr, len| {
llvm::LLVMConstStruct(ptr, len as c_uint, True)
}
}
}
pub fn C_named_struct(T: TypeRef, elts: &[ValueRef]) -> ValueRef {
unsafe {
do vec::as_imm_buf(elts) |ptr, len| {
llvm::LLVMConstNamedStruct(T, ptr, len as c_uint)
}
}
}
2013-03-05 19:36:39 -06:00
pub fn C_array(ty: TypeRef, elts: &[ValueRef]) -> ValueRef {
unsafe {
return llvm::LLVMConstArray(ty, vec::raw::to_ptr(elts),
elts.len() as c_uint);
}
2011-07-27 17:14:59 -05:00
}
2011-08-04 12:46:10 -05:00
2013-03-05 19:36:39 -06:00
pub fn C_bytes(bytes: &[u8]) -> ValueRef {
unsafe {
return llvm::LLVMConstString(
cast::reinterpret_cast(&vec::raw::to_ptr(bytes)),
bytes.len() as c_uint, True);
}
}
2013-03-05 19:36:39 -06:00
pub fn C_bytes_plus_null(bytes: &[u8]) -> ValueRef {
unsafe {
return llvm::LLVMConstString(
cast::reinterpret_cast(&vec::raw::to_ptr(bytes)),
bytes.len() as c_uint, False);
}
2011-08-04 12:46:10 -05:00
}
pub fn C_shape(ccx: @CrateContext, +bytes: ~[u8]) -> ValueRef {
unsafe {
let llshape = C_bytes_plus_null(bytes);
let name = fmt!("shape%u", (ccx.names)(~"shape").repr);
let llglobal = str::as_c_str(name, |buf| {
llvm::LLVMAddGlobal(ccx.llmod, val_ty(llshape), buf)
});
llvm::LLVMSetInitializer(llglobal, llshape);
llvm::LLVMSetGlobalConstant(llglobal, True);
lib::llvm::SetLinkage(llglobal, lib::llvm::InternalLinkage);
return llvm::LLVMConstPointerCast(llglobal, T_ptr(T_i8()));
}
2011-08-04 13:25:09 -05:00
}
pub fn get_param(fndecl: ValueRef, param: uint) -> ValueRef {
unsafe {
llvm::LLVMGetParam(fndecl, param as c_uint)
}
}
// Used to identify cached monomorphized functions and vtables
#[deriving_eq]
pub enum mono_param_id {
2012-08-20 14:23:37 -05:00
mono_precise(ty::t, Option<~[mono_id]>),
mono_any,
mono_repr(uint /* size */,
uint /* align */,
bool /* is_float */,
datum::DatumMode),
}
2012-08-27 18:26:35 -05:00
#[deriving_eq]
pub struct mono_id_ {
def: ast::def_id,
params: ~[mono_param_id],
impl_did_opt: Option<ast::def_id>
}
2012-08-27 18:26:35 -05:00
pub type mono_id = @mono_id_;
2012-08-27 18:26:35 -05:00
impl to_bytes::IterBytes for mono_param_id {
pure fn iter_bytes(&self, +lsb0: bool, f: to_bytes::Cb) {
match /*bad*/copy *self {
mono_precise(t, mids) =>
to_bytes::iter_bytes_3(&0u8, &ty::type_id(t), &mids, lsb0, f),
mono_any => 1u8.iter_bytes(lsb0, f),
mono_repr(ref a, ref b, ref c, ref d) =>
to_bytes::iter_bytes_5(&2u8, a, b, c, d, lsb0, f)
}
}
}
impl to_bytes::IterBytes for mono_id_ {
pure fn iter_bytes(&self, +lsb0: bool, f: to_bytes::Cb) {
to_bytes::iter_bytes_2(&self.def, &self.params, lsb0, f);
}
}
pub fn umax(cx: block, a: ValueRef, b: ValueRef) -> ValueRef {
let cond = build::ICmp(cx, lib::llvm::IntULT, a, b);
2012-08-01 19:30:05 -05:00
return build::Select(cx, cond, b, a);
}
pub fn umin(cx: block, a: ValueRef, b: ValueRef) -> ValueRef {
let cond = build::ICmp(cx, lib::llvm::IntULT, a, b);
2012-08-01 19:30:05 -05:00
return build::Select(cx, cond, a, b);
}
pub fn align_to(cx: block, off: ValueRef, align: ValueRef) -> ValueRef {
let mask = build::Sub(cx, align, C_int(cx.ccx(), 1));
let bumped = build::Add(cx, off, mask);
2012-08-01 19:30:05 -05:00
return build::And(cx, bumped, build::Not(cx, mask));
}
2013-03-05 19:36:39 -06:00
pub fn path_str(sess: session::Session, p: &path) -> ~str {
let mut r = ~"", first = true;
2013-03-05 19:36:39 -06:00
for vec::each(*p) |e| {
match *e {
ast_map::path_name(s) | ast_map::path_mod(s) => {
if first { first = false; }
else { r += ~"::"; }
r += *sess.str_of(s);
}
}
}
r
}
pub fn monomorphize_type(bcx: block, t: ty::t) -> ty::t {
match /*bad*/copy bcx.fcx.param_substs {
Some(substs) => {
ty::subst_tps(bcx.tcx(), substs.tys, substs.self_ty, t)
}
_ => { assert !ty::type_has_params(t); t }
}
}
pub fn node_id_type(bcx: block, id: ast::node_id) -> ty::t {
let tcx = bcx.tcx();
let t = ty::node_id_to_type(tcx, id);
monomorphize_type(bcx, t)
}
pub fn expr_ty(bcx: block, ex: @ast::expr) -> ty::t {
node_id_type(bcx, ex.id)
}
pub fn expr_ty_adjusted(bcx: block, ex: @ast::expr) -> ty::t {
let tcx = bcx.tcx();
let t = ty::expr_ty_adjusted(tcx, ex);
monomorphize_type(bcx, t)
}
pub fn node_id_type_params(bcx: block, id: ast::node_id) -> ~[ty::t] {
let tcx = bcx.tcx();
let params = ty::node_id_to_type_params(tcx, id);
match /*bad*/copy bcx.fcx.param_substs {
2012-08-20 14:23:37 -05:00
Some(substs) => {
do vec::map(params) |t| {
ty::subst_tps(tcx, substs.tys, substs.self_ty, *t)
}
}
2012-08-03 21:59:04 -05:00
_ => params
}
}
pub fn node_vtables(bcx: block, id: ast::node_id)
-> Option<typeck::vtable_res> {
let raw_vtables = bcx.ccx().maps.vtable_map.find(&id);
raw_vtables.map(
|vts| resolve_vtables_in_fn_ctxt(bcx.fcx, *vts))
}
pub fn resolve_vtables_in_fn_ctxt(fcx: fn_ctxt, vts: typeck::vtable_res)
-> typeck::vtable_res {
@vec::map(*vts, |d| resolve_vtable_in_fn_ctxt(fcx, copy *d))
}
// Apply the typaram substitutions in the fn_ctxt to a vtable. This should
// eliminate any vtable_params.
pub fn resolve_vtable_in_fn_ctxt(fcx: fn_ctxt, +vt: typeck::vtable_origin)
-> typeck::vtable_origin {
let tcx = fcx.ccx.tcx;
match vt {
typeck::vtable_static(trait_id, tys, sub) => {
let tys = match /*bad*/copy fcx.param_substs {
Some(substs) => {
do vec::map(tys) |t| {
ty::subst_tps(tcx, substs.tys, substs.self_ty, *t)
}
}
_ => tys
};
typeck::vtable_static(trait_id, tys,
resolve_vtables_in_fn_ctxt(fcx, sub))
}
typeck::vtable_param(n_param, n_bound) => {
match fcx.param_substs {
Some(substs) => {
find_vtable(tcx, substs, n_param, n_bound)
}
_ => {
tcx.sess.bug(fmt!(
"resolve_vtable_in_fn_ctxt: asked to lookup but \
no vtables in the fn_ctxt!"))
}
}
}
}
}
pub fn find_vtable(tcx: ty::ctxt, ps: &param_substs,
n_param: uint, n_bound: uint)
-> typeck::vtable_origin {
debug!("find_vtable_in_fn_ctxt(n_param=%u, n_bound=%u, ps=%?)",
n_param, n_bound, param_substs_to_str(tcx, ps));
// Vtables are stored in a flat array, finding the right one is
// somewhat awkward
let first_n_bounds = ps.bounds.view(0, n_param);
let vtables_to_skip =
ty::count_traits_and_supertraits(tcx, first_n_bounds);
let vtable_off = vtables_to_skip + n_bound;
/*bad*/ copy ps.vtables.get()[vtable_off]
}
pub fn dummy_substs(+tps: ~[ty::t]) -> ty::substs {
substs {
self_r: Some(ty::re_bound(ty::br_self)),
self_ty: None,
tps: tps
}
}
pub fn struct_field(index: uint) -> [uint * 3] {
//! The GEPi sequence to access a field of a record/struct.
[0, 0, index]
}
pub fn struct_dtor() -> [uint * 2] {
//! The GEPi sequence to access the dtor of a struct.
[0, 1]
2012-08-27 18:26:35 -05:00
}
// Casts a Rust bool value to an i1.
pub fn bool_to_i1(bcx: block, llval: ValueRef) -> ValueRef {
build::ICmp(bcx, lib::llvm::IntNE, llval, C_bool(false))
}
//
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End:
//