mod.rs 123.1 KB
Newer Older
1 2
// ignore-tidy-filelength

M
Mateusz Mikuła 已提交
3
#![allow(usage_of_ty_tykind)]
F
flip1995 已提交
4

S
Steven Fackler 已提交
5
pub use self::Variance::*;
6
pub use self::AssociatedItemContainer::*;
S
Steven Fackler 已提交
7 8
pub use self::BorrowKind::*;
pub use self::IntVarValue::*;
9
pub use self::fold::TypeFoldable;
S
Steven Fackler 已提交
10

11
use crate::hir::{map as hir_map, UpvarMap, GlobMap, TraitMap};
12
use crate::hir::Node;
13
use crate::hir::def::{Res, DefKind, CtorOf, CtorKind, ExportMap};
M
Mark Mansi 已提交
14
use crate::hir::def_id::{CrateNum, DefId, LocalDefId, CRATE_DEF_INDEX, LOCAL_CRATE};
15
use rustc_data_structures::svh::Svh;
16
use rustc_macros::HashStable;
M
Mark Mansi 已提交
17 18 19 20 21 22 23 24 25 26 27 28
use crate::ich::Fingerprint;
use crate::ich::StableHashingContext;
use crate::infer::canonical::Canonical;
use crate::middle::lang_items::{FnTraitLangItem, FnMutTraitLangItem, FnOnceTraitLangItem};
use crate::middle::resolve_lifetime::ObjectLifetimeDefault;
use crate::mir::Mir;
use crate::mir::interpret::{GlobalId, ErrorHandled};
use crate::mir::GeneratorLayout;
use crate::session::CrateDisambiguator;
use crate::traits::{self, Reveal};
use crate::ty;
use crate::ty::layout::VariantIdx;
C
csmoe 已提交
29
use crate::ty::subst::{Subst, InternalSubsts, SubstsRef};
M
Mark Mansi 已提交
30 31 32 33
use crate::ty::util::{IntTypeExt, Discr};
use crate::ty::walk::TypeWalker;
use crate::util::captures::Captures;
use crate::util::nodemap::{NodeSet, DefIdMap, FxHashMap};
J
John Kåre Alsaker 已提交
34
use arena::SyncDroplessArena;
M
Mark Mansi 已提交
35
use crate::session::DataTypeKind;
36

37
use serialize::{self, Encodable, Encoder};
W
Wesley Wiser 已提交
38
use std::cell::RefCell;
39
use std::cmp::{self, Ordering};
40
use std::fmt;
41
use std::hash::{Hash, Hasher};
42
use std::ops::Deref;
J
John Kåre Alsaker 已提交
43
use rustc_data_structures::sync::{self, Lrc, ParallelIterator, par_iter};
44
use std::slice;
45
use std::{mem, ptr};
46
use std::ops::Range;
L
ljedrz 已提交
47
use syntax::ast::{self, Name, Ident, NodeId};
48
use syntax::attr;
49
use syntax::ext::hygiene::Mark;
50
use syntax::symbol::{keywords, Symbol, LocalInternedString, InternedString};
51
use syntax_pos::Span;
52

53
use smallvec;
54
use rustc_data_structures::indexed_vec::{Idx, IndexVec};
55 56
use rustc_data_structures::stable_hasher::{StableHasher, StableHasherResult,
                                           HashStable};
O
Oliver Schneider 已提交
57

M
Mark Mansi 已提交
58
use crate::hir;
59

S
scalexm 已提交
60
pub use self::sty::{Binder, BoundTy, BoundTyKind, BoundVar, DebruijnIndex, INNERMOST};
61
pub use self::sty::{FnSig, GenSig, CanonicalPolyFnSig, PolyFnSig, PolyGenSig};
V
varkor 已提交
62
pub use self::sty::{InferTy, ParamTy, ParamConst, InferConst, ProjectionTy, ExistentialPredicate};
63
pub use self::sty::{ClosureSubsts, GeneratorSubsts, UpvarSubsts, TypeAndMut};
V
varkor 已提交
64
pub use self::sty::{TraitRef, TyKind, PolyTraitRef};
65
pub use self::sty::{ExistentialTraitRef, PolyExistentialTraitRef};
O
Oliver Scherer 已提交
66
pub use self::sty::{ExistentialProjection, PolyExistentialProjection, Const};
67
pub use self::sty::{BoundRegion, EarlyBoundRegion, FreeRegion, Region};
N
Niko Matsakis 已提交
68
pub use self::sty::RegionKind;
V
varkor 已提交
69
pub use self::sty::{TyVid, IntVid, FloatVid, ConstVid, RegionVid};
70 71
pub use self::sty::BoundRegion::*;
pub use self::sty::InferTy::*;
N
Niko Matsakis 已提交
72
pub use self::sty::RegionKind::*;
V
varkor 已提交
73
pub use self::sty::TyKind::*;
74

75 76 77
pub use self::binding::BindingMode;
pub use self::binding::BindingMode::*;

78
pub use self::context::{TyCtxt, FreeRegionInfo, GlobalArenas, AllArenas, tls, keep_local};
79
pub use self::context::{Lift, TypeckTables, CtxtInterners, GlobalCtxt};
D
David Wood 已提交
80
pub use self::context::{
81
    UserTypeAnnotationIndex, UserType, CanonicalUserType,
82
    CanonicalUserTypeAnnotation, CanonicalUserTypeAnnotations, ResolvedOpaqueTy,
D
David Wood 已提交
83
};
84

85 86
pub use self::instance::{Instance, InstanceDef};

87
pub use self::trait_def::TraitDef;
88

89
pub use self::query::queries;
90

91
pub mod adjustment;
92
pub mod binding;
93
pub mod cast;
94
#[macro_use]
95
pub mod codec;
96
mod constness;
97
pub mod error;
98
mod erase_regions;
99
pub mod fast_reject;
V
varkor 已提交
100
pub mod flags;
101
pub mod fold;
A
Andrew Cann 已提交
102
pub mod inhabitedness;
103
pub mod layout;
104 105
pub mod _match;
pub mod outlives;
106
pub mod print;
107
pub mod query;
108
pub mod relate;
109
pub mod steal;
110
pub mod subst;
111
pub mod trait_def;
112 113
pub mod walk;
pub mod wf;
114
pub mod util;
115

116
mod context;
117
mod instance;
118 119 120
mod structural_impls;
mod sty;

121
// Data types
122

123 124
#[derive(Clone)]
pub struct Resolutions {
125
    pub upvars: UpvarMap,
126 127
    pub trait_map: TraitMap,
    pub maybe_unused_trait_imports: NodeSet,
128
    pub maybe_unused_extern_crates: Vec<(NodeId, Span)>,
L
ljedrz 已提交
129
    pub export_map: ExportMap<NodeId>,
130
    pub glob_map: GlobMap,
V
Vadim Petrochenkov 已提交
131 132 133
    /// Extern prelude entries. The value is `true` if the entry was introduced
    /// via `extern crate` item and not `--extern` option or compiler built-in.
    pub extern_prelude: FxHashMap<Name, bool>,
134 135
}

136
#[derive(Clone, Copy, PartialEq, Eq, Debug, HashStable)]
137
pub enum AssociatedItemContainer {
N
Niko Matsakis 已提交
138 139
    TraitContainer(DefId),
    ImplContainer(DefId),
140 141
}

142
impl AssociatedItemContainer {
A
Alexander Regueiro 已提交
143 144
    /// Asserts that this is the `DefId` of an associated item declared
    /// in a trait, and returns the trait `DefId`.
145 146 147 148 149 150 151
    pub fn assert_trait(&self) -> DefId {
        match *self {
            TraitContainer(id) => id,
            _ => bug!("associated item has wrong container type: {:?}", self)
        }
    }

N
Niko Matsakis 已提交
152
    pub fn id(&self) -> DefId {
153 154 155 156 157 158 159
        match *self {
            TraitContainer(id) => id,
            ImplContainer(id) => id,
        }
    }
}

A
Aaron Turon 已提交
160 161
/// The "header" of an impl is everything outside the body: a Self type, a trait
/// ref (in the case of a trait impl), and a set of predicates (from the
A
Alexander Regueiro 已提交
162
/// bounds / where-clauses).
A
Aaron Turon 已提交
163 164 165 166 167 168 169 170
#[derive(Clone, PartialEq, Eq, Hash, Debug)]
pub struct ImplHeader<'tcx> {
    pub impl_def_id: DefId,
    pub self_ty: Ty<'tcx>,
    pub trait_ref: Option<TraitRef<'tcx>>,
    pub predicates: Vec<Predicate<'tcx>>,
}

171
#[derive(Copy, Clone, Debug, PartialEq, HashStable)]
172 173
pub struct AssociatedItem {
    pub def_id: DefId,
174
    #[stable_hasher(project(name))]
175
    pub ident: Ident,
176 177 178 179
    pub kind: AssociatedKind,
    pub vis: Visibility,
    pub defaultness: hir::Defaultness,
    pub container: AssociatedItemContainer,
180

181 182 183 184
    /// Whether this is a method with an explicit self
    /// as its first argument, allowing method calls.
    pub method_has_self_argument: bool,
}
185

186
#[derive(Copy, Clone, PartialEq, Eq, Debug, Hash, RustcEncodable, RustcDecodable, HashStable)]
187 188 189
pub enum AssociatedKind {
    Const,
    Method,
O
Oliver Schneider 已提交
190
    Existential,
191 192
    Type
}
193

194
impl AssociatedItem {
195
    pub fn def_kind(&self) -> DefKind {
196
        match self.kind {
197 198 199 200
            AssociatedKind::Const => DefKind::AssociatedConst,
            AssociatedKind::Method => DefKind::Method,
            AssociatedKind::Type => DefKind::AssociatedTy,
            AssociatedKind::Existential => DefKind::AssociatedExistential,
201 202
        }
    }
A
Andrew Cann 已提交
203 204 205 206 207

    /// Tests whether the associated item admits a non-trivial implementation
    /// for !
    pub fn relevant_for_never<'tcx>(&self) -> bool {
        match self.kind {
O
Oliver Schneider 已提交
208 209
            AssociatedKind::Existential |
            AssociatedKind::Const |
A
Andrew Cann 已提交
210
            AssociatedKind::Type => true,
A
Andrew Cann 已提交
211
            // FIXME(canndrew): Be more thorough here, check if any argument is uninhabited.
A
Andrew Cann 已提交
212 213 214
            AssociatedKind::Method => !self.method_has_self_argument,
        }
    }
215

F
flip1995 已提交
216
    pub fn signature<'a, 'tcx>(&self, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> String {
217 218 219 220 221 222
        match self.kind {
            ty::AssociatedKind::Method => {
                // We skip the binder here because the binder would deanonymize all
                // late-bound regions, and we don't want method signatures to show up
                // `as for<'r> fn(&'r MyType)`.  Pretty-printing handles late-bound
                // regions just fine, showing `fn(&MyType)`.
L
ljedrz 已提交
223
                tcx.fn_sig(self.def_id).skip_binder().to_string()
224
            }
225
            ty::AssociatedKind::Type => format!("type {};", self.ident),
O
Oliver Schneider 已提交
226
            ty::AssociatedKind::Existential => format!("existential type {};", self.ident),
227
            ty::AssociatedKind::Const => {
228
                format!("const {}: {:?};", self.ident, tcx.type_of(self.def_id))
229 230 231
            }
        }
    }
232 233
}

234
#[derive(Clone, Debug, PartialEq, Eq, Copy, RustcEncodable, RustcDecodable, HashStable)]
235 236 237 238
pub enum Visibility {
    /// Visible everywhere (including in other crates).
    Public,
    /// Visible only in the given crate-local module.
239
    Restricted(DefId),
240
    /// Not visible anywhere in the local crate. This is the visibility of private external items.
241
    Invisible,
242 243
}

244 245
pub trait DefIdTree: Copy {
    fn parent(self, id: DefId) -> Option<DefId>;
A
Andrew Cann 已提交
246 247 248 249 250 251 252 253 254 255 256 257 258 259

    fn is_descendant_of(self, mut descendant: DefId, ancestor: DefId) -> bool {
        if descendant.krate != ancestor.krate {
            return false;
        }

        while descendant != ancestor {
            match self.parent(descendant) {
                Some(parent) => descendant = parent,
                None => return false,
            }
        }
        true
    }
260 261
}

262 263 264
impl<'a, 'gcx, 'tcx> DefIdTree for TyCtxt<'a, 'gcx, 'tcx> {
    fn parent(self, id: DefId) -> Option<DefId> {
        self.def_key(id).parent.map(|index| DefId { index: index, ..id })
265 266 267
    }
}

268
impl Visibility {
L
ljedrz 已提交
269
    pub fn from_hir(visibility: &hir::Visibility, id: hir::HirId, tcx: TyCtxt<'_, '_, '_>) -> Self {
270
        match visibility.node {
271 272
            hir::VisibilityKind::Public => Visibility::Public,
            hir::VisibilityKind::Crate(_) => Visibility::Restricted(DefId::local(CRATE_DEF_INDEX)),
273
            hir::VisibilityKind::Restricted { ref path, .. } => match path.res {
274 275
                // If there is no resolution, `resolve` will have already reported an error, so
                // assume that the visibility is public to avoid reporting more privacy errors.
276
                Res::Err => Visibility::Public,
277
                def => Visibility::Restricted(def.def_id()),
278
            },
279
            hir::VisibilityKind::Inherited => {
L
ljedrz 已提交
280
                Visibility::Restricted(tcx.hir().get_module_parent_by_hir_id(id))
281
            }
282 283
        }
    }
284

285
    /// Returns `true` if an item with this visibility is accessible from the given block.
A
Andrew Cann 已提交
286
    pub fn is_accessible_from<T: DefIdTree>(self, module: DefId, tree: T) -> bool {
287 288 289 290
        let restriction = match self {
            // Public items are visible everywhere.
            Visibility::Public => return true,
            // Private items from other crates are visible nowhere.
291
            Visibility::Invisible => return false,
292
            // Restricted items are visible in an arbitrary local module.
293
            Visibility::Restricted(other) if other.krate != module.krate => return false,
294 295 296
            Visibility::Restricted(module) => module,
        };

A
Andrew Cann 已提交
297
        tree.is_descendant_of(module, restriction)
298
    }
299

300
    /// Returns `true` if this visibility is at least as accessible as the given visibility
301
    pub fn is_at_least<T: DefIdTree>(self, vis: Visibility, tree: T) -> bool {
302 303
        let vis_restriction = match vis {
            Visibility::Public => return self == Visibility::Public,
304
            Visibility::Invisible => return true,
305 306 307
            Visibility::Restricted(module) => module,
        };

308
        self.is_accessible_from(vis_restriction, tree)
309
    }
310

311
    // Returns `true` if this item is visible anywhere in the local crate.
312 313 314 315 316 317 318
    pub fn is_visible_locally(self) -> bool {
        match self {
            Visibility::Public => true,
            Visibility::Restricted(def_id) => def_id.is_local(),
            Visibility::Invisible => false,
        }
    }
319 320
}

321
#[derive(Copy, Clone, PartialEq, Eq, RustcDecodable, RustcEncodable, Hash, HashStable)]
322
pub enum Variance {
323 324 325 326
    Covariant,      // T<A> <: T<B> iff A <: B -- e.g., function return type
    Invariant,      // T<A> <: T<B> iff B == A -- e.g., type of mutable cell
    Contravariant,  // T<A> <: T<B> iff B <: A -- e.g., function param type
    Bivariant,      // T<A> <: T<B>            -- e.g., unused type parameter
327
}
328

329 330 331 332
/// The crate variances map is computed during typeck and contains the
/// variance of every item in the local crate. You should not use it
/// directly, because to do so will make your pass dependent on the
/// HIR of every item in the local crate. Instead, use
333
/// `tcx.variances_of()` to get the variance for a *particular*
334
/// item.
335
#[derive(HashStable)]
J
John Kåre Alsaker 已提交
336
pub struct CrateVariancesMap<'tcx> {
337
    /// For each item with generics, maps to a vector of the variance
A
Alexander Regueiro 已提交
338
    /// of its generics. If an item has no generics, it will have no
339
    /// entry.
J
John Kåre Alsaker 已提交
340
    pub variances: FxHashMap<DefId, &'tcx [ty::Variance]>,
341 342
}

343 344
impl Variance {
    /// `a.xform(b)` combines the variance of a context with the
A
Alexander Regueiro 已提交
345
    /// variance of a type with the following meaning. If we are in a
346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368
    /// context with variance `a`, and we encounter a type argument in
    /// a position with variance `b`, then `a.xform(b)` is the new
    /// variance with which the argument appears.
    ///
    /// Example 1:
    ///
    ///     *mut Vec<i32>
    ///
    /// Here, the "ambient" variance starts as covariant. `*mut T` is
    /// invariant with respect to `T`, so the variance in which the
    /// `Vec<i32>` appears is `Covariant.xform(Invariant)`, which
    /// yields `Invariant`. Now, the type `Vec<T>` is covariant with
    /// respect to its type argument `T`, and hence the variance of
    /// the `i32` here is `Invariant.xform(Covariant)`, which results
    /// (again) in `Invariant`.
    ///
    /// Example 2:
    ///
    ///     fn(*const Vec<i32>, *mut Vec<i32)
    ///
    /// The ambient variance is covariant. A `fn` type is
    /// contravariant with respect to its parameters, so the variance
    /// within which both pointer types appear is
A
Alexander Regueiro 已提交
369
    /// `Covariant.xform(Contravariant)`, or `Contravariant`. `*const
370 371
    /// T` is covariant with respect to `T`, so the variance within
    /// which the first `Vec<i32>` appears is
A
Alexander Regueiro 已提交
372
    /// `Contravariant.xform(Covariant)` or `Contravariant`. The same
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402
    /// is true for its `i32` argument. In the `*mut T` case, the
    /// variance of `Vec<i32>` is `Contravariant.xform(Invariant)`,
    /// and hence the outermost type is `Invariant` with respect to
    /// `Vec<i32>` (and its `i32` argument).
    ///
    /// Source: Figure 1 of "Taming the Wildcards:
    /// Combining Definition- and Use-Site Variance" published in PLDI'11.
    pub fn xform(self, v: ty::Variance) -> ty::Variance {
        match (self, v) {
            // Figure 1, column 1.
            (ty::Covariant, ty::Covariant) => ty::Covariant,
            (ty::Covariant, ty::Contravariant) => ty::Contravariant,
            (ty::Covariant, ty::Invariant) => ty::Invariant,
            (ty::Covariant, ty::Bivariant) => ty::Bivariant,

            // Figure 1, column 2.
            (ty::Contravariant, ty::Covariant) => ty::Contravariant,
            (ty::Contravariant, ty::Contravariant) => ty::Covariant,
            (ty::Contravariant, ty::Invariant) => ty::Invariant,
            (ty::Contravariant, ty::Bivariant) => ty::Bivariant,

            // Figure 1, column 3.
            (ty::Invariant, _) => ty::Invariant,

            // Figure 1, column 4.
            (ty::Bivariant, _) => ty::Bivariant,
        }
    }
}

403 404 405
// Contains information needed to resolve types and (in the future) look up
// the types of AST nodes.
#[derive(Copy, Clone, PartialEq, Eq, Hash)]
406
pub struct CReaderCacheKey {
407 408 409 410
    pub cnum: CrateNum,
    pub pos: usize,
}

