vp9enc: Allow caps renegotiation
[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     vpx_codec_destroy (&encoder->encoder);
1492   }
1493
1494   g_mutex_lock (&encoder->encoder_lock);
1495   encoder->cfg.g_profile = gst_vp9_enc_get_downstream_profile (encoder);
1496
1497   /* Scale default bitrate to our size */
1498   if (!encoder->rc_target_bitrate_set)
1499     encoder->cfg.rc_target_bitrate =
1500         gst_util_uint64_scale (DEFAULT_RC_TARGET_BITRATE,
1501         GST_VIDEO_INFO_WIDTH (info) * GST_VIDEO_INFO_HEIGHT (info),
1502         320 * 240 * 1000);
1503
1504   encoder->cfg.g_w = GST_VIDEO_INFO_WIDTH (info);
1505   encoder->cfg.g_h = GST_VIDEO_INFO_HEIGHT (info);
1506
1507   if (encoder->timebase_n != 0 && encoder->timebase_d != 0) {
1508     GST_DEBUG_OBJECT (video_encoder, "Using timebase configuration");
1509     encoder->cfg.g_timebase.num = encoder->timebase_n;
1510     encoder->cfg.g_timebase.den = encoder->timebase_d;
1511   } else if (GST_VIDEO_INFO_FPS_D (info) != 0
1512       && GST_VIDEO_INFO_FPS_N (info) != 0) {
1513     /* GstVideoInfo holds either the framerate or max-framerate (if framerate
1514      * is 0) in FPS so this will be used if max-framerate or framerate
1515      * is set */
1516     GST_DEBUG_OBJECT (video_encoder, "Setting timebase from framerate");
1517     encoder->cfg.g_timebase.num = GST_VIDEO_INFO_FPS_D (info);
1518     encoder->cfg.g_timebase.den = GST_VIDEO_INFO_FPS_N (info);
1519   } else {
1520     /* Zero framerate and max-framerate but still need to setup the timebase to avoid
1521      * a divide by zero error. Presuming the lowest common denominator will be RTP -
1522      * VP9 payload draft states clock rate of 90000 which should work for anyone where
1523      * FPS < 90000 (shouldn't be too many cases where it's higher) though wouldn't be optimal. RTP specification
1524      * http://tools.ietf.org/html/draft-ietf-payload-vp9-01 section 6.3.1 */
1525     GST_WARNING_OBJECT (encoder,
1526         "No timebase and zero framerate setting timebase to 1/90000");
1527     encoder->cfg.g_timebase.num = 1;
1528     encoder->cfg.g_timebase.den = 90000;
1529   }
1530
1531   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1532     if (encoder->first_pass_cache_content == NULL) {
1533       encoder->first_pass_cache_content = g_byte_array_sized_new (4096);
1534     }
1535   } else if (encoder->cfg.g_pass == VPX_RC_LAST_PASS) {
1536     GError *err = NULL;
1537
1538     if (!encoder->multipass_cache_file) {
1539       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1540           ("No multipass cache file provided"), (NULL));
1541       g_mutex_unlock (&encoder->encoder_lock);
1542       return FALSE;
1543     }
1544
1545     if (encoder->cfg.rc_twopass_stats_in.buf != NULL) {
1546       g_free (encoder->cfg.rc_twopass_stats_in.buf);
1547       encoder->cfg.rc_twopass_stats_in.buf = NULL;
1548       encoder->cfg.rc_twopass_stats_in.sz = 0;
1549     }
1550
1551     if (!g_file_get_contents (encoder->multipass_cache_file,
1552             (gchar **) & encoder->cfg.rc_twopass_stats_in.buf,
1553             &encoder->cfg.rc_twopass_stats_in.sz, &err)) {
1554       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1555           ("Failed to read multipass cache file provided"), ("%s",
1556               err->message));
1557       g_error_free (err);
1558       g_mutex_unlock (&encoder->encoder_lock);
1559       return FALSE;
1560     }
1561   }
1562
1563   status = vpx_codec_enc_init (&encoder->encoder, &vpx_codec_vp9_cx_algo,
1564       &encoder->cfg, 0);
1565   if (status != VPX_CODEC_OK) {
1566     GST_ELEMENT_ERROR (encoder, LIBRARY, INIT,
1567         ("Failed to initialize encoder"), ("%s", gst_vpx_error_name (status)));
1568     g_mutex_unlock (&encoder->encoder_lock);
1569     return FALSE;
1570   }
1571
1572   /* FIXME: Disabled for now, does not work with VP9 */
1573 #if 0
1574   {
1575     vpx_scaling_mode_t sm;
1576
1577     sm.h_scaling_mode = encoder->h_scaling_mode;
1578     sm.v_scaling_mode = encoder->v_scaling_mode;
1579
1580     status = vpx_codec_control (&encoder->encoder, VP8E_SET_SCALEMODE, &sm);
1581     if (status != VPX_CODEC_OK) {
1582       GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_SCALEMODE: %s",
1583           gst_vpx_error_name (status));
1584     }
1585   }
1586 #endif
1587
1588   status =
1589       vpx_codec_control (&encoder->encoder, VP8E_SET_CPUUSED,
1590       encoder->cpu_used);
1591   if (status != VPX_CODEC_OK) {
1592     GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_CPUUSED: %s",
1593         gst_vpx_error_name (status));
1594   }
1595
1596   status =
1597       vpx_codec_control (&encoder->encoder, VP8E_SET_ENABLEAUTOALTREF,
1598       (encoder->enable_auto_alt_ref ? 1 : 0));
1599   if (status != VPX_CODEC_OK) {
1600     GST_WARNING_OBJECT (encoder,
1601         "Failed to set VP8E_SET_ENABLEAUTOALTREF: %s",
1602         gst_vpx_error_name (status));
1603   }
1604   status = vpx_codec_control (&encoder->encoder, VP8E_SET_NOISE_SENSITIVITY,
1605       encoder->noise_sensitivity);
1606   if (status != VPX_CODEC_OK) {
1607     GST_WARNING_OBJECT (encoder,
1608         "Failed to set VP8E_SET_NOISE_SENSITIVITY: %s",
1609         gst_vpx_error_name (status));
1610   }
1611   status = vpx_codec_control (&encoder->encoder, VP8E_SET_SHARPNESS,
1612       encoder->sharpness);
1613   if (status != VPX_CODEC_OK) {
1614     GST_WARNING_OBJECT (encoder,
1615         "Failed to set VP8E_SET_SHARPNESS: %s", gst_vpx_error_name (status));
1616   }
1617   status = vpx_codec_control (&encoder->encoder, VP8E_SET_STATIC_THRESHOLD,
1618       encoder->static_threshold);
1619   if (status != VPX_CODEC_OK) {
1620     GST_WARNING_OBJECT (encoder,
1621         "Failed to set VP8E_SET_STATIC_THRESHOLD: %s",
1622         gst_vpx_error_name (status));
1623   }
1624   status = vpx_codec_control (&encoder->encoder, VP8E_SET_TOKEN_PARTITIONS,
1625       encoder->token_partitions);
1626   if (status != VPX_CODEC_OK) {
1627     GST_WARNING_OBJECT (encoder,
1628         "Failed to set VP8E_SET_TOKEN_PARTIONS: %s",
1629         gst_vpx_error_name (status));
1630   }
1631   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_MAXFRAMES,
1632       encoder->arnr_maxframes);
1633   if (status != VPX_CODEC_OK) {
1634     GST_WARNING_OBJECT (encoder,
1635         "Failed to set VP8E_SET_ARNR_MAXFRAMES: %s",
1636         gst_vpx_error_name (status));
1637   }
1638   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_STRENGTH,
1639       encoder->arnr_strength);
1640   if (status != VPX_CODEC_OK) {
1641     GST_WARNING_OBJECT (encoder,
1642         "Failed to set VP8E_SET_ARNR_STRENGTH: %s",
1643         gst_vpx_error_name (status));
1644   }
1645   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_TYPE,
1646       encoder->arnr_type);
1647   if (status != VPX_CODEC_OK) {
1648     GST_WARNING_OBJECT (encoder,
1649         "Failed to set VP8E_SET_ARNR_TYPE: %s", gst_vpx_error_name (status));
1650   }
1651   status = vpx_codec_control (&encoder->encoder, VP8E_SET_TUNING,
1652       encoder->tuning);
1653   if (status != VPX_CODEC_OK) {
1654     GST_WARNING_OBJECT (encoder,
1655         "Failed to set VP8E_SET_TUNING: %s", gst_vpx_error_name (status));
1656   }
1657   status = vpx_codec_control (&encoder->encoder, VP8E_SET_CQ_LEVEL,
1658       encoder->cq_level);
1659   if (status != VPX_CODEC_OK) {
1660     GST_WARNING_OBJECT (encoder,
1661         "Failed to set VP8E_SET_CQ_LEVEL: %s", gst_vpx_error_name (status));
1662   }
1663   status = vpx_codec_control (&encoder->encoder, VP8E_SET_MAX_INTRA_BITRATE_PCT,
1664       encoder->max_intra_bitrate_pct);
1665   if (status != VPX_CODEC_OK) {
1666     GST_WARNING_OBJECT (encoder,
1667         "Failed to set VP8E_SET_MAX_INTRA_BITRATE_PCT: %s",
1668         gst_vpx_error_name (status));
1669   }
1670
1671   if (GST_VIDEO_INFO_FPS_D (info) == 0 || GST_VIDEO_INFO_FPS_N (info) == 0) {
1672     gst_video_encoder_set_latency (video_encoder, GST_CLOCK_TIME_NONE,
1673         GST_CLOCK_TIME_NONE);
1674   } else {
1675     gst_video_encoder_set_latency (video_encoder, 0,
1676         gst_util_uint64_scale (encoder->cfg.g_lag_in_frames,
1677             GST_VIDEO_INFO_FPS_D (info) * GST_SECOND,
1678             GST_VIDEO_INFO_FPS_N (info)));
1679   }
1680   encoder->inited = TRUE;
1681
1682   /* Store input state */
1683   if (encoder->input_state)
1684     gst_video_codec_state_unref (encoder->input_state);
1685   encoder->input_state = gst_video_codec_state_ref (state);
1686
1687   /* prepare cached image buffer setup */
1688   image = &encoder->image;
1689   memset (image, 0, sizeof (*image));
1690
1691   switch (encoder->input_state->info.finfo->format) {
1692     case GST_VIDEO_FORMAT_I420:
1693       image->fmt = VPX_IMG_FMT_I420;
1694       image->bps = 12;
1695       image->x_chroma_shift = image->y_chroma_shift = 1;
1696       break;
1697     case GST_VIDEO_FORMAT_YV12:
1698       image->fmt = VPX_IMG_FMT_YV12;
1699       image->bps = 12;
1700       image->x_chroma_shift = image->y_chroma_shift = 1;
1701       break;
1702     case GST_VIDEO_FORMAT_Y42B:
1703       image->fmt = VPX_IMG_FMT_I422;
1704       image->bps = 16;
1705       image->x_chroma_shift = 1;
1706       image->y_chroma_shift = 0;
1707       break;
1708     case GST_VIDEO_FORMAT_Y444:
1709       image->fmt = VPX_IMG_FMT_I444;
1710       image->bps = 24;
1711       image->x_chroma_shift = image->y_chroma_shift = 0;
1712       break;
1713     default:
1714       g_assert_not_reached ();
1715       break;
1716   }
1717   image->w = image->d_w = GST_VIDEO_INFO_WIDTH (info);
1718   image->h = image->d_h = GST_VIDEO_INFO_HEIGHT (info);
1719
1720   image->stride[VPX_PLANE_Y] = GST_VIDEO_INFO_COMP_STRIDE (info, 0);
1721   image->stride[VPX_PLANE_U] = GST_VIDEO_INFO_COMP_STRIDE (info, 1);
1722   image->stride[VPX_PLANE_V] = GST_VIDEO_INFO_COMP_STRIDE (info, 2);
1723
1724   profile_str = g_strdup_printf ("%d", encoder->cfg.g_profile);
1725   caps = gst_caps_new_simple ("video/x-vp9",
1726       "profile", G_TYPE_STRING, profile_str, NULL);
1727   g_free (profile_str);
1728
1729   g_mutex_unlock (&encoder->encoder_lock);
1730
1731   output_state =
1732       gst_video_encoder_set_output_state (video_encoder, caps, state);
1733   gst_video_codec_state_unref (output_state);
1734
1735   gst_video_encoder_negotiate (GST_VIDEO_ENCODER (encoder));
1736
1737   return ret;
1738 }
1739
1740 static GstFlowReturn
1741 gst_vp9_enc_process (GstVP9Enc * encoder)
1742 {
1743   vpx_codec_iter_t iter = NULL;
1744   const vpx_codec_cx_pkt_t *pkt;
1745   GstVideoEncoder *video_encoder;
1746   GstVideoCodecFrame *frame;
1747   GstFlowReturn ret = GST_FLOW_OK;
1748
1749   video_encoder = GST_VIDEO_ENCODER (encoder);
1750
1751   g_mutex_lock (&encoder->encoder_lock);
1752   pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1753   while (pkt != NULL) {
1754     GstBuffer *buffer;
1755     gboolean invisible;
1756
1757     GST_DEBUG_OBJECT (encoder, "packet %u type %d", (guint) pkt->data.frame.sz,
1758         pkt->kind);
1759
1760     if (pkt->kind == VPX_CODEC_STATS_PKT
1761         && encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1762       GST_LOG_OBJECT (encoder, "handling STATS packet");
1763
1764       g_byte_array_append (encoder->first_pass_cache_content,
1765           pkt->data.twopass_stats.buf, pkt->data.twopass_stats.sz);
1766
1767       frame = gst_video_encoder_get_oldest_frame (video_encoder);
1768       if (frame != NULL) {
1769         buffer = gst_buffer_new ();
1770         GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_LIVE);
1771         frame->output_buffer = buffer;
1772         g_mutex_unlock (&encoder->encoder_lock);
1773         ret = gst_video_encoder_finish_frame (video_encoder, frame);
1774         g_mutex_lock (&encoder->encoder_lock);
1775       }
1776
1777       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1778       continue;
1779     } else if (pkt->kind != VPX_CODEC_CX_FRAME_PKT) {
1780       GST_LOG_OBJECT (encoder, "non frame pkt: %d", pkt->kind);
1781       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1782       continue;
1783     }
1784
1785     invisible = (pkt->data.frame.flags & VPX_FRAME_IS_INVISIBLE) != 0;
1786     frame = gst_video_encoder_get_oldest_frame (video_encoder);
1787     g_assert (frame != NULL);
1788     if ((pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0)
1789       GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
1790     else
1791       GST_VIDEO_CODEC_FRAME_UNSET_SYNC_POINT (frame);
1792
1793     /* FIXME : It would be nice to avoid the memory copy ... */
1794     buffer =
1795         gst_buffer_new_wrapped (g_memdup (pkt->data.frame.buf,
1796             pkt->data.frame.sz), pkt->data.frame.sz);
1797
1798     if (invisible) {
1799       g_mutex_unlock (&encoder->encoder_lock);
1800       ret = gst_pad_push (GST_VIDEO_ENCODER_SRC_PAD (encoder), buffer);
1801       g_mutex_lock (&encoder->encoder_lock);
1802       gst_video_codec_frame_unref (frame);
1803     } else {
1804       frame->output_buffer = buffer;
1805       g_mutex_unlock (&encoder->encoder_lock);
1806       ret = gst_video_encoder_finish_frame (video_encoder, frame);
1807       g_mutex_lock (&encoder->encoder_lock);
1808     }
1809
1810     pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1811   }
1812   g_mutex_unlock (&encoder->encoder_lock);
1813
1814   return ret;
1815 }
1816
1817 /* This function should be called holding then stream lock*/
1818 static GstFlowReturn
1819 gst_vp9_enc_drain (GstVideoEncoder * video_encoder)
1820 {
1821   GstVP9Enc *encoder;
1822   int flags = 0;
1823   vpx_codec_err_t status;
1824   gint64 deadline;
1825
1826   encoder = GST_VP9_ENC (video_encoder);
1827
1828   g_mutex_lock (&encoder->encoder_lock);
1829   deadline = encoder->deadline;
1830   g_mutex_unlock (&encoder->encoder_lock);
1831
1832   status =
1833       vpx_codec_encode (&encoder->encoder, NULL, encoder->n_frames, 1, flags,
1834       deadline);
1835
1836   if (status != 0) {
1837     GST_ERROR_OBJECT (encoder, "encode returned %d %s", status,
1838         gst_vpx_error_name (status));
1839     return GST_FLOW_ERROR;
1840   }
1841
1842   /* dispatch remaining frames */
1843   gst_vp9_enc_process (encoder);
1844
1845   g_mutex_lock (&encoder->encoder_lock);
1846   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS && encoder->multipass_cache_file) {
1847     GError *err = NULL;
1848
1849     if (!g_file_set_contents (encoder->multipass_cache_file,
1850             (const gchar *) encoder->first_pass_cache_content->data,
1851             encoder->first_pass_cache_content->len, &err)) {
1852       GST_ELEMENT_ERROR (encoder, RESOURCE, WRITE, (NULL),
1853           ("Failed to write multipass cache file: %s", err->message));
1854       g_error_free (err);
1855     }
1856   }
1857   g_mutex_unlock (&encoder->encoder_lock);
1858
1859   return GST_FLOW_OK;
1860 }
1861
1862 static GstFlowReturn
1863 gst_vp9_enc_finish (GstVideoEncoder * video_encoder)
1864 {
1865   GstFlowReturn ret;
1866
1867   GST_DEBUG_OBJECT (video_encoder, "finish");
1868
1869   ret = gst_vp9_enc_drain (video_encoder);
1870
1871   return ret;
1872 }
1873
1874 static vpx_image_t *
1875 gst_vp9_enc_buffer_to_image (GstVP9Enc * enc, GstVideoFrame * frame)
1876 {
1877   vpx_image_t *image = g_slice_new (vpx_image_t);
1878
1879   memcpy (image, &enc->image, sizeof (*image));
1880
1881   image->planes[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_DATA (frame, 0);
1882   image->planes[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_DATA (frame, 1);
1883   image->planes[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_DATA (frame, 2);
1884
1885   image->stride[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 0);
1886   image->stride[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 1);
1887   image->stride[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 2);
1888
1889   return image;
1890 }
1891
1892 static GstFlowReturn
1893 gst_vp9_enc_handle_frame (GstVideoEncoder * video_encoder,
1894     GstVideoCodecFrame * frame)
1895 {
1896   GstVP9Enc *encoder;
1897   vpx_codec_err_t status;
1898   int flags = 0;
1899   vpx_image_t *image;
1900   GstVideoFrame vframe;
1901
1902   GST_DEBUG_OBJECT (video_encoder, "handle_frame");
1903
1904   encoder = GST_VP9_ENC (video_encoder);
1905
1906   encoder->n_frames++;
1907
1908   GST_DEBUG_OBJECT (video_encoder, "size %d %d",
1909       GST_VIDEO_INFO_WIDTH (&encoder->input_state->info),
1910       GST_VIDEO_INFO_HEIGHT (&encoder->input_state->info));
1911
1912   gst_video_frame_map (&vframe, &encoder->input_state->info,
1913       frame->input_buffer, GST_MAP_READ);
1914   image = gst_vp9_enc_buffer_to_image (encoder, &vframe);
1915
1916   if (GST_VIDEO_CODEC_FRAME_IS_FORCE_KEYFRAME (frame)) {
1917     flags |= VPX_EFLAG_FORCE_KF;
1918   }
1919
1920   g_mutex_lock (&encoder->encoder_lock);
1921   status = vpx_codec_encode (&encoder->encoder, image,
1922       encoder->n_frames, 1, flags, encoder->deadline);
1923   g_mutex_unlock (&encoder->encoder_lock);
1924   gst_video_frame_unmap (&vframe);
1925
1926   if (status != 0) {
1927     GST_ELEMENT_ERROR (encoder, LIBRARY, ENCODE,
1928         ("Failed to encode frame"), ("%s", gst_vpx_error_name (status)));
1929     return FALSE;
1930   }
1931   gst_video_codec_frame_unref (frame);
1932   return gst_vp9_enc_process (encoder);
1933 }
1934
1935 static gboolean
1936 gst_vp9_enc_sink_event (GstVideoEncoder * benc, GstEvent * event)
1937 {
1938   GstVP9Enc *enc = GST_VP9_ENC (benc);
1939
1940   /* FIXME : Move this to base encoder class */
1941
1942   if (GST_EVENT_TYPE (event) == GST_EVENT_TAG) {
1943     GstTagList *list;
1944     GstTagSetter *setter = GST_TAG_SETTER (enc);
1945     const GstTagMergeMode mode = gst_tag_setter_get_tag_merge_mode (setter);
1946
1947     gst_event_parse_tag (event, &list);
1948     gst_tag_setter_merge_tags (setter, list, mode);
1949   }
1950
1951   /* just peeked, baseclass handles the rest */
1952   return GST_VIDEO_ENCODER_CLASS (parent_class)->sink_event (benc, event);
1953 }
1954
1955 static gboolean
1956 gst_vp9_enc_propose_allocation (GstVideoEncoder * encoder, GstQuery * query)
1957 {
1958   gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
1959
1960   return GST_VIDEO_ENCODER_CLASS (parent_class)->propose_allocation (encoder,
1961       query);
1962 }
1963
1964 #endif /* HAVE_VP9_ENCODER */