OpenCV的Mat图像转QImage

QImage Mat2QImage(const cv::Mat& mat)
{
    // 按照Qt帮助文档说明，像素buffer必须在QImage的生命周期内保持有效，且它默认是不会去释放像素buffer，QImage对象之间采用引用计数实现隐式共享底层数据
    if (mat.type() == CV_8UC1)                    // 单通道
    {
        QImage image(mat.cols, mat.rows, QImage::Format_Indexed8);
        image.setColorCount(256);                // 设置灰度级
        for (int i = 0; i < 256; i++)            // 设置颜色表
        {
            image.setColor(i, qRgb(i, i, i));
        }
        const uchar* pSrc = mat.data;    
        for (int row = 0; row < mat.rows; row++)
        {
            uchar *pDest = image.scanLine(row);
            memcpy(pDest, pSrc, mat.cols);
            pSrc += mat.step;
        }

        return image;
    }
    else if (mat.type() == CV_8UC3)                // 三通道
    {
#if 0
        cv::Mat rgbMat;
        cv::cvtColor(mat, rgbMat, cv::COLOR_BGR2RGB);
        // 如果不返回副本，mat在QImage对象生命周期结束之前析构了，则使用QImage对象会崩溃
        return QImage(rgbMat.data, rgbMat.cols, rgbMat.rows, rgbMat.step, QImage::Format_RGB888).copy();
#endif

#if 1
        QImage image(mat.data, mat.cols, mat.rows, mat.cols*mat.channels(), QImage::Format_RGB888);

        // 因为Mat内部的颜色顺序默认是BGR，也可以在构造之前使用cv::cvtColor(mat, Rgb, CV_BGR2RGB);//颜色空间转换
        return image.rgbSwapped();// 返回转换之后的副本，不会改变原来的对象
#endif
    }
    else if (mat.type() == CV_8UC4)
    {
        // mat.step表示以字节为单位的矩阵的宽度，等价于mat.cols*mat.channels()
        QImage image(mat.data, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);   
        return image.copy();// 深拷贝
    }
    else
    {
        return QImage();
    }
}