Skip to content

Commit 8759f00

Browse files
JulianKnodtJulianKnodt
committed
Actually infer args in visitors
1 parent 3605675 commit 8759f00

File tree

30 files changed

+257
-97
lines changed

30 files changed

+257
-97
lines changed

‎compiler/rustc_ast/src/ast.rs‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,6 +336,8 @@ pub enum ParamKindOrd {
336336
// is active. Specifically, if it's only `min_const_generics`, it will still require
337337
// ordering consts after types.
338338
Const { unordered: bool },
339+
// `Infer` is not actually constructed directly from the AST, but is implicitly constructed
340+
// during HIR lowering, and `ParamKindOrd` will implicitly order inferred variables last.
339341
Infer,
340342
}
341343

‎compiler/rustc_ast_lowering/src/lib.rs‎

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ use rustc_hir::def::{DefKind, Namespace, PartialRes, PerNS, Res};
5151
use rustc_hir::def_id::{DefId, DefIdMap, DefPathHash, LocalDefId, CRATE_DEF_ID};
5252
use rustc_hir::definitions::{DefKey, DefPathData, Definitions};
5353
use rustc_hir::intravisit;
54-
use rustc_hir::{ConstArg, GenericArg, ParamName};
54+
use rustc_hir::{ConstArg, GenericArg, InferKind, ParamName};
5555
use rustc_index::vec::{Idx, IndexVec};
5656
use rustc_session::lint::builtin::{BARE_TRAIT_OBJECTS, MISSING_ABI};
5757
use rustc_session::lint::{BuiltinLintDiagnostics, LintBuffer};
@@ -1219,9 +1219,13 @@ impl<'a, 'hir> LoweringContext<'a, 'hir> {
12191219
ast::GenericArg::Lifetime(lt) => GenericArg::Lifetime(self.lower_lifetime(&lt)),
12201220
ast::GenericArg::Type(ty) => {
12211221
match ty.kind {
1222-
TyKind::Infer => {
1222+
TyKind::Infer if self.sess.features_untracked().generic_arg_infer => {
12231223
let hir_id = self.lower_node_id(ty.id);
1224-
return GenericArg::Infer(hir::InferArg { hir_id, span: ty.span });
1224+
return GenericArg::Infer(hir::InferArg {
1225+
hir_id,
1226+
span: ty.span,
1227+
kind: InferKind::Type,
1228+
});
12251229
}
12261230
// We parse const arguments as path types as we cannot distinguish them during
12271231
// parsing. We try to resolve that ambiguity by attempting resolution in both the

‎compiler/rustc_feature/src/active.rs‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -684,6 +684,9 @@ declare_features! (
684684
/// Allows `cfg(target_abi = "...")`.
685685
(active, cfg_target_abi, "1.55.0", Some(80970), None),
686686

687+
/// Infer generic args for both consts and types.
688+
(active, generic_arg_infer, "1.55.0", Some(85077), None),
689+
687690
// -------------------------------------------------------------------------
688691
// feature-group-end: actual feature gates
689692
// -------------------------------------------------------------------------

‎compiler/rustc_hir/src/hir.rs‎

Lines changed: 35 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -253,9 +253,23 @@ pub struct ConstArg {
253253
pub span: Span,
254254
}
255255

256+
#[derive(Copy, Clone, Encodable, Debug, HashStable_Generic)]
257+
pub enum InferKind {
258+
Const,
259+
Type,
260+
}
261+
262+
impl InferKind {
263+
#[inline]
264+
pub fn is_type(self) -> bool {
265+
matches!(self, InferKind::Type)
266+
}
267+
}
268+
256269
#[derive(Encodable, Debug, HashStable_Generic)]
257270
pub struct InferArg {
258271
pub hir_id: HirId,
272+
pub kind: InferKind,
259273
pub span: Span,
260274
}
261275

@@ -365,22 +379,29 @@ impl GenericArgs<'_> {
365379
panic!("GenericArgs::inputs: not a `Fn(T) -> U`");
366380
}
367381

368-
pub fn own_counts(&self) -> GenericParamCount {
369-
// We could cache this as a property of `GenericParamCount`, but
370-
// the aim is to refactor this away entirely eventually and the
371-
// presence of this method will be a constant reminder.
372-
let mut own_counts: GenericParamCount = Default::default();
382+
#[inline]
383+
pub fn has_type_params(&self) -> bool {
384+
self.args.iter().any(|arg| matches!(arg, GenericArg::Type(_)))
385+
}
373386

374-
for arg in self.args {
375-
match arg {
376-
GenericArg::Lifetime(_) => own_counts.lifetimes += 1,
377-
GenericArg::Type(_) => own_counts.types += 1,
378-
GenericArg::Const(_) => own_counts.consts += 1,
379-
GenericArg::Infer(_) => own_counts.infer += 1,
380-
};
381-
}
387+
#[inline]
388+
pub fn num_type_params(&self) -> usize {
389+
self.args.iter().filter(|arg| matches!(arg, GenericArg::Type(_))).count()
390+
}
391+
392+
#[inline]
393+
pub fn num_lifetime_params(&self) -> usize {
394+
self.args.iter().filter(|arg| matches!(arg, GenericArg::Lifetime(_))).count()
395+
}
382396

383-
own_counts
397+
#[inline]
398+
pub fn has_lifetime_params(&self) -> bool {
399+
self.args.iter().any(|arg| matches!(arg, GenericArg::Lifetime(_)))
400+
}
401+
402+
#[inline]
403+
pub fn num_generic_params(&self) -> usize {
404+
self.args.iter().filter(|arg| !matches!(arg, GenericArg::Lifetime(_))).count()
384405
}
385406

386407
/// The span encompassing the text inside the surrounding brackets.

‎compiler/rustc_hir_pretty/src/lib.rs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ impl<'a> State<'a> {
103103
Node::TraitRef(a) => self.print_trait_ref(&a),
104104
Node::Binding(a) | Node::Pat(a) => self.print_pat(&a),
105105
Node::Arm(a) => self.print_arm(&a),
106-
Node::Infer(_) => self.print_string("_", ast::StrStyle::Cooked),
106+
Node::Infer(_) => self.s.word("_"),
107107
Node::Block(a) => {
108108
// Containing cbox, will be closed by print-block at `}`.
109109
self.cbox(INDENT_UNIT);

‎compiler/rustc_privacy/src/lib.rs‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1224,6 +1224,13 @@ impl<'tcx> Visitor<'tcx> for TypePrivacyVisitor<'tcx> {
12241224
}
12251225
}
12261226
} else {
1227+
let local_id = self.tcx.hir().local_def_id(inf.hir_id);
1228+
if let Some(did) = self.tcx.opt_const_param_of(local_id) {
1229+
if self.visit_def_id(did, "inferred", &"").is_break() {
1230+
return;
1231+
}
1232+
}
1233+
12271234
// FIXME see above note for same issue.
12281235
if self.visit(rustc_typeck::hir_ty_to_ty(self.tcx, &inf.to_ty())).is_break() {
12291236
return;

‎compiler/rustc_resolve/src/late/lifetimes.rs‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2559,7 +2559,9 @@ impl<'a, 'tcx> LifetimeContext<'a, 'tcx> {
25592559
}
25602560
GenericArg::Infer(inf) => {
25612561
self.visit_id(inf.hir_id);
2562-
i += 1;
2562+
if inf.kind.is_type() {
2563+
i += 1;
2564+
}
25632565
}
25642566
}
25652567
}

‎compiler/rustc_span/src/symbol.rs‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -622,6 +622,7 @@ symbols! {
622622
generator,
623623
generator_state,
624624
generators,
625+
generic_arg_infer,
625626
generic_associated_types,
626627
generic_param_attrs,
627628
get_context,

‎compiler/rustc_typeck/src/astconv/generics.rs‎

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,13 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
3939
);
4040

4141
if let GenericParamDefKind::Const { .. } = param.kind {
42-
if matches!(
43-
arg,
44-
GenericArg::Type(hir::Ty { kind: hir::TyKind::Infer, .. }) | GenericArg::Infer(_)
45-
) {
42+
if matches!(arg, GenericArg::Type(hir::Ty { kind: hir::TyKind::Infer, .. })) {
4643
err.help("const arguments cannot yet be inferred with `_`");
44+
if sess.is_nightly_build() {
45+
err.help(
46+
"add `#![feature(generic_arg_infer)]` to the crate attributes to enable",
47+
);
48+
}
4749
}
4850
}
4951

@@ -458,8 +460,8 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
458460
let default_counts = gen_params.own_defaults();
459461
let param_counts = gen_params.own_counts();
460462
let named_type_param_count = param_counts.types - has_self as usize;
461-
let arg_counts = gen_args.own_counts();
462-
let infer_lifetimes = gen_pos != GenericArgPosition::Type && arg_counts.lifetimes == 0;
463+
let infer_lifetimes =
464+
gen_pos != GenericArgPosition::Type && !gen_args.has_lifetime_params();
463465

464466
if gen_pos != GenericArgPosition::Type && !gen_args.bindings.is_empty() {
465467
Self::prohibit_assoc_ty_binding(tcx, gen_args.bindings[0].span);
@@ -517,7 +519,7 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
517519

518520
let min_expected_lifetime_args = if infer_lifetimes { 0 } else { param_counts.lifetimes };
519521
let max_expected_lifetime_args = param_counts.lifetimes;
520-
let num_provided_lifetime_args = arg_counts.lifetimes;
522+
let num_provided_lifetime_args = gen_args.num_lifetime_params();
521523

522524
let lifetimes_correct = check_lifetime_args(
523525
min_expected_lifetime_args,
@@ -588,14 +590,14 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
588590
- default_counts.consts
589591
};
590592
debug!("expected_min: {:?}", expected_min);
591-
debug!("arg_counts.lifetimes: {:?}", arg_counts.lifetimes);
593+
debug!("arg_counts.lifetimes: {:?}", gen_args.num_lifetime_params());
592594

593595
check_types_and_consts(
594596
expected_min,
595597
param_counts.consts + named_type_param_count,
596-
arg_counts.consts + arg_counts.types + arg_counts.infer,
598+
gen_args.num_generic_params(),
597599
param_counts.lifetimes + has_self as usize,
598-
arg_counts.lifetimes,
600+
gen_args.num_lifetime_params(),
599601
)
600602
};
601603

@@ -673,8 +675,7 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
673675
position: GenericArgPosition,
674676
) -> ExplicitLateBound {
675677
let param_counts = def.own_counts();
676-
let arg_counts = args.own_counts();
677-
let infer_lifetimes = position != GenericArgPosition::Type && arg_counts.lifetimes == 0;
678+
let infer_lifetimes = position != GenericArgPosition::Type && !args.has_lifetime_params();
678679

679680
if infer_lifetimes {
680681
return ExplicitLateBound::No;
@@ -687,7 +688,7 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
687688
let span = args.args[0].span();
688689

689690
if position == GenericArgPosition::Value
690-
&& arg_counts.lifetimes != param_counts.lifetimes
691+
&& args.num_lifetime_params() != param_counts.lifetimes
691692
{
692693
let mut err = tcx.sess.struct_span_err(span, msg);
693694
err.span_note(span_late, note);

‎compiler/rustc_typeck/src/astconv/mod.rs‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -463,8 +463,10 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
463463
}
464464
(&GenericParamDefKind::Const { has_default }, hir::GenericArg::Infer(inf)) => {
465465
if has_default {
466-
// FIXME(const_generics): Actually infer parameter here?
467466
tcx.const_param_default(param.def_id).into()
467+
} else if self.astconv.allow_ty_infer() {
468+
// FIXME(const_generics): Actually infer parameter here?
469+
todo!()
468470
} else {
469471
self.inferred_params.push(inf.span);
470472
tcx.ty_error().into()
@@ -1963,7 +1965,7 @@ impl<'o, 'tcx> dyn AstConv<'tcx> + 'o {
19631965
}
19641966
has_err = true;
19651967
err_for_ty = true;
1966-
(inf.span, "inferred")
1968+
(inf.span, "generic")
19671969
}
19681970
};
19691971
let mut err = struct_span_err!(

0 commit comments

Comments
 (0)