411 412 413 414
// Flags that we track on types. These flags are propagated upwards
// through the type during type construction, so that we can quickly
// check whether the type has various kinds of types in it without
// recursing over the type itself.
415
bitflags! {
416 417 418 419 420
    pub struct TypeFlags: u32 {
        const HAS_PARAMS         = 1 << 0;
        const HAS_SELF           = 1 << 1;
        const HAS_TY_INFER       = 1 << 2;
        const HAS_RE_INFER       = 1 << 3;
421
        const HAS_RE_PLACEHOLDER = 1 << 4;
422 423 424 425 426

        /// Does this have any `ReEarlyBound` regions? Used to
        /// determine whether substitition is required, since those
        /// represent regions that are bound in a `ty::Generics` and
        /// hence may be substituted.
427
        const HAS_RE_EARLY_BOUND = 1 << 5;
428 429 430

        /// Does this have any region that "appears free" in the type?
        /// Basically anything but `ReLateBound` and `ReErased`.
431
        const HAS_FREE_REGIONS   = 1 << 6;
432 433

        /// Is an error type reachable?
434 435
        const HAS_TY_ERR         = 1 << 7;
        const HAS_PROJECTION     = 1 << 8;
J
John Kåre Alsaker 已提交
436 437

        // FIXME: Rename this to the actual property since it's used for generators too
438
        const HAS_TY_CLOSURE     = 1 << 9;
439

440 441
        /// `true` if there are "names" of types and regions and so forth
        /// that are local to a particular fn
442
        const HAS_FREE_LOCAL_NAMES    = 1 << 10;
443

444 445
        /// Present if the type belongs in a local type context.
        /// Only set for Infer other than Fresh.
446
        const KEEP_IN_LOCAL_TCX  = 1 << 11;
447

448 449
        // Is there a projection that does not involve a bound region?
        // Currently we can't normalize projections w/ bound regions.
450
        const HAS_NORMALIZABLE_PROJECTION = 1 << 12;
451

452 453
        /// Does this have any `ReLateBound` regions? Used to check
        /// if a global bound is safe to evaluate.
454
        const HAS_RE_LATE_BOUND = 1 << 13;
455

S
scalexm 已提交
456 457
        const HAS_TY_PLACEHOLDER = 1 << 14;

V
varkor 已提交
458
        const HAS_CT_INFER = 1 << 15;
V
varkor 已提交
459
        const HAS_CT_PLACEHOLDER = 1 << 16;
V
varkor 已提交
460

461 462
        const NEEDS_SUBST        = TypeFlags::HAS_PARAMS.bits |
                                   TypeFlags::HAS_SELF.bits |
463
                                   TypeFlags::HAS_RE_EARLY_BOUND.bits;
464

465 466 467
        /// Flags representing the nominal content of a type,
        /// computed by FlagsComputation. If you add a new nominal
        /// flag, it should be added here too.
468 469 470 471
        const NOMINAL_FLAGS     = TypeFlags::HAS_PARAMS.bits |
                                  TypeFlags::HAS_SELF.bits |
                                  TypeFlags::HAS_TY_INFER.bits |
                                  TypeFlags::HAS_RE_INFER.bits |
V
varkor 已提交
472
                                  TypeFlags::HAS_CT_INFER.bits |
473
                                  TypeFlags::HAS_RE_PLACEHOLDER.bits |
474 475
                                  TypeFlags::HAS_RE_EARLY_BOUND.bits |
                                  TypeFlags::HAS_FREE_REGIONS.bits |
476
                                  TypeFlags::HAS_TY_ERR.bits |
477
                                  TypeFlags::HAS_PROJECTION.bits |
478
                                  TypeFlags::HAS_TY_CLOSURE.bits |
479
                                  TypeFlags::HAS_FREE_LOCAL_NAMES.bits |
480
                                  TypeFlags::KEEP_IN_LOCAL_TCX.bits |
S
scalexm 已提交
481
                                  TypeFlags::HAS_RE_LATE_BOUND.bits |
V
varkor 已提交
482 483
                                  TypeFlags::HAS_TY_PLACEHOLDER.bits |
                                  TypeFlags::HAS_CT_PLACEHOLDER.bits;
484
    }
485 486
}

487
pub struct TyS<'tcx> {
V
varkor 已提交
488
    pub sty: TyKind<'tcx>,
489
    pub flags: TypeFlags,
490

491 492 493 494 495 496 497 498
    /// This is a kind of confusing thing: it stores the smallest
    /// binder such that
    ///
    /// (a) the binder itself captures nothing but
    /// (b) all the late-bound things within the type are captured
    ///     by some sub-binder.
    ///
    /// So, for a type without any late-bound things, like `u32`, this
499
    /// will be *innermost*, because that is the innermost binder that
500
    /// captures nothing. But for a type `&'D u32`, where `'D` is a
A
Alexander Regueiro 已提交
501
    /// late-bound region with De Bruijn index `D`, this would be `D + 1`
502 503
    /// -- the binder itself does not capture `D`, but `D` is captured
    /// by an inner binder.
504
    ///
505
    /// We call this concept an "exclusive" binder `D` because all
A
Alexander Regueiro 已提交
506
    /// De Bruijn indices within the type are contained within `0..D`
507
    /// (exclusive).
508
    outer_exclusive_binder: ty::DebruijnIndex,
509 510
}

511 512 513 514
// `TyS` is used a lot. Make sure it doesn't unintentionally get bigger.
#[cfg(target_arch = "x86_64")]
static_assert!(MEM_SIZE_OF_TY_S: ::std::mem::size_of::<TyS<'_>>() == 32);

515 516 517 518 519 520 521 522 523 524 525 526
impl<'tcx> Ord for TyS<'tcx> {
    fn cmp(&self, other: &TyS<'tcx>) -> Ordering {
        self.sty.cmp(&other.sty)
    }
}

impl<'tcx> PartialOrd for TyS<'tcx> {
    fn partial_cmp(&self, other: &TyS<'tcx>) -> Option<Ordering> {
        Some(self.sty.cmp(&other.sty))
    }
}

527
impl<'tcx> PartialEq for TyS<'tcx> {
528
    #[inline]
529
    fn eq(&self, other: &TyS<'tcx>) -> bool {
530
        ptr::eq(self, other)
531 532
    }
}
533
impl<'tcx> Eq for TyS<'tcx> {}
534

A
Alex Crichton 已提交
535 536
impl<'tcx> Hash for TyS<'tcx> {
    fn hash<H: Hasher>(&self, s: &mut H) {
537
        (self as *const TyS<'_>).hash(s)
A
Alex Crichton 已提交
538 539
    }
}
540

G
Guillaume Gomez 已提交
541 542 543
impl<'tcx> TyS<'tcx> {
    pub fn is_primitive_ty(&self) -> bool {
        match self.sty {
544
            TyKind::Bool |
545 546 547 548 549 550 551 552
            TyKind::Char |
            TyKind::Int(_) |
            TyKind::Uint(_) |
            TyKind::Float(_) |
            TyKind::Infer(InferTy::IntVar(_)) |
            TyKind::Infer(InferTy::FloatVar(_)) |
            TyKind::Infer(InferTy::FreshIntTy(_)) |
            TyKind::Infer(InferTy::FreshFloatTy(_)) => true,
V
varkor 已提交
553
            TyKind::Ref(_, x, _) => x.is_primitive_ty(),
G
Guillaume Gomez 已提交
554 555 556
            _ => false,
        }
    }
557 558 559

    pub fn is_suggestable(&self) -> bool {
        match self.sty {
560
            TyKind::Opaque(..) |
V
varkor 已提交
561 562 563 564 565 566
            TyKind::FnDef(..) |
            TyKind::FnPtr(..) |
            TyKind::Dynamic(..) |
            TyKind::Closure(..) |
            TyKind::Infer(..) |
            TyKind::Projection(..) => false,
567 568 569
            _ => true,
        }
    }
G
Guillaume Gomez 已提交
570 571
}

572
impl<'a, 'gcx> HashStable<StableHashingContext<'a>> for ty::TyS<'gcx> {
573
    fn hash_stable<W: StableHasherResult>(&self,
574
                                          hcx: &mut StableHashingContext<'a>,
575 576 577 578 579 580 581
                                          hasher: &mut StableHasher<W>) {
        let ty::TyS {
            ref sty,

            // The other fields just provide fast access to information that is
            // also contained in `sty`, so no need to hash them.
            flags: _,
582 583

            outer_exclusive_binder: _,
584 585 586 587 588 589
        } = *self;

        sty.hash_stable(hcx, hasher);
    }
}

590
pub type Ty<'tcx> = &'tcx TyS<'tcx>;
591

592 593
impl<'tcx> serialize::UseSpecializedEncodable for Ty<'tcx> {}
impl<'tcx> serialize::UseSpecializedDecodable for Ty<'tcx> {}
594

595 596
pub type CanonicalTy<'gcx> = Canonical<'gcx, Ty<'gcx>>;

J
John Kåre Alsaker 已提交
597
extern {
V
varkor 已提交
598 599
    /// A dummy type used to force List to by unsized without requiring fat pointers
    type OpaqueListContents;
J
John Kåre Alsaker 已提交
600 601
}

A
Adam Perry 已提交
602
/// A wrapper for slices with the additional invariant
603 604
/// that the slice is interned and no other slice with
/// the same contents can exist in the same context.
J
John Kåre Alsaker 已提交
605
/// This means we can use pointer for both
606
/// equality comparisons and hashing.
V
varkor 已提交
607
/// Note: `Slice` was already taken by the `Ty`.
J
John Kåre Alsaker 已提交
608
#[repr(C)]
V
varkor 已提交
609
pub struct List<T> {
J
John Kåre Alsaker 已提交
610 611
    len: usize,
    data: [T; 0],
V
varkor 已提交
612
    opaque: OpaqueListContents,
J
John Kåre Alsaker 已提交
613 614
}

V
varkor 已提交
615
unsafe impl<T: Sync> Sync for List<T> {}
J
John Kåre Alsaker 已提交
616

V
varkor 已提交
617
impl<T: Copy> List<T> {
J
John Kåre Alsaker 已提交
618
    #[inline]
V
varkor 已提交
619
    fn from_arena<'tcx>(arena: &'tcx SyncDroplessArena, slice: &[T]) -> &'tcx List<T> {
J
John Kåre Alsaker 已提交
620 621 622 623
        assert!(!mem::needs_drop::<T>());
        assert!(mem::size_of::<T>() != 0);
        assert!(slice.len() != 0);

J
John Kåre Alsaker 已提交
624 625 626 627 628 629
        // Align up the size of the len (usize) field
        let align = mem::align_of::<T>();
        let align_mask = align - 1;
        let offset = mem::size_of::<usize>();
        let offset = (offset + align_mask) & !align_mask;

J
John Kåre Alsaker 已提交
630 631
        let size = offset + slice.len() * mem::size_of::<T>();

J
John Kåre Alsaker 已提交
632
        let mem = arena.alloc_raw(
J
John Kåre Alsaker 已提交
633
            size,
J
John Kåre Alsaker 已提交
634
            cmp::max(mem::align_of::<T>(), mem::align_of::<usize>()));
J
John Kåre Alsaker 已提交
635
        unsafe {
V
varkor 已提交
636
            let result = &mut *(mem.as_mut_ptr() as *mut List<T>);
J
John Kåre Alsaker 已提交
637
            // Write the length
J
John Kåre Alsaker 已提交
638
            result.len = slice.len();
J
John Kåre Alsaker 已提交
639 640

            // Write the elements
J
John Kåre Alsaker 已提交
641
            let arena_slice = slice::from_raw_parts_mut(result.data.as_mut_ptr(), result.len);
J
John Kåre Alsaker 已提交
642 643
            arena_slice.copy_from_slice(slice);

J
John Kåre Alsaker 已提交
644
            result
J
John Kåre Alsaker 已提交
645 646 647 648
        }
    }
}

V
varkor 已提交
649
impl<T: fmt::Debug> fmt::Debug for List<T> {
650
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
J
John Kåre Alsaker 已提交
651 652 653 654
        (**self).fmt(f)
    }
}

V
varkor 已提交
655
impl<T: Encodable> Encodable for List<T> {
J
John Kåre Alsaker 已提交
656 657 658 659 660
    #[inline]
    fn encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
        (**self).encode(s)
    }
}
661

V
varkor 已提交
662 663
impl<T> Ord for List<T> where T: Ord {
    fn cmp(&self, other: &List<T>) -> Ordering {
664
        if self == other { Ordering::Equal } else {
J
John Kåre Alsaker 已提交
665
            <[T] as Ord>::cmp(&**self, &**other)
666 667 668 669
        }
    }
}

V
varkor 已提交
670 671
impl<T> PartialOrd for List<T> where T: PartialOrd {
    fn partial_cmp(&self, other: &List<T>) -> Option<Ordering> {
672
        if self == other { Some(Ordering::Equal) } else {
J
John Kåre Alsaker 已提交
673
            <[T] as PartialOrd>::partial_cmp(&**self, &**other)
674 675 676 677
        }
    }
}

V
varkor 已提交
678
impl<T: PartialEq> PartialEq for List<T> {
679
    #[inline]
V
varkor 已提交
680
    fn eq(&self, other: &List<T>) -> bool {
681
        ptr::eq(self, other)
682 683
    }
}
V
varkor 已提交
684
impl<T: Eq> Eq for List<T> {}
685

V
varkor 已提交
686
impl<T> Hash for List<T> {
J
John Kåre Alsaker 已提交
687
    #[inline]
688
    fn hash<H: Hasher>(&self, s: &mut H) {
V
varkor 已提交
689
        (self as *const List<T>).hash(s)
690 691 692
    }
}

V
varkor 已提交
693
impl<T> Deref for List<T> {
694
    type Target = [T];
J
John Kåre Alsaker 已提交
695
    #[inline(always)]
696
    fn deref(&self) -> &[T] {
J
John Kåre Alsaker 已提交
697
        unsafe {
J
John Kåre Alsaker 已提交
698
            slice::from_raw_parts(self.data.as_ptr(), self.len)
J
John Kåre Alsaker 已提交
699
        }
700 701 702
    }
}

V
varkor 已提交
703
impl<'a, T> IntoIterator for &'a List<T> {
704 705
    type Item = &'a T;
    type IntoIter = <&'a [T] as IntoIterator>::IntoIter;
J
John Kåre Alsaker 已提交
706
    #[inline(always)]
707 708 709 710 711
    fn into_iter(self) -> Self::IntoIter {
        self[..].iter()
    }
}

V
varkor 已提交
712
impl<'tcx> serialize::UseSpecializedDecodable for &'tcx List<Ty<'tcx>> {}
713

V
varkor 已提交
714
impl<T> List<T> {
J
John Kåre Alsaker 已提交
715
    #[inline(always)]
V
varkor 已提交
716
    pub fn empty<'a>() -> &'a List<T> {
J
John Kåre Alsaker 已提交
717 718 719 720
        #[repr(align(64), C)]
        struct EmptySlice([u8; 64]);
        static EMPTY_SLICE: EmptySlice = EmptySlice([0; 64]);
        assert!(mem::align_of::<T>() <= 64);
721
        unsafe {
V
varkor 已提交
722
            &*(&EMPTY_SLICE as *const _ as *const List<T>)
723 724 725 726
        }
    }
}

727
#[derive(Clone, Copy, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
728 729 730 731
pub struct UpvarPath {
    pub hir_id: hir::HirId,
}

A
Alexander Regueiro 已提交
732 733 734
/// Upvars do not get their own `NodeId`. Instead, we use the pair of
/// the original var ID (that is, the root variable that is referenced
/// by the upvar) and the ID of the closure expression.
735
#[derive(Clone, Copy, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
736
pub struct UpvarId {
737
    pub var_path: UpvarPath,
738
    pub closure_expr_id: LocalDefId,
739 740
}

741
#[derive(Clone, PartialEq, Eq, Hash, Debug, RustcEncodable, RustcDecodable, Copy, HashStable)]
742 743 744 745
pub enum BorrowKind {
    /// Data must be immutable and is aliasable.
    ImmBorrow,

A
Alexander Regueiro 已提交
746
    /// Data must be immutable but not aliasable. This kind of borrow
747
    /// cannot currently be expressed by the user and is used only in
K
king6cong 已提交
748
    /// implicit closure bindings. It is needed when the closure
749 750
    /// is borrowing or mutating a mutable referent, e.g.:
    ///
751
    ///    let x: &mut isize = ...;
752 753 754 755 756
    ///    let y = || *x += 5;
    ///
    /// If we were to try to translate this closure into a more explicit
    /// form, we'd encounter an error with the code as written:
    ///
757 758
    ///    struct Env { x: & &mut isize }
    ///    let x: &mut isize = ...;
759 760 761 762 763 764 765
    ///    let y = (&mut Env { &x }, fn_ptr);  // Closure is pair of env and fn
    ///    fn fn_ptr(env: &mut Env) { **env.x += 5; }
    ///
    /// This is then illegal because you cannot mutate a `&mut` found
    /// in an aliasable location. To solve, you'd have to translate with
    /// an `&mut` borrow:
    ///
766 767
    ///    struct Env { x: & &mut isize }
    ///    let x: &mut isize = ...;
768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786
    ///    let y = (&mut Env { &mut x }, fn_ptr); // changed from &x to &mut x
    ///    fn fn_ptr(env: &mut Env) { **env.x += 5; }
    ///
    /// Now the assignment to `**env.x` is legal, but creating a
    /// mutable pointer to `x` is not because `x` is not mutable. We
    /// could fix this by declaring `x` as `let mut x`. This is ok in
    /// user code, if awkward, but extra weird for closures, since the
    /// borrow is hidden.
    ///
    /// So we introduce a "unique imm" borrow -- the referent is
    /// immutable, but not aliasable. This solves the problem. For
    /// simplicity, we don't give users the way to express this
    /// borrow, it's just used when translating closures.
    UniqueImmBorrow,

    /// Data is mutable and not aliasable.
    MutBorrow
}

787 788
/// Information describing the capture of an upvar. This is computed
/// during `typeck`, specifically by `regionck`.
789
#[derive(PartialEq, Clone, Debug, Copy, RustcEncodable, RustcDecodable, HashStable)]
790
pub enum UpvarCapture<'tcx> {
791 792 793 794 795 796
    /// Upvar is captured by value. This is always true when the
    /// closure is labeled `move`, but can also be true in other cases
    /// depending on inference.
    ByValue,

    /// Upvar is captured by reference.
797
    ByRef(UpvarBorrow<'tcx>),
798 799
}

800
#[derive(PartialEq, Clone, Copy, RustcEncodable, RustcDecodable, HashStable)]
801
pub struct UpvarBorrow<'tcx> {
802 803 804
    /// The kind of borrow: by-ref upvars have access to shared
    /// immutable borrows, which are not part of the normal language
    /// syntax.
805
    pub kind: BorrowKind,
806 807

    /// Region of the resulting reference.
N
Niko Matsakis 已提交
808
    pub region: ty::Region<'tcx>,
809 810
}

811
pub type UpvarListMap = FxHashMap<DefId, Vec<UpvarId>>;
812
pub type UpvarCaptureMap<'tcx> = FxHashMap<UpvarId, UpvarCapture<'tcx>>;
813

814 815
#[derive(Copy, Clone)]
pub struct ClosureUpvar<'tcx> {
816
    pub res: Res,
817 818 819 820
    pub span: Span,
    pub ty: Ty<'tcx>,
}

821
#[derive(Clone, Copy, PartialEq, Eq)]
822
pub enum IntVarValue {
823 824
    IntType(ast::IntTy),
    UintType(ast::UintTy),
825 826
}

827 828 829
#[derive(Clone, Copy, PartialEq, Eq)]
pub struct FloatVarValue(pub ast::FloatTy);

830
impl ty::EarlyBoundRegion {
831
    pub fn to_bound_region(&self) -> ty::BoundRegion {
832
        ty::BoundRegion::BrNamed(self.def_id, self.name)
833
    }
834 835 836 837 838 839

    /// Does this early bound region have a name? Early bound regions normally
    /// always have names except when using anonymous lifetimes (`'_`).
    pub fn has_name(&self) -> bool {
        self.name != keywords::UnderscoreLifetime.name().as_interned_str()
    }
840 841
}

842
#[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
843
pub enum GenericParamDefKind {
V
varkor 已提交
844
    Lifetime,
845 846 847 848
    Type {
        has_default: bool,
        object_lifetime_default: ObjectLifetimeDefault,
        synthetic: Option<hir::SyntheticTyParamKind>,
V
varkor 已提交
849 850
    },
    Const,
V
varkor 已提交
851 852
}

