#include<opencv2/imgproc/imgproc.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<iostream>

using namespace cv;
using namespace std;
//宏定義
#define WINDOW_NAME "image[procedure window]"

//全局變量聲明
Mat g_srcImage,g_maskImage;
Point prevPt(-1,-1);

//全局函數(shù)聲明
static void on_Mouse(int event,int x,int y,int flags,void*);

//主函數(shù)
int main()
{
  //載入源圖像
  g_srcImage=imread("/Users/new/Desktop/1.jpg");
  if(!g_srcImage.data){printf("讀取源圖像srcImage錯誤～！\n");return false;}

  //顯示源圖像
  imshow(WINDOW_NAME,g_srcImage);
  Mat srcImage,grayImage;
  g_srcImage.copyTo(srcImage);
  //灰度化
  cvtColor(srcImage, g_maskImage, COLOR_BGR2GRAY);
  //imshow("image[mask]",g_maskImage);
  cvtColor(g_maskImage, grayImage, COLOR_GRAY2BGR);
  //imshow("image[gray]",grayImage);
  //掩膜圖像初始化為0
  g_maskImage=Scalar::all(0);

  //設(shè)置鼠標(biāo)回調(diào)函數(shù)
  setMouseCallback(WINDOW_NAME, on_Mouse,0);

  //輪詢按鍵處理
  while(1)
  {
    //獲取鍵值
    int c=waitKey(0);
    //若按鍵為ESC時，退出
    if((char)c == 27)
      break;
    //若按鍵為2時，恢復(fù)原圖
    if((char)c=='2')
    {
      g_maskImage=Scalar::all(0);
      srcImage.copyTo(g_srcImage);
      imshow("image",g_srcImage);
    }
    //若按鍵為1,則進(jìn)行處理
    if((char)c=='1')
    {
      //定義一些參數(shù)
      int i,j,compCount=0;
      vector<vector<Point>>contours;
      vector<Vec4i> hierarchy;
      //尋找輪廓
      findContours(g_maskImage, contours, hierarchy, CV_RETR_CCOMP, CHAIN_APPROX_SIMPLE);
      //輪廓為空時的處理
      if(contours.empty())
        continue;
      //復(fù)制掩膜
      Mat maskImage(g_maskImage.size(),CV_32S);
      maskImage=Scalar::all(0);

      //循環(huán)繪制輪廓
      for(int index=0;index>=0;index=hierarchy[index][0],++compCount)
        drawContours(maskImage, contours, index, Scalar::all(compCount+1),-1,8,hierarchy,INT_MAX);
        //compCount為零時的處理
        if(compCount==0)
          continue;

        //生成隨機(jī)顏色
        vector<Vec3b>colorTab;
        for(int i=0;i<compCount;++i)
        {
          int b=theRNG().uniform(0, 255);
          int g=theRNG().uniform(0, 255);
          int r=theRNG().uniform(0, 255);

          colorTab.push_back(Vec3b((uchar)b,(uchar)g,(uchar)r));
        }
        //計算處理時間并輸出到窗口中
        double dTime=(double)getTickCount();
        //進(jìn)行分水嶺算法
        watershed(srcImage, maskImage);
        dTime=(double)getTickCount()-dTime;
        printf("\t 處理時間=%gms\n",dTime*1000./getTickFrequency());
        //雙層循環(huán)，將分水嶺圖像遍歷存入watershedImage中
        Mat watershedImage(maskImage.size(),CV_8UC3);
        for(i=0;i<maskImage.rows;++i)
          for(j=0;j<maskImage.cols;++j)
          {
            int index=maskImage.at<int>(i,j);
            if(index==-1)
              watershedImage.at<Vec3b>(i,j)=Vec3b(255,255,255);//圖像變白色
            else if(index<=0||index>compCount)
              watershedImage.at<Vec3b>(i,j)=Vec3b(0,0,0);//圖像變黑色
            else
              watershedImage.at<Vec3b>(i,j)=colorTab[index-1];
          }
        //混合灰度圖和分水嶺效果圖并顯示最終的窗口
        watershedImage=watershedImage*0.5+grayImage*0.5;
        imshow("image[watershed]",watershedImage);
    }
  }
    return 0;
}

//回調(diào)函數(shù)定義
void on_Mouse(int event,int x,int y,int flags,void*)
{
  //處理鼠標(biāo)不在窗口中的情況
  if(x<0||x>=g_srcImage.cols||y<0||y>=g_srcImage.rows)
    return;

  //處理鼠標(biāo)左鍵相關(guān)消息
  if(event==EVENT_LBUTTONUP||!(flags & EVENT_FLAG_LBUTTON))//按下左鍵
    prevPt=Point(-1,-1);
  else if(event==EVENT_LBUTTONDOWN)//松開左鍵
    prevPt=Point(x,y);//鼠標(biāo)所指的位置

  //鼠標(biāo)左鍵按下并移動，繪制出白色線條
  else if(event==EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
  {
    Point pt(x,y);
    if(prevPt.x<0)//如果指出去了，返回
      prevPt=pt;
    line(g_maskImage, prevPt, pt, Scalar::all(255),2,8,0);//畫白線
    line(g_srcImage,prevPt,pt,Scalar::all(255),2,8,0);//畫白線
    prevPt=pt;
    imshow(WINDOW_NAME, g_srcImage);

  }
}

//計算處理時間并輸出到窗口中
        double dTime=(double)getTickCount();
        //進(jìn)行分水嶺算法
        watershed(srcImage, maskImage);
        dTime=(double)getTickCount()-dTime;
        printf("\t 處理時間=%gms\n",dTime*1000./getTickFrequency());

對于單通道的圖像：image.at<uchar>(i, j)
對于RGB通道的圖像：image.at<Vec3b>(i, j)[0] 
         image.at<Vec3b>(i, j)[1] 
         image.at<Vec3b>(i, j)[2]