use graph_craft::document::DocumentNode; use graph_craft::proto::{NodeConstructor, TypeErasedPinned}; use graphene_core::ops::IdNode; use graphene_core::quantization::QuantizationChannels; use graphene_core::raster::bbox::AxisAlignedBbox; use graphene_core::raster::color::Color; use graphene_core::structural::Then; use graphene_core::value::{ClonedNode, CopiedNode, ValueNode}; use graphene_core::vector::brush_stroke::BrushStroke; use graphene_core::vector::VectorData; use graphene_core::wasm_application_io::WasmSurfaceHandle; use graphene_core::wasm_application_io::*; use graphene_core::{concrete, generic, value_fn}; use graphene_core::{fn_type, raster::*}; use graphene_core::{Cow, NodeIdentifier, Type, TypeDescriptor}; use graphene_core::{Node, NodeIO, NodeIOTypes}; use graphene_std::any::{ComposeTypeErased, DowncastBothNode, DynAnyInRefNode, DynAnyNode, FutureWrapperNode, IntoTypeErasedNode, TypeErasedPinnedRef}; use graphene_std::brush::*; use graphene_std::memo::{CacheNode, LetNode}; use graphene_std::raster::BlendImageTupleNode; use graphene_std::raster::*; use dyn_any::StaticType; use glam::{DAffine2, DVec2}; use once_cell::sync::Lazy; use std::collections::HashMap; use std::sync::Arc; macro_rules! construct_node { ($args: ident, $path:ty, [$($type:tt),*]) => { async move { let mut args: Vec> = $args.clone(); args.reverse(); let node = <$path>::new($( { let node = graphene_std::any::input_node::<$type>(args.pop().expect("Not enough arguments provided to construct node")); let value = node.eval(()).await; graphene_core::value::ClonedNode::new(value) } ),* ); node }} } macro_rules! register_node { ($path:ty, input: $input:ty, params: [ $($type:ty),*]) => { vec![ ( NodeIdentifier::new(stringify!($path)), |args| { Box::pin(async move { let node = construct_node!(args, $path, [$($type),*]).await; let node = graphene_std::any::FutureWrapperNode::new(node); let any: DynAnyNode<$input, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, { let node = <$path>::new($( graphene_std::any::PanicNode::<(), $type>::new() ),*); let params = vec![$(value_fn!($type)),*]; let mut node_io = <$path as NodeIO<'_, $input>>::to_node_io(&node, params); node_io.input = concrete!(<$input as StaticType>::Static); node_io }, ) ] }; } macro_rules! raster_node { ($path:ty, params: [$($type:ty),*]) => {{ // this function could also be inlined but serves as a workaround for // [wasm-pack#981](https://github.com/rustwasm/wasm-pack/issues/981). // The non-inlining function leads to fewer locals in the resulting // wasm binary. This issue currently only applies to debug builds, so // we guard inlining to only happen on production builds for // optimization purposes. #[cfg_attr(debug_assertions, inline(never))] #[cfg_attr(not(debug_assertions), inline)] fn generate_triples() -> Vec<(NodeIdentifier, NodeConstructor, NodeIOTypes)> { vec![ ( NodeIdentifier::new(stringify!($path)), |args| { Box::pin(async move { let node = construct_node!(args, $path, [$($type),*]).await; let node = graphene_std::any::FutureWrapperNode::new(node); let any: DynAnyNode = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, { let params = vec![$(value_fn!($type)),*]; NodeIOTypes::new(concrete!(Color), concrete!(Color), params) }, ), ( NodeIdentifier::new(stringify!($path)), |args| { Box::pin(async move { let node = construct_node!(args, $path, [$($type),*]).await; let map_node = graphene_std::raster::MapImageNode::new(graphene_core::value::ValueNode::new(node)); let map_node = graphene_std::any::FutureWrapperNode::new(map_node); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(map_node)); Box::pin(any) as TypeErasedPinned }) }, { let params = vec![$(value_fn!($type)),*]; NodeIOTypes::new(concrete!(Image), concrete!(Image), params) }, ), ( NodeIdentifier::new(stringify!($path)), |args| { Box::pin(async move { let node = construct_node!(args, $path, [$($type),*]).await; let map_node = graphene_std::raster::MapImageNode::new(graphene_core::value::ValueNode::new(node)); let map_node = graphene_std::any::FutureWrapperNode::new(map_node); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(map_node)); Box::pin(any) as TypeErasedPinned }) }, { let params = vec![$(value_fn!($type)),*]; NodeIOTypes::new(concrete!(ImageFrame), concrete!(ImageFrame), params) }, ) ] } generate_triples() }}; } //TODO: turn into hashmap fn node_registry() -> HashMap> { let node_types: Vec> = vec![ //register_node!(graphene_core::ops::IdNode, input: Any<'_>, params: []), vec![( NodeIdentifier::new("graphene_core::ops::IdNode"), |_| Box::pin(async move { Box::pin(FutureWrapperNode::new(IdNode::new())) as TypeErasedPinned }), NodeIOTypes::new(generic!(I), generic!(I), vec![]), )], // TODO: create macro to impl for all types register_node!(graphene_core::structural::ConsNode<_, _>, input: u32, params: [u32]), register_node!(graphene_core::structural::ConsNode<_, _>, input: u32, params: [&u32]), register_node!(graphene_core::structural::ConsNode<_, _>, input: &u32, params: [u32]), register_node!(graphene_core::structural::ConsNode<_, _>, input: &u32, params: [&u32]), register_node!(graphene_core::ops::AddNode, input: (u32, u32), params: []), register_node!(graphene_core::ops::AddNode, input: (u32, &u32), params: []), register_node!(graphene_core::ops::CloneNode<_>, input: &ImageFrame, params: []), register_node!(graphene_core::ops::CloneNode<_>, input: &graphene_core::EditorApi, params: []), register_node!(graphene_core::ops::AddParameterNode<_>, input: u32, params: [u32]), register_node!(graphene_core::ops::AddParameterNode<_>, input: &u32, params: [u32]), register_node!(graphene_core::ops::AddParameterNode<_>, input: u32, params: [&u32]), register_node!(graphene_core::ops::AddParameterNode<_>, input: &u32, params: [&u32]), register_node!(graphene_core::ops::AddParameterNode<_>, input: f64, params: [f64]), register_node!(graphene_core::ops::AddParameterNode<_>, input: &f64, params: [f64]), register_node!(graphene_core::ops::AddParameterNode<_>, input: f64, params: [&f64]), register_node!(graphene_core::ops::AddParameterNode<_>, input: &f64, params: [&f64]), register_node!(graphene_core::ops::SomeNode, input: graphene_core::EditorApi, params: []), register_node!(graphene_core::ops::IntoNode<_, ImageFrame>, input: ImageFrame, params: []), register_node!(graphene_core::ops::IntoNode<_, ImageFrame>, input: ImageFrame, params: []), register_node!(graphene_std::raster::DownresNode<_>, input: ImageFrame, params: []), register_node!(graphene_std::raster::MaskImageNode<_, _, _>, input: ImageFrame, params: [ImageFrame]), register_node!(graphene_std::raster::MaskImageNode<_, _, _>, input: ImageFrame, params: [ImageFrame]), register_node!(graphene_std::raster::InsertChannelNode<_, _, _, _>, input: ImageFrame, params: [ImageFrame, RedGreenBlue]), register_node!(graphene_std::raster::InsertChannelNode<_, _, _, _>, input: ImageFrame, params: [ImageFrame, RedGreenBlue]), vec![( NodeIdentifier::new("graphene_std::raster::CombineChannelsNode"), |args| { Box::pin(async move { use graphene_core::raster::*; use graphene_core::value::*; let channel_r: ImageFrame = DowncastBothNode::new(args[0]).eval(()).await; let channel_g: ImageFrame = DowncastBothNode::new(args[1]).eval(()).await; let channel_b: ImageFrame = DowncastBothNode::new(args[2]).eval(()).await; let channel_a: ImageFrame = DowncastBothNode::new(args[3]).eval(()).await; let insert_r = InsertChannelNode::new(ClonedNode::new(channel_r.clone()), CopiedNode::new(RedGreenBlue::Red)); let insert_g = InsertChannelNode::new(ClonedNode::new(channel_g.clone()), CopiedNode::new(RedGreenBlue::Green)); let insert_b = InsertChannelNode::new(ClonedNode::new(channel_b.clone()), CopiedNode::new(RedGreenBlue::Blue)); let complete_node = insert_r.then(insert_g).then(insert_b); let complete_node = complete_node.then(MaskImageNode::new(ClonedNode::new(channel_a.clone()))); // TODO: Move to FN Node for better performance let (mut transform, mut bounds) = (DAffine2::ZERO, glam::UVec2::ZERO); for image in [channel_a, channel_r, channel_g, channel_b] { if image.image.width() > bounds.x { bounds = glam::UVec2::new(image.image.width(), image.image.height()); transform = image.transform; } } let empty_image = ImageFrame { image: Image::new(bounds.x, bounds.y, Color::BLACK), transform, }; let final_image = ClonedNode::new(empty_image).then(complete_node); let final_image = FutureWrapperNode::new(final_image); let any: DynAnyNode<(), _, _> = graphene_std::any::DynAnyNode::new(ValueNode::new(final_image)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(()), concrete!(ImageFrame), vec![value_fn!(ImageFrame), value_fn!(ImageFrame), value_fn!(ImageFrame), value_fn!(ImageFrame)], ), )], register_node!(graphene_std::raster::EmptyImageNode<_, _>, input: DAffine2, params: [Color]), register_node!(graphene_std::memo::MonitorNode<_>, input: ImageFrame, params: []), register_node!(graphene_std::memo::MonitorNode<_>, input: graphene_core::GraphicGroup, params: []), register_node!(graphene_core::wasm_application_io::CreateSurfaceNode, input: &graphene_core::EditorApi, params: []), vec![( NodeIdentifier::new("graphene_core::wasm_application_io::DrawImageFrameNode<_>"), |args| { Box::pin(async move { let surface: DowncastBothNode<(), Arc> = DowncastBothNode::new(args[0]); let node = graphene_core::wasm_application_io::DrawImageFrameNode::new(surface); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(ImageFrame), concrete!(WasmSurfaceHandleFrame), vec![value_fn!(Arc)]), )], #[cfg(feature = "gpu")] vec![( NodeIdentifier::new("graphene_std::executor::MapGpuSingleImageNode<_>"), |args| { Box::pin(async move { let document_node: DowncastBothNode<(), DocumentNode> = DowncastBothNode::new(args[0]); //let document_node = ClonedNode::new(document_node.eval(())); let node = graphene_std::executor::MapGpuNode::new(document_node); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(ImageFrame), concrete!(ImageFrame), vec![value_fn!(DocumentNode)]), )], #[cfg(feature = "gpu")] vec![( NodeIdentifier::new("graphene_std::executor::BlendGpuImageNode<_, _, _>"), |args| { Box::pin(async move { let background: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let blend_mode: DowncastBothNode<(), BlendMode> = DowncastBothNode::new(args[1]); let opacity: DowncastBothNode<(), f32> = DowncastBothNode::new(args[2]); let node = graphene_std::executor::BlendGpuImageNode::new(background, blend_mode, opacity); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(ImageFrame), concrete!(ImageFrame), vec![value_fn!(ImageFrame), value_fn!(BlendMode), value_fn!(f32)], ), )], vec![( NodeIdentifier::new("graphene_core::structural::ComposeNode<_, _, _>"), |args| { Box::pin(async move { let node = ComposeTypeErased::new(args[0], args[1]); node.into_type_erased() }) }, NodeIOTypes::new( generic!(T), generic!(U), vec![Type::Fn(Box::new(generic!(T)), Box::new(generic!(V))), Type::Fn(Box::new(generic!(V)), Box::new(generic!(U)))], ), )], register_node!(graphene_std::brush::IntoIterNode<_>, input: &Vec, params: []), vec![( NodeIdentifier::new("graphene_std::brush::BrushNode"), |args| { use graphene_core::structural::*; use graphene_core::value::*; use graphene_std::brush::*; Box::pin(async move { let image: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let bounds: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[1]); let strokes: DowncastBothNode<(), Vec> = DowncastBothNode::new(args[2]); let strokes = strokes.eval(()).await; let bbox = strokes.iter().map(|s| s.bounding_box()).reduce(|a, b| a.union(&b)).unwrap_or(AxisAlignedBbox::ZERO); let mut background_bounds = CopiedNode::new(bbox.to_transform()); let bounds_transform = bounds.eval(()).await.transform; if bounds_transform != DAffine2::ZERO { background_bounds = CopiedNode::new(bounds_transform); } let blank_image = background_bounds.then(EmptyImageNode::new(CopiedNode::new(Color::TRANSPARENT))); let background = image.and_then(ExtendImageNode::new(blank_image)); let mut blits = Vec::new(); for stroke in strokes { let stamp = BrushStampGeneratorNode::new(CopiedNode::new(stroke.style.color), CopiedNode::new(stroke.style.hardness), CopiedNode::new(stroke.style.flow)); let stamp = stamp.eval(stroke.style.diameter); let transform = DAffine2::from_scale_angle_translation(DVec2::splat(stroke.style.diameter), 0., -DVec2::splat(stroke.style.diameter / 2.0)); let blank_texture = EmptyImageNode::new(CopiedNode::new(Color::TRANSPARENT)).eval(transform); let blend_params = graphene_core::raster::BlendNode::new(CopiedNode::new(BlendMode::Normal), CopiedNode::new(100.)); let blend_executor = BlendImageTupleNode::new(ValueNode::new(blend_params)); let texture = blend_executor.eval((blank_texture, stamp)); let translations: Vec<_> = stroke.compute_blit_points().into_iter().collect(); let blit_node = BlitNode::new(ClonedNode::new(texture), ClonedNode::new(translations), ClonedNode::new(blend_params)); blits.push(blit_node); } let all_blits = ChainApplyNode::new(background); let node = ClonedNode::new(blits.into_iter()).then(all_blits); let any: DynAnyNode<(), _, _> = graphene_std::any::DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(()), concrete!(ImageFrame), vec![value_fn!(ImageFrame), value_fn!(ImageFrame), value_fn!(Vec)], ), )], vec![( NodeIdentifier::new("graphene_std::brush::ReduceNode<_, _>"), |args| { Box::pin(async move { let acc: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let image = acc.eval(()).await; let blend_node = graphene_core::raster::BlendNode::new(ClonedNode::new(BlendMode::Normal), ClonedNode::new(1.0)); let _ = &blend_node as &dyn for<'i> Node<'i, (Color, Color), Output = Color>; let node = ReduceNode::new(ClonedNode::new(image), ValueNode::new(BlendImageTupleNode::new(ValueNode::new(blend_node)))); //let _ = &node as &dyn for<'i> Node<'i, core::slice::Iter>, Output = ImageFrame>; let node = FutureWrapperNode::new(node); let any: DynAnyNode> + Sync + Send>, _, _> = graphene_std::any::DynAnyNode::new(ValueNode::new(node)); any.into_type_erased() }) }, NodeIOTypes::new( concrete!(Box> + Sync + Send>), concrete!(ImageFrame), vec![value_fn!(ImageFrame)], ), )], // Filters raster_node!(graphene_core::raster::LuminanceNode<_>, params: [LuminanceCalculation]), raster_node!(graphene_core::raster::ExtractChannelNode<_>, params: [RedGreenBlue]), raster_node!(graphene_core::raster::ExtractAlphaNode<>, params: []), raster_node!(graphene_core::raster::LevelsNode<_, _, _, _, _>, params: [f64, f64, f64, f64, f64]), register_node!(graphene_std::image_segmentation::ImageSegmentationNode<_>, input: ImageFrame, params: [ImageFrame]), register_node!(graphene_core::raster::IndexNode<_>, input: Vec>, params: [u32]), vec![( NodeIdentifier::new("graphene_core::raster::BlendNode<_, _, _, _>"), |args| { Box::pin(async move { let image: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let blend_mode: DowncastBothNode<(), BlendMode> = DowncastBothNode::new(args[1]); let opacity: DowncastBothNode<(), f64> = DowncastBothNode::new(args[2]); let blend_node = graphene_core::raster::BlendNode::new(CopiedNode::new(blend_mode.eval(()).await), CopiedNode::new(opacity.eval(()).await)); let node = graphene_std::raster::BlendImageNode::new(image, FutureWrapperNode::new(ValueNode::new(blend_node))); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(graphene_core::value::ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(ImageFrame), concrete!(ImageFrame), vec![value_fn!(ImageFrame), value_fn!(BlendMode), value_fn!(f64)], ), )], raster_node!(graphene_core::raster::GrayscaleNode<_, _, _, _, _, _, _>, params: [Color, f64, f64, f64, f64, f64, f64]), raster_node!(graphene_core::raster::HueSaturationNode<_, _, _>, params: [f64, f64, f64]), raster_node!(graphene_core::raster::InvertRGBNode, params: []), raster_node!(graphene_core::raster::ThresholdNode<_, _, _>, params: [f64, f64, LuminanceCalculation]), raster_node!(graphene_core::raster::VibranceNode<_>, params: [f64]), raster_node!( graphene_core::raster::ChannelMixerNode<_, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _>, params: [bool, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64] ), raster_node!( graphene_core::raster::SelectiveColorNode<_, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _, _>, params: [RelativeAbsolute, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64, f64] ), vec![( NodeIdentifier::new("graphene_core::raster::BrightnessContrastNode<_, _, _>"), |args| { Box::pin(async move { use graphene_core::raster::brightness_contrast::*; let brightness: DowncastBothNode<(), f64> = DowncastBothNode::new(args[0]); let brightness = ClonedNode::new(brightness.eval(()).await as f32); let contrast: DowncastBothNode<(), f64> = DowncastBothNode::new(args[1]); let contrast = ClonedNode::new(contrast.eval(()).await as f32); let use_legacy: DowncastBothNode<(), bool> = DowncastBothNode::new(args[2]); if use_legacy.eval(()).await { let generate_brightness_contrast_legacy_mapper_node = GenerateBrightnessContrastLegacyMapperNode::new(brightness, contrast); let map_image_frame_node = graphene_std::raster::MapImageNode::new(ValueNode::new(generate_brightness_contrast_legacy_mapper_node.eval(()))); let map_image_frame_node = FutureWrapperNode::new(map_image_frame_node); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(ValueNode::new(map_image_frame_node)); Box::pin(any) as TypeErasedPinned } else { let generate_brightness_contrast_mapper_node = GenerateBrightnessContrastMapperNode::new(brightness, contrast); let map_image_frame_node = graphene_std::raster::MapImageNode::new(ValueNode::new(generate_brightness_contrast_mapper_node.eval(()))); let map_image_frame_node = FutureWrapperNode::new(map_image_frame_node); let any: DynAnyNode, _, _> = graphene_std::any::DynAnyNode::new(ValueNode::new(map_image_frame_node)); Box::pin(any) as TypeErasedPinned } }) }, NodeIOTypes::new(concrete!(ImageFrame), concrete!(ImageFrame), vec![value_fn!(f64), value_fn!(f64), value_fn!(bool)]), )], raster_node!(graphene_core::raster::OpacityNode<_>, params: [f64]), raster_node!(graphene_core::raster::PosterizeNode<_>, params: [f64]), raster_node!(graphene_core::raster::ExposureNode<_, _, _>, params: [f64, f64, f64]), vec![ ( NodeIdentifier::new("graphene_std::memo::LetNode<_>"), |_| { Box::pin(async move { let node: LetNode> = graphene_std::memo::LetNode::new(); let any = graphene_std::any::DynAnyRefNode::new(node); any.into_type_erased() }) }, NodeIOTypes::new(concrete!(Option>), concrete!(&ImageFrame), vec![]), ), ( NodeIdentifier::new("graphene_std::memo::LetNode<_>"), |_| { Box::pin(async move { let node: LetNode = graphene_std::memo::LetNode::new(); let any = graphene_std::any::DynAnyRefNode::new(node); any.into_type_erased() }) }, NodeIOTypes::new(concrete!(Option), concrete!(&graphene_core::EditorApi), vec![]), ), ( NodeIdentifier::new("graphene_std::memo::EndLetNode<_>"), |args| { Box::pin(async move { let input: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let node = graphene_std::memo::EndLetNode::new(input); let any: DynAnyInRefNode = graphene_std::any::DynAnyInRefNode::new(node); Box::pin(any) as TypeErasedPinned<'_> }) }, NodeIOTypes::new(generic!(T), concrete!(ImageFrame), vec![value_fn!(ImageFrame)]), ), ( NodeIdentifier::new("graphene_std::memo::EndLetNode<_>"), |args| { Box::pin(async move { let input: DowncastBothNode<(), VectorData> = DowncastBothNode::new(args[0]); let node = graphene_std::memo::EndLetNode::new(input); let any: DynAnyInRefNode = graphene_std::any::DynAnyInRefNode::new(node); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(generic!(T), concrete!(VectorData), vec![value_fn!(VectorData)]), ), ( NodeIdentifier::new("graphene_std::memo::EndLetNode<_>"), |args| { Box::pin(async move { let input: DowncastBothNode<(), graphene_core::GraphicGroup> = DowncastBothNode::new(args[0]); let node = graphene_std::memo::EndLetNode::new(input); let any: DynAnyInRefNode = graphene_std::any::DynAnyInRefNode::new(node); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(generic!(T), concrete!(graphene_core::GraphicGroup), vec![value_fn!(graphene_core::GraphicGroup)]), ), ( NodeIdentifier::new("graphene_std::memo::EndLetNode<_>"), |args| { Box::pin(async move { let input: DowncastBothNode<(), graphene_core::Artboard> = DowncastBothNode::new(args[0]); let node = graphene_std::memo::EndLetNode::new(input); let any: DynAnyInRefNode = graphene_std::any::DynAnyInRefNode::new(node); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(generic!(T), concrete!(graphene_core::Artboard), vec![value_fn!(graphene_core::Artboard)]), ), ( NodeIdentifier::new("graphene_std::memo::EndLetNode<_>"), |args| { Box::pin(async move { let input: DowncastBothNode<(), WasmSurfaceHandleFrame> = DowncastBothNode::new(args[0]); let node = graphene_std::memo::EndLetNode::new(input); let any: DynAnyInRefNode = graphene_std::any::DynAnyInRefNode::new(node); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(generic!(T), concrete!(WasmSurfaceHandleFrame), vec![value_fn!(WasmSurfaceHandleFrame)]), ), ( NodeIdentifier::new("graphene_std::memo::RefNode<_, _>"), |args| { Box::pin(async move { let map_fn: DowncastBothNode, &graphene_core::EditorApi> = DowncastBothNode::new(args[0]); //let map_fn = map_fn.then(EvalSyncNode::new()); let node = graphene_std::memo::RefNode::new(map_fn); let any = graphene_std::any::DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(()), concrete!(&graphene_core::EditorApi), vec![fn_type!(Option, &graphene_core::EditorApi)], ), ), /* ( NodeIdentifier::new("graphene_std::raster::ImaginateNode<_>"), |args| { Box::pin(async move { let cached = graphene_std::any::input_node::>>>(args[15]); let cached = cached.then(EvalSyncNode::new()); let node = graphene_std::raster::ImaginateNode::new(cached); let node = FutureWrapperNode::new(node); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new( concrete!(ImageFrame), concrete!(ImageFrame), vec![ value_fn!(f64), value_fn!(Option), value_fn!(f64), value_fn!(ImaginateSamplingMethod), value_fn!(f64), value_fn!(String), value_fn!(String), value_fn!(bool), value_fn!(f64), value_fn!(Option>), value_fn!(bool), value_fn!(f64), value_fn!(ImaginateMaskStartingFill), value_fn!(bool), value_fn!(bool), value_fn!(Option>>), value_fn!(f64), value_fn!(ImaginateStatus), ], ), ), */ /* ( NodeIdentifier::new("graphene_core::raster::BlurNode"), |args| { let radius = DowncastBothNode::<(), u32>::new(args[0]); let sigma = DowncastBothNode::<(), f64>::new(args[1]); let image = DowncastBothRefNode::, Image>::new(args[2]); let image = image.then(EvalSyncNode::new()); let empty_image: ValueNode> = ValueNode::new(Image::empty()); let empty: TypeNode<_, (), Image> = TypeNode::new(empty_image.then(CloneNode::new())); use graphene_core::Node; let radius = ClonedNode::new(radius.eval(())); let sigma = ClonedNode::new(sigma.eval(())); //let image = &image as &dyn for<'a> Node<'a, (), Output = &'a Image>; // dirty hack: we abuse that the cache node will ignore the input if it is evaluated a second time let image = empty.then(image).then(ImageRefNode::new()); let window = WindowNode::new(radius, image.clone()); let map_gaussian = MapSndNode::new(ValueNode::new(DistanceNode.then(GaussianNode::new(sigma)))); let map_distances = MapNode::new(ValueNode::new(map_gaussian)); let gaussian_iter = window.then(map_distances); let avg = gaussian_iter.then(WeightedAvgNode::new()); let avg: TypeNode<_, u32, Color> = TypeNode::new(avg); let blur_iter = MapNode::new(ValueNode::new(avg)); let pixel_iter = image.clone().then(ImageIndexIterNode::new()); let blur = pixel_iter.then(blur_iter); let collect = CollectNode {}; let vec = blur.then(collect); let new_image = MapImageSliceNode::new(vec); let dimensions = image.then(ImageDimensionsNode::new()); let dimensions: TypeNode<_, (), (u32, u32)> = TypeNode::new(dimensions); let new_image = dimensions.then(new_image); let new_image = ForgetNode::new().then(new_image); let new_image = FutureWrapperNode::new(new_image); let node: DynAnyNode<&Image, _, _> = DynAnyNode::new(ValueNode::new(new_image)); Box::pin(node) }, NodeIOTypes::new(concrete!(Image), concrete!(Image), vec![value_fn!(u32), value_fn!(f64)]), ), //register_node!(graphene_std::memo::CacheNode<_>, input: Image, params: []), */ ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode<(), Image> = DowncastBothNode::new(args[0]); let node: CacheNode, _> = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(()), concrete!(Image), vec![value_fn!(Image)]), ), ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode<(), ImageFrame> = DowncastBothNode::new(args[0]); let node: CacheNode, _> = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(()), concrete!(ImageFrame), vec![value_fn!(ImageFrame)]), ), ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode, ImageFrame> = DowncastBothNode::new(args[0]); let node: CacheNode, _> = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(ImageFrame), concrete!(ImageFrame), vec![fn_type!(ImageFrame, ImageFrame)]), ), ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode<(), QuantizationChannels> = DowncastBothNode::new(args[0]); let node: CacheNode = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(()), concrete!(QuantizationChannels), vec![value_fn!(QuantizationChannels)]), ), ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode<(), Vec> = DowncastBothNode::new(args[0]); let node: CacheNode, _> = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(()), concrete!(Vec), vec![value_fn!(Vec)]), ), ( NodeIdentifier::new("graphene_std::memo::CacheNode"), |args| { Box::pin(async move { let input: DowncastBothNode<(), Arc> = DowncastBothNode::new(args[0]); let node: CacheNode, _> = graphene_std::memo::CacheNode::new(input); let any = DynAnyNode::new(ValueNode::new(node)); Box::pin(any) as TypeErasedPinned }) }, NodeIOTypes::new(concrete!(()), concrete!(Arc), vec![value_fn!(Arc)]), ), ], register_node!(graphene_core::structural::ConsNode<_, _>, input: Image, params: [&str]), register_node!(graphene_std::raster::ImageFrameNode<_, _>, input: Image, params: [DAffine2]), #[cfg(feature = "quantization")] register_node!(graphene_std::quantization::GenerateQuantizationNode<_, _>, input: ImageFrame, params: [u32, u32]), raster_node!(graphene_core::quantization::QuantizeNode<_>, params: [QuantizationChannels]), raster_node!(graphene_core::quantization::DeQuantizeNode<_>, params: [QuantizationChannels]), register_node!(graphene_core::ops::CloneNode<_>, input: &QuantizationChannels, params: []), register_node!(graphene_core::transform::TransformNode<_, _, _, _, _>, input: VectorData, params: [DVec2, f64, DVec2, DVec2, DVec2]), register_node!(graphene_core::transform::TransformNode<_, _, _, _, _>, input: ImageFrame, params: [DVec2, f64, DVec2, DVec2, DVec2]), register_node!(graphene_core::transform::TransformNode<_, _, _, _, _>, input: WasmSurfaceHandleFrame, params: [DVec2, f64, DVec2, DVec2, DVec2]), register_node!(graphene_core::transform::SetTransformNode<_>, input: VectorData, params: [VectorData]), register_node!(graphene_core::transform::SetTransformNode<_>, input: ImageFrame, params: [ImageFrame]), register_node!(graphene_core::transform::SetTransformNode<_>, input: VectorData, params: [DAffine2]), register_node!(graphene_core::transform::SetTransformNode<_>, input: ImageFrame, params: [DAffine2]), register_node!(graphene_core::vector::SetFillNode<_, _, _, _, _, _, _>, input: VectorData, params: [graphene_core::vector::style::FillType, Option, graphene_core::vector::style::GradientType, DVec2, DVec2, DAffine2, Vec<(f64, Option)>]), register_node!(graphene_core::vector::SetStrokeNode<_, _, _, _, _, _, _>, input: VectorData, params: [Option, f64, Vec, f64, graphene_core::vector::style::LineCap, graphene_core::vector::style::LineJoin, f64]), register_node!(graphene_core::vector::generator_nodes::UnitCircleGenerator, input: (), params: []), register_node!( graphene_core::vector::generator_nodes::PathGenerator<_>, input: Vec>, params: [Vec] ), register_node!(graphene_core::text::TextGenerator<_, _, _>, input: &graphene_core::EditorApi, params: [String, graphene_core::text::Font, f64]), register_node!(graphene_std::brush::VectorPointsNode, input: VectorData, params: []), register_node!(graphene_core::ExtractImageFrame, input: &graphene_core::EditorApi, params: []), register_node!(graphene_core::ConstructLayerNode<_, _, _, _, _, _, _>, input: graphene_core::vector::VectorData, params: [String, BlendMode, f32, bool, bool, bool, graphene_core::GraphicGroup]), register_node!(graphene_core::ConstructLayerNode<_, _, _, _, _, _, _>, input: ImageFrame, params: [String, BlendMode, f32, bool, bool, bool, graphene_core::GraphicGroup]), register_node!(graphene_core::ConstructLayerNode<_, _, _, _, _, _, _>, input: graphene_core::GraphicGroup, params: [String, BlendMode, f32, bool, bool, bool, graphene_core::GraphicGroup]), register_node!(graphene_core::ConstructLayerNode<_, _, _, _, _, _, _>, input: graphene_core::Artboard, params: [String, BlendMode, f32, bool, bool, bool, graphene_core::GraphicGroup]), register_node!(graphene_core::ConstructArtboardNode<_, _, _>, input: graphene_core::GraphicGroup, params: [glam::IVec2, glam::IVec2, Color]), ]; let mut map: HashMap> = HashMap::new(); for (id, c, types) in node_types.into_iter().flatten() { // TODO: this is a hack to remove the newline from the node new_name // This occurs for the ChannelMixerNode presumably because of the long name. // This might be caused by the stringify! macro let new_name = id.name.replace('\n', " "); let nid = NodeIdentifier { name: Cow::Owned(new_name) }; map.entry(nid).or_default().insert(types.clone(), c); } map } pub static NODE_REGISTRY: Lazy>> = Lazy::new(|| node_registry()); #[cfg(test)] mod protograph_testing { // TODO: adde tests testing the node registry }