Skip to content

Commit

Permalink
less opt in const param of
Browse files Browse the repository at this point in the history
  • Loading branch information
BoxyUwU committed Aug 7, 2021
1 parent 508b328 commit d777cb8
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 2 deletions.
7 changes: 5 additions & 2 deletions compiler/rustc_typeck/src/collect/type_of.rs
Original file line number Diff line number Diff line change
Expand Up @@ -190,8 +190,12 @@ pub(super) fn opt_const_param_of(tcx: TyCtxt<'_>, def_id: LocalDefId) -> Option<
// Try to use the segment resolution if it is valid, otherwise we
// default to the path resolution.
let res = segment.res.filter(|&r| r != Res::Err).unwrap_or(path.res);
use def::CtorOf;
let generics = match res {
Res::Def(DefKind::Ctor(..), def_id) => {
Res::Def(DefKind::Ctor(CtorOf::Variant, _), def_id) => {
tcx.generics_of(tcx.parent(def_id).and_then(|def_id| tcx.parent(def_id)).unwrap())
}
Res::Def(DefKind::Variant | DefKind::Ctor(CtorOf::Struct, _), def_id) => {
tcx.generics_of(tcx.parent(def_id).unwrap())
}
// Other `DefKind`s don't have generics and would ICE when calling
Expand All @@ -200,7 +204,6 @@ pub(super) fn opt_const_param_of(tcx: TyCtxt<'_>, def_id: LocalDefId) -> Option<
DefKind::Struct
| DefKind::Union
| DefKind::Enum
| DefKind::Variant
| DefKind::Trait
| DefKind::OpaqueTy
| DefKind::TyAlias
Expand Down
24 changes: 24 additions & 0 deletions src/test/ui/const-generics/enum-variants.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
// check-pass
pub enum Foo<const N: usize> {
Variant,
Variant2(),
Variant3{},
}

struct Bar<const N: usize>;
struct Bar2<const N: usize>();
struct Bar3<const N: usize> {}

fn main() {
let _ = Foo::Variant::<1>;
let _ = Foo::Variant2::<1>();
let _ = Foo::Variant3::<1>{};

let _ = Foo::<1>::Variant;
let _ = Foo::<1>::Variant2();
let _ = Foo::<1>::Variant3{};

let _ = Bar::<1>;
let _ = Bar2::<1>();
let _ = Bar3::<1>{};
}

0 comments on commit d777cb8

Please sign in to comment.