97 lines
2.3 KiB
C++
97 lines
2.3 KiB
C++
// QT includes
|
|
#include <QDebug>
|
|
#include <QDateTime>
|
|
|
|
// Hyperion includes
|
|
#include <hyperion/Hyperion.h>
|
|
#include <hyperion/ImageProcessorFactory.h>
|
|
#include <hyperion/ImageProcessor.h>
|
|
|
|
// Dispmanx grabber includes
|
|
#include <grabber/DispmanxWrapper.h>
|
|
#include <grabber/DispmanxFrameGrabber.h>
|
|
|
|
|
|
DispmanxWrapper::DispmanxWrapper(const unsigned grabWidth, const unsigned grabHeight, const unsigned updateRate_Hz, const int priority, Hyperion * hyperion) :
|
|
_updateInterval_ms(1000/updateRate_Hz),
|
|
_timeout_ms(2 * _updateInterval_ms),
|
|
_priority(priority),
|
|
_timer(),
|
|
_image(grabWidth, grabHeight),
|
|
_frameGrabber(new DispmanxFrameGrabber(grabWidth, grabHeight)),
|
|
_processor(ImageProcessorFactory::getInstance().newImageProcessor()),
|
|
_ledColors(hyperion->getLedCount(), ColorRgb{0,0,0}),
|
|
_hyperion(hyperion)
|
|
{
|
|
// Configure the timer to generate events every n milliseconds
|
|
_timer.setInterval(_updateInterval_ms);
|
|
_timer.setSingleShot(false);
|
|
|
|
_processor->setSize(grabWidth, grabHeight);
|
|
_forward = _hyperion->getForwarder()->protoForwardingEnabled();
|
|
|
|
// Connect the QTimer to this
|
|
QObject::connect(&_timer, SIGNAL(timeout()), this, SLOT(action()));
|
|
}
|
|
|
|
DispmanxWrapper::~DispmanxWrapper()
|
|
{
|
|
// Cleanup used resources (ImageProcessor and FrameGrabber)
|
|
delete _processor;
|
|
delete _frameGrabber;
|
|
}
|
|
|
|
void DispmanxWrapper::start()
|
|
{
|
|
// Start the timer with the pre configured interval
|
|
_timer.start();
|
|
}
|
|
|
|
void DispmanxWrapper::action()
|
|
{
|
|
// Grab frame into the allocated image
|
|
_frameGrabber->grabFrame(_image);
|
|
|
|
if ( _forward )
|
|
{
|
|
Image<ColorRgb> image_rgb;
|
|
_image.toRgb(image_rgb);
|
|
emit emitImage(_priority, image_rgb, _timeout_ms);
|
|
}
|
|
|
|
_processor->process(_image, _ledColors);
|
|
_hyperion->setColors(_priority, _ledColors, _timeout_ms);
|
|
}
|
|
|
|
void DispmanxWrapper::stop()
|
|
{
|
|
// Stop the timer, effectivly stopping the process
|
|
_timer.stop();
|
|
}
|
|
|
|
void DispmanxWrapper::setGrabbingMode(const GrabbingMode mode)
|
|
{
|
|
switch (mode)
|
|
{
|
|
case GRABBINGMODE_VIDEO:
|
|
_frameGrabber->setFlags(DISPMANX_SNAPSHOT_NO_RGB|DISPMANX_SNAPSHOT_FILL);
|
|
start();
|
|
break;
|
|
case GRABBINGMODE_AUDIO:
|
|
case GRABBINGMODE_PHOTO:
|
|
case GRABBINGMODE_MENU:
|
|
case GRABBINGMODE_INVALID:
|
|
_frameGrabber->setFlags(0);
|
|
start();
|
|
break;
|
|
case GRABBINGMODE_OFF:
|
|
stop();
|
|
break;
|
|
}
|
|
}
|
|
|
|
void DispmanxWrapper::setVideoMode(const VideoMode mode)
|
|
{
|
|
_frameGrabber->setVideoMode(mode);
|
|
}
|