Move V4L2 grabber code to separate library

Former-commit-id: f3003eb0142af3d085ccf93fff1b297ebc2321fc

libsrc/grabber/dispmanx/CMakeLists.txt

@@ -0,0 +1,36 @@
+
+# Find the BCM-package (VC control)
+find_package(BCM REQUIRED)
+include_directories(${BCM_INCLUDE_DIRS})
+
+# Define the current source locations
+SET(CURRENT_HEADER_DIR ${CMAKE_SOURCE_DIR}/include/grabber)
+SET(CURRENT_SOURCE_DIR ${CMAKE_SOURCE_DIR}/libsrc/grabber/dispmanx)
+
+# Group the headers that go through the MOC compiler
+SET(DispmanxGrabberQT_HEADERS
+		${CURRENT_HEADER_DIR}/DispmanxWrapper.h
+)
+
+SET(DispmanxGrabberHEADERS
+		${CURRENT_SOURCE_DIR}/DispmanxFrameGrabber.h
+)
+
+SET(DispmanxGrabberSOURCES
+		${CURRENT_SOURCE_DIR}/DispmanxWrapper.cpp
+		${CURRENT_SOURCE_DIR}/DispmanxFrameGrabber.cpp
+)
+
+QT4_WRAP_CPP(DispmanxGrabberHEADERS_MOC ${DispmanxGrabberQT_HEADERS})
+
+add_library(dispmanx-grabber
+		${DispmanxGrabberHEADERS}
+		${DispmanxGrabberQT_HEADERS}
+		${DispmanxGrabberHEADERS_MOC}
+		${DispmanxGrabberSOURCES}
+)
+
+target_link_libraries(dispmanx-grabber
+		hyperion
+		${QT_LIBRARIES}
+		${BCM_LIBRARIES})

libsrc/grabber/dispmanx/DispmanxFrameGrabber.cpp

@@ -0,0 +1,101 @@
+
+// STL includes
+#include <cassert>
+#include <iostream>
+
+// Local includes
+#include "DispmanxFrameGrabber.h"
+
+DispmanxFrameGrabber::DispmanxFrameGrabber(const unsigned width, const unsigned height) :
+	_vc_display(0),
+	_vc_resource(0),
+	_vc_flags(0),
+	_width(width),
+	_height(height)
+{
+	// Initiase BCM
+	bcm_host_init();
+
+	{
+		// Check if the display can be opened and display the current resolution
+		// Open the connection to the display
+		_vc_display = vc_dispmanx_display_open(0);
+		assert(_vc_display > 0);
+
+		// Obtain the display information
+		DISPMANX_MODEINFO_T vc_info;
+		int result = vc_dispmanx_display_get_info(_vc_display, &vc_info);
+		// Keep compiler happy in 'release' mode
+		(void)result;
+		assert(result == 0);
+		std::cout << "Display opened with resolution: " << vc_info.width << "x" << vc_info.height << std::endl;
+
+		// Close the displaye
+		vc_dispmanx_display_close(_vc_display);
+	}
+
+	// Create the resources for capturing image
+	uint32_t vc_nativeImageHandle;
+	_vc_resource = vc_dispmanx_resource_create(
+			VC_IMAGE_RGBA32,
+			width,
+			height,
+			&vc_nativeImageHandle);
+	assert(_vc_resource);
+
+	// Define the capture rectangle with the same size
+	vc_dispmanx_rect_set(&_rectangle, 0, 0, width, height);
+}
+
+DispmanxFrameGrabber::~DispmanxFrameGrabber()
+{
+	// Clean up resources
+	vc_dispmanx_resource_delete(_vc_resource);
+
+	// De-init BCM
+	bcm_host_deinit();
+}
+
+void DispmanxFrameGrabber::setFlags(const int vc_flags)
+{
+	_vc_flags = vc_flags;
+}
+
+void DispmanxFrameGrabber::setVideoMode(const VideoMode videoMode)
+{
+	switch (videoMode) {
+	case VIDEO_3DSBS:
+		vc_dispmanx_rect_set(&_rectangle, 0, 0, _width/2, _height);
+		break;
+	case VIDEO_3DTAB:
+		vc_dispmanx_rect_set(&_rectangle, 0, 0, _width, _height/2);
+		break;
+	case VIDEO_2D:
+	default:
+		vc_dispmanx_rect_set(&_rectangle, 0, 0, _width, _height);
+		break;
+	}
+}
+
+void DispmanxFrameGrabber::grabFrame(Image<ColorRgba> & image)
+{
+	// resize the given image if needed
+	if (image.width() != unsigned(_rectangle.width) || image.height() != unsigned(_rectangle.height))
+	{
+		image.resize(_rectangle.width, _rectangle.height);
+	}
+
+	// Open the connection to the display
+	_vc_display = vc_dispmanx_display_open(0);
+
+	// Create the snapshot (incl down-scaling)
+	vc_dispmanx_snapshot(_vc_display, _vc_resource, (DISPMANX_TRANSFORM_T) _vc_flags);
+
+	// Read the snapshot into the memory
+	void* image_ptr = image.memptr();
+	const unsigned destPitch = _rectangle.width * sizeof(ColorRgba);
+	vc_dispmanx_resource_read_data(_vc_resource, &_rectangle, image_ptr, destPitch);
+
+	// Close the displaye
+	vc_dispmanx_display_close(_vc_display);
+}

