vp8enc/vp9enc: Protect the encoder with a mutex in all situations
[platform/upstream/gst-plugins-good.git] / ext / vpx / gstvp9enc.c
1 /* VP9
2  * Copyright (C) 2006 David Schleef <ds@schleef.org>
3  * Copyright (C) 2010 Entropy Wave Inc
4  * Copyright (C) 2010-2013 Sebastian Dröge <slomo@circular-chaos.org>
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Library General Public
8  * License as published by the Free Software Foundation; either
9  * version 2 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Library General Public License for more details.
15  *
16  * You should have received a copy of the GNU Library General Public
17  * License along with this library; if not, write to the
18  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  *
21  */
22 /**
23  * SECTION:element-vp9enc
24  * @see_also: vp9dec, webmmux, oggmux
25  *
26  * This element encodes raw video into a VP9 stream.
27  * <ulink url="http://www.webmproject.org">VP9</ulink> is a royalty-free
28  * video codec maintained by <ulink url="http://www.google.com/">Google
29  * </ulink>. It's the successor of On2 VP3, which was the base of the
30  * Theora video codec.
31  *
32  * To control the quality of the encoding, the #GstVP9Enc::target-bitrate,
33  * #GstVP9Enc::min-quantizer, #GstVP9Enc::max-quantizer or #GstVP9Enc::cq-level
34  * properties can be used. Which one is used depends on the mode selected by
35  * the #GstVP9Enc::end-usage property.
36  * See <ulink url="http://www.webmproject.org/docs/encoder-parameters/">Encoder Parameters</ulink>
37  * for explanation, examples for useful encoding parameters and more details
38  * on the encoding parameters.
39  *
40  * <refsect2>
41  * <title>Example pipeline</title>
42  * |[
43  * gst-launch -v videotestsrc num-buffers=1000 ! vp9enc ! webmmux ! filesink location=videotestsrc.webm
44  * ]| This example pipeline will encode a test video source to VP9 muxed in an
45  * WebM container.
46  * </refsect2>
47  */
48
49 #ifdef HAVE_CONFIG_H
50 #include "config.h"
51 #endif
52
53 #ifdef HAVE_VP9_ENCODER
54
55 /* glib decided in 2.32 it would be a great idea to deprecated GValueArray without
56  * providing an alternative
57  *
58  * See https://bugzilla.gnome.org/show_bug.cgi?id=667228
59  * */
60 #define GLIB_DISABLE_DEPRECATION_WARNINGS
61
62 #include <gst/tag/tag.h>
63 #include <gst/video/video.h>
64 #include <string.h>
65
66 #include "gstvp8utils.h"
67 #include "gstvp9enc.h"
68
69 GST_DEBUG_CATEGORY_STATIC (gst_vp9enc_debug);
70 #define GST_CAT_DEFAULT gst_vp9enc_debug
71
72 /* From vp9/vp9_cx_iface.c */
73 #define DEFAULT_PROFILE 0
74
75 #define DEFAULT_RC_END_USAGE VPX_VBR
76 #define DEFAULT_RC_TARGET_BITRATE 256000
77 #define DEFAULT_RC_MIN_QUANTIZER 4
78 #define DEFAULT_RC_MAX_QUANTIZER 63
79
80 #define DEFAULT_RC_DROPFRAME_THRESH 0
81 #define DEFAULT_RC_RESIZE_ALLOWED 0
82 #define DEFAULT_RC_RESIZE_UP_THRESH 30
83 #define DEFAULT_RC_RESIZE_DOWN_THRESH 60
84 #define DEFAULT_RC_UNDERSHOOT_PCT 100
85 #define DEFAULT_RC_OVERSHOOT_PCT 100
86 #define DEFAULT_RC_BUF_SZ 6000
87 #define DEFAULT_RC_BUF_INITIAL_SZ 4000
88 #define DEFAULT_RC_BUF_OPTIMAL_SZ 5000
89 #define DEFAULT_RC_2PASS_VBR_BIAS_PCT 50
90 #define DEFAULT_RC_2PASS_VBR_MINSECTION_PCT 0
91 #define DEFAULT_RC_2PASS_VBR_MAXSECTION_PCT 400
92
93 #define DEFAULT_KF_MODE VPX_KF_AUTO
94 #define DEFAULT_KF_MAX_DIST 128
95
96 #define DEFAULT_MULTIPASS_MODE VPX_RC_ONE_PASS
97 #define DEFAULT_MULTIPASS_CACHE_FILE "multipass.cache"
98
99 #define DEFAULT_TS_NUMBER_LAYERS 1
100 #define DEFAULT_TS_TARGET_BITRATE NULL
101 #define DEFAULT_TS_RATE_DECIMATOR NULL
102 #define DEFAULT_TS_PERIODICITY 0
103 #define DEFAULT_TS_LAYER_ID NULL
104
105 #define DEFAULT_ERROR_RESILIENT 0
106 #define DEFAULT_LAG_IN_FRAMES 0
107
108 #define DEFAULT_THREADS 0
109
110 #define DEFAULT_H_SCALING_MODE VP8E_NORMAL
111 #define DEFAULT_V_SCALING_MODE VP8E_NORMAL
112 #define DEFAULT_CPU_USED 0
113 #define DEFAULT_ENABLE_AUTO_ALT_REF FALSE
114 #define DEFAULT_DEADLINE VPX_DL_BEST_QUALITY
115 #define DEFAULT_NOISE_SENSITIVITY 0
116 #define DEFAULT_SHARPNESS 0
117 #define DEFAULT_STATIC_THRESHOLD 0
118 #define DEFAULT_TOKEN_PARTITIONS 0
119 #define DEFAULT_ARNR_MAXFRAMES 0
120 #define DEFAULT_ARNR_STRENGTH 3
121 #define DEFAULT_ARNR_TYPE 3
122 #define DEFAULT_TUNING VP8_TUNE_PSNR
123 #define DEFAULT_CQ_LEVEL 10
124 #define DEFAULT_MAX_INTRA_BITRATE_PCT 0
125 #define DEFAULT_TIMEBASE_N 0
126 #define DEFAULT_TIMEBASE_D 1
127
128 enum
129 {
130   PROP_0,
131   PROP_RC_END_USAGE,
132   PROP_RC_TARGET_BITRATE,
133   PROP_RC_MIN_QUANTIZER,
134   PROP_RC_MAX_QUANTIZER,
135   PROP_RC_DROPFRAME_THRESH,
136   PROP_RC_RESIZE_ALLOWED,
137   PROP_RC_RESIZE_UP_THRESH,
138   PROP_RC_RESIZE_DOWN_THRESH,
139   PROP_RC_UNDERSHOOT_PCT,
140   PROP_RC_OVERSHOOT_PCT,
141   PROP_RC_BUF_SZ,
142   PROP_RC_BUF_INITIAL_SZ,
143   PROP_RC_BUF_OPTIMAL_SZ,
144   PROP_RC_2PASS_VBR_BIAS_PCT,
145   PROP_RC_2PASS_VBR_MINSECTION_PCT,
146   PROP_RC_2PASS_VBR_MAXSECTION_PCT,
147   PROP_KF_MODE,
148   PROP_KF_MAX_DIST,
149   PROP_TS_NUMBER_LAYERS,
150   PROP_TS_TARGET_BITRATE,
151   PROP_TS_RATE_DECIMATOR,
152   PROP_TS_PERIODICITY,
153   PROP_TS_LAYER_ID,
154   PROP_MULTIPASS_MODE,
155   PROP_MULTIPASS_CACHE_FILE,
156   PROP_ERROR_RESILIENT,
157   PROP_LAG_IN_FRAMES,
158   PROP_THREADS,
159   PROP_DEADLINE,
160   PROP_H_SCALING_MODE,
161   PROP_V_SCALING_MODE,
162   PROP_CPU_USED,
163   PROP_ENABLE_AUTO_ALT_REF,
164   PROP_NOISE_SENSITIVITY,
165   PROP_SHARPNESS,
166   PROP_STATIC_THRESHOLD,
167   PROP_TOKEN_PARTITIONS,
168   PROP_ARNR_MAXFRAMES,
169   PROP_ARNR_STRENGTH,
170   PROP_ARNR_TYPE,
171   PROP_TUNING,
172   PROP_CQ_LEVEL,
173   PROP_MAX_INTRA_BITRATE_PCT,
174   PROP_TIMEBASE
175 };
176
177 #define GST_VP9_ENC_END_USAGE_TYPE (gst_vp9_enc_end_usage_get_type())
178 static GType
179 gst_vp9_enc_end_usage_get_type (void)
180 {
181   static const GEnumValue values[] = {
182     {VPX_VBR, "Variable Bit Rate (VBR) mode", "vbr"},
183     {VPX_CBR, "Constant Bit Rate (CBR) mode", "cbr"},
184     {VPX_CQ, "Constant Quality Mode (CQ) mode", "cq"},
185     {0, NULL, NULL}
186   };
187   static volatile GType id = 0;
188
189   if (g_once_init_enter ((gsize *) & id)) {
190     GType _id;
191
192     _id = g_enum_register_static ("GstVP9EncEndUsage", values);
193
194     g_once_init_leave ((gsize *) & id, _id);
195   }
196
197   return id;
198 }
199
200 #define GST_VP9_ENC_MULTIPASS_MODE_TYPE (gst_vp9_enc_multipass_mode_get_type())
201 static GType
202 gst_vp9_enc_multipass_mode_get_type (void)
203 {
204   static const GEnumValue values[] = {
205     {VPX_RC_ONE_PASS, "One pass encoding (default)", "one-pass"},
206     {VPX_RC_FIRST_PASS, "First pass of multipass encoding", "first-pass"},
207     {VPX_RC_LAST_PASS, "Last pass of multipass encoding", "last-pass"},
208     {0, NULL, NULL}
209   };
210   static volatile GType id = 0;
211
212   if (g_once_init_enter ((gsize *) & id)) {
213     GType _id;
214
215     _id = g_enum_register_static ("GstVP9EncMultipassMode", values);
216
217     g_once_init_leave ((gsize *) & id, _id);
218   }
219
220   return id;
221 }
222
223 #define GST_VP9_ENC_KF_MODE_TYPE (gst_vp9_enc_kf_mode_get_type())
224 static GType
225 gst_vp9_enc_kf_mode_get_type (void)
226 {
227   static const GEnumValue values[] = {
228     {VPX_KF_AUTO, "Determine optimal placement automatically", "auto"},
229     {VPX_KF_DISABLED, "Don't automatically place keyframes", "disabled"},
230     {0, NULL, NULL}
231   };
232   static volatile GType id = 0;
233
234   if (g_once_init_enter ((gsize *) & id)) {
235     GType _id;
236
237     _id = g_enum_register_static ("GstVP9EncKfMode", values);
238
239     g_once_init_leave ((gsize *) & id, _id);
240   }
241
242   return id;
243 }
244
245 #define GST_VP9_ENC_TUNING_TYPE (gst_vp9_enc_tuning_get_type())
246 static GType
247 gst_vp9_enc_tuning_get_type (void)
248 {
249   static const GEnumValue values[] = {
250     {VP8_TUNE_PSNR, "Tune for PSNR", "psnr"},
251     {VP8_TUNE_SSIM, "Tune for SSIM", "ssim"},
252     {0, NULL, NULL}
253   };
254   static volatile GType id = 0;
255
256   if (g_once_init_enter ((gsize *) & id)) {
257     GType _id;
258
259     _id = g_enum_register_static ("GstVP9EncTuning", values);
260
261     g_once_init_leave ((gsize *) & id, _id);
262   }
263
264   return id;
265 }
266
267 #define GST_VP9_ENC_SCALING_MODE_TYPE (gst_vp9_enc_scaling_mode_get_type())
268 static GType
269 gst_vp9_enc_scaling_mode_get_type (void)
270 {
271   static const GEnumValue values[] = {
272     {VP8E_NORMAL, "Normal", "normal"},
273     {VP8E_FOURFIVE, "4:5", "4:5"},
274     {VP8E_THREEFIVE, "3:5", "3:5"},
275     {VP8E_ONETWO, "1:2", "1:2"},
276     {0, NULL, NULL}
277   };
278   static volatile GType id = 0;
279
280   if (g_once_init_enter ((gsize *) & id)) {
281     GType _id;
282
283     _id = g_enum_register_static ("GstVP9EncScalingMode", values);
284
285     g_once_init_leave ((gsize *) & id, _id);
286   }
287
288   return id;
289 }
290
291 #define GST_VP9_ENC_TOKEN_PARTITIONS_TYPE (gst_vp9_enc_token_partitions_get_type())
292 static GType
293 gst_vp9_enc_token_partitions_get_type (void)
294 {
295   static const GEnumValue values[] = {
296     {VP8_ONE_TOKENPARTITION, "One token partition", "1"},
297     {VP8_TWO_TOKENPARTITION, "Two token partitions", "2"},
298     {VP8_FOUR_TOKENPARTITION, "Four token partitions", "4"},
299     {VP8_EIGHT_TOKENPARTITION, "Eight token partitions", "8"},
300     {0, NULL, NULL}
301   };
302   static volatile GType id = 0;
303
304   if (g_once_init_enter ((gsize *) & id)) {
305     GType _id;
306
307     _id = g_enum_register_static ("GstVP9EncTokenPartitions", values);
308
309     g_once_init_leave ((gsize *) & id, _id);
310   }
311
312   return id;
313 }
314
315 #define GST_VP9_ENC_ER_FLAGS_TYPE (gst_vp9_enc_er_flags_get_type())
316 static GType
317 gst_vp9_enc_er_flags_get_type (void)
318 {
319   static const GFlagsValue values[] = {
320     {VPX_ERROR_RESILIENT_DEFAULT, "Default error resilience", "default"},
321     {VPX_ERROR_RESILIENT_PARTITIONS,
322         "Allow partitions to be decoded independently", "partitions"},
323     {0, NULL, NULL}
324   };
325   static volatile GType id = 0;
326
327   if (g_once_init_enter ((gsize *) & id)) {
328     GType _id;
329
330     _id = g_flags_register_static ("GstVP9EncErFlags", values);
331
332     g_once_init_leave ((gsize *) & id, _id);
333   }
334
335   return id;
336 }
337
338 static void gst_vp9_enc_finalize (GObject * object);
339 static void gst_vp9_enc_set_property (GObject * object, guint prop_id,
340     const GValue * value, GParamSpec * pspec);
341 static void gst_vp9_enc_get_property (GObject * object, guint prop_id,
342     GValue * value, GParamSpec * pspec);
343
344 static gboolean gst_vp9_enc_start (GstVideoEncoder * encoder);
345 static gboolean gst_vp9_enc_stop (GstVideoEncoder * encoder);
346 static gboolean gst_vp9_enc_set_format (GstVideoEncoder *
347     video_encoder, GstVideoCodecState * state);
348 static GstFlowReturn gst_vp9_enc_finish (GstVideoEncoder * video_encoder);
349 static GstFlowReturn gst_vp9_enc_handle_frame (GstVideoEncoder *
350     video_encoder, GstVideoCodecFrame * frame);
351 static GstFlowReturn gst_vp9_enc_drain (GstVideoEncoder * video_encoder);
352 static gboolean gst_vp9_enc_sink_event (GstVideoEncoder *
353     video_encoder, GstEvent * event);
354 static gboolean gst_vp9_enc_propose_allocation (GstVideoEncoder * encoder,
355     GstQuery * query);
356
357 /* FIXME: Y42B and Y444 do not work yet it seems */
358 static GstStaticPadTemplate gst_vp9_enc_sink_template =
359 GST_STATIC_PAD_TEMPLATE ("sink",
360     GST_PAD_SINK,
361     GST_PAD_ALWAYS,
362     /*GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ I420, YV12, Y42B, Y444 }")) */
363     GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ I420, YV12 }"))
364     );
365
366 static GstStaticPadTemplate gst_vp9_enc_src_template =
367 GST_STATIC_PAD_TEMPLATE ("src",
368     GST_PAD_SRC,
369     GST_PAD_ALWAYS,
370     GST_STATIC_CAPS ("video/x-vp9, " "profile = (string) {0, 1, 2, 3}")
371     );
372
373 #define parent_class gst_vp9_enc_parent_class
374 G_DEFINE_TYPE_WITH_CODE (GstVP9Enc, gst_vp9_enc, GST_TYPE_VIDEO_ENCODER,
375     G_IMPLEMENT_INTERFACE (GST_TYPE_TAG_SETTER, NULL);
376     G_IMPLEMENT_INTERFACE (GST_TYPE_PRESET, NULL););
377
378 static void
379 gst_vp9_enc_class_init (GstVP9EncClass * klass)
380 {
381   GObjectClass *gobject_class;
382   GstElementClass *element_class;
383   GstVideoEncoderClass *video_encoder_class;
384
385   gobject_class = G_OBJECT_CLASS (klass);
386   element_class = GST_ELEMENT_CLASS (klass);
387   video_encoder_class = GST_VIDEO_ENCODER_CLASS (klass);
388
389   gobject_class->set_property = gst_vp9_enc_set_property;
390   gobject_class->get_property = gst_vp9_enc_get_property;
391   gobject_class->finalize = gst_vp9_enc_finalize;
392
393   gst_element_class_add_pad_template (element_class,
394       gst_static_pad_template_get (&gst_vp9_enc_src_template));
395   gst_element_class_add_pad_template (element_class,
396       gst_static_pad_template_get (&gst_vp9_enc_sink_template));
397
398   gst_element_class_set_static_metadata (element_class,
399       "On2 VP9 Encoder",
400       "Codec/Encoder/Video",
401       "Encode VP9 video streams", "David Schleef <ds@entropywave.com>, "
402       "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
403
404   video_encoder_class->start = gst_vp9_enc_start;
405   video_encoder_class->stop = gst_vp9_enc_stop;
406   video_encoder_class->handle_frame = gst_vp9_enc_handle_frame;
407   video_encoder_class->set_format = gst_vp9_enc_set_format;
408   video_encoder_class->finish = gst_vp9_enc_finish;
409   video_encoder_class->sink_event = gst_vp9_enc_sink_event;
410   video_encoder_class->propose_allocation = gst_vp9_enc_propose_allocation;
411
412   g_object_class_install_property (gobject_class, PROP_RC_END_USAGE,
413       g_param_spec_enum ("end-usage", "Rate control mode",
414           "Rate control mode",
415           GST_VP9_ENC_END_USAGE_TYPE, DEFAULT_RC_END_USAGE,
416           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
417
418   g_object_class_install_property (gobject_class, PROP_RC_TARGET_BITRATE,
419       g_param_spec_int ("target-bitrate", "Target bitrate",
420           "Target bitrate (in bits/sec)",
421           0, G_MAXINT, DEFAULT_RC_TARGET_BITRATE,
422           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
423
424   g_object_class_install_property (gobject_class, PROP_RC_MIN_QUANTIZER,
425       g_param_spec_int ("min-quantizer", "Minimum Quantizer",
426           "Minimum Quantizer (best)",
427           0, 63, DEFAULT_RC_MIN_QUANTIZER,
428           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
429
430   g_object_class_install_property (gobject_class, PROP_RC_MAX_QUANTIZER,
431       g_param_spec_int ("max-quantizer", "Maximum Quantizer",
432           "Maximum Quantizer (worst)",
433           0, 63, DEFAULT_RC_MAX_QUANTIZER,
434           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
435
436   g_object_class_install_property (gobject_class, PROP_RC_DROPFRAME_THRESH,
437       g_param_spec_int ("dropframe-threshold", "Drop Frame Threshold",
438           "Temporal resampling threshold (buf %)",
439           0, 100, DEFAULT_RC_DROPFRAME_THRESH,
440           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
441
442   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_ALLOWED,
443       g_param_spec_boolean ("resize-allowed", "Resize Allowed",
444           "Allow spatial resampling",
445           DEFAULT_RC_RESIZE_ALLOWED,
446           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
447
448   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_UP_THRESH,
449       g_param_spec_int ("resize-up-threshold", "Resize Up Threshold",
450           "Upscale threshold (buf %)",
451           0, 100, DEFAULT_RC_RESIZE_UP_THRESH,
452           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
453
454   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_DOWN_THRESH,
455       g_param_spec_int ("resize-down-threshold", "Resize Down Threshold",
456           "Downscale threshold (buf %)",
457           0, 100, DEFAULT_RC_RESIZE_DOWN_THRESH,
458           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
459
460   g_object_class_install_property (gobject_class, PROP_RC_UNDERSHOOT_PCT,
461       g_param_spec_int ("undershoot", "Undershoot PCT",
462           "Datarate undershoot (min) target (%)",
463           0, 1000, DEFAULT_RC_UNDERSHOOT_PCT,
464           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
465
466   g_object_class_install_property (gobject_class, PROP_RC_OVERSHOOT_PCT,
467       g_param_spec_int ("overshoot", "Overshoot PCT",
468           "Datarate overshoot (max) target (%)",
469           0, 1000, DEFAULT_RC_OVERSHOOT_PCT,
470           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
471
472   g_object_class_install_property (gobject_class, PROP_RC_BUF_SZ,
473       g_param_spec_int ("buffer-size", "Buffer size",
474           "Client buffer size (ms)",
475           0, G_MAXINT, DEFAULT_RC_BUF_SZ,
476           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
477
478   g_object_class_install_property (gobject_class, PROP_RC_BUF_INITIAL_SZ,
479       g_param_spec_int ("buffer-initial-size", "Buffer initial size",
480           "Initial client buffer size (ms)",
481           0, G_MAXINT, DEFAULT_RC_BUF_INITIAL_SZ,
482           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
483
484   g_object_class_install_property (gobject_class, PROP_RC_BUF_OPTIMAL_SZ,
485       g_param_spec_int ("buffer-optimal-size", "Buffer optimal size",
486           "Optimal client buffer size (ms)",
487           0, G_MAXINT, DEFAULT_RC_BUF_OPTIMAL_SZ,
488           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
489
490   g_object_class_install_property (gobject_class, PROP_RC_2PASS_VBR_BIAS_PCT,
491       g_param_spec_int ("twopass-vbr-bias", "2-pass VBR bias",
492           "CBR/VBR bias (0=CBR, 100=VBR)",
493           0, 100, DEFAULT_RC_2PASS_VBR_BIAS_PCT,
494           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
495
496   g_object_class_install_property (gobject_class,
497       PROP_RC_2PASS_VBR_MINSECTION_PCT,
498       g_param_spec_int ("twopass-vbr-minsection", "2-pass GOP min bitrate",
499           "GOP minimum bitrate (% target)", 0, G_MAXINT,
500           DEFAULT_RC_2PASS_VBR_MINSECTION_PCT,
501           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
502
503   g_object_class_install_property (gobject_class,
504       PROP_RC_2PASS_VBR_MAXSECTION_PCT,
505       g_param_spec_int ("twopass-vbr-maxsection", "2-pass GOP max bitrate",
506           "GOP maximum bitrate (% target)", 0, G_MAXINT,
507           DEFAULT_RC_2PASS_VBR_MINSECTION_PCT,
508           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
509
510   g_object_class_install_property (gobject_class, PROP_KF_MODE,
511       g_param_spec_enum ("keyframe-mode", "Keyframe Mode",
512           "Keyframe placement",
513           GST_VP9_ENC_KF_MODE_TYPE, DEFAULT_KF_MODE,
514           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
515
516   g_object_class_install_property (gobject_class, PROP_KF_MAX_DIST,
517       g_param_spec_int ("keyframe-max-dist", "Keyframe max distance",
518           "Maximum distance between keyframes (number of frames)",
519           0, G_MAXINT, DEFAULT_KF_MAX_DIST,
520           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
521
522   g_object_class_install_property (gobject_class, PROP_MULTIPASS_MODE,
523       g_param_spec_enum ("multipass-mode", "Multipass Mode",
524           "Multipass encode mode",
525           GST_VP9_ENC_MULTIPASS_MODE_TYPE, DEFAULT_MULTIPASS_MODE,
526           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
527
528   g_object_class_install_property (gobject_class, PROP_MULTIPASS_CACHE_FILE,
529       g_param_spec_string ("multipass-cache-file", "Multipass Cache File",
530           "Multipass cache file",
531           DEFAULT_MULTIPASS_CACHE_FILE,
532           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
533
534   g_object_class_install_property (gobject_class, PROP_TS_NUMBER_LAYERS,
535       g_param_spec_int ("temporal-scalability-number-layers",
536           "Number of coding layers", "Number of coding layers to use", 1, 5,
537           DEFAULT_TS_NUMBER_LAYERS,
538           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
539
540   g_object_class_install_property (gobject_class, PROP_TS_TARGET_BITRATE,
541       g_param_spec_value_array ("temporal-scalability-target-bitrate",
542           "Coding layer target bitrates",
543           "Target bitrates for coding layers (one per layer, decreasing)",
544           g_param_spec_int ("target-bitrate", "Target bitrate",
545               "Target bitrate", 0, G_MAXINT, DEFAULT_RC_TARGET_BITRATE,
546               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
547           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
548
549   g_object_class_install_property (gobject_class, PROP_TS_RATE_DECIMATOR,
550       g_param_spec_value_array ("temporal-scalability-rate-decimator",
551           "Coding layer rate decimator",
552           "Rate decimation factors for each layer",
553           g_param_spec_int ("rate-decimator", "Rate decimator",
554               "Rate decimator", 0, 1000000000, 0,
555               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
556           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
557
558   g_object_class_install_property (gobject_class, PROP_TS_PERIODICITY,
559       g_param_spec_int ("temporal-scalability-periodicity",
560           "Coding layer periodicity",
561           "Length of sequence that defines layer membership periodicity", 0, 16,
562           DEFAULT_TS_PERIODICITY,
563           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
564
565   g_object_class_install_property (gobject_class, PROP_TS_LAYER_ID,
566       g_param_spec_value_array ("temporal-scalability-layer-id",
567           "Coding layer identification",
568           "Sequence defining coding layer membership",
569           g_param_spec_int ("layer-id", "Layer ID", "Layer ID", 0, 4, 0,
570               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
571           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
572
573   g_object_class_install_property (gobject_class, PROP_LAG_IN_FRAMES,
574       g_param_spec_int ("lag-in-frames", "Lag in frames",
575           "Maximum number of frames to lag",
576           0, 25, DEFAULT_LAG_IN_FRAMES,
577           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
578
579   g_object_class_install_property (gobject_class, PROP_ERROR_RESILIENT,
580       g_param_spec_flags ("error-resilient", "Error resilient",
581           "Error resilience flags",
582           GST_VP9_ENC_ER_FLAGS_TYPE, DEFAULT_ERROR_RESILIENT,
583           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
584
585   g_object_class_install_property (gobject_class, PROP_THREADS,
586       g_param_spec_int ("threads", "Threads",
587           "Number of threads to use",
588           0, 64, DEFAULT_THREADS,
589           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
590
591   g_object_class_install_property (gobject_class, PROP_DEADLINE,
592       g_param_spec_int64 ("deadline", "Deadline",
593           "Deadline per frame (usec, 0=disabled)",
594           0, G_MAXINT64, DEFAULT_DEADLINE,
595           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
596
597   g_object_class_install_property (gobject_class, PROP_H_SCALING_MODE,
598       g_param_spec_enum ("horizontal-scaling-mode", "Horizontal scaling mode",
599           "Horizontal scaling mode",
600           GST_VP9_ENC_SCALING_MODE_TYPE, DEFAULT_H_SCALING_MODE,
601           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
602
603   g_object_class_install_property (gobject_class, PROP_V_SCALING_MODE,
604       g_param_spec_enum ("vertical-scaling-mode", "Vertical scaling mode",
605           "Vertical scaling mode",
606           GST_VP9_ENC_SCALING_MODE_TYPE, DEFAULT_V_SCALING_MODE,
607           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
608
609   g_object_class_install_property (gobject_class, PROP_CPU_USED,
610       g_param_spec_int ("cpu-used", "CPU used",
611           "CPU used",
612           -16, 16, DEFAULT_CPU_USED,
613           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
614
615   g_object_class_install_property (gobject_class, PROP_ENABLE_AUTO_ALT_REF,
616       g_param_spec_boolean ("auto-alt-ref", "Auto alt reference frames",
617           "Automatically generate AltRef frames",
618           DEFAULT_ENABLE_AUTO_ALT_REF,
619           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
620
621   g_object_class_install_property (gobject_class, PROP_NOISE_SENSITIVITY,
622       g_param_spec_int ("noise-sensitivity", "Noise sensitivity",
623           "Noise sensisivity (frames to blur)",
624           0, 6, DEFAULT_NOISE_SENSITIVITY,
625           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
626
627   g_object_class_install_property (gobject_class, PROP_SHARPNESS,
628       g_param_spec_int ("sharpness", "Sharpness",
629           "Filter sharpness",
630           0, 7, DEFAULT_SHARPNESS,
631           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
632
633   g_object_class_install_property (gobject_class, PROP_STATIC_THRESHOLD,
634       g_param_spec_int ("static-threshold", "Static Threshold",
635           "Motion detection threshold",
636           0, G_MAXINT, DEFAULT_STATIC_THRESHOLD,
637           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
638
639   g_object_class_install_property (gobject_class, PROP_TOKEN_PARTITIONS,
640       g_param_spec_enum ("token-partitions", "Token partitions",
641           "Number of token partitions",
642           GST_VP9_ENC_TOKEN_PARTITIONS_TYPE, DEFAULT_TOKEN_PARTITIONS,
643           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
644
645   g_object_class_install_property (gobject_class, PROP_ARNR_MAXFRAMES,
646       g_param_spec_int ("arnr-maxframes", "AltRef max frames",
647           "AltRef maximum number of frames",
648           0, 15, DEFAULT_ARNR_MAXFRAMES,
649           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
650
651   g_object_class_install_property (gobject_class, PROP_ARNR_STRENGTH,
652       g_param_spec_int ("arnr-strength", "AltRef strength",
653           "AltRef strength",
654           0, 6, DEFAULT_ARNR_STRENGTH,
655           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
656
657   g_object_class_install_property (gobject_class, PROP_ARNR_TYPE,
658       g_param_spec_int ("arnr-type", "AltRef type",
659           "AltRef type",
660           1, 3, DEFAULT_ARNR_TYPE,
661           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
662
663   g_object_class_install_property (gobject_class, PROP_TUNING,
664       g_param_spec_enum ("tuning", "Tuning",
665           "Tuning",
666           GST_VP9_ENC_TUNING_TYPE, DEFAULT_TUNING,
667           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
668
669   g_object_class_install_property (gobject_class, PROP_CQ_LEVEL,
670       g_param_spec_int ("cq-level", "Constrained quality level",
671           "Constrained quality level",
672           0, 63, DEFAULT_CQ_LEVEL,
673           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
674
675   g_object_class_install_property (gobject_class, PROP_MAX_INTRA_BITRATE_PCT,
676       g_param_spec_int ("max-intra-bitrate", "Max Intra bitrate",
677           "Maximum Intra frame bitrate",
678           0, G_MAXINT, DEFAULT_MAX_INTRA_BITRATE_PCT,
679           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
680
681   g_object_class_install_property (gobject_class, PROP_TIMEBASE,
682       gst_param_spec_fraction ("timebase", "Shortest interframe time",
683           "Fraction of one second that is the shortest interframe time - normally left as zero which will default to the framerate",
684           0, 1, G_MAXINT, 1, DEFAULT_TIMEBASE_N, DEFAULT_TIMEBASE_D,
685           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
686
687   GST_DEBUG_CATEGORY_INIT (gst_vp9enc_debug, "vp9enc", 0, "VP9 Encoder");
688 }
689
690 static void
691 gst_vp9_enc_init (GstVP9Enc * gst_vp9_enc)
692 {
693   vpx_codec_err_t status;
694
695   GST_DEBUG_OBJECT (gst_vp9_enc, "init");
696
697   status =
698       vpx_codec_enc_config_default (&vpx_codec_vp9_cx_algo, &gst_vp9_enc->cfg,
699       0);
700   if (status != VPX_CODEC_OK) {
701     GST_ERROR_OBJECT (gst_vp9_enc,
702         "Failed to get default encoder configuration: %s",
703         gst_vpx_error_name (status));
704     gst_vp9_enc->have_default_config = FALSE;
705   } else {
706     gst_vp9_enc->have_default_config = TRUE;
707   }
708
709   gst_vp9_enc->cfg.rc_end_usage = DEFAULT_RC_END_USAGE;
710   gst_vp9_enc->cfg.rc_target_bitrate = DEFAULT_RC_TARGET_BITRATE / 1000;
711   gst_vp9_enc->rc_target_bitrate_set = FALSE;
712   gst_vp9_enc->cfg.rc_min_quantizer = DEFAULT_RC_MIN_QUANTIZER;
713   gst_vp9_enc->cfg.rc_max_quantizer = DEFAULT_RC_MAX_QUANTIZER;
714   gst_vp9_enc->cfg.rc_dropframe_thresh = DEFAULT_RC_DROPFRAME_THRESH;
715   gst_vp9_enc->cfg.rc_resize_allowed = DEFAULT_RC_RESIZE_ALLOWED;
716   gst_vp9_enc->cfg.rc_resize_up_thresh = DEFAULT_RC_RESIZE_UP_THRESH;
717   gst_vp9_enc->cfg.rc_resize_down_thresh = DEFAULT_RC_RESIZE_DOWN_THRESH;
718   gst_vp9_enc->cfg.rc_undershoot_pct = DEFAULT_RC_UNDERSHOOT_PCT;
719   gst_vp9_enc->cfg.rc_overshoot_pct = DEFAULT_RC_OVERSHOOT_PCT;
720   gst_vp9_enc->cfg.rc_buf_sz = DEFAULT_RC_BUF_SZ;
721   gst_vp9_enc->cfg.rc_buf_initial_sz = DEFAULT_RC_BUF_INITIAL_SZ;
722   gst_vp9_enc->cfg.rc_buf_optimal_sz = DEFAULT_RC_BUF_OPTIMAL_SZ;
723   gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct = DEFAULT_RC_2PASS_VBR_BIAS_PCT;
724   gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct =
725       DEFAULT_RC_2PASS_VBR_MINSECTION_PCT;
726   gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct =
727       DEFAULT_RC_2PASS_VBR_MAXSECTION_PCT;
728   gst_vp9_enc->cfg.kf_mode = DEFAULT_KF_MODE;
729   gst_vp9_enc->cfg.kf_max_dist = DEFAULT_KF_MAX_DIST;
730   gst_vp9_enc->cfg.g_pass = DEFAULT_MULTIPASS_MODE;
731   gst_vp9_enc->multipass_cache_file = g_strdup (DEFAULT_MULTIPASS_CACHE_FILE);
732   gst_vp9_enc->cfg.ts_number_layers = DEFAULT_TS_NUMBER_LAYERS;
733   gst_vp9_enc->n_ts_target_bitrate = 0;
734   gst_vp9_enc->n_ts_rate_decimator = 0;
735   gst_vp9_enc->cfg.ts_periodicity = DEFAULT_TS_PERIODICITY;
736   gst_vp9_enc->n_ts_layer_id = 0;
737   gst_vp9_enc->cfg.g_error_resilient = DEFAULT_ERROR_RESILIENT;
738   gst_vp9_enc->cfg.g_lag_in_frames = DEFAULT_LAG_IN_FRAMES;
739   gst_vp9_enc->cfg.g_threads = DEFAULT_THREADS;
740   gst_vp9_enc->deadline = DEFAULT_DEADLINE;
741   gst_vp9_enc->h_scaling_mode = DEFAULT_H_SCALING_MODE;
742   gst_vp9_enc->v_scaling_mode = DEFAULT_V_SCALING_MODE;
743   gst_vp9_enc->cpu_used = DEFAULT_CPU_USED;
744   gst_vp9_enc->enable_auto_alt_ref = DEFAULT_ENABLE_AUTO_ALT_REF;
745   gst_vp9_enc->noise_sensitivity = DEFAULT_NOISE_SENSITIVITY;
746   gst_vp9_enc->sharpness = DEFAULT_SHARPNESS;
747   gst_vp9_enc->static_threshold = DEFAULT_STATIC_THRESHOLD;
748   gst_vp9_enc->token_partitions = DEFAULT_TOKEN_PARTITIONS;
749   gst_vp9_enc->arnr_maxframes = DEFAULT_ARNR_MAXFRAMES;
750   gst_vp9_enc->arnr_strength = DEFAULT_ARNR_STRENGTH;
751   gst_vp9_enc->arnr_type = DEFAULT_ARNR_TYPE;
752   gst_vp9_enc->tuning = DEFAULT_TUNING;
753   gst_vp9_enc->cq_level = DEFAULT_CQ_LEVEL;
754   gst_vp9_enc->max_intra_bitrate_pct = DEFAULT_MAX_INTRA_BITRATE_PCT;
755   gst_vp9_enc->timebase_n = DEFAULT_TIMEBASE_N;
756   gst_vp9_enc->timebase_d = DEFAULT_TIMEBASE_D;
757
758   gst_vp9_enc->cfg.g_profile = DEFAULT_PROFILE;
759
760   g_mutex_init (&gst_vp9_enc->encoder_lock);
761 }
762
763 static void
764 gst_vp9_enc_finalize (GObject * object)
765 {
766   GstVP9Enc *gst_vp9_enc;
767
768   GST_DEBUG_OBJECT (object, "finalize");
769
770   g_return_if_fail (GST_IS_VP9_ENC (object));
771   gst_vp9_enc = GST_VP9_ENC (object);
772
773   g_free (gst_vp9_enc->multipass_cache_file);
774   gst_vp9_enc->multipass_cache_file = NULL;
775
776   if (gst_vp9_enc->input_state)
777     gst_video_codec_state_unref (gst_vp9_enc->input_state);
778
779   g_mutex_clear (&gst_vp9_enc->encoder_lock);
780
781   G_OBJECT_CLASS (parent_class)->finalize (object);
782
783 }
784
785 static void
786 gst_vp9_enc_set_property (GObject * object, guint prop_id,
787     const GValue * value, GParamSpec * pspec)
788 {
789   GstVP9Enc *gst_vp9_enc;
790   gboolean global = FALSE;
791   vpx_codec_err_t status;
792
793   g_return_if_fail (GST_IS_VP9_ENC (object));
794   gst_vp9_enc = GST_VP9_ENC (object);
795
796   GST_DEBUG_OBJECT (object, "gst_vp9_enc_set_property");
797   g_mutex_lock (&gst_vp9_enc->encoder_lock);
798   switch (prop_id) {
799     case PROP_RC_END_USAGE:
800       gst_vp9_enc->cfg.rc_end_usage = g_value_get_enum (value);
801       global = TRUE;
802       break;
803     case PROP_RC_TARGET_BITRATE:
804       gst_vp9_enc->cfg.rc_target_bitrate = g_value_get_int (value) / 1000;
805       gst_vp9_enc->rc_target_bitrate_set = TRUE;
806       global = TRUE;
807       break;
808     case PROP_RC_MIN_QUANTIZER:
809       gst_vp9_enc->cfg.rc_min_quantizer = g_value_get_int (value);
810       global = TRUE;
811       break;
812     case PROP_RC_MAX_QUANTIZER:
813       gst_vp9_enc->cfg.rc_max_quantizer = g_value_get_int (value);
814       global = TRUE;
815       break;
816     case PROP_RC_DROPFRAME_THRESH:
817       gst_vp9_enc->cfg.rc_dropframe_thresh = g_value_get_int (value);
818       global = TRUE;
819       break;
820     case PROP_RC_RESIZE_ALLOWED:
821       gst_vp9_enc->cfg.rc_resize_allowed = g_value_get_boolean (value);
822       global = TRUE;
823       break;
824     case PROP_RC_RESIZE_UP_THRESH:
825       gst_vp9_enc->cfg.rc_resize_up_thresh = g_value_get_int (value);
826       global = TRUE;
827       break;
828     case PROP_RC_RESIZE_DOWN_THRESH:
829       gst_vp9_enc->cfg.rc_resize_down_thresh = g_value_get_int (value);
830       global = TRUE;
831       break;
832     case PROP_RC_UNDERSHOOT_PCT:
833       gst_vp9_enc->cfg.rc_undershoot_pct = g_value_get_int (value);
834       global = TRUE;
835       break;
836     case PROP_RC_OVERSHOOT_PCT:
837       gst_vp9_enc->cfg.rc_overshoot_pct = g_value_get_int (value);
838       global = TRUE;
839       break;
840     case PROP_RC_BUF_SZ:
841       gst_vp9_enc->cfg.rc_buf_sz = g_value_get_int (value);
842       global = TRUE;
843       break;
844     case PROP_RC_BUF_INITIAL_SZ:
845       gst_vp9_enc->cfg.rc_buf_initial_sz = g_value_get_int (value);
846       global = TRUE;
847       break;
848     case PROP_RC_BUF_OPTIMAL_SZ:
849       gst_vp9_enc->cfg.rc_buf_optimal_sz = g_value_get_int (value);
850       global = TRUE;
851       break;
852     case PROP_RC_2PASS_VBR_BIAS_PCT:
853       gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct = g_value_get_int (value);
854       global = TRUE;
855       break;
856     case PROP_RC_2PASS_VBR_MINSECTION_PCT:
857       gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct = g_value_get_int (value);
858       global = TRUE;
859       break;
860     case PROP_RC_2PASS_VBR_MAXSECTION_PCT:
861       gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct = g_value_get_int (value);
862       global = TRUE;
863       break;
864     case PROP_KF_MODE:
865       gst_vp9_enc->cfg.kf_mode = g_value_get_enum (value);
866       global = TRUE;
867       break;
868     case PROP_KF_MAX_DIST:
869       gst_vp9_enc->cfg.kf_max_dist = g_value_get_int (value);
870       global = TRUE;
871       break;
872     case PROP_MULTIPASS_MODE:
873       gst_vp9_enc->cfg.g_pass = g_value_get_enum (value);
874       global = TRUE;
875       break;
876     case PROP_MULTIPASS_CACHE_FILE:
877       if (gst_vp9_enc->multipass_cache_file)
878         g_free (gst_vp9_enc->multipass_cache_file);
879       gst_vp9_enc->multipass_cache_file = g_value_dup_string (value);
880       break;
881     case PROP_TS_NUMBER_LAYERS:
882       gst_vp9_enc->cfg.ts_number_layers = g_value_get_int (value);
883       global = TRUE;
884       break;
885     case PROP_TS_TARGET_BITRATE:{
886       GValueArray *va = g_value_get_boxed (value);
887
888       memset (&gst_vp9_enc->cfg.ts_target_bitrate, 0,
889           sizeof (gst_vp9_enc->cfg.ts_target_bitrate));
890       if (va == NULL) {
891         gst_vp9_enc->n_ts_target_bitrate = 0;
892       } else {
893         if (va->n_values > VPX_TS_MAX_LAYERS) {
894           g_warning ("%s: Only %d layers allowed at maximum",
895               GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_LAYERS);
896         } else {
897           gint i;
898
899           for (i = 0; i < va->n_values; i++)
900             gst_vp9_enc->cfg.ts_target_bitrate[i] =
901                 g_value_get_int (g_value_array_get_nth (va, i));
902           gst_vp9_enc->n_ts_target_bitrate = va->n_values;
903         }
904       }
905       global = TRUE;
906       break;
907     }
908     case PROP_TS_RATE_DECIMATOR:{
909       GValueArray *va = g_value_get_boxed (value);
910
911       memset (&gst_vp9_enc->cfg.ts_rate_decimator, 0,
912           sizeof (gst_vp9_enc->cfg.ts_rate_decimator));
913       if (va == NULL) {
914         gst_vp9_enc->n_ts_rate_decimator = 0;
915       } else if (va->n_values > VPX_TS_MAX_LAYERS) {
916         g_warning ("%s: Only %d layers allowed at maximum",
917             GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_LAYERS);
918       } else {
919         gint i;
920
921         for (i = 0; i < va->n_values; i++)
922           gst_vp9_enc->cfg.ts_rate_decimator[i] =
923               g_value_get_int (g_value_array_get_nth (va, i));
924         gst_vp9_enc->n_ts_rate_decimator = va->n_values;
925       }
926       global = TRUE;
927       break;
928     }
929     case PROP_TS_PERIODICITY:
930       gst_vp9_enc->cfg.ts_periodicity = g_value_get_int (value);
931       global = TRUE;
932       break;
933     case PROP_TS_LAYER_ID:{
934       GValueArray *va = g_value_get_boxed (value);
935
936       memset (&gst_vp9_enc->cfg.ts_layer_id, 0,
937           sizeof (gst_vp9_enc->cfg.ts_layer_id));
938       if (va && va->n_values > VPX_TS_MAX_PERIODICITY) {
939         g_warning ("%s: Only %d sized layer sequences allowed at maximum",
940             GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_PERIODICITY);
941       } else if (va) {
942         gint i;
943
944         for (i = 0; i < va->n_values; i++)
945           gst_vp9_enc->cfg.ts_layer_id[i] =
946               g_value_get_int (g_value_array_get_nth (va, i));
947         gst_vp9_enc->n_ts_layer_id = va->n_values;
948       } else {
949         gst_vp9_enc->n_ts_layer_id = 0;
950       }
951       global = TRUE;
952       break;
953     }
954     case PROP_ERROR_RESILIENT:
955       gst_vp9_enc->cfg.g_error_resilient = g_value_get_flags (value);
956       global = TRUE;
957       break;
958     case PROP_LAG_IN_FRAMES:
959       gst_vp9_enc->cfg.g_lag_in_frames = g_value_get_int (value);
960       global = TRUE;
961       break;
962     case PROP_THREADS:
963       gst_vp9_enc->cfg.g_threads = g_value_get_int (value);
964       global = TRUE;
965       break;
966     case PROP_DEADLINE:
967       gst_vp9_enc->deadline = g_value_get_int64 (value);
968       break;
969     case PROP_H_SCALING_MODE:
970       gst_vp9_enc->h_scaling_mode = g_value_get_enum (value);
971       if (gst_vp9_enc->inited) {
972         vpx_scaling_mode_t sm;
973
974         sm.h_scaling_mode = gst_vp9_enc->h_scaling_mode;
975         sm.v_scaling_mode = gst_vp9_enc->v_scaling_mode;
976
977         status =
978             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SCALEMODE, &sm);
979         if (status != VPX_CODEC_OK) {
980           GST_WARNING_OBJECT (gst_vp9_enc,
981               "Failed to set VP8E_SET_SCALEMODE: %s",
982               gst_vpx_error_name (status));
983         }
984       }
985       break;
986     case PROP_V_SCALING_MODE:
987       gst_vp9_enc->v_scaling_mode = g_value_get_enum (value);
988       if (gst_vp9_enc->inited) {
989         vpx_scaling_mode_t sm;
990
991         sm.h_scaling_mode = gst_vp9_enc->h_scaling_mode;
992         sm.v_scaling_mode = gst_vp9_enc->v_scaling_mode;
993
994         status =
995             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SCALEMODE, &sm);
996         if (status != VPX_CODEC_OK) {
997           GST_WARNING_OBJECT (gst_vp9_enc,
998               "Failed to set VP8E_SET_SCALEMODE: %s",
999               gst_vpx_error_name (status));
1000         }
1001       }
1002       break;
1003     case PROP_CPU_USED:
1004       gst_vp9_enc->cpu_used = g_value_get_int (value);
1005       if (gst_vp9_enc->inited) {
1006         status =
1007             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_CPUUSED,
1008             gst_vp9_enc->cpu_used);
1009         if (status != VPX_CODEC_OK) {
1010           GST_WARNING_OBJECT (gst_vp9_enc, "Failed to set VP8E_SET_CPUUSED: %s",
1011               gst_vpx_error_name (status));
1012         }
1013       }
1014       break;
1015     case PROP_ENABLE_AUTO_ALT_REF:
1016       gst_vp9_enc->enable_auto_alt_ref = g_value_get_boolean (value);
1017       if (gst_vp9_enc->inited) {
1018         status =
1019             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ENABLEAUTOALTREF,
1020             (gst_vp9_enc->enable_auto_alt_ref ? 1 : 0));
1021         if (status != VPX_CODEC_OK) {
1022           GST_WARNING_OBJECT (gst_vp9_enc,
1023               "Failed to set VP8E_SET_ENABLEAUTOALTREF: %s",
1024               gst_vpx_error_name (status));
1025         }
1026       }
1027       break;
1028     case PROP_NOISE_SENSITIVITY:
1029       gst_vp9_enc->noise_sensitivity = g_value_get_int (value);
1030       if (gst_vp9_enc->inited) {
1031         status =
1032             vpx_codec_control (&gst_vp9_enc->encoder,
1033             VP8E_SET_NOISE_SENSITIVITY, gst_vp9_enc->noise_sensitivity);
1034         if (status != VPX_CODEC_OK) {
1035           GST_WARNING_OBJECT (gst_vp9_enc,
1036               "Failed to set VP8E_SET_NOISE_SENSITIVITY: %s",
1037               gst_vpx_error_name (status));
1038         }
1039       }
1040       break;
1041     case PROP_SHARPNESS:
1042       gst_vp9_enc->sharpness = g_value_get_int (value);
1043       if (gst_vp9_enc->inited) {
1044         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SHARPNESS,
1045             gst_vp9_enc->sharpness);
1046         if (status != VPX_CODEC_OK) {
1047           GST_WARNING_OBJECT (gst_vp9_enc,
1048               "Failed to set VP8E_SET_SHARPNESS: %s",
1049               gst_vpx_error_name (status));
1050         }
1051       }
1052       break;
1053     case PROP_STATIC_THRESHOLD:
1054       gst_vp9_enc->static_threshold = g_value_get_int (value);
1055       if (gst_vp9_enc->inited) {
1056         status =
1057             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_STATIC_THRESHOLD,
1058             gst_vp9_enc->static_threshold);
1059         if (status != VPX_CODEC_OK) {
1060           GST_WARNING_OBJECT (gst_vp9_enc,
1061               "Failed to set VP8E_SET_STATIC_THRESHOLD: %s",
1062               gst_vpx_error_name (status));
1063         }
1064       }
1065       break;
1066     case PROP_TOKEN_PARTITIONS:
1067       gst_vp9_enc->token_partitions = g_value_get_enum (value);
1068       if (gst_vp9_enc->inited) {
1069         status =
1070             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_TOKEN_PARTITIONS,
1071             gst_vp9_enc->token_partitions);
1072         if (status != VPX_CODEC_OK) {
1073           GST_WARNING_OBJECT (gst_vp9_enc,
1074               "Failed to set VP8E_SET_TOKEN_PARTIONS: %s",
1075               gst_vpx_error_name (status));
1076         }
1077       }
1078       break;
1079     case PROP_ARNR_MAXFRAMES:
1080       gst_vp9_enc->arnr_maxframes = g_value_get_int (value);
1081       if (gst_vp9_enc->inited) {
1082         status =
1083             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ARNR_MAXFRAMES,
1084             gst_vp9_enc->arnr_maxframes);
1085         if (status != VPX_CODEC_OK) {
1086           GST_WARNING_OBJECT (gst_vp9_enc,
1087               "Failed to set VP8E_SET_ARNR_MAXFRAMES: %s",
1088               gst_vpx_error_name (status));
1089         }
1090       }
1091       break;
1092     case PROP_ARNR_STRENGTH:
1093       gst_vp9_enc->arnr_strength = g_value_get_int (value);
1094       if (gst_vp9_enc->inited) {
1095         status =
1096             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ARNR_STRENGTH,
1097             gst_vp9_enc->arnr_strength);
1098         if (status != VPX_CODEC_OK) {
1099           GST_WARNING_OBJECT (gst_vp9_enc,
1100               "Failed to set VP8E_SET_ARNR_STRENGTH: %s",
1101               gst_vpx_error_name (status));
1102         }
1103       }
1104       break;
1105     case PROP_ARNR_TYPE:
1106       gst_vp9_enc->arnr_type = g_value_get_int (value);
1107       if (gst_vp9_enc->inited) {
1108         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ARNR_TYPE,
1109             gst_vp9_enc->arnr_type);
1110         if (status != VPX_CODEC_OK) {
1111           GST_WARNING_OBJECT (gst_vp9_enc,
1112               "Failed to set VP8E_SET_ARNR_TYPE: %s",
1113               gst_vpx_error_name (status));
1114         }
1115       }
1116       break;
1117     case PROP_TUNING:
1118       gst_vp9_enc->tuning = g_value_get_enum (value);
1119       if (gst_vp9_enc->inited) {
1120         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_TUNING,
1121             gst_vp9_enc->tuning);
1122         if (status != VPX_CODEC_OK) {
1123           GST_WARNING_OBJECT (gst_vp9_enc,
1124               "Failed to set VP8E_SET_TUNING: %s", gst_vpx_error_name (status));
1125         }
1126       }
1127       break;
1128     case PROP_CQ_LEVEL:
1129       gst_vp9_enc->cq_level = g_value_get_int (value);
1130       if (gst_vp9_enc->inited) {
1131         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_CQ_LEVEL,
1132             gst_vp9_enc->cq_level);
1133         if (status != VPX_CODEC_OK) {
1134           GST_WARNING_OBJECT (gst_vp9_enc,
1135               "Failed to set VP8E_SET_CQ_LEVEL: %s",
1136               gst_vpx_error_name (status));
1137         }
1138       }
1139       break;
1140     case PROP_MAX_INTRA_BITRATE_PCT:
1141       gst_vp9_enc->max_intra_bitrate_pct = g_value_get_int (value);
1142       if (gst_vp9_enc->inited) {
1143         status =
1144             vpx_codec_control (&gst_vp9_enc->encoder,
1145             VP8E_SET_MAX_INTRA_BITRATE_PCT, gst_vp9_enc->max_intra_bitrate_pct);
1146         if (status != VPX_CODEC_OK) {
1147           GST_WARNING_OBJECT (gst_vp9_enc,
1148               "Failed to set VP8E_SET_MAX_INTRA_BITRATE_PCT: %s",
1149               gst_vpx_error_name (status));
1150         }
1151       }
1152       break;
1153     case PROP_TIMEBASE:
1154       gst_vp9_enc->timebase_n = gst_value_get_fraction_numerator (value);
1155       gst_vp9_enc->timebase_d = gst_value_get_fraction_denominator (value);
1156       break;
1157     default:
1158       break;
1159   }
1160
1161   if (global &&gst_vp9_enc->inited) {
1162     status =
1163         vpx_codec_enc_config_set (&gst_vp9_enc->encoder, &gst_vp9_enc->cfg);
1164     if (status != VPX_CODEC_OK) {
1165       g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1166       GST_ELEMENT_ERROR (gst_vp9_enc, LIBRARY, INIT,
1167           ("Failed to set encoder configuration"), ("%s",
1168               gst_vpx_error_name (status)));
1169     } else {
1170       g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1171     }
1172   } else {
1173     g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1174   }
1175 }
1176
1177 static void
1178 gst_vp9_enc_get_property (GObject * object, guint prop_id, GValue * value,
1179     GParamSpec * pspec)
1180 {
1181   GstVP9Enc *gst_vp9_enc;
1182
1183   g_return_if_fail (GST_IS_VP9_ENC (object));
1184   gst_vp9_enc = GST_VP9_ENC (object);
1185
1186   g_mutex_lock (&gst_vp9_enc->encoder_lock);
1187   switch (prop_id) {
1188     case PROP_RC_END_USAGE:
1189       g_value_set_enum (value, gst_vp9_enc->cfg.rc_end_usage);
1190       break;
1191     case PROP_RC_TARGET_BITRATE:
1192       g_value_set_int (value, gst_vp9_enc->cfg.rc_target_bitrate * 1000);
1193       break;
1194     case PROP_RC_MIN_QUANTIZER:
1195       g_value_set_int (value, gst_vp9_enc->cfg.rc_min_quantizer);
1196       break;
1197     case PROP_RC_MAX_QUANTIZER:
1198       g_value_set_int (value, gst_vp9_enc->cfg.rc_max_quantizer);
1199       break;
1200     case PROP_RC_DROPFRAME_THRESH:
1201       g_value_set_int (value, gst_vp9_enc->cfg.rc_dropframe_thresh);
1202       break;
1203     case PROP_RC_RESIZE_ALLOWED:
1204       g_value_set_boolean (value, gst_vp9_enc->cfg.rc_resize_allowed);
1205       break;
1206     case PROP_RC_RESIZE_UP_THRESH:
1207       g_value_set_int (value, gst_vp9_enc->cfg.rc_resize_up_thresh);
1208       break;
1209     case PROP_RC_RESIZE_DOWN_THRESH:
1210       g_value_set_int (value, gst_vp9_enc->cfg.rc_resize_down_thresh);
1211       break;
1212     case PROP_RC_UNDERSHOOT_PCT:
1213       g_value_set_int (value, gst_vp9_enc->cfg.rc_undershoot_pct);
1214       break;
1215     case PROP_RC_OVERSHOOT_PCT:
1216       g_value_set_int (value, gst_vp9_enc->cfg.rc_overshoot_pct);
1217       break;
1218     case PROP_RC_BUF_SZ:
1219       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_sz);
1220       break;
1221     case PROP_RC_BUF_INITIAL_SZ:
1222       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_initial_sz);
1223       break;
1224     case PROP_RC_BUF_OPTIMAL_SZ:
1225       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_optimal_sz);
1226       break;
1227     case PROP_RC_2PASS_VBR_BIAS_PCT:
1228       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct);
1229       break;
1230     case PROP_RC_2PASS_VBR_MINSECTION_PCT:
1231       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct);
1232       break;
1233     case PROP_RC_2PASS_VBR_MAXSECTION_PCT:
1234       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct);
1235       break;
1236     case PROP_KF_MODE:
1237       g_value_set_enum (value, gst_vp9_enc->cfg.kf_mode);
1238       break;
1239     case PROP_KF_MAX_DIST:
1240       g_value_set_int (value, gst_vp9_enc->cfg.kf_max_dist);
1241       break;
1242     case PROP_MULTIPASS_MODE:
1243       g_value_set_enum (value, gst_vp9_enc->cfg.g_pass);
1244       break;
1245     case PROP_MULTIPASS_CACHE_FILE:
1246       g_value_set_string (value, gst_vp9_enc->multipass_cache_file);
1247       break;
1248     case PROP_TS_NUMBER_LAYERS:
1249       g_value_set_int (value, gst_vp9_enc->cfg.ts_number_layers);
1250       break;
1251     case PROP_TS_TARGET_BITRATE:{
1252       GValueArray *va;
1253
1254       if (gst_vp9_enc->n_ts_target_bitrate == 0) {
1255         g_value_set_boxed (value, NULL);
1256       } else {
1257         gint i;
1258
1259         va = g_value_array_new (gst_vp9_enc->n_ts_target_bitrate);
1260         for (i = 0; i < gst_vp9_enc->n_ts_target_bitrate; i++) {
1261           GValue v = { 0, };
1262
1263           g_value_init (&v, G_TYPE_INT);
1264           g_value_set_int (&v, gst_vp9_enc->cfg.ts_target_bitrate[i]);
1265           g_value_array_append (va, &v);
1266           g_value_unset (&v);
1267         }
1268         g_value_set_boxed (value, va);
1269         g_value_array_free (va);
1270       }
1271       break;
1272     }
1273     case PROP_TS_RATE_DECIMATOR:{
1274       GValueArray *va;
1275
1276       if (gst_vp9_enc->n_ts_rate_decimator == 0) {
1277         g_value_set_boxed (value, NULL);
1278       } else {
1279         gint i;
1280
1281         va = g_value_array_new (gst_vp9_enc->n_ts_rate_decimator);
1282         for (i = 0; i < gst_vp9_enc->n_ts_rate_decimator; i++) {
1283           GValue v = { 0, };
1284
1285           g_value_init (&v, G_TYPE_INT);
1286           g_value_set_int (&v, gst_vp9_enc->cfg.ts_rate_decimator[i]);
1287           g_value_array_append (va, &v);
1288           g_value_unset (&v);
1289         }
1290         g_value_set_boxed (value, va);
1291         g_value_array_free (va);
1292       }
1293       break;
1294     }
1295     case PROP_TS_PERIODICITY:
1296       g_value_set_int (value, gst_vp9_enc->cfg.ts_periodicity);
1297       break;
1298     case PROP_TS_LAYER_ID:{
1299       GValueArray *va;
1300
1301       if (gst_vp9_enc->n_ts_layer_id == 0) {
1302         g_value_set_boxed (value, NULL);
1303       } else {
1304         gint i;
1305
1306         va = g_value_array_new (gst_vp9_enc->n_ts_layer_id);
1307         for (i = 0; i < gst_vp9_enc->n_ts_layer_id; i++) {
1308           GValue v = { 0, };
1309
1310           g_value_init (&v, G_TYPE_INT);
1311           g_value_set_int (&v, gst_vp9_enc->cfg.ts_layer_id[i]);
1312           g_value_array_append (va, &v);
1313           g_value_unset (&v);
1314         }
1315         g_value_set_boxed (value, va);
1316         g_value_array_free (va);
1317       }
1318       break;
1319     }
1320     case PROP_ERROR_RESILIENT:
1321       g_value_set_flags (value, gst_vp9_enc->cfg.g_error_resilient);
1322       break;
1323     case PROP_LAG_IN_FRAMES:
1324       g_value_set_int (value, gst_vp9_enc->cfg.g_lag_in_frames);
1325       break;
1326     case PROP_THREADS:
1327       g_value_set_int (value, gst_vp9_enc->cfg.g_threads);
1328       break;
1329     case PROP_DEADLINE:
1330       g_value_set_int64 (value, gst_vp9_enc->deadline);
1331       break;
1332     case PROP_H_SCALING_MODE:
1333       g_value_set_enum (value, gst_vp9_enc->h_scaling_mode);
1334       break;
1335     case PROP_V_SCALING_MODE:
1336       g_value_set_enum (value, gst_vp9_enc->v_scaling_mode);
1337       break;
1338     case PROP_CPU_USED:
1339       g_value_set_int (value, gst_vp9_enc->cpu_used);
1340       break;
1341     case PROP_ENABLE_AUTO_ALT_REF:
1342       g_value_set_boolean (value, gst_vp9_enc->enable_auto_alt_ref);
1343       break;
1344     case PROP_NOISE_SENSITIVITY:
1345       g_value_set_int (value, gst_vp9_enc->noise_sensitivity);
1346       break;
1347     case PROP_SHARPNESS:
1348       g_value_set_int (value, gst_vp9_enc->sharpness);
1349       break;
1350     case PROP_STATIC_THRESHOLD:
1351       g_value_set_int (value, gst_vp9_enc->static_threshold);
1352       break;
1353     case PROP_TOKEN_PARTITIONS:
1354       g_value_set_enum (value, gst_vp9_enc->token_partitions);
1355       break;
1356     case PROP_ARNR_MAXFRAMES:
1357       g_value_set_int (value, gst_vp9_enc->arnr_maxframes);
1358       break;
1359     case PROP_ARNR_STRENGTH:
1360       g_value_set_int (value, gst_vp9_enc->arnr_strength);
1361       break;
1362     case PROP_ARNR_TYPE:
1363       g_value_set_int (value, gst_vp9_enc->arnr_type);
1364       break;
1365     case PROP_TUNING:
1366       g_value_set_enum (value, gst_vp9_enc->tuning);
1367       break;
1368     case PROP_CQ_LEVEL:
1369       g_value_set_int (value, gst_vp9_enc->cq_level);
1370       break;
1371     case PROP_MAX_INTRA_BITRATE_PCT:
1372       g_value_set_int (value, gst_vp9_enc->max_intra_bitrate_pct);
1373       break;
1374     case PROP_TIMEBASE:
1375       gst_value_set_fraction (value, gst_vp9_enc->timebase_n,
1376           gst_vp9_enc->timebase_d);
1377       break;
1378     default:
1379       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1380       break;
1381   }
1382
1383   g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1384 }
1385
1386 static gboolean
1387 gst_vp9_enc_start (GstVideoEncoder * video_encoder)
1388 {
1389   GstVP9Enc *encoder = GST_VP9_ENC (video_encoder);
1390
1391   GST_DEBUG_OBJECT (video_encoder, "start");
1392
1393   if (!encoder->have_default_config) {
1394     GST_ELEMENT_ERROR (encoder, LIBRARY, INIT,
1395         ("Failed to get default encoder configuration"), (NULL));
1396     return FALSE;
1397   }
1398
1399   return TRUE;
1400 }
1401
1402 static gboolean
1403 gst_vp9_enc_stop (GstVideoEncoder * video_encoder)
1404 {
1405   GstVP9Enc *encoder;
1406
1407   GST_DEBUG_OBJECT (video_encoder, "stop");
1408
1409   encoder = GST_VP9_ENC (video_encoder);
1410
1411   g_mutex_lock (&encoder->encoder_lock);
1412   if (encoder->inited) {
1413     vpx_codec_destroy (&encoder->encoder);
1414     encoder->inited = FALSE;
1415   }
1416
1417   if (encoder->first_pass_cache_content) {
1418     g_byte_array_free (encoder->first_pass_cache_content, TRUE);
1419     encoder->first_pass_cache_content = NULL;
1420   }
1421
1422   if (encoder->cfg.rc_twopass_stats_in.buf) {
1423     g_free (encoder->cfg.rc_twopass_stats_in.buf);
1424     encoder->cfg.rc_twopass_stats_in.buf = NULL;
1425     encoder->cfg.rc_twopass_stats_in.sz = 0;
1426   }
1427   g_mutex_unlock (&encoder->encoder_lock);
1428
1429   gst_tag_setter_reset_tags (GST_TAG_SETTER (encoder));
1430
1431   return TRUE;
1432 }
1433
1434 static gint
1435 gst_vp9_enc_get_downstream_profile (GstVP9Enc * encoder)
1436 {
1437   GstCaps *allowed;
1438   GstStructure *s;
1439   gint profile = DEFAULT_PROFILE;
1440
1441   allowed = gst_pad_get_allowed_caps (GST_VIDEO_ENCODER_SRC_PAD (encoder));
1442   if (allowed) {
1443     allowed = gst_caps_truncate (allowed);
1444     s = gst_caps_get_structure (allowed, 0);
1445     if (gst_structure_has_field (s, "profile")) {
1446       const GValue *v = gst_structure_get_value (s, "profile");
1447       const gchar *profile_str = NULL;
1448
1449       if (GST_VALUE_HOLDS_LIST (v) && gst_value_list_get_size (v) > 0) {
1450         profile_str = g_value_get_string (gst_value_list_get_value (v, 0));
1451       } else if (G_VALUE_HOLDS_STRING (v)) {
1452         profile_str = g_value_get_string (v);
1453       }
1454
1455       if (profile_str) {
1456         gchar *endptr = NULL;
1457
1458         profile = g_ascii_strtoull (profile_str, &endptr, 10);
1459         if (*endptr != '\0' || profile < 0 || profile > 3) {
1460           GST_ERROR_OBJECT (encoder, "Invalid profile '%s'", profile_str);
1461           profile = DEFAULT_PROFILE;
1462         }
1463       }
1464     }
1465     gst_caps_unref (allowed);
1466   }
1467
1468   GST_DEBUG_OBJECT (encoder, "Using profile %d", profile);
1469
1470   return profile;
1471 }
1472
1473 static gboolean
1474 gst_vp9_enc_set_format (GstVideoEncoder * video_encoder,
1475     GstVideoCodecState * state)
1476 {
1477   GstVP9Enc *encoder;
1478   vpx_codec_err_t status;
1479   vpx_image_t *image;
1480   GstCaps *caps;
1481   gboolean ret = TRUE;
1482   GstVideoInfo *info = &state->info;
1483   GstVideoCodecState *output_state;
1484   gchar *profile_str;
1485
1486   encoder = GST_VP9_ENC (video_encoder);
1487   GST_DEBUG_OBJECT (video_encoder, "set_format");
1488
1489   if (encoder->inited) {
1490     gst_vp9_enc_drain (video_encoder);
1491     g_mutex_lock (&encoder->encoder_lock);
1492     vpx_codec_destroy (&encoder->encoder);
1493     encoder->inited = FALSE;
1494   } else {
1495     g_mutex_lock (&encoder->encoder_lock);
1496   }
1497
1498   encoder->cfg.g_profile = gst_vp9_enc_get_downstream_profile (encoder);
1499
1500   /* Scale default bitrate to our size */
1501   if (!encoder->rc_target_bitrate_set)
1502     encoder->cfg.rc_target_bitrate =
1503         gst_util_uint64_scale (DEFAULT_RC_TARGET_BITRATE,
1504         GST_VIDEO_INFO_WIDTH (info) * GST_VIDEO_INFO_HEIGHT (info),
1505         320 * 240 * 1000);
1506
1507   encoder->cfg.g_w = GST_VIDEO_INFO_WIDTH (info);
1508   encoder->cfg.g_h = GST_VIDEO_INFO_HEIGHT (info);
1509
1510   if (encoder->timebase_n != 0 && encoder->timebase_d != 0) {
1511     GST_DEBUG_OBJECT (video_encoder, "Using timebase configuration");
1512     encoder->cfg.g_timebase.num = encoder->timebase_n;
1513     encoder->cfg.g_timebase.den = encoder->timebase_d;
1514   } else if (GST_VIDEO_INFO_FPS_D (info) != 0
1515       && GST_VIDEO_INFO_FPS_N (info) != 0) {
1516     /* GstVideoInfo holds either the framerate or max-framerate (if framerate
1517      * is 0) in FPS so this will be used if max-framerate or framerate
1518      * is set */
1519     GST_DEBUG_OBJECT (video_encoder, "Setting timebase from framerate");
1520     encoder->cfg.g_timebase.num = GST_VIDEO_INFO_FPS_D (info);
1521     encoder->cfg.g_timebase.den = GST_VIDEO_INFO_FPS_N (info);
1522   } else {
1523     /* Zero framerate and max-framerate but still need to setup the timebase to avoid
1524      * a divide by zero error. Presuming the lowest common denominator will be RTP -
1525      * VP9 payload draft states clock rate of 90000 which should work for anyone where
1526      * FPS < 90000 (shouldn't be too many cases where it's higher) though wouldn't be optimal. RTP specification
1527      * http://tools.ietf.org/html/draft-ietf-payload-vp9-01 section 6.3.1 */
1528     GST_WARNING_OBJECT (encoder,
1529         "No timebase and zero framerate setting timebase to 1/90000");
1530     encoder->cfg.g_timebase.num = 1;
1531     encoder->cfg.g_timebase.den = 90000;
1532   }
1533
1534   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1535     if (encoder->first_pass_cache_content == NULL) {
1536       encoder->first_pass_cache_content = g_byte_array_sized_new (4096);
1537     }
1538   } else if (encoder->cfg.g_pass == VPX_RC_LAST_PASS) {
1539     GError *err = NULL;
1540
1541     if (!encoder->multipass_cache_file) {
1542       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1543           ("No multipass cache file provided"), (NULL));
1544       g_mutex_unlock (&encoder->encoder_lock);
1545       return FALSE;
1546     }
1547
1548     if (encoder->cfg.rc_twopass_stats_in.buf != NULL) {
1549       g_free (encoder->cfg.rc_twopass_stats_in.buf);
1550       encoder->cfg.rc_twopass_stats_in.buf = NULL;
1551       encoder->cfg.rc_twopass_stats_in.sz = 0;
1552     }
1553
1554     if (!g_file_get_contents (encoder->multipass_cache_file,
1555             (gchar **) & encoder->cfg.rc_twopass_stats_in.buf,
1556             &encoder->cfg.rc_twopass_stats_in.sz, &err)) {
1557       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1558           ("Failed to read multipass cache file provided"), ("%s",
1559               err->message));
1560       g_error_free (err);
1561       g_mutex_unlock (&encoder->encoder_lock);
1562       return FALSE;
1563     }
1564   }
1565
1566   status = vpx_codec_enc_init (&encoder->encoder, &vpx_codec_vp9_cx_algo,
1567       &encoder->cfg, 0);
1568   if (status != VPX_CODEC_OK) {
1569     GST_ELEMENT_ERROR (encoder, LIBRARY, INIT,
1570         ("Failed to initialize encoder"), ("%s", gst_vpx_error_name (status)));
1571     g_mutex_unlock (&encoder->encoder_lock);
1572     return FALSE;
1573   }
1574
1575   /* FIXME: Disabled for now, does not work with VP9 */
1576 #if 0
1577   {
1578     vpx_scaling_mode_t sm;
1579
1580     sm.h_scaling_mode = encoder->h_scaling_mode;
1581     sm.v_scaling_mode = encoder->v_scaling_mode;
1582
1583     status = vpx_codec_control (&encoder->encoder, VP8E_SET_SCALEMODE, &sm);
1584     if (status != VPX_CODEC_OK) {
1585       GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_SCALEMODE: %s",
1586           gst_vpx_error_name (status));
1587     }
1588   }
1589 #endif
1590
1591   status =
1592       vpx_codec_control (&encoder->encoder, VP8E_SET_CPUUSED,
1593       encoder->cpu_used);
1594   if (status != VPX_CODEC_OK) {
1595     GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_CPUUSED: %s",
1596         gst_vpx_error_name (status));
1597   }
1598
1599   status =
1600       vpx_codec_control (&encoder->encoder, VP8E_SET_ENABLEAUTOALTREF,
1601       (encoder->enable_auto_alt_ref ? 1 : 0));
1602   if (status != VPX_CODEC_OK) {
1603     GST_WARNING_OBJECT (encoder,
1604         "Failed to set VP8E_SET_ENABLEAUTOALTREF: %s",
1605         gst_vpx_error_name (status));
1606   }
1607   status = vpx_codec_control (&encoder->encoder, VP8E_SET_NOISE_SENSITIVITY,
1608       encoder->noise_sensitivity);
1609   if (status != VPX_CODEC_OK) {
1610     GST_WARNING_OBJECT (encoder,
1611         "Failed to set VP8E_SET_NOISE_SENSITIVITY: %s",
1612         gst_vpx_error_name (status));
1613   }
1614   status = vpx_codec_control (&encoder->encoder, VP8E_SET_SHARPNESS,
1615       encoder->sharpness);
1616   if (status != VPX_CODEC_OK) {
1617     GST_WARNING_OBJECT (encoder,
1618         "Failed to set VP8E_SET_SHARPNESS: %s", gst_vpx_error_name (status));
1619   }
1620   status = vpx_codec_control (&encoder->encoder, VP8E_SET_STATIC_THRESHOLD,
1621       encoder->static_threshold);
1622   if (status != VPX_CODEC_OK) {
1623     GST_WARNING_OBJECT (encoder,
1624         "Failed to set VP8E_SET_STATIC_THRESHOLD: %s",
1625         gst_vpx_error_name (status));
1626   }
1627   status = vpx_codec_control (&encoder->encoder, VP8E_SET_TOKEN_PARTITIONS,
1628       encoder->token_partitions);
1629   if (status != VPX_CODEC_OK) {
1630     GST_WARNING_OBJECT (encoder,
1631         "Failed to set VP8E_SET_TOKEN_PARTIONS: %s",
1632         gst_vpx_error_name (status));
1633   }
1634   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_MAXFRAMES,
1635       encoder->arnr_maxframes);
1636   if (status != VPX_CODEC_OK) {
1637     GST_WARNING_OBJECT (encoder,
1638         "Failed to set VP8E_SET_ARNR_MAXFRAMES: %s",
1639         gst_vpx_error_name (status));
1640   }
1641   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_STRENGTH,
1642       encoder->arnr_strength);
1643   if (status != VPX_CODEC_OK) {
1644     GST_WARNING_OBJECT (encoder,
1645         "Failed to set VP8E_SET_ARNR_STRENGTH: %s",
1646         gst_vpx_error_name (status));
1647   }
1648   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_TYPE,
1649       encoder->arnr_type);
1650   if (status != VPX_CODEC_OK) {
1651     GST_WARNING_OBJECT (encoder,
1652         "Failed to set VP8E_SET_ARNR_TYPE: %s", gst_vpx_error_name (status));
1653   }
1654   status = vpx_codec_control (&encoder->encoder, VP8E_SET_TUNING,
1655       encoder->tuning);
1656   if (status != VPX_CODEC_OK) {
1657     GST_WARNING_OBJECT (encoder,
1658         "Failed to set VP8E_SET_TUNING: %s", gst_vpx_error_name (status));
1659   }
1660   status = vpx_codec_control (&encoder->encoder, VP8E_SET_CQ_LEVEL,
1661       encoder->cq_level);
1662   if (status != VPX_CODEC_OK) {
1663     GST_WARNING_OBJECT (encoder,
1664         "Failed to set VP8E_SET_CQ_LEVEL: %s", gst_vpx_error_name (status));
1665   }
1666   status = vpx_codec_control (&encoder->encoder, VP8E_SET_MAX_INTRA_BITRATE_PCT,
1667       encoder->max_intra_bitrate_pct);
1668   if (status != VPX_CODEC_OK) {
1669     GST_WARNING_OBJECT (encoder,
1670         "Failed to set VP8E_SET_MAX_INTRA_BITRATE_PCT: %s",
1671         gst_vpx_error_name (status));
1672   }
1673
1674   if (GST_VIDEO_INFO_FPS_D (info) == 0 || GST_VIDEO_INFO_FPS_N (info) == 0) {
1675     gst_video_encoder_set_latency (video_encoder, GST_CLOCK_TIME_NONE,
1676         GST_CLOCK_TIME_NONE);
1677   } else {
1678     gst_video_encoder_set_latency (video_encoder, 0,
1679         gst_util_uint64_scale (encoder->cfg.g_lag_in_frames,
1680             GST_VIDEO_INFO_FPS_D (info) * GST_SECOND,
1681             GST_VIDEO_INFO_FPS_N (info)));
1682   }
1683   encoder->inited = TRUE;
1684
1685   /* Store input state */
1686   if (encoder->input_state)
1687     gst_video_codec_state_unref (encoder->input_state);
1688   encoder->input_state = gst_video_codec_state_ref (state);
1689
1690   /* prepare cached image buffer setup */
1691   image = &encoder->image;
1692   memset (image, 0, sizeof (*image));
1693
1694   switch (encoder->input_state->info.finfo->format) {
1695     case GST_VIDEO_FORMAT_I420:
1696       image->fmt = VPX_IMG_FMT_I420;
1697       image->bps = 12;
1698       image->x_chroma_shift = image->y_chroma_shift = 1;
1699       break;
1700     case GST_VIDEO_FORMAT_YV12:
1701       image->fmt = VPX_IMG_FMT_YV12;
1702       image->bps = 12;
1703       image->x_chroma_shift = image->y_chroma_shift = 1;
1704       break;
1705     case GST_VIDEO_FORMAT_Y42B:
1706       image->fmt = VPX_IMG_FMT_I422;
1707       image->bps = 16;
1708       image->x_chroma_shift = 1;
1709       image->y_chroma_shift = 0;
1710       break;
1711     case GST_VIDEO_FORMAT_Y444:
1712       image->fmt = VPX_IMG_FMT_I444;
1713       image->bps = 24;
1714       image->x_chroma_shift = image->y_chroma_shift = 0;
1715       break;
1716     default:
1717       g_assert_not_reached ();
1718       break;
1719   }
1720   image->w = image->d_w = GST_VIDEO_INFO_WIDTH (info);
1721   image->h = image->d_h = GST_VIDEO_INFO_HEIGHT (info);
1722
1723   image->stride[VPX_PLANE_Y] = GST_VIDEO_INFO_COMP_STRIDE (info, 0);
1724   image->stride[VPX_PLANE_U] = GST_VIDEO_INFO_COMP_STRIDE (info, 1);
1725   image->stride[VPX_PLANE_V] = GST_VIDEO_INFO_COMP_STRIDE (info, 2);
1726
1727   profile_str = g_strdup_printf ("%d", encoder->cfg.g_profile);
1728   caps = gst_caps_new_simple ("video/x-vp9",
1729       "profile", G_TYPE_STRING, profile_str, NULL);
1730   g_free (profile_str);
1731
1732   g_mutex_unlock (&encoder->encoder_lock);
1733
1734   output_state =
1735       gst_video_encoder_set_output_state (video_encoder, caps, state);
1736   gst_video_codec_state_unref (output_state);
1737
1738   gst_video_encoder_negotiate (GST_VIDEO_ENCODER (encoder));
1739
1740   return ret;
1741 }
1742
1743 static GstFlowReturn
1744 gst_vp9_enc_process (GstVP9Enc * encoder)
1745 {
1746   vpx_codec_iter_t iter = NULL;
1747   const vpx_codec_cx_pkt_t *pkt;
1748   GstVideoEncoder *video_encoder;
1749   GstVideoCodecFrame *frame;
1750   GstFlowReturn ret = GST_FLOW_OK;
1751
1752   video_encoder = GST_VIDEO_ENCODER (encoder);
1753
1754   g_mutex_lock (&encoder->encoder_lock);
1755   pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1756   while (pkt != NULL) {
1757     GstBuffer *buffer;
1758     gboolean invisible;
1759
1760     GST_DEBUG_OBJECT (encoder, "packet %u type %d", (guint) pkt->data.frame.sz,
1761         pkt->kind);
1762
1763     if (pkt->kind == VPX_CODEC_STATS_PKT
1764         && encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1765       GST_LOG_OBJECT (encoder, "handling STATS packet");
1766
1767       g_byte_array_append (encoder->first_pass_cache_content,
1768           pkt->data.twopass_stats.buf, pkt->data.twopass_stats.sz);
1769
1770       frame = gst_video_encoder_get_oldest_frame (video_encoder);
1771       if (frame != NULL) {
1772         buffer = gst_buffer_new ();
1773         GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_LIVE);
1774         frame->output_buffer = buffer;
1775         g_mutex_unlock (&encoder->encoder_lock);
1776         ret = gst_video_encoder_finish_frame (video_encoder, frame);
1777         g_mutex_lock (&encoder->encoder_lock);
1778       }
1779
1780       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1781       continue;
1782     } else if (pkt->kind != VPX_CODEC_CX_FRAME_PKT) {
1783       GST_LOG_OBJECT (encoder, "non frame pkt: %d", pkt->kind);
1784       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1785       continue;
1786     }
1787
1788     invisible = (pkt->data.frame.flags & VPX_FRAME_IS_INVISIBLE) != 0;
1789     frame = gst_video_encoder_get_oldest_frame (video_encoder);
1790     g_assert (frame != NULL);
1791     if ((pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0)
1792       GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
1793     else
1794       GST_VIDEO_CODEC_FRAME_UNSET_SYNC_POINT (frame);
1795
1796     /* FIXME : It would be nice to avoid the memory copy ... */
1797     buffer =
1798         gst_buffer_new_wrapped (g_memdup (pkt->data.frame.buf,
1799             pkt->data.frame.sz), pkt->data.frame.sz);
1800
1801     if (invisible) {
1802       g_mutex_unlock (&encoder->encoder_lock);
1803       ret = gst_pad_push (GST_VIDEO_ENCODER_SRC_PAD (encoder), buffer);
1804       g_mutex_lock (&encoder->encoder_lock);
1805       gst_video_codec_frame_unref (frame);
1806     } else {
1807       frame->output_buffer = buffer;
1808       g_mutex_unlock (&encoder->encoder_lock);
1809       ret = gst_video_encoder_finish_frame (video_encoder, frame);
1810       g_mutex_lock (&encoder->encoder_lock);
1811     }
1812
1813     pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1814   }
1815   g_mutex_unlock (&encoder->encoder_lock);
1816
1817   return ret;
1818 }
1819
1820 /* This function should be called holding then stream lock*/
1821 static GstFlowReturn
1822 gst_vp9_enc_drain (GstVideoEncoder * video_encoder)
1823 {
1824   GstVP9Enc *encoder;
1825   int flags = 0;
1826   vpx_codec_err_t status;
1827   gint64 deadline;
1828
1829   encoder = GST_VP9_ENC (video_encoder);
1830
1831   g_mutex_lock (&encoder->encoder_lock);
1832   deadline = encoder->deadline;
1833
1834   status =
1835       vpx_codec_encode (&encoder->encoder, NULL, encoder->n_frames, 1, flags,
1836       deadline);
1837   g_mutex_unlock (&encoder->encoder_lock);
1838
1839   if (status != 0) {
1840     GST_ERROR_OBJECT (encoder, "encode returned %d %s", status,
1841         gst_vpx_error_name (status));
1842     return GST_FLOW_ERROR;
1843   }
1844
1845   /* dispatch remaining frames */
1846   gst_vp9_enc_process (encoder);
1847
1848   g_mutex_lock (&encoder->encoder_lock);
1849   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS && encoder->multipass_cache_file) {
1850     GError *err = NULL;
1851
1852     if (!g_file_set_contents (encoder->multipass_cache_file,
1853             (const gchar *) encoder->first_pass_cache_content->data,
1854             encoder->first_pass_cache_content->len, &err)) {
1855       GST_ELEMENT_ERROR (encoder, RESOURCE, WRITE, (NULL),
1856           ("Failed to write multipass cache file: %s", err->message));
1857       g_error_free (err);
1858     }
1859   }
1860   g_mutex_unlock (&encoder->encoder_lock);
1861
1862   return GST_FLOW_OK;
1863 }
1864
1865 static GstFlowReturn
1866 gst_vp9_enc_finish (GstVideoEncoder * video_encoder)
1867 {
1868   GstFlowReturn ret;
1869
1870   GST_DEBUG_OBJECT (video_encoder, "finish");
1871
1872   ret = gst_vp9_enc_drain (video_encoder);
1873
1874   return ret;
1875 }
1876
1877 static vpx_image_t *
1878 gst_vp9_enc_buffer_to_image (GstVP9Enc * enc, GstVideoFrame * frame)
1879 {
1880   vpx_image_t *image = g_slice_new (vpx_image_t);
1881
1882   memcpy (image, &enc->image, sizeof (*image));
1883
1884   image->planes[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_DATA (frame, 0);
1885   image->planes[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_DATA (frame, 1);
1886   image->planes[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_DATA (frame, 2);
1887
1888   image->stride[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 0);
1889   image->stride[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 1);
1890   image->stride[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 2);
1891
1892   return image;
1893 }
1894
1895 static GstFlowReturn
1896 gst_vp9_enc_handle_frame (GstVideoEncoder * video_encoder,
1897     GstVideoCodecFrame * frame)
1898 {
1899   GstVP9Enc *encoder;
1900   vpx_codec_err_t status;
1901   int flags = 0;
1902   vpx_image_t *image;
1903   GstVideoFrame vframe;
1904
1905   GST_DEBUG_OBJECT (video_encoder, "handle_frame");
1906
1907   encoder = GST_VP9_ENC (video_encoder);
1908
1909   encoder->n_frames++;
1910
1911   GST_DEBUG_OBJECT (video_encoder, "size %d %d",
1912       GST_VIDEO_INFO_WIDTH (&encoder->input_state->info),
1913       GST_VIDEO_INFO_HEIGHT (&encoder->input_state->info));
1914
1915   gst_video_frame_map (&vframe, &encoder->input_state->info,
1916       frame->input_buffer, GST_MAP_READ);
1917   image = gst_vp9_enc_buffer_to_image (encoder, &vframe);
1918
1919   if (GST_VIDEO_CODEC_FRAME_IS_FORCE_KEYFRAME (frame)) {
1920     flags |= VPX_EFLAG_FORCE_KF;
1921   }
1922
1923   g_mutex_lock (&encoder->encoder_lock);
1924   status = vpx_codec_encode (&encoder->encoder, image,
1925       encoder->n_frames, 1, flags, encoder->deadline);
1926   g_mutex_unlock (&encoder->encoder_lock);
1927   gst_video_frame_unmap (&vframe);
1928
1929   if (status != 0) {
1930     GST_ELEMENT_ERROR (encoder, LIBRARY, ENCODE,
1931         ("Failed to encode frame"), ("%s", gst_vpx_error_name (status)));
1932     return FALSE;
1933   }
1934   gst_video_codec_frame_unref (frame);
1935   return gst_vp9_enc_process (encoder);
1936 }
1937
1938 static gboolean
1939 gst_vp9_enc_sink_event (GstVideoEncoder * benc, GstEvent * event)
1940 {
1941   GstVP9Enc *enc = GST_VP9_ENC (benc);
1942
1943   /* FIXME : Move this to base encoder class */
1944
1945   if (GST_EVENT_TYPE (event) == GST_EVENT_TAG) {
1946     GstTagList *list;
1947     GstTagSetter *setter = GST_TAG_SETTER (enc);
1948     const GstTagMergeMode mode = gst_tag_setter_get_tag_merge_mode (setter);
1949
1950     gst_event_parse_tag (event, &list);
1951     gst_tag_setter_merge_tags (setter, list, mode);
1952   }
1953
1954   /* just peeked, baseclass handles the rest */
1955   return GST_VIDEO_ENCODER_CLASS (parent_class)->sink_event (benc, event);
1956 }
1957
1958 static gboolean
1959 gst_vp9_enc_propose_allocation (GstVideoEncoder * encoder, GstQuery * query)
1960 {
1961   gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
1962
1963   return GST_VIDEO_ENCODER_CLASS (parent_class)->propose_allocation (encoder,
1964       query);
1965 }
1966
1967 #endif /* HAVE_VP9_ENCODER */