/*
* cloning.cpp
*
* Author:
* Siddharth Kherada <siddharthkherada27[at]gmail[dot]com>
*
* This tutorial demonstrates how to use OpenCV seamless cloning
* module.
* Flags:
* 1- NORMAL_CLONE
* 2- MIXED_CLONE
* 3- FEATURE_EXCHANGE

* The program takes as input a source and a destination image
* and ouputs the cloned image.

* Step 1:
* -> In the source image, select the region of interest by left click mouse button. A Polygon ROI will be created by left clicking mouse button.
* -> To set the Polygon ROI, click the right mouse button.
* -> To reset the region selected, click the middle mouse button.

* Step 2:
* -> In the destination image, select the point where you want to place the ROI in the image by left clicking mouse button.
* -> To get the cloned result, click the right mouse button.

* Result: The cloned image will be displayed.
*/

#include "opencv2/photo.hpp"
#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/core.hpp"
#include <iostream>
#include <stdlib.h>

using namespace std;
using namespace cv;

Mat img0, img1, img2, res, res1, final, final1, blend;

Point point;
int drag = 0;
int destx, desty;

int numpts = 100;
Point* pts = new Point[100];
Point* pts2 = new Point[100];
Point* pts_diff = new Point[100];


int var = 0;
int flag = 0;
int flag1 = 0;

int minx,miny,maxx,maxy,lenx,leny;
int minxd,minyd,maxxd,maxyd,lenxd,lenyd;

int channel,num;

void source(int event, int x, int y, int, void*)
{

	if (event == EVENT_LBUTTONDOWN && !drag)
	{
		if(flag1 == 0)
		{
			if(var==0)
				img1 = img0.clone();
			point = Point(x, y);
			circle(img1,point,2,Scalar(0, 0, 255),-1, 8, 0);
			pts[var] = point;
			var++;
			drag  = 1;
			if(var>1)
				line(img1,pts[var-2], point, Scalar(0, 0, 255), 2, 8, 0);

			imshow("Source", img1);
		}
	}


	if (event == EVENT_LBUTTONUP && drag)
	{
		imshow("Source", img1);

		drag = 0;
	}
	if (event == EVENT_RBUTTONDOWN)
	{
		flag1 = 1;
		img1 = img0.clone();
		for(int i = var; i < numpts ; i++)
			pts[i] = point;

		if(var!=0)
		{
			const Point* pts3[1] = {&pts[0]};
			polylines( img1, pts3, &numpts,1, 1, Scalar(0,0,0), 2, 8, 0);
		}

		for(int i=0;i<var;i++)
		{
			minx = min(minx,pts[i].x);
			maxx = max(maxx,pts[i].x);
			miny = min(miny,pts[i].y);
			maxy = max(maxy,pts[i].y);
		}
		lenx = maxx - minx;
		leny = maxy - miny;

        int mid_pointx = minx + lenx/2;
        int mid_pointy = miny + leny/2;

        for(int i=0;i<var;i++)
        {
            pts_diff[i].x = pts[i].x - mid_pointx;
            pts_diff[i].y = pts[i].y - mid_pointy;
        }

		imshow("Source", img1);
	}

	if (event == EVENT_RBUTTONUP)
	{
		flag = var;

		final = Mat::zeros(img0.size(),CV_8UC3);
		res1 = Mat::zeros(img0.size(),CV_8UC1);
		const Point* pts4[1] = {&pts[0]};

		fillPoly(res1, pts4,&numpts, 1, Scalar(255, 255, 255), 8, 0);
		bitwise_and(img0, img0, final,res1);

		imshow("Source", img1);

	}
	if (event == EVENT_MBUTTONDOWN)
	{
		for(int i = 0; i < numpts ; i++)
		{
			pts[i].x=0;
			pts[i].y=0;
		}
		var = 0;
		flag1 = 0;
		minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;
		imshow("Source", img0);
		drag = 0;
	}
}


