Skip to content

Commit

Permalink
rustc: remove obsolete hacks from ppaux, relating to normalization un…
Browse files Browse the repository at this point in the history
…der HRTB.
  • Loading branch information
eddyb committed Mar 15, 2019
1 parent fb53bb9 commit 030cdc9
Show file tree
Hide file tree
Showing 4 changed files with 142 additions and 104 deletions.
47 changes: 25 additions & 22 deletions src/librustc/ty/print/mod.rs
@@ -1,6 +1,6 @@
use crate::hir::map::DefPathData;
use crate::hir::def_id::{CrateNum, DefId};
use crate::ty::{self, DefIdTree, Ty, TyCtxt, TypeFoldable};
use crate::ty::{self, DefIdTree, Ty, TyCtxt};
use crate::ty::subst::{Subst, SubstsRef};

use rustc_data_structures::fx::FxHashSet;
Expand All @@ -16,19 +16,6 @@ pub use self::pretty::*;
// FIXME(eddyb) this module uses `pub(crate)` for things used only
// from `ppaux` - when that is removed, they can be re-privatized.

struct LateBoundRegionNameCollector(FxHashSet<InternedString>);
impl<'tcx> ty::fold::TypeVisitor<'tcx> for LateBoundRegionNameCollector {
fn visit_region(&mut self, r: ty::Region<'tcx>) -> bool {
match *r {
ty::ReLateBound(_, ty::BrNamed(_, name)) => {
self.0.insert(name);
},
_ => {},
}
r.super_visit_with(self)
}
}

#[derive(Default)]
pub(crate) struct PrintConfig {
used_region_names: Option<FxHashSet<InternedString>>,
Expand Down Expand Up @@ -67,14 +54,6 @@ impl<'a, 'gcx, 'tcx, P> PrintCx<'a, 'gcx, 'tcx, P> {
pub(crate) fn with_tls_tcx<R>(printer: P, f: impl FnOnce(PrintCx<'_, '_, '_, P>) -> R) -> R {
ty::tls::with(|tcx| PrintCx::with(tcx, printer, f))
}
fn prepare_late_bound_region_info<T>(&mut self, value: &ty::Binder<T>)
where T: TypeFoldable<'tcx>
{
let mut collector = LateBoundRegionNameCollector(Default::default());
value.visit_with(&mut collector);
self.config.used_region_names = Some(collector.0);
self.config.region_index = 0;
}
}

pub trait Print<'tcx, P> {
Expand Down Expand Up @@ -322,3 +301,27 @@ pub fn characteristic_def_id_of_type(ty: Ty<'_>) -> Option<DefId> {
ty::Float(_) => None,
}
}

impl<P: Printer> Print<'tcx, P> for ty::RegionKind {
type Output = P::Region;
type Error = P::Error;
fn print(&self, cx: PrintCx<'_, '_, 'tcx, P>) -> Result<Self::Output, Self::Error> {
cx.print_region(self)
}
}

impl<P: Printer> Print<'tcx, P> for ty::Region<'_> {
type Output = P::Region;
type Error = P::Error;
fn print(&self, cx: PrintCx<'_, '_, 'tcx, P>) -> Result<Self::Output, Self::Error> {
cx.print_region(self)
}
}

impl<P: Printer> Print<'tcx, P> for Ty<'tcx> {
type Output = P::Type;
type Error = P::Error;
fn print(&self, cx: PrintCx<'_, '_, 'tcx, P>) -> Result<Self::Output, Self::Error> {
cx.print_type(self)
}
}
56 changes: 52 additions & 4 deletions src/librustc/ty/print/pretty.rs
Expand Up @@ -205,6 +205,15 @@ pub trait PrettyPrinter:
self.print_def_path(def_id, substs, iter::empty())
}

fn in_binder<T>(
self: PrintCx<'_, '_, 'tcx, Self>,
value: &ty::Binder<T>,
) -> Result<Self, Self::Error>
where T: Print<'tcx, Self, Output = Self, Error = Self::Error> + TypeFoldable<'tcx>
{
value.skip_binder().print(self)
}

/// Print `<...>` around what `f` prints.
fn generic_delimiters<'gcx, 'tcx>(
self: PrintCx<'_, 'gcx, 'tcx, Self>,
Expand Down Expand Up @@ -784,6 +793,15 @@ impl<F: fmt::Write> PrettyPrinter for FmtPrinter<F> {
Ok(printer)
}

fn in_binder<T>(
self: PrintCx<'_, '_, 'tcx, Self>,
value: &ty::Binder<T>,
) -> Result<Self, Self::Error>
where T: Print<'tcx, Self, Output = Self, Error = Self::Error> + TypeFoldable<'tcx>
{
self.pretty_in_binder(value)
}

fn generic_delimiters<'gcx, 'tcx>(
mut self: PrintCx<'_, 'gcx, 'tcx, Self>,
f: impl FnOnce(PrintCx<'_, 'gcx, 'tcx, Self>) -> Result<Self, Self::Error>,
Expand Down Expand Up @@ -1125,7 +1143,7 @@ impl<'gcx, 'tcx, P: PrettyPrinter> PrintCx<'_, 'gcx, 'tcx, P> {
p!(write(" "), print(witness), write("]"))
},
ty::GeneratorWitness(types) => {
nest!(|cx| cx.pretty_in_binder(&types))
nest!(|cx| cx.in_binder(&types))
}
ty::Closure(did, substs) => {
let upvar_tys = substs.upvar_tys(did, self.tcx);
Expand Down Expand Up @@ -1257,9 +1275,6 @@ impl<'gcx, 'tcx, P: PrettyPrinter> PrintCx<'_, 'gcx, 'tcx, P> {
})
};