853
#[derive(Clone, RustcEncodable, RustcDecodable, HashStable)]
854 855 856 857
pub struct GenericParamDef {
    pub name: InternedString,
    pub def_id: DefId,
    pub index: u32,
V
varkor 已提交
858 859 860 861 862 863

    /// `pure_wrt_drop`, set by the (unsafe) `#[may_dangle]` attribute
    /// on generic parameter `'a`/`T`, asserts data behind the parameter
    /// `'a`/`T` won't be accessed during the parent type's `Drop` impl.
    pub pure_wrt_drop: bool,

864 865 866
    pub kind: GenericParamDefKind,
}

V
varkor 已提交
867
impl GenericParamDef {
868
    pub fn to_early_bound_region_data(&self) -> ty::EarlyBoundRegion {
L
ljedrz 已提交
869 870 871 872 873
        if let GenericParamDefKind::Lifetime = self.kind {
            ty::EarlyBoundRegion {
                def_id: self.def_id,
                index: self.index,
                name: self.name,
874
            }
L
ljedrz 已提交
875 876
        } else {
            bug!("cannot convert a non-lifetime parameter def to an early bound region")
877 878 879 880
        }
    }

    pub fn to_bound_region(&self) -> ty::BoundRegion {
L
ljedrz 已提交
881 882 883 884
        if let GenericParamDefKind::Lifetime = self.kind {
            self.to_early_bound_region_data().to_bound_region()
        } else {
            bug!("cannot convert a non-lifetime parameter def to an early bound region")
V
varkor 已提交
885 886 887 888
        }
    }
}

V
varkor 已提交
889
#[derive(Default)]
890 891 892
pub struct GenericParamCount {
    pub lifetimes: usize,
    pub types: usize,
V
varkor 已提交
893
    pub consts: usize,
894 895
}

896
/// Information about the formal type/lifetime parameters associated
897
/// with an item or method. Analogous to `hir::Generics`.
A
Ariel Ben-Yehuda 已提交
898
///
899 900
/// The ordering of parameters is the same as in `Subst` (excluding child generics):
/// `Self` (optionally), `Lifetime` params..., `Type` params...
901
#[derive(Clone, Debug, RustcEncodable, RustcDecodable, HashStable)]
902
pub struct Generics {
903
    pub parent: Option<DefId>,
V
varkor 已提交
904
    pub parent_count: usize,
V
varkor 已提交
905
    pub params: Vec<GenericParamDef>,
906

V
varkor 已提交
907
    /// Reverse map to the `index` field of each `GenericParamDef`
908
    #[stable_hasher(ignore)]
909
    pub param_def_id_to_index: FxHashMap<DefId, u32>,
910

911
    pub has_self: bool,
912
    pub has_late_bound_regions: Option<Span>,
913 914
}

915
impl<'a, 'gcx, 'tcx> Generics {
916
    pub fn count(&self) -> usize {
V
varkor 已提交
917
        self.parent_count + self.params.len()
918
    }
919

V
varkor 已提交
920
    pub fn own_counts(&self) -> GenericParamCount {
921 922 923
        // We could cache this as a property of `GenericParamCount`, but
        // the aim is to refactor this away entirely eventually and the
        // presence of this method will be a constant reminder.
V
varkor 已提交
924
        let mut own_counts: GenericParamCount = Default::default();
925

V
varkor 已提交
926
        for param in &self.params {
927
            match param.kind {
V
varkor 已提交
928
                GenericParamDefKind::Lifetime => own_counts.lifetimes += 1,
929
                GenericParamDefKind::Type { .. } => own_counts.types += 1,
V
varkor 已提交
930
                GenericParamDefKind::Const => own_counts.consts += 1,
931 932 933
            };
        }

V
varkor 已提交
934
        own_counts
935 936
    }

937
    pub fn requires_monomorphization(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>) -> bool {
V
varkor 已提交
938 939
        if self.own_requires_monomorphization() {
            return true;
V
varkor 已提交
940
        }
V
varkor 已提交
941

V
varkor 已提交
942 943
        if let Some(parent_def_id) = self.parent {
            let parent = tcx.generics_of(parent_def_id);
944
            parent.requires_monomorphization(tcx)
V
varkor 已提交
945 946 947
        } else {
            false
        }
V
varkor 已提交
948 949
    }

V
varkor 已提交
950 951 952 953 954 955 956 957 958 959
    pub fn own_requires_monomorphization(&self) -> bool {
        for param in &self.params {
            match param.kind {
                GenericParamDefKind::Type { .. } | GenericParamDefKind::Const => return true,
                GenericParamDefKind::Lifetime => {}
            }
        }
        false
    }

960 961 962
    pub fn region_param(&'tcx self,
                        param: &EarlyBoundRegion,
                        tcx: TyCtxt<'a, 'gcx, 'tcx>)
963
                        -> &'tcx GenericParamDef
964
    {
V
varkor 已提交
965
        if let Some(index) = param.index.checked_sub(self.parent_count as u32) {
V
varkor 已提交
966
            let param = &self.params[index as usize];
967
            match param.kind {
V
varkor 已提交
968
                GenericParamDefKind::Lifetime => param,
V
varkor 已提交
969
                _ => bug!("expected lifetime parameter, but found another generic parameter")
V
varkor 已提交
970
            }
971
        } else {
972
            tcx.generics_of(self.parent.expect("parent_count > 0 but no parent?"))
L
ljedrz 已提交
973
               .region_param(param, tcx)
974
        }
975 976
    }

977
    /// Returns the `GenericParamDef` associated with this `ParamTy`.
978 979
    pub fn type_param(&'tcx self,
                      param: &ParamTy,
A
Ariel Ben-Yehuda 已提交
980
                      tcx: TyCtxt<'a, 'gcx, 'tcx>)
981
                      -> &'tcx GenericParamDef {
982
        if let Some(index) = param.idx.checked_sub(self.parent_count as u32) {
V
varkor 已提交
983 984
            let param = &self.params[index as usize];
            match param.kind {
V
varkor 已提交
985
                GenericParamDefKind::Type { .. } => param,
V
varkor 已提交
986 987
                _ => bug!("expected type parameter, but found another generic parameter")
            }
988
        } else {
989
            tcx.generics_of(self.parent.expect("parent_count > 0 but no parent?"))
L
ljedrz 已提交
990
               .type_param(param, tcx)
991
        }
992
    }
V
varkor 已提交
993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009

    /// Returns the `ConstParameterDef` associated with this `ParamConst`.
    pub fn const_param(&'tcx self,
                       param: &ParamConst,
                       tcx: TyCtxt<'a, 'gcx, 'tcx>)
                       -> &GenericParamDef {
        if let Some(index) = param.index.checked_sub(self.parent_count as u32) {
            let param = &self.params[index as usize];
            match param.kind {
                GenericParamDefKind::Const => param,
                _ => bug!("expected const parameter, but found another generic parameter")
            }
        } else {
            tcx.generics_of(self.parent.expect("parent_count>0 but no parent?"))
                .const_param(param, tcx)
        }
    }
1010 1011
}

1012
/// Bounds on generics.
1013
#[derive(Clone, Default, Debug, HashStable)]
1014
pub struct GenericPredicates<'tcx> {
1015
    pub parent: Option<DefId>,
1016
    pub predicates: Vec<(Predicate<'tcx>, Span)>,
1017 1018
}

1019 1020 1021
impl<'tcx> serialize::UseSpecializedEncodable for GenericPredicates<'tcx> {}
impl<'tcx> serialize::UseSpecializedDecodable for GenericPredicates<'tcx> {}

1022
impl<'a, 'gcx, 'tcx> GenericPredicates<'tcx> {
C
csmoe 已提交
1023
    pub fn instantiate(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>, substs: SubstsRef<'tcx>)
1024
                       -> InstantiatedPredicates<'tcx> {
1025 1026 1027 1028
        let mut instantiated = InstantiatedPredicates::empty();
        self.instantiate_into(tcx, &mut instantiated, substs);
        instantiated
    }
1029

C
csmoe 已提交
1030
    pub fn instantiate_own(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>, substs: SubstsRef<'tcx>)
1031
                           -> InstantiatedPredicates<'tcx> {
1032
        InstantiatedPredicates {
1033
            predicates: self.predicates.iter().map(|(p, _)| p.subst(tcx, substs)).collect(),
1034 1035 1036 1037 1038
        }
    }

    fn instantiate_into(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>,
                        instantiated: &mut InstantiatedPredicates<'tcx>,
C
csmoe 已提交
1039
                        substs: SubstsRef<'tcx>) {
1040
        if let Some(def_id) = self.parent {
1041
            tcx.predicates_of(def_id).instantiate_into(tcx, instantiated, substs);
1042
        }
1043 1044 1045
        instantiated.predicates.extend(
            self.predicates.iter().map(|(p, _)| p.subst(tcx, substs)),
        );
1046
    }
1047

1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059
    pub fn instantiate_identity(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>)
                                -> InstantiatedPredicates<'tcx> {
        let mut instantiated = InstantiatedPredicates::empty();
        self.instantiate_identity_into(tcx, &mut instantiated);
        instantiated
    }

    fn instantiate_identity_into(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>,
                                 instantiated: &mut InstantiatedPredicates<'tcx>) {
        if let Some(def_id) = self.parent {
            tcx.predicates_of(def_id).instantiate_identity_into(tcx, instantiated);
        }
1060
        instantiated.predicates.extend(self.predicates.iter().map(|&(p, _)| p))
1061 1062
    }

1063
    pub fn instantiate_supertrait(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>,
1064 1065 1066
                                  poly_trait_ref: &ty::PolyTraitRef<'tcx>)
                                  -> InstantiatedPredicates<'tcx>
    {
1067
        assert_eq!(self.parent, None);
1068
        InstantiatedPredicates {
1069
            predicates: self.predicates.iter().map(|(pred, _)| {
1070
                pred.subst_supertrait(tcx, poly_trait_ref)
1071
            }).collect()
1072 1073
        }
    }
1074 1075
}

1076
#[derive(Clone, Copy, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
1077
pub enum Predicate<'tcx> {
1078
    /// Corresponds to `where Foo: Bar<A,B,C>`. `Foo` here would be
N
Niko Matsakis 已提交
1079
    /// the `Self` type of the trait reference and `A`, `B`, and `C`
1080
    /// would be the type parameters.
1081
    Trait(PolyTraitPredicate<'tcx>),
1082

1083
    /// where `'a: 'b`
1084
    RegionOutlives(PolyRegionOutlivesPredicate<'tcx>),
1085

1086
    /// where `T: 'a`
1087
    TypeOutlives(PolyTypeOutlivesPredicate<'tcx>),
1088

1089 1090
    /// where `<T as TraitRef>::Name == X`, approximately.
    /// See the `ProjectionPredicate` struct for details.
1091
    Projection(PolyProjectionPredicate<'tcx>),
1092

1093
    /// no syntax: `T` well-formed
1094 1095 1096
    WellFormed(Ty<'tcx>),

    /// trait must be object-safe
N
Niko Matsakis 已提交
1097
    ObjectSafe(DefId),
1098

1099
    /// No direct syntax. May be thought of as `where T: FnFoo<...>`
1100
    /// for some substitutions `...` and `T` being a closure type.
1101
    /// Satisfied (or refuted) once we know the closure's kind.
1102
    ClosureKind(DefId, ClosureSubsts<'tcx>, ClosureKind),
N
Niko Matsakis 已提交
1103 1104 1105

    /// `T1 <: T2`
    Subtype(PolySubtypePredicate<'tcx>),
1106 1107

    /// Constant initializer must evaluate successfully.
C
csmoe 已提交
1108
    ConstEvaluatable(DefId, SubstsRef<'tcx>),
1109 1110
}

1111 1112 1113 1114 1115 1116
/// The crate outlives map is computed during typeck and contains the
/// outlives of every item in the local crate. You should not use it
/// directly, because to do so will make your pass dependent on the
/// HIR of every item in the local crate. Instead, use
/// `tcx.inferred_outlives_of()` to get the outlives for a *particular*
/// item.
1117
#[derive(HashStable)]
1118 1119 1120 1121
pub struct CratePredicatesMap<'tcx> {
    /// For each struct with outlive bounds, maps to a vector of the
    /// predicate of its outlive bounds. If an item has no outlives
    /// bounds, it will have no entry.
J
John Kåre Alsaker 已提交
1122
    pub predicates: FxHashMap<DefId, &'tcx [ty::Predicate<'tcx>]>,
1123 1124
}

1125 1126 1127 1128 1129 1130
impl<'tcx> AsRef<Predicate<'tcx>> for Predicate<'tcx> {
    fn as_ref(&self) -> &Predicate<'tcx> {
        self
    }
}

1131
impl<'a, 'gcx, 'tcx> Predicate<'tcx> {
1132
    /// Performs a substitution suitable for going from a
1133 1134
    /// poly-trait-ref to supertraits that must hold if that
    /// poly-trait-ref holds. This is slightly different from a normal
A
Alexander Regueiro 已提交
1135
    /// substitution in terms of what happens with bound regions. See
1136
    /// lengthy comment below for details.
1137
    pub fn subst_supertrait(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>,
1138 1139 1140 1141 1142 1143 1144 1145
                            trait_ref: &ty::PolyTraitRef<'tcx>)
                            -> ty::Predicate<'tcx>
    {
        // The interaction between HRTB and supertraits is not entirely
        // obvious. Let me walk you (and myself) through an example.
        //
        // Let's start with an easy case. Consider two traits:
        //
1146
        //     trait Foo<'a>: Bar<'a,'a> { }
1147 1148
        //     trait Bar<'b,'c> { }
        //
1149 1150
        // Now, if we have a trait reference `for<'x> T: Foo<'x>`, then
        // we can deduce that `for<'x> T: Bar<'x,'x>`. Basically, if we
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162
        // knew that `Foo<'x>` (for any 'x) then we also know that
        // `Bar<'x,'x>` (for any 'x). This more-or-less falls out from
        // normal substitution.
        //
        // In terms of why this is sound, the idea is that whenever there
        // is an impl of `T:Foo<'a>`, it must show that `T:Bar<'a,'a>`
        // holds.  So if there is an impl of `T:Foo<'a>` that applies to
        // all `'a`, then we must know that `T:Bar<'a,'a>` holds for all
        // `'a`.
        //
        // Another example to be careful of is this:
        //
1163
        //     trait Foo1<'a>: for<'b> Bar1<'a,'b> { }
1164 1165
        //     trait Bar1<'b,'c> { }
        //
1166 1167
        // Here, if we have `for<'x> T: Foo1<'x>`, then what do we know?
        // The answer is that we know `for<'x,'b> T: Bar1<'x,'b>`. The
1168
        // reason is similar to the previous example: any impl of
1169
        // `T:Foo1<'x>` must show that `for<'b> T: Bar1<'x, 'b>`.  So
1170 1171 1172 1173 1174 1175
        // basically we would want to collapse the bound lifetimes from
        // the input (`trait_ref`) and the supertraits.
        //
        // To achieve this in practice is fairly straightforward. Let's
        // consider the more complicated scenario:
        //
1176 1177
        // - We start out with `for<'x> T: Foo1<'x>`. In this case, `'x`
        //   has a De Bruijn index of 1. We want to produce `for<'x,'b> T: Bar1<'x,'b>`,
1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200
        //   where both `'x` and `'b` would have a DB index of 1.
        //   The substitution from the input trait-ref is therefore going to be
        //   `'a => 'x` (where `'x` has a DB index of 1).
        // - The super-trait-ref is `for<'b> Bar1<'a,'b>`, where `'a` is an
        //   early-bound parameter and `'b' is a late-bound parameter with a
        //   DB index of 1.
        // - If we replace `'a` with `'x` from the input, it too will have
        //   a DB index of 1, and thus we'll have `for<'x,'b> Bar1<'x,'b>`
        //   just as we wanted.
        //
        // There is only one catch. If we just apply the substitution `'a
        // => 'x` to `for<'b> Bar1<'a,'b>`, the substitution code will
        // adjust the DB index because we substituting into a binder (it
        // tries to be so smart...) resulting in `for<'x> for<'b>
        // Bar1<'x,'b>` (we have no syntax for this, so use your
        // imagination). Basically the 'x will have DB index of 2 and 'b
        // will have DB index of 1. Not quite what we want. So we apply
        // the substitution to the *contents* of the trait reference,
        // rather than the trait reference itself (put another way, the
        // substitution code expects equal binding levels in the values
        // from the substitution and the value being substituted into, and
        // this trick achieves that).

1201
        let substs = &trait_ref.skip_binder().substs;
1202
        match *self {
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
            Predicate::Trait(ref binder) =>
                Predicate::Trait(binder.map_bound(|data| data.subst(tcx, substs))),
            Predicate::Subtype(ref binder) =>
                Predicate::Subtype(binder.map_bound(|data| data.subst(tcx, substs))),
            Predicate::RegionOutlives(ref binder) =>
                Predicate::RegionOutlives(binder.map_bound(|data| data.subst(tcx, substs))),
            Predicate::TypeOutlives(ref binder) =>
                Predicate::TypeOutlives(binder.map_bound(|data| data.subst(tcx, substs))),
            Predicate::Projection(ref binder) =>
                Predicate::Projection(binder.map_bound(|data| data.subst(tcx, substs))),
1213 1214 1215 1216
            Predicate::WellFormed(data) =>
                Predicate::WellFormed(data.subst(tcx, substs)),
            Predicate::ObjectSafe(trait_def_id) =>
                Predicate::ObjectSafe(trait_def_id),
1217 1218
            Predicate::ClosureKind(closure_def_id, closure_substs, kind) =>
                Predicate::ClosureKind(closure_def_id, closure_substs.subst(tcx, substs), kind),
1219 1220
            Predicate::ConstEvaluatable(def_id, const_substs) =>
                Predicate::ConstEvaluatable(def_id, const_substs.subst(tcx, substs)),
1221 1222 1223 1224
        }
    }
}

1225
#[derive(Clone, Copy, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
1226
pub struct TraitPredicate<'tcx> {
1227
    pub trait_ref: TraitRef<'tcx>
1228
}
1229

1230 1231 1232
pub type PolyTraitPredicate<'tcx> = ty::Binder<TraitPredicate<'tcx>>;

impl<'tcx> TraitPredicate<'tcx> {
N
Niko Matsakis 已提交
1233
    pub fn def_id(&self) -> DefId {
1234 1235 1236
        self.trait_ref.def_id
    }

1237
    pub fn input_types<'a>(&'a self) -> impl DoubleEndedIterator<Item=Ty<'tcx>> + 'a {
1238
        self.trait_ref.input_types()
1239 1240 1241 1242 1243 1244 1245 1246
    }

    pub fn self_ty(&self) -> Ty<'tcx> {
        self.trait_ref.self_ty()
    }
}

impl<'tcx> PolyTraitPredicate<'tcx> {
N
Niko Matsakis 已提交
1247
    pub fn def_id(&self) -> DefId {
1248
        // ok to skip binder since trait def-id does not care about regions
1249
        self.skip_binder().def_id()
1250
    }
1251 1252
}

1253 1254
#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord,
         Hash, Debug, RustcEncodable, RustcDecodable, HashStable)]
1255
pub struct OutlivesPredicate<A,B>(pub A, pub B); // `A: B`
1256
pub type PolyOutlivesPredicate<A,B> = ty::Binder<OutlivesPredicate<A,B>>;
S
scalexm 已提交
1257 1258 1259 1260 1261 1262
pub type RegionOutlivesPredicate<'tcx> = OutlivesPredicate<ty::Region<'tcx>,
                                                           ty::Region<'tcx>>;
pub type TypeOutlivesPredicate<'tcx> = OutlivesPredicate<Ty<'tcx>,
                                                         ty::Region<'tcx>>;
pub type PolyRegionOutlivesPredicate<'tcx> = ty::Binder<RegionOutlivesPredicate<'tcx>>;
pub type PolyTypeOutlivesPredicate<'tcx> = ty::Binder<TypeOutlivesPredicate<'tcx>>;
1263

