ext/libav/gstavviddec.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  17  * Boston, MA 02110-1301, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include "config.h"
  22 #endif
  23
  24 #include <assert.h>
  25 #include <string.h>
  26
  27 #include <libavcodec/avcodec.h>
  28 #include <libavutil/stereo3d.h>
  29
  30 #include "gstav.h"
  31 #include "gstavcodecmap.h"
  32 #include "gstavutils.h"
  33 #include "gstavviddec.h"
  34
  35 #define MAX_TS_MASK 0xff
  36
  37 #define DEFAULT_LOWRES                  0
  38 #define DEFAULT_SKIPFRAME               0
  39 #define DEFAULT_DIRECT_RENDERING        TRUE
  40 #define DEFAULT_DEBUG_MV                FALSE
  41 #define DEFAULT_MAX_THREADS             0
  42 #define DEFAULT_OUTPUT_CORRUPT          TRUE
  43 #define REQUIRED_POOL_MAX_BUFFERS       32
  44 #define DEFAULT_STRIDE_ALIGN            31
  45 #define DEFAULT_ALLOC_PARAM             { 0, DEFAULT_STRIDE_ALIGN, 0, 0, }
  46
  47 enum
  48 {
  49   PROP_0,
  50   PROP_LOWRES,
  51   PROP_SKIPFRAME,
  52   PROP_DIRECT_RENDERING,
  53   PROP_DEBUG_MV,
  54   PROP_MAX_THREADS,
  55   PROP_OUTPUT_CORRUPT,
  56   PROP_LAST
  57 };
  58
  59 /* A number of function prototypes are given so we can refer to them later. */
  60 static void gst_ffmpegviddec_base_init (GstFFMpegVidDecClass * klass);
  61 static void gst_ffmpegviddec_class_init (GstFFMpegVidDecClass * klass);
  62 static void gst_ffmpegviddec_init (GstFFMpegVidDec * ffmpegdec);
  63 static void gst_ffmpegviddec_finalize (GObject * object);
  64
  65 static gboolean gst_ffmpegviddec_set_format (GstVideoDecoder * decoder,
  66     GstVideoCodecState * state);
  67 static GstFlowReturn gst_ffmpegviddec_handle_frame (GstVideoDecoder * decoder,
  68     GstVideoCodecFrame * frame);
  69 static gboolean gst_ffmpegviddec_start (GstVideoDecoder * decoder);
  70 static gboolean gst_ffmpegviddec_stop (GstVideoDecoder * decoder);
  71 static gboolean gst_ffmpegviddec_flush (GstVideoDecoder * decoder);
  72 static gboolean gst_ffmpegviddec_decide_allocation (GstVideoDecoder * decoder,
  73     GstQuery * query);
  74 static gboolean gst_ffmpegviddec_propose_allocation (GstVideoDecoder * decoder,
  75     GstQuery * query);
  76
  77 static void gst_ffmpegviddec_set_property (GObject * object,
  78     guint prop_id, const GValue * value, GParamSpec * pspec);
  79 static void gst_ffmpegviddec_get_property (GObject * object,
  80     guint prop_id, GValue * value, GParamSpec * pspec);
  81
  82 static gboolean gst_ffmpegviddec_negotiate (GstFFMpegVidDec * ffmpegdec,
  83     AVCodecContext * context, AVFrame * picture);
  84
  85 /* some sort of bufferpool handling, but different */
  86 static int gst_ffmpegviddec_get_buffer2 (AVCodecContext * context,
  87     AVFrame * picture, int flags);
  88
  89 static GstFlowReturn gst_ffmpegviddec_finish (GstVideoDecoder * decoder);
  90 static GstFlowReturn gst_ffmpegviddec_drain (GstVideoDecoder * decoder);
  91
  92 static gboolean picture_changed (GstFFMpegVidDec * ffmpegdec,
  93     AVFrame * picture);
  94 static gboolean context_changed (GstFFMpegVidDec * ffmpegdec,
  95     AVCodecContext * context);
  96
  97 #define GST_FFDEC_PARAMS_QDATA g_quark_from_static_string("avdec-params")
  98
  99 static GstElementClass *parent_class = NULL;
 100
 101 #define GST_FFMPEGVIDDEC_TYPE_LOWRES (gst_ffmpegviddec_lowres_get_type())
 102 static GType
 103 gst_ffmpegviddec_lowres_get_type (void)
 104 {
 105   static GType ffmpegdec_lowres_type = 0;
 106
 107   if (!ffmpegdec_lowres_type) {
 108     static const GEnumValue ffmpegdec_lowres[] = {
 109       {0, "0", "full"},
 110       {1, "1", "1/2-size"},
 111       {2, "2", "1/4-size"},
 112       {0, NULL, NULL},
 113     };
 114
 115     ffmpegdec_lowres_type =
 116         g_enum_register_static ("GstLibAVVidDecLowres", ffmpegdec_lowres);
 117   }
 118
 119   return ffmpegdec_lowres_type;
 120 }
 121
 122 #define GST_FFMPEGVIDDEC_TYPE_SKIPFRAME (gst_ffmpegviddec_skipframe_get_type())
 123 static GType
 124 gst_ffmpegviddec_skipframe_get_type (void)
 125 {
 126   static GType ffmpegdec_skipframe_type = 0;
 127
 128   if (!ffmpegdec_skipframe_type) {
 129     static const GEnumValue ffmpegdec_skipframe[] = {
 130       {0, "0", "Skip nothing"},
 131       {1, "1", "Skip B-frames"},
 132       {2, "2", "Skip IDCT/Dequantization"},
 133       {5, "5", "Skip everything"},
 134       {0, NULL, NULL},
 135     };
 136
 137     ffmpegdec_skipframe_type =
 138         g_enum_register_static ("GstLibAVVidDecSkipFrame", ffmpegdec_skipframe);
 139   }
 140
 141   return ffmpegdec_skipframe_type;
 142 }
 143
 144 static void
 145 gst_ffmpegviddec_base_init (GstFFMpegVidDecClass * klass)
 146 {
 147   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
 148   GstPadTemplate *sinktempl, *srctempl;
 149   GstCaps *sinkcaps, *srccaps;
 150   AVCodec *in_plugin;
 151   gchar *longname, *description;
 152
 153   in_plugin =
 154       (AVCodec *) g_type_get_qdata (G_OBJECT_CLASS_TYPE (klass),
 155       GST_FFDEC_PARAMS_QDATA);
 156   g_assert (in_plugin != NULL);
 157
 158   /* construct the element details struct */
 159   longname = g_strdup_printf ("libav %s decoder", in_plugin->long_name);
 160   description = g_strdup_printf ("libav %s decoder", in_plugin->name);
 161   gst_element_class_set_metadata (element_class, longname,
 162       "Codec/Decoder/Video", description,
 163       "Wim Taymans <wim.taymans@gmail.com>, "
 164       "Ronald Bultje <rbultje@ronald.bitfreak.net>, "
 165       "Edward Hervey <bilboed@bilboed.com>");
 166   g_free (longname);
 167   g_free (description);
 168
 169   /* get the caps */
 170   sinkcaps = gst_ffmpeg_codecid_to_caps (in_plugin->id, NULL, FALSE);
 171   if (!sinkcaps) {
 172     GST_DEBUG ("Couldn't get sink caps for decoder '%s'", in_plugin->name);
 173     sinkcaps = gst_caps_new_empty_simple ("unknown/unknown");
 174   }
 175   srccaps = gst_ffmpeg_codectype_to_video_caps (NULL,
 176       in_plugin->id, FALSE, in_plugin);
 177   if (!srccaps) {
 178     GST_DEBUG ("Couldn't get source caps for decoder '%s'", in_plugin->name);
 179     srccaps = gst_caps_from_string ("video/x-raw");
 180   }
 181
 182   /* pad templates */
 183   sinktempl = gst_pad_template_new ("sink", GST_PAD_SINK,
 184       GST_PAD_ALWAYS, sinkcaps);
 185   srctempl = gst_pad_template_new ("src", GST_PAD_SRC, GST_PAD_ALWAYS, srccaps);
 186
 187   gst_element_class_add_pad_template (element_class, srctempl);
 188   gst_element_class_add_pad_template (element_class, sinktempl);
 189
 190   gst_caps_unref (sinkcaps);
 191   gst_caps_unref (srccaps);
 192
 193   klass->in_plugin = in_plugin;
 194 }
 195
 196 static void
 197 gst_ffmpegviddec_class_init (GstFFMpegVidDecClass * klass)
 198 {
 199   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 200   GstVideoDecoderClass *viddec_class = GST_VIDEO_DECODER_CLASS (klass);
 201   int caps;
 202
 203   parent_class = g_type_class_peek_parent (klass);
 204
 205   gobject_class->finalize = gst_ffmpegviddec_finalize;
 206
 207   gobject_class->set_property = gst_ffmpegviddec_set_property;
 208   gobject_class->get_property = gst_ffmpegviddec_get_property;
 209
 210   g_object_class_install_property (gobject_class, PROP_SKIPFRAME,
 211       g_param_spec_enum ("skip-frame", "Skip frames",
 212           "Which types of frames to skip during decoding",
 213           GST_FFMPEGVIDDEC_TYPE_SKIPFRAME, 0,
 214           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 215   g_object_class_install_property (gobject_class, PROP_LOWRES,
 216       g_param_spec_enum ("lowres", "Low resolution",
 217           "At which resolution to decode images",
 218           GST_FFMPEGVIDDEC_TYPE_LOWRES, 0,
 219           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 220   g_object_class_install_property (gobject_class, PROP_DIRECT_RENDERING,
 221       g_param_spec_boolean ("direct-rendering", "Direct Rendering",
 222           "Enable direct rendering", DEFAULT_DIRECT_RENDERING,
 223           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 224   g_object_class_install_property (gobject_class, PROP_DEBUG_MV,
 225       g_param_spec_boolean ("debug-mv", "Debug motion vectors",
 226           "Whether libav should print motion vectors on top of the image",
 227           DEFAULT_DEBUG_MV, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 228   g_object_class_install_property (gobject_class, PROP_OUTPUT_CORRUPT,
 229       g_param_spec_boolean ("output-corrupt", "Output corrupt buffers",
 230           "Whether libav should output frames even if corrupted",
 231           DEFAULT_OUTPUT_CORRUPT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 232
 233   caps = klass->in_plugin->capabilities;
 234   if (caps & (CODEC_CAP_FRAME_THREADS | CODEC_CAP_SLICE_THREADS)) {
 235     g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MAX_THREADS,
 236         g_param_spec_int ("max-threads", "Maximum decode threads",
 237             "Maximum number of worker threads to spawn. (0 = auto)",
 238             0, G_MAXINT, DEFAULT_MAX_THREADS,
 239             G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 240   }
 241
 242   viddec_class->set_format = gst_ffmpegviddec_set_format;
 243   viddec_class->handle_frame = gst_ffmpegviddec_handle_frame;
 244   viddec_class->start = gst_ffmpegviddec_start;
 245   viddec_class->stop = gst_ffmpegviddec_stop;
 246   viddec_class->flush = gst_ffmpegviddec_flush;
 247   viddec_class->finish = gst_ffmpegviddec_finish;
 248   viddec_class->drain = gst_ffmpegviddec_drain;
 249   viddec_class->decide_allocation = gst_ffmpegviddec_decide_allocation;
 250   viddec_class->propose_allocation = gst_ffmpegviddec_propose_allocation;
 251 }
 252
 253 static void
 254 gst_ffmpegviddec_init (GstFFMpegVidDec * ffmpegdec)
 255 {
 256   GstFFMpegVidDecClass *klass =
 257       (GstFFMpegVidDecClass *) G_OBJECT_GET_CLASS (ffmpegdec);
 258
 259   /* some ffmpeg data */
 260   ffmpegdec->context = avcodec_alloc_context3 (klass->in_plugin);
 261   ffmpegdec->context->opaque = ffmpegdec;
 262   ffmpegdec->picture = av_frame_alloc ();
 263   ffmpegdec->opened = FALSE;
 264   ffmpegdec->skip_frame = ffmpegdec->lowres = 0;
 265   ffmpegdec->direct_rendering = DEFAULT_DIRECT_RENDERING;
 266   ffmpegdec->debug_mv = DEFAULT_DEBUG_MV;
 267   ffmpegdec->max_threads = DEFAULT_MAX_THREADS;
 268   ffmpegdec->output_corrupt = DEFAULT_OUTPUT_CORRUPT;
 269
 270   GST_PAD_SET_ACCEPT_TEMPLATE (GST_VIDEO_DECODER_SINK_PAD (ffmpegdec));
 271   gst_video_decoder_set_use_default_pad_acceptcaps (GST_VIDEO_DECODER_CAST
 272       (ffmpegdec), TRUE);
 273
 274   gst_video_decoder_set_needs_format (GST_VIDEO_DECODER (ffmpegdec), TRUE);
 275 }
 276
 277 static void
 278 gst_ffmpegviddec_finalize (GObject * object)
 279 {
 280   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) object;
 281
 282   av_frame_free (&ffmpegdec->picture);
 283
 284   if (ffmpegdec->context != NULL) {
 285     gst_ffmpeg_avcodec_close (ffmpegdec->context);
 286     av_free (ffmpegdec->context);
 287     ffmpegdec->context = NULL;
 288   }
 289
 290   G_OBJECT_CLASS (parent_class)->finalize (object);
 291 }
 292
 293 static void
 294 gst_ffmpegviddec_context_set_flags (AVCodecContext * context, guint flags,
 295     gboolean enable)
 296 {
 297   g_return_if_fail (context != NULL);
 298
 299   if (enable)
 300     context->flags |= flags;
 301   else
 302     context->flags &= ~flags;
 303 }
 304
 305 /* with LOCK */
 306 static gboolean
 307 gst_ffmpegviddec_close (GstFFMpegVidDec * ffmpegdec, gboolean reset)
 308 {
 309   GstFFMpegVidDecClass *oclass;
 310   gint i;
 311
 312   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
 313
 314   GST_LOG_OBJECT (ffmpegdec, "closing ffmpeg codec");
 315
 316   gst_caps_replace (&ffmpegdec->last_caps, NULL);
 317
 318   gst_ffmpeg_avcodec_close (ffmpegdec->context);
 319   ffmpegdec->opened = FALSE;
 320
 321   for (i = 0; i < G_N_ELEMENTS (ffmpegdec->stride); i++)
 322     ffmpegdec->stride[i] = -1;
 323
 324   gst_buffer_replace (&ffmpegdec->palette, NULL);
 325
 326   if (ffmpegdec->context->extradata) {
 327     av_free (ffmpegdec->context->extradata);
 328     ffmpegdec->context->extradata = NULL;
 329   }
 330   if (reset) {
 331     if (avcodec_get_context_defaults3 (ffmpegdec->context,
 332             oclass->in_plugin) < 0) {
 333       GST_DEBUG_OBJECT (ffmpegdec, "Failed to set context defaults");
 334       return FALSE;
 335     }
 336     ffmpegdec->context->opaque = ffmpegdec;
 337   }
 338   return TRUE;
 339 }
 340
 341 /* with LOCK */
 342 static gboolean
 343 gst_ffmpegviddec_open (GstFFMpegVidDec * ffmpegdec)
 344 {
 345   GstFFMpegVidDecClass *oclass;
 346   gint i;
 347
 348   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
 349
 350   if (gst_ffmpeg_avcodec_open (ffmpegdec->context, oclass->in_plugin) < 0)
 351     goto could_not_open;
 352
 353   for (i = 0; i < G_N_ELEMENTS (ffmpegdec->stride); i++)
 354     ffmpegdec->stride[i] = -1;
 355
 356   ffmpegdec->opened = TRUE;
 357
 358   GST_LOG_OBJECT (ffmpegdec, "Opened libav codec %s, id %d",
 359       oclass->in_plugin->name, oclass->in_plugin->id);
 360
 361   gst_ffmpegviddec_context_set_flags (ffmpegdec->context,
 362       CODEC_FLAG_OUTPUT_CORRUPT, ffmpegdec->output_corrupt);
 363
 364   return TRUE;
 365
 366   /* ERRORS */
 367 could_not_open:
 368   {
 369     gst_ffmpegviddec_close (ffmpegdec, TRUE);
 370     GST_DEBUG_OBJECT (ffmpegdec, "avdec_%s: Failed to open libav codec",
 371         oclass->in_plugin->name);
 372     return FALSE;
 373   }
 374 }
 375
 376 static void
 377 gst_ffmpegviddec_get_palette (GstFFMpegVidDec * ffmpegdec,
 378     GstVideoCodecState * state)
 379 {
 380   GstStructure *str = gst_caps_get_structure (state->caps, 0);
 381   const GValue *palette_v;
 382   GstBuffer *palette;
 383
 384   /* do we have a palette? */
 385   if ((palette_v = gst_structure_get_value (str, "palette_data"))) {
 386     palette = gst_value_get_buffer (palette_v);
 387     GST_DEBUG ("got palette data %p", palette);
 388     if (gst_buffer_get_size (palette) >= AVPALETTE_SIZE) {
 389       gst_buffer_replace (&ffmpegdec->palette, palette);
 390     }
 391   }
 392 }
 393
 394
 395 static gboolean
 396 gst_ffmpegviddec_set_format (GstVideoDecoder * decoder,
 397     GstVideoCodecState * state)
 398 {
 399   GstFFMpegVidDec *ffmpegdec;
 400   GstFFMpegVidDecClass *oclass;
 401   GstClockTime latency = GST_CLOCK_TIME_NONE;
 402   gboolean ret = FALSE;
 403
 404   ffmpegdec = (GstFFMpegVidDec *) decoder;
 405   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
 406
 407   if (ffmpegdec->last_caps != NULL &&
 408       gst_caps_is_equal (ffmpegdec->last_caps, state->caps)) {
 409     return TRUE;
 410   }
 411
 412   GST_DEBUG_OBJECT (ffmpegdec, "setcaps called");
 413
 414   GST_OBJECT_LOCK (ffmpegdec);
 415   /* stupid check for VC1 */
 416   if ((oclass->in_plugin->id == AV_CODEC_ID_WMV3) ||
 417       (oclass->in_plugin->id == AV_CODEC_ID_VC1))
 418     oclass->in_plugin->id = gst_ffmpeg_caps_to_codecid (state->caps, NULL);
 419
 420   /* close old session */
 421   if (ffmpegdec->opened) {
 422     GST_OBJECT_UNLOCK (ffmpegdec);
 423     gst_ffmpegviddec_finish (decoder);
 424     GST_OBJECT_LOCK (ffmpegdec);
 425     if (!gst_ffmpegviddec_close (ffmpegdec, TRUE)) {
 426       GST_OBJECT_UNLOCK (ffmpegdec);
 427       return FALSE;
 428     }
 429     ffmpegdec->pic_pix_fmt = 0;
 430     ffmpegdec->pic_width = 0;
 431     ffmpegdec->pic_height = 0;
 432     ffmpegdec->pic_par_n = 0;
 433     ffmpegdec->pic_par_d = 0;
 434     ffmpegdec->pic_interlaced = 0;
 435     ffmpegdec->pic_field_order = 0;
 436     ffmpegdec->pic_field_order_changed = FALSE;
 437     ffmpegdec->ctx_ticks = 0;
 438     ffmpegdec->ctx_time_n = 0;
 439     ffmpegdec->ctx_time_d = 0;
 440     ffmpegdec->cur_multiview_mode = GST_VIDEO_MULTIVIEW_MODE_NONE;
 441     ffmpegdec->cur_multiview_flags = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
 442   }
 443
 444   gst_caps_replace (&ffmpegdec->last_caps, state->caps);
 445
 446   /* set buffer functions */
 447   ffmpegdec->context->get_buffer2 = gst_ffmpegviddec_get_buffer2;
 448   ffmpegdec->context->draw_horiz_band = NULL;
 449
 450   /* reset coded_width/_height to prevent it being reused from last time when
 451    * the codec is opened again, causing a mismatch and possible
 452    * segfault/corruption. (Common scenario when renegotiating caps) */
 453   ffmpegdec->context->coded_width = 0;
 454   ffmpegdec->context->coded_height = 0;
 455
 456   GST_LOG_OBJECT (ffmpegdec, "size %dx%d", ffmpegdec->context->width,
 457       ffmpegdec->context->height);
 458
 459   /* FIXME : Create a method that takes GstVideoCodecState instead */
 460   /* get size and so */
 461   gst_ffmpeg_caps_with_codecid (oclass->in_plugin->id,
 462       oclass->in_plugin->type, state->caps, ffmpegdec->context);
 463
 464   GST_LOG_OBJECT (ffmpegdec, "size after %dx%d", ffmpegdec->context->width,
 465       ffmpegdec->context->height);
 466
 467   gst_ffmpegviddec_get_palette (ffmpegdec, state);
 468
 469   if (!ffmpegdec->context->time_base.den || !ffmpegdec->context->time_base.num) {
 470     GST_DEBUG_OBJECT (ffmpegdec, "forcing 25/1 framerate");
 471     ffmpegdec->context->time_base.num = 1;
 472     ffmpegdec->context->time_base.den = 25;
 473   }
 474
 475   /* workaround encoder bugs */
 476   ffmpegdec->context->workaround_bugs |= FF_BUG_AUTODETECT;
 477   ffmpegdec->context->err_recognition = 1;
 478
 479   /* for slow cpus */
 480   ffmpegdec->context->lowres = ffmpegdec->lowres;
 481   ffmpegdec->context->skip_frame = ffmpegdec->skip_frame;
 482
 483   /* ffmpeg can draw motion vectors on top of the image (not every decoder
 484    * supports it) */
 485   ffmpegdec->context->debug_mv = ffmpegdec->debug_mv;
 486
 487   {
 488     GstQuery *query;
 489     gboolean is_live;
 490
 491     if (ffmpegdec->max_threads == 0) {
 492       if (!(oclass->in_plugin->capabilities & CODEC_CAP_AUTO_THREADS))
 493         ffmpegdec->context->thread_count = gst_ffmpeg_auto_max_threads ();
 494       else
 495         ffmpegdec->context->thread_count = 0;
 496     } else
 497       ffmpegdec->context->thread_count = ffmpegdec->max_threads;
 498
 499     query = gst_query_new_latency ();
 500     is_live = FALSE;
 501     /* Check if upstream is live. If it isn't we can enable frame based
 502      * threading, which is adding latency */
 503     if (gst_pad_peer_query (GST_VIDEO_DECODER_SINK_PAD (ffmpegdec), query)) {
 504       gst_query_parse_latency (query, &is_live, NULL, NULL);
 505     }
 506     gst_query_unref (query);
 507
 508     if (is_live)
 509       ffmpegdec->context->thread_type = FF_THREAD_SLICE;
 510     else
 511       ffmpegdec->context->thread_type = FF_THREAD_SLICE | FF_THREAD_FRAME;
 512   }
 513
 514   /* open codec - we don't select an output pix_fmt yet,
 515    * simply because we don't know! We only get it
 516    * during playback... */
 517   if (!gst_ffmpegviddec_open (ffmpegdec))
 518     goto open_failed;
 519
 520   if (ffmpegdec->input_state)
 521     gst_video_codec_state_unref (ffmpegdec->input_state);
 522   ffmpegdec->input_state = gst_video_codec_state_ref (state);
 523
 524   if (ffmpegdec->input_state->info.fps_n) {
 525     GstVideoInfo *info = &ffmpegdec->input_state->info;
 526     latency = gst_util_uint64_scale_ceil (
 527         (ffmpegdec->context->has_b_frames) * GST_SECOND, info->fps_d,
 528         info->fps_n);
 529   }
 530
 531   ret = TRUE;
 532
 533 done:
 534   GST_OBJECT_UNLOCK (ffmpegdec);
 535
 536   if (GST_CLOCK_TIME_IS_VALID (latency))
 537     gst_video_decoder_set_latency (decoder, latency, latency);
 538
 539   return ret;
 540
 541   /* ERRORS */
 542 open_failed:
 543   {
 544     GST_DEBUG_OBJECT (ffmpegdec, "Failed to open");
 545     goto done;
 546   }
 547 }
 548
 549 typedef struct
 550 {
 551   GstFFMpegVidDec *ffmpegdec;
 552   GstVideoCodecFrame *frame;
 553   gboolean mapped;
 554   GstVideoFrame vframe;
 555   GstBuffer *buffer;
 556   AVBufferRef *avbuffer;
 557 } GstFFMpegVidDecVideoFrame;
 558
 559 static GstFFMpegVidDecVideoFrame *
 560 gst_ffmpegviddec_video_frame_new (GstFFMpegVidDec * ffmpegdec,
 561     GstVideoCodecFrame * frame)
 562 {
 563   GstFFMpegVidDecVideoFrame *dframe;
 564
 565   dframe = g_slice_new0 (GstFFMpegVidDecVideoFrame);
 566   dframe->ffmpegdec = ffmpegdec;
 567   dframe->frame = frame;
 568
 569   GST_DEBUG_OBJECT (ffmpegdec, "new video frame %p", dframe);
 570
 571   return dframe;
 572 }
 573
 574 static void
 575 gst_ffmpegviddec_video_frame_free (GstFFMpegVidDec * ffmpegdec,
 576     GstFFMpegVidDecVideoFrame * frame)
 577 {
 578   GST_DEBUG_OBJECT (ffmpegdec, "free video frame %p", frame);
 579
 580   if (frame->mapped)
 581     gst_video_frame_unmap (&frame->vframe);
 582   gst_video_decoder_release_frame (GST_VIDEO_DECODER (ffmpegdec), frame->frame);
 583   gst_buffer_replace (&frame->buffer, NULL);
 584   if (frame->avbuffer) {
 585     av_buffer_unref (&frame->avbuffer);
 586   }
 587   g_slice_free (GstFFMpegVidDecVideoFrame, frame);
 588 }
 589
 590 static void
 591 dummy_free_buffer (void *opaque, uint8_t * data)
 592 {
 593   GstFFMpegVidDecVideoFrame *frame = opaque;
 594
 595   gst_ffmpegviddec_video_frame_free (frame->ffmpegdec, frame);
 596 }
 597
 598 /* This function prepares the pool configuration for direct rendering. To use
 599  * this method, the codec should support direct rendering and the pool should
 600  * support video meta and video alignment */
 601 static void
 602 gst_ffmpegvideodec_prepare_dr_pool (GstFFMpegVidDec * ffmpegdec,
 603     GstBufferPool * pool, GstVideoInfo * info, GstStructure * config)
 604 {
 605   GstAllocationParams params;
 606   GstVideoAlignment align;
 607   GstAllocator *allocator = NULL;
 608   gint width, height;
 609   gint linesize_align[4];
 610   gint i;
 611   guint edge;
 612   gsize max_align;
 613
 614   width = GST_VIDEO_INFO_WIDTH (info);
 615   height = GST_VIDEO_INFO_HEIGHT (info);
 616
 617   /* let ffmpeg find the alignment and padding */
 618   avcodec_align_dimensions2 (ffmpegdec->context, &width, &height,
 619       linesize_align);
 620
 621   if (ffmpegdec->context->flags & CODEC_FLAG_EMU_EDGE)
 622     edge = 0;
 623   else
 624     edge = avcodec_get_edge_width ();
 625
 626   /* increase the size for the padding */
 627   width += edge << 1;
 628   height += edge << 1;
 629
 630   align.padding_top = edge;
 631   align.padding_left = edge;
 632   align.padding_right = width - GST_VIDEO_INFO_WIDTH (info) - edge;
 633   align.padding_bottom = height - GST_VIDEO_INFO_HEIGHT (info) - edge;
 634
 635   /* add extra padding to match libav buffer allocation sizes */
 636   align.padding_bottom++;
 637
 638   gst_buffer_pool_config_get_allocator (config, &allocator, &params);
 639
 640   max_align = DEFAULT_STRIDE_ALIGN;
 641   max_align |= params.align;
 642
 643   for (i = 0; i < GST_VIDEO_MAX_PLANES; i++) {
 644     if (linesize_align[i] > 0)
 645       max_align |= linesize_align[i] - 1;
 646   }
 647
 648   for (i = 0; i < GST_VIDEO_MAX_PLANES; i++)
 649     align.stride_align[i] = max_align;
 650
 651   params.align = max_align;
 652
 653   gst_buffer_pool_config_set_allocator (config, allocator, &params);
 654
 655   GST_DEBUG_OBJECT (ffmpegdec, "aligned dimension %dx%d -> %dx%d "
 656       "padding t:%u l:%u r:%u b:%u, stride_align %d:%d:%d:%d",
 657       GST_VIDEO_INFO_WIDTH (info),
 658       GST_VIDEO_INFO_HEIGHT (info), width, height, align.padding_top,
 659       align.padding_left, align.padding_right, align.padding_bottom,
 660       align.stride_align[0], align.stride_align[1], align.stride_align[2],
 661       align.stride_align[3]);
 662
 663   gst_buffer_pool_config_add_option (config,
 664       GST_BUFFER_POOL_OPTION_VIDEO_ALIGNMENT);
 665   gst_buffer_pool_config_set_video_alignment (config, &align);
 666 }
 667
 668 static void
 669 gst_ffmpegviddec_ensure_internal_pool (GstFFMpegVidDec * ffmpegdec,
 670     AVFrame * picture)
 671 {
 672   GstAllocationParams params = DEFAULT_ALLOC_PARAM;
 673   GstVideoInfo info;
 674   GstVideoFormat format;
 675   GstCaps *caps;
 676   GstStructure *config;
 677   gint i;
 678
 679   if (ffmpegdec->internal_pool != NULL &&
 680       ffmpegdec->pool_width == picture->width &&
 681       ffmpegdec->pool_height == picture->height &&
 682       ffmpegdec->pool_format == picture->format)
 683     return;
 684
 685   GST_DEBUG_OBJECT (ffmpegdec, "Updating internal pool (%i, %i)",
 686       picture->width, picture->height);
 687
 688   format = gst_ffmpeg_pixfmt_to_videoformat (picture->format);
 689   gst_video_info_set_format (&info, format, picture->width, picture->height);
 690
 691   for (i = 0; i < G_N_ELEMENTS (ffmpegdec->stride); i++)
 692     ffmpegdec->stride[i] = -1;
 693
 694   if (ffmpegdec->internal_pool)
 695     gst_object_unref (ffmpegdec->internal_pool);
 696
 697   ffmpegdec->internal_pool = gst_video_buffer_pool_new ();
 698   config = gst_buffer_pool_get_config (ffmpegdec->internal_pool);
 699
 700   caps = gst_video_info_to_caps (&info);
 701   gst_buffer_pool_config_set_params (config, caps, info.size, 2, 0);
 702   gst_buffer_pool_config_set_allocator (config, NULL, &params);
 703   gst_buffer_pool_config_add_option (config, GST_BUFFER_POOL_OPTION_VIDEO_META);
 704
 705   gst_ffmpegvideodec_prepare_dr_pool (ffmpegdec,
 706       ffmpegdec->internal_pool, &info, config);
 707   /* generic video pool never fails */
 708   gst_buffer_pool_set_config (ffmpegdec->internal_pool, config);
 709   gst_caps_unref (caps);
 710
 711   gst_buffer_pool_set_active (ffmpegdec->internal_pool, TRUE);
 712
 713   /* Remember pool size so we can detect changes */
 714   ffmpegdec->pool_width = picture->width;
 715   ffmpegdec->pool_height = picture->height;
 716   ffmpegdec->pool_format = picture->format;
 717   ffmpegdec->pool_info = info;
 718 }
 719
 720 static gboolean
 721 gst_ffmpegviddec_can_direct_render (GstFFMpegVidDec * ffmpegdec)
 722 {
 723   GstFFMpegVidDecClass *oclass;
 724
 725   if (!ffmpegdec->direct_rendering)
 726     return FALSE;
 727
 728   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
 729   return ((oclass->in_plugin->capabilities & CODEC_CAP_DR1) == CODEC_CAP_DR1);
 730 }
 731
 732 /* called when ffmpeg wants us to allocate a buffer to write the decoded frame
 733  * into. We try to give it memory from our pool */
 734 static int
 735 gst_ffmpegviddec_get_buffer2 (AVCodecContext * context, AVFrame * picture,
 736     int flags)
 737 {
 738   GstVideoCodecFrame *frame;
 739   GstFFMpegVidDecVideoFrame *dframe;
 740   GstFFMpegVidDec *ffmpegdec;
 741   gint c;
 742   GstFlowReturn ret;
 743
 744   ffmpegdec = (GstFFMpegVidDec *) context->opaque;
 745
 746   GST_DEBUG_OBJECT (ffmpegdec, "getting buffer picture %p", picture);
 747
 748   /* apply the last info we have seen to this picture, when we get the
 749    * picture back from ffmpeg we can use this to correctly timestamp the output
 750    * buffer */
 751   GST_DEBUG_OBJECT (ffmpegdec, "opaque value SN %d",
 752       (gint32) picture->reordered_opaque);
 753
 754   frame =
 755       gst_video_decoder_get_frame (GST_VIDEO_DECODER (ffmpegdec),
 756       picture->reordered_opaque);
 757   if (G_UNLIKELY (frame == NULL))
 758     goto no_frame;
 759
 760   /* now it has a buffer allocated, so it is real and will also
 761    * be _released */
 762   GST_VIDEO_CODEC_FRAME_FLAG_UNSET (frame,
 763       GST_VIDEO_CODEC_FRAME_FLAG_DECODE_ONLY);
 764
 765   if (G_UNLIKELY (frame->output_buffer != NULL))
 766     goto duplicate_frame;
 767
 768   /* GstFFMpegVidDecVideoFrame receives the frame ref */
 769   if (picture->opaque) {
 770     dframe = picture->opaque;
 771     dframe->frame = frame;
 772   } else {
 773     picture->opaque = dframe =
 774         gst_ffmpegviddec_video_frame_new (ffmpegdec, frame);
 775   }
 776
 777   GST_DEBUG_OBJECT (ffmpegdec, "storing opaque %p", dframe);
 778
 779   if (!gst_ffmpegviddec_can_direct_render (ffmpegdec))
 780     goto no_dr;
 781
 782   gst_ffmpegviddec_ensure_internal_pool (ffmpegdec, picture);
 783
 784   ret = gst_buffer_pool_acquire_buffer (ffmpegdec->internal_pool,
 785       &frame->output_buffer, NULL);
 786   if (ret != GST_FLOW_OK)
 787     goto alloc_failed;
 788
 789   /* piggy-backed alloc'ed on the frame,
 790    * and there was much rejoicing and we are grateful.
 791    * Now take away buffer from frame, we will give it back later when decoded.
 792    * This allows multiple request for a buffer per frame; unusual but possible. */
 793   gst_buffer_replace (&dframe->buffer, frame->output_buffer);
 794   gst_buffer_replace (&frame->output_buffer, NULL);
 795
 796   /* Fill avpicture */
 797   if (!gst_video_frame_map (&dframe->vframe, &ffmpegdec->pool_info,
 798           dframe->buffer, GST_MAP_READWRITE))
 799     goto map_failed;
 800   dframe->mapped = TRUE;
 801
 802   for (c = 0; c < AV_NUM_DATA_POINTERS; c++) {
 803     if (c < GST_VIDEO_INFO_N_PLANES (&ffmpegdec->pool_info)) {
 804       picture->data[c] = GST_VIDEO_FRAME_PLANE_DATA (&dframe->vframe, c);
 805       picture->linesize[c] = GST_VIDEO_FRAME_PLANE_STRIDE (&dframe->vframe, c);
 806
 807       if (ffmpegdec->stride[c] == -1)
 808         ffmpegdec->stride[c] = picture->linesize[c];
 809
 810       /* libav does not allow stride changes, decide allocation should check
 811        * before replacing the internal pool with a downstream pool.
 812        * https://bugzilla.gnome.org/show_bug.cgi?id=704769
 813        * https://bugzilla.libav.org/show_bug.cgi?id=556
 814        */
 815       g_assert (picture->linesize[c] == ffmpegdec->stride[c]);
 816     } else {
 817       picture->data[c] = NULL;
 818       picture->linesize[c] = 0;
 819     }
 820     GST_LOG_OBJECT (ffmpegdec, "linesize %d, data %p", picture->linesize[c],
 821         picture->data[c]);
 822   }
 823
 824   picture->buf[0] = av_buffer_create (NULL, 0, dummy_free_buffer, dframe, 0);
 825
 826   GST_LOG_OBJECT (ffmpegdec, "returned frame %p", dframe->buffer);
 827
 828   return 0;
 829
 830 no_dr:
 831   {
 832     int c;
 833     int ret = avcodec_default_get_buffer2 (context, picture, flags);
 834
 835     GST_LOG_OBJECT (ffmpegdec, "direct rendering disabled, fallback alloc");
 836
 837     for (c = 0; c < AV_NUM_DATA_POINTERS; c++) {
 838       ffmpegdec->stride[c] = picture->linesize[c];
 839     }
 840     /* Wrap our buffer around the default one to be able to have a callback
 841      * when our data can be freed. Just putting our data into the first free
 842      * buffer might not work if there are too many allocated already
 843      */
 844     if (picture->buf[0]) {
 845       dframe->avbuffer = picture->buf[0];
 846       picture->buf[0] =
 847           av_buffer_create (picture->buf[0]->data, picture->buf[0]->size,
 848           dummy_free_buffer, dframe, 0);
 849     } else {
 850       picture->buf[0] =
 851           av_buffer_create (NULL, 0, dummy_free_buffer, dframe, 0);
 852     }
 853
 854     return ret;
 855   }
 856 alloc_failed:
 857   {
 858     GST_ELEMENT_ERROR (ffmpegdec, RESOURCE, FAILED,
 859         ("Unable to allocate memory"),
 860         ("The downstream pool failed to allocated buffer."));
 861     return -1;
 862   }
 863 map_failed:
 864   {
 865     GST_ELEMENT_ERROR (ffmpegdec, RESOURCE, OPEN_READ_WRITE,
 866         ("Cannot access memory for read and write operation."),
 867         ("The video memory allocated from downstream pool could not mapped for"
 868             "read and write."));
 869     return -1;
 870   }
 871 duplicate_frame:
 872   {
 873     GST_WARNING_OBJECT (ffmpegdec, "already alloc'ed output buffer for frame");
 874     gst_video_codec_frame_unref (frame);
 875     return -1;
 876   }
 877 no_frame:
 878   {
 879     GST_WARNING_OBJECT (ffmpegdec, "Couldn't get codec frame !");
 880     return -1;
 881   }
 882 }
 883
 884 static gboolean
 885 picture_changed (GstFFMpegVidDec * ffmpegdec, AVFrame * picture)
 886 {
 887   gint pic_field_order = 0;
 888
 889   if (picture->interlaced_frame) {
 890     if (picture->repeat_pict)
 891       pic_field_order |= GST_VIDEO_BUFFER_FLAG_RFF;
 892     if (picture->top_field_first)
 893       pic_field_order |= GST_VIDEO_BUFFER_FLAG_TFF;
 894   }
 895
 896   return !(ffmpegdec->pic_width == picture->width
 897       && ffmpegdec->pic_height == picture->height
 898       && ffmpegdec->pic_pix_fmt == picture->format
 899       && ffmpegdec->pic_par_n == picture->sample_aspect_ratio.num
 900       && ffmpegdec->pic_par_d == picture->sample_aspect_ratio.den
 901       && ffmpegdec->pic_interlaced == picture->interlaced_frame
 902       && ffmpegdec->pic_field_order == pic_field_order
 903       && ffmpegdec->cur_multiview_mode == ffmpegdec->picture_multiview_mode
 904       && ffmpegdec->cur_multiview_flags == ffmpegdec->picture_multiview_flags);
 905 }
 906
 907 static gboolean
 908 context_changed (GstFFMpegVidDec * ffmpegdec, AVCodecContext * context)
 909 {
 910   return !(ffmpegdec->ctx_ticks == context->ticks_per_frame
 911       && ffmpegdec->ctx_time_n == context->time_base.num
 912       && ffmpegdec->ctx_time_d == context->time_base.den);
 913 }
 914
 915 static gboolean
 916 update_video_context (GstFFMpegVidDec * ffmpegdec, AVCodecContext * context,
 917     AVFrame * picture)
 918 {
 919   gint pic_field_order = 0;
 920
 921   if (picture->interlaced_frame) {
 922     if (picture->repeat_pict)
 923       pic_field_order |= GST_VIDEO_BUFFER_FLAG_RFF;
 924     if (picture->top_field_first)
 925       pic_field_order |= GST_VIDEO_BUFFER_FLAG_TFF;
 926   }
 927
 928   if (!picture_changed (ffmpegdec, picture)
 929       && !context_changed (ffmpegdec, context))
 930     return FALSE;
 931
 932   GST_DEBUG_OBJECT (ffmpegdec,
 933       "Renegotiating video from %dx%d@ %d:%d PAR %d/%d fps pixfmt %d to %dx%d@ %d:%d PAR %d/%d fps pixfmt %d",
 934       ffmpegdec->pic_width, ffmpegdec->pic_height,
 935       ffmpegdec->pic_par_n, ffmpegdec->pic_par_d,
 936       ffmpegdec->ctx_time_n, ffmpegdec->ctx_time_d,
 937       ffmpegdec->pic_pix_fmt,
 938       picture->width, picture->height,
 939       picture->sample_aspect_ratio.num,
 940       picture->sample_aspect_ratio.den,
 941       context->time_base.num, context->time_base.den, picture->format);
 942
 943   ffmpegdec->pic_pix_fmt = picture->format;
 944   ffmpegdec->pic_width = picture->width;
 945   ffmpegdec->pic_height = picture->height;
 946   ffmpegdec->pic_par_n = picture->sample_aspect_ratio.num;
 947   ffmpegdec->pic_par_d = picture->sample_aspect_ratio.den;
 948   ffmpegdec->cur_multiview_mode = ffmpegdec->picture_multiview_mode;
 949   ffmpegdec->cur_multiview_flags = ffmpegdec->picture_multiview_flags;
 950
 951   /* Remember if we have interlaced content and the field order changed
 952    * at least once. If that happens, we must be interlace-mode=mixed
 953    */
 954   if (ffmpegdec->pic_field_order_changed ||
 955       (ffmpegdec->pic_field_order != pic_field_order &&
 956           ffmpegdec->pic_interlaced))
 957     ffmpegdec->pic_field_order_changed = TRUE;
 958
 959   ffmpegdec->pic_field_order = pic_field_order;
 960   ffmpegdec->pic_interlaced = picture->interlaced_frame;
 961
 962   if (!ffmpegdec->pic_interlaced)
 963     ffmpegdec->pic_field_order_changed = FALSE;
 964
 965   ffmpegdec->ctx_ticks = context->ticks_per_frame;
 966   ffmpegdec->ctx_time_n = context->time_base.num;
 967   ffmpegdec->ctx_time_d = context->time_base.den;
 968
 969   return TRUE;
 970 }
 971
 972 static void
 973 gst_ffmpegviddec_update_par (GstFFMpegVidDec * ffmpegdec,
 974     GstVideoInfo * in_info, GstVideoInfo * out_info)
 975 {
 976   gboolean demuxer_par_set = FALSE;
 977   gboolean decoder_par_set = FALSE;
 978   gint demuxer_num = 1, demuxer_denom = 1;
 979   gint decoder_num = 1, decoder_denom = 1;
 980
 981   if (in_info->par_n && in_info->par_d) {
 982     demuxer_num = in_info->par_n;
 983     demuxer_denom = in_info->par_d;
 984     demuxer_par_set = TRUE;
 985     GST_DEBUG_OBJECT (ffmpegdec, "Demuxer PAR: %d:%d", demuxer_num,
 986         demuxer_denom);
 987   }
 988
 989   if (ffmpegdec->pic_par_n && ffmpegdec->pic_par_d) {
 990     decoder_num = ffmpegdec->pic_par_n;
 991     decoder_denom = ffmpegdec->pic_par_d;
 992     decoder_par_set = TRUE;
 993     GST_DEBUG_OBJECT (ffmpegdec, "Decoder PAR: %d:%d", decoder_num,
 994         decoder_denom);
 995   }
 996
 997   if (!demuxer_par_set && !decoder_par_set)
 998     goto no_par;
 999
1000   if (demuxer_par_set && !decoder_par_set)
1001     goto use_demuxer_par;
1002
1003   if (decoder_par_set && !demuxer_par_set)
1004     goto use_decoder_par;
1005
1006   /* Both the demuxer and the decoder provide a PAR. If one of
1007    * the two PARs is 1:1 and the other one is not, use the one
1008    * that is not 1:1. */
1009   if (demuxer_num == demuxer_denom && decoder_num != decoder_denom)
1010     goto use_decoder_par;
1011
1012   if (decoder_num == decoder_denom && demuxer_num != demuxer_denom)
1013     goto use_demuxer_par;
1014
1015   /* Both PARs are non-1:1, so use the PAR provided by the demuxer */
1016   goto use_demuxer_par;
1017
1018 use_decoder_par:
1019   {
1020     GST_DEBUG_OBJECT (ffmpegdec,
1021         "Setting decoder provided pixel-aspect-ratio of %u:%u", decoder_num,
1022         decoder_denom);
1023     out_info->par_n = decoder_num;
1024     out_info->par_d = decoder_denom;
1025     return;
1026   }
1027 use_demuxer_par:
1028   {
1029     GST_DEBUG_OBJECT (ffmpegdec,
1030         "Setting demuxer provided pixel-aspect-ratio of %u:%u", demuxer_num,
1031         demuxer_denom);
1032     out_info->par_n = demuxer_num;
1033     out_info->par_d = demuxer_denom;
1034     return;
1035   }
1036 no_par:
1037   {
1038     GST_DEBUG_OBJECT (ffmpegdec,
1039         "Neither demuxer nor codec provide a pixel-aspect-ratio");
1040     out_info->par_n = 1;
1041     out_info->par_d = 1;
1042     return;
1043   }
1044 }
1045
1046 static GstVideoMultiviewMode
1047 stereo_av_to_gst (enum AVStereo3DType type)
1048 {
1049   switch (type) {
1050     case AV_STEREO3D_SIDEBYSIDE:
1051       return GST_VIDEO_MULTIVIEW_MODE_SIDE_BY_SIDE;
1052     case AV_STEREO3D_TOPBOTTOM:
1053       return GST_VIDEO_MULTIVIEW_MODE_TOP_BOTTOM;
1054     case AV_STEREO3D_FRAMESEQUENCE:
1055       return GST_VIDEO_MULTIVIEW_MODE_FRAME_BY_FRAME;
1056     case AV_STEREO3D_CHECKERBOARD:
1057       return GST_VIDEO_MULTIVIEW_MODE_CHECKERBOARD;
1058     case AV_STEREO3D_SIDEBYSIDE_QUINCUNX:
1059       return GST_VIDEO_MULTIVIEW_MODE_SIDE_BY_SIDE_QUINCUNX;
1060     case AV_STEREO3D_LINES:
1061       return GST_VIDEO_MULTIVIEW_MODE_ROW_INTERLEAVED;
1062     case AV_STEREO3D_COLUMNS:
1063       return GST_VIDEO_MULTIVIEW_MODE_COLUMN_INTERLEAVED;
1064     default:
1065       break;
1066   }
1067
1068   return GST_VIDEO_MULTIVIEW_MODE_NONE;
1069 }
1070
1071 static gboolean
1072 gst_ffmpegviddec_negotiate (GstFFMpegVidDec * ffmpegdec,
1073     AVCodecContext * context, AVFrame * picture)
1074 {
1075   GstVideoFormat fmt;
1076   GstVideoInfo *in_info, *out_info;
1077   GstVideoCodecState *output_state;
1078   gint fps_n, fps_d;
1079   GstClockTime latency;
1080   GstStructure *in_s;
1081
1082   if (!update_video_context (ffmpegdec, context, picture))
1083     return TRUE;
1084
1085   fmt = gst_ffmpeg_pixfmt_to_videoformat (ffmpegdec->pic_pix_fmt);
1086   if (G_UNLIKELY (fmt == GST_VIDEO_FORMAT_UNKNOWN))
1087     goto unknown_format;
1088
1089   output_state =
1090       gst_video_decoder_set_output_state (GST_VIDEO_DECODER (ffmpegdec), fmt,
1091       ffmpegdec->pic_width, ffmpegdec->pic_height, ffmpegdec->input_state);
1092   if (ffmpegdec->output_state)
1093     gst_video_codec_state_unref (ffmpegdec->output_state);
1094   ffmpegdec->output_state = output_state;
1095
1096   in_info = &ffmpegdec->input_state->info;
1097   out_info = &ffmpegdec->output_state->info;
1098
1099   /* set the interlaced flag */
1100   in_s = gst_caps_get_structure (ffmpegdec->input_state->caps, 0);
1101
1102   if (!gst_structure_has_field (in_s, "interlace-mode")) {
1103     if (ffmpegdec->pic_interlaced) {
1104       if (ffmpegdec->pic_field_order_changed ||
1105           (ffmpegdec->pic_field_order & GST_VIDEO_BUFFER_FLAG_RFF)) {
1106         out_info->interlace_mode = GST_VIDEO_INTERLACE_MODE_MIXED;
1107       } else {
1108         out_info->interlace_mode = GST_VIDEO_INTERLACE_MODE_INTERLEAVED;
1109         if ((ffmpegdec->pic_field_order & GST_VIDEO_BUFFER_FLAG_TFF))
1110           GST_VIDEO_INFO_FIELD_ORDER (out_info) =
1111               GST_VIDEO_FIELD_ORDER_TOP_FIELD_FIRST;
1112         else
1113           GST_VIDEO_INFO_FIELD_ORDER (out_info) =
1114               GST_VIDEO_FIELD_ORDER_BOTTOM_FIELD_FIRST;
1115       }
1116     } else {
1117       out_info->interlace_mode = GST_VIDEO_INTERLACE_MODE_PROGRESSIVE;
1118     }
1119   }
1120
1121   if (!gst_structure_has_field (in_s, "chroma-site")) {
1122     switch (context->chroma_sample_location) {
1123       case AVCHROMA_LOC_LEFT:
1124         out_info->chroma_site = GST_VIDEO_CHROMA_SITE_MPEG2;
1125         break;
1126       case AVCHROMA_LOC_CENTER:
1127         out_info->chroma_site = GST_VIDEO_CHROMA_SITE_JPEG;
1128         break;
1129       case AVCHROMA_LOC_TOPLEFT:
1130         out_info->chroma_site = GST_VIDEO_CHROMA_SITE_DV;
1131         break;
1132       case AVCHROMA_LOC_TOP:
1133         out_info->chroma_site = GST_VIDEO_CHROMA_SITE_V_COSITED;
1134         break;
1135       default:
1136         break;
1137     }
1138   }
1139
1140   if (!gst_structure_has_field (in_s, "colorimetry")
1141       || in_info->colorimetry.primaries == GST_VIDEO_COLOR_PRIMARIES_UNKNOWN) {
1142     switch (context->color_primaries) {
1143       case AVCOL_PRI_BT709:
1144         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_BT709;
1145         break;
1146       case AVCOL_PRI_BT470M:
1147         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_BT470M;
1148         break;
1149       case AVCOL_PRI_BT470BG:
1150         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_BT470BG;
1151         break;
1152       case AVCOL_PRI_SMPTE170M:
1153         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_SMPTE170M;
1154         break;
1155       case AVCOL_PRI_SMPTE240M:
1156         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_SMPTE240M;
1157         break;
1158       case AVCOL_PRI_FILM:
1159         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_FILM;
1160         break;
1161       case AVCOL_PRI_BT2020:
1162         out_info->colorimetry.primaries = GST_VIDEO_COLOR_PRIMARIES_BT2020;
1163         break;
1164       default:
1165         break;
1166     }
1167   }
1168
1169   if (!gst_structure_has_field (in_s, "colorimetry")
1170       || in_info->colorimetry.transfer == GST_VIDEO_TRANSFER_UNKNOWN) {
1171     switch (context->color_trc) {
1172       case AVCOL_TRC_BT2020_10:
1173       case AVCOL_TRC_BT709:
1174       case AVCOL_TRC_SMPTE170M:
1175         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_BT709;
1176         break;
1177       case AVCOL_TRC_GAMMA22:
1178         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_GAMMA22;
1179         break;
1180       case AVCOL_TRC_GAMMA28:
1181         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_GAMMA28;
1182         break;
1183       case AVCOL_TRC_SMPTE240M:
1184         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_SMPTE240M;
1185         break;
1186       case AVCOL_TRC_LINEAR:
1187         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_GAMMA10;
1188         break;
1189       case AVCOL_TRC_LOG:
1190         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_LOG100;
1191         break;
1192       case AVCOL_TRC_LOG_SQRT:
1193         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_LOG316;
1194         break;
1195       case AVCOL_TRC_BT2020_12:
1196         out_info->colorimetry.transfer = GST_VIDEO_TRANSFER_BT2020_12;
1197         break;
1198       default:
1199         break;
1200     }
1201   }
1202
1203   if (!gst_structure_has_field (in_s, "colorimetry")
1204       || in_info->colorimetry.matrix == GST_VIDEO_COLOR_MATRIX_UNKNOWN) {
1205     switch (context->colorspace) {
1206       case AVCOL_SPC_RGB:
1207         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_RGB;
1208         break;
1209       case AVCOL_SPC_BT709:
1210         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_BT709;
1211         break;
1212       case AVCOL_SPC_FCC:
1213         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_FCC;
1214         break;
1215       case AVCOL_SPC_BT470BG:
1216       case AVCOL_SPC_SMPTE170M:
1217         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_BT601;
1218         break;
1219       case AVCOL_SPC_SMPTE240M:
1220         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_SMPTE240M;
1221         break;
1222       case AVCOL_SPC_BT2020_NCL:
1223         out_info->colorimetry.matrix = GST_VIDEO_COLOR_MATRIX_BT2020;
1224         break;
1225       default:
1226         break;
1227     }
1228   }
1229
1230   if (!gst_structure_has_field (in_s, "colorimetry")
1231       || in_info->colorimetry.range == GST_VIDEO_COLOR_RANGE_UNKNOWN) {
1232     if (context->color_range == AVCOL_RANGE_JPEG) {
1233       out_info->colorimetry.range = GST_VIDEO_COLOR_RANGE_0_255;
1234     } else {
1235       out_info->colorimetry.range = GST_VIDEO_COLOR_RANGE_16_235;
1236     }
1237   }
1238
1239   /* try to find a good framerate */
1240   if ((in_info->fps_d && in_info->fps_n) ||
1241       GST_VIDEO_INFO_FLAG_IS_SET (in_info, GST_VIDEO_FLAG_VARIABLE_FPS)) {
1242     /* take framerate from input when it was specified (#313970) */
1243     fps_n = in_info->fps_n;
1244     fps_d = in_info->fps_d;
1245   } else {
1246     fps_n = ffmpegdec->ctx_time_d / ffmpegdec->ctx_ticks;
1247     fps_d = ffmpegdec->ctx_time_n;
1248
1249     if (!fps_d) {
1250       GST_LOG_OBJECT (ffmpegdec, "invalid framerate: %d/0, -> %d/1", fps_n,
1251           fps_n);
1252       fps_d = 1;
1253     }
1254     if (gst_util_fraction_compare (fps_n, fps_d, 1000, 1) > 0) {
1255       GST_LOG_OBJECT (ffmpegdec, "excessive framerate: %d/%d, -> 0/1", fps_n,
1256           fps_d);
1257       fps_n = 0;
1258       fps_d = 1;
1259     }
1260   }
1261
1262   GST_LOG_OBJECT (ffmpegdec, "setting framerate: %d/%d", fps_n, fps_d);
1263   out_info->fps_n = fps_n;
1264   out_info->fps_d = fps_d;
1265
1266   /* calculate and update par now */
1267   gst_ffmpegviddec_update_par (ffmpegdec, in_info, out_info);
1268
1269   GST_VIDEO_INFO_MULTIVIEW_MODE (out_info) = ffmpegdec->cur_multiview_mode;
1270   GST_VIDEO_INFO_MULTIVIEW_FLAGS (out_info) = ffmpegdec->cur_multiview_flags;
1271
1272   if (!gst_video_decoder_negotiate (GST_VIDEO_DECODER (ffmpegdec)))
1273     goto negotiate_failed;
1274
1275   /* The decoder is configured, we now know the true latency */
1276   if (fps_n) {
1277     latency =
1278         gst_util_uint64_scale_ceil (ffmpegdec->context->has_b_frames *
1279         GST_SECOND, fps_d, fps_n);
1280     gst_video_decoder_set_latency (GST_VIDEO_DECODER (ffmpegdec), latency,
1281         latency);
1282   }
1283
1284   return TRUE;
1285
1286   /* ERRORS */
1287 unknown_format:
1288   {
1289     GST_ERROR_OBJECT (ffmpegdec,
1290         "decoder requires a video format unsupported by GStreamer");
1291     return FALSE;
1292   }
1293 negotiate_failed:
1294   {
1295     /* Reset so we try again next time even if force==FALSE */
1296     ffmpegdec->pic_pix_fmt = 0;
1297     ffmpegdec->pic_width = 0;
1298     ffmpegdec->pic_height = 0;
1299     ffmpegdec->pic_par_n = 0;
1300     ffmpegdec->pic_par_d = 0;
1301     ffmpegdec->pic_interlaced = 0;
1302     ffmpegdec->pic_field_order = 0;
1303     ffmpegdec->pic_field_order_changed = FALSE;
1304     ffmpegdec->ctx_ticks = 0;
1305     ffmpegdec->ctx_time_n = 0;
1306     ffmpegdec->ctx_time_d = 0;
1307
1308     GST_ERROR_OBJECT (ffmpegdec, "negotiation failed");
1309     return FALSE;
1310   }
1311 }
1312
1313 /* perform qos calculations before decoding the next frame.
1314  *
1315  * Sets the skip_frame flag and if things are really bad, skips to the next
1316  * keyframe.
1317  *
1318  */
1319 static void
1320 gst_ffmpegviddec_do_qos (GstFFMpegVidDec * ffmpegdec,
1321     GstVideoCodecFrame * frame, gboolean * mode_switch)
1322 {
1323   GstClockTimeDiff diff;
1324   GstSegmentFlags skip_flags =
1325       GST_VIDEO_DECODER_INPUT_SEGMENT (ffmpegdec).flags;
1326
1327   *mode_switch = FALSE;
1328
1329   if (frame == NULL)
1330     return;
1331
1332   if (skip_flags & GST_SEGMENT_FLAG_TRICKMODE_KEY_UNITS) {
1333     ffmpegdec->context->skip_frame = AVDISCARD_NONKEY;
1334     *mode_switch = TRUE;
1335   } else if (skip_flags & GST_SEGMENT_FLAG_TRICKMODE) {
1336     ffmpegdec->context->skip_frame = AVDISCARD_NONREF;
1337     *mode_switch = TRUE;
1338   }
1339
1340   if (*mode_switch == TRUE) {
1341     /* We've already switched mode, we can return straight away
1342      * without any further calculation */
1343     return;
1344   }
1345
1346   diff =
1347       gst_video_decoder_get_max_decode_time (GST_VIDEO_DECODER (ffmpegdec),
1348       frame);
1349
1350   /* if we don't have timing info, then we don't do QoS */
1351   if (G_UNLIKELY (diff == G_MAXINT64)) {
1352     /* Ensure the skipping strategy is the default one */
1353     ffmpegdec->context->skip_frame = ffmpegdec->skip_frame;
1354     return;
1355   }
1356
1357   GST_DEBUG_OBJECT (ffmpegdec, "decoding time %" G_GINT64_FORMAT, diff);
1358
1359   if (diff > 0 && ffmpegdec->context->skip_frame != AVDISCARD_DEFAULT) {
1360     ffmpegdec->context->skip_frame = AVDISCARD_DEFAULT;
1361     *mode_switch = TRUE;
1362     GST_DEBUG_OBJECT (ffmpegdec, "QOS: normal mode");
1363   }
1364
1365   else if (diff <= 0 && ffmpegdec->context->skip_frame != AVDISCARD_NONREF) {
1366     ffmpegdec->context->skip_frame = AVDISCARD_NONREF;
1367     *mode_switch = TRUE;
1368     GST_DEBUG_OBJECT (ffmpegdec,
1369         "QOS: hurry up, diff %" G_GINT64_FORMAT " >= 0", diff);
1370   }
1371 }
1372
1373 /* get an outbuf buffer with the current picture */
1374 static GstFlowReturn
1375 get_output_buffer (GstFFMpegVidDec * ffmpegdec, GstVideoCodecFrame * frame)
1376 {
1377   GstFlowReturn ret = GST_FLOW_OK;
1378   AVFrame pic, *outpic;
1379   GstVideoFrame vframe;
1380   GstVideoInfo *info;
1381   gint c;
1382
1383   GST_LOG_OBJECT (ffmpegdec, "get output buffer");
1384
1385   if (!ffmpegdec->output_state)
1386     goto not_negotiated;
1387
1388   ret =
1389       gst_video_decoder_allocate_output_frame (GST_VIDEO_DECODER (ffmpegdec),
1390       frame);
1391   if (G_UNLIKELY (ret != GST_FLOW_OK))
1392     goto alloc_failed;
1393
1394   /* original ffmpeg code does not handle odd sizes correctly.
1395    * This patched up version does */
1396   /* Fill avpicture */
1397   info = &ffmpegdec->output_state->info;
1398   if (!gst_video_frame_map (&vframe, info, frame->output_buffer,
1399           GST_MAP_READ | GST_MAP_WRITE))
1400     goto map_failed;
1401
1402   memset (&pic, 0, sizeof (pic));
1403   pic.format = ffmpegdec->pic_pix_fmt;
1404   pic.width = GST_VIDEO_FRAME_WIDTH (&vframe);
1405   pic.height = GST_VIDEO_FRAME_HEIGHT (&vframe);
1406   for (c = 0; c < AV_NUM_DATA_POINTERS; c++) {
1407     if (c < GST_VIDEO_INFO_N_PLANES (info)) {
1408       pic.data[c] = GST_VIDEO_FRAME_PLANE_DATA (&vframe, c);
1409       pic.linesize[c] = GST_VIDEO_FRAME_PLANE_STRIDE (&vframe, c);
1410       GST_LOG_OBJECT (ffmpegdec, "[%i] linesize %d, data %p", c,
1411           pic.linesize[c], pic.data[c]);
1412     } else {
1413       pic.data[c] = NULL;
1414       pic.linesize[c] = 0;
1415     }
1416   }
1417
1418   outpic = ffmpegdec->picture;
1419
1420   if (av_frame_copy (&pic, outpic) != 0) {
1421     GST_ERROR_OBJECT (ffmpegdec, "Failed to copy output frame");
1422     ret = GST_FLOW_ERROR;
1423   }
1424
1425   gst_video_frame_unmap (&vframe);
1426
1427   ffmpegdec->picture->reordered_opaque = -1;
1428
1429   return ret;
1430
1431   /* special cases */
1432 alloc_failed:
1433   {
1434     GST_ELEMENT_ERROR (ffmpegdec, RESOURCE, FAILED,
1435         ("Unable to allocate memory"),
1436         ("The downstream pool failed to allocated buffer."));
1437     return ret;
1438   }
1439 map_failed:
1440   {
1441     GST_ELEMENT_ERROR (ffmpegdec, RESOURCE, OPEN_READ_WRITE,
1442         ("Cannot access memory for read and write operation."),
1443         ("The video memory allocated from downstream pool could not mapped for"
1444             "read and write."));
1445     return ret;
1446   }
1447 not_negotiated:
1448   {
1449     GST_DEBUG_OBJECT (ffmpegdec, "not negotiated");
1450     return GST_FLOW_NOT_NEGOTIATED;
1451   }
1452 }
1453
1454 static void
1455 gst_avpacket_init (AVPacket * packet, guint8 * data, guint size)
1456 {
1457   memset (packet, 0, sizeof (AVPacket));
1458   packet->data = data;
1459   packet->size = size;
1460 }
1461
1462 /* gst_ffmpegviddec_[video|audio]_frame:
1463  * ffmpegdec:
1464  * data: pointer to the data to decode
1465  * size: size of data in bytes
1466  * in_timestamp: incoming timestamp.
1467  * in_duration: incoming duration.
1468  * in_offset: incoming offset (frame number).
1469  * ret: Return flow.
1470  *
1471  * Returns: number of bytes used in decoding. The check for successful decode is
1472  *   outbuf being non-NULL.
1473  */
1474 static gint
1475 gst_ffmpegviddec_video_frame (GstFFMpegVidDec * ffmpegdec,
1476     guint8 * data, guint size, gint * have_data, GstVideoCodecFrame * frame,
1477     GstFlowReturn * ret)
1478 {
1479   gint len = -1;
1480   gboolean mode_switch;
1481   GstVideoCodecFrame *out_frame;
1482   GstFFMpegVidDecVideoFrame *out_dframe;
1483   AVPacket packet;
1484   GstBufferPool *pool;
1485
1486   *ret = GST_FLOW_OK;
1487
1488   /* in case we skip frames */
1489   ffmpegdec->picture->pict_type = -1;
1490
1491   /* run QoS code, we don't stop decoding the frame when we are late because
1492    * else we might skip a reference frame */
1493   gst_ffmpegviddec_do_qos (ffmpegdec, frame, &mode_switch);
1494
1495   if (frame) {
1496     /* save reference to the timing info */
1497     ffmpegdec->context->reordered_opaque = (gint64) frame->system_frame_number;
1498     ffmpegdec->picture->reordered_opaque = (gint64) frame->system_frame_number;
1499
1500     GST_DEBUG_OBJECT (ffmpegdec, "stored opaque values idx %d",
1501         frame->system_frame_number);
1502   }
1503
1504   /* now decode the frame */
1505   gst_avpacket_init (&packet, data, size);
1506
1507   if (ffmpegdec->palette) {
1508     guint8 *pal;
1509
1510     pal = av_packet_new_side_data (&packet, AV_PKT_DATA_PALETTE,
1511         AVPALETTE_SIZE);
1512     gst_buffer_extract (ffmpegdec->palette, 0, pal, AVPALETTE_SIZE);
1513     GST_DEBUG_OBJECT (ffmpegdec, "copy pal %p %p", &packet, pal);
1514   }
1515
1516   /* This might call into get_buffer() from another thread,
1517    * which would cause a deadlock. Release the lock here
1518    * and taking it again later seems safe
1519    * See https://bugzilla.gnome.org/show_bug.cgi?id=726020
1520    */
1521   GST_VIDEO_DECODER_STREAM_UNLOCK (ffmpegdec);
1522   len = avcodec_decode_video2 (ffmpegdec->context,
1523       ffmpegdec->picture, have_data, &packet);
1524   GST_VIDEO_DECODER_STREAM_LOCK (ffmpegdec);
1525
1526   GST_DEBUG_OBJECT (ffmpegdec, "after decode: len %d, have_data %d",
1527       len, *have_data);
1528
1529   /* when we are in skip_frame mode, don't complain when ffmpeg returned
1530    * no data because we told it to skip stuff. */
1531   if (len < 0 && (mode_switch || ffmpegdec->context->skip_frame))
1532     len = 0;
1533
1534   /* no data, we're done */
1535   if (len < 0 || *have_data == 0)
1536     goto beach;
1537
1538   /* get the output picture timing info again */
1539   out_dframe = ffmpegdec->picture->opaque;
1540   out_frame = gst_video_codec_frame_ref (out_dframe->frame);
1541
1542   /* also give back a buffer allocated by the frame, if any */
1543   gst_buffer_replace (&out_frame->output_buffer, out_dframe->buffer);
1544   gst_buffer_replace (&out_dframe->buffer, NULL);
1545
1546   /* Extract auxilliary info not stored in the main AVframe */
1547   {
1548     GstVideoInfo *in_info = &ffmpegdec->input_state->info;
1549     /* Take multiview mode from upstream if present */
1550     ffmpegdec->picture_multiview_mode = GST_VIDEO_INFO_MULTIVIEW_MODE (in_info);
1551     ffmpegdec->picture_multiview_flags =
1552         GST_VIDEO_INFO_MULTIVIEW_FLAGS (in_info);
1553
1554     /* Otherwise, see if there's info in the frame */
1555     if (ffmpegdec->picture_multiview_mode == GST_VIDEO_MULTIVIEW_MODE_NONE) {
1556       AVFrameSideData *side_data =
1557           av_frame_get_side_data (ffmpegdec->picture, AV_FRAME_DATA_STEREO3D);
1558       if (side_data) {
1559         AVStereo3D *stereo = (AVStereo3D *) side_data->data;
1560         ffmpegdec->picture_multiview_mode = stereo_av_to_gst (stereo->type);
1561         if (stereo->flags & AV_STEREO3D_FLAG_INVERT) {
1562           ffmpegdec->picture_multiview_flags =
1563               GST_VIDEO_MULTIVIEW_FLAGS_RIGHT_VIEW_FIRST;
1564         } else {
1565           ffmpegdec->picture_multiview_flags = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
1566         }
1567       }
1568     }
1569   }
1570
1571   GST_DEBUG_OBJECT (ffmpegdec,
1572       "pts %" G_GUINT64_FORMAT " duration %" G_GUINT64_FORMAT,
1573       out_frame->pts, out_frame->duration);
1574   GST_DEBUG_OBJECT (ffmpegdec, "picture: pts %" G_GUINT64_FORMAT,
1575       (guint64) ffmpegdec->picture->pts);
1576   GST_DEBUG_OBJECT (ffmpegdec, "picture: num %d",
1577       ffmpegdec->picture->coded_picture_number);
1578   GST_DEBUG_OBJECT (ffmpegdec, "picture: display %d",
1579       ffmpegdec->picture->display_picture_number);
1580   GST_DEBUG_OBJECT (ffmpegdec, "picture: opaque %p",
1581       ffmpegdec->picture->opaque);
1582   GST_DEBUG_OBJECT (ffmpegdec, "picture: reordered opaque %" G_GUINT64_FORMAT,
1583       (guint64) ffmpegdec->picture->reordered_opaque);
1584   GST_DEBUG_OBJECT (ffmpegdec, "repeat_pict:%d",
1585       ffmpegdec->picture->repeat_pict);
1586   GST_DEBUG_OBJECT (ffmpegdec, "corrupted frame: %d",
1587       ! !(ffmpegdec->picture->flags & AV_FRAME_FLAG_CORRUPT));
1588
1589   if (!gst_ffmpegviddec_negotiate (ffmpegdec, ffmpegdec->context,
1590           ffmpegdec->picture))
1591     goto negotiation_error;
1592
1593   pool = gst_video_decoder_get_buffer_pool (GST_VIDEO_DECODER (ffmpegdec));
1594   if (G_UNLIKELY (out_frame->output_buffer == NULL)) {
1595     *ret = get_output_buffer (ffmpegdec, out_frame);
1596   } else if (G_UNLIKELY (out_frame->output_buffer->pool != pool)) {
1597     GstBuffer *tmp = out_frame->output_buffer;
1598     out_frame->output_buffer = NULL;
1599     *ret = get_output_buffer (ffmpegdec, out_frame);
1600     gst_buffer_unref (tmp);
1601   }
1602 #ifndef G_DISABLE_ASSERT
1603   else {
1604     GstVideoMeta *vmeta = gst_buffer_get_video_meta (out_frame->output_buffer);
1605     if (vmeta) {
1606       GstVideoInfo *info = &ffmpegdec->output_state->info;
1607       g_assert (vmeta->width == GST_VIDEO_INFO_WIDTH (info));
1608       g_assert (vmeta->height == GST_VIDEO_INFO_HEIGHT (info));
1609     }
1610   }
1611 #endif
1612   gst_object_unref (pool);
1613
1614   if (G_UNLIKELY (*ret != GST_FLOW_OK))
1615     goto no_output;
1616
1617   /* Mark corrupted frames as corrupted */
1618   if (ffmpegdec->picture->flags & AV_FRAME_FLAG_CORRUPT)
1619     GST_BUFFER_FLAG_SET (out_frame->output_buffer, GST_BUFFER_FLAG_CORRUPTED);
1620
1621   if (ffmpegdec->pic_interlaced) {
1622     /* set interlaced flags */
1623     if (ffmpegdec->picture->repeat_pict)
1624       GST_BUFFER_FLAG_SET (out_frame->output_buffer, GST_VIDEO_BUFFER_FLAG_RFF);
1625     if (ffmpegdec->picture->top_field_first)
1626       GST_BUFFER_FLAG_SET (out_frame->output_buffer, GST_VIDEO_BUFFER_FLAG_TFF);
1627     if (ffmpegdec->picture->interlaced_frame)
1628       GST_BUFFER_FLAG_SET (out_frame->output_buffer,
1629           GST_VIDEO_BUFFER_FLAG_INTERLACED);
1630   }
1631
1632   {
1633     AVFrameSideData *side_data =
1634         av_frame_get_side_data (ffmpegdec->picture, AV_FRAME_DATA_A53_CC);
1635     if (side_data) {
1636       GST_LOG_OBJECT (ffmpegdec, "Found CC of size %d", side_data->size);
1637       GST_MEMDUMP ("A53 CC", side_data->data, side_data->size);
1638       out_frame->output_buffer =
1639           gst_buffer_make_writable (out_frame->output_buffer);
1640       gst_buffer_add_video_caption_meta (out_frame->output_buffer,
1641           GST_VIDEO_CAPTION_TYPE_CEA708_RAW, side_data->data, side_data->size);
1642     }
1643   }
1644
1645   /* cleaning time */
1646   /* so we decoded this frame, frames preceding it in decoding order
1647    * that still do not have a buffer allocated seem rather useless,
1648    * and can be discarded, due to e.g. misparsed bogus frame
1649    * or non-keyframe in skipped decoding, ...
1650    * In any case, not likely to be seen again, so discard those,
1651    * before they pile up and/or mess with timestamping */
1652   {
1653     GList *l, *ol;
1654     GstVideoDecoder *dec = GST_VIDEO_DECODER (ffmpegdec);
1655     gboolean old = TRUE;
1656
1657     ol = l = gst_video_decoder_get_frames (dec);
1658     while (l) {
1659       GstVideoCodecFrame *tmp = l->data;
1660
1661       if (tmp == frame)
1662         old = FALSE;
1663
1664       if (old && GST_VIDEO_CODEC_FRAME_IS_DECODE_ONLY (tmp)) {
1665         GST_LOG_OBJECT (dec,
1666             "discarding ghost frame %p (#%d) PTS:%" GST_TIME_FORMAT " DTS:%"
1667             GST_TIME_FORMAT, tmp, tmp->system_frame_number,
1668             GST_TIME_ARGS (tmp->pts), GST_TIME_ARGS (tmp->dts));
1669         /* drop extra ref and remove from frame list */
1670         gst_video_decoder_release_frame (dec, tmp);
1671       } else {
1672         /* drop extra ref we got */
1673         gst_video_codec_frame_unref (tmp);
1674       }
1675       l = l->next;
1676     }
1677     g_list_free (ol);
1678   }
1679
1680   av_frame_unref (ffmpegdec->picture);
1681
1682   /* FIXME: Ideally we would remap the buffer read-only now before pushing but
1683    * libav might still have a reference to it!
1684    */
1685   *ret =
1686       gst_video_decoder_finish_frame (GST_VIDEO_DECODER (ffmpegdec), out_frame);
1687
1688 beach:
1689   GST_DEBUG_OBJECT (ffmpegdec, "return flow %s, len %d",
1690       gst_flow_get_name (*ret), len);
1691   return len;
1692
1693   /* special cases */
1694 no_output:
1695   {
1696     GST_DEBUG_OBJECT (ffmpegdec, "no output buffer");
1697     gst_video_decoder_drop_frame (GST_VIDEO_DECODER (ffmpegdec), out_frame);
1698     len = -1;
1699     goto beach;
1700   }
1701
1702 negotiation_error:
1703   {
1704     if (GST_PAD_IS_FLUSHING (GST_VIDEO_DECODER_SRC_PAD (ffmpegdec))) {
1705       *ret = GST_FLOW_FLUSHING;
1706       goto beach;
1707     }
1708     GST_WARNING_OBJECT (ffmpegdec, "Error negotiating format");
1709     *ret = GST_FLOW_NOT_NEGOTIATED;
1710     goto beach;
1711   }
1712 }
1713
1714
1715 /* gst_ffmpegviddec_frame:
1716  * ffmpegdec:
1717  * data: pointer to the data to decode
1718  * size: size of data in bytes
1719  * got_data: 0 if no data was decoded, != 0 otherwise.
1720  * in_time: timestamp of data
1721  * in_duration: duration of data
1722  * ret: GstFlowReturn to return in the chain function
1723  *
1724  * Decode the given frame and pushes it downstream.
1725  *
1726  * Returns: Number of bytes used in decoding, -1 on error/failure.
1727  */
1728
1729 static gint
1730 gst_ffmpegviddec_frame (GstFFMpegVidDec * ffmpegdec,
1731     guint8 * data, guint size, gint * have_data, GstVideoCodecFrame * frame,
1732     GstFlowReturn * ret)
1733 {
1734   GstFFMpegVidDecClass *oclass;
1735   gint len = 0;
1736
1737   if (G_UNLIKELY (ffmpegdec->context->codec == NULL))
1738     goto no_codec;
1739
1740   GST_LOG_OBJECT (ffmpegdec, "data:%p, size:%d", data, size);
1741
1742   *ret = GST_FLOW_OK;
1743   ffmpegdec->context->frame_number++;
1744
1745   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
1746
1747   len =
1748       gst_ffmpegviddec_video_frame (ffmpegdec, data, size, have_data, frame,
1749       ret);
1750
1751   if (len < 0) {
1752     GST_WARNING_OBJECT (ffmpegdec,
1753         "avdec_%s: decoding error (len: %d, have_data: %d)",
1754         oclass->in_plugin->name, len, *have_data);
1755   }
1756
1757   return len;
1758
1759   /* ERRORS */
1760 no_codec:
1761   {
1762     GST_ERROR_OBJECT (ffmpegdec, "no codec context");
1763     *ret = GST_FLOW_NOT_NEGOTIATED;
1764     return -1;
1765   }
1766 }
1767
1768 static GstFlowReturn
1769 gst_ffmpegviddec_drain (GstVideoDecoder * decoder)
1770 {
1771   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
1772   GstFFMpegVidDecClass *oclass;
1773
1774   if (!ffmpegdec->opened)
1775     return GST_FLOW_OK;
1776
1777   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
1778
1779   if (oclass->in_plugin->capabilities & CODEC_CAP_DELAY) {
1780     gint have_data, len;
1781     GstFlowReturn ret;
1782
1783     GST_LOG_OBJECT (ffmpegdec,
1784         "codec has delay capabilities, calling until ffmpeg has drained everything");
1785
1786     do {
1787       len = gst_ffmpegviddec_frame (ffmpegdec, NULL, 0, &have_data, NULL, &ret);
1788     } while (len >= 0 && have_data == 1 && ret == GST_FLOW_OK);
1789   }
1790
1791   return GST_FLOW_OK;
1792 }
1793
1794 static GstFlowReturn
1795 gst_ffmpegviddec_handle_frame (GstVideoDecoder * decoder,
1796     GstVideoCodecFrame * frame)
1797 {
1798   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
1799   guint8 *data, *bdata;
1800   gint size, len, have_data, bsize;
1801   GstMapInfo minfo;
1802   GstFlowReturn ret = GST_FLOW_OK;
1803   gboolean do_padding;
1804
1805   GST_LOG_OBJECT (ffmpegdec,
1806       "Received new data of size %" G_GSIZE_FORMAT ", dts %" GST_TIME_FORMAT
1807       ", pts:%" GST_TIME_FORMAT ", dur:%" GST_TIME_FORMAT,
1808       gst_buffer_get_size (frame->input_buffer), GST_TIME_ARGS (frame->dts),
1809       GST_TIME_ARGS (frame->pts), GST_TIME_ARGS (frame->duration));
1810
1811   if (!gst_buffer_map (frame->input_buffer, &minfo, GST_MAP_READ)) {
1812     GST_ELEMENT_ERROR (ffmpegdec, STREAM, DECODE, ("Decoding problem"),
1813         ("Failed to map buffer for reading"));
1814     return GST_FLOW_ERROR;
1815   }
1816
1817   /* treat frame as void until a buffer is requested for it */
1818   GST_VIDEO_CODEC_FRAME_FLAG_SET (frame,
1819       GST_VIDEO_CODEC_FRAME_FLAG_DECODE_ONLY);
1820
1821   bdata = minfo.data;
1822   bsize = minfo.size;
1823
1824   if (bsize > 0 && (!GST_MEMORY_IS_ZERO_PADDED (minfo.memory)
1825           || (minfo.maxsize - minfo.size) < AV_INPUT_BUFFER_PADDING_SIZE)) {
1826     /* add padding */
1827     if (ffmpegdec->padded_size < bsize + AV_INPUT_BUFFER_PADDING_SIZE) {
1828       ffmpegdec->padded_size = bsize + AV_INPUT_BUFFER_PADDING_SIZE;
1829       ffmpegdec->padded = g_realloc (ffmpegdec->padded, ffmpegdec->padded_size);
1830       GST_LOG_OBJECT (ffmpegdec, "resized padding buffer to %d",
1831           ffmpegdec->padded_size);
1832     }
1833     GST_CAT_TRACE_OBJECT (CAT_PERFORMANCE, ffmpegdec,
1834         "Copy input to add padding");
1835     memcpy (ffmpegdec->padded, bdata, bsize);
1836     memset (ffmpegdec->padded + bsize, 0, AV_INPUT_BUFFER_PADDING_SIZE);
1837
1838     bdata = ffmpegdec->padded;
1839     do_padding = TRUE;
1840   } else {
1841     do_padding = FALSE;
1842   }
1843
1844   do {
1845     guint8 tmp_padding[AV_INPUT_BUFFER_PADDING_SIZE];
1846
1847     /* parse, if at all possible */
1848     data = bdata;
1849     size = bsize;
1850
1851     if (do_padding) {
1852       /* add temporary padding */
1853       GST_CAT_TRACE_OBJECT (CAT_PERFORMANCE, ffmpegdec,
1854           "Add temporary input padding");
1855       memcpy (tmp_padding, data + size, AV_INPUT_BUFFER_PADDING_SIZE);
1856       memset (data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
1857     }
1858
1859     /* decode a frame of audio/video now */
1860     len =
1861         gst_ffmpegviddec_frame (ffmpegdec, data, size, &have_data, frame, &ret);
1862
1863     if (ret != GST_FLOW_OK) {
1864       GST_LOG_OBJECT (ffmpegdec, "breaking because of flow ret %s",
1865           gst_flow_get_name (ret));
1866       /* bad flow return, make sure we discard all data and exit */
1867       bsize = 0;
1868       break;
1869     }
1870
1871     if (do_padding) {
1872       memcpy (data + size, tmp_padding, AV_INPUT_BUFFER_PADDING_SIZE);
1873     }
1874
1875     if (len == 0 && have_data == 0) {
1876       /* nothing was decoded, this could be because no data was available or
1877        * because we were skipping frames.
1878        * If we have no context we must exit and wait for more data, we keep the
1879        * data we tried. */
1880       GST_LOG_OBJECT (ffmpegdec, "Decoding didn't return any data, breaking");
1881       break;
1882     }
1883
1884     if (len < 0) {
1885       /* a decoding error happened, we must break and try again with next data. */
1886       GST_LOG_OBJECT (ffmpegdec, "Decoding error, breaking");
1887       bsize = 0;
1888       break;
1889     }
1890
1891     /* prepare for the next round, for codecs with a context we did this
1892      * already when using the parser. */
1893     bsize -= len;
1894     bdata += len;
1895
1896     do_padding = TRUE;
1897
1898     GST_LOG_OBJECT (ffmpegdec, "Before (while bsize>0).  bsize:%d , bdata:%p",
1899         bsize, bdata);
1900   } while (bsize > 0);
1901
1902   if (bsize > 0)
1903     GST_DEBUG_OBJECT (ffmpegdec, "Dropping %d bytes of data", bsize);
1904
1905   gst_buffer_unmap (frame->input_buffer, &minfo);
1906   gst_video_codec_frame_unref (frame);
1907
1908   return ret;
1909 }
1910
1911 static gboolean
1912 gst_ffmpegviddec_start (GstVideoDecoder * decoder)
1913 {
1914   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
1915   GstFFMpegVidDecClass *oclass;
1916
1917   oclass = (GstFFMpegVidDecClass *) (G_OBJECT_GET_CLASS (ffmpegdec));
1918
1919   GST_OBJECT_LOCK (ffmpegdec);
1920   gst_ffmpeg_avcodec_close (ffmpegdec->context);
1921   if (avcodec_get_context_defaults3 (ffmpegdec->context, oclass->in_plugin) < 0) {
1922     GST_DEBUG_OBJECT (ffmpegdec, "Failed to set context defaults");
1923     GST_OBJECT_UNLOCK (ffmpegdec);
1924     return FALSE;
1925   }
1926   ffmpegdec->context->opaque = ffmpegdec;
1927   GST_OBJECT_UNLOCK (ffmpegdec);
1928
1929   return TRUE;
1930 }
1931
1932 static gboolean
1933 gst_ffmpegviddec_stop (GstVideoDecoder * decoder)
1934 {
1935   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
1936
1937   GST_OBJECT_LOCK (ffmpegdec);
1938   gst_ffmpegviddec_close (ffmpegdec, FALSE);
1939   GST_OBJECT_UNLOCK (ffmpegdec);
1940   g_free (ffmpegdec->padded);
1941   ffmpegdec->padded = NULL;
1942   ffmpegdec->padded_size = 0;
1943   if (ffmpegdec->input_state)
1944     gst_video_codec_state_unref (ffmpegdec->input_state);
1945   ffmpegdec->input_state = NULL;
1946   if (ffmpegdec->output_state)
1947     gst_video_codec_state_unref (ffmpegdec->output_state);
1948   ffmpegdec->output_state = NULL;
1949
1950   if (ffmpegdec->internal_pool)
1951     gst_object_unref (ffmpegdec->internal_pool);
1952   ffmpegdec->internal_pool = NULL;
1953
1954   ffmpegdec->pic_pix_fmt = 0;
1955   ffmpegdec->pic_width = 0;
1956   ffmpegdec->pic_height = 0;
1957   ffmpegdec->pic_par_n = 0;
1958   ffmpegdec->pic_par_d = 0;
1959   ffmpegdec->pic_interlaced = 0;
1960   ffmpegdec->pic_field_order = 0;
1961   ffmpegdec->pic_field_order_changed = FALSE;
1962   ffmpegdec->ctx_ticks = 0;
1963   ffmpegdec->ctx_time_n = 0;
1964   ffmpegdec->ctx_time_d = 0;
1965
1966   ffmpegdec->pool_width = 0;
1967   ffmpegdec->pool_height = 0;
1968   ffmpegdec->pool_format = 0;
1969
1970   return TRUE;
1971 }
1972
1973 static GstFlowReturn
1974 gst_ffmpegviddec_finish (GstVideoDecoder * decoder)
1975 {
1976   gst_ffmpegviddec_drain (decoder);
1977   /* note that finish can and should clean up more drastically,
1978    * but drain is also invoked on e.g. packet loss in GAP handling */
1979   gst_ffmpegviddec_flush (decoder);
1980
1981   return GST_FLOW_OK;
1982 }
1983
1984 static gboolean
1985 gst_ffmpegviddec_flush (GstVideoDecoder * decoder)
1986 {
1987   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
1988
1989   if (ffmpegdec->opened) {
1990     GST_LOG_OBJECT (decoder, "flushing buffers");
1991     avcodec_flush_buffers (ffmpegdec->context);
1992   }
1993
1994   return TRUE;
1995 }
1996
1997 static gboolean
1998 gst_ffmpegviddec_decide_allocation (GstVideoDecoder * decoder, GstQuery * query)
1999 {
2000   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) decoder;
2001   GstVideoCodecState *state;
2002   GstBufferPool *pool;
2003   guint size, min, max;
2004   GstStructure *config;
2005   gboolean have_pool, have_videometa, have_alignment, update_pool = FALSE;
2006   GstAllocator *allocator = NULL;
2007   GstAllocationParams params = DEFAULT_ALLOC_PARAM;
2008
2009   have_pool = (gst_query_get_n_allocation_pools (query) != 0);
2010
2011   if (!GST_VIDEO_DECODER_CLASS (parent_class)->decide_allocation (decoder,
2012           query))
2013     return FALSE;
2014
2015   state = gst_video_decoder_get_output_state (decoder);
2016
2017   if (gst_query_get_n_allocation_params (query) > 0) {
2018     gst_query_parse_nth_allocation_param (query, 0, &allocator, &params);
2019     params.align = MAX (params.align, DEFAULT_STRIDE_ALIGN);
2020   } else {
2021     gst_query_add_allocation_param (query, allocator, &params);
2022   }
2023
2024   gst_query_parse_nth_allocation_pool (query, 0, &pool, &size, &min, &max);
2025
2026   /* Don't use pool that can't grow, as we don't know how many buffer we'll
2027    * need, otherwise we may stall */
2028   if (max != 0 && max < REQUIRED_POOL_MAX_BUFFERS) {
2029     gst_object_unref (pool);
2030     pool = gst_video_buffer_pool_new ();
2031     max = 0;
2032     update_pool = TRUE;
2033     have_pool = FALSE;
2034
2035     /* if there is an allocator, also drop it, as it might be the reason we
2036      * have this limit. Default will be used */
2037     if (allocator) {
2038       gst_object_unref (allocator);
2039       allocator = NULL;
2040     }
2041   }
2042
2043   config = gst_buffer_pool_get_config (pool);
2044   gst_buffer_pool_config_set_params (config, state->caps, size, min, max);
2045   gst_buffer_pool_config_set_allocator (config, allocator, &params);
2046
2047   have_videometa =
2048       gst_query_find_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
2049
2050   if (have_videometa)
2051     gst_buffer_pool_config_add_option (config,
2052         GST_BUFFER_POOL_OPTION_VIDEO_META);
2053
2054   have_alignment =
2055       gst_buffer_pool_has_option (pool, GST_BUFFER_POOL_OPTION_VIDEO_ALIGNMENT);
2056
2057   /* If we have videometa, we never have to copy */
2058   if (have_videometa && have_pool && have_alignment &&
2059       gst_ffmpegviddec_can_direct_render (ffmpegdec)) {
2060     GstStructure *config_copy = gst_structure_copy (config);
2061
2062     gst_ffmpegvideodec_prepare_dr_pool (ffmpegdec, pool, &state->info,
2063         config_copy);
2064
2065     /* FIXME validate and retry */
2066     if (gst_buffer_pool_set_config (pool, config_copy)) {
2067       GstFlowReturn ret;
2068       GstBuffer *tmp;
2069
2070       gst_buffer_pool_set_active (pool, TRUE);
2071       ret = gst_buffer_pool_acquire_buffer (pool, &tmp, NULL);
2072       if (ret == GST_FLOW_OK) {
2073         GstVideoMeta *vmeta = gst_buffer_get_video_meta (tmp);
2074         gboolean same_stride = TRUE;
2075         gint i;
2076
2077         for (i = 0; i < vmeta->n_planes; i++) {
2078           if (vmeta->stride[i] != ffmpegdec->stride[i]) {
2079             same_stride = FALSE;
2080             break;
2081           }
2082         }
2083
2084         gst_buffer_unref (tmp);
2085
2086         if (same_stride) {
2087           if (ffmpegdec->internal_pool)
2088             gst_object_unref (ffmpegdec->internal_pool);
2089           ffmpegdec->internal_pool = gst_object_ref (pool);
2090           ffmpegdec->pool_info = state->info;
2091           gst_structure_free (config);
2092           goto done;
2093         }
2094       }
2095     }
2096   }
2097
2098   if (have_videometa && ffmpegdec->internal_pool
2099       && ffmpegdec->pool_width == state->info.width
2100       && ffmpegdec->pool_height == state->info.height) {
2101     update_pool = TRUE;
2102     gst_object_unref (pool);
2103     pool = gst_object_ref (ffmpegdec->internal_pool);
2104     gst_structure_free (config);
2105     goto done;
2106   }
2107
2108   /* configure */
2109   if (!gst_buffer_pool_set_config (pool, config)) {
2110     gboolean working_pool = FALSE;
2111     config = gst_buffer_pool_get_config (pool);
2112
2113     if (gst_buffer_pool_config_validate_params (config, state->caps, size, min,
2114             max)) {
2115       working_pool = gst_buffer_pool_set_config (pool, config);
2116     } else {
2117       gst_structure_free (config);
2118     }
2119
2120     if (!working_pool) {
2121       gst_object_unref (pool);
2122       pool = gst_video_buffer_pool_new ();
2123       config = gst_buffer_pool_get_config (pool);
2124       gst_buffer_pool_config_set_params (config, state->caps, size, min, max);
2125       gst_buffer_pool_config_set_allocator (config, NULL, &params);
2126       gst_buffer_pool_set_config (pool, config);
2127       update_pool = TRUE;
2128     }
2129   }
2130
2131 done:
2132   /* and store */
2133   if (update_pool)
2134     gst_query_set_nth_allocation_pool (query, 0, pool, size, min, max);
2135
2136   gst_object_unref (pool);
2137   if (allocator)
2138     gst_object_unref (allocator);
2139   gst_video_codec_state_unref (state);
2140
2141   return TRUE;
2142 }
2143
2144 static gboolean
2145 gst_ffmpegviddec_propose_allocation (GstVideoDecoder * decoder,
2146     GstQuery * query)
2147 {
2148   GstAllocationParams params;
2149
2150   gst_allocation_params_init (&params);
2151   params.flags = GST_MEMORY_FLAG_ZERO_PADDED;
2152   params.align = DEFAULT_STRIDE_ALIGN;
2153   params.padding = AV_INPUT_BUFFER_PADDING_SIZE;
2154   /* we would like to have some padding so that we don't have to
2155    * memcpy. We don't suggest an allocator. */
2156   gst_query_add_allocation_param (query, NULL, &params);
2157
2158   return GST_VIDEO_DECODER_CLASS (parent_class)->propose_allocation (decoder,
2159       query);
2160 }
2161
2162 static void
2163 gst_ffmpegviddec_set_property (GObject * object,
2164     guint prop_id, const GValue * value, GParamSpec * pspec)
2165 {
2166   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) object;
2167
2168   switch (prop_id) {
2169     case PROP_LOWRES:
2170       ffmpegdec->lowres = ffmpegdec->context->lowres = g_value_get_enum (value);
2171       break;
2172     case PROP_SKIPFRAME:
2173       ffmpegdec->skip_frame = ffmpegdec->context->skip_frame =
2174           g_value_get_enum (value);
2175       break;
2176     case PROP_DIRECT_RENDERING:
2177       ffmpegdec->direct_rendering = g_value_get_boolean (value);
2178       break;
2179     case PROP_DEBUG_MV:
2180       ffmpegdec->debug_mv = ffmpegdec->context->debug_mv =
2181           g_value_get_boolean (value);
2182       break;
2183     case PROP_MAX_THREADS:
2184       ffmpegdec->max_threads = g_value_get_int (value);
2185       break;
2186     case PROP_OUTPUT_CORRUPT:
2187       ffmpegdec->output_corrupt = g_value_get_boolean (value);
2188       break;
2189     default:
2190       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2191       break;
2192   }
2193 }
2194
2195 static void
2196 gst_ffmpegviddec_get_property (GObject * object,
2197     guint prop_id, GValue * value, GParamSpec * pspec)
2198 {
2199   GstFFMpegVidDec *ffmpegdec = (GstFFMpegVidDec *) object;
2200
2201   switch (prop_id) {
2202     case PROP_LOWRES:
2203       g_value_set_enum (value, ffmpegdec->context->lowres);
2204       break;
2205     case PROP_SKIPFRAME:
2206       g_value_set_enum (value, ffmpegdec->context->skip_frame);
2207       break;
2208     case PROP_DIRECT_RENDERING:
2209       g_value_set_boolean (value, ffmpegdec->direct_rendering);
2210       break;
2211     case PROP_DEBUG_MV:
2212       g_value_set_boolean (value, ffmpegdec->context->debug_mv);
2213       break;
2214     case PROP_MAX_THREADS:
2215       g_value_set_int (value, ffmpegdec->max_threads);
2216       break;
2217     case PROP_OUTPUT_CORRUPT:
2218       g_value_set_boolean (value, ffmpegdec->output_corrupt);
2219       break;
2220     default:
2221       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
2222       break;
2223   }
2224 }
2225
2226 gboolean
2227 gst_ffmpegviddec_register (GstPlugin * plugin)
2228 {
2229   GTypeInfo typeinfo = {
2230     sizeof (GstFFMpegVidDecClass),
2231     (GBaseInitFunc) gst_ffmpegviddec_base_init,
2232     NULL,
2233     (GClassInitFunc) gst_ffmpegviddec_class_init,
2234     NULL,
2235     NULL,
2236     sizeof (GstFFMpegVidDec),
2237     0,
2238     (GInstanceInitFunc) gst_ffmpegviddec_init,
2239   };
2240   GType type;
2241   AVCodec *in_plugin;
2242   gint rank;
2243
2244   in_plugin = av_codec_next (NULL);
2245
2246   GST_LOG ("Registering decoders");
2247
2248   while (in_plugin) {
2249     gchar *type_name;
2250     gchar *plugin_name;
2251
2252     /* only video decoders */
2253     if (!av_codec_is_decoder (in_plugin)
2254         || in_plugin->type != AVMEDIA_TYPE_VIDEO)
2255       goto next;
2256
2257     /* no quasi codecs, please */
2258     if (in_plugin->id == AV_CODEC_ID_RAWVIDEO ||
2259         in_plugin->id == AV_CODEC_ID_V210 ||
2260         in_plugin->id == AV_CODEC_ID_V210X ||
2261         in_plugin->id == AV_CODEC_ID_V308 ||
2262         in_plugin->id == AV_CODEC_ID_V408 ||
2263         in_plugin->id == AV_CODEC_ID_V410 ||
2264         in_plugin->id == AV_CODEC_ID_R210
2265         || in_plugin->id == AV_CODEC_ID_AYUV
2266         || in_plugin->id == AV_CODEC_ID_Y41P
2267         || in_plugin->id == AV_CODEC_ID_012V
2268         || in_plugin->id == AV_CODEC_ID_YUV4
2269 #if AV_VERSION_INT (LIBAVCODEC_VERSION_MAJOR, LIBAVCODEC_VERSION_MINOR, LIBAVCODEC_VERSION_MICRO) >= \
2270         AV_VERSION_INT (57,4,0)
2271         || in_plugin->id == AV_CODEC_ID_WRAPPED_AVFRAME
2272 #endif
2273         || in_plugin->id == AV_CODEC_ID_ZLIB) {
2274       goto next;
2275     }
2276
2277     /* No decoders depending on external libraries (we don't build them, but
2278      * people who build against an external ffmpeg might have them.
2279      * We have native gstreamer plugins for all of those libraries anyway. */
2280     if (!strncmp (in_plugin->name, "lib", 3)) {
2281       GST_DEBUG
2282           ("Not using external library decoder %s. Use the gstreamer-native ones instead.",
2283           in_plugin->name);
2284       goto next;
2285     }
2286
2287     /* No vdpau plugins until we can figure out how to properly use them
2288      * outside of ffmpeg. */
2289     if (g_str_has_suffix (in_plugin->name, "_vdpau")) {
2290       GST_DEBUG
2291           ("Ignoring VDPAU decoder %s. We can't handle this outside of ffmpeg",
2292           in_plugin->name);
2293       goto next;
2294     }
2295
2296     if (g_str_has_suffix (in_plugin->name, "_xvmc")) {
2297       GST_DEBUG
2298           ("Ignoring XVMC decoder %s. We can't handle this outside of ffmpeg",
2299           in_plugin->name);
2300       goto next;
2301     }
2302
2303     if (strstr (in_plugin->name, "vaapi")) {
2304       GST_DEBUG
2305           ("Ignoring VAAPI decoder %s. We can't handle this outside of ffmpeg",
2306           in_plugin->name);
2307       goto next;
2308     }
2309
2310     if (g_str_has_suffix (in_plugin->name, "_qsv")) {
2311       GST_DEBUG
2312           ("Ignoring qsv decoder %s. We can't handle this outside of ffmpeg",
2313           in_plugin->name);
2314       goto next;
2315     }
2316
2317     if (g_str_has_suffix (in_plugin->name, "_cuvid")) {
2318       GST_DEBUG
2319           ("Ignoring CUVID decoder %s. We can't handle this outside of ffmpeg",
2320           in_plugin->name);
2321       goto next;
2322     }
2323
2324     if (g_str_has_suffix (in_plugin->name, "_v4l2m2m")) {
2325       GST_DEBUG
2326           ("Ignoring V4L2 mem-to-mem decoder %s. We can't handle this outside of ffmpeg",
2327           in_plugin->name);
2328       goto next;
2329     }
2330
2331     GST_DEBUG ("Trying plugin %s [%s]", in_plugin->name, in_plugin->long_name);
2332
2333     /* no codecs for which we're GUARANTEED to have better alternatives */
2334     /* MPEG1VIDEO : the mpeg2video decoder is preferred */
2335     /* MP1 : Use MP3 for decoding */
2336     /* MP2 : Use MP3 for decoding */
2337     /* Theora: Use libtheora based theoradec */
2338     if (!strcmp (in_plugin->name, "theora") ||
2339         !strcmp (in_plugin->name, "mpeg1video") ||
2340         strstr (in_plugin->name, "crystalhd") != NULL ||
2341         !strcmp (in_plugin->name, "ass") ||
2342         !strcmp (in_plugin->name, "srt") ||
2343         !strcmp (in_plugin->name, "pgssub") ||
2344         !strcmp (in_plugin->name, "dvdsub") ||
2345         !strcmp (in_plugin->name, "dvbsub")) {
2346       GST_LOG ("Ignoring decoder %s", in_plugin->name);
2347       goto next;
2348     }
2349
2350     /* construct the type */
2351     if (!strcmp (in_plugin->name, "hevc")) {
2352       plugin_name = g_strdup ("h265");
2353     } else {
2354       plugin_name = g_strdup ((gchar *) in_plugin->name);
2355     }
2356     g_strdelimit (plugin_name, NULL, '_');
2357     type_name = g_strdup_printf ("avdec_%s", plugin_name);
2358     g_free (plugin_name);
2359
2360     type = g_type_from_name (type_name);
2361
2362     if (!type) {
2363       /* create the gtype now */
2364       type =
2365           g_type_register_static (GST_TYPE_VIDEO_DECODER, type_name, &typeinfo,
2366           0);
2367       g_type_set_qdata (type, GST_FFDEC_PARAMS_QDATA, (gpointer) in_plugin);
2368     }
2369
2370     /* (Ronald) MPEG-4 gets a higher priority because it has been well-
2371      * tested and by far outperforms divxdec/xviddec - so we prefer it.
2372      * msmpeg4v3 same, as it outperforms divxdec for divx3 playback.
2373      * VC1/WMV3 are not working and thus unpreferred for now. */
2374     switch (in_plugin->id) {
2375       case AV_CODEC_ID_MPEG1VIDEO:
2376       case AV_CODEC_ID_MPEG2VIDEO:
2377       case AV_CODEC_ID_MPEG4:
2378       case AV_CODEC_ID_MSMPEG4V3:
2379       case AV_CODEC_ID_H264:
2380       case AV_CODEC_ID_HEVC:
2381       case AV_CODEC_ID_RV10:
2382       case AV_CODEC_ID_RV20:
2383       case AV_CODEC_ID_RV30:
2384       case AV_CODEC_ID_RV40:
2385         rank = GST_RANK_PRIMARY;
2386         break;
2387         /* DVVIDEO: we have a good dv decoder, fast on both ppc as well as x86.
2388          * They say libdv's quality is better though. leave as secondary.
2389          * note: if you change this, see the code in gstdv.c in good/ext/dv.
2390          */
2391       case AV_CODEC_ID_DVVIDEO:
2392         rank = GST_RANK_SECONDARY;
2393         break;
2394       default:
2395         rank = GST_RANK_MARGINAL;
2396         break;
2397     }
2398     if (!gst_element_register (plugin, type_name, rank, type)) {
2399       g_warning ("Failed to register %s", type_name);
2400       g_free (type_name);
2401       return FALSE;
2402     }
2403
2404     g_free (type_name);
2405
2406   next:
2407     in_plugin = av_codec_next (in_plugin);
2408   }
2409
2410   GST_LOG ("Finished Registering decoders");
2411
2412   return TRUE;
2413 }