mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-25 01:30:38 +00:00
444 lines
12 KiB
C
444 lines
12 KiB
C
/*
|
|
* test-windows.c - Test GstVaapiWindow
|
|
*
|
|
* gstreamer-vaapi (C) 2010 Splitted-Desktop Systems
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <gst/vaapi/gstvaapidisplay_x11.h>
|
|
#include <gst/vaapi/gstvaapiwindow_x11.h>
|
|
#include <gst/vaapi/gstvaapisurface.h>
|
|
#include <gst/vaapi/gstvaapiimage.h>
|
|
|
|
static inline void pause(void)
|
|
{
|
|
g_print("Press any key to continue...\n");
|
|
getchar();
|
|
}
|
|
|
|
typedef void (*DrawRectFunc)(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
);
|
|
|
|
static void draw_rect_ARGB(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
guint i, j;
|
|
|
|
color = GUINT32_TO_BE(color);
|
|
|
|
for (j = 0; j < height; j++) {
|
|
guint32 *p = (guint32 *)(pixels[0] + (y + j) * stride[0] + x * 4);
|
|
for (i = 0; i < width; i++)
|
|
p[i] = color;
|
|
}
|
|
}
|
|
|
|
static void draw_rect_BGRA(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
// Converts ARGB color to BGRA
|
|
color = GUINT32_SWAP_LE_BE(color);
|
|
|
|
draw_rect_ARGB(pixels, stride, x, y, width, height, color);
|
|
}
|
|
|
|
static void draw_rect_RGBA(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
// Converts ARGB color to RGBA
|
|
color = ((color >> 24) & 0xff) | ((color & 0xffffff) << 8);
|
|
|
|
draw_rect_ARGB(pixels, stride, x, y, width, height, color);
|
|
}
|
|
|
|
static void draw_rect_ABGR(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
// Converts ARGB color to ABGR
|
|
color = ((color & 0xff00ff00) |
|
|
((color >> 16) & 0xff) |
|
|
((color & 0xff) << 16));
|
|
|
|
draw_rect_ARGB(pixels, stride, x, y, width, height, color);
|
|
}
|
|
|
|
static void draw_rect_NV12( // Y, UV planes
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
const guchar Y = color >> 16;
|
|
const guchar Cb = color >> 8;
|
|
const guchar Cr = color;
|
|
guchar *dst;
|
|
guint i, j;
|
|
|
|
dst = pixels[0] + y * stride[0] + x;
|
|
for (j = 0; j < height; j++, dst += stride[0])
|
|
for (i = 0; i < width; i++)
|
|
dst[i] = Y;
|
|
|
|
x /= 2;
|
|
y /= 2;
|
|
width /= 2;
|
|
height /= 2;
|
|
|
|
dst = pixels[1] + y * stride[1] + x * 2;
|
|
for (j = 0; j < height; j++, dst += stride[1])
|
|
for (i = 0; i < width; i++) {
|
|
dst[2*i + 0] = Cb;
|
|
dst[2*i + 1] = Cr;
|
|
}
|
|
}
|
|
|
|
static void draw_rect_YV12( // Y, U, V planes
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
const guchar Y = color >> 16;
|
|
const guchar Cb = color >> 8;
|
|
const guchar Cr = color;
|
|
guchar *pY, *pU, *pV;
|
|
guint i, j;
|
|
|
|
pY = pixels[0] + y * stride[0] + x;
|
|
for (j = 0; j < height; j++, pY += stride[0])
|
|
for (i = 0; i < width; i++)
|
|
pY[i] = Y;
|
|
|
|
x /= 2;
|
|
y /= 2;
|
|
width /= 2;
|
|
height /= 2;
|
|
|
|
pU = pixels[1] + y * stride[1] + x;
|
|
pV = pixels[2] + y * stride[2] + x;
|
|
for (j = 0; j < height; j++, pU += stride[1], pV += stride[2])
|
|
for (i = 0; i < width; i++) {
|
|
pU[i] = Cb;
|
|
pV[i] = Cr;
|
|
}
|
|
}
|
|
|
|
static void draw_rect_AYUV(
|
|
guchar *pixels[3],
|
|
guint stride[3],
|
|
gint x,
|
|
gint y,
|
|
guint width,
|
|
guint height,
|
|
guint32 color
|
|
)
|
|
{
|
|
guint i, j;
|
|
|
|
color = color | 0xff000000;
|
|
|
|
for (j = 0; j < height; j++) {
|
|
guint32 *p = (guint32 *)(pixels[0] + (y + j) * stride[0] + x * 4);
|
|
for (i = 0; i < width; i++)
|
|
p[i] = color;
|
|
}
|
|
}
|
|
|
|
static gboolean draw_rgb_rects(GstVaapiImage *image)
|
|
{
|
|
GstVaapiImageFormat format = GST_VAAPI_IMAGE_FORMAT(image);
|
|
guint w = GST_VAAPI_IMAGE_WIDTH(image);
|
|
guint h = GST_VAAPI_IMAGE_HEIGHT(image);
|
|
guchar *pixels[3];
|
|
guint stride[3];
|
|
guint32 red_color, green_color, blue_color, black_color;
|
|
DrawRectFunc draw_rect;
|
|
|
|
if (!gst_vaapi_image_map(image))
|
|
return FALSE;
|
|
|
|
switch (format) {
|
|
case GST_VAAPI_IMAGE_ARGB:
|
|
draw_rect = draw_rect_ARGB;
|
|
goto RGB_colors;
|
|
case GST_VAAPI_IMAGE_BGRA:
|
|
draw_rect = draw_rect_BGRA;
|
|
goto RGB_colors;
|
|
case GST_VAAPI_IMAGE_RGBA:
|
|
draw_rect = draw_rect_RGBA;
|
|
goto RGB_colors;
|
|
case GST_VAAPI_IMAGE_ABGR:
|
|
draw_rect = draw_rect_ABGR;
|
|
RGB_colors:
|
|
pixels[0] = gst_vaapi_image_get_plane(image, 0);
|
|
stride[0] = gst_vaapi_image_get_pitch(image, 0);
|
|
red_color = 0xffff0000;
|
|
green_color = 0xff00ff00;
|
|
blue_color = 0xff0000ff;
|
|
black_color = 0xff000000;
|
|
break;
|
|
case GST_VAAPI_IMAGE_NV12:
|
|
draw_rect = draw_rect_NV12;
|
|
pixels[0] = gst_vaapi_image_get_plane(image, 0);
|
|
stride[0] = gst_vaapi_image_get_pitch(image, 0);
|
|
pixels[1] = gst_vaapi_image_get_plane(image, 1);
|
|
stride[1] = gst_vaapi_image_get_pitch(image, 1);
|
|
goto YUV_colors;
|
|
case GST_VAAPI_IMAGE_YV12:
|
|
draw_rect = draw_rect_YV12;
|
|
pixels[0] = gst_vaapi_image_get_plane(image, 0);
|
|
stride[0] = gst_vaapi_image_get_pitch(image, 0);
|
|
pixels[1] = gst_vaapi_image_get_plane(image, 2);
|
|
stride[1] = gst_vaapi_image_get_pitch(image, 2);
|
|
pixels[2] = gst_vaapi_image_get_plane(image, 1);
|
|
stride[2] = gst_vaapi_image_get_pitch(image, 1);
|
|
goto YUV_colors;
|
|
case GST_VAAPI_IMAGE_I420:
|
|
draw_rect = draw_rect_YV12;
|
|
pixels[0] = gst_vaapi_image_get_plane(image, 0);
|
|
stride[0] = gst_vaapi_image_get_pitch(image, 0);
|
|
pixels[1] = gst_vaapi_image_get_plane(image, 1);
|
|
stride[1] = gst_vaapi_image_get_pitch(image, 1);
|
|
pixels[2] = gst_vaapi_image_get_plane(image, 2);
|
|
stride[2] = gst_vaapi_image_get_pitch(image, 2);
|
|
goto YUV_colors;
|
|
case GST_VAAPI_IMAGE_AYUV:
|
|
draw_rect = draw_rect_AYUV;
|
|
pixels[0] = gst_vaapi_image_get_plane(image, 0);
|
|
stride[0] = gst_vaapi_image_get_pitch(image, 0);
|
|
YUV_colors:
|
|
red_color = 0x515af0;
|
|
green_color = 0x913622;
|
|
blue_color = 0x29f06e;
|
|
black_color = 0x108080;
|
|
break;
|
|
default:
|
|
gst_vaapi_image_unmap(image);
|
|
return FALSE;
|
|
}
|
|
|
|
draw_rect(pixels, stride, 0, 0, w/2, h/2, red_color);
|
|
draw_rect(pixels, stride, w/2, 0, w/2, h/2, green_color);
|
|
draw_rect(pixels, stride, 0, h/2, w/2, h/2, blue_color);
|
|
draw_rect(pixels, stride, w/2, h/2, w/2, h/2, black_color);
|
|
|
|
if (!gst_vaapi_image_unmap(image))
|
|
return FALSE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean
|
|
upload_image(GstVaapiSurface *surface, GstVaapiImage *image)
|
|
{
|
|
GstVaapiDisplay *display;
|
|
GstVaapiImageFormat format;
|
|
GstVaapiSubpicture *subpicture;
|
|
|
|
display = gst_vaapi_object_get_display(GST_VAAPI_OBJECT(surface));
|
|
if (!display)
|
|
return FALSE;
|
|
|
|
format = gst_vaapi_image_get_format(image);
|
|
if (!format)
|
|
return FALSE;
|
|
|
|
if (gst_vaapi_surface_put_image(surface, image))
|
|
return TRUE;
|
|
|
|
g_print("could not upload %" GST_FOURCC_FORMAT" image to surface\n",
|
|
GST_FOURCC_ARGS(format));
|
|
|
|
if (!gst_vaapi_display_has_subpicture_format(display, format))
|
|
return FALSE;
|
|
|
|
g_print("trying as a subpicture\n");
|
|
|
|
subpicture = gst_vaapi_subpicture_new(image);
|
|
if (!subpicture)
|
|
g_error("could not create Gst/VA subpicture");
|
|
|
|
if (!gst_vaapi_surface_associate_subpicture(surface, subpicture,
|
|
NULL, NULL))
|
|
g_error("could not associate subpicture to surface");
|
|
|
|
/* The surface holds a reference to the subpicture. This is safe */
|
|
g_object_unref(subpicture);
|
|
return TRUE;
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
GstVaapiDisplay *display;
|
|
GstVaapiWindow *window;
|
|
GstVaapiSurface *surface;
|
|
GstVaapiImage *image = NULL;
|
|
guint flags = GST_VAAPI_PICTURE_STRUCTURE_FRAME;
|
|
guint i;
|
|
|
|
static const GstVaapiImageFormat image_formats[] = {
|
|
GST_VAAPI_IMAGE_NV12,
|
|
GST_VAAPI_IMAGE_YV12,
|
|
GST_VAAPI_IMAGE_I420,
|
|
GST_VAAPI_IMAGE_AYUV,
|
|
GST_VAAPI_IMAGE_ARGB,
|
|
GST_VAAPI_IMAGE_BGRA,
|
|
GST_VAAPI_IMAGE_RGBA,
|
|
GST_VAAPI_IMAGE_ABGR,
|
|
0
|
|
};
|
|
|
|
static const GstVaapiChromaType chroma_type = GST_VAAPI_CHROMA_TYPE_YUV420;
|
|
static const guint width = 320;
|
|
static const guint height = 240;
|
|
static const guint win_width = 640;
|
|
static const guint win_height = 480;
|
|
|
|
gst_init(&argc, &argv);
|
|
|
|
display = gst_vaapi_display_x11_new(NULL);
|
|
if (!display)
|
|
g_error("could not create Gst/VA display");
|
|
|
|
surface = gst_vaapi_surface_new(display, chroma_type, width, height);
|
|
if (!surface)
|
|
g_error("could not create Gst/VA surface");
|
|
|
|
for (i = 0; image_formats[i]; i++) {
|
|
const GstVaapiImageFormat format = image_formats[i];
|
|
|
|
image = gst_vaapi_image_new(display, format, width, height);
|
|
if (!image)
|
|
continue;
|
|
|
|
if (!draw_rgb_rects(image))
|
|
g_error("could not draw RGB rectangles");
|
|
|
|
if (upload_image(surface, image))
|
|
break;
|
|
}
|
|
if (!image)
|
|
g_error("could not create Gst/VA image");
|
|
|
|
if (!gst_vaapi_surface_sync(surface))
|
|
g_error("could not complete image upload");
|
|
|
|
g_print("#\n");
|
|
g_print("# Create window with gst_vaapi_window_x11_new()\n");
|
|
g_print("#\n");
|
|
{
|
|
window = gst_vaapi_window_x11_new(display, win_width, win_height);
|
|
if (!window)
|
|
g_error("could not create window");
|
|
|
|
gst_vaapi_window_show(window);
|
|
|
|
if (!gst_vaapi_window_put_surface(window, surface, NULL, NULL, flags))
|
|
g_error("could not render surface");
|
|
|
|
pause();
|
|
g_object_unref(window);
|
|
}
|
|
|
|
g_print("#\n");
|
|
g_print("# Create window with gst_vaapi_window_x11_new_with_xid()\n");
|
|
g_print("#\n");
|
|
{
|
|
Display * const dpy = GST_VAAPI_DISPLAY_XDISPLAY(display);
|
|
Window rootwin, win;
|
|
int screen;
|
|
unsigned long white_pixel, black_pixel;
|
|
|
|
screen = DefaultScreen(dpy);
|
|
rootwin = RootWindow(dpy, screen);
|
|
white_pixel = WhitePixel(dpy, screen);
|
|
black_pixel = BlackPixel(dpy, screen);
|
|
|
|
win = XCreateSimpleWindow(
|
|
dpy,
|
|
rootwin,
|
|
0, 0, win_width, win_height,
|
|
0, black_pixel,
|
|
white_pixel
|
|
);
|
|
if (!win)
|
|
g_error("could not create X window");
|
|
|
|
window = gst_vaapi_window_x11_new_with_xid(display, win);
|
|
if (!window)
|
|
g_error("could not create window");
|
|
|
|
gst_vaapi_window_show(window);
|
|
|
|
if (!gst_vaapi_window_put_surface(window, surface, NULL, NULL, flags))
|
|
g_error("could not render surface");
|
|
|
|
pause();
|
|
g_object_unref(window);
|
|
XUnmapWindow(dpy, win);
|
|
XDestroyWindow(dpy, win);
|
|
}
|
|
|
|
g_object_unref(image);
|
|
g_object_unref(surface);
|
|
g_object_unref(display);
|
|
gst_deinit();
|
|
return 0;
|
|
}
|