1264
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, RustcEncodable, RustcDecodable, HashStable)]
N
Niko Matsakis 已提交
1265 1266 1267 1268 1269 1270 1271
pub struct SubtypePredicate<'tcx> {
    pub a_is_expected: bool,
    pub a: Ty<'tcx>,
    pub b: Ty<'tcx>
}
pub type PolySubtypePredicate<'tcx> = ty::Binder<SubtypePredicate<'tcx>>;

1272 1273 1274
/// This kind of predicate has no *direct* correspondent in the
/// syntax, but it roughly corresponds to the syntactic forms:
///
A
Alexander Regueiro 已提交
1275
/// 1. `T: TraitRef<..., Item = Type>`
1276 1277 1278
/// 2. `<T as TraitRef<...>>::Item == Type` (NYI)
///
/// In particular, form #1 is "desugared" to the combination of a
A
Alexander Regueiro 已提交
1279
/// normal trait predicate (`T: TraitRef<...>`) and one of these
1280
/// predicates. Form #2 is a broader form in that it also permits
1281 1282
/// equality between arbitrary types. Processing an instance of
/// Form #2 eventually yields one of these `ProjectionPredicate`
1283
/// instances to normalize the LHS.
1284
#[derive(Copy, Clone, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, HashStable)]
1285 1286 1287 1288 1289 1290 1291
pub struct ProjectionPredicate<'tcx> {
    pub projection_ty: ProjectionTy<'tcx>,
    pub ty: Ty<'tcx>,
}

pub type PolyProjectionPredicate<'tcx> = Binder<ProjectionPredicate<'tcx>>;

1292
impl<'tcx> PolyProjectionPredicate<'tcx> {
1293
    /// Returns the `DefId` of the associated item being projected.
1294 1295 1296 1297
    pub fn item_def_id(&self) -> DefId {
        self.skip_binder().projection_ty.item_def_id
    }

1298
    #[inline]
1299
    pub fn to_poly_trait_ref(&self, tcx: TyCtxt<'_, '_, '_>) -> PolyTraitRef<'tcx> {
1300 1301
        // Note: unlike with `TraitRef::to_poly_trait_ref()`,
        // `self.0.trait_ref` is permitted to have escaping regions.
1302 1303 1304
        // This is because here `self` has a `Binder` and so does our
        // return value, so we are preserving the number of binding
        // levels.
1305
        self.map_bound(|predicate| predicate.projection_ty.trait_ref(tcx))
1306
    }
1307 1308

    pub fn ty(&self) -> Binder<Ty<'tcx>> {
1309 1310 1311
        self.map_bound(|predicate| predicate.ty)
    }

1312
    /// The `DefId` of the `TraitItem` for the associated type.
1313
    ///
1314 1315
    /// Note that this is not the `DefId` of the `TraitRef` containing this
    /// associated type, which is in `tcx.associated_item(projection_def_id()).container`.
1316
    pub fn projection_def_id(&self) -> DefId {
1317
        // okay to skip binder since trait def-id does not care about regions
1318
        self.skip_binder().projection_ty.item_def_id
1319
    }
1320 1321
}

1322 1323 1324 1325
pub trait ToPolyTraitRef<'tcx> {
    fn to_poly_trait_ref(&self) -> PolyTraitRef<'tcx>;
}

1326
impl<'tcx> ToPolyTraitRef<'tcx> for TraitRef<'tcx> {
1327
    fn to_poly_trait_ref(&self) -> PolyTraitRef<'tcx> {
1328
        ty::Binder::dummy(self.clone())
1329 1330 1331 1332 1333
    }
}

impl<'tcx> ToPolyTraitRef<'tcx> for PolyTraitPredicate<'tcx> {
    fn to_poly_trait_ref(&self) -> PolyTraitRef<'tcx> {
1334
        self.map_bound_ref(|trait_pred| trait_pred.trait_ref)
1335 1336 1337
    }
}

1338 1339
pub trait ToPredicate<'tcx> {
    fn to_predicate(&self) -> Predicate<'tcx>;
1340 1341
}

1342 1343
impl<'tcx> ToPredicate<'tcx> for TraitRef<'tcx> {
    fn to_predicate(&self) -> Predicate<'tcx> {
1344
        ty::Predicate::Trait(ty::Binder::dummy(ty::TraitPredicate {
1345 1346 1347 1348 1349
            trait_ref: self.clone()
        }))
    }
}

1350 1351
impl<'tcx> ToPredicate<'tcx> for PolyTraitRef<'tcx> {
    fn to_predicate(&self) -> Predicate<'tcx> {
1352
        ty::Predicate::Trait(self.to_poly_trait_predicate())
1353 1354 1355
    }
}

1356
impl<'tcx> ToPredicate<'tcx> for PolyRegionOutlivesPredicate<'tcx> {
1357
    fn to_predicate(&self) -> Predicate<'tcx> {
1358 1359 1360 1361
        Predicate::RegionOutlives(self.clone())
    }
}

1362 1363
impl<'tcx> ToPredicate<'tcx> for PolyTypeOutlivesPredicate<'tcx> {
    fn to_predicate(&self) -> Predicate<'tcx> {
1364 1365
        Predicate::TypeOutlives(self.clone())
    }
1366 1367
}

1368 1369
impl<'tcx> ToPredicate<'tcx> for PolyProjectionPredicate<'tcx> {
    fn to_predicate(&self) -> Predicate<'tcx> {
1370 1371 1372 1373
        Predicate::Projection(self.clone())
    }
}

1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418
// A custom iterator used by Predicate::walk_tys.
enum WalkTysIter<'tcx, I, J, K>
    where I: Iterator<Item = Ty<'tcx>>,
          J: Iterator<Item = Ty<'tcx>>,
          K: Iterator<Item = Ty<'tcx>>
{
    None,
    One(Ty<'tcx>),
    Two(Ty<'tcx>, Ty<'tcx>),
    Types(I),
    InputTypes(J),
    ProjectionTypes(K)
}

impl<'tcx, I, J, K> Iterator for WalkTysIter<'tcx, I, J, K>
    where I: Iterator<Item = Ty<'tcx>>,
          J: Iterator<Item = Ty<'tcx>>,
          K: Iterator<Item = Ty<'tcx>>
{
    type Item = Ty<'tcx>;

    fn next(&mut self) -> Option<Ty<'tcx>> {
        match *self {
            WalkTysIter::None => None,
            WalkTysIter::One(item) => {
                *self = WalkTysIter::None;
                Some(item)
            },
            WalkTysIter::Two(item1, item2) => {
                *self = WalkTysIter::One(item2);
                Some(item1)
            },
            WalkTysIter::Types(ref mut iter) => {
                iter.next()
            },
            WalkTysIter::InputTypes(ref mut iter) => {
                iter.next()
            },
            WalkTysIter::ProjectionTypes(ref mut iter) => {
                iter.next()
            }
        }
    }
}

1419
impl<'tcx> Predicate<'tcx> {
1420 1421 1422
    /// Iterates over the types in this predicate. Note that in all
    /// cases this is skipping over a binder, so late-bound regions
    /// with depth 0 are bound by the predicate.
1423 1424
    pub fn walk_tys(&'a self) -> impl Iterator<Item = Ty<'tcx>> + 'a {
        match *self {
1425
            ty::Predicate::Trait(ref data) => {
1426
                WalkTysIter::InputTypes(data.skip_binder().input_types())
1427
            }
1428 1429
            ty::Predicate::Subtype(binder) => {
                let SubtypePredicate { a, b, a_is_expected: _ } = binder.skip_binder();
1430
                WalkTysIter::Two(a, b)
N
Niko Matsakis 已提交
1431
            }
1432
            ty::Predicate::TypeOutlives(binder) => {
1433
                WalkTysIter::One(binder.skip_binder().0)
1434 1435
            }
            ty::Predicate::RegionOutlives(..) => {
1436
                WalkTysIter::None
1437 1438
            }
            ty::Predicate::Projection(ref data) => {
1439
                let inner = data.skip_binder();
1440 1441
                WalkTysIter::ProjectionTypes(
                    inner.projection_ty.substs.types().chain(Some(inner.ty)))
1442
            }
1443
            ty::Predicate::WellFormed(data) => {
1444
                WalkTysIter::One(data)
1445 1446
            }
            ty::Predicate::ObjectSafe(_trait_def_id) => {
1447
                WalkTysIter::None
1448
            }
1449
            ty::Predicate::ClosureKind(_closure_def_id, closure_substs, _kind) => {
1450
                WalkTysIter::Types(closure_substs.substs.types())
1451
            }
1452
            ty::Predicate::ConstEvaluatable(_, substs) => {
1453
                WalkTysIter::Types(substs.types())
1454
            }
1455
        }
1456 1457
    }

1458
    pub fn to_opt_poly_trait_ref(&self) -> Option<PolyTraitRef<'tcx>> {
1459 1460
        match *self {
            Predicate::Trait(ref t) => {
1461
                Some(t.to_poly_trait_ref())
1462
            }
1463
            Predicate::Projection(..) |
N
Niko Matsakis 已提交
1464
            Predicate::Subtype(..) |
1465
            Predicate::RegionOutlives(..) |
1466 1467
            Predicate::WellFormed(..) |
            Predicate::ObjectSafe(..) |
1468
            Predicate::ClosureKind(..) |
1469 1470
            Predicate::TypeOutlives(..) |
            Predicate::ConstEvaluatable(..) => {
1471 1472
                None
            }
1473 1474
        }
    }
1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492

    pub fn to_opt_type_outlives(&self) -> Option<PolyTypeOutlivesPredicate<'tcx>> {
        match *self {
            Predicate::TypeOutlives(data) => {
                Some(data)
            }
            Predicate::Trait(..) |
            Predicate::Projection(..) |
            Predicate::Subtype(..) |
            Predicate::RegionOutlives(..) |
            Predicate::WellFormed(..) |
            Predicate::ObjectSafe(..) |
            Predicate::ClosureKind(..) |
            Predicate::ConstEvaluatable(..) => {
                None
            }
        }
    }
1493 1494
}

S
Steve Klabnik 已提交
1495
/// Represents the bounds declared on a particular set of type
A
Alexander Regueiro 已提交
1496 1497
/// parameters. Should eventually be generalized into a flag list of
/// where-clauses. You can obtain a `InstantiatedPredicates` list from a
1498 1499 1500 1501
/// `GenericPredicates` by using the `instantiate` method. Note that this method
/// reflects an important semantic invariant of `InstantiatedPredicates`: while
/// the `GenericPredicates` are expressed in terms of the bound type
/// parameters of the impl/trait/whatever, an `InstantiatedPredicates` instance
S
Steve Klabnik 已提交
1502 1503 1504 1505 1506 1507 1508 1509
/// represented a set of bounds for some particular instantiation,
/// meaning that the generic parameters have been substituted with
/// their values.
///
/// Example:
///
///     struct Foo<T,U:Bar<T>> { ... }
///
1510
/// Here, the `GenericPredicates` for `Foo` would contain a list of bounds like
A
Alexander Regueiro 已提交
1511
/// `[[], [U:Bar<T>]]`. Now if there were some particular reference
1512 1513
/// like `Foo<isize,usize>`, then the `InstantiatedPredicates` would be `[[],
/// [usize:Bar<isize>]]`.
1514
#[derive(Clone, Debug)]
1515
pub struct InstantiatedPredicates<'tcx> {
1516
    pub predicates: Vec<Predicate<'tcx>>,
1517 1518
}

1519 1520
impl<'tcx> InstantiatedPredicates<'tcx> {
    pub fn empty() -> InstantiatedPredicates<'tcx> {
1521
        InstantiatedPredicates { predicates: vec![] }
1522 1523
    }

1524 1525
    pub fn is_empty(&self) -> bool {
        self.predicates.is_empty()
1526
    }
1527 1528
}

1529
newtype_index! {
1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
    /// "Universes" are used during type- and trait-checking in the
    /// presence of `for<..>` binders to control what sets of names are
    /// visible. Universes are arranged into a tree: the root universe
    /// contains names that are always visible. Each child then adds a new
    /// set of names that are visible, in addition to those of its parent.
    /// We say that the child universe "extends" the parent universe with
    /// new names.
    ///
    /// To make this more concrete, consider this program:
    ///
    /// ```
    /// struct Foo { }
    /// fn bar<T>(x: T) {
    ///   let y: for<'a> fn(&'a u8, Foo) = ...;
    /// }
    /// ```
    ///
    /// The struct name `Foo` is in the root universe U0. But the type
    /// parameter `T`, introduced on `bar`, is in an extended universe U1
    /// -- i.e., within `bar`, we can name both `T` and `Foo`, but outside
    /// of `bar`, we cannot name `T`. Then, within the type of `y`, the
    /// region `'a` is in a universe U2 that extends U1, because we can
    /// name it inside the fn type but not outside.
    ///
    /// Universes are used to do type- and trait-checking around these
    /// "forall" binders (also called **universal quantification**). The
    /// idea is that when, in the body of `bar`, we refer to `T` as a
    /// type, we aren't referring to any type in particular, but rather a
    /// kind of "fresh" type that is distinct from all other types we have
    /// actually declared. This is called a **placeholder** type, and we
    /// use universes to talk about this. In other words, a type name in
    /// universe 0 always corresponds to some "ground" type that the user
    /// declared, but a type name in a non-zero universe is a placeholder
    /// type -- an idealized representative of "types in general" that we
    /// use for checking generic functions.
1565 1566 1567 1568
    pub struct UniverseIndex {
        DEBUG_FORMAT = "U{}",
    }
}
1569 1570

impl_stable_hash_for!(struct UniverseIndex { private });
N
Niko Matsakis 已提交
1571 1572

impl UniverseIndex {
1573
    pub const ROOT: UniverseIndex = UniverseIndex::from_u32_const(0);
N
Niko Matsakis 已提交
1574

1575 1576
    /// Returns the "next" universe index in order -- this new index
    /// is considered to extend all previous universes. This
A
Alexander Regueiro 已提交
1577
    /// corresponds to entering a `forall` quantifier. So, for
1578
    /// example, suppose we have this type in universe `U`:
N
Niko Matsakis 已提交
1579 1580 1581 1582 1583 1584
    ///
    /// ```
    /// for<'a> fn(&'a u32)
    /// ```
    ///
    /// Once we "enter" into this `for<'a>` quantifier, we are in a
1585 1586 1587 1588
    /// new universe that extends `U` -- in this new universe, we can
    /// name the region `'a`, but that region was not nameable from
    /// `U` because it was not in scope there.
    pub fn next_universe(self) -> UniverseIndex {
1589
        UniverseIndex::from_u32(self.private.checked_add(1).unwrap())
1590 1591
    }

1592
    /// Returns `true` if `self` can name a name from `other` -- in other words,
1593
    /// if the set of names in `self` is a superset of those in
N
Niko Matsakis 已提交
1594
    /// `other` (`self >= other`).
1595 1596
    pub fn can_name(self, other: UniverseIndex) -> bool {
        self.private >= other.private
N
Niko Matsakis 已提交
1597
    }
N
Niko Matsakis 已提交
1598

1599
    /// Returns `true` if `self` cannot name some names from `other` -- in other
N
Niko Matsakis 已提交
1600 1601 1602 1603 1604
    /// words, if the set of names in `self` is a strict subset of
    /// those in `other` (`self < other`).
    pub fn cannot_name(self, other: UniverseIndex) -> bool {
        self.private < other.private
    }
N
Niko Matsakis 已提交
1605 1606
}

1607 1608 1609 1610 1611 1612 1613
/// The "placeholder index" fully defines a placeholder region.
/// Placeholder regions are identified by both a **universe** as well
/// as a "bound-region" within that universe. The `bound_region` is
/// basically a name -- distinct bound regions within the same
/// universe are just two regions with an unknown relationship to one
/// another.
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, RustcEncodable, RustcDecodable, PartialOrd, Ord)]
S
scalexm 已提交
1614
pub struct Placeholder<T> {
1615
    pub universe: UniverseIndex,
S
scalexm 已提交
1616
    pub name: T,
1617 1618
}

S
scalexm 已提交
1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
impl<'a, 'gcx, T> HashStable<StableHashingContext<'a>> for Placeholder<T>
    where T: HashStable<StableHashingContext<'a>>
{
    fn hash_stable<W: StableHasherResult>(
        &self,
        hcx: &mut StableHashingContext<'a>,
        hasher: &mut StableHasher<W>
    ) {
        self.universe.hash_stable(hcx, hasher);
        self.name.hash_stable(hcx, hasher);
    }
}

pub type PlaceholderRegion = Placeholder<BoundRegion>;

pub type PlaceholderType = Placeholder<BoundVar>;
1635

V
varkor 已提交
1636 1637
pub type PlaceholderConst = Placeholder<BoundVar>;

1638
/// When type checking, we use the `ParamEnv` to track
1639 1640
/// details about the set of where-clauses that are in scope at this
/// particular point.
1641
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash, HashStable)]
1642
pub struct ParamEnv<'tcx> {
1643 1644
    /// Obligations that the caller must satisfy. This is basically
    /// the set of bounds on the in-scope type parameters, translated
1645
    /// into Obligations, and elaborated and normalized.
V
varkor 已提交
1646
    pub caller_bounds: &'tcx List<ty::Predicate<'tcx>>,
1647

I
Irina Popa 已提交
1648
    /// Typically, this is `Reveal::UserFacing`, but during codegen we
1649 1650 1651
    /// want `Reveal::All` -- note that this is always paired with an
    /// empty environment. To get that, use `ParamEnv::reveal()`.
    pub reveal: traits::Reveal,
S
scalexm 已提交
1652 1653 1654 1655 1656

    /// If this `ParamEnv` comes from a call to `tcx.param_env(def_id)`,
    /// register that `def_id` (useful for transitioning to the chalk trait
    /// solver).
    pub def_id: Option<DefId>,
1657
}
1658

1659
impl<'tcx> ParamEnv<'tcx> {
1660
    /// Construct a trait environment suitable for contexts where
A
Alexander Regueiro 已提交
1661
    /// there are no where-clauses in scope. Hidden types (like `impl
1662 1663
    /// Trait`) are left hidden, so this is suitable for ordinary
    /// type-checking.
1664
    #[inline]
1665
    pub fn empty() -> Self {
S
scalexm 已提交
1666
        Self::new(List::empty(), Reveal::UserFacing, None)
1667 1668
    }

A
Alexander Regueiro 已提交
1669
    /// Construct a trait environment with no where-clauses in scope
1670 1671
    /// where the values of all `impl Trait` and other hidden types
    /// are revealed. This is suitable for monomorphized, post-typeck
I
Irina Popa 已提交
1672
    /// environments like codegen or doing optimizations.
1673
    ///
A
Alexander Regueiro 已提交
1674
    /// N.B., if you want to have predicates in scope, use `ParamEnv::new`,
1675
    /// or invoke `param_env.with_reveal_all()`.
1676
    #[inline]
1677
    pub fn reveal_all() -> Self {
S
scalexm 已提交
1678
        Self::new(List::empty(), Reveal::All, None)
1679 1680 1681
    }

    /// Construct a trait environment with the given set of predicates.
1682
    #[inline]
S
scalexm 已提交
1683 1684 1685 1686 1687 1688
    pub fn new(
        caller_bounds: &'tcx List<ty::Predicate<'tcx>>,
        reveal: Reveal,
        def_id: Option<DefId>
    ) -> Self {
        ty::ParamEnv { caller_bounds, reveal, def_id }
1689 1690 1691 1692
    }

    /// Returns a new parameter environment with the same clauses, but
    /// which "reveals" the true results of projections in all cases
A
Alexander Regueiro 已提交
1693
    /// (even for associated types that are specializable). This is
I
Irina Popa 已提交
1694
    /// the desired behavior during codegen and certain other special
1695 1696 1697 1698 1699 1700 1701 1702
    /// contexts; normally though we want to use `Reveal::UserFacing`,
    /// which is the default.
    pub fn with_reveal_all(self) -> Self {
        ty::ParamEnv { reveal: Reveal::All, ..self }
    }

