|
|
|
|
@@ -1,11 +1,14 @@
|
|
|
|
|
use core_types::bounds::{BoundingBox, RenderBoundingBox};
|
|
|
|
|
use core_types::list::{AttributeValueDyn, Item, List, ListDyn, NodeIdPath};
|
|
|
|
|
use core_types::registry::types::{Angle, SignedInteger};
|
|
|
|
|
use core_types::registry::types::{Angle, SeedValue, SignedInteger};
|
|
|
|
|
use core_types::{ATTR_EDITOR_LAYER_PATH, ATTR_EDITOR_MERGED_LAYERS, ATTR_TRANSFORM, AnyHash, BlendMode, CacheHash, CloneVarArgs, Color, Context, Ctx, ExtractAll, OwnedContextImpl};
|
|
|
|
|
use glam::{DAffine2, DVec2};
|
|
|
|
|
use graphic_types::graphic::{Graphic, IntoGraphicList};
|
|
|
|
|
use graphic_types::{Artboard, Vector};
|
|
|
|
|
use rand::SeedableRng;
|
|
|
|
|
use rand::seq::SliceRandom;
|
|
|
|
|
use raster_types::{CPU, GPU, Raster};
|
|
|
|
|
use std::cmp::Ordering;
|
|
|
|
|
use vector_types::gradient::{GradientSpreadMethod, GradientType};
|
|
|
|
|
use vector_types::{Gradient, GradientStop, ReferencePoint};
|
|
|
|
|
|
|
|
|
|
@@ -45,12 +48,12 @@ pub fn remove_at_index<T: graphic_types::graphic::OmitIndex + Clone + Default>(
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Returns the item at the specified index in a `List`, keeping its attributes.
|
|
|
|
|
/// Returns the item at the specified index in a list, keeping its attributes.
|
|
|
|
|
/// If no value exists at that index, the element type's default is returned.
|
|
|
|
|
#[node_macro::node(category("General"), name("Item at Index"))]
|
|
|
|
|
pub fn item_at_index<T: Clone + Default + Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The `List` of data to take the item from.
|
|
|
|
|
/// The list of data to take the item from.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
@@ -85,6 +88,328 @@ pub fn item_at_index<T: Clone + Default + Send + Sync + 'static>(
|
|
|
|
|
list.clone_item(resolved).unwrap_or_default()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Keeps chosen items from a list (those corresponding to `true` values) and discards the others (those corresponding to `false` values) based on the *Keep Pattern* bool list. A short pattern is repeated over the remainder of the filtered list, allowing a pattern like `[true, false]` to keep every other item starting from the first. An empty pattern keeps all items.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn filter<T: Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list of data to filter.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
List<f32>,
|
|
|
|
|
List<f64>,
|
|
|
|
|
List<u32>,
|
|
|
|
|
List<u64>,
|
|
|
|
|
List<DVec2>,
|
|
|
|
|
List<DAffine2>,
|
|
|
|
|
List<Vector>,
|
|
|
|
|
List<Graphic>,
|
|
|
|
|
List<Raster<CPU>>,
|
|
|
|
|
List<Raster<GPU>>,
|
|
|
|
|
List<Color>,
|
|
|
|
|
List<Gradient>,
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
/// The list of true and false values that determines which corresponding items are kept (`true`) and discarded (`false`). The pattern may repeat if it is shorter than the list of data.
|
|
|
|
|
keep_pattern: List<bool>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
// Tile the keep pattern over the items, so a short pattern repeats from the start
|
|
|
|
|
let pattern = keep_pattern.iter_element_values().as_slice();
|
|
|
|
|
if pattern.is_empty() {
|
|
|
|
|
return list;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
list.into_iter().enumerate().filter_map(|(index, item)| pattern[index % pattern.len()].then_some(item)).collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Reverses the order of the items in a list, so the last item comes first and the first comes last.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn reverse<T: Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list of data to reverse.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
List<f32>,
|
|
|
|
|
List<f64>,
|
|
|
|
|
List<u32>,
|
|
|
|
|
List<u64>,
|
|
|
|
|
List<DVec2>,
|
|
|
|
|
List<DAffine2>,
|
|
|
|
|
List<Vector>,
|
|
|
|
|
List<Graphic>,
|
|
|
|
|
List<Raster<CPU>>,
|
|
|
|
|
List<Raster<GPU>>,
|
|
|
|
|
List<Color>,
|
|
|
|
|
List<Gradient>,
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
list.into_iter().rev().collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Shifts the items in a list by a number of positions. With wrapping, items pushed off one end reappear at the other. Otherwise they are dropped, shortening the list.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn shift<T: Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list of data to shift.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
List<f32>,
|
|
|
|
|
List<f64>,
|
|
|
|
|
List<u32>,
|
|
|
|
|
List<u64>,
|
|
|
|
|
List<DVec2>,
|
|
|
|
|
List<DAffine2>,
|
|
|
|
|
List<Vector>,
|
|
|
|
|
List<Graphic>,
|
|
|
|
|
List<Raster<CPU>>,
|
|
|
|
|
List<Raster<GPU>>,
|
|
|
|
|
List<Color>,
|
|
|
|
|
List<Gradient>,
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
/// How many positions to shift each item. Positive values shift items toward the start of the list, negative toward the end.
|
|
|
|
|
amount: Item<SignedInteger>,
|
|
|
|
|
/// Whether items shifted off one end wrap around to the other. When off, they are dropped and the list gets shorter.
|
|
|
|
|
#[default(true)]
|
|
|
|
|
wrap: Item<bool>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
let amount = amount.into_element() as i64;
|
|
|
|
|
let wrap = wrap.into_element();
|
|
|
|
|
let len = list.len() as i64;
|
|
|
|
|
if len == 0 {
|
|
|
|
|
return list;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let mut items: Vec<Item<T>> = list.into_iter().collect();
|
|
|
|
|
if wrap {
|
|
|
|
|
items.rotate_left((((amount % len) + len) % len) as usize);
|
|
|
|
|
items.into_iter().collect()
|
|
|
|
|
} else if amount >= 0 {
|
|
|
|
|
items.into_iter().skip(amount.min(len) as usize).collect()
|
|
|
|
|
} else {
|
|
|
|
|
items.into_iter().take((len + amount).max(0) as usize).collect()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Randomly reorders the items in a list. The same seed always produces the same ordering.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn shuffle<T: Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list to have its items randomly reordered.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
List<f32>,
|
|
|
|
|
List<f64>,
|
|
|
|
|
List<u32>,
|
|
|
|
|
List<u64>,
|
|
|
|
|
List<DVec2>,
|
|
|
|
|
List<DAffine2>,
|
|
|
|
|
List<Vector>,
|
|
|
|
|
List<Graphic>,
|
|
|
|
|
List<Raster<CPU>>,
|
|
|
|
|
List<Raster<GPU>>,
|
|
|
|
|
List<Color>,
|
|
|
|
|
List<Gradient>,
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
/// Seed to determine the unique variation of the random shuffle ordering. The same seed always produces the same ordering.
|
|
|
|
|
seed: Item<SeedValue>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
let seed = seed.into_element();
|
|
|
|
|
let mut items: Vec<Item<T>> = list.into_iter().collect();
|
|
|
|
|
|
|
|
|
|
let mut rng = rand::rngs::StdRng::seed_from_u64(seed.into());
|
|
|
|
|
items.shuffle(&mut rng);
|
|
|
|
|
|
|
|
|
|
items.into_iter().collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Generates a list of evenly spaced numbers, starting at a value and progressing by a step (which may be positive, negative, or zero) for a given count.
|
|
|
|
|
#[node_macro::node(category("General"), name("Number Sequence"))]
|
|
|
|
|
fn number_sequence(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
_primary: (),
|
|
|
|
|
/// The first number in the sequence.
|
|
|
|
|
start: Item<f64>,
|
|
|
|
|
/// The amount added to reach each successive number.
|
|
|
|
|
#[default(1.)]
|
|
|
|
|
step: Item<f64>,
|
|
|
|
|
/// How many numbers to generate.
|
|
|
|
|
#[default(10)]
|
|
|
|
|
count: Item<u32>,
|
|
|
|
|
) -> List<f64> {
|
|
|
|
|
let (start, step, count) = (*start.element(), *step.element(), count.into_element());
|
|
|
|
|
|
|
|
|
|
(0..count).map(|i| Item::new_from_element(start + step * i as f64)).collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Counts out the index of each item in a list (0, 1, 2, and so on), producing a list of numbers with one for each item.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn list_indices(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list whose items are counted.
|
|
|
|
|
list: ListDyn,
|
|
|
|
|
/// The number that the count begins from for the first item.
|
|
|
|
|
start_index: Item<SignedInteger>,
|
|
|
|
|
) -> List<f64> {
|
|
|
|
|
let start_index = start_index.into_element();
|
|
|
|
|
|
|
|
|
|
(0..list.len()).map(|index| Item::new_from_element(start_index + index as f64)).collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Extracts a portion of a list, starting at "Start" and ending before "End".
|
|
|
|
|
///
|
|
|
|
|
/// Negative indices count from the end of the list. If the index of "Start" equals or exceeds "End", the result is an empty list.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn list_slice<T: Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list of data to take a portion of.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
List<f32>,
|
|
|
|
|
List<f64>,
|
|
|
|
|
List<u32>,
|
|
|
|
|
List<u64>,
|
|
|
|
|
List<DVec2>,
|
|
|
|
|
List<DAffine2>,
|
|
|
|
|
List<Vector>,
|
|
|
|
|
List<Graphic>,
|
|
|
|
|
List<Raster<CPU>>,
|
|
|
|
|
List<Raster<GPU>>,
|
|
|
|
|
List<Color>,
|
|
|
|
|
List<Gradient>,
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
/// The index of the first item in the portion. Negative indices count from the end of the list.
|
|
|
|
|
start: Item<SignedInteger>,
|
|
|
|
|
/// The index the portion ends before, which is not included. Zero or negative indices count from the end of the list.
|
|
|
|
|
end: Item<SignedInteger>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
let (start, end) = (start.into_element(), end.into_element());
|
|
|
|
|
let total_items = list.len();
|
|
|
|
|
|
|
|
|
|
let start = if start < 0. {
|
|
|
|
|
total_items.saturating_sub(start.abs() as usize)
|
|
|
|
|
} else {
|
|
|
|
|
(start as usize).min(total_items)
|
|
|
|
|
};
|
|
|
|
|
let end = if end <= 0. {
|
|
|
|
|
total_items.saturating_sub(end.abs() as usize)
|
|
|
|
|
} else {
|
|
|
|
|
(end as usize).min(total_items)
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if start >= end {
|
|
|
|
|
return List::new();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
list.into_iter().skip(start).take(end - start).collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Pairwise ordering used by the Sort node for element values. Types without a natural
|
|
|
|
|
/// order compare as equal, so the stable sort leaves their items in their original relative positions.
|
|
|
|
|
pub trait ElementOrder {
|
|
|
|
|
fn element_order(&self, _other: &Self) -> Ordering {
|
|
|
|
|
Ordering::Equal
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for String {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for bool {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for f32 {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.total_cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for f64 {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.total_cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for u32 {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for u64 {
|
|
|
|
|
fn element_order(&self, other: &Self) -> Ordering {
|
|
|
|
|
self.cmp(other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
impl ElementOrder for DVec2 {}
|
|
|
|
|
impl ElementOrder for DAffine2 {}
|
|
|
|
|
impl ElementOrder for Vector {}
|
|
|
|
|
impl ElementOrder for Graphic {}
|
|
|
|
|
impl ElementOrder for Raster<CPU> {}
|
|
|
|
|
impl ElementOrder for Raster<GPU> {}
|
|
|
|
|
impl ElementOrder for Color {}
|
|
|
|
|
impl ElementOrder for Gradient {}
|
|
|
|
|
impl ElementOrder for Artboard {}
|
|
|
|
|
|
|
|
|
|
/// Reorders a list's items from smallest to largest, either by each item's own value or by a parallel list of sortable values in the *Sort Order* input. The sort is stable, so items with the same sort order retain their relative positions.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
fn sort<T: ElementOrder + Clone + Send + Sync + 'static, U: ElementOrder + Send + Sync + 'static>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The list of data to reorder.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>, List<bool>, List<f32>, List<f64>, List<u32>, List<u64>, List<DVec2>, List<DAffine2>, List<Vector>, List<Graphic>, List<Raster<CPU>>, List<Raster<GPU>>, List<Color>, List<Gradient>, List<Artboard>,
|
|
|
|
|
List<String>, List<bool>, List<f32>, List<f64>, List<u32>, List<u64>, List<DVec2>, List<DAffine2>, List<Vector>, List<Graphic>, List<Raster<CPU>>, List<Raster<GPU>>, List<Color>, List<Gradient>, List<Artboard>,
|
|
|
|
|
List<String>, List<bool>, List<f32>, List<f64>, List<u32>, List<u64>, List<DVec2>, List<DAffine2>, List<Vector>, List<Graphic>, List<Raster<CPU>>, List<Raster<GPU>>, List<Color>, List<Gradient>, List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
list: List<T>,
|
|
|
|
|
/// The optional list of orderable values, corresponding item-to-item with the input list, to sort by instead of the items' own values.
|
|
|
|
|
#[expose]
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>, List<f64>,
|
|
|
|
|
List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>, List<String>,
|
|
|
|
|
List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>, List<bool>,
|
|
|
|
|
)]
|
|
|
|
|
sort_order: List<U>,
|
|
|
|
|
/// Reverses the sorted list order, following descending order instead of ascending (numbers largest-to-smallest, strings Z-to-A, etc.).
|
|
|
|
|
reverse: Item<bool>,
|
|
|
|
|
) -> List<T> {
|
|
|
|
|
let reverse = reverse.into_element();
|
|
|
|
|
|
|
|
|
|
// Order by the parallel keys when provided (repeating the last if there are fewer keys than items), otherwise by the element values themselves
|
|
|
|
|
let keys = sort_order.iter_element_values().as_slice();
|
|
|
|
|
let elements: Vec<&T> = list.iter_element_values().collect();
|
|
|
|
|
|
|
|
|
|
let mut order: Vec<usize> = (0..list.len()).collect();
|
|
|
|
|
order.sort_by(|&a, &b| {
|
|
|
|
|
let ordering = match keys {
|
|
|
|
|
[] => elements[a].element_order(elements[b]),
|
|
|
|
|
keys => keys[a.min(keys.len() - 1)].element_order(&keys[b.min(keys.len() - 1)]),
|
|
|
|
|
};
|
|
|
|
|
if reverse { ordering.reverse() } else { ordering }
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let mut result = List::new();
|
|
|
|
|
for index in order {
|
|
|
|
|
if let Some(item) = list.clone_item(index) {
|
|
|
|
|
result.push(item);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
result
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
async fn map<Item: AnyHash + Send + Sync + CacheHash>(
|
|
|
|
|
ctx: impl Ctx + CloneVarArgs + ExtractAll,
|
|
|
|
|
@@ -215,8 +540,8 @@ pub fn path_of_subgraph(_: impl Ctx, node_path: Item<NodeIdPath>) -> Item<NodeId
|
|
|
|
|
Item::new_from_element(NodeIdPath(node_path.into_iter().take(len.saturating_sub(1)).collect()))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Sets a named attribute on the input `List`, computing one value per item via the value-producing input. That input
|
|
|
|
|
/// is evaluated once per item, with the item's index and the item itself (as a `List` containing only that item,
|
|
|
|
|
/// Sets a named attribute on the input list, computing one value per item via the value-producing input. That input
|
|
|
|
|
/// is evaluated once per item, with the item's index and the item itself (as a list containing only that item,
|
|
|
|
|
/// passed as a vararg) provided via context, so the upstream pipeline can return a different value per item that may
|
|
|
|
|
/// be derived from the item's own data. If the attribute already exists, its values are replaced; if not, it's added.
|
|
|
|
|
/// The value is type-erased into an `Item<AttributeValueDyn>` by the auto-inserted input adapter, so this node only
|
|
|
|
|
@@ -224,7 +549,7 @@ pub fn path_of_subgraph(_: impl Ctx, node_path: Item<NodeIdPath>) -> Item<NodeId
|
|
|
|
|
#[node_macro::node(category("Attributes: Write"))]
|
|
|
|
|
async fn write_attribute<T: AnyHash + Clone + Send + Sync + CacheHash>(
|
|
|
|
|
ctx: impl ExtractAll + CloneVarArgs + Ctx,
|
|
|
|
|
/// The `List` to set the named attribute on (one value per item).
|
|
|
|
|
/// The list to set the named attribute on (one value per item).
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
@@ -473,11 +798,11 @@ fn read_attribute_raster(
|
|
|
|
|
result
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Joins two `List`s of the same type, extending the base `List` with the items from the new `List`.
|
|
|
|
|
/// Joins two lists of the same type, extending the base list with the items from the new list.
|
|
|
|
|
#[node_macro::node(category("General"))]
|
|
|
|
|
pub async fn extend<T: 'n + Send + Clone>(
|
|
|
|
|
_: impl Ctx,
|
|
|
|
|
/// The `List` whose items will appear at the start of the extended `List`.
|
|
|
|
|
/// The list whose items will appear at the start of the extended list.
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
List<bool>,
|
|
|
|
|
@@ -496,7 +821,7 @@ pub async fn extend<T: 'n + Send + Clone>(
|
|
|
|
|
List<Artboard>,
|
|
|
|
|
)]
|
|
|
|
|
base: List<T>,
|
|
|
|
|
/// The `List` whose items will appear at the end of the extended `List`.
|
|
|
|
|
/// The list whose items will appear at the end of the extended list.
|
|
|
|
|
#[expose]
|
|
|
|
|
#[implementations(
|
|
|
|
|
List<String>,
|
|
|
|
|
@@ -724,3 +1049,128 @@ fn colors_to_gradient<T: IntoGraphicList>(_: impl Ctx, #[implementations(List<Gr
|
|
|
|
|
});
|
|
|
|
|
Item::new_from_element(Gradient::new(colors))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
|
mod test {
|
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
|
|
fn list_of<T>(elements: impl IntoIterator<Item = T>) -> List<T> {
|
|
|
|
|
elements.into_iter().map(Item::new_from_element).collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn elements<T: Clone>(list: &List<T>) -> Vec<T> {
|
|
|
|
|
list.iter_element_values().cloned().collect()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn sorts_elements_by_their_natural_order() {
|
|
|
|
|
let list = list_of(["banana".to_string(), "apple".to_string(), "cherry".to_string()]);
|
|
|
|
|
let sorted = sort((), list, List::<f64>::new(), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), ["apple", "banana", "cherry"]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn sorts_elements_in_reverse() {
|
|
|
|
|
let list = list_of([3., 1., 2.]);
|
|
|
|
|
let sorted = sort((), list, List::<f64>::new(), Item::new_from_element(true));
|
|
|
|
|
assert_eq!(elements(&sorted), [3., 2., 1.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn sort_order_keys_override_element_order() {
|
|
|
|
|
let list = list_of(["apple".to_string(), "banana".to_string(), "cherry".to_string()]);
|
|
|
|
|
let sorted = sort((), list, list_of([2., 0., 1.]), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), ["banana", "cherry", "apple"]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn short_sort_order_repeats_its_last_key() {
|
|
|
|
|
let list = list_of(["a".to_string(), "b".to_string(), "c".to_string()]);
|
|
|
|
|
let sorted = sort((), list, list_of([2., 1.]), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), ["b", "c", "a"]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn long_sort_order_ignores_its_extra_keys() {
|
|
|
|
|
let list = list_of([1., 2.]);
|
|
|
|
|
let sorted = sort((), list, list_of([3., 1., 0., 5.]), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), [2., 1.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn text_sort_order_keys_order_items_alphabetically() {
|
|
|
|
|
let list = list_of([1., 2., 3.]);
|
|
|
|
|
let sorted = sort((), list, list_of(["c".to_string(), "a".to_string(), "b".to_string()]), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), [2., 3., 1.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn unsortable_elements_keep_their_original_order() {
|
|
|
|
|
let list = list_of([DVec2::new(3., 3.), DVec2::new(1., 1.), DVec2::new(2., 2.)]);
|
|
|
|
|
let sorted = sort((), list, List::<f64>::new(), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&sorted), [DVec2::new(3., 3.), DVec2::new(1., 1.), DVec2::new(2., 2.)]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn shift_wraps_items_around() {
|
|
|
|
|
let forward = shift((), list_of([1., 2., 3., 4.]), Item::new_from_element(1.), Item::new_from_element(true));
|
|
|
|
|
assert_eq!(elements(&forward), [2., 3., 4., 1.]);
|
|
|
|
|
|
|
|
|
|
let backward = shift((), list_of([1., 2., 3., 4.]), Item::new_from_element(-1.), Item::new_from_element(true));
|
|
|
|
|
assert_eq!(elements(&backward), [4., 1., 2., 3.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn shift_without_wrapping_drops_items() {
|
|
|
|
|
let dropped_front = shift((), list_of([1., 2., 3., 4.]), Item::new_from_element(1.), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&dropped_front), [2., 3., 4.]);
|
|
|
|
|
|
|
|
|
|
let dropped_back = shift((), list_of([1., 2., 3., 4.]), Item::new_from_element(-1.), Item::new_from_element(false));
|
|
|
|
|
assert_eq!(elements(&dropped_back), [1., 2., 3.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn shuffle_is_deterministic_and_preserves_elements() {
|
|
|
|
|
let original = [1., 2., 3., 4., 5., 6., 7., 8.];
|
|
|
|
|
let first = shuffle((), list_of(original), Item::new_from_element(42_u32));
|
|
|
|
|
let second = shuffle((), list_of(original), Item::new_from_element(42_u32));
|
|
|
|
|
assert_eq!(elements(&first), elements(&second), "the same seed should always produce the same ordering");
|
|
|
|
|
|
|
|
|
|
let mut recovered = elements(&first);
|
|
|
|
|
recovered.sort_by(|a, b| a.partial_cmp(b).unwrap());
|
|
|
|
|
assert_eq!(recovered, original, "shuffling should preserve all the elements");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn number_sequence_generates_evenly_spaced_numbers() {
|
|
|
|
|
let sequence = number_sequence((), (), Item::new_from_element(0.), Item::new_from_element(2.), Item::new_from_element(4_u32));
|
|
|
|
|
assert_eq!(elements(&sequence), [0., 2., 4., 6.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn list_indices_counts_each_item() {
|
|
|
|
|
let indices = list_indices((), ListDyn::from(list_of(["a".to_string(), "b".to_string(), "c".to_string()])), Item::new_from_element(0.));
|
|
|
|
|
assert_eq!(elements(&indices), [0., 1., 2.]);
|
|
|
|
|
|
|
|
|
|
let from_one = list_indices((), ListDyn::from(list_of(["a".to_string(), "b".to_string(), "c".to_string()])), Item::new_from_element(1.));
|
|
|
|
|
assert_eq!(elements(&from_one), [1., 2., 3.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn list_slice_takes_the_portion_between_start_and_end() {
|
|
|
|
|
let portion = list_slice((), list_of([1., 2., 3., 4., 5.]), Item::new_from_element(1.), Item::new_from_element(3.));
|
|
|
|
|
assert_eq!(elements(&portion), [2., 3.]);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn list_slice_resolves_negative_indices_from_the_end() {
|
|
|
|
|
let portion = list_slice((), list_of([1., 2., 3., 4., 5.]), Item::new_from_element(-2.), Item::new_from_element(0.));
|
|
|
|
|
assert_eq!(elements(&portion), [4., 5.], "an end of zero reaches through the end of the list");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
|
fn list_slice_yields_nothing_when_start_reaches_end() {
|
|
|
|
|
let portion = list_slice((), list_of([1., 2., 3., 4., 5.]), Item::new_from_element(3.), Item::new_from_element(3.));
|
|
|
|
|
assert!(elements(&portion).is_empty());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|