#include "opencv2/imgcodecs.hpp"

#include "opencv2/highgui.hpp"

#include "opencv2/imgproc.hpp"

#include "opencv2/photo.hpp"

#include <iostream>

using namespace cv;

using namespace std;

static void help( char** argv )

{

cout << "\nCool inpainging demo. Inpainting repairs damage to images by floodfilling the damage \n"

<< "with surrounding image areas.\n"

"Using OpenCV version %s\n" << CV_VERSION << "\n"

"Usage:\n" << argv[0] <<" [image_name -- Default fruits.jpg]\n" << endl;

cout << "Hot keys: \n"

"\tESC - quit the program\n"

"\tr - restore the original image\n"

"\ti or SPACE - run inpainting algorithm\n"

"\t\t(before running it, paint something on the image)\n" << endl;

}

Mat img, inpaintMask;

Point prevPt(-1,-1);

static void onMouse( int event, int x, int y, int flags, void* )

{

if( event == EVENT_LBUTTONUP || !(flags & EVENT_FLAG_LBUTTON) )

prevPt = Point(-1,-1);

else if( event == EVENT_LBUTTONDOWN )

prevPt = Point(x,y);

else if( event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON) )

{

Point pt(x,y);

if( prevPt.x < 0 )

prevPt = pt;

line( inpaintMask, prevPt, pt, Scalar::all(255), 5, 8, 0 );

line( img, prevPt, pt, Scalar::all(255), 5, 8, 0 );

prevPt = pt;

imshow("image", img);

}

}

int main( int argc, char** argv )

{

cv::CommandLineParser parser(argc, argv, "{@image|fruits.jpg|}");

help(argv);

string filename = samples::findFile(parser.get<string>("@image"));

Mat img0 = imread(filename, IMREAD_COLOR);

if(img0.empty())

{

cout << "Couldn't open the image " << filename << ". Usage: inpaint <image_name>\n" << endl;

return 0;

}

namedWindow("image", WINDOW_AUTOSIZE);

img = img0.clone();

inpaintMask = Mat::zeros(img.size(), CV_8U);

imshow("image", img);

setMouseCallback( "image", onMouse, NULL);

for(;;)

{

char c = (char)waitKey();

if( c == 27 )

break;

if( c == 'r' )

{

inpaintMask = Scalar::all(0);

img0.copyTo(img);

imshow("image", img);

}

if( c == 'i' || c == ' ' )

{

Mat inpainted;

inpaint(img, inpaintMask, inpainted, 3, INPAINT_TELEA);

imshow("inpainted image", inpainted);

}

}

return 0;

}