DirectX9 Grabber (#1039)

include/grabber/DirectXGrabber.h

@@ -0,0 +1,79 @@
+#pragma once
+
+#include <QObject>
+
+// DirectX 9 header
+#include <d3d9.h>
+#include <d3dx9.h>
+
+// Hyperion-utils includes
+#include <utils/ColorRgb.h>
+#include <hyperion/Grabber.h>
+
+///
+/// @brief The DirectX9 capture implementation
+///
+class DirectXGrabber : public Grabber
+{
+public:
+
+	DirectXGrabber(int cropLeft, int cropRight, int cropTop, int cropBottom, int pixelDecimation, int display);
+
+	virtual ~DirectXGrabber();
+
+	///
+	/// 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
+	///
+	virtual int grabFrame(Image<ColorRgb> & image);
+
+	///
+	/// @brief Set a new video mode
+	///
+	virtual void setVideoMode(VideoMode mode);
+
+	///
+	/// @brief Apply new width/height values, overwrite Grabber.h implementation
+	///
+	virtual bool setWidthHeight(int width, int height) { return true; };
+
+	///
+	/// @brief Apply new pixelDecimation
+	///
+	virtual void setPixelDecimation(int pixelDecimation);
+
+	///
+	/// Set the crop values
+	/// @param  cropLeft    Left pixel crop
+	/// @param  cropRight   Right pixel crop
+	/// @param  cropTop     Top pixel crop
+	/// @param  cropBottom  Bottom pixel crop
+	///
+	virtual void setCropping(unsigned cropLeft, unsigned cropRight, unsigned cropTop, unsigned cropBottom);
+
+private:
+	///
+	/// @brief Setup a new capture display, will free the previous one
+	/// @return True on success, false if no display is found
+	///
+	bool setupDisplay();
+
+	///
+	/// @brief free the _screen pointer
+	///
+	void freeResources();
+
+private:
+	int _pixelDecimation;
+	unsigned _displayWidth;
+	unsigned _displayHeight;
+	RECT* _srcRect;
+
+	IDirect3D9* _d3d9;
+	IDirect3DDevice9* _device;
+	IDirect3DSurface9* _surface;
+	IDirect3DSurface9* _surfaceDest;
+};

include/grabber/DirectXWrapper.h

@@ -0,0 +1,36 @@
+#pragma once
+
+#include <hyperion/GrabberWrapper.h>
+#include <grabber/DirectXGrabber.h>
+
+class DirectXWrapper: public GrabberWrapper
+{
+public:
+	///
+	/// Constructs the DirectX grabber with a specified grab size and update rate.
+	///
+	/// @param[in] cropLeft          Remove from left [pixels]
+	/// @param[in] cropRight 	     Remove from right [pixels]
+	/// @param[in] cropTop           Remove from top [pixels]
+	/// @param[in] cropBottom        Remove from bottom [pixels]
+	/// @param[in] pixelDecimation   Decimation factor for image [pixels]
+	/// @param[in] display           The display used[index]
+	/// @param[in] updateRate_Hz     The image grab rate [Hz]
+	///
+	DirectXWrapper(int cropLeft, int cropRight, int cropTop, int cropBottom, int pixelDecimation, int display, const unsigned updateRate_Hz);
+
+	///
+	/// Destructor of this DirectX grabber. Releases any claimed resources.
+	///
+	virtual ~DirectXWrapper() {};
+
+public slots:
+	///
+	/// Performs a single frame grab and computes the led-colors
+	///
+	virtual void action();
+
+private:
+	/// The actual grabber
+	DirectXGrabber _grabber;
+};