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