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, ¶ms);
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, ¶ms);
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, ¶ms);
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, ¶ms);
2281 params.align = MAX (params.align, DEFAULT_STRIDE_ALIGN);
2282 } else {
2283 gst_query_add_allocation_param (query, allocator, ¶ms);
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, ¶ms);
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, ¶ms);
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 (¶ms);
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, ¶ms);
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