From 6df1a451ecd4f19f9b51f7396119ace8161eada2 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Thu, 7 Dec 2023 10:45:50 +0100
Subject: [PATCH] added face detection

---
 Lab11.iml                          |  3 +++
 src/livefilter/FaceDetection.scala | 36 ++++++++++++++++++++++++++++++
 src/livefilter/LiveFilter.scala    | 11 +++++++--
 3 files changed, 48 insertions(+), 2 deletions(-)
 create mode 100644 src/livefilter/FaceDetection.scala

diff --git a/Lab11.iml b/Lab11.iml
index c703763..fe799c4 100644
--- a/Lab11.iml
+++ b/Lab11.iml
@@ -3,6 +3,7 @@
   <component name="NewModuleRootManager" inherit-compiler-output="true">
     <exclude-output />
     <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/res" type="java-resource" />
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
     </content>
     <orderEntry type="inheritedJdk" />
@@ -11,5 +12,7 @@
     <orderEntry type="library" name="bridj-0.7.0" level="project" />
     <orderEntry type="library" name="slf4j-api-1.7.2" level="project" />
     <orderEntry type="library" name="webcam-capture-0.3.12" level="project" />
+    <orderEntry type="library" name="bytedeco.javacv" level="project" />
+    <orderEntry type="library" name="bytedeco.javacv.platform" level="project" />
   </component>
 </module>
\ No newline at end of file
diff --git a/src/livefilter/FaceDetection.scala b/src/livefilter/FaceDetection.scala
new file mode 100644
index 0000000..ee28618
--- /dev/null
+++ b/src/livefilter/FaceDetection.scala
@@ -0,0 +1,36 @@
+package livefilter
+
+import org.bytedeco.javacpp.Loader
+import org.bytedeco.javacpp.indexer.UByteRawIndexer
+import org.bytedeco.opencv.global.opencv_core.CV_8U
+import org.bytedeco.opencv.opencv_core.{Mat, Rect, RectVector}
+import org.bytedeco.opencv.opencv_objdetect.CascadeClassifier
+
+import java.io.File
+import java.net.URL
+
+object FaceDetection {
+  val url: URL = new URL("https://raw.github.com/opencv/opencv/master/data/haarcascades/haarcascade_frontalface_alt.xml")
+  val file: File = Loader.cacheResource(url)
+  val classifierName: String = file.getAbsolutePath
+  val cascade: CascadeClassifier = new CascadeClassifier(classifierName)
+
+  def detectFaces(src: Array[Array[Int]]): Array[Rect] = {
+    val grayFrame: Mat = new Mat()
+    val width: Int = src.length
+    val height: Int = if (width == 0) 0 else src(0).length
+    grayFrame.create(height, width, CV_8U)
+
+    val idx: UByteRawIndexer = grayFrame.createIndexer()
+
+    for (x: Int <- 0 until width) {
+      for (y: Int <- 0 until height) {
+        idx.put(y, x, src(x)(y))
+      }
+    }
+
+    val faces: RectVector = new RectVector()
+    cascade.detectMultiScale(grayFrame, faces)
+    return faces.get()
+  }
+}
diff --git a/src/livefilter/LiveFilter.scala b/src/livefilter/LiveFilter.scala
index 08bc521..167f008 100644
--- a/src/livefilter/LiveFilter.scala
+++ b/src/livefilter/LiveFilter.scala
@@ -3,6 +3,7 @@ package livefilter
 import com.github.sarxos.webcam.WebcamPanel.DrawMode
 import com.github.sarxos.webcam.{Webcam, WebcamImageTransformer, WebcamPanel, WebcamResolution}
 import imagefilters.ImageFilters
+import org.bytedeco.opencv.opencv_core.{Rect, RectVector}
 
 import java.awt.image.BufferedImage
 import java.awt.{Color, Dimension}
@@ -87,8 +88,14 @@ class LiveFilter extends WebcamImageTransformer {
 	override def transform(image: BufferedImage): BufferedImage = {
 		val img: Array[Array[Color]] = toArray(image)
 
-		// TODO Complete here by changing the assignment with your filters
-		val filtered: Array[Array[Color]] = ImageFilters.mask(ImageFilters.sepia(ImageFilters.noise(img, 10)), videoMask)
+		val bw: Array[Array[Int]] = toBW(image)
+		val faces: Array[Rect] = FaceDetection.detectFaces(bw)
+		var filtered: Array[Array[Color]] = img
+		for (rect: Rect <- faces) {
+      filtered = ImageFilters.pixelize(filtered, 10, rect.x, rect.y, rect.x+rect.width, rect.y+rect.height)
+    }
+
+		//val filtered: Array[Array[Color]] = ImageFilters.mask(ImageFilters.sepia(ImageFilters.noise(img, 10)), videoMask)
 
 		updateImage(filtered, image)
 		return image