DispmanxFrameGrabber.h

#pragma once

// BCM includes
#ifdef PLATFORM_RPI
    #pragma GCC system_header
    #include <bcm_host.h>
#else
    #include <grabber/DispmanxFrameGrabberMock.h>
#endif

// Utils includes
#include <utils/Image.h>
#include <utils/ColorRgba.h>
#include <hyperion/Grabber.h>

class DispmanxFrameGrabber : public Grabber
{
public:
    DispmanxFrameGrabber(const unsigned width, const unsigned height);
    ~DispmanxFrameGrabber();


    int grabFrame(Image<ColorRgb> & image);

    virtual bool setWidthHeight(int width, int height);

private:
    void setFlags(const int vc_flags);

    void freeResources();

    DISPMANX_DISPLAY_HANDLE_T _vc_display;

    DISPMANX_RESOURCE_HANDLE_T _vc_resource;

    VC_RECT_T _rectangle;

    int _vc_flags;

    // temp buffer when capturing with unsupported pitch size or
    // when we need to crop the image
    ColorRgba* _captureBuffer;

    // size of the capture buffer in Pixels
    unsigned _captureBufferSize;

    // rgba output buffer
    Image<ColorRgba>  _image_rgba;

};