working on python wrappers for stitching

modules/python/CMakeLists.txt

@@ -10,7 +10,7 @@ if(ANDROID OR IOS OR NOT PYTHONLIBS_FOUND OR NOT PYTHON_USE_NUMPY)
 endif()
 
 set(the_description "The python bindings")
-ocv_add_module(python BINDINGS opencv_core opencv_flann opencv_imgproc opencv_video opencv_ml opencv_features2d opencv_highgui opencv_calib3d opencv_photo opencv_nonfree opencv_objdetect opencv_legacy opencv_contrib)
+ocv_add_module(python BINDINGS opencv_core opencv_flann opencv_imgproc opencv_video opencv_ml opencv_features2d opencv_highgui opencv_calib3d opencv_photo opencv_nonfree opencv_objdetect opencv_legacy opencv_contrib opencv_stitching)
 
 ocv_include_directories(${PYTHON_INCLUDE_PATH})
 ocv_include_directories(
@@ -29,6 +29,7 @@ ocv_include_directories(
     "${OpenCV_SOURCE_DIR}/modules/nonfree/include"
     "${OpenCV_SOURCE_DIR}/modules/legacy/include"
     "${OpenCV_SOURCE_DIR}/modules/contrib/include"
+    "${OpenCV_SOURCE_DIR}/modules/stitching/include"
     )
 
 include_directories(${CMAKE_CURRENT_BINARY_DIR})
@@ -44,7 +45,8 @@ set(opencv_hdrs "${OpenCV_SOURCE_DIR}/modules/core/include/opencv2/core/core.hpp
     "${OpenCV_SOURCE_DIR}/modules/features2d/include/opencv2/features2d/features2d.hpp"
     "${OpenCV_SOURCE_DIR}/modules/nonfree/include/opencv2/nonfree/features2d.hpp"
     "${OpenCV_SOURCE_DIR}/modules/calib3d/include/opencv2/calib3d/calib3d.hpp"
-    "${OpenCV_SOURCE_DIR}/modules/objdetect/include/opencv2/objdetect/objdetect.hpp")
+    "${OpenCV_SOURCE_DIR}/modules/objdetect/include/opencv2/objdetect/objdetect.hpp"
+    "${OpenCV_SOURCE_DIR}/modules/stitching/include/opencv2/stitching/detail/seam_finders.hpp")
 
 if(MSVC)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /W3")
@@ -75,7 +77,7 @@ add_custom_command(
 
 set(cv2_target "opencv_python")
 add_library(${cv2_target} SHARED src2/cv2.cpp ${CMAKE_CURRENT_BINARY_DIR}/generated0.i ${cv2_generated_hdrs} src2/cv2.cv.hpp)
-target_link_libraries(${cv2_target} ${PYTHON_LIBRARIES} opencv_core opencv_flann opencv_imgproc opencv_video opencv_ml opencv_features2d opencv_highgui opencv_calib3d opencv_objdetect opencv_legacy opencv_contrib opencv_photo)
+target_link_libraries(${cv2_target} ${PYTHON_LIBRARIES} opencv_core opencv_flann opencv_imgproc opencv_video opencv_ml opencv_features2d opencv_highgui opencv_calib3d opencv_objdetect opencv_legacy opencv_contrib opencv_photo opencv_stitching)
     
 set_target_properties(${cv2_target} PROPERTIES PREFIX "")
 set_target_properties(${cv2_target} PROPERTIES OUTPUT_NAME "cv2")

modules/python/src2/cv2.cpp

@@ -20,9 +20,11 @@
 #include "opencv2/photo/photo.hpp"
 #include "opencv2/nonfree/nonfree.hpp"
 #include "opencv2/highgui/highgui.hpp"
+#include "opencv2/stitching/stitcher.hpp"
 
 using cv::flann::IndexParams;
 using cv::flann::SearchParams;
+using cv::detail::GraphCutSeamFinder;
 
 static PyObject* opencv_error = 0;
 

modules/python/src2/hdr_parser.py

@@ -11,7 +11,8 @@ opencv_hdr_list = [
 "../../video/include/opencv2/video/tracking.hpp",
 "../../video/include/opencv2/video/background_segm.hpp",
 "../../objdetect/include/opencv2/objdetect/objdetect.hpp",
-"../../highgui/include/opencv2/highgui/highgui.hpp"
+"../../highgui/include/opencv2/highgui/highgui.hpp",
+"../../stitching/include/opencv2/stitching/detail/seam_finders.hpp"
 ]
 
 """

modules/stitching/include/opencv2/stitching/detail/seam_finders.hpp

@@ -49,11 +49,11 @@
 namespace cv {
 namespace detail {
 
-class CV_EXPORTS SeamFinder
+class CV_EXPORTS_AS(SeamFinder) SeamFinder
 {
 public:
     virtual ~SeamFinder() {}
-    virtual void find(const std::vector<Mat> &src, const std::vector<Point> &corners,
+    CV_WRAP virtual void find(const std::vector<Mat> &src, const std::vector<Point> &corners,
                       std::vector<Mat> &masks) = 0;
 };
 
@@ -99,10 +99,10 @@ public:
 };
 
 
-class CV_EXPORTS GraphCutSeamFinder : public GraphCutSeamFinderBase, public SeamFinder
+class CV_EXPORTS_AS(GraphCutSeamFinder) GraphCutSeamFinder : public SeamFinder, public GraphCutSeamFinderBase
 {
 public:
-    GraphCutSeamFinder(int cost_type = COST_COLOR_GRAD, float terminal_cost = 10000.f,
+    CV_WRAP GraphCutSeamFinder(int cost_type = GraphCutSeamFinder::COST_COLOR_GRAD, float terminal_cost = 10000.f,
                        float bad_region_penalty = 1000.f);
 
     ~GraphCutSeamFinder();