Create filter trait.

Author: jschlumpp

src/bin/cli.rs

@@ -5,7 +5,7 @@ extern crate clap;
 
 use clap::{Arg, App, AppSettings};
 
-use geometrify::RandomPointGenerator;
+use geometrify::{RandomPointGenerator, Filter};
 use geometrify::geometrify::Geometrify;
 
 use image::open;
@@ -67,7 +67,7 @@ fn main() {
     );
 
 
-    let outputbuf = filter.apply(sourcebuf);
+    let outputbuf = filter.apply(&sourcebuf);
 
     outputbuf
         .save(&Path::new(matches.value_of("OUTPUT").expect("expected output file")))

src/geometrify.rs

@@ -1,4 +1,4 @@
-use super::{Point, BoundingBox, PointGenerator};
+use super::{Point, BoundingBox, PointGenerator, Filter};
 use image::{Rgba, RgbaImage, Pixel};
 
 use pbr::ProgressBar;
@@ -287,8 +287,10 @@ impl Geometrify {
 
         result
     }
+}
 
-    pub fn apply(&mut self, image: RgbaImage) -> RgbaImage {
+impl Filter for Geometrify {
+    fn apply(&mut self, image: &RgbaImage) -> RgbaImage {
         let mut progress = ProgressBar::new(self.iterations as u64);
         progress.format("|#--|");
 

src/lib.rs

@@ -5,6 +5,8 @@ extern crate pbr;
 
 pub mod geometrify;
 
+
+use image::RgbaImage;
 use rand::Rng;
 
 #[derive(Clone, Copy, Debug)]
@@ -54,6 +56,10 @@ impl PointGenerator for RandomPointGenerator {
     }
 }
 
+pub trait Filter {
+    fn apply(&mut self, image: &RgbaImage) -> RgbaImage;
+}
+
 #[cfg(test)]
 mod tests {
     #[test]