WIP3

Author: Firestar99

node-graph/node-macro/src/parsing.rs

@@ -39,7 +39,7 @@ pub(crate) struct ParsedNodeFn {
 	pub(crate) description: String,
 }
 
-#[derive(Debug, Default)]
+#[derive(Debug, Default, Clone)]
 pub(crate) struct NodeFnAttributes {
 	pub(crate) category: Option<LitStr>,
 	pub(crate) display_name: Option<LitStr>,

node-graph/node-macro/src/shader_nodes/mod.rs

@@ -1,10 +1,11 @@
-use crate::parsing::{NodeFnAttributes, ParsedNodeFn};
+use crate::parsing::{Input, NodeFnAttributes, ParsedNodeFn};
 use crate::shader_nodes::per_pixel_adjust::PerPixelAdjust;
 use proc_macro2::{Ident, TokenStream};
-use quote::quote;
+use quote::{format_ident, quote};
+use std::{todo, vec};
 use strum::VariantNames;
-use syn::Error;
 use syn::parse::{Parse, ParseStream};
+use syn::{Error, Path, Type, TypePath};
 
 pub mod per_pixel_adjust;
 
@@ -19,7 +20,7 @@ pub fn modify_cfg(attributes: &NodeFnAttributes) -> TokenStream {
 	}
 }
 
-#[derive(Debug, VariantNames)]
+#[derive(Debug, Clone, VariantNames)]
 pub(crate) enum ShaderNodeType {
 	PerPixelAdjust(PerPixelAdjust),
 }
@@ -36,6 +37,7 @@ impl Parse for ShaderNodeType {
 
 pub trait CodegenShaderEntryPoint {
 	fn codegen_shader_entry_point(&self, parsed: &ParsedNodeFn) -> syn::Result<TokenStream>;
+	fn codegen_gpu_node(&self, parsed: &ParsedNodeFn) -> syn::Result<TokenStream>;
 }
 
 impl CodegenShaderEntryPoint for ShaderNodeType {

node-graph/node-macro/src/shader_nodes/per_pixel_adjust.rs

@@ -1,11 +1,13 @@
-use crate::parsing::{ParsedField, ParsedNodeFn};
+use crate::parsing::{Input, NodeFnAttributes, ParsedField, ParsedNodeFn};
 use crate::shader_nodes::CodegenShaderEntryPoint;
+use convert_case::{Case, Casing};
 use proc_macro2::{Ident, TokenStream};
 use quote::{ToTokens, format_ident, quote};
 use std::borrow::Cow;
 use syn::parse::{Parse, ParseStream};
+use syn::{Path, Type, TypePath};
 
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct PerPixelAdjust {}
 
 impl Parse for PerPixelAdjust {
@@ -17,7 +19,7 @@ impl Parse for PerPixelAdjust {
 impl CodegenShaderEntryPoint for PerPixelAdjust {
 	fn codegen_shader_entry_point(&self, parsed: &ParsedNodeFn) -> syn::Result<TokenStream> {
 		let fn_name = &parsed.fn_name;
-		let gpu_mod = format_ident!("{}_gpu", parsed.fn_name);
+		let gpu_mod = format_ident!("{}_gpu_entry_point", parsed.fn_name);
 		let spirv_image_ty = quote!(Image2d);
 
 		// bindings for images start at 1
@@ -93,6 +95,42 @@ impl CodegenShaderEntryPoint for PerPixelAdjust {
 			}
 		})
 	}
+
+	fn codegen_gpu_node(&self, parsed: &ParsedNodeFn) -> syn::Result<TokenStream> {
+		let fn_name = format_ident!("{}_gpu", parsed.fn_name);
+		let struct_name = format_ident!("{}", fn_name.to_string().to_case(Case::Pascal));
+		let mod_name = fn_name.clone();
+
+		let fields = todo!();
+		let body = todo!();
+
+		crate::codegen::generate_node_code(&ParsedNodeFn {
+			vis: parsed.vis.clone(),
+			attributes: NodeFnAttributes {
+				shader_node: None,
+				..parsed.attributes.clone()
+			},
+			fn_name,
+			struct_name,
+			mod_name,
+			fn_generics: vec![],
+			where_clause: None,
+			input: Input {
+				pat_ident: parsed.input.pat_ident.clone(),
+				ty: Type::Path(TypePath {
+					path: Path::from(format_ident!("Ctx")),
+					qself: None,
+				}),
+				implementations: Default::default(),
+			},
+			output_type: parsed.output_type.clone(),
+			is_async: true,
+			fields,
+			body,
+			crate_name: parsed.crate_name.clone(),
+			description: "".to_string(),
+		})
+	}
 }
 
 struct Param<'a> {