520 lines
15 KiB
Rust
520 lines
15 KiB
Rust
use crate::blending::AlphaBlending;
|
|
use crate::bounds::{BoundingBox, RenderBoundingBox};
|
|
use crate::raster_types::{CPU, GPU, Raster};
|
|
use crate::table::{Table, TableRow};
|
|
use crate::uuid::NodeId;
|
|
use crate::vector::Vector;
|
|
use crate::{Artboard, Color, Ctx};
|
|
use dyn_any::DynAny;
|
|
use glam::{DAffine2, DVec2};
|
|
use std::hash::Hash;
|
|
|
|
/// The possible forms of graphical content that can be rendered by the Render node into either an image or SVG syntax.
|
|
#[derive(Clone, Debug, Hash, PartialEq, DynAny, serde::Serialize, serde::Deserialize)]
|
|
pub enum Graphic {
|
|
Graphic(Table<Graphic>),
|
|
Vector(Table<Vector>),
|
|
RasterCPU(Table<Raster<CPU>>),
|
|
RasterGPU(Table<Raster<GPU>>),
|
|
Color(Table<Color>),
|
|
}
|
|
|
|
impl Default for Graphic {
|
|
fn default() -> Self {
|
|
Self::Graphic(Table::new())
|
|
}
|
|
}
|
|
|
|
// Graphic
|
|
impl From<Table<Graphic>> for Graphic {
|
|
fn from(graphic: Table<Graphic>) -> Self {
|
|
Graphic::Graphic(graphic)
|
|
}
|
|
}
|
|
|
|
// Vector
|
|
impl From<Vector> for Graphic {
|
|
fn from(vector: Vector) -> Self {
|
|
Graphic::Vector(Table::new_from_element(vector))
|
|
}
|
|
}
|
|
impl From<Table<Vector>> for Graphic {
|
|
fn from(vector: Table<Vector>) -> Self {
|
|
Graphic::Vector(vector)
|
|
}
|
|
}
|
|
impl From<Vector> for Table<Graphic> {
|
|
fn from(vector: Vector) -> Self {
|
|
Table::new_from_element(Graphic::Vector(Table::new_from_element(vector)))
|
|
}
|
|
}
|
|
impl From<Table<Vector>> for Table<Graphic> {
|
|
fn from(vector: Table<Vector>) -> Self {
|
|
Table::new_from_element(Graphic::Vector(vector))
|
|
}
|
|
}
|
|
|
|
// Raster<CPU>
|
|
impl From<Raster<CPU>> for Graphic {
|
|
fn from(raster: Raster<CPU>) -> Self {
|
|
Graphic::RasterCPU(Table::new_from_element(raster))
|
|
}
|
|
}
|
|
impl From<Table<Raster<CPU>>> for Graphic {
|
|
fn from(raster: Table<Raster<CPU>>) -> Self {
|
|
Graphic::RasterCPU(raster)
|
|
}
|
|
}
|
|
impl From<Raster<CPU>> for Table<Graphic> {
|
|
fn from(raster: Raster<CPU>) -> Self {
|
|
Table::new_from_element(Graphic::RasterCPU(Table::new_from_element(raster)))
|
|
}
|
|
}
|
|
impl From<Table<Raster<CPU>>> for Table<Graphic> {
|
|
fn from(raster: Table<Raster<CPU>>) -> Self {
|
|
Table::new_from_element(Graphic::RasterCPU(raster))
|
|
}
|
|
}
|
|
|
|
// Raster<GPU>
|
|
impl From<Raster<GPU>> for Graphic {
|
|
fn from(raster: Raster<GPU>) -> Self {
|
|
Graphic::RasterGPU(Table::new_from_element(raster))
|
|
}
|
|
}
|
|
impl From<Table<Raster<GPU>>> for Graphic {
|
|
fn from(raster: Table<Raster<GPU>>) -> Self {
|
|
Graphic::RasterGPU(raster)
|
|
}
|
|
}
|
|
impl From<Raster<GPU>> for Table<Graphic> {
|
|
fn from(raster: Raster<GPU>) -> Self {
|
|
Table::new_from_element(Graphic::RasterGPU(Table::new_from_element(raster)))
|
|
}
|
|
}
|
|
impl From<Table<Raster<GPU>>> for Table<Graphic> {
|
|
fn from(raster: Table<Raster<GPU>>) -> Self {
|
|
Table::new_from_element(Graphic::RasterGPU(raster))
|
|
}
|
|
}
|
|
|
|
// Color
|
|
impl From<Color> for Graphic {
|
|
fn from(color: Color) -> Self {
|
|
Graphic::Color(Table::new_from_element(color))
|
|
}
|
|
}
|
|
impl From<Table<Color>> for Graphic {
|
|
fn from(color: Table<Color>) -> Self {
|
|
Graphic::Color(color)
|
|
}
|
|
}
|
|
impl From<Color> for Table<Graphic> {
|
|
fn from(color: Color) -> Self {
|
|
Table::new_from_element(Graphic::Color(Table::new_from_element(color)))
|
|
}
|
|
}
|
|
impl From<Table<Color>> for Table<Graphic> {
|
|
fn from(color: Table<Color>) -> Self {
|
|
Table::new_from_element(Graphic::Color(color))
|
|
}
|
|
}
|
|
|
|
// Option<Color>
|
|
impl From<Option<Color>> for Graphic {
|
|
fn from(color: Option<Color>) -> Self {
|
|
if let Some(color) = color {
|
|
Graphic::Color(Table::new_from_element(color))
|
|
} else {
|
|
Graphic::default()
|
|
}
|
|
}
|
|
}
|
|
impl From<Option<Color>> for Table<Graphic> {
|
|
fn from(color: Option<Color>) -> Self {
|
|
if let Some(color) = color {
|
|
Table::new_from_element(Graphic::Color(Table::new_from_element(color)))
|
|
} else {
|
|
Table::new()
|
|
}
|
|
}
|
|
}
|
|
|
|
// DAffine2
|
|
impl From<DAffine2> for Graphic {
|
|
fn from(_: DAffine2) -> Self {
|
|
Graphic::default()
|
|
}
|
|
}
|
|
impl From<DAffine2> for Table<Graphic> {
|
|
fn from(_: DAffine2) -> Self {
|
|
Table::new()
|
|
}
|
|
}
|
|
|
|
impl Graphic {
|
|
pub fn as_graphic(&self) -> Option<&Table<Graphic>> {
|
|
match self {
|
|
Graphic::Graphic(graphic) => Some(graphic),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_graphic_mut(&mut self) -> Option<&mut Table<Graphic>> {
|
|
match self {
|
|
Graphic::Graphic(graphic) => Some(graphic),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_vector(&self) -> Option<&Table<Vector>> {
|
|
match self {
|
|
Graphic::Vector(vector) => Some(vector),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_vector_mut(&mut self) -> Option<&mut Table<Vector>> {
|
|
match self {
|
|
Graphic::Vector(vector) => Some(vector),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_raster(&self) -> Option<&Table<Raster<CPU>>> {
|
|
match self {
|
|
Graphic::RasterCPU(raster) => Some(raster),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn as_raster_mut(&mut self) -> Option<&mut Table<Raster<CPU>>> {
|
|
match self {
|
|
Graphic::RasterCPU(raster) => Some(raster),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
pub fn had_clip_enabled(&self) -> bool {
|
|
match self {
|
|
Graphic::Vector(vector) => vector.iter().all(|row| row.alpha_blending.clip),
|
|
Graphic::Graphic(graphic) => graphic.iter().all(|row| row.alpha_blending.clip),
|
|
Graphic::RasterCPU(raster) => raster.iter().all(|row| row.alpha_blending.clip),
|
|
Graphic::RasterGPU(raster) => raster.iter().all(|row| row.alpha_blending.clip),
|
|
Graphic::Color(color) => color.iter().all(|row| row.alpha_blending.clip),
|
|
}
|
|
}
|
|
|
|
pub fn can_reduce_to_clip_path(&self) -> bool {
|
|
match self {
|
|
Graphic::Vector(vector) => vector.iter().all(|row| {
|
|
let style = &row.element.style;
|
|
let alpha_blending = &row.alpha_blending;
|
|
(alpha_blending.opacity > 1. - f32::EPSILON) && style.fill().is_opaque() && style.stroke().is_none_or(|stroke| !stroke.has_renderable_stroke())
|
|
}),
|
|
_ => false,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl BoundingBox for Graphic {
|
|
fn bounding_box(&self, transform: DAffine2, include_stroke: bool) -> RenderBoundingBox {
|
|
match self {
|
|
Graphic::Vector(vector) => vector.bounding_box(transform, include_stroke),
|
|
Graphic::RasterCPU(raster) => raster.bounding_box(transform, include_stroke),
|
|
Graphic::RasterGPU(raster) => raster.bounding_box(transform, include_stroke),
|
|
Graphic::Graphic(graphic) => graphic.bounding_box(transform, include_stroke),
|
|
Graphic::Color(color) => color.bounding_box(transform, include_stroke),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[node_macro::node(category(""))]
|
|
async fn source_node_id<I: 'n + Send + Clone>(
|
|
_: impl Ctx,
|
|
#[implementations(Table<Artboard>, Table<Graphic>, Table<Vector>, Table<Raster<CPU>>, Table<Raster<GPU>>, Table<Color>)] content: Table<I>,
|
|
node_path: Vec<NodeId>,
|
|
) -> Table<I> {
|
|
// Get the penultimate element of the node path, or None if the path is too short
|
|
// This is used to get the ID of the user-facing parent layer-style node (which encapsulates this internal node).
|
|
let source_node_id = node_path.get(node_path.len().wrapping_sub(2)).copied();
|
|
|
|
let mut content = content;
|
|
for row in content.iter_mut() {
|
|
*row.source_node_id = source_node_id;
|
|
}
|
|
|
|
content
|
|
}
|
|
|
|
/// Joins two tables of the same type, extending the base table with the rows of the new table.
|
|
#[node_macro::node(category("General"))]
|
|
async fn extend<I: 'n + Send + Clone>(
|
|
_: impl Ctx,
|
|
/// The table whose rows will appear at the start of the extended table.
|
|
#[implementations(Table<Artboard>, Table<Graphic>, Table<Vector>, Table<Raster<CPU>>, Table<Raster<GPU>>, Table<Color>)]
|
|
base: Table<I>,
|
|
/// The table whose rows will appear at the end of the extended table.
|
|
#[expose]
|
|
#[implementations(Table<Artboard>, Table<Graphic>, Table<Vector>, Table<Raster<CPU>>, Table<Raster<GPU>>, Table<Color>)]
|
|
new: Table<I>,
|
|
) -> Table<I> {
|
|
let mut base = base;
|
|
base.extend(new);
|
|
|
|
base
|
|
}
|
|
|
|
// TODO: Eventually remove this document upgrade code
|
|
#[node_macro::node(category(""))]
|
|
async fn legacy_layer_extend<I: 'n + Send + Clone>(
|
|
_: impl Ctx,
|
|
#[implementations(Table<Artboard>, Table<Graphic>, Table<Vector>, Table<Raster<CPU>>, Table<Raster<GPU>>, Table<Color>)] base: Table<I>,
|
|
#[expose]
|
|
#[implementations(Table<Artboard>, Table<Graphic>, Table<Vector>, Table<Raster<CPU>>, Table<Raster<GPU>>, Table<Color>)]
|
|
new: Table<I>,
|
|
nested_node_path: Vec<NodeId>,
|
|
) -> Table<I> {
|
|
// Get the penultimate element of the node path, or None if the path is too short
|
|
// This is used to get the ID of the user-facing parent layer-style node (which encapsulates this internal node).
|
|
let source_node_id = nested_node_path.get(nested_node_path.len().wrapping_sub(2)).copied();
|
|
|
|
let mut base = base;
|
|
for row in new.into_iter() {
|
|
base.push(TableRow { source_node_id, ..row });
|
|
}
|
|
|
|
base
|
|
}
|
|
|
|
/// Places a table of graphical content into an element of a new wrapper graphic table.
|
|
#[node_macro::node(category("General"))]
|
|
async fn wrap_graphic<T: Into<Graphic> + 'n>(
|
|
_: impl Ctx,
|
|
#[implementations(
|
|
Table<Graphic>,
|
|
Table<Vector>,
|
|
Table<Raster<CPU>>,
|
|
Table<Raster<GPU>>,
|
|
Table<Color>,
|
|
Color,
|
|
Option<Color>,
|
|
DAffine2,
|
|
)]
|
|
content: T,
|
|
) -> Table<Graphic> {
|
|
Table::new_from_element(content.into())
|
|
}
|
|
|
|
/// Converts a table of graphical content into a graphic table by placing it into an element of a new wrapper graphic table.
|
|
/// If it is already a graphic table, it is not wrapped again. Use the 'Wrap Graphic' node if wrapping is always desired.
|
|
#[node_macro::node(category("Type Conversion"))]
|
|
async fn to_graphic<T: Into<Table<Graphic>> + 'n>(
|
|
_: impl Ctx,
|
|
#[implementations(
|
|
Table<Graphic>,
|
|
Table<Vector>,
|
|
Table<Raster<CPU>>,
|
|
Table<Raster<GPU>>,
|
|
Table<Color>,
|
|
Color,
|
|
Option<Color>,
|
|
)]
|
|
content: T,
|
|
) -> Table<Graphic> {
|
|
content.into()
|
|
}
|
|
|
|
#[node_macro::node(category("General"))]
|
|
async fn flatten_graphic(_: impl Ctx, content: Table<Graphic>, fully_flatten: bool) -> Table<Graphic> {
|
|
// TODO: Avoid mutable reference, instead return a new Table<Graphic>?
|
|
fn flatten_table(output_graphic_table: &mut Table<Graphic>, current_graphic_table: Table<Graphic>, fully_flatten: bool, recursion_depth: usize) {
|
|
for current_row in current_graphic_table.iter() {
|
|
let current_element = current_row.element.clone();
|
|
let reference = *current_row.source_node_id;
|
|
|
|
let recurse = fully_flatten || recursion_depth == 0;
|
|
|
|
match current_element {
|
|
// If we're allowed to recurse, flatten any graphics we encounter
|
|
Graphic::Graphic(mut current_element) if recurse => {
|
|
// Apply the parent graphic's transform to all child elements
|
|
for graphic in current_element.iter_mut() {
|
|
*graphic.transform = *current_row.transform * *graphic.transform;
|
|
}
|
|
|
|
flatten_table(output_graphic_table, current_element, fully_flatten, recursion_depth + 1);
|
|
}
|
|
// Push any leaf Graphic elements we encounter, which can be either Graphic table elements beyond the recursion depth, or table elements other than Graphic tables
|
|
_ => {
|
|
output_graphic_table.push(TableRow {
|
|
element: current_element,
|
|
transform: *current_row.transform,
|
|
alpha_blending: *current_row.alpha_blending,
|
|
source_node_id: reference,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut output = Table::new();
|
|
flatten_table(&mut output, content, fully_flatten, 0);
|
|
|
|
output
|
|
}
|
|
|
|
#[node_macro::node(category("Vector"))]
|
|
async fn flatten_vector(_: impl Ctx, content: Table<Graphic>) -> Table<Vector> {
|
|
// TODO: Avoid mutable reference, instead return a new Table<Graphic>?
|
|
fn flatten_table(output_vector_table: &mut Table<Vector>, current_graphic_table: Table<Graphic>) {
|
|
for current_graphic_row in current_graphic_table.iter() {
|
|
let current_graphic = current_graphic_row.element.clone();
|
|
let source_node_id = *current_graphic_row.source_node_id;
|
|
|
|
match current_graphic {
|
|
// If we're allowed to recurse, flatten any tables we encounter
|
|
Graphic::Graphic(mut current_graphic_table) => {
|
|
// Apply the parent graphic's transform to all child elements
|
|
for graphic in current_graphic_table.iter_mut() {
|
|
*graphic.transform = *current_graphic_row.transform * *graphic.transform;
|
|
}
|
|
|
|
flatten_table(output_vector_table, current_graphic_table);
|
|
}
|
|
// Push any leaf Vector elements we encounter
|
|
Graphic::Vector(vector_table) => {
|
|
for current_vector_row in vector_table.iter() {
|
|
output_vector_table.push(TableRow {
|
|
element: current_vector_row.element.clone(),
|
|
transform: *current_graphic_row.transform * *current_vector_row.transform,
|
|
alpha_blending: AlphaBlending {
|
|
blend_mode: current_vector_row.alpha_blending.blend_mode,
|
|
opacity: current_graphic_row.alpha_blending.opacity * current_vector_row.alpha_blending.opacity,
|
|
fill: current_vector_row.alpha_blending.fill,
|
|
clip: current_vector_row.alpha_blending.clip,
|
|
},
|
|
source_node_id,
|
|
});
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut output = Table::new();
|
|
flatten_table(&mut output, content);
|
|
|
|
output
|
|
}
|
|
|
|
/// Returns the value at the specified index in the collection.
|
|
/// If that index has no value, the type's default value is returned.
|
|
#[node_macro::node(category("General"))]
|
|
fn index<T: AtIndex + Clone + Default>(
|
|
_: impl Ctx,
|
|
/// The collection of data, such as a list or table.
|
|
#[implementations(
|
|
Vec<Color>,
|
|
Vec<Option<Color>>,
|
|
Vec<f64>, Vec<u64>,
|
|
Vec<DVec2>,
|
|
Table<Vector>,
|
|
Table<Raster<CPU>>,
|
|
Table<Graphic>,
|
|
)]
|
|
collection: T,
|
|
/// The index of the item to retrieve, starting from 0 for the first item.
|
|
index: u32,
|
|
) -> T::Output
|
|
where
|
|
T::Output: Clone + Default,
|
|
{
|
|
collection.at_index(index as usize).unwrap_or_default()
|
|
}
|
|
|
|
pub trait AtIndex {
|
|
type Output;
|
|
fn at_index(&self, index: usize) -> Option<Self::Output>;
|
|
}
|
|
impl<T: Clone> AtIndex for Vec<T> {
|
|
type Output = T;
|
|
|
|
fn at_index(&self, index: usize) -> Option<Self::Output> {
|
|
self.get(index).cloned()
|
|
}
|
|
}
|
|
impl<T: Clone> AtIndex for Table<T> {
|
|
type Output = Table<T>;
|
|
|
|
fn at_index(&self, index: usize) -> Option<Self::Output> {
|
|
let mut result_table = Self::default();
|
|
if let Some(row) = self.iter().nth(index) {
|
|
result_table.push(row.into_cloned());
|
|
Some(result_table)
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO: Eventually remove this migration document upgrade code
|
|
pub fn migrate_graphic<'de, D: serde::Deserializer<'de>>(deserializer: D) -> Result<Table<Graphic>, D::Error> {
|
|
use serde::Deserialize;
|
|
|
|
#[derive(Clone, Debug, PartialEq, DynAny, Default, serde::Serialize, serde::Deserialize)]
|
|
pub struct OldGraphicGroup {
|
|
elements: Vec<(Graphic, Option<NodeId>)>,
|
|
transform: DAffine2,
|
|
alpha_blending: AlphaBlending,
|
|
}
|
|
#[derive(Clone, Debug, PartialEq, DynAny, Default, serde::Serialize, serde::Deserialize)]
|
|
pub struct GraphicGroup {
|
|
elements: Vec<(Graphic, Option<NodeId>)>,
|
|
}
|
|
|
|
#[derive(serde::Serialize, serde::Deserialize)]
|
|
#[serde(untagged)]
|
|
enum EitherFormat {
|
|
OldGraphicGroup(OldGraphicGroup),
|
|
Table(serde_json::Value),
|
|
}
|
|
|
|
Ok(match EitherFormat::deserialize(deserializer)? {
|
|
EitherFormat::OldGraphicGroup(old) => {
|
|
let mut graphic_table = Table::new();
|
|
for (graphic, source_node_id) in old.elements {
|
|
graphic_table.push(TableRow {
|
|
element: graphic,
|
|
transform: old.transform,
|
|
alpha_blending: old.alpha_blending,
|
|
source_node_id,
|
|
});
|
|
}
|
|
graphic_table
|
|
}
|
|
EitherFormat::Table(value) => {
|
|
// Try to deserialize as either table format
|
|
if let Ok(old_table) = serde_json::from_value::<Table<GraphicGroup>>(value.clone()) {
|
|
let mut graphic_table = Table::new();
|
|
for row in old_table.iter() {
|
|
for (graphic, source_node_id) in &row.element.elements {
|
|
graphic_table.push(TableRow {
|
|
element: graphic.clone(),
|
|
transform: *row.transform,
|
|
alpha_blending: *row.alpha_blending,
|
|
source_node_id: *source_node_id,
|
|
});
|
|
}
|
|
}
|
|
graphic_table
|
|
} else if let Ok(new_table) = serde_json::from_value::<Table<Graphic>>(value) {
|
|
new_table
|
|
} else {
|
|
return Err(serde::de::Error::custom("Failed to deserialize Table<Graphic>"));
|
|
}
|
|
}
|
|
})
|
|
}
|