fork/Graphite
1
0
Fork 0
mirror of https://github.com/GraphiteEditor/Graphite.git synced 2026-09-16 23:08:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

GPU: add support for N-many input images and not just exactly one

Browse Source
This commit is contained in:
firestar99
2026-05-24 19:28:41 +02:00
committed by Keavon Chambers
parent 7a6e4f6f25
commit 953a9312f0
3 changed files with 94 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
node-graph/node-macro/src/shader_nodes/per_pixel_adjust.rs
View File

@@ -248,16 +248,15 @@ impl PerPixelAdjustCodegen<'_> {
is_data_field: false,
});
// find exactly one gpu_image field, runtime doesn't support more than 1 atm
let gpu_image_field = {
let mut iter = fields.iter().filter(|f| matches!(f.ty, ParsedFieldType::Regular(RegularParsedField { gpu_image: true, .. })));
match (iter.next(), iter.next()) {
(Some(v), None) => Ok(v),
(Some(_), Some(more)) => Err(syn::Error::new_spanned(&more.pat_ident, "No more than one parameter must be annotated with `#[gpu_image]`")),
(None, _) => Err(syn::Error::new_spanned(&self.parsed.fn_name, "At least one parameter must be annotated with `#[gpu_image]`")),
}?
};
let gpu_image = &gpu_image_field.pat_ident.ident;
// find gpu_image fields
let gpu_images = fields
.iter()
.filter_map(|f| match f.ty {
ParsedFieldType::Regular(RegularParsedField { gpu_image: true, .. }) => Some(&f.pat_ident.ident),
_ => None,
})
.collect::<Vec<_>>();
let input_images = gpu_images.len();
// uniform buffer struct construction
let has_uniform = self.has_uniform;
@@ -287,7 +286,8 @@ impl PerPixelAdjustCodegen<'_> {
wgsl_shader: crate::WGSL_SHADER,
fragment_shader_name: super::#entry_point_name,
has_uniform: #has_uniform,
}, #gpu_image, #uniform_buffer).await
input_images: #input_images,
}, &[#(#gpu_images),*], #uniform_buffer).await
}
};