    /// Returns this same environment but with no caller bounds.
    pub fn without_caller_bounds(self) -> Self {
V
varkor 已提交
1703
        ty::ParamEnv { caller_bounds: List::empty(), ..self }
1704 1705
    }

1706
    /// Creates a suitable environment in which to perform trait
1707 1708 1709 1710 1711
    /// queries on the given value. When type-checking, this is simply
    /// the pair of the environment plus value. But when reveal is set to
    /// All, then if `value` does not reference any type parameters, we will
    /// pair it with the empty environment. This improves caching and is generally
    /// invisible.
1712
    ///
1713
    /// N.B., we preserve the environment when type-checking because it
1714
    /// is possible for the user to have wacky where-clauses like
1715
    /// `where Box<u32>: Copy`, which are clearly never
1716 1717
    /// satisfiable. We generally want to behave as if they were true,
    /// although the surrounding function is never reachable.
1718
    pub fn and<T: TypeFoldable<'tcx>>(self, value: T) -> ParamEnvAnd<'tcx, T> {
1719 1720 1721 1722 1723 1724
        match self.reveal {
            Reveal::UserFacing => {
                ParamEnvAnd {
                    param_env: self,
                    value,
                }
1725
            }
1726 1727

            Reveal::All => {
1728
                if value.has_placeholders()
1729 1730 1731 1732
                    || value.needs_infer()
                    || value.has_param_types()
                    || value.has_self_ty()
                {
1733 1734 1735 1736 1737 1738 1739 1740 1741 1742
                    ParamEnvAnd {
                        param_env: self,
                        value,
                    }
                } else {
                    ParamEnvAnd {
                        param_env: self.without_caller_bounds(),
                        value,
                    }
                }
1743
            }
1744 1745 1746
        }
    }
}
1747

1748
#[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
1749 1750
pub struct ParamEnvAnd<'tcx, T> {
    pub param_env: ParamEnv<'tcx>,
1751
    pub value: T,
1752 1753
}

1754 1755
impl<'tcx, T> ParamEnvAnd<'tcx, T> {
    pub fn into_parts(self) -> (ParamEnv<'tcx>, T) {
1756
        (self.param_env, self.value)
1757
    }
1758 1759
}

1760 1761
impl<'a, 'gcx, T> HashStable<StableHashingContext<'a>> for ParamEnvAnd<'gcx, T>
    where T: HashStable<StableHashingContext<'a>>
1762 1763
{
    fn hash_stable<W: StableHasherResult>(&self,
1764
                                          hcx: &mut StableHashingContext<'a>,
1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775
                                          hasher: &mut StableHasher<W>) {
        let ParamEnvAnd {
            ref param_env,
            ref value
        } = *self;

        param_env.hash_stable(hcx, hasher);
        value.hash_stable(hcx, hasher);
    }
}

1776
#[derive(Copy, Clone, Debug, HashStable)]
1777
pub struct Destructor {
A
Alexander Regueiro 已提交
1778
    /// The `DefId` of the destructor method
1779 1780 1781
    pub did: DefId,
}

1782
bitflags! {
1783
    #[derive(HashStable)]
1784 1785
    pub struct AdtFlags: u32 {
        const NO_ADT_FLAGS        = 0;
1786
        /// Indicates whether the ADT is an enum.
1787
        const IS_ENUM             = 1 << 0;
1788
        /// Indicates whether the ADT is a union.
A
Alexander Regueiro 已提交
1789
        const IS_UNION            = 1 << 1;
1790
        /// Indicates whether the ADT is a struct.
A
Alexander Regueiro 已提交
1791
        const IS_STRUCT           = 1 << 2;
1792
        /// Indicates whether the ADT is a struct and has a constructor.
1793
        const HAS_CTOR            = 1 << 3;
1794
        /// Indicates whether the type is a `PhantomData`.
A
Alexander Regueiro 已提交
1795
        const IS_PHANTOM_DATA     = 1 << 4;
1796
        /// Indicates whether the type has a `#[fundamental]` attribute.
A
Alexander Regueiro 已提交
1797
        const IS_FUNDAMENTAL      = 1 << 5;
1798
        /// Indicates whether the type is a `Box`.
A
Alexander Regueiro 已提交
1799
        const IS_BOX              = 1 << 6;
1800
        /// Indicates whether the type is an `Arc`.
A
Alexander Regueiro 已提交
1801
        const IS_ARC              = 1 << 7;
1802
        /// Indicates whether the type is an `Rc`.
A
Alexander Regueiro 已提交
1803
        const IS_RC               = 1 << 8;
1804 1805
        /// Indicates whether the variant list of this ADT is `#[non_exhaustive]`.
        /// (i.e., this flag is never set unless this ADT is an enum).
A
Alexander Regueiro 已提交
1806
        const IS_VARIANT_LIST_NON_EXHAUSTIVE = 1 << 9;
1807 1808 1809 1810
    }
}

bitflags! {
1811
    #[derive(HashStable)]
1812 1813 1814 1815
    pub struct VariantFlags: u32 {
        const NO_VARIANT_FLAGS        = 0;
        /// Indicates whether the field list of this variant is `#[non_exhaustive]`.
        const IS_FIELD_LIST_NON_EXHAUSTIVE = 1 << 0;
1816 1817 1818
    }
}

1819
/// Definition of a variant -- a struct's fields or a enum variant.
1820
#[derive(Debug)]
1821
pub struct VariantDef {
1822 1823 1824 1825 1826 1827
    /// `DefId` that identifies the variant itself.
    /// If this variant belongs to a struct or union, then this is a copy of its `DefId`.
    pub def_id: DefId,
    /// `DefId` that identifies the variant's constructor.
    /// If this variant is a struct variant, then this is `None`.
    pub ctor_def_id: Option<DefId>,
1828 1829 1830
    /// Variant or struct name.
    pub ident: Ident,
    /// Discriminant of this variant.
1831
    pub discr: VariantDiscr,
1832
    /// Fields of this variant.
1833
    pub fields: Vec<FieldDef>,
1834
    /// Type of constructor of variant.
1835
    pub ctor_kind: CtorKind,
1836
    /// Flags of the variant (e.g. is field list non-exhaustive)?
1837
    flags: VariantFlags,
1838
    /// Recovered?
1839
    pub recovered: bool,
1840 1841
}

1842
impl<'a, 'gcx, 'tcx> VariantDef {
A
Alexander Regueiro 已提交
1843
    /// Creates a new `VariantDef`.
1844
    ///
1845 1846
    /// `variant_did` is the `DefId` that identifies the enum variant (if this `VariantDef`
    /// represents an enum variant).
1847
    ///
1848 1849 1850 1851 1852 1853 1854
    /// `ctor_did` is the `DefId` that identifies the constructor of unit or
    /// tuple-variants/structs. If this is a `struct`-variant then this should be `None`.
    ///
    /// `parent_did` is the `DefId` of the `AdtDef` representing the enum or struct that
    /// owns this variant. It is used for checking if a struct has `#[non_exhaustive]` w/out having
    /// to go through the redirect of checking the ctor's attributes - but compiling a small crate
    /// requires loading the `AdtDef`s for all the structs in the universe (e.g., coherence for any
1855 1856 1857
    /// built-in trait), and we do not want to load attributes twice.
    ///
    /// If someone speeds up attribute loading to not be a performance concern, they can
A
Alexander Regueiro 已提交
1858
    /// remove this hack and use the constructor `DefId` everywhere.
1859 1860 1861
    pub fn new(
        tcx: TyCtxt<'a, 'gcx, 'tcx>,
        ident: Ident,
1862
        variant_did: Option<DefId>,
1863
        ctor_def_id: Option<DefId>,
1864 1865 1866
        discr: VariantDiscr,
        fields: Vec<FieldDef>,
        ctor_kind: CtorKind,
1867 1868
        adt_kind: AdtKind,
        parent_did: DefId,
1869 1870
        recovered: bool,
    ) -> Self {
1871
        debug!(
1872
            "VariantDef::new(ident = {:?}, variant_did = {:?}, ctor_def_id = {:?}, discr = {:?},
1873
             fields = {:?}, ctor_kind = {:?}, adt_kind = {:?}, parent_did = {:?})",
1874
             ident, variant_did, ctor_def_id, discr, fields, ctor_kind, adt_kind, parent_did,
1875 1876
        );

1877
        let mut flags = VariantFlags::NO_VARIANT_FLAGS;
1878 1879
        if adt_kind == AdtKind::Struct && tcx.has_attr(parent_did, "non_exhaustive") {
            debug!("found non-exhaustive field list for {:?}", parent_did);
1880
            flags = flags | VariantFlags::IS_FIELD_LIST_NON_EXHAUSTIVE;
1881 1882 1883 1884 1885
        } else if let Some(variant_did) = variant_did {
            if tcx.has_attr(variant_did, "non_exhaustive") {
                debug!("found non-exhaustive field list for {:?}", variant_did);
                flags = flags | VariantFlags::IS_FIELD_LIST_NON_EXHAUSTIVE;
            }
1886
        }
1887

1888
        VariantDef {
1889 1890
            def_id: variant_did.unwrap_or(parent_did),
            ctor_def_id,
1891
            ident,
1892 1893 1894
            discr,
            fields,
            ctor_kind,
1895 1896
            flags,
            recovered,
1897 1898 1899
        }
    }

1900
    /// Is this field list non-exhaustive?
1901 1902 1903 1904
    #[inline]
    pub fn is_field_list_non_exhaustive(&self) -> bool {
        self.flags.intersects(VariantFlags::IS_FIELD_LIST_NON_EXHAUSTIVE)
    }
1905 1906
}

1907
impl_stable_hash_for!(struct VariantDef {
1908 1909
    def_id,
    ctor_def_id,
1910
    ident -> (ident.name),
1911 1912 1913
    discr,
    fields,
    ctor_kind,
1914 1915
    flags,
    recovered
1916 1917
});

1918
#[derive(Copy, Clone, Debug, PartialEq, Eq, RustcEncodable, RustcDecodable, HashStable)]
1919
pub enum VariantDiscr {
1920
    /// Explicit value for this variant, i.e., `X = 123`.
1921 1922 1923 1924 1925 1926 1927
    /// The `DefId` corresponds to the embedded constant.
    Explicit(DefId),

    /// The previous variant's discriminant plus one.
    /// For efficiency reasons, the distance from the
    /// last `Explicit` discriminant is being stored,
    /// or `0` for the first variant, if it has none.
1928
    Relative(u32),
1929 1930
}

1931
#[derive(Debug, HashStable)]
1932
pub struct FieldDef {
1933
    pub did: DefId,
1934
    #[stable_hasher(project(name))]
1935
    pub ident: Ident,
1936
    pub vis: Visibility,
1937 1938
}

1939
/// The definition of an abstract data type -- a struct or enum.
A
Ariel Ben-Yehuda 已提交
1940
///
1941
/// These are all interned (by `intern_adt_def`) into the `adt_defs` table.
1942
pub struct AdtDef {
1943
    /// `DefId` of the struct, enum or union item.
1944
    pub did: DefId,
1945
    /// Variants of the ADT. If this is a struct or enum, then there will be a single variant.
1946
    pub variants: IndexVec<self::layout::VariantIdx, VariantDef>,
1947
    /// Flags of the ADT (e.g. is this a struct? is this non-exhaustive?)
1948
    flags: AdtFlags,
1949
    /// Repr options provided by the user.
1950
    pub repr: ReprOptions,
1951 1952
}

1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
impl PartialOrd for AdtDef {
    fn partial_cmp(&self, other: &AdtDef) -> Option<Ordering> {
        Some(self.cmp(&other))
    }
}

/// There should be only one AdtDef for each `did`, therefore
/// it is fine to implement `Ord` only based on `did`.
impl Ord for AdtDef {
    fn cmp(&self, other: &AdtDef) -> Ordering {
        self.did.cmp(&other.did)
    }
}

1967 1968
impl PartialEq for AdtDef {
    // AdtDef are always interned and this is part of TyS equality
1969
    #[inline]
1970
    fn eq(&self, other: &Self) -> bool { ptr::eq(self, other) }
1971 1972
}

1973
impl Eq for AdtDef {}
1974

1975
impl Hash for AdtDef {
1976 1977
    #[inline]
    fn hash<H: Hasher>(&self, s: &mut H) {
1978
        (self as *const AdtDef).hash(s)
1979 1980 1981
    }
}

1982
impl<'tcx> serialize::UseSpecializedEncodable for &'tcx AdtDef {
1983
    fn default_encode<S: Encoder>(&self, s: &mut S) -> Result<(), S::Error> {
1984 1985 1986 1987
        self.did.encode(s)
    }
}

1988
impl<'tcx> serialize::UseSpecializedDecodable for &'tcx AdtDef {}
1989

1990

1991
impl<'a> HashStable<StableHashingContext<'a>> for AdtDef {
1992
    fn hash_stable<W: StableHasherResult>(&self,
1993
                                          hcx: &mut StableHashingContext<'a>,
1994
                                          hasher: &mut StableHasher<W>) {
W
Wesley Wiser 已提交
1995
        thread_local! {
1996
            static CACHE: RefCell<FxHashMap<usize, Fingerprint>> = Default::default();
W
Wesley Wiser 已提交
1997
        }
1998

W
Wesley Wiser 已提交
1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019
        let hash: Fingerprint = CACHE.with(|cache| {
            let addr = self as *const AdtDef as usize;
            *cache.borrow_mut().entry(addr).or_insert_with(|| {
                let ty::AdtDef {
                    did,
                    ref variants,
                    ref flags,
                    ref repr,
                } = *self;

                let mut hasher = StableHasher::new();
                did.hash_stable(hcx, &mut hasher);
                variants.hash_stable(hcx, &mut hasher);
                flags.hash_stable(hcx, &mut hasher);
                repr.hash_stable(hcx, &mut hasher);

                hasher.finish()
           })
        });

        hash.hash_stable(hcx, hasher);
2020 2021 2022
    }
}

2023
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash)]
2024
pub enum AdtKind { Struct, Union, Enum }
2025

2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
impl Into<DataTypeKind> for AdtKind {
    fn into(self) -> DataTypeKind {
        match self {
            AdtKind::Struct => DataTypeKind::Struct,
            AdtKind::Union => DataTypeKind::Union,
            AdtKind::Enum => DataTypeKind::Enum,
        }
    }
}

2036 2037
bitflags! {
    #[derive(RustcEncodable, RustcDecodable, Default)]
2038 2039
    pub struct ReprFlags: u8 {
        const IS_C               = 1 << 0;
2040 2041
        const IS_SIMD            = 1 << 1;
        const IS_TRANSPARENT     = 1 << 2;
2042
        // Internal only for now. If true, don't reorder fields.
2043
        const IS_LINEAR          = 1 << 3;
2044 2045 2046 2047

        // Any of these flags being set prevent field reordering optimisation.
        const IS_UNOPTIMISABLE   = ReprFlags::IS_C.bits |
                                   ReprFlags::IS_SIMD.bits |
2048
                                   ReprFlags::IS_LINEAR.bits;
2049 2050 2051 2052 2053 2054 2055
    }
}

impl_stable_hash_for!(struct ReprFlags {
    bits
});

2056
/// Represents the repr options provided by the user,
2057
#[derive(Copy, Clone, Debug, Eq, PartialEq, RustcEncodable, RustcDecodable, Default)]
2058 2059
pub struct ReprOptions {
    pub int: Option<attr::IntType>,
2060
    pub align: u32,
2061
    pub pack: u32,
2062
    pub flags: ReprFlags,
2063 2064
}

2065
impl_stable_hash_for!(struct ReprOptions {
2066
    align,
2067
    pack,
2068
    int,
2069
    flags
2070 2071
});

2072
impl ReprOptions {
2073
    pub fn new(tcx: TyCtxt<'_, '_, '_>, did: DefId) -> ReprOptions {
2074 2075
        let mut flags = ReprFlags::empty();
        let mut size = None;
2076
        let mut max_align = 0;
2077
        let mut min_pack = 0;
2078
        for attr in tcx.get_attrs(did).iter() {
C
csmoe 已提交
2079
            for r in attr::find_repr_attrs(&tcx.sess.parse_sess, attr) {
2080
                flags.insert(match r {
2081
                    attr::ReprC => ReprFlags::IS_C,
2082 2083 2084 2085 2086 2087 2088 2089
                    attr::ReprPacked(pack) => {
                        min_pack = if min_pack > 0 {
                            cmp::min(pack, min_pack)
                        } else {
                            pack
                        };
                        ReprFlags::empty()
                    },
R
Robin Kruppe 已提交
2090
                    attr::ReprTransparent => ReprFlags::IS_TRANSPARENT,
2091 2092 2093 2094 2095
                    attr::ReprSimd => ReprFlags::IS_SIMD,
                    attr::ReprInt(i) => {
                        size = Some(i);
                        ReprFlags::empty()
                    },
2096 2097 2098 2099
                    attr::ReprAlign(align) => {
                        max_align = cmp::max(align, max_align);
                        ReprFlags::empty()
                    },
2100
                });
2101 2102
            }
        }
2103

2104
        // This is here instead of layout because the choice must make it into metadata.
2105
        if !tcx.consider_optimizing(|| format!("Reorder fields of {:?}", tcx.def_path_str(did))) {
2106 2107
            flags.insert(ReprFlags::IS_LINEAR);
        }
2108
        ReprOptions { int: size, align: max_align, pack: min_pack, flags: flags }
2109
    }
2110

2111 2112 2113 2114 2115
    #[inline]
    pub fn simd(&self) -> bool { self.flags.contains(ReprFlags::IS_SIMD) }
    #[inline]
    pub fn c(&self) -> bool { self.flags.contains(ReprFlags::IS_C) }
    #[inline]
2116
    pub fn packed(&self) -> bool { self.pack > 0 }
2117
    #[inline]
R
Robin Kruppe 已提交
2118 2119
    pub fn transparent(&self) -> bool { self.flags.contains(ReprFlags::IS_TRANSPARENT) }
    #[inline]
2120 2121
    pub fn linear(&self) -> bool { self.flags.contains(ReprFlags::IS_LINEAR) }

2122
    pub fn discr_type(&self) -> attr::IntType {
2123
        self.int.unwrap_or(attr::SignedInt(ast::IntTy::Isize))
2124
    }
2125

2126
    /// Returns `true` if this `#[repr()]` should inhabit "smart enum
2127 2128 2129
    /// layout" optimizations, such as representing `Foo<&T>` as a
    /// single pointer.
    pub fn inhibit_enum_layout_opt(&self) -> bool {
2130
        self.c() || self.int.is_some()
2131
    }
2132

2133
    /// Returns `true` if this `#[repr()]` should inhibit struct field reordering
A
Alexander Regueiro 已提交
2134
    /// optimizations, such as with `repr(C)`, `repr(packed(1))`, or `repr(<int>)`.
2135
    pub fn inhibit_struct_field_reordering_opt(&self) -> bool {
2136 2137
        self.flags.intersects(ReprFlags::IS_UNOPTIMISABLE) || self.pack == 1 ||
            self.int.is_some()
2138
    }
2139

A
Alexander Regueiro 已提交
2140
    /// Returns `true` if this `#[repr()]` should inhibit union ABI optimisations.
2141 2142 2143 2144
    pub fn inhibit_union_abi_opt(&self) -> bool {
        self.c()
    }

2145 2146
}

