// Laborator 3 - lucrarea 4

Point Pstart, Pend;

void callBack(int event, int x, int y, int flags, void *userdata)
{
	Mat* H = (Mat*)userdata;
	if (event == EVENT_RBUTTONDOWN)
	{
		int height = (*H).rows;
		int width = (*H).cols;
		Mat labels = Mat::zeros((*H).size(), CV_8UC1);
		Mat dst = Mat::zeros((*H).size(), CV_8UC1);

		std::queue<Point> que;
		int w = 1;
		float hue_avg = (*H).at<uchar>(y, x);
		int hue_std = 5;
		float k1 = 2.8;
		// float T = hue_std * k1;
		float T = 15;
		/*
		for (int i = x - w; i <= x + w; i++)
		{
			for (int j = y - w; j <= y + w; j++)
			{
				hue_avg += (*H).at<uchar>(i, j);
				printf("hue_avg = %d", hue_avg);
			}
		}
		*/

		printf("hue avg = %f", hue_avg);

		int k = 1; //ethiceta curenta
		int N = 1; // numarul de pixeli din regiune

		que.push(Point(x, y)); // adauga element (seed point) in coada
		// acesta primeste eticheta k
		while (!que.empty())
		{ // Retine poz. celui mai vechi element din coada
			Point oldest = que.front();
			que.pop(); // scoate element din coada
			int xx = oldest.x; // coordonatele lui
			int yy = oldest.y;

			for (int i = yy - w; i <= yy + w; i++)
			{
				for (int j = xx - w; j <= xx + w; j++)
				{
					if (inside(i, j, height, width))
					{
						if ((labels.at<uchar>(i, j) == 0) && (abs((*H).at<uchar>(i, j) - hue_avg) < T))
						{
							que.push(Point(j, i));
							labels.at<uchar>(i, j) = k;
							hue_avg = (N * hue_avg + (*H).at<uchar>(i, j)) / (N + 1);
							N++;
						}
					}
				}
			}
		}
		for (int i = 0; i < height; i++)
		{
			for (int j = 0; j < width; j++)
			{
				if (labels.at<uchar>(i, j) == 1)
				{
					dst.at<uchar>(i, j) = 255;
				}
				else
				{
					dst.at<uchar>(i, j) = 0;
				}
			}
		}
		imshow("Imaginea destinatie inainte de postprocesare", dst);

		// creare element structural de dimensiune 3x3 de tip patrat (V8)
		Mat element = getStructuringElement(MORPH_RECT, Size(3, 3));

		// dilatare cu acest element structural (aplicata n)
		dilate(dst, dst, element, Point(-1, -1), 3);

		// dilatare cu acest element structural (aplicata n)
		erode(dst, dst, element, Point(-1, -1), 3);

		imshow("Imaginea destinatie", dst);
	}
}

void regionGrowing()
{
	char fname[MAX_PATH];
	while (openFileDlg(fname))
	{
		Mat src = imread(fname);
		//imshow("orignial", src);
		Mat hue = Mat(src.rows, src.cols, CV_8UC1);
		Mat labels = Mat::zeros(hue.size(), CV_8UC1);
		vector<Mat> hsv;

		GaussianBlur(src, src, Size(5, 5), 0, 0);

		hsv = BGR2HSV(src);
		hue = hsv[0];

		//Create a window
		namedWindow("My Window", 1);

		//set the callback function for any mouse event
		setMouseCallback("My Window", callBack, &hue);

		//show the image
		imshow("My Window", src);

		// Wait until user press some key
		waitKey(0);
	}
}