Update X11 Grabber (scaling improvements) (#191)

* Add transform Matrix for scaling

* Add Image scaling by XRender

* Coding Style correction
This commit is contained in:
Paulchen-Panther 2016-08-22 11:59:07 +02:00 committed by redPanther
parent c8318ff0c7
commit 6acf68c87e
2 changed files with 185 additions and 85 deletions

View File

@ -73,6 +73,10 @@ private:
Picture _srcPicture; Picture _srcPicture;
Picture _dstPicture; Picture _dstPicture;
XTransform _transform;
int _horizontalDecimation;
int _verticalDecimation;
unsigned _screenWidth; unsigned _screenWidth;
unsigned _screenHeight; unsigned _screenHeight;
unsigned _croppedWidth; unsigned _croppedWidth;

View File

@ -6,29 +6,29 @@
// X11Grabber includes // X11Grabber includes
#include <grabber/X11Grabber.h> #include <grabber/X11Grabber.h>
X11Grabber::X11Grabber(bool useXGetImage, int cropLeft, int cropRight, int cropTop, int cropBottom, int horizontalPixelDecimation, int verticalPixelDecimation) : X11Grabber::X11Grabber(bool useXGetImage, int cropLeft, int cropRight, int cropTop, int cropBottom, int horizontalPixelDecimation, int verticalPixelDecimation)
_imageResampler(), : _imageResampler()
_useXGetImage(useXGetImage), , _useXGetImage(useXGetImage)
_cropLeft(cropLeft), , _cropLeft(cropLeft)
_cropRight(cropRight), , _cropRight(cropRight)
_cropTop(cropTop), , _cropTop(cropTop)
_cropBottom(cropBottom), , _cropBottom(cropBottom)
_x11Display(nullptr), , _x11Display(nullptr)
_pixmap(None), , _pixmap(None)
_srcFormat(nullptr), , _srcFormat(nullptr)
_dstFormat(nullptr), , _dstFormat(nullptr)
_srcPicture(None), , _srcPicture(None)
_dstPicture(None), , _dstPicture(None)
_screenWidth(0), , _horizontalDecimation(horizontalPixelDecimation)
_screenHeight(0), , _verticalDecimation(verticalPixelDecimation)
_croppedWidth(0), , _screenWidth(0)
_croppedHeight(0), , _screenHeight(0)
_image(0,0), , _croppedWidth(0)
_log(Logger::getInstance("X11GRABBER")) , _croppedHeight(0)
, _image(0,0)
, _log(Logger::getInstance("X11GRABBER"))
{ {
_imageResampler.setHorizontalPixelDecimation(horizontalPixelDecimation); _imageResampler.setCropping(0, 0, 0, 0); // cropping is performed by XRender, XShmGetImage or XGetImage
_imageResampler.setVerticalPixelDecimation(verticalPixelDecimation);
_imageResampler.setCropping(0, 0, 0, 0); // cropping is performed by XShmGetImage or XGetImage
memset(&_pictAttr, 0, sizeof(_pictAttr)); memset(&_pictAttr, 0, sizeof(_pictAttr));
_pictAttr.repeat = RepeatNone; _pictAttr.repeat = RepeatNone;
} }
@ -51,12 +51,14 @@ void X11Grabber::freeResources()
{ {
// Cleanup allocated resources of the X11 grab // Cleanup allocated resources of the X11 grab
XDestroyImage(_xImage); XDestroyImage(_xImage);
if(_XShmAvailable && !_useXGetImage) { if(_XShmAvailable && !_useXGetImage)
{
XShmDetach(_x11Display, &_shminfo); XShmDetach(_x11Display, &_shminfo);
shmdt(_shminfo.shmaddr); shmdt(_shminfo.shmaddr);
shmctl(_shminfo.shmid, IPC_RMID, 0); shmctl(_shminfo.shmid, IPC_RMID, 0);
} }
if (_XRenderAvailable && !_useXGetImage) { if (_XRenderAvailable && !_useXGetImage)
{
XRenderFreePicture(_x11Display, _srcPicture); XRenderFreePicture(_x11Display, _srcPicture);
XRenderFreePicture(_x11Display, _dstPicture); XRenderFreePicture(_x11Display, _dstPicture);
XFreePixmap(_x11Display, _pixmap); XFreePixmap(_x11Display, _pixmap);
@ -65,29 +67,31 @@ void X11Grabber::freeResources()
void X11Grabber::setupResources() void X11Grabber::setupResources()
{ {
if(_XShmAvailable && !_useXGetImage) { if(_XShmAvailable && !_useXGetImage)
{
_xImage = XShmCreateImage(_x11Display, _windowAttr.visual, _xImage = XShmCreateImage(_x11Display, _windowAttr.visual,
_windowAttr.depth, ZPixmap, NULL, &_shminfo, _windowAttr.depth, ZPixmap, NULL, &_shminfo,
_croppedWidth, _croppedHeight); _croppedWidth, _croppedHeight);
_shminfo.shmid = shmget(IPC_PRIVATE, _xImage->bytes_per_line * _xImage->height, IPC_CREAT|0777); _shminfo.shmid = shmget(IPC_PRIVATE, _xImage->bytes_per_line * _xImage->height, IPC_CREAT|0777);
_xImage->data = (char*)shmat(_shminfo.shmid,0,0); _xImage->data = (char*)shmat(_shminfo.shmid,0,0);
_shminfo.shmaddr = _xImage->data; _shminfo.shmaddr = _xImage->data;
_shminfo.readOnly = False; _shminfo.readOnly = False;
XShmAttach(_x11Display, &_shminfo);
XShmAttach(_x11Display, &_shminfo);
} }
if (_XRenderAvailable && !_useXGetImage) { if (_XRenderAvailable && !_useXGetImage)
if(_XShmPixmapAvailable) { {
if(_XShmPixmapAvailable)
{
_pixmap = XShmCreatePixmap(_x11Display, _window, _xImage->data, &_shminfo, _croppedWidth, _croppedHeight, _windowAttr.depth); _pixmap = XShmCreatePixmap(_x11Display, _window, _xImage->data, &_shminfo, _croppedWidth, _croppedHeight, _windowAttr.depth);
} else { }
else
{
_pixmap = XCreatePixmap(_x11Display, _window, _croppedWidth, _croppedHeight, _windowAttr.depth); _pixmap = XCreatePixmap(_x11Display, _window, _croppedWidth, _croppedHeight, _windowAttr.depth);
} }
_srcFormat = XRenderFindVisualFormat(_x11Display, _windowAttr.visual); _srcFormat = XRenderFindVisualFormat(_x11Display, _windowAttr.visual);
_dstFormat = XRenderFindVisualFormat(_x11Display, _windowAttr.visual); _dstFormat = XRenderFindVisualFormat(_x11Display, _windowAttr.visual);
_srcPicture = XRenderCreatePicture(_x11Display, _window, _srcFormat, CPRepeat, &_pictAttr); _srcPicture = XRenderCreatePicture(_x11Display, _window, _srcFormat, CPRepeat, &_pictAttr);
_dstPicture = XRenderCreatePicture(_x11Display, _pixmap, _dstFormat, CPRepeat, &_pictAttr); _dstPicture = XRenderCreatePicture(_x11Display, _pixmap, _dstFormat, CPRepeat, &_pictAttr);
XRenderSetPictureFilter(_x11Display, _srcPicture, "bilinear", NULL, 0);
} }
} }
@ -97,9 +101,12 @@ bool X11Grabber::Setup()
if (_x11Display == nullptr) if (_x11Display == nullptr)
{ {
Error(_log, "Unable to open display"); Error(_log, "Unable to open display");
if (getenv("DISPLAY")) { if (getenv("DISPLAY"))
{
Error(_log, "%s",getenv("DISPLAY")); Error(_log, "%s",getenv("DISPLAY"));
} else { }
else
{
Error(_log, "DISPLAY environment variable not set"); Error(_log, "DISPLAY environment variable not set");
} }
return false; return false;
@ -114,6 +121,10 @@ bool X11Grabber::Setup()
XShmQueryVersion(_x11Display, &dummy, &dummy, &pixmaps_supported); XShmQueryVersion(_x11Display, &dummy, &dummy, &pixmaps_supported);
_XShmPixmapAvailable = pixmaps_supported && XShmPixmapFormat(_x11Display) == ZPixmap; _XShmPixmapAvailable = pixmaps_supported && XShmPixmapFormat(_x11Display) == ZPixmap;
// Image scaling is performed by XRender when available, otherwise by ImageResampler
_imageResampler.setHorizontalPixelDecimation(_XRenderAvailable ? 1 : _horizontalDecimation);
_imageResampler.setVerticalPixelDecimation(_XRenderAvailable ? 1 : _verticalDecimation);
return true; return true;
} }
@ -121,32 +132,68 @@ Image<ColorRgb> & X11Grabber::grab()
{ {
updateScreenDimensions(); updateScreenDimensions();
if (_XRenderAvailable && !_useXGetImage) { if (_XRenderAvailable && !_useXGetImage)
XRenderComposite( _x11Display, // *dpy, {
PictOpSrc, // op, double scale_x = static_cast<double>(_windowAttr.width / _horizontalDecimation) / static_cast<double>(_windowAttr.width);
_srcPicture, // src double scale_y = static_cast<double>(_windowAttr.height / _verticalDecimation) / static_cast<double>(_windowAttr.height);
None, // mask double scale = std::min(scale_y, scale_x);
_dstPicture, // dst
_cropLeft, // src_x _transform =
_cropTop, // src_y {
0, // mask_x {
0, // mask_y {
0, // dst_x XDoubleToFixed(1),
0, // dst_y XDoubleToFixed(0),
_croppedWidth, // width XDoubleToFixed(0)
_croppedHeight); // height },
{
XDoubleToFixed(0),
XDoubleToFixed(1),
XDoubleToFixed(0)
},
{
XDoubleToFixed(0),
XDoubleToFixed(0),
XDoubleToFixed(scale)
}
}
};
XRenderSetPictureTransform (_x11Display, _srcPicture, &_transform);
XRenderSetPictureFilter(_x11Display, _srcPicture, FilterBilinear, NULL, 0);
XRenderComposite( _x11Display, // dpy
PictOpSrc, // op
_srcPicture, // src
None, // mask
_dstPicture, // dst
_cropLeft / _horizontalDecimation, // src_x _cropLeft
_cropTop / _verticalDecimation, // src_y _cropTop
0, // mask_x
0, // mask_y
0, // dst_x
0, // dst_y
_croppedWidth, // width
_croppedHeight); // height
XSync(_x11Display, False); XSync(_x11Display, False);
if (_XShmAvailable) { if (!_XShmAvailable)
{
XShmGetImage(_x11Display, _pixmap, _xImage, 0, 0, AllPlanes); XShmGetImage(_x11Display, _pixmap, _xImage, 0, 0, AllPlanes);
} else { }
else
{
_xImage = XGetImage(_x11Display, _pixmap, 0, 0, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap); _xImage = XGetImage(_x11Display, _pixmap, 0, 0, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap);
} }
} else { }
else
{
if (_XShmAvailable && !_useXGetImage) { if (_XShmAvailable && !_useXGetImage) {
XShmGetImage(_x11Display, _window, _xImage, _cropLeft, _cropTop, AllPlanes); XShmGetImage(_x11Display, _window, _xImage, _cropLeft, _cropTop, AllPlanes);
} else { }
else
{
_xImage = XGetImage(_x11Display, _window, _cropLeft, _cropTop, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap); _xImage = XGetImage(_x11Display, _window, _cropLeft, _cropTop, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap);
} }
} }
@ -164,32 +211,68 @@ Image<ColorRgb> & X11Grabber::grab()
int X11Grabber::grabFrame(Image<ColorRgb> & image) int X11Grabber::grabFrame(Image<ColorRgb> & image)
{ {
if (_XRenderAvailable && !_useXGetImage) { if (_XRenderAvailable && !_useXGetImage)
XRenderComposite( _x11Display, // *dpy, {
PictOpSrc, // op, double scale_x = static_cast<double>(_windowAttr.width / _horizontalDecimation) / static_cast<double>(_windowAttr.width);
_srcPicture, // src double scale_y = static_cast<double>(_windowAttr.height / _verticalDecimation) / static_cast<double>(_windowAttr.height);
None, // mask double scale = std::min(scale_y, scale_x);
_dstPicture, // dst
_cropLeft, // src_x _transform =
_cropTop, // src_y {
0, // mask_x {
0, // mask_y {
0, // dst_x XDoubleToFixed(1),
0, // dst_y XDoubleToFixed(0),
_croppedWidth, // width XDoubleToFixed(0)
_croppedHeight); // height },
{
XDoubleToFixed(0),
XDoubleToFixed(1),
XDoubleToFixed(0)
},
{
XDoubleToFixed(0),
XDoubleToFixed(0),
XDoubleToFixed(scale)
}
}
};
XRenderSetPictureTransform (_x11Display, _srcPicture, &_transform);
XRenderSetPictureFilter(_x11Display, _srcPicture, FilterBilinear, NULL, 0);
XRenderComposite( _x11Display, // dpy
PictOpSrc, // op
_srcPicture, // src
None, // mask
_dstPicture, // dst
_cropLeft / _horizontalDecimation, // src_x _cropLeft
_cropTop / _verticalDecimation, // src_y _cropTop
0, // mask_x
0, // mask_y
0, // dst_x
0, // dst_y
_croppedWidth, // width
_croppedHeight); // height
XSync(_x11Display, False); XSync(_x11Display, False);
if (_XShmAvailable) { if (!_XShmAvailable)
{
XShmGetImage(_x11Display, _pixmap, _xImage, 0, 0, AllPlanes); XShmGetImage(_x11Display, _pixmap, _xImage, 0, 0, AllPlanes);
} else { }
else
{
_xImage = XGetImage(_x11Display, _pixmap, 0, 0, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap); _xImage = XGetImage(_x11Display, _pixmap, 0, 0, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap);
} }
} else { }
else
{
if (_XShmAvailable && !_useXGetImage) { if (_XShmAvailable && !_useXGetImage) {
XShmGetImage(_x11Display, _window, _xImage, _cropLeft, _cropTop, AllPlanes); XShmGetImage(_x11Display, _window, _xImage, _cropLeft, _cropTop, AllPlanes);
} else { }
else
{
_xImage = XGetImage(_x11Display, _window, _cropLeft, _cropTop, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap); _xImage = XGetImage(_x11Display, _window, _cropLeft, _cropTop, _croppedWidth, _croppedHeight, AllPlanes, ZPixmap);
} }
} }
@ -220,7 +303,8 @@ int X11Grabber::updateScreenDimensions()
return 0; return 0;
} }
if (_screenWidth || _screenHeight) { if (_screenWidth || _screenHeight)
{
freeResources(); freeResources();
} }
@ -228,17 +312,29 @@ int X11Grabber::updateScreenDimensions()
_screenWidth = _windowAttr.width; _screenWidth = _windowAttr.width;
_screenHeight = _windowAttr.height; _screenHeight = _windowAttr.height;
_croppedWidth = (_screenWidth > unsigned(_cropLeft + _cropRight)) // Image scaling is performed by XRender when available, otherwise by ImageResampler
? (_screenWidth - _cropLeft - _cropRight) if (_XRenderAvailable && !_useXGetImage)
: _screenWidth; {
_croppedWidth = (_screenWidth > unsigned(_cropLeft + _cropRight))
? ((_screenWidth - _cropLeft - _cropRight) / _horizontalDecimation)
: _screenWidth / _horizontalDecimation;
_croppedHeight = (_screenHeight > unsigned(_cropTop + _cropBottom)) _croppedHeight = (_screenHeight > unsigned(_cropTop + _cropBottom))
? (_screenHeight - _cropTop - _cropBottom) ? ((_screenHeight - _cropTop - _cropBottom) / _verticalDecimation)
: _screenHeight; : _screenHeight / _verticalDecimation;
if (_XRenderAvailable && !_useXGetImage) {
Info(_log, "Using XRender for grabbing"); Info(_log, "Using XRender for grabbing");
} else { }
else
{
_croppedWidth = (_screenWidth > unsigned(_cropLeft + _cropRight))
? (_screenWidth - _cropLeft - _cropRight)
: _screenWidth;
_croppedHeight = (_screenHeight > unsigned(_cropTop + _cropBottom))
? (_screenHeight - _cropTop - _cropBottom)
: _screenHeight;
Info(_log, "Using XGetImage for grabbing"); Info(_log, "Using XGetImage for grabbing");
} }