2147
impl<'a, 'gcx, 'tcx> AdtDef {
2148 2149 2150 2151 2152 2153 2154 2155
    /// Creates a new `AdtDef`.
    fn new(
        tcx: TyCtxt<'_, '_, '_>,
        did: DefId,
        kind: AdtKind,
        variants: IndexVec<VariantIdx, VariantDef>,
        repr: ReprOptions
    ) -> Self {
2156
        debug!("AdtDef::new({:?}, {:?}, {:?}, {:?})", did, kind, variants, repr);
A
Ariel Ben-Yehuda 已提交
2157
        let mut flags = AdtFlags::NO_ADT_FLAGS;
A
Alexander Regueiro 已提交
2158 2159 2160 2161 2162

        if kind == AdtKind::Enum && tcx.has_attr(did, "non_exhaustive") {
            debug!("found non-exhaustive variant list for {:?}", did);
            flags = flags | AdtFlags::IS_VARIANT_LIST_NON_EXHAUSTIVE;
        }
2163

A
Alexander Regueiro 已提交
2164 2165 2166 2167 2168 2169
        flags |= match kind {
            AdtKind::Enum => AdtFlags::IS_ENUM,
            AdtKind::Union => AdtFlags::IS_UNION,
            AdtKind::Struct => AdtFlags::IS_STRUCT,
        };

2170
        if kind == AdtKind::Struct && variants[VariantIdx::new(0)].ctor_def_id.is_some() {
2171
            flags |= AdtFlags::HAS_CTOR;
A
Alexander Regueiro 已提交
2172 2173
        }

A
Ariel Ben-Yehuda 已提交
2174
        let attrs = tcx.get_attrs(did);
2175
        if attr::contains_name(&attrs, "fundamental") {
A
Alexander Regueiro 已提交
2176
            flags |= AdtFlags::IS_FUNDAMENTAL;
2177
        }
2178
        if Some(did) == tcx.lang_items().phantom_data() {
A
Alexander Regueiro 已提交
2179
            flags |= AdtFlags::IS_PHANTOM_DATA;
2180
        }
2181
        if Some(did) == tcx.lang_items().owned_box() {
A
Alexander Regueiro 已提交
2182
            flags |= AdtFlags::IS_BOX;
2183
        }
2184
        if Some(did) == tcx.lang_items().arc() {
A
Alexander Regueiro 已提交
2185
            flags |= AdtFlags::IS_ARC;
2186 2187
        }
        if Some(did) == tcx.lang_items().rc() {
A
Alexander Regueiro 已提交
2188
            flags |= AdtFlags::IS_RC;
2189
        }
A
Alexander Regueiro 已提交
2190

2191
        AdtDef {
2192 2193 2194 2195
            did,
            variants,
            flags,
            repr,
2196 2197 2198
        }
    }

2199
    /// Returns `true` if this is a struct.
2200 2201
    #[inline]
    pub fn is_struct(&self) -> bool {
A
Alexander Regueiro 已提交
2202 2203 2204
        self.flags.contains(AdtFlags::IS_STRUCT)
    }

2205
    /// Returns `true` if this is a union.
2206 2207
    #[inline]
    pub fn is_union(&self) -> bool {
A
Alexander Regueiro 已提交
2208
        self.flags.contains(AdtFlags::IS_UNION)
2209 2210
    }

2211
    /// Returns `true` if this is a enum.
2212 2213
    #[inline]
    pub fn is_enum(&self) -> bool {
A
Alexander Regueiro 已提交
2214
        self.flags.contains(AdtFlags::IS_ENUM)
2215 2216
    }

2217
    /// Returns `true` if the variant list of this ADT is `#[non_exhaustive]`.
2218
    #[inline]
2219
    pub fn is_variant_list_non_exhaustive(&self) -> bool {
A
Alexander Regueiro 已提交
2220
        self.flags.contains(AdtFlags::IS_VARIANT_LIST_NON_EXHAUSTIVE)
2221 2222
    }

A
Alexander Regueiro 已提交
2223
    /// Returns the kind of the ADT.
2224
    #[inline]
A
Ariel Ben-Yehuda 已提交
2225
    pub fn adt_kind(&self) -> AdtKind {
2226
        if self.is_enum() {
A
Ariel Ben-Yehuda 已提交
2227
            AdtKind::Enum
2228
        } else if self.is_union() {
2229
            AdtKind::Union
2230
        } else {
A
Ariel Ben-Yehuda 已提交
2231
            AdtKind::Struct
2232 2233 2234
        }
    }

2235
    /// Returns a description of this abstract data type.
2236 2237 2238 2239 2240 2241 2242 2243
    pub fn descr(&self) -> &'static str {
        match self.adt_kind() {
            AdtKind::Struct => "struct",
            AdtKind::Union => "union",
            AdtKind::Enum => "enum",
        }
    }

2244
    /// Returns a description of a variant of this abstract data type.
2245
    #[inline]
2246 2247 2248 2249 2250 2251 2252 2253
    pub fn variant_descr(&self) -> &'static str {
        match self.adt_kind() {
            AdtKind::Struct => "struct",
            AdtKind::Union => "union",
            AdtKind::Enum => "variant",
        }
    }

2254 2255 2256 2257 2258 2259
    /// If this function returns `true`, it implies that `is_struct` must return `true`.
    #[inline]
    pub fn has_ctor(&self) -> bool {
        self.flags.contains(AdtFlags::HAS_CTOR)
    }

A
Alexander Regueiro 已提交
2260
    /// Returns `true` if this type is `#[fundamental]` for the purposes
A
Ariel Ben-Yehuda 已提交
2261
    /// of coherence checking.
2262 2263
    #[inline]
    pub fn is_fundamental(&self) -> bool {
A
Alexander Regueiro 已提交
2264
        self.flags.contains(AdtFlags::IS_FUNDAMENTAL)
2265 2266
    }

A
Alexander Regueiro 已提交
2267
    /// Returns `true` if this is `PhantomData<T>`.
2268 2269
    #[inline]
    pub fn is_phantom_data(&self) -> bool {
A
Alexander Regueiro 已提交
2270
        self.flags.contains(AdtFlags::IS_PHANTOM_DATA)
2271 2272
    }

2273 2274
    /// Returns `true` if this is `Arc<T>`.
    pub fn is_arc(&self) -> bool {
A
Alexander Regueiro 已提交
2275
        self.flags.contains(AdtFlags::IS_ARC)
2276 2277 2278 2279
    }

    /// Returns `true` if this is `Rc<T>`.
    pub fn is_rc(&self) -> bool {
A
Alexander Regueiro 已提交
2280
        self.flags.contains(AdtFlags::IS_RC)
2281 2282
    }

2283
    /// Returns `true` if this is Box<T>.
2284 2285
    #[inline]
    pub fn is_box(&self) -> bool {
A
Alexander Regueiro 已提交
2286
        self.flags.contains(AdtFlags::IS_BOX)
2287 2288
    }

A
Alexander Regueiro 已提交
2289
    /// Returns `true` if this type has a destructor.
2290 2291
    pub fn has_dtor(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>) -> bool {
        self.destructor(tcx).is_some()
2292 2293
    }

2294 2295 2296
    /// Asserts this is a struct or union and returns its unique variant.
    pub fn non_enum_variant(&self) -> &VariantDef {
        assert!(self.is_struct() || self.is_union());
2297
        &self.variants[VariantIdx::new(0)]
2298 2299 2300
    }

    #[inline]
2301
    pub fn predicates(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>) -> Lrc<GenericPredicates<'gcx>> {
2302
        tcx.predicates_of(self.did)
2303
    }
2304

A
Ariel Ben-Yehuda 已提交
2305 2306
    /// Returns an iterator over all fields contained
    /// by this ADT.
2307
    #[inline]
2308 2309
    pub fn all_fields<'s>(&'s self) -> impl Iterator<Item = &'s FieldDef> {
        self.variants.iter().flat_map(|v| v.fields.iter())
2310 2311 2312 2313 2314 2315 2316
    }

    pub fn is_payloadfree(&self) -> bool {
        !self.variants.is_empty() &&
            self.variants.iter().all(|v| v.fields.is_empty())
    }

2317
    /// Return a `VariantDef` given a variant id.
2318 2319 2320
    pub fn variant_with_id(&self, vid: DefId) -> &VariantDef {
        self.variants.iter().find(|v| v.def_id == vid)
            .expect("variant_with_id: unknown variant")
2321 2322
    }

2323
    /// Return a `VariantDef` given a constructor id.
2324
    pub fn variant_with_ctor_id(&self, cid: DefId) -> &VariantDef {
2325
        self.variants.iter().find(|v| v.ctor_def_id == Some(cid))
2326 2327 2328
            .expect("variant_with_ctor_id: unknown variant")
    }

2329
    /// Return the index of `VariantDef` given a variant id.
2330 2331 2332
    pub fn variant_index_with_id(&self, vid: DefId) -> VariantIdx {
        self.variants.iter_enumerated().find(|(_, v)| v.def_id == vid)
            .expect("variant_index_with_id: unknown variant").0
2333 2334
    }

2335
    /// Return the index of `VariantDef` given a constructor id.
2336
    pub fn variant_index_with_ctor_id(&self, cid: DefId) -> VariantIdx {
2337 2338
        self.variants.iter_enumerated().find(|(_, v)| v.ctor_def_id == Some(cid))
            .expect("variant_index_with_ctor_id: unknown variant").0
N
Niko Matsakis 已提交
2339 2340
    }

2341 2342 2343 2344 2345 2346 2347 2348
    pub fn variant_of_res(&self, res: Res) -> &VariantDef {
        match res {
            Res::Def(DefKind::Variant, vid) => self.variant_with_id(vid),
            Res::Def(DefKind::Ctor(..), cid) => self.variant_with_ctor_id(cid),
            Res::Def(DefKind::Struct, _) | Res::Def(DefKind::Union, _) |
            Res::Def(DefKind::TyAlias, _) | Res::Def(DefKind::AssociatedTy, _) | Res::SelfTy(..) |
            Res::SelfCtor(..) => self.non_enum_variant(),
            _ => bug!("unexpected res {:?} in variant_of_res", res)
2349 2350
        }
    }
2351

O
Oliver Schneider 已提交
2352
    #[inline]
2353
    pub fn eval_explicit_discr(
O
Oliver Schneider 已提交
2354 2355 2356 2357
        &self,
        tcx: TyCtxt<'a, 'gcx, 'tcx>,
        expr_did: DefId,
    ) -> Option<Discr<'tcx>> {
2358
        let param_env = ParamEnv::empty();
O
Oliver Schneider 已提交
2359
        let repr_type = self.repr.discr_type();
C
csmoe 已提交
2360
        let substs = InternalSubsts::identity_for_item(tcx.global_tcx(), expr_did);
O
Oliver Schneider 已提交
2361 2362 2363 2364 2365 2366
        let instance = ty::Instance::new(expr_did, substs);
        let cid = GlobalId {
            instance,
            promoted: None
        };
        match tcx.const_eval(param_env.and(cid)) {
2367 2368
            Ok(val) => {
                // FIXME: Find the right type and use it instead of `val.ty` here
O
Oliver Schneider 已提交
2369
                if let Some(b) = val.assert_bits(tcx.global_tcx(), param_env.and(val.ty)) {
2370 2371 2372 2373 2374 2375 2376
                    trace!("discriminants: {} ({:?})", b, repr_type);
                    Some(Discr {
                        val: b,
                        ty: val.ty,
                    })
                } else {
                    info!("invalid enum discriminant: {:#?}", val);
M
Mark Mansi 已提交
2377
                    crate::mir::interpret::struct_error(
2378
                        tcx.at(tcx.def_span(expr_did)),
2379 2380 2381 2382
                        "constant evaluation of enum discriminant resulted in non-integer",
                    ).emit();
                    None
                }
O
Oliver Schneider 已提交
2383
            }
2384
            Err(ErrorHandled::Reported) => {
O
Oliver Schneider 已提交
2385 2386 2387
                if !expr_did.is_local() {
                    span_bug!(tcx.def_span(expr_did),
                        "variant discriminant evaluation succeeded \
L
ljedrz 已提交
2388
                         in its crate but failed locally");
O
Oliver Schneider 已提交
2389 2390 2391
                }
                None
            }
2392 2393 2394 2395
            Err(ErrorHandled::TooGeneric) => span_bug!(
                tcx.def_span(expr_did),
                "enum discriminant depends on generic arguments",
            ),
O
Oliver Schneider 已提交
2396 2397 2398
        }
    }

2399
    #[inline]
2400 2401 2402
    pub fn discriminants(
        &'a self,
        tcx: TyCtxt<'a, 'gcx, 'tcx>,
2403
    ) -> impl Iterator<Item=(VariantIdx, Discr<'tcx>)> + Captures<'gcx> + 'a {
2404
        let repr_type = self.repr.discr_type();
2405
        let initial = repr_type.initial_discriminant(tcx.global_tcx());
O
Oliver Schneider 已提交
2406
        let mut prev_discr = None::<Discr<'tcx>>;
2407
        self.variants.iter_enumerated().map(move |(i, v)| {
O
Oliver Schneider 已提交
2408
            let mut discr = prev_discr.map_or(initial, |d| d.wrap_incr(tcx));
2409
            if let VariantDiscr::Explicit(expr_did) = v.discr {
O
Oliver Schneider 已提交
2410 2411
                if let Some(new_discr) = self.eval_explicit_discr(tcx, expr_did) {
                    discr = new_discr;
2412 2413 2414 2415
                }
            }
            prev_discr = Some(discr);

2416
            (i, discr)
2417 2418 2419
        })
    }

2420 2421 2422 2423 2424
    #[inline]
    pub fn variant_range(&self) -> Range<VariantIdx> {
        (VariantIdx::new(0)..VariantIdx::new(self.variants.len()))
    }

A
Alexander Regueiro 已提交
2425
    /// Computes the discriminant value used by a specific variant.
2426 2427 2428 2429
    /// Unlike `discriminants`, this is (amortized) constant-time,
    /// only doing at most one query for evaluating an explicit
    /// discriminant (the last one before the requested variant),
    /// assuming there are no constant-evaluation errors there.
2430
    #[inline]
2431 2432
    pub fn discriminant_for_variant(&self,
                                    tcx: TyCtxt<'a, 'gcx, 'tcx>,
2433
                                    variant_index: VariantIdx)
O
Oliver Schneider 已提交
2434
                                    -> Discr<'tcx> {
2435 2436 2437 2438 2439 2440 2441
        let (val, offset) = self.discriminant_def_for_variant(variant_index);
        let explicit_value = val
            .and_then(|expr_did| self.eval_explicit_discr(tcx, expr_did))
            .unwrap_or_else(|| self.repr.discr_type().initial_discriminant(tcx.global_tcx()));
        explicit_value.checked_add(tcx, offset as u128).0
    }

A
Alexander Regueiro 已提交
2442
    /// Yields a `DefId` for the discriminant and an offset to add to it
2443
    /// Alternatively, if there is no explicit discriminant, returns the
A
Alexander Regueiro 已提交
2444
    /// inferred discriminant directly.
2445 2446
    pub fn discriminant_def_for_variant(
        &self,
2447 2448 2449
        variant_index: VariantIdx,
    ) -> (Option<DefId>, u32) {
        let mut explicit_index = variant_index.as_u32();
2450
        let expr_did;
2451
        loop {
2452
            match self.variants[VariantIdx::from_u32(explicit_index)].discr {
2453 2454 2455 2456
                ty::VariantDiscr::Relative(0) => {
                    expr_did = None;
                    break;
                },
2457 2458 2459
                ty::VariantDiscr::Relative(distance) => {
                    explicit_index -= distance;
                }
2460 2461 2462
                ty::VariantDiscr::Explicit(did) => {
                    expr_did = Some(did);
                    break;
2463 2464 2465
                }
            }
        }
2466
        (expr_did, variant_index.as_u32() - explicit_index)
2467 2468
    }

2469
    pub fn destructor(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>) -> Option<Destructor> {
2470
        tcx.adt_destructor(self.did)
2471 2472
    }

2473
    /// Returns a list of types such that `Self: Sized` if and only
A
Alexander Regueiro 已提交
2474
    /// if that type is `Sized`, or `TyErr` if this type is recursive.
A
Ariel Ben-Yehuda 已提交
2475
    ///
A
Alexander Regueiro 已提交
2476
    /// Oddly enough, checking that the sized-constraint is `Sized` is
A
Ariel Ben-Yehuda 已提交
2477
    /// actually more expressive than checking all members:
A
Alexander Regueiro 已提交
2478 2479
    /// the `Sized` trait is inductive, so an associated type that references
    /// `Self` would prevent its containing ADT from being `Sized`.
A
Ariel Ben-Yehuda 已提交
2480 2481
    ///
    /// Due to normalization being eager, this applies even if
A
Alexander Regueiro 已提交
2482
    /// the associated type is behind a pointer (e.g., issue #31299).
2483
    pub fn sized_constraint(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>) -> &'tcx [Ty<'tcx>] {
2484
        tcx.adt_sized_constraint(self.did).0
2485
    }
2486

2487 2488 2489 2490
    fn sized_constraint_for_ty(&self,
                               tcx: TyCtxt<'a, 'tcx, 'tcx>,
                               ty: Ty<'tcx>)
                               -> Vec<Ty<'tcx>> {
2491
        let result = match ty.sty {
2492
            Bool | Char | Int(..) | Uint(..) | Float(..) |
V
varkor 已提交
2493 2494
            RawPtr(..) | Ref(..) | FnDef(..) | FnPtr(_) |
            Array(..) | Closure(..) | Generator(..) | Never => {
A
Ariel Ben-Yehuda 已提交
2495
                vec![]
2496 2497
            }

2498
            Str |
V
varkor 已提交
2499 2500
            Dynamic(..) |
            Slice(_) |
V
varkor 已提交
2501
            Foreign(..) |
V
varkor 已提交
2502 2503
            Error |
            GeneratorWitness(..) => {
2504
                // these are never sized - return the target type
A
Ariel Ben-Yehuda 已提交
2505
                vec![ty]
2506 2507
            }

V
varkor 已提交
2508
            Tuple(ref tys) => {
2509 2510
                match tys.last() {
                    None => vec![],
V
varkor 已提交
2511
                    Some(ty) => self.sized_constraint_for_ty(tcx, ty.expect_ty()),
2512
                }
2513 2514
            }

V
varkor 已提交
2515
            Adt(adt, substs) => {
2516
                // recursive case
2517
                let adt_tys = adt.sized_constraint(tcx);
2518
                debug!("sized_constraint_for_ty({:?}) intermediate = {:?}",
2519 2520
                       ty, adt_tys);
                adt_tys.iter()
L
ljedrz 已提交
2521 2522 2523
                       .map(|ty| ty.subst(tcx, substs))
                       .flat_map(|ty| self.sized_constraint_for_ty(tcx, ty))
                       .collect()
2524 2525
            }

2526
            Projection(..) | Opaque(..) => {
2527 2528
                // must calculate explicitly.
                // FIXME: consider special-casing always-Sized projections
A
Ariel Ben-Yehuda 已提交
2529
                vec![ty]
2530 2531
            }

2532 2533
            UnnormalizedProjection(..) => bug!("only used with chalk-engine"),

V
varkor 已提交
2534
            Param(..) => {
A
Ariel Ben-Yehuda 已提交
2535 2536 2537 2538
                // perf hack: if there is a `T: Sized` bound, then
                // we know that `T` is Sized and do not need to check
                // it on the impl.

2539
                let sized_trait = match tcx.lang_items().sized_trait() {
2540
                    Some(x) => x,
A
Ariel Ben-Yehuda 已提交
2541
                    _ => return vec![ty]
2542
                };
2543
                let sized_predicate = Binder::dummy(TraitRef {
2544
                    def_id: sized_trait,
2545
                    substs: tcx.mk_substs_trait(ty, &[])
2546
                }).to_predicate();
2547 2548
                let predicates = &tcx.predicates_of(self.did).predicates;
                if predicates.iter().any(|(p, _)| *p == sized_predicate) {
A
Ariel Ben-Yehuda 已提交
2549
                    vec![]
2550
                } else {
A
Ariel Ben-Yehuda 已提交
2551
                    vec![ty]
2552 2553 2554
                }
            }

S
scalexm 已提交
2555
            Placeholder(..) |
S
scalexm 已提交
2556
            Bound(..) |
V
varkor 已提交
2557
            Infer(..) => {
2558 2559 2560 2561 2562 2563 2564
                bug!("unexpected type `{:?}` in sized_constraint_for_ty",
                     ty)
            }
        };
        debug!("sized_constraint_for_ty({:?}) = {:?}", ty, result);
        result
    }
