Files
Graphite/node-graph/nodes/text/src/lib.rs
Keavon Chambers 2090e9979a Make the data model use Item and List types universally, with nodes authored as rank-polymorphic kernels (#4335)
* Add rank polymorphism node audit classifying all 271 nodes

* Implement StaticType for Item<T>

* Generate Item and mapped List wire variants for nodes declaring an Item<T> primary input

* Migrate nine nodes to Item element-wise kernels, dissolving the blending trait boilerplate

* Document the Item kernel implementation and staging plan

* Route Item<Vector> through TaggedValue::TypeDefault

* Add executor integration tests covering the Item and List wire variants

* Collapse element-wise Item/List wire pairs to the List form for conversion insertion

* Migrate sixteen vector modifier nodes to Item element-wise kernels

* Migrate Sample Image, Extend Image to Bounds, and Dehaze to Item element-wise kernels

* Fix bevel_with_transform test to actually exercise the transform attribute

* Implement From<T> for Item<T>

* Register PromoteNode rank adapters wrapping bare values into Item wires

* Insert PromoteNode adapters for Item/List wire pair fields in the preprocessor

* Define a real promote node backing the PromoteNode registry identifiers

* Zip ranked Item connectors by frame slot in the mapped element-wise variant

* Register ItemToListNode singleton raise adapters

* Resolve Item wires against List connectors by inserting promotion adapters at construction

* Rank the Offset Points distance connector and prove mixed-rank resolution end-to-end

* Implement Clampable for Item and List wires with per-variant clamp bounds

* Rank the Round Corners radius connector, exercising hard bounds on a ranked wire

* Implement ApplyTransform for Item

* Add Item wire implementations to the Transform node, keeping rank-0 chains rank 0

* Detect element-wise nodes by lazy primary connectors declaring Output = Item

* Convert Transform to an Item kernel with ranked parameters, delivering the broadcast milestone

* Rename Apply Transform to Bake Transform, baking item transforms on Vector, DAffine2, and DVec2

* Promote bare wires onto Item connectors at resolution via WrapItemNode adapters

* Rank the numeric, vector, and boolean parameters across the migrated element-wise nodes

* Rank the enum, integer, and seed parameters, registering their rank adapters via a consolidated macro

* Amend the audit with the DashPattern value type resolution

* Migrate the string family to Item element-wise kernels

* Unwrap Item wires into bare legacy connectors at resolution via UnwrapItemNode adapters

* Shadow owned node parameters in bodies instead of mut in signatures

* Migrate the math family and string measure nodes to Item element-wise kernels

* Convert the comparison and clamp nodes to Item kernels, dropping unreachable &str rows

* Flat-map expander kernels returning List under the mapped variant's frame

* Migrate the expander nodes to Item kernels flat-mapping under the frame

* Remove the unused peel_list helper

* Rank the raster adjustment and blending kernels, recontextualizing shader nodes onto an Item stand-in

Migrate the 16 adjustment nodes, Mix, Color Overlay, and Gradient Map from whole-List kernels to rank-0 Item kernels, letting the macro derive the List-mapped (zip) variants. Move the Adjust and Blend per-element seams off List onto the element types (add the Raster<CPU> impls, drop the now-dead List impls).

Shader nodes keep their bodies verbatim: PerPixelAdjust re-emits the identical kernel against a transparent no_std Item stand-in, so every Item<T> connector and .element() call resolves to a zero-cost identity on the GPU while the uniform buffer stays bare repr(C). The macro peels Item off ranked uniform params, wraps the fetched texel and uniforms at the entry point, and unwraps the result. This drops the shader_node/Item incompatibility guard. Register rank adapters for the adjustment enums.

* Update the rank polymorphism roadmap for the landed shader-node and adjustments chunk

* Rename the GPU Item stand-in to ShaderItem, aliased as Item at its shader-node import sites

* Flip the vector shape generators to emit rank-0 Item<Vector>

The shape generators (Rectangle, Circle, Ellipse, Arc, Spiral, Polygon, Star, Arrow, Line, Grid, QR Code) each produced exactly one shape wrapped in a singleton List<Vector>. Emit Item<Vector> directly so they connect to the rank-0 content connector of the migrated Transform node. Downstream List consumers receive the value through the existing Item to List promotion.

Relax the element-wise validation so a `()` (generator) primary may return Item<T> without being element-wise. Adapt the Repeat on Points test, which still takes a List content connector, by raising the generator's Item output through a singleton wrapper node.

* Parse ranked Item<T> parameter defaults against the bare element type

A ranked `Item<T>` parameter's default value is a bare, unranked `T` (promoted to the wire at resolution), but the preprocessor was handed the wrapped `Item<T>` type and could not parse the literal, flooding the console with warnings and dropping the defaults. Key the field's default_type metadata off the peeled element type for concrete ranked parameters, leaving generic `Item<T>` primaries and skip_impl nodes untouched.

* Parse an element-wise primary's scalar default against the bare element type

An element-wise node's primary reports its default_type as the List wire form so an unconnected primary defaults to an empty list. But when the primary carries a scalar `#[default]` (such as Root's radicand), that literal must parse as a bare element, not a List. Key the primary's default_type off the bare element type when it has a Default value source, keeping the List form otherwise.

* Add the DashPattern value type for stroke dash sequences

Introduce a rank-0 DashPattern value type (a Vec<f64> of alternating dash and gap lengths) so a stroke's dash pattern is a single frameable value rather than a rank-1 List<f64>. Register it as an auto-generated TaggedValue variant, parse its default from a comma or space separated string, and register its rank adapters. Not yet wired into the Stroke node.

* Rank the Fill and Stroke nodes element-wise and give Stroke a DashPattern connector

Migrate Fill and Stroke to element-wise Item<V> primaries (over Vector and Graphic element types) via a new element-level VectorItemMut trait, so styling one shape yields one shape and rank is preserved instead of promoting the input to a singleton List and emitting a List. The macro derives the List-mapped variant for genuine collections.

