#include <opencv2/opencv.hpp>
#include <iostream>
using namespace cv;
using namespace std;

Mat img;
int threshval = 150;

static void on_trackbar(int, void*) //滑动条回调函数
{
    Mat bw;
    threshold(img, bw, threshval, 255, THRESH_BINARY); //二值化，0、255
    imshow("bw", bw);//bw是0、255的“二值图”
    Mat labelImage;
    int nLabels = connectedComponents(bw, labelImage);//0、255的bw图变成0、1、2……不同区域标识的labelImage(32位短整型)
    std::vector<Vec3b> colors(nLabels); //colors有nLabels个元素，每个元素有3维，即nLabels个彩色像素
    colors[0] = Vec3b(0, 0, 0);//背景色，黑色
    for (int label = 1; label < nLabels; ++label)//label从1开始，因为0所在区域是背景
    {
        colors[label] = Vec3b((rand() & 255), (rand() & 255), (rand() & 255));//用于不同区域的随机彩色
    }
    Mat dst(img.size(), CV_8UC3); //与img等大的3通道的画板dst
    for (int r = 0; r < dst.rows; ++r)
    {
        for (int c = 0; c < dst.cols; ++c)
        {
            int label = (int)labelImage.at<int>(r, c);//at<int>中int对应labelImage中元素类型（短整型）
            dst.at<Vec3b>(r, c) = colors[label]; //dst画板上涂色，依据检索的连通域标识图labelImage中的值（0、1、2……）涂不同颜色
        }
    }
    imshow("Connected Components", dst);
}

int main(int argc, const char** argv)
{
    img = imread("lena.jpg", IMREAD_GRAYSCALE);//灰度读取
    imshow("Image", img);
    namedWindow("Connected Components", WINDOW_AUTOSIZE);//先窗体，后滑动条
    createTrackbar("Threshold", "Connected Components", &threshval, 255, on_trackbar);
    on_trackbar(threshval, 0);//先执行一次，出图。0无意义，装饰品

    waitKey(0);
    return 0;
}