Bundle Graphite using Tauri (#873)

* Setup tauri component for graphite editor Integrate graphite into tauri app Split interpreted-executor out of graph-craft * Add gpu execution node * General Cleanup
2026-09-16 23:08:05 +08:00 · 2022-12-07 12:49:34 +01:00
parent 52cc770a1e
commit 7d8f94462a
109 changed files with 5661 additions and 544 deletions
--- a/node-graph/gstd/src/executor.rs
+++ b/node-graph/gstd/src/executor.rs
@@ -0,0 +1,135 @@
+use bytemuck::Pod;
+use core::marker::PhantomData;
+use dyn_any::StaticTypeSized;
+use graph_craft::document::*;
+use graph_craft::proto::*;
+use graphene_core::{raster::Image, value::ValueNode, Node};
+
+pub struct MapGpuNode<NN: Node<()>, I: IntoIterator<Item = S>, S: StaticTypeSized + Sync + Send + Pod, O: StaticTypeSized + Sync + Send + Pod>(pub NN, PhantomData<(S, I, O)>);
+
+impl<'n, I: IntoIterator<Item = S>, NN: Node<(), Output = &'n NodeNetwork> + Copy, S: StaticTypeSized + Sync + Send + Pod, O: StaticTypeSized + Sync + Send + Pod> Node<I>
+	for &MapGpuNode<NN, I, S, O>
+{
+	type Output = Vec<O>;
+	fn eval(self, input: I) -> Self::Output {
+		let network = self.0.eval(());
+
+		use graph_craft::executor::Compiler;
+		use graph_craft::executor::Executor;
+		use graph_craft::gpu::compiler::Metadata;
+		let compiler = Compiler {};
+		let proto_network = compiler.compile(network.clone(), true);
+
+		let m = Metadata::new("project".to_owned(), vec!["test@example.com".to_owned()]);
+		let temp_dir = tempfile::tempdir().expect("failed to create tempdir");
+
+		use graph_craft::gpu::context::Context;
+		use graph_craft::gpu::executor::GpuExecutor;
+		let executor: GpuExecutor<S, O> = GpuExecutor::new(Context::new(), proto_network, m, temp_dir.path()).unwrap();
+
+		let data: Vec<_> = input.into_iter().collect();
+		let result = executor.execute(Box::new(data)).unwrap();
+		let result = dyn_any::downcast::<Vec<O>>(result).unwrap();
+		*result
+	}
+}
+impl<'n, I: IntoIterator<Item = S>, NN: Node<(), Output = &'n NodeNetwork> + Copy, S: StaticTypeSized + Sync + Send + Pod, O: StaticTypeSized + Sync + Send + Pod> Node<I> for MapGpuNode<NN, I, S, O> {
+	type Output = Vec<O>;
+	fn eval(self, input: I) -> Self::Output {
+		let network = self.0.eval(());
+
+		use graph_craft::executor::Compiler;
+		use graph_craft::executor::Executor;
+		use graph_craft::gpu::compiler::Metadata;
+		let compiler = Compiler {};
+		let proto_network = compiler.compile(network.clone(), true);
+
+		let m = Metadata::new("project".to_owned(), vec!["test@example.com".to_owned()]);
+		let temp_dir = tempfile::tempdir().expect("failed to create tempdir");
+
+		use graph_craft::gpu::context::Context;
+		use graph_craft::gpu::executor::GpuExecutor;
+		let executor: GpuExecutor<S, O> = GpuExecutor::new(Context::new(), proto_network, m, temp_dir.path()).unwrap();
+
+		let data: Vec<_> = input.into_iter().collect();
+		let result = executor.execute(Box::new(data)).unwrap();
+		let result = dyn_any::downcast::<Vec<O>>(result).unwrap();
+		*result
+	}
+}
+
+impl<I: IntoIterator<Item = S>, NN: Node<()>, S: StaticTypeSized + Sync + Pod + Send, O: StaticTypeSized + Sync + Send + Pod> MapGpuNode<NN, I, S, O> {
+	pub const fn new(network: NN) -> Self {
+		MapGpuNode(network, PhantomData)
+	}
+}
+
+pub struct MapGpuSingleImageNode<NN: Node<(), Output = String>>(pub NN);
+
+impl<NN: Node<(), Output = String> + Copy> Node<Image> for MapGpuSingleImageNode<NN> {
+	type Output = Image;
+	fn eval(self, input: Image) -> Self::Output {
+		let node = self.0.eval(());
+		use graph_craft::document::*;
+
+		let identifier = NodeIdentifier {
+			name: std::borrow::Cow::Owned(node),
+			types: std::borrow::Cow::Borrowed(&[]),
+		};
+
+		let network = NodeNetwork {
+			inputs: vec![0],
+			output: 0,
+			nodes: [(
+				0,
+				DocumentNode {
+					name: "Image filter Node".into(),
+					inputs: vec![NodeInput::Network],
+					implementation: DocumentNodeImplementation::Unresolved(identifier),
+					metadata: DocumentNodeMetadata::default(),
+				},
+			)]
+			.into_iter()
+			.collect(),
+		};
+
+		let value_network = ValueNode::new(network);
+		let map_node = MapGpuNode::new(&value_network);
+		let data = map_node.eval(input.data.clone());
+		Image { data, ..input }
+	}
+}
+
+impl<NN: Node<(), Output = String> + Copy> Node<Image> for &MapGpuSingleImageNode<NN> {
+	type Output = Image;
+	fn eval(self, input: Image) -> Self::Output {
+		let node = self.0.eval(());
+		use graph_craft::document::*;
+
+		let identifier = NodeIdentifier {
+			name: std::borrow::Cow::Owned(node),
+			types: std::borrow::Cow::Borrowed(&[]),
+		};
+
+		let network = NodeNetwork {
+			inputs: vec![0],
+			output: 0,
+			nodes: [(
+				0,
+				DocumentNode {
+					name: "Image filter Node".into(),
+					inputs: vec![NodeInput::Network],
+					implementation: DocumentNodeImplementation::Unresolved(identifier),
+					metadata: DocumentNodeMetadata::default(),
+				},
+			)]
+			.into_iter()
+			.collect(),
+		};
+
+		let value_network = ValueNode::new(network);
+		let map_node = MapGpuNode::new(&value_network);
+		let data = map_node.eval(input.data.clone());
+		Image { data, ..input }
+	}
+}