libsrc/grabber/dispmanx/DispmanxFrameGrabber.h

@@ -0,0 +1,71 @@
+#pragma once
+
+// BCM includes
+#pragma GCC system_header
+#include <bcm_host.h>
+
+// STL includes
+#include <cstdint>
+
+// Utils includes
+#include <utils/Image.h>
+#include <utils/ColorRgba.h>
+#include <utils/VideoMode.h>
+
+///
+/// The DispmanxFrameGrabber is used for creating snapshots of the display (screenshots) with a
+/// downsized and scaled resolution.
+///
+class DispmanxFrameGrabber
+{
+public:
+	///
+	/// Construct a DispmanxFrameGrabber that will capture snapshots with specified dimensions.
+	///
+	/// @param[in] width  The width of the captured screenshot
+	/// @param[in] height The heigth of the captured screenshot
+	///
+	DispmanxFrameGrabber(const unsigned width, const unsigned height);
+	~DispmanxFrameGrabber();
+
+	///
+	/// Updates the frame-grab flags as used by the VC library for frame grabbing
+	///
+	/// @param vc_flags  The snapshot grabbing mask
+	///
+	void setFlags(const int vc_flags);
+
+	///
+	/// Set the video mode (2D/3D)
+	/// @param[in] mode The new video mode
+	///
+	void setVideoMode(const VideoMode videoMode);
+
+	///
+	/// Captures a single snapshot of the display and writes the data to the given image. The
+	/// provided image should have the same dimensions as the configured values (_width and
+	/// _height)
+	///
+	/// @param[out] image  The snapped screenshot (should be initialized with correct width and
+	/// height)
+	///
+	void grabFrame(Image<ColorRgba> & image);
+
+private:
+	/// Handle to the display that is being captured
+	DISPMANX_DISPLAY_HANDLE_T _vc_display;
+
+	/// Handle to the resource for storing the captured snapshot
+	DISPMANX_RESOURCE_HANDLE_T _vc_resource;
+
+	/// Rectangle of the captured resource that is transfered to user space
+	VC_RECT_T _rectangle;
+
+	/// Flags (transforms) for creating snapshots
+	int _vc_flags;
+
+	/// With of the captured snapshot [pixels]
+	const unsigned _width;
+	/// Height of the captured snapshot [pixels]
+	const unsigned _height;
+};

libsrc/grabber/dispmanx/DispmanxWrapper.cpp

@@ -0,0 +1,88 @@
+// 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 "DispmanxFrameGrabber.h"
+
+
+DispmanxWrapper::DispmanxWrapper(const unsigned grabWidth, const unsigned grabHeight, const unsigned updateRate_Hz, Hyperion * hyperion) :
+	_updateInterval_ms(1000/updateRate_Hz),
+	_timeout_ms(2 * _updateInterval_ms),
+	_priority(1000),
+	_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);
+
+	// 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);
+
+	_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);
+}