2565 2566
}

2567
impl<'a, 'gcx, 'tcx> FieldDef {
C
csmoe 已提交
2568
    pub fn ty(&self, tcx: TyCtxt<'a, 'gcx, 'tcx>, subst: SubstsRef<'tcx>) -> Ty<'tcx> {
2569
        tcx.type_of(self.did).subst(tcx, subst)
2570
    }
2571 2572
}

A
Alexander Regueiro 已提交
2573
/// Represents the various closure traits in the language. This
2574
/// will determine the type of the environment (`self`, in the
A
Andy Russell 已提交
2575
/// desugaring) argument that the closure expects.
2576 2577 2578
///
/// You can get the environment type of a closure using
/// `tcx.closure_env_ty()`.
2579 2580
#[derive(Clone, Copy, PartialOrd, Ord, PartialEq, Eq, Hash, Debug,
         RustcEncodable, RustcDecodable, HashStable)]
2581
pub enum ClosureKind {
2582 2583 2584
    // Warning: Ordering is significant here! The ordering is chosen
    // because the trait Fn is a subtrait of FnMut and so in turn, and
    // hence we order it so that Fn < FnMut < FnOnce.
2585 2586 2587
    Fn,
    FnMut,
    FnOnce,
2588 2589
}

2590
impl<'a, 'tcx> ClosureKind {
2591 2592 2593
    // This is the initial value used when doing upvar inference.
    pub const LATTICE_BOTTOM: ClosureKind = ClosureKind::Fn;

2594
    pub fn trait_did(&self, tcx: TyCtxt<'a, 'tcx, 'tcx>) -> DefId {
2595 2596
        match *self {
            ClosureKind::Fn => tcx.require_lang_item(FnTraitLangItem),
2597
            ClosureKind::FnMut => {
2598
                tcx.require_lang_item(FnMutTraitLangItem)
2599
            }
2600
            ClosureKind::FnOnce => {
2601
                tcx.require_lang_item(FnOnceTraitLangItem)
2602 2603 2604
            }
        }
    }
2605

2606
    /// Returns `true` if this a type that impls this closure kind
2607 2608 2609
    /// must also implement `other`.
    pub fn extends(self, other: ty::ClosureKind) -> bool {
        match (self, other) {
2610 2611 2612 2613 2614 2615
            (ClosureKind::Fn, ClosureKind::Fn) => true,
            (ClosureKind::Fn, ClosureKind::FnMut) => true,
            (ClosureKind::Fn, ClosureKind::FnOnce) => true,
            (ClosureKind::FnMut, ClosureKind::FnMut) => true,
            (ClosureKind::FnMut, ClosureKind::FnOnce) => true,
            (ClosureKind::FnOnce, ClosureKind::FnOnce) => true,
2616 2617 2618
            _ => false,
        }
    }
2619 2620 2621 2622 2623 2624 2625 2626 2627

    /// Returns the representative scalar type for this closure kind.
    /// See `TyS::to_opt_closure_kind` for more details.
    pub fn to_ty(self, tcx: TyCtxt<'_, '_, 'tcx>) -> Ty<'tcx> {
        match self {
            ty::ClosureKind::Fn => tcx.types.i8,
            ty::ClosureKind::FnMut => tcx.types.i16,
            ty::ClosureKind::FnOnce => tcx.types.i32,
        }
2628
    }
2629 2630
}

2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643
impl<'tcx> TyS<'tcx> {
    /// Iterator that walks `self` and any types reachable from
    /// `self`, in depth-first order. Note that just walks the types
    /// that appear in `self`, it does not descend into the fields of
    /// structs or variants. For example:
    ///
    /// ```notrust
    /// isize => { isize }
    /// Foo<Bar<isize>> => { Foo<Bar<isize>>, Bar<isize>, isize }
    /// [isize] => { [isize], isize }
    /// ```
    pub fn walk(&'tcx self) -> TypeWalker<'tcx> {
        TypeWalker::new(self)
2644 2645
    }

A
Alexander Regueiro 已提交
2646
    /// Iterator that walks the immediate children of `self`. Hence
2647 2648
    /// `Foo<Bar<i32>, u32>` yields the sequence `[Bar<i32>, u32]`
    /// (but not `i32`, like `walk`).
2649
    pub fn walk_shallow(&'tcx self) -> smallvec::IntoIter<walk::TypeWalkerArray<'tcx>> {
2650
        walk::walk_shallow(self)
2651 2652
    }

2653
    /// Walks `ty` and any types appearing within `ty`, invoking the
A
Alexander Regueiro 已提交
2654
    /// callback `f` on each type. If the callback returns `false`, then the
2655 2656 2657 2658
    /// children of the current type are ignored.
    ///
    /// Note: prefer `ty.walk()` where possible.
    pub fn maybe_walk<F>(&'tcx self, mut f: F)
2659
        where F: FnMut(Ty<'tcx>) -> bool
2660 2661 2662 2663 2664 2665 2666
    {
        let mut walker = self.walk();
        while let Some(ty) = walker.next() {
            if !f(ty) {
                walker.skip_current_subtree();
            }
        }
2667
    }
2668
}
2669

2670
impl BorrowKind {
2671
    pub fn from_mutbl(m: hir::Mutability) -> BorrowKind {
2672
        match m {
2673 2674
            hir::MutMutable => MutBorrow,
            hir::MutImmutable => ImmBorrow,
2675 2676
        }
    }
2677

2678 2679 2680 2681
    /// Returns a mutability `m` such that an `&m T` pointer could be used to obtain this borrow
    /// kind. Because borrow kinds are richer than mutabilities, we sometimes have to pick a
    /// mutability that is stronger than necessary so that it at least *would permit* the borrow in
    /// question.
2682
    pub fn to_mutbl_lossy(self) -> hir::Mutability {
2683
        match self {
2684 2685
            MutBorrow => hir::MutMutable,
            ImmBorrow => hir::MutImmutable,
2686 2687 2688 2689

            // We have no type corresponding to a unique imm borrow, so
            // use `&mut`. It gives all the capabilities of an `&uniq`
            // and hence is a safe "over approximation".
2690
            UniqueImmBorrow => hir::MutMutable,
2691
        }
2692
    }
2693

2694 2695 2696 2697 2698 2699
    pub fn to_user_str(&self) -> &'static str {
        match *self {
            MutBorrow => "mutable",
            ImmBorrow => "immutable",
            UniqueImmBorrow => "uniquely immutable",
        }
2700 2701 2702
    }
}

2703 2704
#[derive(Debug, Clone)]
pub enum Attributes<'gcx> {
2705
    Owned(Lrc<[ast::Attribute]>),
2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719
    Borrowed(&'gcx [ast::Attribute])
}

impl<'gcx> ::std::ops::Deref for Attributes<'gcx> {
    type Target = [ast::Attribute];

    fn deref(&self) -> &[ast::Attribute] {
        match self {
            &Attributes::Owned(ref data) => &data,
            &Attributes::Borrowed(data) => data
        }
    }
}

2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758
#[derive(Debug, PartialEq, Eq)]
pub enum ImplOverlapKind {
    /// These impls are always allowed to overlap.
    Permitted,
    /// These impls are allowed to overlap, but that raises
    /// an issue #33140 future-compatibility warning.
    ///
    /// Some background: in Rust 1.0, the trait-object types `Send + Sync` (today's
    /// `dyn Send + Sync`) and `Sync + Send` (now `dyn Sync + Send`) were different.
    ///
    /// The widely-used version 0.1.0 of the crate `traitobject` had accidentally relied
    /// that difference, making what reduces to the following set of impls:
    ///
    /// ```
    /// trait Trait {}
    /// impl Trait for dyn Send + Sync {}
    /// impl Trait for dyn Sync + Send {}
    /// ```
    ///
    /// Obviously, once we made these types be identical, that code causes a coherence
    /// error and a fairly big headache for us. However, luckily for us, the trait
    /// `Trait` used in this case is basically a marker trait, and therefore having
    /// overlapping impls for it is sound.
    ///
    /// To handle this, we basically regard the trait as a marker trait, with an additional
    /// future-compatibility warning. To avoid accidentally "stabilizing" this feature,
    /// it has the following restrictions:
    ///
    /// 1. The trait must indeed be a marker-like trait (i.e., no items), and must be
    /// positive impls.
    /// 2. The trait-ref of both impls must be equal.
    /// 3. The trait-ref of both impls must be a trait object type consisting only of
    /// marker traits.
    /// 4. Neither of the impls can have any where-clauses.
    ///
    /// Once `traitobject` 0.1.0 is no longer an active concern, this hack can be removed.
    Issue33140
}

2759
impl<'a, 'gcx, 'tcx> TyCtxt<'a, 'gcx, 'tcx> {
2760
    pub fn body_tables(self, body: hir::BodyId) -> &'gcx TypeckTables<'gcx> {
2761
        self.typeck_tables_of(self.hir().body_owner_def_id(body))
2762 2763
    }

A
Alexander Regueiro 已提交
2764
    /// Returns an iterator of the `DefId`s for all body-owners in this
N
Niko Matsakis 已提交
2765
    /// crate. If you would prefer to iterate over the bodies
2766
    /// themselves, you can do `self.hir().krate().body_ids.iter()`.
2767 2768 2769
    pub fn body_owners(
        self,
    ) -> impl Iterator<Item = DefId> + Captures<'tcx> + Captures<'gcx> + 'a {
2770 2771 2772 2773
        self.hir().krate()
                  .body_ids
                  .iter()
                  .map(move |&body_id| self.hir().body_owner_def_id(body_id))
N
Niko Matsakis 已提交
2774 2775
    }

J
John Kåre Alsaker 已提交
2776
    pub fn par_body_owners<F: Fn(DefId) + sync::Sync + sync::Send>(self, f: F) {
2777 2778
        par_iter(&self.hir().krate().body_ids).for_each(|&body_id| {
            f(self.hir().body_owner_def_id(body_id))
J
John Kåre Alsaker 已提交
2779 2780 2781
        });
    }

2782
    pub fn expr_span(self, id: NodeId) -> Span {
2783
        match self.hir().find(id) {
V
varkor 已提交
2784
            Some(Node::Expr(e)) => {
2785 2786 2787
                e.span
            }
            Some(f) => {
2788
                bug!("Node id {} is not an expr: {:?}", id, f);
2789 2790
            }
            None => {
2791
                bug!("Node id {} is not present in the node map", id);
2792
            }
2793
        }
2794 2795
    }

2796 2797
    pub fn provided_trait_methods(self, id: DefId) -> Vec<AssociatedItem> {
        self.associated_items(id)
2798
            .filter(|item| item.kind == AssociatedKind::Method && item.defaultness.has_value())
2799
            .collect()
2800 2801
    }

A
Andrew Cann 已提交
2802 2803 2804 2805 2806 2807
    pub fn trait_relevant_for_never(self, did: DefId) -> bool {
        self.associated_items(did).any(|item| {
            item.relevant_for_never()
        })
    }

2808
    pub fn opt_associated_item(self, def_id: DefId) -> Option<AssociatedItem> {
2809 2810
        let is_associated_item = if let Some(hir_id) = self.hir().as_local_hir_id(def_id) {
            match self.hir().get_by_hir_id(hir_id) {
V
varkor 已提交
2811
                Node::TraitItem(_) | Node::ImplItem(_) => true,
2812 2813 2814
                _ => false,
            }
        } else {
2815 2816 2817 2818
            match self.def_kind(def_id).expect("no def for def-id") {
                DefKind::AssociatedConst
                | DefKind::Method
                | DefKind::AssociatedTy => true,
2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829
                _ => false,
            }
        };

        if is_associated_item {
            Some(self.associated_item(def_id))
        } else {
            None
        }
    }

2830 2831
    fn associated_item_from_trait_item_ref(self,
                                           parent_def_id: DefId,
2832
                                           parent_vis: &hir::Visibility,
2833
                                           trait_item_ref: &hir::TraitItemRef)
2834
                                           -> AssociatedItem {
L
ljedrz 已提交
2835
        let def_id = self.hir().local_def_id_from_hir_id(trait_item_ref.id.hir_id);
2836 2837 2838 2839
        let (kind, has_self) = match trait_item_ref.kind {
            hir::AssociatedItemKind::Const => (ty::AssociatedKind::Const, false),
            hir::AssociatedItemKind::Method { has_self } => {
                (ty::AssociatedKind::Method, has_self)
2840
            }
2841
            hir::AssociatedItemKind::Type => (ty::AssociatedKind::Type, false),
O
Oliver Schneider 已提交
2842
            hir::AssociatedItemKind::Existential => bug!("only impls can have existentials"),
2843 2844 2845
        };

        AssociatedItem {
2846
            ident: trait_item_ref.ident,
2847
            kind,
2848
            // Visibility of trait items is inherited from their traits.
L
ljedrz 已提交
2849
            vis: Visibility::from_hir(parent_vis, trait_item_ref.id.hir_id, self),
2850
            defaultness: trait_item_ref.defaultness,
2851
            def_id,
2852 2853 2854 2855 2856 2857 2858 2859 2860
            container: TraitContainer(parent_def_id),
            method_has_self_argument: has_self
        }
    }

    fn associated_item_from_impl_item_ref(self,
                                          parent_def_id: DefId,
                                          impl_item_ref: &hir::ImplItemRef)
                                          -> AssociatedItem {
L
ljedrz 已提交
2861
        let def_id = self.hir().local_def_id_from_hir_id(impl_item_ref.id.hir_id);
2862 2863 2864 2865 2866 2867
        let (kind, has_self) = match impl_item_ref.kind {
            hir::AssociatedItemKind::Const => (ty::AssociatedKind::Const, false),
            hir::AssociatedItemKind::Method { has_self } => {
                (ty::AssociatedKind::Method, has_self)
            }
            hir::AssociatedItemKind::Type => (ty::AssociatedKind::Type, false),
O
Oliver Schneider 已提交
2868
            hir::AssociatedItemKind::Existential => (ty::AssociatedKind::Existential, false),
2869 2870
        };

2871 2872
        AssociatedItem {
            ident: impl_item_ref.ident,
2873
            kind,
2874
            // Visibility of trait impl items doesn't matter.
L
ljedrz 已提交
2875
            vis: ty::Visibility::from_hir(&impl_item_ref.vis, impl_item_ref.id.hir_id, self),
2876
            defaultness: impl_item_ref.defaultness,
2877
            def_id,
2878 2879 2880 2881 2882
            container: ImplContainer(parent_def_id),
            method_has_self_argument: has_self
        }
    }

L
ljedrz 已提交
2883
    pub fn field_index(self, hir_id: hir::HirId, tables: &TypeckTables<'_>) -> usize {
2884 2885 2886 2887 2888
        tables.field_indices().get(hir_id).cloned().expect("no index for a field")
    }

    pub fn find_field_index(self, ident: Ident, variant: &VariantDef) -> Option<usize> {
        variant.fields.iter().position(|field| {
2889
            self.adjust_ident(ident, variant.def_id, hir::DUMMY_HIR_ID).0 == field.ident.modern()
2890 2891 2892
        })
    }

2893 2894 2895
    pub fn associated_items(
        self,
        def_id: DefId,
2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906
    ) -> AssociatedItemsIterator<'a, 'gcx, 'tcx> {
        // Ideally, we would use `-> impl Iterator` here, but it falls
        // afoul of the conservative "capture [restrictions]" we put
        // in place, so we use a hand-written iterator.
        //
        // [restrictions]: https://github.com/rust-lang/rust/issues/34511#issuecomment-373423999
        AssociatedItemsIterator {
            tcx: self,
            def_ids: self.associated_item_def_ids(def_id),
            next_index: 0,
        }
2907 2908
    }

2909
    /// Returns `true` if the impls are the same polarity and the trait either
2910
    /// has no items or is annotated #[marker] and prevents item overrides.
2911 2912 2913 2914
    pub fn impls_are_allowed_to_overlap(self, def_id1: DefId, def_id2: DefId)
                                        -> Option<ImplOverlapKind>
    {
        let is_legit = if self.features().overlapping_marker_traits {
2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925
            let trait1_is_empty = self.impl_trait_ref(def_id1)
                .map_or(false, |trait_ref| {
                    self.associated_item_def_ids(trait_ref.def_id).is_empty()
                });
            let trait2_is_empty = self.impl_trait_ref(def_id2)
                .map_or(false, |trait_ref| {
                    self.associated_item_def_ids(trait_ref.def_id).is_empty()
                });
            self.impl_polarity(def_id1) == self.impl_polarity(def_id2)
                && trait1_is_empty
                && trait2_is_empty
2926
        } else {
2927 2928 2929 2930 2931 2932 2933
            let is_marker_impl = |def_id: DefId| -> bool {
                let trait_ref = self.impl_trait_ref(def_id);
                trait_ref.map_or(false, |tr| self.trait_def(tr.def_id).is_marker)
            };
            self.impl_polarity(def_id1) == self.impl_polarity(def_id2)
                && is_marker_impl(def_id1)
                && is_marker_impl(def_id2)
2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956
        };

        if is_legit {
            debug!("impls_are_allowed_to_overlap({:?}, {:?}) = Some(Permitted)",
                  def_id1, def_id2);
            Some(ImplOverlapKind::Permitted)
        } else {
            if let Some(self_ty1) = self.issue33140_self_ty(def_id1) {
                if let Some(self_ty2) = self.issue33140_self_ty(def_id2) {
                    if self_ty1 == self_ty2 {
                        debug!("impls_are_allowed_to_overlap({:?}, {:?}) - issue #33140 HACK",
                               def_id1, def_id2);
                        return Some(ImplOverlapKind::Issue33140);
                    } else {
                        debug!("impls_are_allowed_to_overlap({:?}, {:?}) - found {:?} != {:?}",
                              def_id1, def_id2, self_ty1, self_ty2);
                    }
                }
            }

            debug!("impls_are_allowed_to_overlap({:?}, {:?}) = None",
                  def_id1, def_id2);
            None
2957
        }
S
Sean Griffin 已提交
2958 2959
    }

2960
    /// Returns `ty::VariantDef` if `res` refers to a struct,
2961
    /// or variant or their constructors, panics otherwise.
2962 2963 2964
    pub fn expect_variant_res(self, res: Res) -> &'tcx VariantDef {
        match res {
            Res::Def(DefKind::Variant, did) => {
2965
                let enum_did = self.parent(did).unwrap();
2966
                self.adt_def(enum_did).variant_with_id(did)
2967
            }
2968
            Res::Def(DefKind::Struct, did) | Res::Def(DefKind::Union, did) => {
2969
                self.adt_def(did).non_enum_variant()
2970
            }
2971
            Res::Def(DefKind::Ctor(CtorOf::Variant, ..), variant_ctor_did) => {
2972 2973 2974 2975
                let variant_did = self.parent(variant_ctor_did).unwrap();
                let enum_did = self.parent(variant_did).unwrap();
                self.adt_def(enum_did).variant_with_ctor_id(variant_ctor_did)
            }
2976
            Res::Def(DefKind::Ctor(CtorOf::Struct, ..), ctor_did) => {
2977 2978
                let struct_did = self.parent(ctor_did).expect("struct ctor has no parent");
                self.adt_def(struct_did).non_enum_variant()
2979
            }
2980
            _ => bug!("expect_variant_res used with unexpected res {:?}", res)
2981 2982 2983
        }
    }

2984
    pub fn item_name(self, id: DefId) -> InternedString {
2985
        if id.index == CRATE_DEF_INDEX {
2986
            self.original_crate_name(id.krate).as_interned_str()
2987
        } else {
2988
            let def_key = self.def_key(id);
2989
            match def_key.disambiguated_data.data {
2990 2991
                // The name of a constructor is that of its parent.
                hir_map::DefPathData::Ctor =>
2992 2993 2994 2995 2996
                    self.item_name(DefId {
                        krate: id.krate,
                        index: def_key.parent.unwrap()
                    }),
                _ => def_key.disambiguated_data.data.get_opt_name().unwrap_or_else(|| {
2997
                    bug!("item_name: no name for {:?}", self.def_path(id));
2998
                }),
2999
            }
3000 3001 3002
        }
    }

A
Alexander Regueiro 已提交
3003
    /// Returns the possibly-auto-generated MIR of a `(DefId, Subst)` pair.
3004
    pub fn instance_mir(self, instance: ty::InstanceDef<'gcx>)
3005
                        -> &'gcx Mir<'gcx>
3006 3007
    {
        match instance {
N
Niko Matsakis 已提交
3008
            ty::InstanceDef::Item(did) => {
3009
                self.optimized_mir(did)
N
Niko Matsakis 已提交
3010
            }
M
Masaki Hara 已提交
3011
            ty::InstanceDef::VtableShim(..) |
N
Niko Matsakis 已提交
3012 3013 3014 3015
            ty::InstanceDef::Intrinsic(..) |
            ty::InstanceDef::FnPtrShim(..) |
            ty::InstanceDef::Virtual(..) |
            ty::InstanceDef::ClosureOnceShim { .. } |
3016
            ty::InstanceDef::DropGlue(..) |
S
scalexm 已提交
3017
            ty::InstanceDef::CloneShim(..) => {
3018
                self.mir_shims(instance)
N
Niko Matsakis 已提交
3019
            }
3020 3021 3022
        }
    }

A
Alexander Regueiro 已提交
3023
    /// Gets the attributes of a definition.
3024
    pub fn get_attrs(self, did: DefId) -> Attributes<'gcx> {
L
ljedrz 已提交
3025 3026
        if let Some(id) = self.hir().as_local_hir_id(did) {
            Attributes::Borrowed(self.hir().attrs_by_hir_id(id))
3027
        } else {
A
achernyak 已提交
3028
            Attributes::Owned(self.item_attrs(did))
3029
        }
3030 3031
    }

A
Alexander Regueiro 已提交
3032
    /// Determines whether an item is annotated with an attribute.
3033
    pub fn has_attr(self, did: DefId, attr: &str) -> bool {
3034
        attr::contains_name(&self.get_attrs(did), attr)
3035
    }
3036

3037
    /// Returns `true` if this is an `auto trait`.
3038
    pub fn trait_is_auto(self, trait_def_id: DefId) -> bool {
3039
        self.trait_def(trait_def_id).has_auto_impl
3040
    }
3041

J
John Kåre Alsaker 已提交
3042 3043 3044 3045
    pub fn generator_layout(self, def_id: DefId) -> &'tcx GeneratorLayout<'tcx> {
        self.optimized_mir(def_id).generator_layout.as_ref().unwrap()
    }

A
Alexander Regueiro 已提交
3046 3047
    /// Given the `DefId` of an impl, returns the `DefId` of the trait it implements.
    /// If it implements no trait, returns `None`.
3048
    pub fn trait_id_of_impl(self, def_id: DefId) -> Option<DefId> {
3049
        self.impl_trait_ref(def_id).map(|tr| tr.def_id)
3050
    }
3051

A
Alexander Regueiro 已提交
3052 3053
    /// If the given defid describes a method belonging to an impl, returns the
    /// `DefId` of the impl that the method belongs to; otherwise, returns `None`.
3054
    pub fn impl_of_method(self, def_id: DefId) -> Option<DefId> {
3055
        let item = if def_id.krate != LOCAL_CRATE {
3056
            if let Some(DefKind::Method) = self.def_kind(def_id) {
3057 3058 3059 3060 3061
                Some(self.associated_item(def_id))
            } else {
                None
            }
        } else {
3062
            self.opt_associated_item(def_id)
3063 3064
        };

3065 3066 3067 3068
        item.and_then(|trait_item|
            match trait_item.container {
                TraitContainer(_) => None,
                ImplContainer(def_id) => Some(def_id),
3069
            }
3070
        )
3071 3072
    }

3073 3074
    /// Looks up the span of `impl_did` if the impl is local; otherwise returns `Err`
    /// with the name of the crate containing the impl.
3075
    pub fn span_of_impl(self, impl_did: DefId) -> Result<Span, Symbol> {
3076
        if impl_did.is_local() {
L
ljedrz 已提交
3077 3078
            let hir_id = self.hir().as_local_hir_id(impl_did).unwrap();
            Ok(self.hir().span_by_hir_id(hir_id))
3079
        } else {
3080
            Err(self.crate_name(impl_did.krate))
3081 3082
        }
    }
J
Jeffrey Seyfried 已提交
3083

A
Alexander Regueiro 已提交
3084 3085 3086
    /// Hygienically compares a use-site name (`use_name`) for a field or an associated item with
    /// its supposed definition name (`def_name`). The method also needs `DefId` of the supposed
    /// definition's parent/scope to perform comparison.
3087
    pub fn hygienic_eq(self, use_name: Ident, def_name: Ident, def_parent_def_id: DefId) -> bool {
L
ljedrz 已提交
3088
        self.adjust_ident(use_name, def_parent_def_id, hir::DUMMY_HIR_ID).0 == def_name.modern()
J
Jeffrey Seyfried 已提交
3089 3090
    }

L
ljedrz 已提交
3091
    pub fn adjust_ident(self, mut ident: Ident, scope: DefId, block: hir::HirId) -> (Ident, DefId) {
3092 3093
        ident = ident.modern();
        let target_expansion = match scope.krate {
3094
            LOCAL_CRATE => self.hir().definitions().expansion_that_defined(scope.index),
J
Jeffrey Seyfried 已提交
3095 3096
            _ => Mark::root(),
        };
3097 3098
        let scope = match ident.span.adjust(target_expansion) {
            Some(actual_expansion) =>
3099
                self.hir().definitions().parent_module_of_macro_def(actual_expansion),
L
ljedrz 已提交
3100 3101
            None if block == hir::DUMMY_HIR_ID => DefId::local(CRATE_DEF_INDEX), // Dummy DefId
            None => self.hir().get_module_parent_by_hir_id(block),
J
Jeffrey Seyfried 已提交
3102 3103 3104
        };
        (ident, scope)
    }
3105
}
3106

3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122
pub struct AssociatedItemsIterator<'a, 'gcx: 'tcx, 'tcx: 'a> {
    tcx: TyCtxt<'a, 'gcx, 'tcx>,
    def_ids: Lrc<Vec<DefId>>,
    next_index: usize,
}

