initial commit

samples/cpp/inpaint.cpp

@@ -0,0 +1,91 @@
+#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;
+}