void destination(int event, int x, int y, int, void*)
{


	Mat im1;
	minxd = INT_MAX; minyd = INT_MAX; maxxd = INT_MIN; maxyd = INT_MIN;
	im1 = img2.clone();
	if (event == EVENT_LBUTTONDOWN)
	{
		if(flag1 == 1)
		{
			point = Point(x, y);

            for(int i=0;i<var;i++)
            {
                pts2[i].x = point.x + pts_diff[i].x;
                pts2[i].y = point.y + pts_diff[i].y;
            }
            
            for(int i=var;i<numpts;i++)
            {
                pts2[i].x = point.x + pts_diff[0].x;
                pts2[i].y = point.y + pts_diff[0].y;
            }

			const Point* pts5[1] = {&pts2[0]};
			polylines( im1, pts5, &numpts,1, 1, Scalar(0,0,255), 2, 8, 0);

			destx = x;
			desty = y;

			imshow("Destination", im1);
		}
	}
	if (event == EVENT_RBUTTONUP)
	{
		for(int i=0;i<flag;i++)
		{
			minxd = min(minxd,pts2[i].x);
			maxxd = max(maxxd,pts2[i].x);
			minyd = min(minyd,pts2[i].y);
			maxyd = max(maxyd,pts2[i].y);
		}

		if(maxxd > im1.size().width || maxyd > im1.size().height || minxd < 0 || minyd < 0)
		{
			cout << "Index out of range" << endl;
			exit(0);
		}

		final1 = Mat::zeros(img2.size(),CV_8UC3);
		res = Mat::zeros(img2.size(),CV_8UC1);
		for(int i=miny, k=minyd;i<(miny+leny);i++,k++)
			for(int j=minx,l=minxd ;j<(minx+lenx);j++,l++)
			{
				for(int c=0;c<channel;c++)
				{
					final1.at<uchar>(k,l*channel+c) = final.at<uchar>(i,j*channel+c);

				}
			}


		const Point* pts6[1] = {&pts2[0]};
		fillPoly(res, pts6, &numpts, 1, Scalar(255, 255, 255), 8, 0);

		if(num == 1 || num == 2 || num == 3)
		{
            seamlessClone(img0,img2,res1,point,blend,num);        
			imshow("Cloned Image", blend);
            imwrite("cloned.png",blend);
			waitKey(0);
		}

		for(int i = 0; i < flag ; i++)
		{
			pts2[i].x=0;
			pts2[i].y=0;
		}

		minxd = INT_MAX; minyd = INT_MAX; maxxd = INT_MIN; maxyd = INT_MIN;
	}

	im1.release();
}

int main(int argc, char **argv)
{

    if (argc != 3) 
    {
        cout << "usage: " << argv[0] << "  <source image>" << "  <destination image>" << endl;
        exit(1);
    }

    Mat src = imread(argv[1]);
    Mat dest = imread(argv[2]);

    cout << "Flags:" << endl;
    cout << "1- NORMAL_CLONE" << endl;
    cout << "2- MIXED_CLONE" << endl;
    cout << "3- FEATURE_EXCHANGE" << endl << endl;

    cout << "Enter Flag:";
    cin >> num;

	minx = INT_MAX; miny = INT_MAX; maxx = INT_MIN; maxy = INT_MIN;

	minxd = INT_MAX; minyd = INT_MAX; maxxd = INT_MIN; maxyd = INT_MIN;

	img0 = src;
	img2 = dest;

	channel = img0.channels();

	res = Mat::zeros(img2.size(),CV_8UC1);
	res1 = Mat::zeros(img0.size(),CV_8UC1);
	final = Mat::zeros(img0.size(),CV_8UC3);
	final1 = Mat::zeros(img2.size(),CV_8UC3);
	//////////// source image ///////////////////

	namedWindow("Source", 1);
	setMouseCallback("Source", source, NULL);
	imshow("Source", img0);

	/////////// destination image ///////////////

	namedWindow("Destination", 1);
	setMouseCallback("Destination", destination, NULL);
	imshow("Destination",img2);
	waitKey(0);

	img0.release();
	img1.release();
	img2.release();
}