mirrors/Graphite
1
0
Fork 0
mirror of https://github.com/GraphiteEditor/Graphite.git synced 2025-08-22 05:54:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

shader-rt: replace gpu node's args and ret types with Raster<GPU>

Browse source
This commit is contained in:
firestar99 2025-08-19 15:46:53 +02:00 • committed by Keavon Chambers
parent 35d7896450
commit 61b4017398
1 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
node-graph/node-macro/src/shader_nodes/per_pixel_adjust.rs
View file

@ -1,6 +1,7 @@
use crate::parsing::{Input, NodeFnAttributes, ParsedField, ParsedFieldType, ParsedNodeFn, RegularParsedField};
use crate::shader_nodes::{CodegenShaderEntryPoint, ShaderNodeType};
use convert_case::{Case, Casing};
use proc_macro_crate::FoundCrate;
use proc_macro2::{Ident, Span, TokenStream};
use quote::{ToTokens, format_ident, quote};
use std::borrow::Cow;
@ -105,14 +106,24 @@ impl CodegenShaderEntryPoint for PerPixelAdjust {
let struct_name = format_ident!("{}", fn_name.to_string().to_case(Case::Pascal));
let mod_name = fn_name.clone();
let gcore = match &parsed.crate_name {
FoundCrate::Itself => format_ident!("crate"),
FoundCrate::Name(name) => format_ident!("{name}"),
};
let raster_gpu = syn::parse2::<Type>(quote!(#gcore::table::Table<#gcore::raster_types::Raster<#gcore::raster_types::GPU>>))?;
let fields = parsed
.fields
.iter()
.map(|f| match &f.ty {
ParsedFieldType::Regular(reg) => Ok(ParsedField {
ty: ParsedFieldType::Regular(RegularParsedField { gpu_image: false, ..reg.clone() }),
ParsedFieldType::Regular(reg @ RegularParsedField { gpu_image: true, .. }) => Ok(ParsedField {
ty: ParsedFieldType::Regular(RegularParsedField {
ty: raster_gpu.clone(),
..reg.clone()
}),
..f.clone()
}),
ParsedFieldType::Regular(RegularParsedField { gpu_image: false, .. }) => Ok(f.clone()),
ParsedFieldType::Node { .. } => Err(syn::Error::new_spanned(&f.pat_ident, "PerPixelAdjust shader nodes cannot accept other nodes as generics")),
})
.collect::<syn::Result<_>>()?;
@ -146,7 +157,7 @@ impl CodegenShaderEntryPoint for PerPixelAdjust {
}),
implementations: Default::default(),
},
output_type: parsed.output_type.clone(),
output_type: raster_gpu,
is_async: true,
fields,
body,