Wire the Stroke dash sequence to the new rank-0 DashPattern value type, collapsing the old content x paint x dash cartesian and dropping the IntoF64Vec trait. Update the stroke properties dash widget, the drawing tool, and graph-operation plumbing to read and write DashPattern, and migrate legacy F64Array, F64, and String dash inputs on document open.

Assign Colors stays a whole-collection node: each element's gradient position depends on its index among all siblings, which the element frame does not expose, so it keeps its List primary and the VectorListIterMut trait.

* Register rank adapters for the ranked Stroke enum parameters

The element-wise Stroke node ranks its align, cap, and paint order parameters as Item<StrokeAlign>, Item<StrokeCap>, and Item<PaintOrder>, but those enums lacked promotion adapters, so a bare default enum value could not be promoted to its Item wire and no Stroke variant resolved ("No construct found for node"). Register their rank adapters alongside StrokeJoin.

* Display Item wires in the Data panel without a List's ID column

Add a TableItemLayout impl for Item<T> and recognize Item wire types when introspecting graph data. An Item holds a single element, so it renders as a one-row table of the element plus its attributes with no leading index column, and it labels as its element type T rather than a List's T[]. Add ItemAttributeValues::get_any for the attribute widget dispatch.

* Register MonitorNode for Item wire types so the Data panel introspects them directly

Graph introspection wraps the inspected output in a generic MonitorNode typed to the wire. Without Item<T> monitor registrations, an Item<Vector> output could only be monitored after an Item to List promotion, so the Data panel captured and displayed a List<Vector> despite the connector being Item<Vector>. Register monitors for the Item types the element-wise nodes emit, and add the matching Data panel downcast entries.

* Color and double Item/List wires and cleave layer-stack connectors in the node graph

* Route wire color and rank through hidden nodes and refresh them on type changes

* Rework the DashPattern connector conversions with element-wise promotion and an explicit reducer node

* Rank the remaining value, context, aggregation, and transform nodes onto Item<T> wires

* Back DashPattern with a List<f64> so the Data panel can introspect its lengths

* Carry a single Item<T> through varargs so the Read context nodes emit Item<T> not List<T>

* Relax rank validation for aggregation shapes, add element adapters, and match variants by fewest promotions

* Rank the remaining bare and unnecessarily-List connectors across the node catalog

* Add Graphic::None and the FillChoice paint value, making colors and gradients plain values

* Rename GradientStops to Gradient and the legacy Gradient/Fill structs to LegacyGradient/LegacyFill

* Restore generator frame-from-params ranking to the roadmap as a planned stage

* Rename the ranked-field adapter identifier from PromoteNode to FieldAdapterNode to reflect its full contract

* Unload only the wires whose displayed style changed when types update

* Peel wire rank in the editor's semantic type checks so rank-0 layers are recognized

* Restore the whole-List Transform variant so rank-1 content wires resolve again

* Register the Item wire forms for the Memoize and Context Modification infrastructure nodes

* Give every ranked connector a field adapter and add numeric cast variants for legacy wires

* Key a ranked param's type default off its Item wire form when no literal default exists

* Inherit the layer's content value when splicing a node into an empty chain

* Migrate stale List-form TypeDefault inputs to the definition's current default

* Generate the mapped wire variant only when the element-wise node has a frame source

* Let a bare wire feed a List connector via a wrap-raise adapter, costed as two rank steps

* Add a zip companion to the whole-List Transform so ranked List parameters pair per slot

* Add the Sum, Average, Minimum, Maximum, Any, and All list reducers

* Convert the measure family to element-wise Item kernels per the audit classification

* Prefer the bare element value over the Item type default so ranked params keep their widgets

* Rename GradientStopsUI to GradientUI

* Split Fill's optional transform into a _has_transform bool and a ranked _transform matrix

* Rename the migration-only OptionalDAffine2 TaggedValue to LegacyOptionalDAffine2

* Flow byte buffers as Item<Resource> instead of List<u8> across the byte nodes

* Macro-generate the list-content wire variant, retiring the hand-written Transform-zip, Area, and Centroid companions

* Let ()-primary generators take ranked params and frame over them via the mapped variant, ranking Circle's radius

* Rank the vector shape generators' params to Item, adding a rank-aware input grab to the introspection harness

* Rank the value, color, and text generator params to Item

* Rank the raster, web-request, and context-reader generator params to Item

* Fix the repeat and brush test wirings left behind by the param-ranking sweeps

* Delete the vestigial Some, Unwrap Option, and Size Of debug nodes

* Delete the Attach Attribute node, folding its role into Write Attribute

* Add the Filter and Sort list companion nodes

* Guard the removed-definition migration swap target with a test

* Add the Box Corners value type in place of the rectangle corner radius list

* Split Text to Vector's per-glyph mode into a Text to Vector Glyphs node

* Rank the Combine Channels node's channel connectors to Item

* Make Map Points an element-wise node

* Delete the deprecated Upload Texture node

* Update the implementation roadmap to reflect the landed stages

* Let monitor introspection read rank-0 wires, locking in the layer coercion promotion path

* Prefer the rank-0 default when disconnecting a rank-capable input

* Make Path Modify an element-wise node

* Wrap node paths in a NodeIdPath newtype so they flow as a single Item

* Give Item<Raster<CPU>> a default so an unconnected Brush background resolves

* Stop the Brush node from setting layer attributes its paint operation doesn't produce

* Present-gate Flatten Path's adopted layer path like its fill and stroke

* Gate carried layer attributes on static column presence, not runtime values

* Give the remaining graphic Item<T> types a default so unconnected primaries resolve

* Dispatch a ranked param's Properties widget from its rank-0 element type

* Make Extract Transform an element-wise node, restoring the Origins to Polyline body

* Rename Flatten Path to Combine Paths

* Stamp Legacy Layer Extend's adopted layer path as a readable NodeIdPath

