Sharpen filter

daemon/src/app/settings.rs

@@ -17,6 +17,7 @@ pub enum Proc {
  Pixelate { k: i32 },
  Sepia,
  Edges { t1: f64, t2: f64 },
+ Sharpen,
 }
 
 impl Settings {

daemon/src/filters/mod.rs

@@ -4,6 +4,7 @@ pub mod blur;
 pub mod edges;
 pub mod pixelate;
 pub mod sepia;
+pub mod sharpen;
 mod utils;
 
 use crate::types::Frame;

daemon/src/filters/sharpen.rs

@@ -0,0 +1,43 @@
+use opencv::core::{BorderTypes, Mat, Point};
+use opencv::imgproc;
+
+use super::{utils, Filter};
+
+use crate::types::Frame;
+
+#[derive(Clone, Debug)]
+pub struct Sharpen {
+ kernel: Mat,
+ out: Frame,
+}
+
+pub fn new(out: Frame) -> Sharpen {
+ let kernel =
+ Mat::from_slice_2d(&vec![vec![-1, -1, -1], vec![-1, 9, -1], vec![-1, -1, -1]]).unwrap();
+
+ Sharpen { kernel, out }
+}
+
+impl Filter for Sharpen {
+ fn run(&mut self, src_frame: Frame) -> Frame {
+ let src = utils::frame_to_mat(src_frame);
+ let mut dst = utils::frame_to_mat(self.out);
+
+ imgproc::filter_2d(
+ &src,
+ &mut dst,
+ -1,
+ &self.kernel,
+ Point::new(-1, -1),
+ 0.0,
+ BorderTypes::BORDER_REPLICATE as i32,
+ )
+ .unwrap();
+
+ self.out
+ }
+
+ fn output(&self) -> Frame {
+ self.out
+ }
+}

daemon/src/pipeline.rs

@@ -74,6 +74,7 @@ pub fn alloc_filters(args: &Args, settings: &Settings) -> Vec<Box<dyn Filter>> {
  Pixelate { k } => Box::new(pixelate::new(*k, frame)),
  Sepia => Box::new(sepia::new(frame)),
  Edges { t1, t2 } => Box::new(edges::new(*t1, *t2, frame)),
+ Sharpen => Box::new(sharpen::new(frame)),
  }
  })
  .collect()