#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();
}