#include “opencv2/imgproc.hpp”
#include “opencv2/highgui.hpp”
#include <iostream>

using namespace cv;
using namespace std;

Mat src, erosion_dst, dilation_dst;

int erosion_elem = 0;
int erosion_size = 0;
int dilation_elem = 0;
int dilation_size = 0;
int const max_elem = 2;
int const max_kernel_size = 21;

void Erosion(int, void*);
void Dilation(int, void*);
Mat ReadData();

int main(){

src = ReadData();

namedWindow(“Erosion Demo”, WINDOW_NORMAL);
namedWindow(“Dilation Demo”, WINDOW_NORMAL);
moveWindow(“Dilation Demo”, src.cols, 0);

createTrackbar(“Element:\n 0: Rect \n 1: Cross \n 2: Ellipse”, “Erosion Demo”,
&erosion_elem, max_elem,
Erosion);

createTrackbar(“Kernel size:\n 2n +1”, “Erosion Demo”,
&erosion_size, max_kernel_size,
Erosion);

createTrackbar(“Element:\n 0: Rect \n 1: Cross \n 2: Ellipse”, “Dilation Demo”,
&dilation_elem, max_elem,
Dilation);

createTrackbar(“Kernel size:\n 2n +1”, “Dilation Demo”,
&dilation_size, max_kernel_size,
Dilation);

Erosion(0, 0);
Dilation(0, 0);
waitKey(0);
return 0;

}

void Erosion(int, void*){

int erosion_type = 0;
if (erosion_elem == 0) { erosion_type = MORPH_RECT; }
else if (erosion_elem == 1) { erosion_type = MORPH_CROSS; }
else if (erosion_elem == 2) { erosion_type = MORPH_ELLIPSE; }
Mat element = getStructuringElement(erosion_type,
Size(2 * erosion_size + 1, 2 * erosion_size + 1),
Point(erosion_size, erosion_size));
erode(src, erosion_dst, element);
imshow(“Erosion Demo”, erosion_dst);

}

void Dilation(int, void*){

int dilation_type = 0;
if (dilation_elem == 0) { dilation_type = MORPH_RECT; }
else if (dilation_elem == 1) { dilation_type = MORPH_CROSS; }
else if (dilation_elem == 2) { dilation_type = MORPH_ELLIPSE; }
Mat element = getStructuringElement(dilation_type,
Size(2 * dilation_size + 1, 2 * dilation_size + 1),
Point(dilation_size, dilation_size));
dilate(src, dilation_dst, element);
imshow(“Dilation Demo”, dilation_dst);

}

Mat ReadData(){

Mat I;

cout << “Read Done.” << endl;
// Read the image file
I = imread(“D:/images/ThumbsUp.jpg”, IMREAD_COLOR); // Read the file

// Check for failure
if (I.empty())
{
cout << “Could not open or find the image” << endl;
cin.get(); //wait for any key press
}
else
{
cout << “Loading input image: ” << endl;
}
return I;

}