diff options
Diffstat (limited to 'tests/test-decode.c')
| -rw-r--r-- | tests/test-decode.c | 152 |
1 files changed, 152 insertions, 0 deletions
diff --git a/tests/test-decode.c b/tests/test-decode.c new file mode 100644 index 00000000..a08383f7 --- /dev/null +++ b/tests/test-decode.c @@ -0,0 +1,152 @@ +/* + * test-decode.c - Test GstVaapiDecoder + * + * 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 <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-mpeg2.h" +#include "test-h264.h" +#include "test-vc1.h" + +/* Default timeout to wait for the first decoded frame (10 ms) */ +/* Defined to -1 if the application indefintely waits for the decoded frame */ +#define TIMEOUT -1 + +typedef void (*GetVideoDataFunc)(const guchar **data, guint *size); + +typedef struct _CodecDefs CodecDefs; +struct _CodecDefs { + const gchar *codec_str; + GstVaapiCodec codec; + GetVideoDataFunc get_video_data; +}; + +static const CodecDefs g_codec_defs[] = { + { "mpeg2", GST_VAAPI_CODEC_MPEG2, mpeg2_get_video_data }, + { "h264", GST_VAAPI_CODEC_H264, h264_get_video_data }, + { "vc1", GST_VAAPI_CODEC_VC1, vc1_get_video_data }, + { 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; + GstVaapiDecoderStatus status; + const CodecDefs *codec; + GstVaapiSurfaceProxy *proxy; + const guchar *vdata; + guint vdata_size; + + 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_data(&vdata, &vdata_size); + decoder = gst_vaapi_decoder_ffmpeg_new(display, codec->codec); + if (!decoder) + g_error("could not create FFmpeg decoder"); + + if (!gst_vaapi_decoder_put_buffer_data(decoder, vdata, vdata_size)) + g_error("could not send video data to the decoder"); + if (0 && !gst_vaapi_decoder_put_buffer(decoder, NULL)) + g_error("could not send EOS to the decoder"); + + if (TIMEOUT < 0) { + proxy = gst_vaapi_decoder_get_surface(decoder, &status); + if (!proxy) + g_error("could not get decoded surface"); + } + else { + proxy = gst_vaapi_decoder_timed_get_surface(decoder, TIMEOUT, &status); + if (!proxy) + g_warning("Could not get decoded surface after %d us", TIMEOUT); + } + + gst_vaapi_window_show(window); + + if (!gst_vaapi_window_put_surface(window, proxy->surface, NULL, NULL, + GST_VAAPI_PICTURE_STRUCTURE_FRAME)) + g_error("could not render surface"); + + pause(); + + 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; +} |