// NOTE(eddyb) this must be below `start_or_continue`'s definition
// as that also has a `define_scoped_cx` and that kind of shadowing
// is disallowed (name resolution thinks `scoped_cx!` is ambiguous).
define_scoped_cx!(self);

let old_region_index = self.config.region_index;
Expand Down Expand Up @@ -1302,10 +1317,43 @@ impl<'gcx, 'tcx, P: PrettyPrinter> PrintCx<'_, 'gcx, 'tcx, P> {
result
}

fn prepare_late_bound_region_info<T>(&mut self, value: &ty::Binder<T>)
where T: TypeFoldable<'tcx>
{

struct LateBoundRegionNameCollector(FxHashSet<InternedString>);
impl<'tcx> ty::fold::TypeVisitor<'tcx> for LateBoundRegionNameCollector {
fn visit_region(&mut self, r: ty::Region<'tcx>) -> bool {
match *r {
ty::ReLateBound(_, ty::BrNamed(_, name)) => {
self.0.insert(name);
},
_ => {},
}
r.super_visit_with(self)
}
}

let mut collector = LateBoundRegionNameCollector(Default::default());
value.visit_with(&mut collector);
self.config.used_region_names = Some(collector.0);
self.config.region_index = 0;
}

fn is_name_used(&self, name: &InternedString) -> bool {
match self.config.used_region_names {
Some(ref names) => names.contains(name),
None => false,
}
}
}

impl<T, P: PrettyPrinter> Print<'tcx, P> for ty::Binder<T>
where T: Print<'tcx, P, Output = P, Error = P::Error> + TypeFoldable<'tcx>
{
type Output = P;
type Error = P::Error;
fn print(&self, cx: PrintCx<'_, '_, 'tcx, P>) -> Result<Self::Output, Self::Error> {
cx.in_binder(self)
}
}
22 changes: 0 additions & 22 deletions src/librustc/ty/structural_impls.rs
Expand Up @@ -775,28 +775,6 @@ BraceStructLiftImpl! {
}
}

// FIXME(eddyb) this is like what some of the macros above generate,
// except that macros *also* generate a foldable impl, which we don't
// want (with it we'd risk bypassing `fold_region` / `fold_const`).
impl<'tcx> Lift<'tcx> for ty::RegionKind {
type Lifted = ty::RegionKind;
fn lift_to_tcx<'b, 'gcx>(&self, _: TyCtxt<'b, 'gcx, 'tcx>) -> Option<Self::Lifted> {
Some(self.clone())
}
}

impl<'a, 'tcx> Lift<'tcx> for ty::LazyConst<'a> {
type Lifted = ty::LazyConst<'tcx>;
fn lift_to_tcx<'b, 'gcx>(&self, tcx: TyCtxt<'b, 'gcx, 'tcx>) -> Option<Self::Lifted> {
match self {
ty::LazyConst::Evaluated(v) => Some(ty::LazyConst::Evaluated(tcx.lift(v)?)),
ty::LazyConst::Unevaluated(def_id, substs) => {
Some(ty::LazyConst::Unevaluated(*def_id, tcx.lift(substs)?))
}
}
}
}

BraceStructLiftImpl! {
impl<'a, 'tcx> Lift<'tcx> for ty::Const<'a> {
type Lifted = ty::Const<'tcx>;
Expand Down
121 changes: 65 additions & 56 deletions src/librustc/util/ppaux.rs
@@ -1,7 +1,7 @@
use crate::hir;
use crate::hir::def::Namespace;
use crate::ty::subst::{Kind, UnpackedKind};
use crate::ty::{self, ParamConst, Ty};
use crate::ty::{self, ParamConst, Ty, TyCtxt};
use crate::ty::print::{FmtPrinter, PrettyPrinter, PrintCx, Print};
use crate::mir::interpret::ConstValue;

Expand All @@ -10,13 +10,60 @@ use std::iter;

use rustc_target::spec::abi::Abi;

pub trait LiftAndPrintToFmt<'tcx> {
fn lift_and_print_to_fmt(
&self,
tcx: TyCtxt<'_, '_, 'tcx>,
f: &mut fmt::Formatter<'_>,
) -> fmt::Result;
}

impl<T> LiftAndPrintToFmt<'tcx> for T
where T: ty::Lift<'tcx>,
for<'a, 'b> <T as ty::Lift<'tcx>>::Lifted:
Print<'tcx, FmtPrinter<&'a mut fmt::Formatter<'b>>, Error = fmt::Error>
{
fn lift_and_print_to_fmt(
&self,
tcx: TyCtxt<'_, '_, 'tcx>,
f: &mut fmt::Formatter<'_>,
) -> fmt::Result {
PrintCx::with(tcx, FmtPrinter::new(f, Namespace::TypeNS), |cx| {
cx.tcx.lift(self).expect("could not lift for printing").print(cx)?;
Ok(())
})
}
}

// HACK(eddyb) this is separate because `ty::RegionKind` doesn't need lifting.
impl LiftAndPrintToFmt<'tcx> for ty::RegionKind {
fn lift_and_print_to_fmt(
&self,
tcx: TyCtxt<'_, '_, 'tcx>,
f: &mut fmt::Formatter<'_>,
) -> fmt::Result {
PrintCx::with(tcx, FmtPrinter::new(f, Namespace::TypeNS), |cx| {
self.print(cx)?;
Ok(())
})
}
}