impl Iterator for AssociatedItemsIterator<'_, '_, '_> {
    type Item = AssociatedItem;

    fn next(&mut self) -> Option<AssociatedItem> {
        let def_id = self.def_ids.get(self.next_index)?;
        self.next_index += 1;
        Some(self.tcx.associated_item(*def_id))
    }
}

L
ljedrz 已提交
3123
fn associated_item<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, def_id: DefId) -> AssociatedItem {
L
ljedrz 已提交
3124 3125 3126 3127
    let id = tcx.hir().as_local_hir_id(def_id).unwrap();
    let parent_id = tcx.hir().get_parent_item(id);
    let parent_def_id = tcx.hir().local_def_id_from_hir_id(parent_id);
    let parent_item = tcx.hir().expect_item_by_hir_id(parent_id);
3128
    match parent_item.node {
C
csmoe 已提交
3129
        hir::ItemKind::Impl(.., ref impl_item_refs) => {
L
ljedrz 已提交
3130
            if let Some(impl_item_ref) = impl_item_refs.iter().find(|i| i.id.hir_id == id) {
3131 3132
                let assoc_item = tcx.associated_item_from_impl_item_ref(parent_def_id,
                                                                        impl_item_ref);
3133 3134
                debug_assert_eq!(assoc_item.def_id, def_id);
                return assoc_item;
3135 3136 3137
            }
        }

C
csmoe 已提交
3138
        hir::ItemKind::Trait(.., ref trait_item_refs) => {
L
ljedrz 已提交
3139
            if let Some(trait_item_ref) = trait_item_refs.iter().find(|i| i.id.hir_id == id) {
3140 3141 3142
                let assoc_item = tcx.associated_item_from_trait_item_ref(parent_def_id,
                                                                         &parent_item.vis,
                                                                         trait_item_ref);
3143 3144
                debug_assert_eq!(assoc_item.def_id, def_id);
                return assoc_item;
3145 3146 3147
            }
        }

3148
        _ => { }
3149
    }
3150 3151 3152 3153

    span_bug!(parent_item.span,
              "unexpected parent of trait or impl item or item not found: {:?}",
              parent_item.node)
3154 3155
}

3156
#[derive(Clone, HashStable)]
3157 3158
pub struct AdtSizedConstraint<'tcx>(pub &'tcx [Ty<'tcx>]);

A
Alexander Regueiro 已提交
3159
/// Calculates the `Sized` constraint.
3160
///
3161
/// In fact, there are only a few options for the types in the constraint:
3162 3163 3164 3165
///     - an obviously-unsized type
///     - a type parameter or projection whose Sizedness can't be known
///     - a tuple of type parameters or projections, if there are multiple
///       such.
V
varkor 已提交
3166
///     - a Error, if a type contained itself. The representability
3167 3168 3169
///       check should catch this case.
fn adt_sized_constraint<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                  def_id: DefId)
3170
                                  -> AdtSizedConstraint<'tcx> {
3171
    let def = tcx.adt_def(def_id);
3172

3173
    let result = tcx.mk_type_list(def.variants.iter().flat_map(|v| {
3174 3175
        v.fields.last()
    }).flat_map(|f| {
3176
        def.sized_constraint_for_ty(tcx, tcx.type_of(f.did))
3177
    }));
3178

3179
    debug!("adt_sized_constraint: {:?} => {:?}", def, result);
3180

3181
    AdtSizedConstraint(result)
3182 3183
}

3184 3185
fn associated_item_def_ids<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                     def_id: DefId)
3186
                                     -> Lrc<Vec<DefId>> {
L
ljedrz 已提交
3187 3188
    let id = tcx.hir().as_local_hir_id(def_id).unwrap();
    let item = tcx.hir().expect_item_by_hir_id(id);
3189
    let vec: Vec<_> = match item.node {
C
csmoe 已提交
3190
        hir::ItemKind::Trait(.., ref trait_item_refs) => {
3191 3192
            trait_item_refs.iter()
                           .map(|trait_item_ref| trait_item_ref.id)
L
ljedrz 已提交
3193
                           .map(|id| tcx.hir().local_def_id_from_hir_id(id.hir_id))
3194 3195
                           .collect()
        }
C
csmoe 已提交
3196
        hir::ItemKind::Impl(.., ref impl_item_refs) => {
3197 3198
            impl_item_refs.iter()
                          .map(|impl_item_ref| impl_item_ref.id)
L
ljedrz 已提交
3199
                          .map(|id| tcx.hir().local_def_id_from_hir_id(id.hir_id))
3200 3201
                          .collect()
        }
C
csmoe 已提交
3202
        hir::ItemKind::TraitAlias(..) => vec![],
3203 3204
        _ => span_bug!(item.span, "associated_item_def_ids: not impl or trait")
    };
3205
    Lrc::new(vec)
3206 3207
}

A
achernyak 已提交
3208
fn def_span<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, def_id: DefId) -> Span {
3209
    tcx.hir().span_if_local(def_id).unwrap()
A
achernyak 已提交
3210 3211
}

A
Alexander Regueiro 已提交
3212 3213 3214
/// If the given `DefId` describes an item belonging to a trait,
/// returns the `DefId` of the trait that the trait item belongs to;
/// otherwise, returns `None`.
A
achernyak 已提交
3215 3216 3217 3218 3219 3220 3221 3222 3223 3224
fn trait_of_item<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>, def_id: DefId) -> Option<DefId> {
    tcx.opt_associated_item(def_id)
        .and_then(|associated_item| {
            match associated_item.container {
                TraitContainer(def_id) => Some(def_id),
                ImplContainer(_) => None
            }
        })
}

3225
/// Yields the parent function's `DefId` if `def_id` is an `impl Trait` definition.
3226
pub fn is_impl_trait_defn(tcx: TyCtxt<'_, '_, '_>, def_id: DefId) -> Option<DefId> {
3227 3228
    if let Some(hir_id) = tcx.hir().as_local_hir_id(def_id) {
        if let Node::Item(item) = tcx.hir().get_by_hir_id(hir_id) {
3229 3230 3231 3232 3233 3234 3235 3236
            if let hir::ItemKind::Existential(ref exist_ty) = item.node {
                return exist_ty.impl_trait_fn;
            }
        }
    }
    None
}

3237
/// See `ParamEnv` struct definition for details.
3238
fn param_env<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
3239
                       def_id: DefId)
L
ljedrz 已提交
3240 3241
                       -> ParamEnv<'tcx>
{
O
Oliver Schneider 已提交
3242
    // The param_env of an impl Trait type is its defining function's param_env
3243 3244
    if let Some(parent) = is_impl_trait_defn(tcx, def_id) {
        return param_env(tcx, parent);
3245
    }
3246 3247
    // Compute the bounds on Self and the type parameters.

N
Niko Matsakis 已提交
3248 3249
    let InstantiatedPredicates { predicates } =
        tcx.predicates_of(def_id).instantiate_identity(tcx);
3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262

    // Finally, we have to normalize the bounds in the environment, in
    // case they contain any associated type projections. This process
    // can yield errors if the put in illegal associated types, like
    // `<i32 as Foo>::Bar` where `i32` does not implement `Foo`. We
    // report these errors right here; this doesn't actually feel
    // right to me, because constructing the environment feels like a
    // kind of a "idempotent" action, but I'm not sure where would be
    // a better place. In practice, we construct environments for
    // every fn once during type checking, and we'll abort if there
    // are any errors at that point, so after type checking you can be
    // sure that this will succeed without errors anyway.

S
scalexm 已提交
3263 3264 3265
    let unnormalized_env = ty::ParamEnv::new(
        tcx.intern_predicates(&predicates),
        traits::Reveal::UserFacing,
3266
        if tcx.sess.opts.debugging_opts.chalk { Some(def_id) } else { None }
S
scalexm 已提交
3267
    );
3268

L
ljedrz 已提交
3269 3270
    let body_id = tcx.hir().as_local_hir_id(def_id).map_or(hir::DUMMY_HIR_ID, |id| {
        tcx.hir().maybe_body_owned_by_by_hir_id(id).map_or(id, |body| body.hir_id)
3271 3272 3273 3274
    });
    let cause = traits::ObligationCause::misc(tcx.def_span(def_id), body_id);
    traits::normalize_param_env_or_error(tcx, def_id, unnormalized_env, cause)
}
A
achernyak 已提交
3275

3276
fn crate_disambiguator<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
3277
                                 crate_num: CrateNum) -> CrateDisambiguator {
3278 3279 3280 3281
    assert_eq!(crate_num, LOCAL_CRATE);
    tcx.sess.local_crate_disambiguator()
}

3282 3283 3284 3285 3286 3287
fn original_crate_name<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                 crate_num: CrateNum) -> Symbol {
    assert_eq!(crate_num, LOCAL_CRATE);
    tcx.crate_name.clone()
}

3288 3289 3290 3291
fn crate_hash<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                        crate_num: CrateNum)
                        -> Svh {
    assert_eq!(crate_num, LOCAL_CRATE);
3292
    tcx.hir().crate_hash
3293 3294
}

V
varkor 已提交
3295 3296 3297 3298
fn instance_def_size_estimate<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                        instance_def: InstanceDef<'tcx>)
                                        -> usize {
    match instance_def {
3299 3300 3301
        InstanceDef::Item(..) |
        InstanceDef::DropGlue(..) => {
            let mir = tcx.instance_mir(instance_def);
V
varkor 已提交
3302 3303
            mir.basic_blocks().iter().map(|bb| bb.statements.len()).sum()
        },
3304
        // Estimate the size of other compiler-generated shims to be 1.
V
varkor 已提交
3305 3306 3307 3308
        _ => 1
    }
}

A
Alexander Regueiro 已提交
3309
/// If `def_id` is an issue 33140 hack impl, returns its self type; otherwise, returns `None`.
3310
///
A
Alexander Regueiro 已提交
3311
/// See [`ImplOverlapKind::Issue33140`] for more details.
3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360
fn issue33140_self_ty<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                def_id: DefId)
                                -> Option<Ty<'tcx>>
{
    debug!("issue33140_self_ty({:?})", def_id);

    let trait_ref = tcx.impl_trait_ref(def_id).unwrap_or_else(|| {
        bug!("issue33140_self_ty called on inherent impl {:?}", def_id)
    });

    debug!("issue33140_self_ty({:?}), trait-ref={:?}", def_id, trait_ref);

    let is_marker_like =
        tcx.impl_polarity(def_id) == hir::ImplPolarity::Positive &&
        tcx.associated_item_def_ids(trait_ref.def_id).is_empty();

    // Check whether these impls would be ok for a marker trait.
    if !is_marker_like {
        debug!("issue33140_self_ty - not marker-like!");
        return None;
    }

    // impl must be `impl Trait for dyn Marker1 + Marker2 + ...`
    if trait_ref.substs.len() != 1 {
        debug!("issue33140_self_ty - impl has substs!");
        return None;
    }

    let predicates = tcx.predicates_of(def_id);
    if predicates.parent.is_some() || !predicates.predicates.is_empty() {
        debug!("issue33140_self_ty - impl has predicates {:?}!", predicates);
        return None;
    }

    let self_ty = trait_ref.self_ty();
    let self_ty_matches = match self_ty.sty {
        ty::Dynamic(ref data, ty::ReStatic) => data.principal().is_none(),
        _ => false
    };

    if self_ty_matches {
        debug!("issue33140_self_ty - MATCHES!");
        Some(self_ty)
    } else {
        debug!("issue33140_self_ty - non-matching self type");
        None
    }
}

3361
pub fn provide(providers: &mut ty::query::Providers<'_>) {
3362
    context::provide(providers);
3363
    erase_regions::provide(providers);
3364 3365
    layout::provide(providers);
    util::provide(providers);
3366
    constness::provide(providers);
3367
    *providers = ty::query::Providers {
3368
        associated_item,
3369
        associated_item_def_ids,
3370
        adt_sized_constraint,
A
achernyak 已提交
3371
        def_span,
3372
        param_env,
A
achernyak 已提交
3373
        trait_of_item,
3374
        crate_disambiguator,
3375
        original_crate_name,
3376
        crate_hash,
3377
        trait_impls_of: trait_def::trait_impls_of_provider,
V
varkor 已提交
3378
        instance_def_size_estimate,
3379
        issue33140_self_ty,
3380 3381 3382 3383
        ..*providers
    };
}

3384 3385 3386
/// A map for the local crate mapping each type to a vector of its
/// inherent impls. This is not meant to be used outside of coherence;
/// rather, you should request the vector for a specific type via
3387 3388
/// `tcx.inherent_impls(def_id)` so as to minimize your dependencies
/// (constructing this map requires touching the entire crate).
3389
#[derive(Clone, Debug, Default, HashStable)]
3390
pub struct CrateInherentImpls {
3391
    pub inherent_impls: DefIdMap<Lrc<Vec<DefId>>>,
3392
}
A
Ariel Ben-Yehuda 已提交
3393

3394
#[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, RustcEncodable, RustcDecodable)]
3395 3396 3397 3398 3399 3400
pub struct SymbolName {
    // FIXME: we don't rely on interning or equality here - better have
    // this be a `&'tcx str`.
    pub name: InternedString
}

3401 3402 3403 3404
impl_stable_hash_for!(struct self::SymbolName {
    name
});

3405 3406 3407
impl SymbolName {
    pub fn new(name: &str) -> SymbolName {
        SymbolName {
3408
            name: Symbol::intern(name).as_interned_str()
3409 3410
        }
    }
3411

3412 3413 3414
    pub fn as_str(&self) -> LocalInternedString {
        self.name.as_str()
    }
3415 3416 3417
}

impl fmt::Display for SymbolName {
3418
    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
3419 3420 3421
        fmt::Display::fmt(&self.name, fmt)
    }
}
3422 3423

impl fmt::Debug for SymbolName {
3424
    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
3425 3426 3427
        fmt::Display::fmt(&self.name, fmt)
    }
}