use alloc::{string::String, vec::Vec};
use core::fmt::{self, Write};
use types::{BoundingBox, Matrix};
pub trait PathSink {
fn move_to(&mut self, x: f32, y: f32);
fn line_to(&mut self, x: f32, y: f32);
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32);
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32);
fn close(&mut self);
}
#[derive(Copy, Clone, PartialEq, PartialOrd, Debug)]
pub enum PathElement {
MoveTo { x: f32, y: f32 },
LineTo { x: f32, y: f32 },
QuadTo { cx0: f32, cy0: f32, x: f32, y: f32 },
CurveTo {
cx0: f32,
cy0: f32,
cx1: f32,
cy1: f32,
x: f32,
y: f32,
},
Close,
}
impl PathSink for Vec<PathElement> {
fn move_to(&mut self, x: f32, y: f32) {
self.push(PathElement::MoveTo { x, y })
}
fn line_to(&mut self, x: f32, y: f32) {
self.push(PathElement::LineTo { x, y })
}
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32) {
self.push(PathElement::QuadTo { cx0, cy0, x, y })
}
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32) {
self.push(PathElement::CurveTo {
cx0,
cy0,
cx1,
cy1,
x,
y,
})
}
fn close(&mut self) {
self.push(PathElement::Close)
}
}
pub struct NullSink;
impl PathSink for NullSink {
fn move_to(&mut self, _x: f32, _y: f32) {}
fn line_to(&mut self, _x: f32, _y: f32) {}
fn quad_to(&mut self, _cx0: f32, _cy0: f32, _x: f32, _y: f32) {}
fn curve_to(&mut self, _cx0: f32, _cy0: f32, _cx1: f32, _cy1: f32, _x: f32, _y: f32) {}
fn close(&mut self) {}
}
#[derive(Clone, Default, Debug)]
pub struct SvgSink(String, Option<usize>);
impl SvgSink {
pub fn new() -> Self {
Self::default()
}
pub fn with_precision(precision: usize) -> Self {
Self(String::default(), Some(precision))
}
pub fn clear(&mut self) {
self.0.clear();
}
fn maybe_push_space(&mut self) {
if !self.0.is_empty() {
self.0.push(' ');
}
}
}
impl core::ops::Deref for SvgSink {
type Target = str;
fn deref(&self) -> &Self::Target {
self.0.as_str()
}
}
impl PathSink for SvgSink {
fn move_to(&mut self, x: f32, y: f32) {
self.maybe_push_space();
let _ = if let Some(prec) = self.1 {
write!(self.0, "M{x:.0$},{y:.0$}", prec)
} else {
write!(self.0, "M{x},{y}")
};
}
fn line_to(&mut self, x: f32, y: f32) {
self.maybe_push_space();
let _ = if let Some(prec) = self.1 {
write!(self.0, "L{x:.0$},{y:.0$}", prec)
} else {
write!(self.0, "L{x},{y}")
};
}
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32) {
self.maybe_push_space();
let _ = if let Some(prec) = self.1 {
write!(self.0, "Q{cx0:.0$},{cy0:.0$} {x:.0$},{y:.0$}", prec)
} else {
write!(self.0, "Q{cx0},{cy0} {x},{y}")
};
}
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32) {
self.maybe_push_space();
let _ = if let Some(prec) = self.1 {
write!(
self.0,
"C{cx0:.0$},{cy0:.0$} {cx1:.0$},{cy1:.0$} {x:.0$},{y:.0$}",
prec
)
} else {
write!(self.0, "C{cx0},{cy0} {cx1},{cy1} {x},{y}")
};
}
fn close(&mut self) {
self.maybe_push_space();
self.0.push('Z');
}
}
impl AsRef<str> for SvgSink {
fn as_ref(&self) -> &str {
self.0.as_ref()
}
}
impl From<String> for SvgSink {
fn from(value: String) -> Self {
Self(value, None)
}
}
impl From<SvgSink> for String {
fn from(value: SvgSink) -> Self {
value.0
}
}
impl fmt::Display for SvgSink {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.0)
}
}
#[derive(Clone, Default, Debug)]
pub struct ControlBoundsSink(Option<BoundingBox<f32>>);
impl ControlBoundsSink {
pub fn new() -> Self {
Self(None)
}
pub fn bounding_box(&self) -> Option<BoundingBox<f32>> {
self.0
}
fn update_bounds(&mut self, x: f32, y: f32) {
if let Some(bb) = &mut self.0 {
bb.x_min = bb.x_min.min(x);
bb.y_min = bb.y_min.min(y);
bb.x_max = bb.x_max.max(x);
bb.y_max = bb.y_max.max(y);
} else {
self.0 = Some(BoundingBox {
x_min: x,
y_min: y,
x_max: x,
y_max: y,
});
}
}
}
impl PathSink for ControlBoundsSink {
fn move_to(&mut self, x: f32, y: f32) {
self.update_bounds(x, y);
}
fn line_to(&mut self, x: f32, y: f32) {
self.update_bounds(x, y);
}
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32) {
self.update_bounds(cx0, cy0);
self.update_bounds(x, y);
}
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32) {
self.update_bounds(cx0, cy0);
self.update_bounds(cx1, cy1);
self.update_bounds(x, y);
}
fn close(&mut self) {}
}
#[derive(Clone, Debug)]
pub struct TransformSink<S> {
sink: S,
transform: Matrix<f32>,
}
impl<S> TransformSink<S> {
pub fn new(sink: S, transform: Matrix<f32>) -> Self {
Self { sink, transform }
}
pub fn transform(&self) -> Matrix<f32> {
self.transform
}
pub fn into_inner(self) -> S {
self.sink
}
}
impl<S: PathSink> PathSink for TransformSink<S> {
fn move_to(&mut self, x: f32, y: f32) {
let (x, y) = self.transform.transform(x, y);
self.sink.move_to(x, y);
}
fn line_to(&mut self, x: f32, y: f32) {
let (x, y) = self.transform.transform(x, y);
self.sink.line_to(x, y);
}
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32) {
let (cx0, cy0) = self.transform.transform(cx0, cy0);
let (x, y) = self.transform.transform(x, y);
self.sink.quad_to(cx0, cy0, x, y);
}
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32) {
let (cx0, cy0) = self.transform.transform(cx0, cy0);
let (cx1, cy1) = self.transform.transform(cx1, cy1);
let (x, y) = self.transform.transform(x, y);
self.sink.curve_to(cx0, cy0, cx1, cy1, x, y);
}
fn close(&mut self) {
self.sink.close();
}
}
impl<P: PathSink + ?Sized> PathSink for &mut P {
fn move_to(&mut self, x: f32, y: f32) {
(**self).move_to(x, y)
}
fn line_to(&mut self, x: f32, y: f32) {
(**self).line_to(x, y)
}
fn quad_to(&mut self, cx0: f32, cy0: f32, x: f32, y: f32) {
(**self).quad_to(cx0, cy0, x, y)
}
fn curve_to(&mut self, cx0: f32, cy0: f32, cx1: f32, cy1: f32, x: f32, y: f32) {
(**self).curve_to(cx0, cy0, cx1, cy1, x, y)
}
fn close(&mut self) {
(**self).close()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn transform_sink_moves_every_point() {
let transform = Matrix::from_elements([2.0, 1.0, 0.0, 1.0, 10.0, 20.0]);
let mut path = Vec::new();
let mut sink = TransformSink::new(&mut path, transform);
sink.move_to(1.0, 2.0);
sink.line_to(3.0, 4.0);
sink.quad_to(1.0, 0.0, 0.0, 1.0);
sink.curve_to(1.0, 1.0, 2.0, 2.0, 3.0, 3.0);
sink.close();
assert_eq!(sink.transform(), transform);
assert_eq!(
path,
[
PathElement::MoveTo { x: 12.0, y: 23.0 },
PathElement::LineTo { x: 16.0, y: 27.0 },
PathElement::QuadTo {
cx0: 12.0,
cy0: 21.0,
x: 10.0,
y: 21.0
},
PathElement::CurveTo {
cx0: 12.0,
cy0: 22.0,
cx1: 14.0,
cy1: 24.0,
x: 16.0,
y: 26.0
},
PathElement::Close,
]
);
}
#[test]
fn svg_sink_precision() {
let svg_data = [None, Some(1), Some(4)].map(|prec| {
let mut pen = match prec {
None => SvgSink::new(),
Some(prec) => SvgSink::with_precision(prec),
};
pen.move_to(1.0, 2.45556);
pen.line_to(1.2, 4.0);
pen.quad_to(2.0345, 3.56789, -0.157, -425.07);
pen.curve_to(-37.0010, 4.5, 2.0, 1.0, -0.5, -0.25);
pen.close();
pen.to_string()
});
let expected = [
"M1,2.45556 L1.2,4 Q2.0345,3.56789 -0.157,-425.07 C-37.001,4.5 2,1 -0.5,-0.25 Z",
"M1.0,2.5 L1.2,4.0 Q2.0,3.6 -0.2,-425.1 C-37.0,4.5 2.0,1.0 -0.5,-0.2 Z",
"M1.0000,2.4556 L1.2000,4.0000 Q2.0345,3.5679 -0.1570,-425.0700 C-37.0010,4.5000 2.0000,1.0000 -0.5000,-0.2500 Z"
];
for (result, expected) in svg_data.iter().zip(&expected) {
assert_eq!(result, expected);
}
}
}