CVRawImage.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__EXAMPLE__CV_RAW_IMAGE_H_
#define __PBIO_API__PBIO__EXAMPLE__CV_RAW_IMAGE_H_

#include <opencv2/opencv.hpp>

#include "pbio/IRawImage.h"

namespace pbio
{

class CVRawImage : public pbio::IRawImage
{
public:

    CVRawImage(){}

    virtual ~CVRawImage(){}

    virtual const unsigned char* data() const;


    virtual int32_t width() const;

    virtual int32_t height() const;


    virtual int32_t format() const;


    cv::Mat& mat();


    const cv::Mat& mat() const;


private:
    cv::Mat _mat;
};

}  // pbio namespace





namespace pbio
{

inline
const unsigned char* CVRawImage::data() const
{
    if( _mat.empty() ||
        _mat.dims != 2 ||
        ( _mat.type() != CV_8UC1 && _mat.type() != CV_8UC3 ) ||
        !_mat.isContinuous() )
    {
        return NULL;
    }

    return _mat.data;
}

inline
int32_t CVRawImage::width() const
{
    return _mat.cols;
}


inline
int32_t CVRawImage::height() const
{
    return _mat.rows;
}


inline
int32_t CVRawImage::format() const
{
    if(_mat.type() == CV_8UC1)
    {
        return (int32_t) FORMAT_GRAY;
    }

    if(_mat.type() == CV_8UC3)
    {
        return (int32_t) FORMAT_BGR;
    }

    return -1;
}



inline
cv::Mat& CVRawImage::mat()
{
    return _mat;
}

inline
const cv::Mat& CVRawImage::mat() const
{
    return _mat;
}

}  // pbio namespace

#endif  // __PBIO_API__PBIO__EXAMPLE__CV_RAW_IMAGE_H_