#include"imfill.h"

Mat imfill(Mat cop,int n)
{
	Mat data = ~cop;
	Mat labels, stats, centroids;
	connectedComponentsWithStats(data, labels, stats, centroids, 4, CV_16U);
	int regions_count = stats.rows - 1;
	int regions_size, regions_x1, regions_y1, regions_x2, regions_y2;

	for (int i = 1; i <= regions_count; i++)
	{
		regions_size = stats.ptr<int>(i)[4];
		if (regions_size < n)
		{
			regions_x1 = stats.ptr<int>(i)[0];
			regions_y1 = stats.ptr<int>(i)[1];
			regions_x2 = regions_x1 + stats.ptr<int>(i)[2];
			regions_y2 = regions_y1 + stats.ptr<int>(i)[3];

			for (int j = regions_y1; j < regions_y2; j++)
			{
				for (int k = regions_x1; k < regions_x2; k++)
				{
					if (labels.ptr<ushort>(j)[k] == i)
						data.ptr<uchar>(j)[k] = 0;
				}
			}
		}
	}
	data = ~data;
	return data;
}

#include<opencv2/opencv.hpp>
#include<iostream>

using namespace std;
using namespace cv;

#include"imfill.h"

Mat src,dst, gray, erzhi,holef;

int main()
{
	src = imread("5.jpg");
	//imshow("万丈高楼第一步",src);

	blur(src, dst, Size(3, 3), Point(-1, -1));
	//imshow("均值滤波",dst);

	cvtColor(dst, gray, COLOR_BGR2GRAY);
	//imshow("灰度图",gray);

	Canny(gray, erzhi, 100, 200, 3, false);
	//imshow("边缘检测",erzhi);

	holef = imfill(erzhi,1000);
	imshow("填洞", holef);

	waitKey(0);
	return 0;
}