mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-07-07 07:05:52 +00:00
tests: add test for subpictures.
Signed-off-by: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
This commit is contained in:
Thibault Saunier
Gwenole Beauchesne
parent
cbf81d7d51
commit
bb72c55555
|
|
@ -3,6 +3,7 @@ noinst_PROGRAMS = \
|
|||
test-display \
|
||||
test-surfaces \
|
||||
test-windows \
|
||||
test-subpicture \
|
||||
$(NULL)
|
||||
|
||||
if USE_GLX
|
||||
|
|
@ -55,6 +56,10 @@ test_surfaces_SOURCES = test-surfaces.c
|
|||
test_surfaces_CFLAGS = $(TEST_CFLAGS) $(TEST_X11_CFLAGS)
|
||||
test_surfaces_LDADD = $(TEST_LIBS) $(TEST_X11_LIBS)
|
||||
|
||||
test_subpicture_SOURCES = test-subpicture.c test-h264.c test-subpicture-data.c
|
||||
test_subpicture_CFLAGS = $(TEST_CFLAGS) $(TEST_X11_CFLAGS)
|
||||
test_subpicture_LDADD = $(TEST_LIBS) $(TEST_X11_LIBS)
|
||||
|
||||
test_windows_SOURCES = test-windows.c image.c
|
||||
test_windows_CFLAGS = $(TEST_CFLAGS) $(TEST_X11_CFLAGS)
|
||||
test_windows_LDADD = $(TEST_LIBS) $(TEST_X11_LIBS)
|
||||
|
|
|
|||
1430
tests/test-subpicture-data.c
Normal file
1430
tests/test-subpicture-data.c
Normal file
File diff suppressed because it is too large
Load diff
40
tests/test-subpicture-data.h
Normal file
40
tests/test-subpicture-data.h
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
/*
|
||||
* test-subpicture-data.h - subpicture data
|
||||
*
|
||||
* Copyright (C) <2011> Intel Corporation
|
||||
* Copyright (C) <2011> Collabora Ltd.
|
||||
* Copyright (C) <2011> Thibault Saunier <thibault.saunier@collabora.com>
|
||||
*
|
||||
* 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
|
||||
*/
|
||||
|
||||
#ifndef TEST_SUBPICTURE_DATA
|
||||
#define TEST_SUBPICTURE_DATA
|
||||
|
||||
#include <glib.h>
|
||||
#include "test-decode.h"
|
||||
|
||||
typedef struct _VideoSubpictureInfo VideoSubpictureInfo;
|
||||
|
||||
struct _VideoSubpictureInfo {
|
||||
guint width;
|
||||
guint height;
|
||||
const guint32 *data;
|
||||
guint data_size;
|
||||
};
|
||||
|
||||
void subpicture_get_info(VideoSubpictureInfo *info);
|
||||
|
||||
#endif /* TEST_SUBPICTURE_DATA*/
|
||||
207
tests/test-subpicture.c
Normal file
207
tests/test-subpicture.c
Normal file
|
|
@ -0,0 +1,207 @@
|
|||
/*
|
||||
* test-subpicture.c - Test GstVaapiSubpicture
|
||||
*
|
||||
* Copyright (C) <2011> Intel Corporation
|
||||
* Copyright (C) <2011> Collabora Ltd.
|
||||
* Copyright (C) <2011> Thibault Saunier <thibault.saunier@collabora.com>
|
||||
*
|
||||
* 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 <string.h>
|
||||
#include <gst/vaapi/gstvaapidisplay_x11.h>
|
||||
#include <gst/vaapi/gstvaapiwindow_x11.h>
|
||||
#include <gst/vaapi/gstvaapidecoder.h>
|
||||
#include <gst/vaapi/gstvaapidecoder_ffmpeg.h>
|
||||
#include <gst/vaapi/gstvaapisurface.h>
|
||||
#include "test-h264.h"
|
||||
#include "test-subpicture-data.h"
|
||||
|
||||
typedef void (*GetVideoInfoFunc)(VideoDecodeInfo *info);
|
||||
|
||||
typedef struct _CodecDefs CodecDefs;
|
||||
struct _CodecDefs {
|
||||
const gchar *codec_str;
|
||||
GetVideoInfoFunc get_video_info;
|
||||
};
|
||||
|
||||
static const CodecDefs g_codec_defs[] = {
|
||||
#define INIT_FUNCS(CODEC) { #CODEC, CODEC##_get_video_info }
|
||||
INIT_FUNCS(h264),
|
||||
#undef INIT_FUNCS
|
||||
{ NULL, }
|
||||
};
|
||||
|
||||
static const CodecDefs *
|
||||
get_codec_defs(const gchar *codec_str)
|
||||
{
|
||||
const CodecDefs *c;
|
||||
for (c = g_codec_defs; c->codec_str; c++)
|
||||
if (strcmp(codec_str, c->codec_str) == 0)
|
||||
return c;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static inline void pause(void)
|
||||
{
|
||||
g_print("Press any key to continue...\n");
|
||||
getchar();
|
||||
}
|
||||
|
||||
static gchar *g_codec_str;
|
||||
|
||||
static GOptionEntry g_options[] = {
|
||||
{ "codec", 'c',
|
||||
0,
|
||||
G_OPTION_ARG_STRING, &g_codec_str,
|
||||
"codec to test", NULL },
|
||||
{ NULL, }
|
||||
};
|
||||
|
||||
int
|
||||
main(int argc, char *argv[])
|
||||
{
|
||||
GOptionContext *options;
|
||||
GstVaapiDisplay *display;
|
||||
GstVaapiWindow *window;
|
||||
GstVaapiDecoder *decoder;
|
||||
GstCaps *decoder_caps;
|
||||
GstStructure *structure;
|
||||
GstVaapiDecoderStatus status;
|
||||
const CodecDefs *codec;
|
||||
GstBuffer *buffer;
|
||||
GstVaapiSurfaceProxy *proxy;
|
||||
VideoDecodeInfo info;
|
||||
VideoSubpictureInfo subinfo;
|
||||
GstVaapiImage *subtitle_image;
|
||||
GstVaapiSubpicture *subpicture;
|
||||
GstCaps *argbcaps;
|
||||
GstVaapiRectangle sub_rect;
|
||||
|
||||
static const guint win_width = 640;
|
||||
static const guint win_height = 480;
|
||||
|
||||
gst_init(&argc, &argv);
|
||||
|
||||
options = g_option_context_new(" - test-decode options");
|
||||
g_option_context_add_main_entries(options, g_options, NULL);
|
||||
g_option_context_parse(options, &argc, &argv, NULL);
|
||||
g_option_context_free(options);
|
||||
|
||||
if (!g_codec_str)
|
||||
g_codec_str = g_strdup("h264");
|
||||
|
||||
g_print("Test %s decode\n", g_codec_str);
|
||||
codec = get_codec_defs(g_codec_str);
|
||||
if (!codec)
|
||||
g_error("no %s codec data found", g_codec_str);
|
||||
|
||||
display = gst_vaapi_display_x11_new(NULL);
|
||||
if (!display)
|
||||
g_error("could not create VA display");
|
||||
|
||||
window = gst_vaapi_window_x11_new(display, win_width, win_height);
|
||||
if (!window)
|
||||
g_error("could not create window");
|
||||
|
||||
codec->get_video_info(&info);
|
||||
decoder_caps = gst_vaapi_profile_get_caps(info.profile);
|
||||
if (!decoder_caps)
|
||||
g_error("could not create decoder caps");
|
||||
|
||||
structure = gst_caps_get_structure(decoder_caps, 0);
|
||||
if (info.width > 0 && info.height > 0)
|
||||
gst_structure_set(
|
||||
structure,
|
||||
"width", G_TYPE_INT, info.width,
|
||||
"height", G_TYPE_INT, info.height,
|
||||
NULL
|
||||
);
|
||||
|
||||
decoder = gst_vaapi_decoder_ffmpeg_new(display, decoder_caps);
|
||||
if (!decoder)
|
||||
g_error("could not create FFmpeg decoder");
|
||||
gst_caps_unref(decoder_caps);
|
||||
|
||||
buffer = gst_buffer_new();
|
||||
if (!buffer)
|
||||
g_error("could not create encoded data buffer");
|
||||
gst_buffer_set_data(buffer, (guchar *)info.data, info.data_size);
|
||||
|
||||
if (!gst_vaapi_decoder_put_buffer(decoder, buffer))
|
||||
g_error("could not send video data to the decoder");
|
||||
gst_buffer_unref(buffer);
|
||||
|
||||
if (!gst_vaapi_decoder_put_buffer(decoder, NULL))
|
||||
g_error("could not send EOS to the decoder");
|
||||
|
||||
proxy = gst_vaapi_decoder_get_surface(decoder, &status);
|
||||
if (!proxy)
|
||||
g_error("could not get decoded surface (decoder status %d)", status);
|
||||
|
||||
subpicture_get_info (&subinfo);
|
||||
|
||||
/* Adding subpicture */
|
||||
argbcaps = gst_caps_new_simple ("video/x-raw-rgb",
|
||||
"endianness", G_TYPE_INT, 1,
|
||||
"bpp", G_TYPE_INT, 32,
|
||||
"width", G_TYPE_INT, subinfo.width,
|
||||
"height", G_TYPE_INT, subinfo.height,
|
||||
NULL);
|
||||
|
||||
buffer = gst_buffer_new ();
|
||||
gst_buffer_set_data(buffer, (guchar *)subinfo.data, subinfo.data_size);
|
||||
gst_buffer_set_caps (buffer, argbcaps);
|
||||
|
||||
subtitle_image = gst_vaapi_image_new (display,
|
||||
GST_VAAPI_IMAGE_RGBA, subinfo.width, subinfo.height);
|
||||
|
||||
gst_vaapi_image_update_from_buffer (subtitle_image, buffer);
|
||||
|
||||
subpicture = gst_vaapi_subpicture_new (subtitle_image);
|
||||
|
||||
/* We position it as a subtitle, centered at the bottom. */
|
||||
sub_rect.x = (win_width - subinfo.width) / 2;
|
||||
sub_rect.y = win_height - subinfo.height - 10;
|
||||
sub_rect.height = subinfo.height;
|
||||
sub_rect.width = subinfo.height;
|
||||
|
||||
if (!gst_vaapi_surface_associate_subpicture (
|
||||
GST_VAAPI_SURFACE_PROXY_SURFACE(proxy),
|
||||
subpicture,
|
||||
NULL,
|
||||
&sub_rect))
|
||||
g_error("could not associate subpicture");
|
||||
|
||||
gst_vaapi_window_show(window);
|
||||
|
||||
if (!gst_vaapi_window_put_surface(window,
|
||||
GST_VAAPI_SURFACE_PROXY_SURFACE(proxy),
|
||||
NULL,
|
||||
NULL,
|
||||
GST_VAAPI_PICTURE_STRUCTURE_FRAME))
|
||||
g_error("could not render surface");
|
||||
|
||||
pause();
|
||||
|
||||
gst_buffer_unref(buffer);
|
||||
g_object_unref(proxy);
|
||||
g_object_unref(decoder);
|
||||
g_object_unref(window);
|
||||
g_object_unref(display);
|
||||
g_free(g_codec_str);
|
||||
gst_deinit();
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in a new issue