connected components labeling

modules/gpu/include/opencv2/gpu/gpu.hpp

@@ -917,6 +917,9 @@ CV_EXPORTS void graphcut(GpuMat& terminals, GpuMat& leftTransp, GpuMat& rightTra
                          GpuMat& labels,
                          GpuMat& buf, Stream& stream = Stream::Null());
 
+//! performs connected componnents labeling.
+CV_EXPORTS void labelComponents(const GpuMat& image, GpuMat& mask, GpuMat& components, const cv::Scalar& lo, const cv::Scalar& hi);
+
 ////////////////////////////////// Histograms //////////////////////////////////
 
 //! Compute levels with even distribution. levels will have 1 row and nLevels cols and CV_32SC1 type.

modules/gpu/src/graphcuts.cpp

@@ -47,8 +47,26 @@
 void cv::gpu::graphcut(GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, Stream&) { throw_nogpu(); }
 void cv::gpu::graphcut(GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, GpuMat&, Stream&) { throw_nogpu(); }
 
+void cv::gpu::labelComponents(const GpuMat& image, GpuMat& mask, GpuMat& components, const cv::Scalar& lo, const cv::Scalar& hi) { throw_nogpu(); }
+
 #else /* !defined (HAVE_CUDA) */
 
+namespace cv { namespace gpu { namespace device
+{
+    namespace ccl
+    {
+        void labelComponents(const DevMem2D& edges, DevMem2Di comps);
+        void computeEdges(const DevMem2D& image, DevMem2D edges, const int lo, const int hi);
+    }
+}}}
+
+void cv::gpu::labelComponents(const GpuMat& image, GpuMat& mask, GpuMat& components, const cv::Scalar& lo, const cv::Scalar& hi)
+{
+    device::ccl::computeEdges(image, mask, lo[0], hi[0]);
+    device::ccl::labelComponents(mask, components);
+}
+
+
 namespace
 {
     typedef NppStatus (*init_func_t)(NppiSize oSize, NppiGraphcutState** ppState, Npp8u* pDeviceMem);

modules/gpu/test/test_labeling.cpp

+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+// By downloading, copying, installing or using the software you agree to this license.
+// If you do not agree to this license, do not download, install,
+// copy or use the software.
+//
+//
+//                          License Agreement
+//               For Open Source Computer Vision Library
+//
+// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
+// Copyright (C) 2008-2011, Willow Garage Inc., all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//  * Redistributions of source code must retain the above copyright notice,
+//    this list of conditions and the following disclaimer.
+//
+//  * Redistributions in binary form must reproduce the above copyright notice,
+//    this list of conditions and the following disclaimer in the documentation
+//    and/or other materials provided with the distribution.
+//
+//  * The name of the copyright holders may not be used to endorse or promote products
+//    derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Intel Corporation or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//M*/
+
+#include "precomp.hpp"
+#include <string>
+#include <iostream>
+
+struct Labeling : testing::TestWithParam<cv::gpu::DeviceInfo>
+{
+    cv::gpu::DeviceInfo devInfo;
+
+    virtual void SetUp()
+    {
+        devInfo = GetParam();
+        cv::gpu::setDevice(devInfo.deviceID());
+    }
+
+    cv::Mat loat_image()
+    {
+        return cv::imread(std::string( cvtest::TS::ptr()->get_data_path() ) + "labeling/label.png");
+    }
+};
+
+TEST_P(Labeling, ConnectedComponents)
+{
+    cv::Mat image;
+    cvtColor(loat_image(), image, CV_BGR2GRAY);
+    cv::Mat image_cpu = image.clone();
+
+    // cv::floodFill(image, cv::Point(1,1),cv::Scalar::all(64), 0, cv::Scalar::all(0), cv::Scalar::all(256));
+
+    cv::gpu::GpuMat mask;
+    mask.create(image.rows, image.cols, CV_8UC1);
+
+    cv::gpu::GpuMat components;
+    components.create(image.rows, image.cols, CV_32SC1);
+
+    std::cout << "summary: " << image.cols << " " << image.rows << " "
+    << cv::gpu::GpuMat(image).cols << " " << cv::gpu::GpuMat(image).rows<< " "
+    << mask.cols << " " << mask.rows<< " "
+    << components.cols << " " << components.rows<< std::endl;
+
+
+    cv::gpu::labelComponents(cv::gpu::GpuMat(image), mask, components, cv::Scalar::all(0), cv::Scalar::all(2));
+
+    // // for(int i = 0; i + 32 < image.rows; i += 32)
+    // //     for(int j = 0; j + 32 < image.cols; j += 32)
+    // //     {
+    // //         std::cout << cv::Mat(cv::Mat(mask), cv::Rect(j, i, 32, 32 ))<< std::endl;
+    // //         std::cout << cv::Mat(cv::Mat(components), cv::Rect(j, i, 32, 32 )) << std::endl;
+    // //     }
+
+    // std::cout << cv::Mat(components) << std::endl;
+    // cv::imshow("test", image);
+    // cv::waitKey(0);
+
+    // for(int i = 0; i + 32 < image.rows; i += 32)
+    //     for(int j = 0; j + 32 < image.cols; j += 32)
+    //         cv::rectangle(image, cv::Rect(j, i, 32, 32) , CV_RGB(255, 255, 255));
+
+    cv::imshow("test", image);
+    cv::waitKey(0);
+    cv::imshow("test", cv::Mat(mask) * 10);
+    cv::waitKey(0);
+    cv::imshow("test", cv::Mat(components) * 2);
+    cv::waitKey(0);
+    std::cout << "test! " << image.cols << std::endl;
+}
+
+INSTANTIATE_TEST_CASE_P(ConnectedComponents, Labeling, ALL_DEVICES);
\ No newline at end of file