#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
using namespace std;
using namespace cv;
/// <summary>
/// 随机添加椒盐噪点 ---- 和课件上不一样
/// </summary>
/// <param name="img"></param>
/// <param name="n"></param>
void add_salt(Mat image, Mat &img, int n)
{
    image.copyTo(img);
    for (int k = 0; k < n; k++)
    {
        int i = rand() % img.cols;
        int j = rand() % img.rows;
        //通过通道数量判断图像的类型
        if (img.channels() == 1) {
            //设置为白色像素
            img.at<uchar>(i, j) = 255;
        }
        else
        {
            img.at<Vec3b>(i, j)[0] = 255;
            img.at<Vec3b>(i, j)[1] = 255;
            img.at<Vec3b>(i, j)[2] = 255;
        }
    }
}
int main() {
    //读取图像
    Mat img = imread("G:/image/xs.jpg");
    //改为灰度图像
    cvtColor(img, img, COLOR_BGR2GRAY);
    Mat noise, gaussian, median, element;
    resize(img, img, Size(256, 256), INTER_AREA);
    //添加椒盐噪点
    add_salt(img, noise, 1000);
    //高斯滤波
    int kernel_size = 3;
    element = getStructuringElement(MORPH_RECT, Size(5, 5));
    GaussianBlur(noise, gaussian, Size(kernel_size, kernel_size),1);
    //中值滤波
    medianBlur(noise, median, kernel_size);
    //生成图像
    imshow("Original image", img);
    imshow("noise_image", noise);
    imshow("gaussian_image", gaussian);
    imshow("median_image", median);
    waitKey(0);
    return 0;
}