fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2026-05-08 02:38:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

nak: Implement Srcs/DstsAsType directly for Op

Browse source 
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/41410>
This commit is contained in:
Faith Ekstrand 2026-05-06 19:39:49 -04:00 • committed by Marge Bot
parent 7d0c50f20a
commit 57d7544896
1 changed files with 88 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

90
src/nouveau/compiler/nak/ir_proc.rs
View file

@ -14,12 +14,98 @@ use syn::*;
#[proc_macro_derive(SrcsAsSlice, attributes(src_type))]
pub fn derive_srcs_as_slice(input: TokenStream) -> TokenStream {
derive_as_slice(input, "Src", "src_type", "SrcType")
let input2 = input.clone();
let DeriveInput { ident, data, .. } = parse_macro_input!(input2);
if let Data::Enum(e) = data {
let mut as_slice_cases = TokenStream2::new();
let mut as_mut_slice_cases = TokenStream2::new();
let mut types_cases = TokenStream2::new();
for v in e.variants {
let case = v.ident;
as_slice_cases.extend(quote! {
#ident::#case(x) => x.srcs_as_slice(),
});
as_mut_slice_cases.extend(quote! {
#ident::#case(x) => x.srcs_as_mut_slice(),
});
types_cases.extend(quote! {
#ident::#case(x) => x.src_types(),
});
}
quote! {
impl SrcsAsSlice for #ident {
fn srcs_as_slice(&self) -> &[Src] {
match self {
#as_slice_cases
}
}
fn srcs_as_mut_slice(&mut self) -> &mut [Src] {
match self {
#as_mut_slice_cases
}
}
fn src_types(&self) -> SrcTypeList {
match self {
#types_cases
}
}
}
}
.into()
} else {
derive_as_slice(input, "Src", "src_type", "SrcType")
}
}
#[proc_macro_derive(DstsAsSlice, attributes(dst_type))]
pub fn derive_dsts_as_slice(input: TokenStream) -> TokenStream {
derive_as_slice(input, "Dst", "dst_type", "DstType")
let input2 = input.clone();
let DeriveInput { ident, data, .. } = parse_macro_input!(input2);
if let Data::Enum(e) = data {
let mut as_slice_cases = TokenStream2::new();
let mut as_mut_slice_cases = TokenStream2::new();
let mut types_cases = TokenStream2::new();
for v in e.variants {
let case = v.ident;
as_slice_cases.extend(quote! {
#ident::#case(x) => x.dsts_as_slice(),
});
as_mut_slice_cases.extend(quote! {
#ident::#case(x) => x.dsts_as_mut_slice(),
});
types_cases.extend(quote! {
#ident::#case(x) => x.dst_types(),
});
}
quote! {
impl DstsAsSlice for #ident {
fn dsts_as_slice(&self) -> &[Dst] {
match self {
#as_slice_cases
}
}
fn dsts_as_mut_slice(&mut self) -> &mut [Dst] {
match self {
#as_mut_slice_cases
}
}
fn dst_types(&self) -> DstTypeList {
match self {
#types_cases
}
}
}
}
.into()
} else {
derive_as_slice(input, "Dst", "dst_type", "DstType")
}
}
#[proc_macro_derive(DisplayOp)]