Inline box_region macro calls
This commit is contained in:
parent
8f397bc8a0
commit
86b3ebe2da
@ -78,83 +78,3 @@ pub enum YieldType<I, A> {
|
|||||||
Initial(I),
|
Initial(I),
|
||||||
Accessor(Marker<A>),
|
Accessor(Marker<A>),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[macro_export]
|
|
||||||
#[allow_internal_unstable(fn_traits)]
|
|
||||||
macro_rules! declare_box_region_type {
|
|
||||||
($v:vis $name: ident, ($($args:ty),*) -> ($reti:ty, $retc:ty)) => {
|
|
||||||
$v struct $name($crate::box_region::PinnedGenerator<
|
|
||||||
$reti,
|
|
||||||
fn(($($args,)*)),
|
|
||||||
$retc
|
|
||||||
>);
|
|
||||||
|
|
||||||
impl $name {
|
|
||||||
fn new<T>(generator: T) -> ($reti, Self)
|
|
||||||
where T: ::std::ops::Generator<
|
|
||||||
$crate::box_region::Action,
|
|
||||||
Yield = $crate::box_region::YieldType<$reti, fn(($($args,)*))>,
|
|
||||||
Return = $retc,
|
|
||||||
> + 'static {
|
|
||||||
let (initial, pinned) = $crate::box_region::PinnedGenerator::new(generator);
|
|
||||||
(initial, $name(pinned))
|
|
||||||
}
|
|
||||||
|
|
||||||
$v fn access<F: FnOnce($($args,)*) -> R, R>(&mut self, f: F) -> R {
|
|
||||||
// Turn the FnOnce closure into *mut dyn FnMut()
|
|
||||||
// so we can pass it in to the generator
|
|
||||||
let mut r = None;
|
|
||||||
let mut f = Some(f);
|
|
||||||
let mut_f: &mut dyn FnMut(($($args,)*)) =
|
|
||||||
&mut |args| {
|
|
||||||
let f = f.take().unwrap();
|
|
||||||
r = Some(FnOnce::call_once(f, args));
|
|
||||||
};
|
|
||||||
let mut_f = mut_f as *mut dyn FnMut(($($args,)*));
|
|
||||||
|
|
||||||
// Get the generator to call our closure
|
|
||||||
unsafe {
|
|
||||||
self.0.access(::std::mem::transmute(mut_f));
|
|
||||||
}
|
|
||||||
|
|
||||||
// Unwrap the result
|
|
||||||
r.unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
$v fn complete(mut self) -> $retc {
|
|
||||||
self.0.complete()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn initial_yield(
|
|
||||||
value: $reti,
|
|
||||||
) -> $crate::box_region::YieldType<$reti, fn(($($args,)*))> {
|
|
||||||
$crate::box_region::YieldType::Initial(value)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
#[macro_export]
|
|
||||||
#[allow_internal_unstable(fn_traits)]
|
|
||||||
macro_rules! box_region_allow_access {
|
|
||||||
(($($args:ty),*), ($($exprs:expr),*), $action:ident) => {
|
|
||||||
loop {
|
|
||||||
match $action {
|
|
||||||
$crate::box_region::Action::Access(accessor) => {
|
|
||||||
let accessor: &mut dyn FnMut($($args),*) = unsafe {
|
|
||||||
::std::mem::transmute(accessor.get())
|
|
||||||
};
|
|
||||||
(*accessor)(($($exprs),*));
|
|
||||||
unsafe {
|
|
||||||
let marker = $crate::box_region::Marker::<
|
|
||||||
fn(($($args,)*))
|
|
||||||
>::new();
|
|
||||||
$action = yield $crate::box_region::YieldType::Accessor(marker);
|
|
||||||
};
|
|
||||||
}
|
|
||||||
$crate::box_region::Action::Complete => break,
|
|
||||||
$crate::box_region::Action::Initial => panic!("unexpected box_region action: Initial"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
@ -6,10 +6,10 @@
|
|||||||
use rustc_ast::{self as ast, visit};
|
use rustc_ast::{self as ast, visit};
|
||||||
use rustc_codegen_ssa::back::link::emit_metadata;
|
use rustc_codegen_ssa::back::link::emit_metadata;
|
||||||
use rustc_codegen_ssa::traits::CodegenBackend;
|
use rustc_codegen_ssa::traits::CodegenBackend;
|
||||||
|
use rustc_data_structures::parallel;
|
||||||
use rustc_data_structures::steal::Steal;
|
use rustc_data_structures::steal::Steal;
|
||||||
use rustc_data_structures::sync::{par_iter, Lrc, OnceCell, ParallelIterator, WorkerLocal};
|
use rustc_data_structures::sync::{par_iter, Lrc, OnceCell, ParallelIterator, WorkerLocal};
|
||||||
use rustc_data_structures::temp_dir::MaybeTempDir;
|
use rustc_data_structures::temp_dir::MaybeTempDir;
|
||||||
use rustc_data_structures::{box_region_allow_access, declare_box_region_type, parallel};
|
|
||||||
use rustc_errors::{ErrorReported, PResult};
|
use rustc_errors::{ErrorReported, PResult};
|
||||||
use rustc_expand::base::ExtCtxt;
|
use rustc_expand::base::ExtCtxt;
|
||||||
use rustc_hir::def_id::LOCAL_CRATE;
|
use rustc_hir::def_id::LOCAL_CRATE;
|
||||||
@ -85,11 +85,62 @@ fn count_nodes(krate: &ast::Crate) -> usize {
|
|||||||
counter.count
|
counter.count
|
||||||
}
|
}
|
||||||
|
|
||||||
declare_box_region_type!(
|
pub struct BoxedResolver(
|
||||||
pub BoxedResolver,
|
rustc_data_structures::box_region::PinnedGenerator<
|
||||||
(&mut Resolver<'_>) -> (Result<ast::Crate>, ResolverOutputs)
|
Result<ast::Crate>,
|
||||||
|
fn(&mut Resolver<'_>),
|
||||||
|
ResolverOutputs,
|
||||||
|
>,
|
||||||
);
|
);
|
||||||
|
|
||||||
|
impl BoxedResolver {
|
||||||
|
fn new<T>(generator: T) -> (Result<ast::Crate>, Self)
|
||||||
|
where
|
||||||
|
T: ::std::ops::Generator<
|
||||||
|
rustc_data_structures::box_region::Action,
|
||||||
|
Yield = rustc_data_structures::box_region::YieldType<
|
||||||
|
Result<ast::Crate>,
|
||||||
|
fn(&mut Resolver<'_>),
|
||||||
|
>,
|
||||||
|
Return = ResolverOutputs,
|
||||||
|
> + 'static,
|
||||||
|
{
|
||||||
|
let (initial, pinned) = rustc_data_structures::box_region::PinnedGenerator::new(generator);
|
||||||
|
(initial, BoxedResolver(pinned))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn access<F: FnOnce(&mut Resolver<'_>) -> R, R>(&mut self, f: F) -> R {
|
||||||
|
// Turn the FnOnce closure into *mut dyn FnMut()
|
||||||
|
// so we can pass it in to the generator
|
||||||
|
let mut r = None;
|
||||||
|
let mut f = Some(f);
|
||||||
|
let mut_f: &mut dyn FnMut(&mut Resolver<'_>) = &mut |resolver| {
|
||||||
|
let f = f.take().unwrap();
|
||||||
|
r = Some(f(resolver));
|
||||||
|
};
|
||||||
|
let mut_f = mut_f as *mut dyn FnMut(&mut Resolver<'_>);
|
||||||
|
|
||||||
|
// Get the generator to call our closure
|
||||||
|
unsafe {
|
||||||
|
self.0.access(::std::mem::transmute(mut_f));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unwrap the result
|
||||||
|
r.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn complete(mut self) -> ResolverOutputs {
|
||||||
|
self.0.complete()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn initial_yield(
|
||||||
|
value: Result<ast::Crate>,
|
||||||
|
) -> rustc_data_structures::box_region::YieldType<Result<ast::Crate>, fn(&mut Resolver<'_>)>
|
||||||
|
{
|
||||||
|
rustc_data_structures::box_region::YieldType::Initial(value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Runs the "early phases" of the compiler: initial `cfg` processing, loading compiler plugins,
|
/// Runs the "early phases" of the compiler: initial `cfg` processing, loading compiler plugins,
|
||||||
/// syntax expansion, secondary `cfg` expansion, synthesis of a test
|
/// syntax expansion, secondary `cfg` expansion, synthesis of a test
|
||||||
/// harness if one is to be provided, injection of a dependency on the
|
/// harness if one is to be provided, injection of a dependency on the
|
||||||
@ -132,7 +183,28 @@ pub fn configure_and_expand(
|
|||||||
resolver
|
resolver
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
box_region_allow_access!((&mut Resolver<'_>), (&mut resolver), action);
|
|
||||||
|
loop {
|
||||||
|
match action {
|
||||||
|
rustc_data_structures::box_region::Action::Access(accessor) => {
|
||||||
|
let accessor: &mut dyn FnMut(&mut Resolver<'_>) =
|
||||||
|
unsafe { ::std::mem::transmute(accessor.get()) };
|
||||||
|
(*accessor)(&mut resolver);
|
||||||
|
unsafe {
|
||||||
|
let marker = rustc_data_structures::box_region::Marker::<
|
||||||
|
fn(&mut Resolver<'_>),
|
||||||
|
>::new();
|
||||||
|
action =
|
||||||
|
yield rustc_data_structures::box_region::YieldType::Accessor(marker);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
rustc_data_structures::box_region::Action::Complete => break,
|
||||||
|
rustc_data_structures::box_region::Action::Initial => {
|
||||||
|
panic!("unexpected box_region action: Initial")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
resolver.into_outputs()
|
resolver.into_outputs()
|
||||||
});
|
});
|
||||||
result.map(|k| (k, resolver))
|
result.map(|k| (k, resolver))
|
||||||
|
Loading…
Reference in New Issue
Block a user