#ifndef WATERSHEDSEGMENTER

#define WATERSHEDSEGMENTER

#include <opencv2/core/core.hpp>

#include <opencv2/imgproc/imgproc.hpp>

class WatershedSegmenter {

  private:

      //用來表示標記（圖）

      cv::Mat markers;

  public:

       //設置標記圖

      void setMarkers(const cv::Mat& markerImage) {

        //watershed()的輸入參數必須為一個32位有符號的標記，所以要先進行轉換

        markerImage.convertTo(markers,CV_32S);

      }

      //執行watershed（）

      cv::Mat process(const cv::Mat &image) {

        // Apply watershed

        cv::watershed(image,markers);

        return markers;

      }

      // 以圖像形式返回結果

      cv::Mat getSegmentation() {

        cv::Mat tmp;

    // 從32S到8U（0-255）會進行飽和運算，所以像素高于255的一律復制為255

        markers.convertTo(tmp,CV_8U);//

        return tmp;

      }

      // 以圖像形式返回分水嶺（我理解的是分割線）

      cv::Mat getWatersheds() {

        cv::Mat tmp;

        //在設置標記圖像，即執行setMarkers（）后，邊緣的像素會被賦值為-1，其他的用正整數表示

        //下面的這個轉換可以讓邊緣像素變為-1*255+255=0，即黑色，其余的溢出，賦值為255，即白色。

        markers.convertTo(tmp,CV_8U,,);

        return tmp;

      }

};

#endif // WATERSHEDSEGMENTER

#include <iostream>

#include <opencv2/core/core.hpp>

#include <opencv2/imgproc/imgproc.hpp>

#include <opencv2/highgui/highgui.hpp>

#include "watershedsegmenter.h"

using namespace std;

using namespace cv;

int main()

{

    // Read input image 原圖

    Mat image= imread("/home/osksh/skin_c/dulani_anuradha4.jpg");

    if (!image.data)

        return ;

    // Display the image

    namedWindow("Original Image");

    imshow("Original Image",image);

    Mat binary,fg,bg;

    cvtColor(image,binary,CV_RGBA2GRAY);

    //Display the binary image

    namedWindow("Binary Image");

    imshow("Binary Image",binary);

    // 由二值圖像獲得前景。腐蝕。移除噪點與微小物體

    erode(binary,fg,cv::Mat(),cv::Point(-,-),);

    // Display the foreground image

    namedWindow("Foreground Image");

    imshow("Foreground Image",fg);

    //膨脹二值圖來獲取背景（只有草地，沒有樹林）

    dilate(binary,bg,cv::Mat(),cv::Point(-,-),);

    threshold(bg,bg,,,cv::THRESH_BINARY_INV);

    //最后一個參數的表示 ifsrc>1,dst=0,else dst=128。這樣就使背景全為灰色（128）

    // Display the background image

    namedWindow("Background Image");

    imshow("Background Image",bg);

    // Show markers image

    Mat markers(binary.size(),CV_8U,cv::Scalar());

    markers= fg+bg;//使用重載操作符+

    namedWindow("Markers");

    imshow("Markers",markers);

    // Create watershed segmentation object

    WatershedSegmenter segmenter;

    // Set markers and process

    segmenter.setMarkers(markers);

    segmenter.process(image);

    // Display segmentation result

    namedWindow("Segmentation");

    imshow("Segmentation",segmenter.getSegmentation());

    // Display watersheds

    namedWindow("Watersheds");

    imshow("Watersheds",segmenter.getWatersheds());

    waitKey();

    return ;

}