hyperion.ng/include/grabber/DirectXGrabber.h
2020-10-26 21:13:13 +01:00

80 lines
1.8 KiB
C++

#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;
};