commonimage.h

#ifndef COMMONIMAGE_H
#define COMMONIMAGE_H

#include <QImage>
#include <string>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>

enum CVConversionType
{
    TO_RGB,
    TO_YUV
};

class CommonImage : public QImage
{
    //Q_OBJECT
public:
    CommonImage() :
        QImage(),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(const char * const xpm[]) :
        QImage(xpm),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(const QImage &q) :
        QImage(q),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(QImage* q) :
        QImage(*q),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(const QSize &size, Format format) :
        QImage(size, format),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(const uchar *data, int width, int height, Format format, QImageCleanupFunction cleanupFunction, void *cleanupInfo) :
        QImage(data, width, height, format, cleanupFunction, cleanupInfo),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(const uchar *data, int width, int height, int bytesPerLine, Format format, QImageCleanupFunction cleanupFunction, void *cleanupInfo) :
        QImage(data, width, height, bytesPerLine, format, cleanupFunction, cleanupInfo),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(int width, int height, Format format) :
        QImage(width, height, format),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(uchar *data, int width, int height, Format format, QImageCleanupFunction cleanupFunction, void *cleanupInfo) :
        QImage(data, width, height, format, cleanupFunction, cleanupInfo),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(uchar *data, int width, int height, int bytesPerLine, Format format, QImageCleanupFunction cleanupFunction, void *cleanupInfo) :
        QImage(data, width, height, bytesPerLine, format, cleanupFunction, cleanupInfo),
        is_camera_beingused(false), is_file_beingused(false)
    {}
    CommonImage(cv::Mat& source);
    CommonImage(const QString &fileName, const char *format);
    CommonImage(int devNumber);

    ~CommonImage();

    using QImage::operator =;

    // Warning. These functions allocate new data.
    static QImage       formatconv_CV2QT(const cv::Mat& input);
    static cv::Mat      formatconv_QT2CV(const QImage input);
    static cv::Mat      formatconv_QT2CV_YUV(const QImage input);

    cv::Mat             toCVMat();
    void                toCVMat(cv::Mat& output);
    cv::Mat             toCVYUVMat();

    void                applyCVMask(const cv::Mat& mask);

    bool                captureFromCamera(int device);
    bool                captureFromVideoFile(std::string fn);
    double              getCaptureAttribute(int propid);
    int                 getCaptureFPS();
    int                 getCaptureFrameInterval();
    bool                captureSkipToTime(double secs);

    bool                grabCap();
    void                closeCap();

    void                convertFromMat(const cv::Mat& input);

    CommonImage&        operator = (const cv::Mat input);

protected:
    void                cleanseCommonImageData();
    void                cleanseQImageData(int w = 0, int h = 0, Format fmt = QImage::Format_Invalid);

private:
    cv::Mat             cvimg;
    cv::VideoCapture    vcapturer;

    std::string         capt_filename;

    bool                is_camera_beingused;
    bool                is_file_beingused;
    bool                is_mat_beingedited;

signals:
    
public slots:
    
};

#endif // COMMONIMAGE_H