use crate::math; use skia_safe as skia; use uuid::Uuid; use crate::render::{BlendMode, Renderable}; mod bools; mod fills; mod images; mod matrix; mod paths; mod renderable; mod strokes; pub use bools::*; pub use fills::*; pub use images::*; use matrix::*; pub use paths::*; pub use strokes::*; #[derive(Debug, Clone, PartialEq)] pub enum Kind { Rect(math::Rect), Circle(math::Rect), Path(Path), Bool(BoolType, Path), } pub type Color = skia::Color; #[derive(Debug, Clone)] #[allow(dead_code)] pub struct Shape { id: Uuid, children: Vec, kind: Kind, selrect: math::Rect, transform: Matrix, rotation: f32, clip_content: bool, fills: Vec, strokes: Vec, blend_mode: BlendMode, opacity: f32, hidden: bool, } impl Shape { pub fn new(id: Uuid) -> Self { Self { id, children: Vec::::new(), kind: Kind::Rect(math::Rect::new_empty()), selrect: math::Rect::new_empty(), transform: Matrix::identity(), rotation: 0., clip_content: true, fills: vec![], strokes: vec![], blend_mode: BlendMode::default(), opacity: 1., hidden: false, } } pub fn kind(&self) -> Kind { self.kind.clone() } pub fn set_selrect(&mut self, left: f32, top: f32, right: f32, bottom: f32) { self.selrect.set_ltrb(left, top, right, bottom); match self.kind { Kind::Rect(_) => { self.kind = Kind::Rect(self.selrect.to_owned()); } Kind::Circle(_) => { self.kind = Kind::Circle(self.selrect.to_owned()); } _ => {} }; } pub fn set_kind(&mut self, kind: Kind) { self.kind = kind; } pub fn set_clip(&mut self, value: bool) { self.clip_content = value; } pub fn set_rotation(&mut self, angle: f32) { self.rotation = angle; } pub fn set_transform(&mut self, a: f32, b: f32, c: f32, d: f32, e: f32, f: f32) { self.transform = Matrix::new(a, b, c, d, e, f); } pub fn set_opacity(&mut self, opacity: f32) { self.opacity = opacity; } pub fn set_hidden(&mut self, value: bool) { self.hidden = value; } pub fn add_child(&mut self, id: Uuid) { self.children.push(id); } pub fn clear_children(&mut self) { self.children.clear(); } pub fn fills(&self) -> std::slice::Iter { self.fills.iter() } pub fn add_fill(&mut self, f: Fill) { self.fills.push(f) } pub fn clear_fills(&mut self) { self.fills.clear(); } pub fn add_fill_gradient_stops(&mut self, buffer: Vec) -> Result<(), String> { let fill = self.fills.last_mut().ok_or("Shape has no fills")?; let gradient = match fill { Fill::LinearGradient(g) => Ok(g), Fill::RadialGradient(g) => Ok(g), _ => Err("Active fill is not a gradient"), }?; for stop in buffer.into_iter() { gradient.add_stop(stop.color(), stop.offset()); } Ok(()) } pub fn strokes(&self) -> std::slice::Iter { self.strokes.iter() } pub fn add_stroke(&mut self, s: Stroke) { self.strokes.push(s) } pub fn set_stroke_fill(&mut self, f: Fill) -> Result<(), String> { let stroke = self.strokes.last_mut().ok_or("Shape has no strokes")?; stroke.fill = f; Ok(()) } pub fn add_stroke_gradient_stops(&mut self, buffer: Vec) -> Result<(), String> { let stroke = self.strokes.last_mut().ok_or("Shape has no strokes")?; let fill = &mut stroke.fill; let gradient = match fill { Fill::LinearGradient(g) => Ok(g), Fill::RadialGradient(g) => Ok(g), _ => Err("Active stroke is not a gradient"), }?; for stop in buffer.into_iter() { gradient.add_stop(stop.color(), stop.offset()); } Ok(()) } pub fn clear_strokes(&mut self) { self.strokes.clear(); } pub fn set_path_segments(&mut self, buffer: Vec) -> Result<(), String> { let p = Path::try_from(buffer)?; let kind = match &self.kind { Kind::Bool(bool_type, _) => Kind::Bool(*bool_type, p), _ => Kind::Path(p), }; self.kind = kind; Ok(()) } pub fn set_blend_mode(&mut self, mode: BlendMode) { self.blend_mode = mode; } pub fn set_bool_type(&mut self, bool_type: BoolType) { let kind = match &self.kind { Kind::Bool(_, path) => Kind::Bool(bool_type, path.clone()), _ => Kind::Bool(bool_type, Path::default()), }; self.kind = kind; } fn to_path_transform(&self) -> Option { match self.kind { Kind::Path(_) | Kind::Bool(_, _) => { let center = self.bounds().center(); let mut matrix = skia::Matrix::new_identity(); matrix.pre_translate(center); matrix.pre_concat(&self.transform.no_translation().to_skia_matrix().invert()?); matrix.pre_translate(-center); Some(matrix) } _ => None, } } } #[cfg(test)] mod tests { use super::*; fn any_shape() -> Shape { Shape::new(Uuid::nil()) } #[test] fn add_fill_pushes_a_new_fill() { let mut shape = any_shape(); assert_eq!(shape.fills.len(), 0); shape.add_fill(Fill::Solid(Color::TRANSPARENT)); assert_eq!(shape.fills.get(0), Some(&Fill::Solid(Color::TRANSPARENT))) } }