Qt QImageとOpenCVのMatタイプの相互変換(改良)


Qt QImageとOpenCVのMatタイプの相互変換(2)
私の以前のブログでは、実は変換方法を紹介していました.https://blog.csdn.net/liyuanbhu/article/details/46662115
しかし、そのブログのコードは十分ではありません.主に、変換後の画像と変換前の画像がメモリデータを共有しているかどうかの比較が乱れている.共有メモリの場合もあれば、共有メモリがない場合もあります.コードの説明もはっきりしていない.私のこのコードを使っている人に迷惑をかけるかもしれません.だからまたブログを書きました.コードをさらに改善する.
このバージョンのコードは、メモリを共有するかどうかを制御したり、RGS 24形式の画像に対して、RとBの位置を交換するかどうかを制御したりすることができます.この2つの改善により、いくつかのアプリケーションシーンでコードの実行速度を大幅に向上させることができます.
コードは次のとおりです.
/****************************************************************************
** file: convert.h
** brief: cv::Mat   QImage        。  Qt       OpenCV    。
** Copyright (C) LiYuan
** Author: LiYuan
** E-Mail: 18069211#qq(.)com
** Version 2.0.1
** Last modified: 2019.01.11
** Modified By: LiYuan
****************************************************************************/


#ifndef QT_OPENCV_CONVERT_H
#define QT_OPENCV_CONVERT_H

#include 
#include 
#include "opencv2/core/core.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/opencv.hpp"
/**
 * @brief   OpenCV   cv::Mat         QImage   
 * @param mat       ,   CV_8UC1、CV_8UC3、CV_8UC4   OpenCV      
 * @param clone true     Mat      ,      mat         ,false     mat     
 * @param rb_swap     CV_8UC3   ,   true      R   B RGB->BGR,                  
 * @return      QImage   
*/
QImage cvMat2QImage(const cv::Mat& mat, bool clone = true, bool rb_swap = true);

/**
 * @brief   QImage          cv::Mat   
 * @param image       ,   Format_Indexed8/Format_Grayscale、24    、32      ,
 * @param clone true     QImage      ,      mat         ,false     QImage     
 * @param rg_swap     RGB888   ,   true      R   B RGB->BGR,                  
 * @return      cv::Mat   
*/
cv::Mat QImage2cvMat(QImage &image, bool clone = false, bool rb_swap = true);

#endif


/****************************************************************************
** file: convert.cpp
** brief: cv::Mat   QImage        。  Qt       OpenCV    。
** Copyright (C) LiYuan
** Author: LiYuan
** E-Mail: 18069211#qq(.)com
** Version 2.0.1
** Last modified: 2019.01.11
** Modified By: LiYuan
****************************************************************************/

#include "convert.h"

QImage cvMat2QImage(const cv::Mat& mat, bool clone, bool rb_swap)
{
    const uchar *pSrc = (const uchar*)mat.data;
    // 8-bits unsigned, NO. OF CHANNELS = 1
    if(mat.type() == CV_8UC1)
    {
        //QImage image(mat.cols, mat.rows, QImage::Format_Grayscale8);
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_Grayscale8);
        if(clone) return image.copy();
        return image;
    }
    // 8-bits unsigned, NO. OF CHANNELS = 3
    else if(mat.type() == CV_8UC3)
    {
        // Create QImage with same dimensions as input Mat
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
        if(clone)
        {
            if(rb_swap) return image.rgbSwapped();
            return image.copy();
        }
        else
        {
            if(rb_swap)
            {
                cv::cvtColor(mat, mat, CV_BGR2RGB);
            }
            return image;
        }

    }
    else if(mat.type() == CV_8UC4)
    {
        qDebug() << "CV_8UC4";
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);
        if(clone) return image.copy();
        return image;
    }
    else
    {
        qDebug() << "ERROR: Mat could not be converted to QImage.";
        return QImage();
    }
}

cv::Mat QImage2cvMat(QImage &image, bool clone, bool rb_swap)
{
    cv::Mat mat;
    //qDebug() << image.format();
    switch(image.format())
    {
    case QImage::Format_ARGB32:
    case QImage::Format_RGB32:
    case QImage::Format_ARGB32_Premultiplied:
        mat = cv::Mat(image.height(), image.width(), CV_8UC4, (void *)image.constBits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        break;
    case QImage::Format_RGB888:
        mat = cv::Mat(image.height(), image.width(), CV_8UC3, (void *)image.constBits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        if(rb_swap) cv::cvtColor(mat, mat, CV_BGR2RGB);
        break;
    case QImage::Format_Indexed8:
    case QImage::Format_Grayscale8:
        mat = cv::Mat(image.height(), image.width(), CV_8UC1, (void *)image.bits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        break;
    }
    return mat;
}

皆さんのご利用を歓迎します.フィードバック使用中に発生した問題も歓迎します.