imagepro.h

Ir a la documentación de este archivo.

#ifndef IMAGEPRO_H
#define IMAGEPRO_H

#include <QtWidgets>
#include "opencv2/opencv.hpp"
#include "opencv2/highgui/highgui.hpp"

namespace
{
inline QImage mat_2_qimage_ref_policy(cv::Mat &mat, QImage::Format format)
{
    return QImage(mat.data, mat.cols, mat.rows, mat.step, format);
}

inline QImage mat_2_qimage_cpy_policy(cv::Mat const &mat, QImage::Format format)
{
    return QImage(mat.data, mat.cols, mat.rows, mat.step, format).copy();
}

struct qimage_2_mat_cpy_policy
{
    static cv::Mat start(QImage const &img, int format)
    {
        return cv::Mat(img.height(), img.width(), format, const_cast<uchar*>(img.bits()), img.bytesPerLine()).clone();
    }
};

struct qimage_2_mat_ref_policy
{
    static cv::Mat start(QImage &img, int format)
    {
        return cv::Mat(img.height(), img.width(), format, img.bits(), img.bytesPerLine());
    }
};

template<typename Policy>
struct qimage_2_mat
{
    template<typename Image>
    static cv::Mat run(Image &img, bool swap);
};

template<typename Policy>
template<typename Image>
cv::Mat qimage_2_mat<Policy>::run(Image &img, bool swap)
{
    if(img.isNull()){
        return cv::Mat();
    }

    switch (img.format()) {
    case QImage::Format_RGB888:{
        cv::Mat result = Policy::start(img, CV_8UC3);
        if(swap){
            cv::cvtColor(result, result, CV_RGB2BGR);
        }
        return result;
    }
    case QImage::Format_Indexed8:{
        return Policy::start(img, CV_8U);
    }
    case QImage::Format_RGB32:
    case QImage::Format_ARGB32:
    case QImage::Format_ARGB32_Premultiplied:{
        return Policy::start(img, CV_8UC4);
    }
    default:
        break;
    }

    return cv::Mat();
}
}

using namespace cv;

class ImagePro
{

public:
    ImagePro();

    QImage mat_2_qimage_cpy(Mat const &mat, bool swap);
    Mat qimage_2_mat_cpy(QImage const &img, bool swap);
};

#endif // IMAGEPRO_H