use svg::{Document, Node};
use crate::canvas::Canvas;
use crate::render::Renderer;
use crate::shape::{LinePoint, Shape};
use crate::style::Style;
use crate::{Drawing, Point, RGB};
use svg::node::element::path::Data;
use svg::node::element::{tag, Element};
pub struct SvgRenderer {}
impl SvgRenderer {
pub fn new() -> SvgRenderer {
SvgRenderer {}
}
}
impl Renderer for SvgRenderer {
fn render(&self, canvas: &Canvas) -> Vec<u8> {
let mut document = Document::new().set("viewBox", (0, 0, canvas.width, canvas.height));
if let Some(shape) = &canvas.background {
let origin = Point::new(0.0, 0.0);
document = render_shape(shape, &origin, &Style::default(), document);
}
for drawing in canvas.drawings() {
document = render_drawing(drawing, document)
}
document.to_string().into_bytes()
}
}
fn render_drawing(drawing: &Drawing, mut document: Document) -> Document {
if let Some(shape) = &drawing.shape {
document = render_shape(shape, &drawing.position, &drawing.style, document);
}
for drawing in &drawing.display_list.drawings {
document = render_drawing(drawing, document);
}
document
}
fn render_shape(shape: &Shape, position: &Point, style: &Style, document: Document) -> Document {
let mut element;
match shape {
Shape::Rectangle { width, height } => {
element = rect(*width, *height, &position);
}
Shape::Circle { radius } => {
element = circle(*radius, &position);
}
Shape::Line { start, points } => {
element = line(*start, points);
}
}
if let Some(fill) = &style.fill {
element.assign("fill", rgb_to_str(&fill.color));
} else {
element.assign("fill", "transparent");
}
if let Some(stroke) = &style.stroke {
element.assign("stroke", rgb_to_str(&stroke.color));
element.assign("stroke-width", format!("{}", stroke.width));
}
document.add(element)
}
fn rgb_to_str(color: &RGB) -> String {
format!("rgb({},{},{})", color.r, color.g, color.b)
}
fn rect(width: u32, height: u32, position: &Point) -> Element {
let mut element = Element::new(tag::Rectangle);
element.assign("width", width);
element.assign("height", height);
element.assign("x", position.x);
element.assign("y", position.y);
element
}
fn circle(radius: u32, position: &Point) -> Element {
let mut element = Element::new(tag::Circle);
element.assign("r", radius);
element.assign("cx", position.x);
element.assign("cy", position.y);
element
}
fn line(start: Point, points: &Vec<LinePoint>) -> Element {
let mut data = Data::new().move_to((start.x, start.y));
for point in points {
match *point {
LinePoint::Straight { point } => {
data = data.line_to((point.x, point.y));
}
LinePoint::QuadraticBezierCurve { point, curve } => {
data = data.quadratic_curve_to((curve.x, curve.y, point.x, point.y));
}
_ => unimplemented!(),
}
}
let mut element = Element::new(tag::Path);
element.assign("d", data);
element
}