* Drop the dead List<u8> and List<NodeId> wire rows

* Rank Flatten Graphic's Fully Flatten toggle to Item

* Update the implementation roadmap with the endgame scope

* Make Combine Paths a reducer that collapses the whole frame into one path

* Stop type-converter nodes from carrying the source's unrelated attributes

* Format the Origins to Polyline regression test

* Wrap the Brush node's trace in a BrushTrace newtype so it flows as one value

* Make Switch a framed element-wise select, bundling whole collections

* Widen and align element-type coverage across the list and graphic nodes

* Register the compiler's cache chain pair for every ranked enum and newtype wire

* Fix wire colors for Passthrough outputs, bundled lists, and bools, and widen list wires

* Represent List wire types structurally with Type::List, replacing name-parsed rank promotion

* Treat scope and data fields as environment, rank scope wires as Item, and feed the render boundary through a context vararg

* Delete the vestigial Clone debug node

* Reinstate Upload Texture as an element-wise node and fix the GPU variants' scope executor and rank adapters

* Rename Combine Paths back to Flatten Path, deferring that rename to its own PR

* Deduplicate the promotion adapter registrations into the field adapter macro

* Rank Write Attribute's value connector to Item<AttributeValueDyn>, retiring the UnwrapItem bridge

* Vertical wire styling

* Store the editor layer path attribute as a bare NodeIdPath, not an Item<NodeIdPath>

* Rank Context Modification's features connector to Item<ContextFeatures>, dropping the dead memoize row

* Rank Path Modify's modification parameter to Item<Box<VectorModification>>

* Rename the field adapter node family to input adapter

* Drop the dead bare scalar rows from Context Modification's implementations list

* Move the dynamic executor's test module into its own file

* Drop the registry's unreachable bare rows for Memoize, the cache chain, and ConvertNode

* Materialize stored TaggedValues as ranked Item wires at the source

* Remove the bare-wire promotion and adapter machinery made dead by ranked value materialization

* Plant the input adapter for List-only inputs, composing position conversion from standard rows

* Consolidate Into/Convert conversions into the input adapter umbrella and rename the rank adapter identifiers

* Fix grouped layers gaining a phantom None stack element from the FillChoice default hijacking every List<Graphic> disconnect

* Enforce ranked node inputs in the macro, rejecting bare wire declarations

* Remove the unit Context => () machinery rows, leaving () purely as the no-primary sentinel

* Add a --signatures rank-audit mode to node-docs for the ranked-wire migration

* Remove the node-docs --signatures rank-audit mode now that ranked wires are enforced

* Migrate legacy no-color values on the Black & White, Color Overlay, and Empty Image color inputs

* Rewrite the element-wise accessor wire type at the primary input, not raw index 0

* Register the cache chain for Resource wires, replacing the lone hand-written Monitor row

* Gate the remaining Raster<GPU> registry rows behind the gpu feature

* Let List<DVec2> wires erase to ListDyn for the attribute reader and element counter

* Rename Extract Element to Item at Index, Count Elements to List Length, and Omit Element to Remove at Index

* Store paint picks as plain color/gradient values, removing the FillChoice value type

* Code review restructuring

* Sort by the consumed sort_key attribute or natural element order, adding the Sort Key node

* Remove the new list-combinator and reducer nodes to defer them to a follow-up PR

* Parse Fill and Stroke color defaults through the paint wire's Graphic element

* Emit ranked implementation-row default types structurally so their element TypeIds survive to default-literal parsing

* Exempt the deliberate no-paint choice from the stale List-form TypeDefault migration

* Migrate the legacy 4-input Fill directly to the split has-transform shape

* Upgrade the demo artwork

* Fix the valid AI review findings: Item eq/hash contract, table-era no-paint migration, quantize List rows, and other smaller issues

* Remove the rank polymorphism working documents

* Hash Item attribute values directly instead of debug-formatting them, speeding up cached evaluation

* Replace the data panel's dead bare-wire downcast arms with full coverage of the ranked monitor row types

* Derive PartialEq for Item now that attributes participate in equality

* Extend the data panel's attribute dispatchers with the newly supported scalar and choice enum types

* Add List monitor rows for the framed numeric conversion outputs so inspecting them resolves, with matching data panel arms
2026-07-15 19:03:01 -07:00

892 lines
32 KiB
Rust

