#include "core/core.hpp"
#include "highgui/highgui.hpp"
#include "imgproc/imgproc.hpp"
#include <time.h>
 
using namespace cv;
 
Mat Sourceimage, Spilteimage, Rebuildimage, Dstimage;
int rows, cols;
int Roirows, Roicols;
vector<Mat>arraryimage;
void Randarrary(vector<Mat> &vectorMat); //隨機排列子圖像序列函數
static int vectornumber = 0;
void OnMouseAction(int event, int x, int y, int flags, void *ustc); //鼠標回調事件函數
 
int mainFun()
{
 Sourceimage = imread("D:\\test\\lena.jpg");
 imshow("Source image", Sourceimage);
 rows = 6; //將圖像分割成rows行
 cols = 4; //將圖像分割成cols列
 Roirows = Sourceimage.rows / rows;
 Roicols = Sourceimage.cols / cols;
 Spilteimage = Mat::zeros(Sourceimage.rows, Sourceimage.cols, Sourceimage.type());
 Dstimage = Mat::zeros(Sourceimage.rows, Sourceimage.cols, Sourceimage.type());
 for (int i = 0; i<rows; i++)
 {
 for (int j = 0; j<cols; j++)
 {
 Mat SourceRoi = Sourceimage(Rect(j*Roicols, i*Roirows, Roicols - 1, Roirows - 1));
 arraryimage.push_back(SourceRoi);
 }
 }
 // 隨機函數
 Randarrary(arraryimage);
 
 for (int i = 0; i<rows; i++)
 {
 for (int j = 0; j<cols; j++)
 {
 Mat SpilterRoi = Spilteimage(Rect(j*Roicols, i*Roirows, Roicols - 1, Roirows - 1));
 addWeighted(SpilterRoi, 0, arraryimage[vectornumber], 1, 0, SpilterRoi);
 vectornumber++;
 imshow("Splite image", Spilteimage);
 waitKey(150);
 }
 }
 setMouseCallback("Splite image", OnMouseAction);
 waitKey();
 
 return 0;
 
}
 
//*******************************************************************//
//隨機調換所有的子圖像序列的位置，用于在 Splite image中顯示
//*******************************************************************//
void Randarrary(vector<Mat>& vectorMat)
{
 for (int i = 0; i<vectorMat.size(); i++)
 {
 srand(int(time(0)));
 int a = rand() % (vectorMat.size() - i) + i;
 swap(vectorMat[i], vectorMat[a]);
 }
 
}
 
//*******************************************************************//
//鼠標回調函數，用于獲取需要查找的子圖像在原圖像中的位置，并在疊加顯示在目標圖像中
//*******************************************************************//
void OnMouseAction(int event, int x, int y, int flags, void *ustc)
{
 if (event == CV_EVENT_LBUTTONDOWN)
 {
 Mat RoiSpilte, RoiSource;
 int rows = (y / Roirows)*Roirows;
 int clos = (x / Roicols)*Roicols;
 
 RoiSpilte = Spilteimage(Rect(clos, rows, Roicols, Roirows));
 imshow("Slice", RoiSpilte);
 
 Mat image = Mat::zeros(Sourceimage.rows - Roirows, Sourceimage.cols - Roicols, CV_32FC1);
 matchTemplate(Sourceimage, RoiSpilte, image, 1);
 normalize(image, image, 0, 1, NORM_MINMAX);
 
 double minV = 0;
 double maxV = 0;
 Point minP, maxP;
 
 minMaxLoc(image, &minV, &maxV, &minP, &maxP);
 
 Mat ROIDst = Dstimage(Rect(minP.x, minP.y, Roicols, Roirows));
 addWeighted(ROIDst, 0, RoiSpilte, 1, 0, ROIDst, -1);
 imshow("Jigsaw image", Dstimage);
 }
}
 
 
//-----開始------
void COpenCVLearningDlg::OnBnClickedStartButton()
{
 mainFun();
}