From bee9d1f89ca89e85a06b6ec613d0d0cf3f55e955 Mon Sep 17 00:00:00 2001
From: Patrycjusz Mania <worms308@wp.pl>
Date: Wed, 16 Jun 2021 20:54:14 +0200
Subject: [PATCH] zadanie - 10 (odszumianie)

---
 src/core/transformations/noise_bilateral.cpp | 43 ++++++++++++++++----
 src/core/transformations/noise_median.cpp    | 28 +++++++++++--
 2 files changed, 59 insertions(+), 12 deletions(-)

diff --git a/src/core/transformations/noise_bilateral.cpp b/src/core/transformations/noise_bilateral.cpp
index cc6579f..7288a9b 100644
--- a/src/core/transformations/noise_bilateral.cpp
+++ b/src/core/transformations/noise_bilateral.cpp
@@ -17,32 +17,59 @@ PNM* NoiseBilateral::transform()
 
     PNM* newImage = new PNM(width, height, image->format());
 
-    sigma_d = getParameter("sigma_d").toInt();
+    sigma_d = getParameter("sigma_d").toInt() * 1.8;
     sigma_r = getParameter("sigma_r").toInt();
     radius = sigma_d;
 
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    for (int x=0;x<width;++x) {
+        for (int y=0;y<height;++y) {
+            int r = calcVal(x,y, RChannel);
+            int g = calcVal(x,y, GChannel);
+            int b = calcVal(x,y, BChannel);
+            newImage->setPixel(x, y, QColor(r,g,b).rgb());
+        }
+    }
 
     return newImage;
 }
 
 int NoiseBilateral::calcVal(int x, int y, Channel channel)
 {
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    math::matrix<float> window = getWindow(x, y, radius, channel, RepeatEdge);
 
-    return 0;
+    float sumCounter = 0;
+    float sumDenominator = 0;
+    int pixel = image->pixel(x,y);
+    switch (channel) {
+        case RChannel: pixel = qRed(pixel);
+        case GChannel: pixel = qGreen(pixel);
+        case BChannel: pixel = qBlue(pixel);
+    }
+    for (int i=0;i<radius;++i) {
+        for (int j=0;j<radius;++j) {
+            int pixelVal = window[i][j];
+            QPoint point((i - radius/2) + x, (j - radius/2) + y);
+            QPoint point2(x,y);
+            sumCounter += pixelVal * colorCloseness(pixelVal, pixel) * spatialCloseness(point, point2);
+            sumDenominator += colorCloseness(pixelVal, pixel) * spatialCloseness(point, point2);
+        }
+    }
+    return sumDenominator != 0 ? sumCounter / sumDenominator : 0;
 }
 
 float NoiseBilateral::colorCloseness(int val1, int val2)
 {
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    float counter = pow(val1 - val2, 2);
+    float denominator = 2 * pow(sigma_r, 2);
 
-    return 0;
+    return exp(-(counter / denominator));
 }
 
 float NoiseBilateral::spatialCloseness(QPoint point1, QPoint point2)
 {
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    float X = pow(point1.x() - point2.x(), 2);
+    float Y = pow(point1.y() - point2.y(), 2);
+    float denominator = 2 * pow(sigma_d, 2);
 
-    return 0;
+    return exp(-(X + Y) / denominator);
 }
diff --git a/src/core/transformations/noise_median.cpp b/src/core/transformations/noise_median.cpp
index 1d8a9f7..41e2901 100644
--- a/src/core/transformations/noise_median.cpp
+++ b/src/core/transformations/noise_median.cpp
@@ -17,16 +17,36 @@ PNM* NoiseMedian::transform()
 
     PNM* newImage = new PNM(width, height, image->format());
 
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    for (int x=0;x<width;++x) {
+        for (int y=0;y<height;++y) {
+            int r = getMedian(x,y, RChannel);
+            int g = getMedian(x,y, GChannel);
+            int b = getMedian(x,y, BChannel);
+            newImage->setPixel(x, y, QColor(r,g,b).rgb());
+        }
+    }
 
     return newImage;
 }
 
 int NoiseMedian::getMedian(int x, int y, Channel channel)
 {
-    int radius = getParameter("radius").toInt();
+    int radius = getParameter("radius").toInt() * 2 + 1;
+    math::matrix<float> window = getWindow(x, y, radius, channel, RepeatEdge);
+    int items = radius * radius;
+    int tmp[items];
 
-    qDebug() << Q_FUNC_INFO << "Not implemented yet!";
+    int i=0;
+    for (int x=0;x<radius;++x) {
+        for (int y=0;y<radius;++y) {
+            tmp[i++] = window[x][y];
+        }
+    }
 
-    return 0;
+    std::sort(tmp, tmp + (items - 1));
+
+    if (items % 2 != 0)
+        return tmp[items / 2];
+
+    return (tmp[(items - 1) / 2] + tmp[items / 2]) / 2.0;
 }