gstreamer/subprojects/gstreamer-vaapi/tests/internal/decoder.c

/*
 *  decoder.h - Decoder utilities for the tests
 *
 *  Copyright (C) 2013 Intel Corporation
 *    Author: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public License
 *  as published by the Free Software Foundation; either version 2.1
 *  of the License, or (at your option) any later version.
 *
 *  This library 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
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free
 *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA 02110-1301 USA
 */

#include "gst/vaapi/sysdeps.h"
#include <gst/vaapi/gstvaapidecoder_h264.h>
#include <gst/vaapi/gstvaapidecoder_jpeg.h>
#include <gst/vaapi/gstvaapidecoder_mpeg2.h>
#include <gst/vaapi/gstvaapidecoder_mpeg4.h>
#include <gst/vaapi/gstvaapidecoder_vc1.h>
#include "decoder.h"
#include "test-jpeg.h"
#include "test-mpeg2.h"
#include "test-mpeg4.h"
#include "test-h264.h"
#include "test-vc1.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 (jpeg),
  INIT_FUNCS (mpeg2),
  INIT_FUNCS (mpeg4),
  INIT_FUNCS (h264),
  INIT_FUNCS (vc1),
#undef INIT_FUNCS
  {NULL,}
};

static const CodecDefs *
find_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 const CodecDefs *
get_codec_defs (GstVaapiDecoder * decoder)
{
  return gst_vaapi_decoder_get_user_data (decoder);
}

static inline void
set_codec_defs (GstVaapiDecoder * decoder, const CodecDefs * c)
{
  gst_vaapi_decoder_set_user_data (decoder, (gpointer) c);
}

GstVaapiDecoder *
decoder_new (GstVaapiDisplay * display, const gchar * codec_name)
{
  GstVaapiDecoder *decoder;
  const CodecDefs *codec;
  GstCaps *caps;
  VideoDecodeInfo info;

  if (!codec_name)
    codec_name = "h264";

  codec = find_codec_defs (codec_name);
  if (!codec) {
    GST_ERROR ("failed to find %s codec data", codec_name);
    return NULL;
  }

  codec->get_video_info (&info);
  caps = gst_vaapi_profile_get_caps (info.profile);
  if (!caps) {
    GST_ERROR ("failed to create decoder caps");
    return NULL;
  }

  if (info.width > 0 && info.height > 0)
    gst_caps_set_simple (caps,
        "width", G_TYPE_INT, info.width,
        "height", G_TYPE_INT, info.height, NULL);

  switch (gst_vaapi_profile_get_codec (info.profile)) {
    case GST_VAAPI_CODEC_H264:
      decoder = gst_vaapi_decoder_h264_new (display, caps);
      break;
    case GST_VAAPI_CODEC_JPEG:
      decoder = gst_vaapi_decoder_jpeg_new (display, caps);
      break;
    case GST_VAAPI_CODEC_MPEG2:
      decoder = gst_vaapi_decoder_mpeg2_new (display, caps);
      break;
    case GST_VAAPI_CODEC_MPEG4:
      decoder = gst_vaapi_decoder_mpeg4_new (display, caps);
      break;
    case GST_VAAPI_CODEC_VC1:
      decoder = gst_vaapi_decoder_vc1_new (display, caps);
      break;
    default:
      decoder = NULL;
      break;
  }
  gst_caps_unref (caps);
  if (!decoder) {
    GST_ERROR ("failed to create %s decoder", codec->codec_str);
    return NULL;
  }

  set_codec_defs (decoder, codec);
  return decoder;
}

gboolean
decoder_put_buffers (GstVaapiDecoder * decoder)
{
  const CodecDefs *codec;
  VideoDecodeInfo info;
  GstBuffer *buffer;
  gboolean success;

  g_return_val_if_fail (decoder != NULL, FALSE);

  codec = get_codec_defs (decoder);
  g_return_val_if_fail (codec != NULL, FALSE);

  codec->get_video_info (&info);
  buffer = gst_buffer_new_wrapped_full (GST_MEMORY_FLAG_READONLY,
      (guchar *) info.data, info.data_size, 0, info.data_size, NULL, NULL);
  if (!buffer) {
    GST_ERROR ("failed to create encoded data buffer");
    return FALSE;
  }

  success = gst_vaapi_decoder_put_buffer (decoder, buffer);
  gst_buffer_unref (buffer);
  if (!success) {
    GST_ERROR ("failed to send video data to the decoder");
    return FALSE;
  }

  if (!gst_vaapi_decoder_put_buffer (decoder, NULL)) {
    GST_ERROR ("failed to submit <end-of-stream> to the decoder");
    return FALSE;
  }
  return TRUE;
}

GstVaapiSurfaceProxy *
decoder_get_surface (GstVaapiDecoder * decoder)
{
  GstVaapiSurfaceProxy *proxy;
  GstVaapiDecoderStatus status;

  g_return_val_if_fail (decoder != NULL, NULL);

  status = gst_vaapi_decoder_get_surface (decoder, &proxy);
  if (status != GST_VAAPI_DECODER_STATUS_SUCCESS) {
    GST_ERROR ("failed to get decoded surface (decoder status %d)", status);
    return NULL;
  }
  return proxy;
}

const gchar *
decoder_get_codec_name (GstVaapiDecoder * decoder)
{
  const CodecDefs *codec;

  g_return_val_if_fail (decoder != NULL, NULL);

  codec = get_codec_defs (decoder);
  g_return_val_if_fail (codec != NULL, FALSE);

  return codec->codec_str;
}