macro_rules! define_print {
([$($target:ty),+] $vars:tt $def:tt) => {
$(define_print!($target, $vars $def);)+
(<$($T:ident),*> $target:ty) => {
impl<$($T),*> fmt::Display for $target
where Self: for<'a> LiftAndPrintToFmt<'a>
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
ty::tls::with(|tcx| self.lift_and_print_to_fmt(tcx, f))
}
}
};

($target:ty, ($self:ident, $cx:ident) { display $disp:block }) => {
impl<P: PrettyPrinter> Print<'tcx, P> for $target {
(<$($T:ident),*> $target:ty, ($self:ident, $cx:ident) { display $disp:block }) => {
impl<$($T,)* P: PrettyPrinter> Print<'tcx, P> for $target
where $($T: Print<'tcx, P, Output = P, Error = P::Error>),*
{
type Output = P;
type Error = fmt::Error;
fn print(&$self, $cx: PrintCx<'_, '_, 'tcx, P>) -> Result<Self::Output, Self::Error> {
Expand All @@ -29,14 +76,15 @@ macro_rules! define_print {
}
}

impl fmt::Display for $target {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
PrintCx::with_tls_tcx(FmtPrinter::new(f, Namespace::TypeNS), |cx| {
cx.tcx.lift(self).expect("could not lift for printing").print(cx)?;
Ok(())
})
}
}
define_print!(<$($T),*> $target);
};

($target:ty) => {
define_print!(<> $target);
};

($target:ty, ($self:ident, $cx:ident) { display $disp:block }) => {
define_print!(<> $target, ($self, $cx) { display $disp });
};
}

Expand Down Expand Up @@ -172,11 +220,7 @@ define_print! {
}

define_print! {
ty::RegionKind, (self, cx) {
display {
return cx.print_region(self);
}
}
ty::RegionKind
}

define_print! {
Expand Down Expand Up @@ -215,34 +259,8 @@ define_print! {
}
}

// The generic impl doesn't work yet because projections are not
// normalized under HRTB.
/*impl<T> fmt::Display for ty::Binder<T>
where T: fmt::Display + for<'a> ty::Lift<'a>,
for<'a> <T as ty::Lift<'a>>::Lifted: fmt::Display + TypeFoldable<'a>
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
PrintCx::with_tls_tcx(|cx| cx.pretty_in_binder(cx.tcx.lift(self)
.expect("could not lift for printing")))
}
}*/

define_print! {
[
ty::Binder<&'tcx ty::List<ty::ExistentialPredicate<'tcx>>>,
ty::Binder<ty::TraitRef<'tcx>>,
ty::Binder<ty::FnSig<'tcx>>,
ty::Binder<ty::TraitPredicate<'tcx>>,
ty::Binder<ty::SubtypePredicate<'tcx>>,
ty::Binder<ty::ProjectionPredicate<'tcx>>,
ty::Binder<ty::OutlivesPredicate<Ty<'tcx>, ty::Region<'tcx>>>,
ty::Binder<ty::OutlivesPredicate<ty::Region<'tcx>, ty::Region<'tcx>>>
]
(self, cx) {
display {
nest!(|cx| cx.pretty_in_binder(self))
}
}
<T> ty::Binder<T>
}

define_print! {
Expand All @@ -254,11 +272,7 @@ define_print! {
}

define_print! {
Ty<'tcx>, (self, cx) {
display {
return cx.print_type(self);
}
}
Ty<'tcx>
}

define_print! {
Expand Down Expand Up @@ -309,13 +323,8 @@ define_print! {
}
}

// Similar problem to `Binder<T>`, can't define a generic impl.
define_print! {
[
ty::OutlivesPredicate<Ty<'tcx>, ty::Region<'tcx>>,
ty::OutlivesPredicate<ty::Region<'tcx>, ty::Region<'tcx>>
]
(self, cx) {
<T, U> ty::OutlivesPredicate<T, U>, (self, cx) {
display {
p!(print(self.0), write(" : "), print(self.1))
}
Expand Down

0 comments on commit 030cdc9

Please sign in to comment.