pub mod fallback;
mod font;
pub mod json;
mod path_builder;
pub mod regex;
mod text_context;
mod to_path;
use convert_case::{Boundary, Converter, pattern};
use core_types::graphene_hash::CacheHash;
use core_types::list::{Item, List};
use core_types::registry::types::{SignedInteger, TextArea};
use core_types::{CloneVarArgs, Context, Ctx, ExtractAll, ExtractVarArgs, OwnedContextImpl};
use dyn_any::DynAny;
use glam::{DAffine2, DVec2};
use unicode_segmentation::UnicodeSegmentation;
// Re-export for convenience
pub use core_types as gcore;
pub use fallback::FALLBACK_FONT_RESOURCE;
pub use font::*;
pub use text_context::{TextContext, for_each_styled_glyph_run};
pub use to_path::*;
pub use vector_types;
/// Alignment of lines of type within a text block.
#[repr(C)]
#[cfg_attr(feature = "wasm", derive(tsify::Tsify))]
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash, CacheHash, DynAny, node_macro::ChoiceType)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[widget(Radio)]
pub enum TextAlign {
#[default]
#[icon("TextAlignLeft")]
#[cfg_attr(feature = "serde", serde(alias = "Left"))]
AlignLeft,
#[icon("TextAlignCenter")]
#[cfg_attr(feature = "serde", serde(alias = "Center"))]
AlignCenter,
#[icon("TextAlignRight")]
#[cfg_attr(feature = "serde", serde(alias = "Right"))]
AlignRight,
#[icon("TextJustifyLeft")]
JustifyLeft,
#[icon("TextJustifyCenter")]
JustifyCenter,
#[icon("TextJustifyRight")]
JustifyRight,
#[icon("TextJustifyAll")]
JustifyAll,
}
impl From<TextAlign> for parley::Alignment {
fn from(val: TextAlign) -> Self {
match val {
TextAlign::AlignLeft => parley::Alignment::Left,
TextAlign::AlignCenter => parley::Alignment::Center,
TextAlign::AlignRight => parley::Alignment::Right,
_ => parley::Alignment::Justify,
}
}
}
impl TextAlign {
/// What `parley::Alignment` to apply as a post-correction to the last line of a paragraph, or `None` if parley's default already handles it.
///
/// `JustifyLeft` returns `None` because parley already left-aligns the last line of a `Justify` layout. The other justify modes need
/// the last line shifted (`Center`/`Right`) or its inter-word spaces redistributed (`Justify` / `JustifyAll`).
pub fn last_line_correction(self) -> Option<parley::Alignment> {
match self {
Self::JustifyCenter => Some(parley::Alignment::Center),
Self::JustifyRight => Some(parley::Alignment::Right),
Self::JustifyAll => Some(parley::Alignment::Justify),
_ => None,
}
}
/// CSS `(text-align, text-align-last)` values approximating this alignment for the `contenteditable` text overlay.
pub fn css(self) -> (&'static str, &'static str) {
match self {
Self::AlignLeft => ("left", "auto"),
Self::AlignCenter => ("center", "auto"),
Self::AlignRight => ("right", "auto"),
Self::JustifyLeft => ("justify", "auto"),
Self::JustifyCenter => ("justify", "center"),
Self::JustifyRight => ("justify", "right"),
Self::JustifyAll => ("justify", "justify"),
}
}
}
#[derive(PartialEq, Clone, Copy, Debug)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct TypesettingConfig {
pub font_size: f64,
pub line_height_ratio: f64,
pub letter_spacing: f64,
pub letter_tilt: f64,
pub max_width: Option<f64>,
pub max_height: Option<f64>,
pub align: TextAlign,
}
impl Default for TypesettingConfig {
fn default() -> Self {
Self {
font_size: 24.,
line_height_ratio: 1.2,
letter_spacing: 0.,
letter_tilt: 0.,
max_width: None,
max_height: None,
align: TextAlign::default(),
}
}
}
/// Converts escape sequence representations (`\n`, `\r`, `\t`, `\0`, `\\`) into their corresponding control characters.
/// Unrecognized escape sequences (e.g. `\x`) are preserved as-is.
fn unescape_string(input: String) -> String {
let mut result = String::with_capacity(input.len());
let mut chars = input.chars();
while let Some(c) = chars.next() {
if c == '\\' {
match chars.next() {
Some('n') => result.push('\n'),
Some('r') => result.push('\r'),
Some('t') => result.push('\t'),
Some('0') => result.push('\0'),
Some('\\') => result.push('\\'),
Some(unrecognized) => result.extend(['\\', unrecognized]),
None => result.push('\\'),
}
} else {
result.push(c);
}
}
result
}
/// Converts control characters (newline, carriage return, tab, null, backslash) back into their escape sequence representations.
fn escape_string(input: String) -> String {
let mut result = String::with_capacity(input.len());
for c in input.chars() {
match c {
'\n' => result.push_str("\\n"),
'\r' => result.push_str("\\r"),
'\t' => result.push_str("\\t"),
'\0' => result.push_str("\\0"),
'\\' => result.push_str("\\\\"),
other => result.push(other),
}
}
result
}
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash, CacheHash, dyn_any::DynAny, node_macro::ChoiceType, serde::Serialize, serde::Deserialize)]
#[widget(Dropdown)]
pub enum StringCapitalization {
/// "on the origin of species" — Converts all letters to lower case.
#[default]
#[label("lower case")]
LowerCase,
/// "ON THE ORIGIN OF SPECIES" — Converts all letters to upper case.
#[label("UPPER CASE")]
UpperCase,
/// "On The Origin Of Species" — Converts the first letter of every word to upper case.
#[label("Capital Case")]
CapitalCase,
/// "On the Origin of Species" — Converts the first letter of significant words to upper case.
#[label("Headline Case")]
HeadlineCase,
/// "On the origin of species" — Converts the first letter of every word to lower case, except the initial word which is made upper case.
#[label("Sentence case")]
SentenceCase,
/// "on The Origin Of Species" — Converts the first letter of every word to upper case, except the initial word which is made lower case.
#[label("camel Case")]
CamelCase,
}
/// Constructs a string value which may be set to any plain text.
#[node_macro::node(category("Value"))]
fn string_value(_: impl Ctx, _primary: (), string: Item<TextArea>) -> Item<String> {
string
}
/// Type-asserts a value to be a string.
#[node_macro::node(category("Debug"))]
fn as_string(_: impl Ctx, value: Item<String>) -> Item<String> {
value
}
/// Joins two strings together.
#[node_macro::node(category("Text"))]
fn string_concatenate(_: impl Ctx, #[implementations(String)] first: Item<String>, second: Item<TextArea>) -> Item<String> {
let mut first = first;
first.element_mut().push_str(second.element());
first
}
/// Replaces all occurrences of "From" with "To" in the input string.
#[node_macro::node(category("Text"))]
fn string_replace(_: impl Ctx, string: Item<String>, from: Item<TextArea>, to: Item<TextArea>) -> Item<String> {
let mut string = string;
let result = string.element().replace(from.element().as_str(), to.element());
*string.element_mut() = result;
string
}
/// Extracts a substring from the input string, starting at "Start" and ending before "End".
///
/// Negative indices count from the end of the string. If the index of "Start" equals or exceeds "End", the result is an empty string.
#[node_macro::node(category("Text"))]
fn string_slice(_: impl Ctx, string: Item<String>, start: Item<SignedInteger>, end: Item<SignedInteger>) -> Item<String> {
let mut string = string;
let (start, end) = (*start.element(), *end.element());
let total_graphemes = string.element().graphemes(true).count();
let start = if start < 0. {
total_graphemes.saturating_sub(start.abs() as usize)
} else {
(start as usize).min(total_graphemes)
};
let end = if end <= 0. {
total_graphemes.saturating_sub(end.abs() as usize)
} else {
(end as usize).min(total_graphemes)
};
let result = if start >= end {
String::new()
} else {
string.element().graphemes(true).skip(start).take(end - start).collect()
};
*string.element_mut() = result;
string
}
/// Clips the string to a maximum character length, optionally appending a suffix (like "…") when truncation occurs. Strings already within the limit are not modified.
#[node_macro::node(category("Text"))]
fn string_truncate(
_: impl Ctx,
/// The string to truncate.
string: Item<String>,
/// The maximum number of characters allowed, including the suffix if one is appended.
#[default(80)]
length: Item<u32>,
/// A suffix appended to indicate truncation occurred, unless empty. Its length counts towards the character budget.
#[default("")]
suffix: Item<String>,
) -> Item<String> {
let mut string = string;
let max_length = *length.element() as usize;
let grapheme_count = string.element().graphemes(true).count();
if grapheme_count <= max_length {
return string;
}
let suffix: String = suffix.element().graphemes(true).take(max_length).collect();
let keep = max_length - suffix.graphemes(true).count();
let mut truncated: String = string.element().graphemes(true).take(keep).collect();
truncated.push_str(&suffix);
*string.element_mut() = truncated;
string
}
/// Formats a number as a string with control over decimal places, decimal separator, and thousands grouping.
#[node_macro::node(category("Text"), properties("format_number_properties"))]
fn format_number(
_: impl Ctx,
/// The number to format as a string.
number: Item<f64>,
/// The amount of digits after the decimal point. The value is rounded to fit. Set to 0 to show only whole numbers.
#[default(2)]
decimal_places: Item<u32>,
/// The character(s) used as the decimal point.
#[default(".")]
decimal_separator: Item<String>,
/// Always show the exact number of decimal places, even if they are trailing zeros.
#[default(true)]
fixed_decimals: Item<bool>,
/// Whether to group digits with a thousands separator.
use_thousands_separator: Item<bool>,
/// The character(s) inserted between digit groups.
#[default(",")]
thousands_separator: Item<String>,
/// Don't group 4-digit numbers with a thousands separator (only start grouping at 10,000 and above).
#[name("Start at 10,000")]
start_at_10000: Item<bool>,
) -> Item<String> {
let (number, attributes) = number.into_parts();
let (decimal_places, fixed_decimals, use_thousands_separator, start_at_10000) =
(*decimal_places.element(), *fixed_decimals.element(), *use_thousands_separator.element(), *start_at_10000.element());
let decimal_separator = decimal_separator.element().clone();
let thousands_separator = thousands_separator.element().clone();
// Find the maximum meaningful decimal precision by detecting where float noise begins.
// This works correctly whether the value originated as f32 or f64, since we find the
// shortest decimal representation that round-trips back to the same f64 value.
let requested_places = decimal_places as usize;
let max_places = {
let whole_digits = if number == 0. { 1 } else { (number.abs().log10().floor() as usize).saturating_add(1) };
let upper_bound = 17_usize.saturating_sub(whole_digits);
let mut meaningful = upper_bound;
for p in 0..=upper_bound {
let s = format!("{number:.p$}");
if s.parse::<f64>() == Ok(number) {
meaningful = p;
break;
}
}
meaningful
};
let places = requested_places.min(max_places);
let formatted = format!("{number:.places$}");
// If the user requested more decimal places than the float can represent, pad with zeros
let extra_zeros = requested_places.saturating_sub(places);
// Split into sign, whole, and decimal parts
let (sign, unsigned) = if let Some(rest) = formatted.strip_prefix('-') { ("-", rest) } else { ("", formatted.as_str()) };
let (whole_string, decimal_string) = match unsigned.split_once('.') {
Some((w, d)) => {
let padded = if extra_zeros > 0 { format!("{d}{:0>width$}", "", width = extra_zeros) } else { d.to_string() };
(w.to_string(), Some(padded))
}
None => (unsigned.to_string(), None),
};
// Apply thousands grouping to the whole number part
let grouped_whole = if use_thousands_separator && !thousands_separator.is_empty() {
let skip = start_at_10000 && whole_string.len() <= 4;
if skip {
whole_string.clone()
} else {
let mut result = String::new();
for (i, ch) in whole_string.chars().rev().enumerate() {
if i > 0 && i % 3 == 0 {
result.push_str(&thousands_separator.chars().rev().collect::<String>());
}
result.push(ch);
}
result.chars().rev().collect()
}
} else {
whole_string
};
// Build the final string
let result = match decimal_string {
None if fixed_decimals && requested_places > 0 => {
let zeros = "0".repeat(requested_places);
format!("{sign}{grouped_whole}{decimal_separator}{zeros}")
}
None => format!("{sign}{grouped_whole}"),
Some(decimal_string) if fixed_decimals => format!("{sign}{grouped_whole}{decimal_separator}{decimal_string}"),
Some(decimal_string) => {
let trimmed = decimal_string.trim_end_matches('0');
if trimmed.is_empty() {
format!("{sign}{grouped_whole}")
} else {
format!("{sign}{grouped_whole}{decimal_separator}{trimmed}")
}
}
};
Item::from_parts(result, attributes)
}
/// Parses a string into a number. Falls back to the chosen value if the string is not a valid number.
#[node_macro::node(category("Text"), name("String to Number"))]
fn string_to_number(
_: impl Ctx,
/// The string containing a number. Surrounding whitespace is ignored, a decimal point (.) may be included, sign prefixes (+/-) are respected, and scientific notation (e.g. "1e-3") is supported.
string: Item<String>,
/// The value of the result if the string cannot be parsed as a valid number.
fallback: Item<f64>,
) -> Item<f64> {
let (string, attributes) = string.into_parts();
Item::from_parts(string.trim().parse::<f64>().unwrap_or(*fallback.element()), attributes)
}
/// Removes leading and/or trailing whitespace from a string. Common whitespace characters include spaces, tabs, and newlines.
#[node_macro::node(category("Text"))]
fn string_trim(
_: impl Ctx,
/// The string that may contain leading and trailing whitespace that should be removed.
string: Item<String>,
/// Whether the start of the string should have its whitespace removed.
#[default(true)]
start: Item<bool>,
/// Whether the end of the string should have its whitespace removed.
#[default(true)]
end: Item<bool>,
) -> Item<String> {
let mut string = string;
let (start, end) = (*start.element(), *end.element());
let result = match (start, end) {
(true, true) => string.element().trim().to_string(),
(true, false) => string.element().trim_start().to_string(),
(false, true) => string.element().trim_end().to_string(),
(false, false) => return string,
};
*string.element_mut() = result;
string
}
/// Converts between literal escape sequences and their corresponding control characters within a string.
///
/// Unescape: `\n` (newline), `\r` (carriage return), `\t` (tab), `\0` (null), and `\\` (backslash) are converted into the actual special characters.
/// Escape: the actual special characters are converted back into their escape sequence representations.
#[node_macro::node(category("Text"))]
fn string_escape(
_: impl Ctx,
/// The string that contains either literal escape sequences or control characters to be converted to the opposite representation.
string: Item<String>,
/// Convert the control characters back into their escape sequence representations.
#[default(true)]
unescape: Item<bool>,
) -> Item<String> {
let mut string = string;
let input = std::mem::take(string.element_mut());
let result = if *unescape.element() { unescape_string(input) } else { escape_string(input) };
*string.element_mut() = result;
string
}
/// Reverses the sequence of characters making up the string so it reads back-to-front. ("Backwards text" becomes "txet sdrawkcaB".)
#[node_macro::node(category("Text"))]
fn string_reverse(
_: impl Ctx,
/// The string to be reversed.
string: Item<String>,
) -> Item<String> {
let mut string = string;
let result: String = string.element().graphemes(true).rev().collect();
*string.element_mut() = result;
string
}
/// Repeats the string a given number of times, optionally with a separator between each repetition.
#[node_macro::node(category("Text"))]
fn string_repeat(
_: impl Ctx,
/// The string to be repeated.
string: Item<String>,
/// The number of times the string should appear in the output.
#[default(2)]
#[hard(1..)]
count: Item<u32>,
/// The string placed between each repetition.
#[default("\\n")]
separator: Item<String>,
/// Whether to convert escape sequences found in the separator into their corresponding characters:
/// "\n" (newline), "\r" (carriage return), "\t" (tab), "\0" (null), and "\\" (backslash).
#[default(true)]
separator_escaping: Item<bool>,
) -> Item<String> {
let mut string = string;
let separator = separator.element().clone();
let separator = if *separator_escaping.element() { unescape_string(separator) } else { separator };
let count = *count.element() as usize;
let mut result = String::with_capacity((string.element().len() + separator.len()) * count);
for i in 0..count {
if i > 0 {
result.push_str(&separator);
}
result.push_str(string.element());
}
*string.element_mut() = result;
string
}
/// Pads the string to a target length by filling with the given repeated substring. If the string already meets or exceeds the target length, it is returned unchanged.
#[node_macro::node(category("Text"))]
fn string_pad(
_: impl Ctx,
/// The string to be padded to a target length.
string: Item<String>,
/// The target character length after padding. When "Up To" is set, this length concerns only the portion before (or after) that substring.
#[default(10)]
length: Item<u32>,
/// The repeated substring used to fill the remaining space. A multi-charcter substring may end partway through its final repetition.
#[default("#")]
padding: Item<String>,
/// Pad only the length of the string encountered before the start of the first (or after the end of the last) occurrence of this substring, if given and present (otherwise the full string is considered).
///
/// For example, this can pad numbers with leading zeros to align them before the decimal point.
up_to: Item<String>,
/// Pad at the end of the string instead of the start.
from_end: Item<bool>,
) -> Item<String> {
let mut string = string;
let target_length = *length.element() as usize;
let padding = padding.element().clone();
let up_to = up_to.element().clone();
let from_end = *from_end.element();
if padding.is_empty() {
return string;
}
// Split the string at the "up to" substring if provided, and only pad that portion
if !up_to.is_empty()
&& let Some(position) = if from_end { string.element().rfind(&*up_to) } else { string.element().find(&*up_to) }
{
let (before, after) = string.element().split_at(position);
if from_end {
// Pad the portion after the substring
let after_substring = &after[up_to.len()..];
let current_length = after_substring.graphemes(true).count();
if current_length >= target_length {
return string;
}
let pad_length = target_length - current_length;
let padding: String = padding.graphemes(true).cycle().take(pad_length).collect();
let result = format!("{before}{up_to}{after_substring}{padding}");
*string.element_mut() = result;
return string;
} else {
// Pad the portion before the substring
let current_length = before.graphemes(true).count();
if current_length >= target_length {
return string;
}
let pad_length = target_length - current_length;
let padding: String = padding.graphemes(true).cycle().take(pad_length).collect();
let result = format!("{padding}{before}{after}");
*string.element_mut() = result;
return string;
}
}
let current_length = string.element().graphemes(true).count();
if current_length >= target_length {
return string;
}
let pad_length = target_length - current_length;
let padding: String = padding.graphemes(true).cycle().take(pad_length).collect();
let result = if from_end { string.element().clone() + &padding } else { padding + string.element() };
*string.element_mut() = result;
string
}
/// Checks whether the string contains the given substring. Optionally restricts the match to only the start and/or end of the string.
#[node_macro::node(category("Text"))]
fn string_contains(
_: impl Ctx,
/// The string to search within.
string: Item<String>,
/// The substring to search for.
substring: Item<String>,
/// Only match if the substring appears at the start of the string.
at_start: Item<bool>,
/// Only match if the substring appears at the end of the string.
at_end: Item<bool>,
) -> Item<bool> {
let (string, attributes) = string.into_parts();
let substring = substring.element().as_str();
let (at_start, at_end) = (*at_start.element(), *at_end.element());
let result = match (at_start, at_end) {
(true, true) => string.starts_with(substring) && string.ends_with(substring),
(true, false) => string.starts_with(substring),
(false, true) => string.ends_with(substring),
(false, false) => string.contains(substring),
};
Item::from_parts(result, attributes)
}
/// Similar to the **String Contains** node, this searches within the input string for the first (or last) occurrence of a substring and returns the index of where that begins, or -1 if not found.
#[node_macro::node(category("Text"))]
fn string_find_index(
_: impl Ctx,
/// The string to search within.
string: Item<String>,
/// The substring to search for.
substring: Item<String>,
/// Find the start index of the last occurrence instead of the first.
from_end: Item<bool>,
) -> Item<f64> {
let (string, attributes) = string.into_parts();
let substring = substring.element().as_str();
let from_end = *from_end.element();
if substring.is_empty() {
let result = if from_end { string.graphemes(true).count() as f64 } else { 0. };
return Item::from_parts(result, attributes);
}
let result = if from_end {
// Search backwards by finding all byte-level matches and taking the last one
string
.rmatch_indices(substring)
.next()
.map_or(-1., |(byte_index, _)| string[..byte_index].graphemes(true).count() as f64)
} else {
string
.match_indices(substring)
.next()
.map_or(-1., |(byte_index, _)| string[..byte_index].graphemes(true).count() as f64)
};
Item::from_parts(result, attributes)
}
/// Counts the number of occurrences of a substring within the string.
#[node_macro::node(category("Text"))]
fn string_occurrences(
_: impl Ctx,
/// The string to search within.
string: Item<String>,
/// The substring to count occurrences of.
substring: Item<String>,
/// Whether to count overlapping occurrences, using the substring as a sliding window.
///
/// For example, "aa" occurs twice in "aaaa" without overlapping but three times with overlapping.
overlapping: Item<bool>,
) -> Item<f64> {
let (string, attributes) = string.into_parts();
let substring = substring.element().as_str();
if substring.is_empty() {
return Item::from_parts(0., attributes);
}
// NON-OVERLAPPING: Simple linear scan.
// O(n), where n = string length
if !*overlapping.element() {
return Item::from_parts(string.matches(substring).count() as f64, attributes);
}
// OVERLAPPING: KMP (Knuth-Morris-Pratt) algorithm.
// O(n + m), where n = string length, m = substring length
let pattern: Vec<char> = substring.chars().collect();
let text: Vec<char> = string.chars().collect();
// Build the KMP failure function:
// For each position in the pattern, the length of the longest proper prefix that is also a suffix.
// This lets us skip ahead on mismatches instead of restarting from scratch.
let mut failure = vec![0_usize; pattern.len()];
let mut k = 0;
for i in 1..pattern.len() {
while k > 0 && pattern[k] != pattern[i] {
k = failure[k - 1];
}
if pattern[k] == pattern[i] {
k += 1;
}
failure[i] = k;
}
// Scan the text, advancing the pattern cursor without ever backtracking in the text
let mut count: usize = 0;
let mut pattern_cursor = 0;
for &text_char in &text {
while pattern_cursor > 0 && pattern[pattern_cursor] != text_char {
pattern_cursor = failure[pattern_cursor - 1];
}
if pattern[pattern_cursor] == text_char {
pattern_cursor += 1;
}
if pattern_cursor == pattern.len() {
count += 1;
// Reset using failure function to allow overlapping matches
pattern_cursor = failure[pattern_cursor - 1];
}
}
Item::from_parts(count as f64, attributes)
}
/// Converts a string's capitalization style to another of the common upper and lower case patterns, optionally joining words with a chosen separator.
#[node_macro::node(category("Text"), properties("string_capitalization_properties"))]
fn string_capitalization(
_: impl Ctx,
/// The string to have its letter capitalization converted.
string: Item<String>,
/// The capitalization style to apply.
capitalization: Item<StringCapitalization>,
/// Whether to split the string into words and reconnect with the chosen joiner. When disabled, the existing word structure separators are preserved.
use_joiner: Item<bool>,
/// The string placed between each word.
joiner: Item<String>,
) -> Item<String> {
let mut string = string;
let capitalization = *capitalization.element();
let use_joiner = *use_joiner.element();
let joiner = joiner.element().clone();
let input = std::mem::take(string.element_mut());
// When the joiner is enabled, apply word-level casing and optionally reconnect words with the selected joiner
let result = if use_joiner {
match capitalization {
// Simple case mappings that preserve the string's existing structure
StringCapitalization::LowerCase => input.to_lowercase(),
StringCapitalization::UpperCase => input.to_uppercase(),
// Word-aware capitalizations that split on word boundaries and rejoin with the joiner
StringCapitalization::CapitalCase => Converter::new().set_boundaries(&Boundary::defaults()).set_pattern(pattern::capital).set_delim(&joiner).convert(&input),
StringCapitalization::HeadlineCase => {
// First split into words with convert_case so word boundaries like "AlphaNumeric" are detected consistently with other modes,
// then apply the titlecase crate for smart capitalization (lowercasing short words like "of", "the", etc.),
// then rejoin with the custom joiner without mangling the capitalization
let spaced = Converter::new().set_boundaries(&Boundary::defaults()).set_pattern(pattern::capital).set_delim(" ").convert(&input);
let headline = titlecase::titlecase(&spaced);
Converter::new().set_boundaries(&[Boundary::SPACE]).set_pattern(pattern::noop).set_delim(&joiner).convert(&headline)
}
StringCapitalization::SentenceCase => Converter::new().set_boundaries(&Boundary::defaults()).set_pattern(pattern::sentence).set_delim(&joiner).convert(&input),
StringCapitalization::CamelCase => Converter::new().set_boundaries(&Boundary::defaults()).set_pattern(pattern::camel).set_delim(&joiner).convert(&input),
}
}
// When the joiner is disabled, apply only character-level casing while preserving the string's existing structure
else {
match capitalization {
StringCapitalization::LowerCase => input.to_lowercase(),
StringCapitalization::UpperCase => input.to_uppercase(),
StringCapitalization::CapitalCase => {
let mut capitalize_next = true;
input.chars().fold(String::with_capacity(input.len()), |mut result, c| {
if c.is_whitespace() || c == '_' || c == '-' {
capitalize_next = true;
result.push(c);
} else if capitalize_next {
capitalize_next = false;
result.extend(c.to_uppercase());
} else {
result.push(c);
}
result
})
}
StringCapitalization::HeadlineCase => titlecase::titlecase(&input),
StringCapitalization::SentenceCase => {
let mut chars = input.chars();
match chars.next() {
Some(first) => first.to_uppercase().to_string() + &chars.as_str().to_lowercase(),
None => String::new(),
}
}
StringCapitalization::CamelCase => {
let mut capitalize_next = false;
input.chars().fold(String::with_capacity(input.len()), |mut result, c| {
if c.is_whitespace() || c == '_' || c == '-' {
capitalize_next = true;
result.push(c);
} else if capitalize_next {
capitalize_next = false;
result.extend(c.to_uppercase());
} else {
result.extend(c.to_lowercase());
}
result
})
}
}
};
*string.element_mut() = result;
string
}
// TODO: Return u32, u64, or usize instead of f64 after #1621 is resolved and has allowed us to implement automatic type conversion in the node graph for nodes with generic type inputs.
// TODO: (Currently automatic type conversion only works for concrete types, via the Graphene preprocessor and not the full Graphene type system.)
/// Counts the number of characters in a string.
#[node_macro::node(category("Text"))]
fn string_length(_: impl Ctx, string: Item<String>) -> Item<f64> {
let (string, attributes) = string.into_parts();
Item::from_parts(string.graphemes(true).count() as f64, attributes)
}
/// Splits a string into a list of substrings based on the specified delimiter. This is the inverse of the **String Join** node.
///
/// For example, splitting "a, b, c" with delimiter ", " produces `["a", "b", "c"]`.
#[node_macro::node(category("Text"))]
fn string_split(
_: impl Ctx,
/// The string to split into substrings.
string: Item<String>,
/// The character(s) that separate the substrings. These are not included in the outputs.
#[default("\\n")]
delimiter: Item<String>,
/// Whether to convert escape sequences found in the delimiter into their corresponding characters:
/// "\n" (newline), "\r" (carriage return), "\t" (tab), "\0" (null), and "\\" (backslash).
#[default(true)]
delimiter_escaping: Item<bool>,
) -> List<String> {
let delimiter = delimiter.element().clone();
let delimiter = if *delimiter_escaping.element() { unescape_string(delimiter) } else { delimiter };
string.element().split(&delimiter).map(str::to_string).map(Item::new_from_element).collect()
}
/// Joins a list of strings together with a separator between each pair. This is the inverse of the **String Split** node.
///
/// For example, joining `["a", "b", "c"]` with separator ", " produces "a, b, c".
#[node_macro::node(category("Text"))]
fn string_join(
_: impl Ctx,
/// The list of strings to join together.
strings: List<String>,
/// The text placed between each pair of strings.
#[default(", ")]
separator: Item<String>,
/// Whether to convert escape sequences found in the separator into their corresponding characters:
/// "\n" (newline), "\r" (carriage return), "\t" (tab), "\0" (null), and "\\" (backslash).
#[default(true)]
separator_escaping: Item<bool>,
) -> Item<String> {
let (separator, separator_escaping) = (separator.into_element(), separator_escaping.into_element());
let separator = if separator_escaping { unescape_string(separator) } else { separator };
let joined = strings.iter_element_values().map(|s| s.as_str()).collect::<Vec<_>>().join(&separator);
Item::new_from_element(joined)
}
/// Iterates over a list of strings, evaluating the mapped operation for each one. Use the **Read String** node to access the current string inside the loop.
#[node_macro::node(category("Text"))]
async fn map_string(
ctx: impl Ctx + CloneVarArgs + ExtractAll,
strings: List<String>,
#[expose]
#[implementations(Context -> Item<String>)]
mapped: impl Node<Context<'static>, Output = Item<String>>,
) -> List<String> {
let mut result = List::new();
for (i, row) in strings.into_iter().enumerate() {
let owned_ctx = OwnedContextImpl::from(ctx.clone());
let owned_ctx = owned_ctx.with_vararg(Box::new(row)).with_index(i);
let mapped_string = mapped.eval(owned_ctx.into_context()).await;
result.push(mapped_string);
}
result
}
/// Reads the current string from within a **Map String** node's loop.
#[node_macro::node(category("Context"))]
fn read_string(ctx: impl Ctx + ExtractVarArgs) -> Item<String> {
let Ok(var_arg) = ctx.vararg(0) else { return Item::new_from_element(String::new()) };
let var_arg = var_arg as &dyn std::any::Any;
var_arg.downcast_ref::<Item<String>>().cloned().unwrap_or_default()
}
/// Converts a value to a JSON string representation.
#[node_macro::node(category("Debug"))]
fn serialize<T: serde::Serialize>(_: impl Ctx, #[implementations(String, bool, f64, u32, u64, DVec2, DAffine2)] value: Item<T>) -> Item<String> {
let (value, attributes) = value.into_parts();
let result = serde_json::to_string(&value).unwrap_or_else(|_| "Serialization Error".to_string());
Item::from_parts(result, attributes)
}