1 //! Simplification of where-clauses and parameter bounds into a prettier and
2 //! more canonical form.
4 //! Currently all cross-crate-inlined function use `rustc::ty` to reconstruct
5 //! the AST (e.g., see all of `clean::inline`), but this is not always a
6 //! non-lossy transformation. The current format of storage for where-clauses
7 //! for functions and such is simply a list of predicates. One example of this
8 //! is that the AST predicate of: `where T: Trait<Foo = Bar>` is encoded as:
9 //! `where T: Trait, <T as Trait>::Foo = Bar`.
11 //! This module attempts to reconstruct the original where and/or parameter
12 //! bounds by special casing scenarios such as these. Fun!
15 use std::collections::BTreeMap;
17 use rustc::hir::def_id::DefId;
20 use crate::clean::GenericArgs as PP;
21 use crate::clean::WherePredicate as WP;
23 use crate::core::DocContext;
25 pub fn where_clauses(cx: &DocContext<'_>, clauses: Vec<WP>) -> Vec<WP> {
26 // First, partition the where clause into its separate components
27 let mut params: BTreeMap<_, Vec<_>> = BTreeMap::new();
28 let mut lifetimes = Vec::new();
29 let mut equalities = Vec::new();
30 let mut tybounds = Vec::new();
32 for clause in clauses {
34 WP::BoundPredicate { ty, bounds } => {
36 clean::Generic(s) => params.entry(s).or_default()
38 t => tybounds.push((t, ty_bounds(bounds))),
41 WP::RegionPredicate { lifetime, bounds } => {
42 lifetimes.push((lifetime, bounds));
44 WP::EqPredicate { lhs, rhs } => equalities.push((lhs, rhs)),
48 // Simplify the type parameter bounds on all the generics
49 let mut params = params.into_iter().map(|(k, v)| {
51 }).collect::<BTreeMap<_, _>>();
53 // Look for equality predicates on associated types that can be merged into
54 // general bound predicates
55 equalities.retain(|&(ref lhs, ref rhs)| {
56 let (self_, trait_did, name) = if let Some(p) = lhs.projection() {
61 let generic = match self_ {
62 clean::Generic(s) => s,
65 let bounds = match params.get_mut(generic) {
70 merge_bounds(cx, bounds, trait_did, name, rhs)
73 // And finally, let's reassemble everything
74 let mut clauses = Vec::new();
75 clauses.extend(lifetimes.into_iter().map(|(lt, bounds)| {
76 WP::RegionPredicate { lifetime: lt, bounds: bounds }
78 clauses.extend(params.into_iter().map(|(k, v)| {
80 ty: clean::Generic(k),
84 clauses.extend(tybounds.into_iter().map(|(ty, bounds)| {
85 WP::BoundPredicate { ty: ty, bounds: bounds }
87 clauses.extend(equalities.into_iter().map(|(lhs, rhs)| {
88 WP::EqPredicate { lhs: lhs, rhs: rhs }
94 cx: &clean::DocContext<'_>,
95 bounds: &mut Vec<clean::GenericBound>,
100 !bounds.iter_mut().any(|b| {
101 let trait_ref = match *b {
102 clean::GenericBound::TraitBound(ref mut tr, _) => tr,
103 clean::GenericBound::Outlives(..) => return false,
105 let (did, path) = match trait_ref.trait_ {
106 clean::ResolvedPath { did, ref mut path, ..} => (did, path),
109 // If this QPath's trait `trait_did` is the same as, or a supertrait
110 // of, the bound's trait `did` then we can keep going, otherwise
111 // this is just a plain old equality bound.
112 if !trait_is_same_or_supertrait(cx, did, trait_did) {
115 let last = path.segments.last_mut().expect("segments were empty");
117 PP::AngleBracketed { ref mut bindings, .. } => {
118 bindings.push(clean::TypeBinding {
119 name: name.to_string(),
120 kind: clean::TypeBindingKind::Equality {
125 PP::Parenthesized { ref mut output, .. } => match output {
126 Some(o) => assert!(o == rhs),
127 None => if *rhs != clean::Type::Tuple(Vec::new()) {
128 *output = Some(rhs.clone());
136 pub fn ty_params(mut params: Vec<clean::GenericParamDef>) -> Vec<clean::GenericParamDef> {
137 for param in &mut params {
139 clean::GenericParamDefKind::Type { ref mut bounds, .. } => {
140 *bounds = ty_bounds(mem::take(bounds));
142 _ => panic!("expected only type parameters"),
148 fn ty_bounds(bounds: Vec<clean::GenericBound>) -> Vec<clean::GenericBound> {
152 fn trait_is_same_or_supertrait(cx: &DocContext<'_>, child: DefId,
153 trait_: DefId) -> bool {
157 let predicates = cx.tcx.super_predicates_of(child);
158 debug_assert!(cx.tcx.generics_of(child).has_self);
159 let self_ty = cx.tcx.types.self_param;
160 predicates.predicates.iter().filter_map(|(pred, _)| {
161 if let ty::Predicate::Trait(ref pred) = *pred {
162 if pred.skip_binder().trait_ref.self_ty() == self_ty {
170 }).any(|did| trait_is_same_or_supertrait(cx, did, trait_))