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

using namespace std;
using namespace cv;

bool g_bDrawing = false;
Point g_CurrPoint, g_OrgPoint;
int g_nThick = 5, g_nBlue = 255, g_nGreen = 255, g_nRed = 0;
int g_nImageOneValue = 49;
Mat srcImage;
Mat grayImage;
Mat maskImage;

/*注意：不能在毀掉函式中寫入未初始化的矩陣類，所以需要用時，需要寫一個標誌位，然後再在while(1)迴圈內使用*/
void onMouse(int event, int x, int y, int flag, void *param)
{
	Mat &img = *(cv::Mat*)param;

	switch (event)
	{
		//移動滑鼠的時候
	case CV_EVENT_MOUSEMOVE:
	{
							   g_OrgPoint = g_CurrPoint;
							   g_CurrPoint = Point(x, y);

							   if (g_bDrawing == 1)
							   {
								   line(srcImage, g_CurrPoint, g_OrgPoint, Scalar(g_nBlue, g_nGreen, g_nRed), g_nThick);
								   imshow("【滑鼠事件視窗】", srcImage);

								   //在掩膜圖上進行顯示
								   line(maskImage, g_CurrPoint, g_OrgPoint, Scalar(g_nBlue, g_nGreen, g_nRed), g_nThick);
								   imshow("【掩膜影象】", maskImage);
							   }
	}
		break;
		//點選滑鼠左鍵時
	case CV_EVENT_LBUTTONDOWN:
	{
								 g_bDrawing = true;
								 g_OrgPoint = Point(x, y);
								 g_CurrPoint = g_OrgPoint;
	}
		break;
		//鬆開滑鼠左鍵時
	case CV_EVENT_LBUTTONUP:
	{
							   g_bDrawing = false;
	}
		break;
	}
}

int main()
{
	Mat tempImage;
	RNG &rng = theRNG();

	srcImage = imread("lili.jpg");

	//用一個變數來儲存原影象
	Mat g_srcImage;
	srcImage.copyTo(g_srcImage);

	//為掩膜圖 分配空間
	maskImage.create(srcImage.size(), CV_8UC1);
	maskImage = Scalar::all(0);

	namedWindow("【滑鼠事件視窗】");
	setMouseCallback("【滑鼠事件視窗】", onMouse, 0);

	namedWindow("【滾動條視窗】", 0);
	createTrackbar("thick", "【滾動條視窗】", &g_nThick, 100, 0);
	createTrackbar("Blue", "【滾動條視窗】", &g_nBlue, 255, 0);
	createTrackbar("Green", "【滾動條視窗】", &g_nGreen, 255, 0);
	createTrackbar("Red", "【滾動條視窗】", &g_nRed, 255, 0);

	char key;
	while (1)
	{
		imshow("【滑鼠事件視窗】", srcImage);
		key = waitKey();
		if (key == 27)
			break;

		//如果檢測到 鍵值是1 則恢復原圖
		if (key == '1')
		{
			g_srcImage.copyTo(srcImage);
			maskImage = Scalar::all(0);
			imshow("【滑鼠事件視窗】", srcImage);
		}

		//如果檢測到空格 則開始執行影象修復
		Mat dstImage;
		dstImage.create(srcImage.size(), srcImage.type());
		if (key == ' ')
		{
			inpaint(srcImage, maskImage, dstImage, 3, INPAINT_TELEA);
			imshow("【修補後的影象】", dstImage);
		}
	}

	return 0;
}