Add width to line style

Author: milliams

examples/function_svg.rs

@@ -8,7 +8,7 @@ fn main() {
     let f2 = plotlib::function::Function::new(|x| x.powi(2), 0., 10.)
         .style(plotlib::function::Style::new().colour("darkolivegreen"));
     let f3 = plotlib::function::Function::new(|x| x.sqrt() * 20., 0., 10.)
-        .style(plotlib::function::Style::new().colour("brown"));
+        .style(plotlib::function::Style::new().colour("brown").width(1));
     let v = plotlib::view::View::new().add(&f1).add(&f2).add(&f3);
     plotlib::page::Page::single(&v).save("function.svg");
 }

src/function.rs

@@ -25,18 +25,27 @@ use style;
 #[derive(Debug, Default)]
 pub struct Style {
     colour: Option<String>,
+    width: Option<u8>,
 }
 
 impl Style {
     pub fn new() -> Self {
-        Style { colour: None }
+        Style {
+            colour: None,
+            width: None,
+        }
     }
 
     pub fn overlay(&mut self, other: &Self) {
         match other.colour {
             Some(ref v) => self.colour = Some(v.clone()),
             None => {}
         }
+
+        match other.width {
+            Some(ref v) => self.width = Some(v.clone()),
+            None => {}
+        }
     }
 }
 
@@ -52,6 +61,18 @@ impl style::Line for Style {
     fn get_colour(&self) -> &Option<String> {
         &self.colour
     }
+
+    fn width<T>(&mut self, value: T) -> &mut Self
+    where
+        T: Into<u8>,
+    {
+        self.width = Some(value.into());
+        self
+    }
+
+    fn get_width(&self) -> &Option<u8> {
+        &self.width
+    }
 }
 
 pub struct Function {

src/style.rs

@@ -10,6 +10,12 @@ pub trait Line {
         T: Into<String>;
 
     fn get_colour(&self) -> &Option<String>;
+
+    fn width<T>(&mut self, value: T) -> &mut Self
+    where
+        T: Into<u8>;
+
+    fn get_width(&self) -> &Option<u8>;
 }
 
 /**

src/svg_render.rs

@@ -231,7 +231,7 @@ where
         node::element::Path::new()
             .set("fill", "none")
             .set("stroke", style.get_colour().clone().unwrap_or("".into()))
-            .set("stroke-width", 2)
+            .set("stroke-width", style.get_width().clone().unwrap_or(2))
             .set("d", path),
     );