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 gboolean gst_vp9_enc_flush (GstVideoEncoder * video_encoder);
 350 static GstFlowReturn gst_vp9_enc_drain (GstVideoEncoder * video_encoder);
 351 static GstFlowReturn gst_vp9_enc_handle_frame (GstVideoEncoder *
 352     video_encoder, GstVideoCodecFrame * frame);
 353 static GstFlowReturn gst_vp9_enc_drain (GstVideoEncoder * video_encoder);
 354 static gboolean gst_vp9_enc_sink_event (GstVideoEncoder *
 355     video_encoder, GstEvent * event);
 356 static gboolean gst_vp9_enc_propose_allocation (GstVideoEncoder * encoder,
 357     GstQuery * query);
 358
 359 /* FIXME: Y42B and Y444 do not work yet it seems */
 360 static GstStaticPadTemplate gst_vp9_enc_sink_template =
 361 GST_STATIC_PAD_TEMPLATE ("sink",
 362     GST_PAD_SINK,
 363     GST_PAD_ALWAYS,
 364     /*GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ I420, YV12, Y42B, Y444 }")) */
 365     GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ I420, YV12 }"))
 366     );
 367
 368 static GstStaticPadTemplate gst_vp9_enc_src_template =
 369 GST_STATIC_PAD_TEMPLATE ("src",
 370     GST_PAD_SRC,
 371     GST_PAD_ALWAYS,
 372     GST_STATIC_CAPS ("video/x-vp9, " "profile = (string) {0, 1, 2, 3}")
 373     );
 374
 375 #define parent_class gst_vp9_enc_parent_class
 376 G_DEFINE_TYPE_WITH_CODE (GstVP9Enc, gst_vp9_enc, GST_TYPE_VIDEO_ENCODER,
 377     G_IMPLEMENT_INTERFACE (GST_TYPE_TAG_SETTER, NULL);
 378     G_IMPLEMENT_INTERFACE (GST_TYPE_PRESET, NULL););
 379
 380 static void
 381 gst_vp9_enc_class_init (GstVP9EncClass * klass)
 382 {
 383   GObjectClass *gobject_class;
 384   GstElementClass *element_class;
 385   GstVideoEncoderClass *video_encoder_class;
 386
 387   gobject_class = G_OBJECT_CLASS (klass);
 388   element_class = GST_ELEMENT_CLASS (klass);
 389   video_encoder_class = GST_VIDEO_ENCODER_CLASS (klass);
 390
 391   gobject_class->set_property = gst_vp9_enc_set_property;
 392   gobject_class->get_property = gst_vp9_enc_get_property;
 393   gobject_class->finalize = gst_vp9_enc_finalize;
 394
 395   gst_element_class_add_pad_template (element_class,
 396       gst_static_pad_template_get (&gst_vp9_enc_src_template));
 397   gst_element_class_add_pad_template (element_class,
 398       gst_static_pad_template_get (&gst_vp9_enc_sink_template));
 399
 400   gst_element_class_set_static_metadata (element_class,
 401       "On2 VP9 Encoder",
 402       "Codec/Encoder/Video",
 403       "Encode VP9 video streams", "David Schleef <ds@entropywave.com>, "
 404       "Sebastian Dröge <sebastian.droege@collabora.co.uk>");
 405
 406   video_encoder_class->start = gst_vp9_enc_start;
 407   video_encoder_class->stop = gst_vp9_enc_stop;
 408   video_encoder_class->handle_frame = gst_vp9_enc_handle_frame;
 409   video_encoder_class->set_format = gst_vp9_enc_set_format;
 410   video_encoder_class->flush = gst_vp9_enc_flush;
 411   video_encoder_class->finish = gst_vp9_enc_finish;
 412   video_encoder_class->sink_event = gst_vp9_enc_sink_event;
 413   video_encoder_class->propose_allocation = gst_vp9_enc_propose_allocation;
 414
 415   g_object_class_install_property (gobject_class, PROP_RC_END_USAGE,
 416       g_param_spec_enum ("end-usage", "Rate control mode",
 417           "Rate control mode",
 418           GST_VP9_ENC_END_USAGE_TYPE, DEFAULT_RC_END_USAGE,
 419           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 420
 421   g_object_class_install_property (gobject_class, PROP_RC_TARGET_BITRATE,
 422       g_param_spec_int ("target-bitrate", "Target bitrate",
 423           "Target bitrate (in bits/sec)",
 424           0, G_MAXINT, DEFAULT_RC_TARGET_BITRATE,
 425           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 426
 427   g_object_class_install_property (gobject_class, PROP_RC_MIN_QUANTIZER,
 428       g_param_spec_int ("min-quantizer", "Minimum Quantizer",
 429           "Minimum Quantizer (best)",
 430           0, 63, DEFAULT_RC_MIN_QUANTIZER,
 431           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 432
 433   g_object_class_install_property (gobject_class, PROP_RC_MAX_QUANTIZER,
 434       g_param_spec_int ("max-quantizer", "Maximum Quantizer",
 435           "Maximum Quantizer (worst)",
 436           0, 63, DEFAULT_RC_MAX_QUANTIZER,
 437           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 438
 439   g_object_class_install_property (gobject_class, PROP_RC_DROPFRAME_THRESH,
 440       g_param_spec_int ("dropframe-threshold", "Drop Frame Threshold",
 441           "Temporal resampling threshold (buf %)",
 442           0, 100, DEFAULT_RC_DROPFRAME_THRESH,
 443           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 444
 445   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_ALLOWED,
 446       g_param_spec_boolean ("resize-allowed", "Resize Allowed",
 447           "Allow spatial resampling",
 448           DEFAULT_RC_RESIZE_ALLOWED,
 449           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 450
 451   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_UP_THRESH,
 452       g_param_spec_int ("resize-up-threshold", "Resize Up Threshold",
 453           "Upscale threshold (buf %)",
 454           0, 100, DEFAULT_RC_RESIZE_UP_THRESH,
 455           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 456
 457   g_object_class_install_property (gobject_class, PROP_RC_RESIZE_DOWN_THRESH,
 458       g_param_spec_int ("resize-down-threshold", "Resize Down Threshold",
 459           "Downscale threshold (buf %)",
 460           0, 100, DEFAULT_RC_RESIZE_DOWN_THRESH,
 461           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 462
 463   g_object_class_install_property (gobject_class, PROP_RC_UNDERSHOOT_PCT,
 464       g_param_spec_int ("undershoot", "Undershoot PCT",
 465           "Datarate undershoot (min) target (%)",
 466           0, 1000, DEFAULT_RC_UNDERSHOOT_PCT,
 467           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 468
 469   g_object_class_install_property (gobject_class, PROP_RC_OVERSHOOT_PCT,
 470       g_param_spec_int ("overshoot", "Overshoot PCT",
 471           "Datarate overshoot (max) target (%)",
 472           0, 1000, DEFAULT_RC_OVERSHOOT_PCT,
 473           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 474
 475   g_object_class_install_property (gobject_class, PROP_RC_BUF_SZ,
 476       g_param_spec_int ("buffer-size", "Buffer size",
 477           "Client buffer size (ms)",
 478           0, G_MAXINT, DEFAULT_RC_BUF_SZ,
 479           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 480
 481   g_object_class_install_property (gobject_class, PROP_RC_BUF_INITIAL_SZ,
 482       g_param_spec_int ("buffer-initial-size", "Buffer initial size",
 483           "Initial client buffer size (ms)",
 484           0, G_MAXINT, DEFAULT_RC_BUF_INITIAL_SZ,
 485           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 486
 487   g_object_class_install_property (gobject_class, PROP_RC_BUF_OPTIMAL_SZ,
 488       g_param_spec_int ("buffer-optimal-size", "Buffer optimal size",
 489           "Optimal client buffer size (ms)",
 490           0, G_MAXINT, DEFAULT_RC_BUF_OPTIMAL_SZ,
 491           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 492
 493   g_object_class_install_property (gobject_class, PROP_RC_2PASS_VBR_BIAS_PCT,
 494       g_param_spec_int ("twopass-vbr-bias", "2-pass VBR bias",
 495           "CBR/VBR bias (0=CBR, 100=VBR)",
 496           0, 100, DEFAULT_RC_2PASS_VBR_BIAS_PCT,
 497           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 498
 499   g_object_class_install_property (gobject_class,
 500       PROP_RC_2PASS_VBR_MINSECTION_PCT,
 501       g_param_spec_int ("twopass-vbr-minsection", "2-pass GOP min bitrate",
 502           "GOP minimum bitrate (% target)", 0, G_MAXINT,
 503           DEFAULT_RC_2PASS_VBR_MINSECTION_PCT,
 504           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 505
 506   g_object_class_install_property (gobject_class,
 507       PROP_RC_2PASS_VBR_MAXSECTION_PCT,
 508       g_param_spec_int ("twopass-vbr-maxsection", "2-pass GOP max bitrate",
 509           "GOP maximum bitrate (% target)", 0, G_MAXINT,
 510           DEFAULT_RC_2PASS_VBR_MINSECTION_PCT,
 511           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 512
 513   g_object_class_install_property (gobject_class, PROP_KF_MODE,
 514       g_param_spec_enum ("keyframe-mode", "Keyframe Mode",
 515           "Keyframe placement",
 516           GST_VP9_ENC_KF_MODE_TYPE, DEFAULT_KF_MODE,
 517           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 518
 519   g_object_class_install_property (gobject_class, PROP_KF_MAX_DIST,
 520       g_param_spec_int ("keyframe-max-dist", "Keyframe max distance",
 521           "Maximum distance between keyframes (number of frames)",
 522           0, G_MAXINT, DEFAULT_KF_MAX_DIST,
 523           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 524
 525   g_object_class_install_property (gobject_class, PROP_MULTIPASS_MODE,
 526       g_param_spec_enum ("multipass-mode", "Multipass Mode",
 527           "Multipass encode mode",
 528           GST_VP9_ENC_MULTIPASS_MODE_TYPE, DEFAULT_MULTIPASS_MODE,
 529           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 530
 531   g_object_class_install_property (gobject_class, PROP_MULTIPASS_CACHE_FILE,
 532       g_param_spec_string ("multipass-cache-file", "Multipass Cache File",
 533           "Multipass cache file",
 534           DEFAULT_MULTIPASS_CACHE_FILE,
 535           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 536
 537   g_object_class_install_property (gobject_class, PROP_TS_NUMBER_LAYERS,
 538       g_param_spec_int ("temporal-scalability-number-layers",
 539           "Number of coding layers", "Number of coding layers to use", 1, 5,
 540           DEFAULT_TS_NUMBER_LAYERS,
 541           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 542
 543   g_object_class_install_property (gobject_class, PROP_TS_TARGET_BITRATE,
 544       g_param_spec_value_array ("temporal-scalability-target-bitrate",
 545           "Coding layer target bitrates",
 546           "Target bitrates for coding layers (one per layer, decreasing)",
 547           g_param_spec_int ("target-bitrate", "Target bitrate",
 548               "Target bitrate", 0, G_MAXINT, DEFAULT_RC_TARGET_BITRATE,
 549               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
 550           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 551
 552   g_object_class_install_property (gobject_class, PROP_TS_RATE_DECIMATOR,
 553       g_param_spec_value_array ("temporal-scalability-rate-decimator",
 554           "Coding layer rate decimator",
 555           "Rate decimation factors for each layer",
 556           g_param_spec_int ("rate-decimator", "Rate decimator",
 557               "Rate decimator", 0, 1000000000, 0,
 558               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
 559           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 560
 561   g_object_class_install_property (gobject_class, PROP_TS_PERIODICITY,
 562       g_param_spec_int ("temporal-scalability-periodicity",
 563           "Coding layer periodicity",
 564           "Length of sequence that defines layer membership periodicity", 0, 16,
 565           DEFAULT_TS_PERIODICITY,
 566           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 567
 568   g_object_class_install_property (gobject_class, PROP_TS_LAYER_ID,
 569       g_param_spec_value_array ("temporal-scalability-layer-id",
 570           "Coding layer identification",
 571           "Sequence defining coding layer membership",
 572           g_param_spec_int ("layer-id", "Layer ID", "Layer ID", 0, 4, 0,
 573               G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS),
 574           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 575
 576   g_object_class_install_property (gobject_class, PROP_LAG_IN_FRAMES,
 577       g_param_spec_int ("lag-in-frames", "Lag in frames",
 578           "Maximum number of frames to lag",
 579           0, 25, DEFAULT_LAG_IN_FRAMES,
 580           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 581
 582   g_object_class_install_property (gobject_class, PROP_ERROR_RESILIENT,
 583       g_param_spec_flags ("error-resilient", "Error resilient",
 584           "Error resilience flags",
 585           GST_VP9_ENC_ER_FLAGS_TYPE, DEFAULT_ERROR_RESILIENT,
 586           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 587
 588   g_object_class_install_property (gobject_class, PROP_THREADS,
 589       g_param_spec_int ("threads", "Threads",
 590           "Number of threads to use",
 591           0, 64, DEFAULT_THREADS,
 592           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 593
 594   g_object_class_install_property (gobject_class, PROP_DEADLINE,
 595       g_param_spec_int64 ("deadline", "Deadline",
 596           "Deadline per frame (usec, 0=disabled)",
 597           0, G_MAXINT64, DEFAULT_DEADLINE,
 598           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 599
 600   g_object_class_install_property (gobject_class, PROP_H_SCALING_MODE,
 601       g_param_spec_enum ("horizontal-scaling-mode", "Horizontal scaling mode",
 602           "Horizontal scaling mode",
 603           GST_VP9_ENC_SCALING_MODE_TYPE, DEFAULT_H_SCALING_MODE,
 604           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 605
 606   g_object_class_install_property (gobject_class, PROP_V_SCALING_MODE,
 607       g_param_spec_enum ("vertical-scaling-mode", "Vertical scaling mode",
 608           "Vertical scaling mode",
 609           GST_VP9_ENC_SCALING_MODE_TYPE, DEFAULT_V_SCALING_MODE,
 610           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 611
 612   g_object_class_install_property (gobject_class, PROP_CPU_USED,
 613       g_param_spec_int ("cpu-used", "CPU used",
 614           "CPU used",
 615           -16, 16, DEFAULT_CPU_USED,
 616           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 617
 618   g_object_class_install_property (gobject_class, PROP_ENABLE_AUTO_ALT_REF,
 619       g_param_spec_boolean ("auto-alt-ref", "Auto alt reference frames",
 620           "Automatically generate AltRef frames",
 621           DEFAULT_ENABLE_AUTO_ALT_REF,
 622           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 623
 624   g_object_class_install_property (gobject_class, PROP_NOISE_SENSITIVITY,
 625       g_param_spec_int ("noise-sensitivity", "Noise sensitivity",
 626           "Noise sensisivity (frames to blur)",
 627           0, 6, DEFAULT_NOISE_SENSITIVITY,
 628           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 629
 630   g_object_class_install_property (gobject_class, PROP_SHARPNESS,
 631       g_param_spec_int ("sharpness", "Sharpness",
 632           "Filter sharpness",
 633           0, 7, DEFAULT_SHARPNESS,
 634           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 635
 636   g_object_class_install_property (gobject_class, PROP_STATIC_THRESHOLD,
 637       g_param_spec_int ("static-threshold", "Static Threshold",
 638           "Motion detection threshold",
 639           0, G_MAXINT, DEFAULT_STATIC_THRESHOLD,
 640           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 641
 642   g_object_class_install_property (gobject_class, PROP_TOKEN_PARTITIONS,
 643       g_param_spec_enum ("token-partitions", "Token partitions",
 644           "Number of token partitions",
 645           GST_VP9_ENC_TOKEN_PARTITIONS_TYPE, DEFAULT_TOKEN_PARTITIONS,
 646           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 647
 648   g_object_class_install_property (gobject_class, PROP_ARNR_MAXFRAMES,
 649       g_param_spec_int ("arnr-maxframes", "AltRef max frames",
 650           "AltRef maximum number of frames",
 651           0, 15, DEFAULT_ARNR_MAXFRAMES,
 652           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 653
 654   g_object_class_install_property (gobject_class, PROP_ARNR_STRENGTH,
 655       g_param_spec_int ("arnr-strength", "AltRef strength",
 656           "AltRef strength",
 657           0, 6, DEFAULT_ARNR_STRENGTH,
 658           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 659
 660   g_object_class_install_property (gobject_class, PROP_ARNR_TYPE,
 661       g_param_spec_int ("arnr-type", "AltRef type",
 662           "AltRef type",
 663           1, 3, DEFAULT_ARNR_TYPE,
 664           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS |
 665               G_PARAM_DEPRECATED)));
 666
 667   g_object_class_install_property (gobject_class, PROP_TUNING,
 668       g_param_spec_enum ("tuning", "Tuning",
 669           "Tuning",
 670           GST_VP9_ENC_TUNING_TYPE, DEFAULT_TUNING,
 671           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 672
 673   g_object_class_install_property (gobject_class, PROP_CQ_LEVEL,
 674       g_param_spec_int ("cq-level", "Constrained quality level",
 675           "Constrained quality level",
 676           0, 63, DEFAULT_CQ_LEVEL,
 677           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 678
 679   g_object_class_install_property (gobject_class, PROP_MAX_INTRA_BITRATE_PCT,
 680       g_param_spec_int ("max-intra-bitrate", "Max Intra bitrate",
 681           "Maximum Intra frame bitrate",
 682           0, G_MAXINT, DEFAULT_MAX_INTRA_BITRATE_PCT,
 683           (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)));
 684
 685   g_object_class_install_property (gobject_class, PROP_TIMEBASE,
 686       gst_param_spec_fraction ("timebase", "Shortest interframe time",
 687           "Fraction of one second that is the shortest interframe time - normally left as zero which will default to the framerate",
 688           0, 1, G_MAXINT, 1, DEFAULT_TIMEBASE_N, DEFAULT_TIMEBASE_D,
 689           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 690
 691   GST_DEBUG_CATEGORY_INIT (gst_vp9enc_debug, "vp9enc", 0, "VP9 Encoder");
 692 }
 693
 694 static void
 695 gst_vp9_enc_init (GstVP9Enc * gst_vp9_enc)
 696 {
 697   vpx_codec_err_t status;
 698
 699   GST_DEBUG_OBJECT (gst_vp9_enc, "init");
 700
 701   status =
 702       vpx_codec_enc_config_default (&vpx_codec_vp9_cx_algo, &gst_vp9_enc->cfg,
 703       0);
 704   if (status != VPX_CODEC_OK) {
 705     GST_ERROR_OBJECT (gst_vp9_enc,
 706         "Failed to get default encoder configuration: %s",
 707         gst_vpx_error_name (status));
 708     gst_vp9_enc->have_default_config = FALSE;
 709   } else {
 710     gst_vp9_enc->have_default_config = TRUE;
 711   }
 712
 713   gst_vp9_enc->cfg.rc_end_usage = DEFAULT_RC_END_USAGE;
 714   gst_vp9_enc->cfg.rc_target_bitrate = DEFAULT_RC_TARGET_BITRATE / 1000;
 715   gst_vp9_enc->rc_target_bitrate_set = FALSE;
 716   gst_vp9_enc->cfg.rc_min_quantizer = DEFAULT_RC_MIN_QUANTIZER;
 717   gst_vp9_enc->cfg.rc_max_quantizer = DEFAULT_RC_MAX_QUANTIZER;
 718   gst_vp9_enc->cfg.rc_dropframe_thresh = DEFAULT_RC_DROPFRAME_THRESH;
 719   gst_vp9_enc->cfg.rc_resize_allowed = DEFAULT_RC_RESIZE_ALLOWED;
 720   gst_vp9_enc->cfg.rc_resize_up_thresh = DEFAULT_RC_RESIZE_UP_THRESH;
 721   gst_vp9_enc->cfg.rc_resize_down_thresh = DEFAULT_RC_RESIZE_DOWN_THRESH;
 722   gst_vp9_enc->cfg.rc_undershoot_pct = DEFAULT_RC_UNDERSHOOT_PCT;
 723   gst_vp9_enc->cfg.rc_overshoot_pct = DEFAULT_RC_OVERSHOOT_PCT;
 724   gst_vp9_enc->cfg.rc_buf_sz = DEFAULT_RC_BUF_SZ;
 725   gst_vp9_enc->cfg.rc_buf_initial_sz = DEFAULT_RC_BUF_INITIAL_SZ;
 726   gst_vp9_enc->cfg.rc_buf_optimal_sz = DEFAULT_RC_BUF_OPTIMAL_SZ;
 727   gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct = DEFAULT_RC_2PASS_VBR_BIAS_PCT;
 728   gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct =
 729       DEFAULT_RC_2PASS_VBR_MINSECTION_PCT;
 730   gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct =
 731       DEFAULT_RC_2PASS_VBR_MAXSECTION_PCT;
 732   gst_vp9_enc->cfg.kf_mode = DEFAULT_KF_MODE;
 733   gst_vp9_enc->cfg.kf_max_dist = DEFAULT_KF_MAX_DIST;
 734   gst_vp9_enc->cfg.g_pass = DEFAULT_MULTIPASS_MODE;
 735   gst_vp9_enc->multipass_cache_file = g_strdup (DEFAULT_MULTIPASS_CACHE_FILE);
 736   gst_vp9_enc->cfg.ts_number_layers = DEFAULT_TS_NUMBER_LAYERS;
 737   gst_vp9_enc->n_ts_target_bitrate = 0;
 738   gst_vp9_enc->n_ts_rate_decimator = 0;
 739   gst_vp9_enc->cfg.ts_periodicity = DEFAULT_TS_PERIODICITY;
 740   gst_vp9_enc->n_ts_layer_id = 0;
 741   gst_vp9_enc->cfg.g_error_resilient = DEFAULT_ERROR_RESILIENT;
 742   gst_vp9_enc->cfg.g_lag_in_frames = DEFAULT_LAG_IN_FRAMES;
 743   gst_vp9_enc->cfg.g_threads = DEFAULT_THREADS;
 744   gst_vp9_enc->deadline = DEFAULT_DEADLINE;
 745   gst_vp9_enc->h_scaling_mode = DEFAULT_H_SCALING_MODE;
 746   gst_vp9_enc->v_scaling_mode = DEFAULT_V_SCALING_MODE;
 747   gst_vp9_enc->cpu_used = DEFAULT_CPU_USED;
 748   gst_vp9_enc->enable_auto_alt_ref = DEFAULT_ENABLE_AUTO_ALT_REF;
 749   gst_vp9_enc->noise_sensitivity = DEFAULT_NOISE_SENSITIVITY;
 750   gst_vp9_enc->sharpness = DEFAULT_SHARPNESS;
 751   gst_vp9_enc->static_threshold = DEFAULT_STATIC_THRESHOLD;
 752   gst_vp9_enc->token_partitions = DEFAULT_TOKEN_PARTITIONS;
 753   gst_vp9_enc->arnr_maxframes = DEFAULT_ARNR_MAXFRAMES;
 754   gst_vp9_enc->arnr_strength = DEFAULT_ARNR_STRENGTH;
 755   gst_vp9_enc->arnr_type = DEFAULT_ARNR_TYPE;
 756   gst_vp9_enc->tuning = DEFAULT_TUNING;
 757   gst_vp9_enc->cq_level = DEFAULT_CQ_LEVEL;
 758   gst_vp9_enc->max_intra_bitrate_pct = DEFAULT_MAX_INTRA_BITRATE_PCT;
 759   gst_vp9_enc->timebase_n = DEFAULT_TIMEBASE_N;
 760   gst_vp9_enc->timebase_d = DEFAULT_TIMEBASE_D;
 761
 762   gst_vp9_enc->cfg.g_profile = DEFAULT_PROFILE;
 763
 764   g_mutex_init (&gst_vp9_enc->encoder_lock);
 765 }
 766
 767 static void
 768 gst_vp9_enc_finalize (GObject * object)
 769 {
 770   GstVP9Enc *gst_vp9_enc;
 771
 772   GST_DEBUG_OBJECT (object, "finalize");
 773
 774   g_return_if_fail (GST_IS_VP9_ENC (object));
 775   gst_vp9_enc = GST_VP9_ENC (object);
 776
 777   g_free (gst_vp9_enc->multipass_cache_file);
 778   gst_vp9_enc->multipass_cache_file = NULL;
 779
 780   if (gst_vp9_enc->input_state)
 781     gst_video_codec_state_unref (gst_vp9_enc->input_state);
 782
 783   g_mutex_clear (&gst_vp9_enc->encoder_lock);
 784
 785   G_OBJECT_CLASS (parent_class)->finalize (object);
 786 }
 787
 788 static void
 789 gst_vp9_enc_set_property (GObject * object, guint prop_id,
 790     const GValue * value, GParamSpec * pspec)
 791 {
 792   GstVP9Enc *gst_vp9_enc;
 793   gboolean global = FALSE;
 794   vpx_codec_err_t status;
 795
 796   g_return_if_fail (GST_IS_VP9_ENC (object));
 797   gst_vp9_enc = GST_VP9_ENC (object);
 798
 799   GST_DEBUG_OBJECT (object, "gst_vp9_enc_set_property");
 800   g_mutex_lock (&gst_vp9_enc->encoder_lock);
 801   switch (prop_id) {
 802     case PROP_RC_END_USAGE:
 803       gst_vp9_enc->cfg.rc_end_usage = g_value_get_enum (value);
 804       global = TRUE;
 805       break;
 806     case PROP_RC_TARGET_BITRATE:
 807       gst_vp9_enc->cfg.rc_target_bitrate = g_value_get_int (value) / 1000;
 808       gst_vp9_enc->rc_target_bitrate_set = TRUE;
 809       global = TRUE;
 810       break;
 811     case PROP_RC_MIN_QUANTIZER:
 812       gst_vp9_enc->cfg.rc_min_quantizer = g_value_get_int (value);
 813       global = TRUE;
 814       break;
 815     case PROP_RC_MAX_QUANTIZER:
 816       gst_vp9_enc->cfg.rc_max_quantizer = g_value_get_int (value);
 817       global = TRUE;
 818       break;
 819     case PROP_RC_DROPFRAME_THRESH:
 820       gst_vp9_enc->cfg.rc_dropframe_thresh = g_value_get_int (value);
 821       global = TRUE;
 822       break;
 823     case PROP_RC_RESIZE_ALLOWED:
 824       gst_vp9_enc->cfg.rc_resize_allowed = g_value_get_boolean (value);
 825       global = TRUE;
 826       break;
 827     case PROP_RC_RESIZE_UP_THRESH:
 828       gst_vp9_enc->cfg.rc_resize_up_thresh = g_value_get_int (value);
 829       global = TRUE;
 830       break;
 831     case PROP_RC_RESIZE_DOWN_THRESH:
 832       gst_vp9_enc->cfg.rc_resize_down_thresh = g_value_get_int (value);
 833       global = TRUE;
 834       break;
 835     case PROP_RC_UNDERSHOOT_PCT:
 836       gst_vp9_enc->cfg.rc_undershoot_pct = g_value_get_int (value);
 837       global = TRUE;
 838       break;
 839     case PROP_RC_OVERSHOOT_PCT:
 840       gst_vp9_enc->cfg.rc_overshoot_pct = g_value_get_int (value);
 841       global = TRUE;
 842       break;
 843     case PROP_RC_BUF_SZ:
 844       gst_vp9_enc->cfg.rc_buf_sz = g_value_get_int (value);
 845       global = TRUE;
 846       break;
 847     case PROP_RC_BUF_INITIAL_SZ:
 848       gst_vp9_enc->cfg.rc_buf_initial_sz = g_value_get_int (value);
 849       global = TRUE;
 850       break;
 851     case PROP_RC_BUF_OPTIMAL_SZ:
 852       gst_vp9_enc->cfg.rc_buf_optimal_sz = g_value_get_int (value);
 853       global = TRUE;
 854       break;
 855     case PROP_RC_2PASS_VBR_BIAS_PCT:
 856       gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct = g_value_get_int (value);
 857       global = TRUE;
 858       break;
 859     case PROP_RC_2PASS_VBR_MINSECTION_PCT:
 860       gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct = g_value_get_int (value);
 861       global = TRUE;
 862       break;
 863     case PROP_RC_2PASS_VBR_MAXSECTION_PCT:
 864       gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct = g_value_get_int (value);
 865       global = TRUE;
 866       break;
 867     case PROP_KF_MODE:
 868       gst_vp9_enc->cfg.kf_mode = g_value_get_enum (value);
 869       global = TRUE;
 870       break;
 871     case PROP_KF_MAX_DIST:
 872       gst_vp9_enc->cfg.kf_max_dist = g_value_get_int (value);
 873       global = TRUE;
 874       break;
 875     case PROP_MULTIPASS_MODE:
 876       gst_vp9_enc->cfg.g_pass = g_value_get_enum (value);
 877       global = TRUE;
 878       break;
 879     case PROP_MULTIPASS_CACHE_FILE:
 880       if (gst_vp9_enc->multipass_cache_file)
 881         g_free (gst_vp9_enc->multipass_cache_file);
 882       gst_vp9_enc->multipass_cache_file = g_value_dup_string (value);
 883       break;
 884     case PROP_TS_NUMBER_LAYERS:
 885       gst_vp9_enc->cfg.ts_number_layers = g_value_get_int (value);
 886       global = TRUE;
 887       break;
 888     case PROP_TS_TARGET_BITRATE:{
 889       GValueArray *va = g_value_get_boxed (value);
 890
 891       memset (&gst_vp9_enc->cfg.ts_target_bitrate, 0,
 892           sizeof (gst_vp9_enc->cfg.ts_target_bitrate));
 893       if (va == NULL) {
 894         gst_vp9_enc->n_ts_target_bitrate = 0;
 895       } else {
 896         if (va->n_values > VPX_TS_MAX_LAYERS) {
 897           g_warning ("%s: Only %d layers allowed at maximum",
 898               GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_LAYERS);
 899         } else {
 900           gint i;
 901
 902           for (i = 0; i < va->n_values; i++)
 903             gst_vp9_enc->cfg.ts_target_bitrate[i] =
 904                 g_value_get_int (g_value_array_get_nth (va, i));
 905           gst_vp9_enc->n_ts_target_bitrate = va->n_values;
 906         }
 907       }
 908       global = TRUE;
 909       break;
 910     }
 911     case PROP_TS_RATE_DECIMATOR:{
 912       GValueArray *va = g_value_get_boxed (value);
 913
 914       memset (&gst_vp9_enc->cfg.ts_rate_decimator, 0,
 915           sizeof (gst_vp9_enc->cfg.ts_rate_decimator));
 916       if (va == NULL) {
 917         gst_vp9_enc->n_ts_rate_decimator = 0;
 918       } else if (va->n_values > VPX_TS_MAX_LAYERS) {
 919         g_warning ("%s: Only %d layers allowed at maximum",
 920             GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_LAYERS);
 921       } else {
 922         gint i;
 923
 924         for (i = 0; i < va->n_values; i++)
 925           gst_vp9_enc->cfg.ts_rate_decimator[i] =
 926               g_value_get_int (g_value_array_get_nth (va, i));
 927         gst_vp9_enc->n_ts_rate_decimator = va->n_values;
 928       }
 929       global = TRUE;
 930       break;
 931     }
 932     case PROP_TS_PERIODICITY:
 933       gst_vp9_enc->cfg.ts_periodicity = g_value_get_int (value);
 934       global = TRUE;
 935       break;
 936     case PROP_TS_LAYER_ID:{
 937       GValueArray *va = g_value_get_boxed (value);
 938
 939       memset (&gst_vp9_enc->cfg.ts_layer_id, 0,
 940           sizeof (gst_vp9_enc->cfg.ts_layer_id));
 941       if (va && va->n_values > VPX_TS_MAX_PERIODICITY) {
 942         g_warning ("%s: Only %d sized layer sequences allowed at maximum",
 943             GST_ELEMENT_NAME (gst_vp9_enc), VPX_TS_MAX_PERIODICITY);
 944       } else if (va) {
 945         gint i;
 946
 947         for (i = 0; i < va->n_values; i++)
 948           gst_vp9_enc->cfg.ts_layer_id[i] =
 949               g_value_get_int (g_value_array_get_nth (va, i));
 950         gst_vp9_enc->n_ts_layer_id = va->n_values;
 951       } else {
 952         gst_vp9_enc->n_ts_layer_id = 0;
 953       }
 954       global = TRUE;
 955       break;
 956     }
 957     case PROP_ERROR_RESILIENT:
 958       gst_vp9_enc->cfg.g_error_resilient = g_value_get_flags (value);
 959       global = TRUE;
 960       break;
 961     case PROP_LAG_IN_FRAMES:
 962       gst_vp9_enc->cfg.g_lag_in_frames = g_value_get_int (value);
 963       global = TRUE;
 964       break;
 965     case PROP_THREADS:
 966       gst_vp9_enc->cfg.g_threads = g_value_get_int (value);
 967       global = TRUE;
 968       break;
 969     case PROP_DEADLINE:
 970       gst_vp9_enc->deadline = g_value_get_int64 (value);
 971       break;
 972     case PROP_H_SCALING_MODE:
 973       gst_vp9_enc->h_scaling_mode = g_value_get_enum (value);
 974       if (gst_vp9_enc->inited) {
 975         vpx_scaling_mode_t sm;
 976
 977         sm.h_scaling_mode = gst_vp9_enc->h_scaling_mode;
 978         sm.v_scaling_mode = gst_vp9_enc->v_scaling_mode;
 979
 980         status =
 981             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SCALEMODE, &sm);
 982         if (status != VPX_CODEC_OK) {
 983           GST_WARNING_OBJECT (gst_vp9_enc,
 984               "Failed to set VP8E_SET_SCALEMODE: %s",
 985               gst_vpx_error_name (status));
 986         }
 987       }
 988       break;
 989     case PROP_V_SCALING_MODE:
 990       gst_vp9_enc->v_scaling_mode = g_value_get_enum (value);
 991       if (gst_vp9_enc->inited) {
 992         vpx_scaling_mode_t sm;
 993
 994         sm.h_scaling_mode = gst_vp9_enc->h_scaling_mode;
 995         sm.v_scaling_mode = gst_vp9_enc->v_scaling_mode;
 996
 997         status =
 998             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SCALEMODE, &sm);
 999         if (status != VPX_CODEC_OK) {
1000           GST_WARNING_OBJECT (gst_vp9_enc,
1001               "Failed to set VP8E_SET_SCALEMODE: %s",
1002               gst_vpx_error_name (status));
1003         }
1004       }
1005       break;
1006     case PROP_CPU_USED:
1007       gst_vp9_enc->cpu_used = g_value_get_int (value);
1008       if (gst_vp9_enc->inited) {
1009         status =
1010             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_CPUUSED,
1011             gst_vp9_enc->cpu_used);
1012         if (status != VPX_CODEC_OK) {
1013           GST_WARNING_OBJECT (gst_vp9_enc, "Failed to set VP8E_SET_CPUUSED: %s",
1014               gst_vpx_error_name (status));
1015         }
1016       }
1017       break;
1018     case PROP_ENABLE_AUTO_ALT_REF:
1019       gst_vp9_enc->enable_auto_alt_ref = g_value_get_boolean (value);
1020       if (gst_vp9_enc->inited) {
1021         status =
1022             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ENABLEAUTOALTREF,
1023             (gst_vp9_enc->enable_auto_alt_ref ? 1 : 0));
1024         if (status != VPX_CODEC_OK) {
1025           GST_WARNING_OBJECT (gst_vp9_enc,
1026               "Failed to set VP8E_SET_ENABLEAUTOALTREF: %s",
1027               gst_vpx_error_name (status));
1028         }
1029       }
1030       break;
1031     case PROP_NOISE_SENSITIVITY:
1032       gst_vp9_enc->noise_sensitivity = g_value_get_int (value);
1033       if (gst_vp9_enc->inited) {
1034         status =
1035             vpx_codec_control (&gst_vp9_enc->encoder,
1036             VP8E_SET_NOISE_SENSITIVITY, gst_vp9_enc->noise_sensitivity);
1037         if (status != VPX_CODEC_OK) {
1038           GST_WARNING_OBJECT (gst_vp9_enc,
1039               "Failed to set VP8E_SET_NOISE_SENSITIVITY: %s",
1040               gst_vpx_error_name (status));
1041         }
1042       }
1043       break;
1044     case PROP_SHARPNESS:
1045       gst_vp9_enc->sharpness = g_value_get_int (value);
1046       if (gst_vp9_enc->inited) {
1047         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_SHARPNESS,
1048             gst_vp9_enc->sharpness);
1049         if (status != VPX_CODEC_OK) {
1050           GST_WARNING_OBJECT (gst_vp9_enc,
1051               "Failed to set VP8E_SET_SHARPNESS: %s",
1052               gst_vpx_error_name (status));
1053         }
1054       }
1055       break;
1056     case PROP_STATIC_THRESHOLD:
1057       gst_vp9_enc->static_threshold = g_value_get_int (value);
1058       if (gst_vp9_enc->inited) {
1059         status =
1060             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_STATIC_THRESHOLD,
1061             gst_vp9_enc->static_threshold);
1062         if (status != VPX_CODEC_OK) {
1063           GST_WARNING_OBJECT (gst_vp9_enc,
1064               "Failed to set VP8E_SET_STATIC_THRESHOLD: %s",
1065               gst_vpx_error_name (status));
1066         }
1067       }
1068       break;
1069     case PROP_TOKEN_PARTITIONS:
1070       gst_vp9_enc->token_partitions = g_value_get_enum (value);
1071       if (gst_vp9_enc->inited) {
1072         status =
1073             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_TOKEN_PARTITIONS,
1074             gst_vp9_enc->token_partitions);
1075         if (status != VPX_CODEC_OK) {
1076           GST_WARNING_OBJECT (gst_vp9_enc,
1077               "Failed to set VP8E_SET_TOKEN_PARTIONS: %s",
1078               gst_vpx_error_name (status));
1079         }
1080       }
1081       break;
1082     case PROP_ARNR_MAXFRAMES:
1083       gst_vp9_enc->arnr_maxframes = g_value_get_int (value);
1084       if (gst_vp9_enc->inited) {
1085         status =
1086             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ARNR_MAXFRAMES,
1087             gst_vp9_enc->arnr_maxframes);
1088         if (status != VPX_CODEC_OK) {
1089           GST_WARNING_OBJECT (gst_vp9_enc,
1090               "Failed to set VP8E_SET_ARNR_MAXFRAMES: %s",
1091               gst_vpx_error_name (status));
1092         }
1093       }
1094       break;
1095     case PROP_ARNR_STRENGTH:
1096       gst_vp9_enc->arnr_strength = g_value_get_int (value);
1097       if (gst_vp9_enc->inited) {
1098         status =
1099             vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_ARNR_STRENGTH,
1100             gst_vp9_enc->arnr_strength);
1101         if (status != VPX_CODEC_OK) {
1102           GST_WARNING_OBJECT (gst_vp9_enc,
1103               "Failed to set VP8E_SET_ARNR_STRENGTH: %s",
1104               gst_vpx_error_name (status));
1105         }
1106       }
1107       break;
1108     case PROP_ARNR_TYPE:
1109       gst_vp9_enc->arnr_type = g_value_get_int (value);
1110       g_warning ("arnr-type is a no-op since control has been deprecated "
1111           "in libvpx");
1112       break;
1113     case PROP_TUNING:
1114       gst_vp9_enc->tuning = g_value_get_enum (value);
1115       if (gst_vp9_enc->inited) {
1116         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_TUNING,
1117             gst_vp9_enc->tuning);
1118         if (status != VPX_CODEC_OK) {
1119           GST_WARNING_OBJECT (gst_vp9_enc,
1120               "Failed to set VP8E_SET_TUNING: %s", gst_vpx_error_name (status));
1121         }
1122       }
1123       break;
1124     case PROP_CQ_LEVEL:
1125       gst_vp9_enc->cq_level = g_value_get_int (value);
1126       if (gst_vp9_enc->inited) {
1127         status = vpx_codec_control (&gst_vp9_enc->encoder, VP8E_SET_CQ_LEVEL,
1128             gst_vp9_enc->cq_level);
1129         if (status != VPX_CODEC_OK) {
1130           GST_WARNING_OBJECT (gst_vp9_enc,
1131               "Failed to set VP8E_SET_CQ_LEVEL: %s",
1132               gst_vpx_error_name (status));
1133         }
1134       }
1135       break;
1136     case PROP_MAX_INTRA_BITRATE_PCT:
1137       gst_vp9_enc->max_intra_bitrate_pct = g_value_get_int (value);
1138       if (gst_vp9_enc->inited) {
1139         status =
1140             vpx_codec_control (&gst_vp9_enc->encoder,
1141             VP8E_SET_MAX_INTRA_BITRATE_PCT, gst_vp9_enc->max_intra_bitrate_pct);
1142         if (status != VPX_CODEC_OK) {
1143           GST_WARNING_OBJECT (gst_vp9_enc,
1144               "Failed to set VP8E_SET_MAX_INTRA_BITRATE_PCT: %s",
1145               gst_vpx_error_name (status));
1146         }
1147       }
1148       break;
1149     case PROP_TIMEBASE:
1150       gst_vp9_enc->timebase_n = gst_value_get_fraction_numerator (value);
1151       gst_vp9_enc->timebase_d = gst_value_get_fraction_denominator (value);
1152       break;
1153     default:
1154       break;
1155   }
1156
1157   if (global &&gst_vp9_enc->inited) {
1158     status =
1159         vpx_codec_enc_config_set (&gst_vp9_enc->encoder, &gst_vp9_enc->cfg);
1160     if (status != VPX_CODEC_OK) {
1161       g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1162       GST_ELEMENT_ERROR (gst_vp9_enc, LIBRARY, INIT,
1163           ("Failed to set encoder configuration"), ("%s",
1164               gst_vpx_error_name (status)));
1165     } else {
1166       g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1167     }
1168   } else {
1169     g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1170   }
1171 }
1172
1173 static void
1174 gst_vp9_enc_get_property (GObject * object, guint prop_id, GValue * value,
1175     GParamSpec * pspec)
1176 {
1177   GstVP9Enc *gst_vp9_enc;
1178
1179   g_return_if_fail (GST_IS_VP9_ENC (object));
1180   gst_vp9_enc = GST_VP9_ENC (object);
1181
1182   g_mutex_lock (&gst_vp9_enc->encoder_lock);
1183   switch (prop_id) {
1184     case PROP_RC_END_USAGE:
1185       g_value_set_enum (value, gst_vp9_enc->cfg.rc_end_usage);
1186       break;
1187     case PROP_RC_TARGET_BITRATE:
1188       g_value_set_int (value, gst_vp9_enc->cfg.rc_target_bitrate * 1000);
1189       break;
1190     case PROP_RC_MIN_QUANTIZER:
1191       g_value_set_int (value, gst_vp9_enc->cfg.rc_min_quantizer);
1192       break;
1193     case PROP_RC_MAX_QUANTIZER:
1194       g_value_set_int (value, gst_vp9_enc->cfg.rc_max_quantizer);
1195       break;
1196     case PROP_RC_DROPFRAME_THRESH:
1197       g_value_set_int (value, gst_vp9_enc->cfg.rc_dropframe_thresh);
1198       break;
1199     case PROP_RC_RESIZE_ALLOWED:
1200       g_value_set_boolean (value, gst_vp9_enc->cfg.rc_resize_allowed);
1201       break;
1202     case PROP_RC_RESIZE_UP_THRESH:
1203       g_value_set_int (value, gst_vp9_enc->cfg.rc_resize_up_thresh);
1204       break;
1205     case PROP_RC_RESIZE_DOWN_THRESH:
1206       g_value_set_int (value, gst_vp9_enc->cfg.rc_resize_down_thresh);
1207       break;
1208     case PROP_RC_UNDERSHOOT_PCT:
1209       g_value_set_int (value, gst_vp9_enc->cfg.rc_undershoot_pct);
1210       break;
1211     case PROP_RC_OVERSHOOT_PCT:
1212       g_value_set_int (value, gst_vp9_enc->cfg.rc_overshoot_pct);
1213       break;
1214     case PROP_RC_BUF_SZ:
1215       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_sz);
1216       break;
1217     case PROP_RC_BUF_INITIAL_SZ:
1218       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_initial_sz);
1219       break;
1220     case PROP_RC_BUF_OPTIMAL_SZ:
1221       g_value_set_int (value, gst_vp9_enc->cfg.rc_buf_optimal_sz);
1222       break;
1223     case PROP_RC_2PASS_VBR_BIAS_PCT:
1224       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_bias_pct);
1225       break;
1226     case PROP_RC_2PASS_VBR_MINSECTION_PCT:
1227       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_minsection_pct);
1228       break;
1229     case PROP_RC_2PASS_VBR_MAXSECTION_PCT:
1230       g_value_set_int (value, gst_vp9_enc->cfg.rc_2pass_vbr_maxsection_pct);
1231       break;
1232     case PROP_KF_MODE:
1233       g_value_set_enum (value, gst_vp9_enc->cfg.kf_mode);
1234       break;
1235     case PROP_KF_MAX_DIST:
1236       g_value_set_int (value, gst_vp9_enc->cfg.kf_max_dist);
1237       break;
1238     case PROP_MULTIPASS_MODE:
1239       g_value_set_enum (value, gst_vp9_enc->cfg.g_pass);
1240       break;
1241     case PROP_MULTIPASS_CACHE_FILE:
1242       g_value_set_string (value, gst_vp9_enc->multipass_cache_file);
1243       break;
1244     case PROP_TS_NUMBER_LAYERS:
1245       g_value_set_int (value, gst_vp9_enc->cfg.ts_number_layers);
1246       break;
1247     case PROP_TS_TARGET_BITRATE:{
1248       GValueArray *va;
1249
1250       if (gst_vp9_enc->n_ts_target_bitrate == 0) {
1251         g_value_set_boxed (value, NULL);
1252       } else {
1253         gint i;
1254
1255         va = g_value_array_new (gst_vp9_enc->n_ts_target_bitrate);
1256         for (i = 0; i < gst_vp9_enc->n_ts_target_bitrate; i++) {
1257           GValue v = { 0, };
1258
1259           g_value_init (&v, G_TYPE_INT);
1260           g_value_set_int (&v, gst_vp9_enc->cfg.ts_target_bitrate[i]);
1261           g_value_array_append (va, &v);
1262           g_value_unset (&v);
1263         }
1264         g_value_set_boxed (value, va);
1265         g_value_array_free (va);
1266       }
1267       break;
1268     }
1269     case PROP_TS_RATE_DECIMATOR:{
1270       GValueArray *va;
1271
1272       if (gst_vp9_enc->n_ts_rate_decimator == 0) {
1273         g_value_set_boxed (value, NULL);
1274       } else {
1275         gint i;
1276
1277         va = g_value_array_new (gst_vp9_enc->n_ts_rate_decimator);
1278         for (i = 0; i < gst_vp9_enc->n_ts_rate_decimator; i++) {
1279           GValue v = { 0, };
1280
1281           g_value_init (&v, G_TYPE_INT);
1282           g_value_set_int (&v, gst_vp9_enc->cfg.ts_rate_decimator[i]);
1283           g_value_array_append (va, &v);
1284           g_value_unset (&v);
1285         }
1286         g_value_set_boxed (value, va);
1287         g_value_array_free (va);
1288       }
1289       break;
1290     }
1291     case PROP_TS_PERIODICITY:
1292       g_value_set_int (value, gst_vp9_enc->cfg.ts_periodicity);
1293       break;
1294     case PROP_TS_LAYER_ID:{
1295       GValueArray *va;
1296
1297       if (gst_vp9_enc->n_ts_layer_id == 0) {
1298         g_value_set_boxed (value, NULL);
1299       } else {
1300         gint i;
1301
1302         va = g_value_array_new (gst_vp9_enc->n_ts_layer_id);
1303         for (i = 0; i < gst_vp9_enc->n_ts_layer_id; i++) {
1304           GValue v = { 0, };
1305
1306           g_value_init (&v, G_TYPE_INT);
1307           g_value_set_int (&v, gst_vp9_enc->cfg.ts_layer_id[i]);
1308           g_value_array_append (va, &v);
1309           g_value_unset (&v);
1310         }
1311         g_value_set_boxed (value, va);
1312         g_value_array_free (va);
1313       }
1314       break;
1315     }
1316     case PROP_ERROR_RESILIENT:
1317       g_value_set_flags (value, gst_vp9_enc->cfg.g_error_resilient);
1318       break;
1319     case PROP_LAG_IN_FRAMES:
1320       g_value_set_int (value, gst_vp9_enc->cfg.g_lag_in_frames);
1321       break;
1322     case PROP_THREADS:
1323       g_value_set_int (value, gst_vp9_enc->cfg.g_threads);
1324       break;
1325     case PROP_DEADLINE:
1326       g_value_set_int64 (value, gst_vp9_enc->deadline);
1327       break;
1328     case PROP_H_SCALING_MODE:
1329       g_value_set_enum (value, gst_vp9_enc->h_scaling_mode);
1330       break;
1331     case PROP_V_SCALING_MODE:
1332       g_value_set_enum (value, gst_vp9_enc->v_scaling_mode);
1333       break;
1334     case PROP_CPU_USED:
1335       g_value_set_int (value, gst_vp9_enc->cpu_used);
1336       break;
1337     case PROP_ENABLE_AUTO_ALT_REF:
1338       g_value_set_boolean (value, gst_vp9_enc->enable_auto_alt_ref);
1339       break;
1340     case PROP_NOISE_SENSITIVITY:
1341       g_value_set_int (value, gst_vp9_enc->noise_sensitivity);
1342       break;
1343     case PROP_SHARPNESS:
1344       g_value_set_int (value, gst_vp9_enc->sharpness);
1345       break;
1346     case PROP_STATIC_THRESHOLD:
1347       g_value_set_int (value, gst_vp9_enc->static_threshold);
1348       break;
1349     case PROP_TOKEN_PARTITIONS:
1350       g_value_set_enum (value, gst_vp9_enc->token_partitions);
1351       break;
1352     case PROP_ARNR_MAXFRAMES:
1353       g_value_set_int (value, gst_vp9_enc->arnr_maxframes);
1354       break;
1355     case PROP_ARNR_STRENGTH:
1356       g_value_set_int (value, gst_vp9_enc->arnr_strength);
1357       break;
1358     case PROP_ARNR_TYPE:
1359       g_value_set_int (value, gst_vp9_enc->arnr_type);
1360       break;
1361     case PROP_TUNING:
1362       g_value_set_enum (value, gst_vp9_enc->tuning);
1363       break;
1364     case PROP_CQ_LEVEL:
1365       g_value_set_int (value, gst_vp9_enc->cq_level);
1366       break;
1367     case PROP_MAX_INTRA_BITRATE_PCT:
1368       g_value_set_int (value, gst_vp9_enc->max_intra_bitrate_pct);
1369       break;
1370     case PROP_TIMEBASE:
1371       gst_value_set_fraction (value, gst_vp9_enc->timebase_n,
1372           gst_vp9_enc->timebase_d);
1373       break;
1374     default:
1375       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1376       break;
1377   }
1378
1379   g_mutex_unlock (&gst_vp9_enc->encoder_lock);
1380 }
1381
1382 static gboolean
1383 gst_vp9_enc_start (GstVideoEncoder * video_encoder)
1384 {
1385   GstVP9Enc *encoder = GST_VP9_ENC (video_encoder);
1386
1387   GST_DEBUG_OBJECT (video_encoder, "start");
1388
1389   if (!encoder->have_default_config) {
1390     GST_ELEMENT_ERROR (encoder, LIBRARY, INIT,
1391         ("Failed to get default encoder configuration"), (NULL));
1392     return FALSE;
1393   }
1394
1395   return TRUE;
1396 }
1397
1398 static void
1399 gst_vp9_enc_destroy_encoder (GstVP9Enc * encoder)
1400 {
1401   g_mutex_lock (&encoder->encoder_lock);
1402   if (encoder->inited) {
1403     vpx_codec_destroy (&encoder->encoder);
1404     encoder->inited = FALSE;
1405   }
1406
1407   if (encoder->first_pass_cache_content) {
1408     g_byte_array_free (encoder->first_pass_cache_content, TRUE);
1409     encoder->first_pass_cache_content = NULL;
1410   }
1411
1412   if (encoder->cfg.rc_twopass_stats_in.buf) {
1413     g_free (encoder->cfg.rc_twopass_stats_in.buf);
1414     encoder->cfg.rc_twopass_stats_in.buf = NULL;
1415     encoder->cfg.rc_twopass_stats_in.sz = 0;
1416   }
1417   g_mutex_unlock (&encoder->encoder_lock);
1418 }
1419
1420 static gboolean
1421 gst_vp9_enc_stop (GstVideoEncoder * video_encoder)
1422 {
1423   GstVP9Enc *encoder;
1424
1425   GST_DEBUG_OBJECT (video_encoder, "stop");
1426
1427   encoder = GST_VP9_ENC (video_encoder);
1428
1429   gst_vp9_enc_destroy_encoder (encoder);
1430
1431   gst_tag_setter_reset_tags (GST_TAG_SETTER (encoder));
1432
1433   return TRUE;
1434 }
1435
1436 static gint
1437 gst_vp9_enc_get_downstream_profile (GstVP9Enc * encoder)
1438 {
1439   GstCaps *allowed;
1440   GstStructure *s;
1441   gint profile = DEFAULT_PROFILE;
1442
1443   allowed = gst_pad_get_allowed_caps (GST_VIDEO_ENCODER_SRC_PAD (encoder));
1444   if (allowed) {
1445     allowed = gst_caps_truncate (allowed);
1446     s = gst_caps_get_structure (allowed, 0);
1447     if (gst_structure_has_field (s, "profile")) {
1448       const GValue *v = gst_structure_get_value (s, "profile");
1449       const gchar *profile_str = NULL;
1450
1451       if (GST_VALUE_HOLDS_LIST (v) && gst_value_list_get_size (v) > 0) {
1452         profile_str = g_value_get_string (gst_value_list_get_value (v, 0));
1453       } else if (G_VALUE_HOLDS_STRING (v)) {
1454         profile_str = g_value_get_string (v);
1455       }
1456
1457       if (profile_str) {
1458         gchar *endptr = NULL;
1459
1460         profile = g_ascii_strtoull (profile_str, &endptr, 10);
1461         if (*endptr != '\0' || profile < 0 || profile > 3) {
1462           GST_ERROR_OBJECT (encoder, "Invalid profile '%s'", profile_str);
1463           profile = DEFAULT_PROFILE;
1464         }
1465       }
1466     }
1467     gst_caps_unref (allowed);
1468   }
1469
1470   GST_DEBUG_OBJECT (encoder, "Using profile %d", profile);
1471
1472   return profile;
1473 }
1474
1475 static gboolean
1476 gst_vp9_enc_set_format (GstVideoEncoder * video_encoder,
1477     GstVideoCodecState * state)
1478 {
1479   GstVP9Enc *encoder;
1480   vpx_codec_err_t status;
1481   vpx_image_t *image;
1482   GstCaps *caps;
1483   gboolean ret = TRUE;
1484   GstVideoInfo *info = &state->info;
1485   GstVideoCodecState *output_state;
1486   gchar *profile_str;
1487   GstClockTime latency;
1488
1489   encoder = GST_VP9_ENC (video_encoder);
1490   GST_DEBUG_OBJECT (video_encoder, "set_format");
1491
1492   if (encoder->inited) {
1493     gst_vp9_enc_drain (video_encoder);
1494     g_mutex_lock (&encoder->encoder_lock);
1495     vpx_codec_destroy (&encoder->encoder);
1496     encoder->inited = FALSE;
1497   } else {
1498     g_mutex_lock (&encoder->encoder_lock);
1499   }
1500
1501   encoder->cfg.g_profile = gst_vp9_enc_get_downstream_profile (encoder);
1502
1503   /* Scale default bitrate to our size */
1504   if (!encoder->rc_target_bitrate_set)
1505     encoder->cfg.rc_target_bitrate =
1506         gst_util_uint64_scale (DEFAULT_RC_TARGET_BITRATE,
1507         GST_VIDEO_INFO_WIDTH (info) * GST_VIDEO_INFO_HEIGHT (info),
1508         320 * 240 * 1000);
1509
1510   encoder->cfg.g_w = GST_VIDEO_INFO_WIDTH (info);
1511   encoder->cfg.g_h = GST_VIDEO_INFO_HEIGHT (info);
1512
1513   if (encoder->timebase_n != 0 && encoder->timebase_d != 0) {
1514     GST_DEBUG_OBJECT (video_encoder, "Using timebase configuration");
1515     encoder->cfg.g_timebase.num = encoder->timebase_n;
1516     encoder->cfg.g_timebase.den = encoder->timebase_d;
1517   } else if (GST_VIDEO_INFO_FPS_D (info) != 0
1518       && GST_VIDEO_INFO_FPS_N (info) != 0) {
1519     /* GstVideoInfo holds either the framerate or max-framerate (if framerate
1520      * is 0) in FPS so this will be used if max-framerate or framerate
1521      * is set */
1522     GST_DEBUG_OBJECT (video_encoder, "Setting timebase from framerate");
1523     encoder->cfg.g_timebase.num = GST_VIDEO_INFO_FPS_D (info);
1524     encoder->cfg.g_timebase.den = GST_VIDEO_INFO_FPS_N (info);
1525   } else {
1526     /* Zero framerate and max-framerate but still need to setup the timebase to avoid
1527      * a divide by zero error. Presuming the lowest common denominator will be RTP -
1528      * VP9 payload draft states clock rate of 90000 which should work for anyone where
1529      * FPS < 90000 (shouldn't be too many cases where it's higher) though wouldn't be optimal. RTP specification
1530      * http://tools.ietf.org/html/draft-ietf-payload-vp9-01 section 6.3.1 */
1531     GST_WARNING_OBJECT (encoder,
1532         "No timebase and zero framerate setting timebase to 1/90000");
1533     encoder->cfg.g_timebase.num = 1;
1534     encoder->cfg.g_timebase.den = 90000;
1535   }
1536
1537   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1538     if (encoder->first_pass_cache_content == NULL) {
1539       encoder->first_pass_cache_content = g_byte_array_sized_new (4096);
1540     }
1541   } else if (encoder->cfg.g_pass == VPX_RC_LAST_PASS) {
1542     GError *err = NULL;
1543
1544     if (!encoder->multipass_cache_file) {
1545       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1546           ("No multipass cache file provided"), (NULL));
1547       g_mutex_unlock (&encoder->encoder_lock);
1548       return FALSE;
1549     }
1550
1551     if (encoder->cfg.rc_twopass_stats_in.buf != NULL) {
1552       g_free (encoder->cfg.rc_twopass_stats_in.buf);
1553       encoder->cfg.rc_twopass_stats_in.buf = NULL;
1554       encoder->cfg.rc_twopass_stats_in.sz = 0;
1555     }
1556
1557     if (!g_file_get_contents (encoder->multipass_cache_file,
1558             (gchar **) & encoder->cfg.rc_twopass_stats_in.buf,
1559             &encoder->cfg.rc_twopass_stats_in.sz, &err)) {
1560       GST_ELEMENT_ERROR (encoder, RESOURCE, OPEN_READ,
1561           ("Failed to read multipass cache file provided"), ("%s",
1562               err->message));
1563       g_error_free (err);
1564       g_mutex_unlock (&encoder->encoder_lock);
1565       return FALSE;
1566     }
1567   }
1568
1569   status = vpx_codec_enc_init (&encoder->encoder, &vpx_codec_vp9_cx_algo,
1570       &encoder->cfg, 0);
1571   if (status != VPX_CODEC_OK) {
1572     GST_ELEMENT_ERROR (encoder, LIBRARY, INIT,
1573         ("Failed to initialize encoder"), ("%s", gst_vpx_error_name (status)));
1574     g_mutex_unlock (&encoder->encoder_lock);
1575     return FALSE;
1576   }
1577
1578   /* FIXME: Disabled for now, does not work with VP9 */
1579 #if 0
1580   {
1581     vpx_scaling_mode_t sm;
1582
1583     sm.h_scaling_mode = encoder->h_scaling_mode;
1584     sm.v_scaling_mode = encoder->v_scaling_mode;
1585
1586     status = vpx_codec_control (&encoder->encoder, VP8E_SET_SCALEMODE, &sm);
1587     if (status != VPX_CODEC_OK) {
1588       GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_SCALEMODE: %s",
1589           gst_vpx_error_name (status));
1590     }
1591   }
1592 #endif
1593
1594   status =
1595       vpx_codec_control (&encoder->encoder, VP8E_SET_CPUUSED,
1596       encoder->cpu_used);
1597   if (status != VPX_CODEC_OK) {
1598     GST_WARNING_OBJECT (encoder, "Failed to set VP8E_SET_CPUUSED: %s",
1599         gst_vpx_error_name (status));
1600   }
1601
1602   status =
1603       vpx_codec_control (&encoder->encoder, VP8E_SET_ENABLEAUTOALTREF,
1604       (encoder->enable_auto_alt_ref ? 1 : 0));
1605   if (status != VPX_CODEC_OK) {
1606     GST_WARNING_OBJECT (encoder,
1607         "Failed to set VP8E_SET_ENABLEAUTOALTREF: %s",
1608         gst_vpx_error_name (status));
1609   }
1610   status = vpx_codec_control (&encoder->encoder, VP8E_SET_NOISE_SENSITIVITY,
1611       encoder->noise_sensitivity);
1612   if (status != VPX_CODEC_OK) {
1613     GST_WARNING_OBJECT (encoder,
1614         "Failed to set VP8E_SET_NOISE_SENSITIVITY: %s",
1615         gst_vpx_error_name (status));
1616   }
1617   status = vpx_codec_control (&encoder->encoder, VP8E_SET_SHARPNESS,
1618       encoder->sharpness);
1619   if (status != VPX_CODEC_OK) {
1620     GST_WARNING_OBJECT (encoder,
1621         "Failed to set VP8E_SET_SHARPNESS: %s", gst_vpx_error_name (status));
1622   }
1623   status = vpx_codec_control (&encoder->encoder, VP8E_SET_STATIC_THRESHOLD,
1624       encoder->static_threshold);
1625   if (status != VPX_CODEC_OK) {
1626     GST_WARNING_OBJECT (encoder,
1627         "Failed to set VP8E_SET_STATIC_THRESHOLD: %s",
1628         gst_vpx_error_name (status));
1629   }
1630   status = vpx_codec_control (&encoder->encoder, VP8E_SET_TOKEN_PARTITIONS,
1631       encoder->token_partitions);
1632   if (status != VPX_CODEC_OK) {
1633     GST_WARNING_OBJECT (encoder,
1634         "Failed to set VP8E_SET_TOKEN_PARTIONS: %s",
1635         gst_vpx_error_name (status));
1636   }
1637   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_MAXFRAMES,
1638       encoder->arnr_maxframes);
1639   if (status != VPX_CODEC_OK) {
1640     GST_WARNING_OBJECT (encoder,
1641         "Failed to set VP8E_SET_ARNR_MAXFRAMES: %s",
1642         gst_vpx_error_name (status));
1643   }
1644   status = vpx_codec_control (&encoder->encoder, VP8E_SET_ARNR_STRENGTH,
1645       encoder->arnr_strength);
1646   if (status != VPX_CODEC_OK) {
1647     GST_WARNING_OBJECT (encoder,
1648         "Failed to set VP8E_SET_ARNR_STRENGTH: %s",
1649         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     /* FIXME: Assume 25fps for unknown framerates. Better than reporting
1673      * that we introduce no latency while we actually do
1674      */
1675     latency = gst_util_uint64_scale (encoder->cfg.g_lag_in_frames,
1676         1 * GST_SECOND, 25);
1677   } else {
1678     latency = gst_util_uint64_scale (encoder->cfg.g_lag_in_frames,
1679         GST_VIDEO_INFO_FPS_D (info) * GST_SECOND, GST_VIDEO_INFO_FPS_N (info));
1680   }
1681   gst_video_encoder_set_latency (video_encoder, latency, latency);
1682   encoder->inited = TRUE;
1683
1684   /* Store input state */
1685   if (encoder->input_state)
1686     gst_video_codec_state_unref (encoder->input_state);
1687   encoder->input_state = gst_video_codec_state_ref (state);
1688
1689   /* prepare cached image buffer setup */
1690   image = &encoder->image;
1691   memset (image, 0, sizeof (*image));
1692
1693   switch (encoder->input_state->info.finfo->format) {
1694     case GST_VIDEO_FORMAT_I420:
1695       image->fmt = VPX_IMG_FMT_I420;
1696       image->bps = 12;
1697       image->x_chroma_shift = image->y_chroma_shift = 1;
1698       break;
1699     case GST_VIDEO_FORMAT_YV12:
1700       image->fmt = VPX_IMG_FMT_YV12;
1701       image->bps = 12;
1702       image->x_chroma_shift = image->y_chroma_shift = 1;
1703       break;
1704     case GST_VIDEO_FORMAT_Y42B:
1705       image->fmt = VPX_IMG_FMT_I422;
1706       image->bps = 16;
1707       image->x_chroma_shift = 1;
1708       image->y_chroma_shift = 0;
1709       break;
1710     case GST_VIDEO_FORMAT_Y444:
1711       image->fmt = VPX_IMG_FMT_I444;
1712       image->bps = 24;
1713       image->x_chroma_shift = image->y_chroma_shift = 0;
1714       break;
1715     default:
1716       g_assert_not_reached ();
1717       break;
1718   }
1719   image->w = image->d_w = GST_VIDEO_INFO_WIDTH (info);
1720   image->h = image->d_h = GST_VIDEO_INFO_HEIGHT (info);
1721
1722   image->stride[VPX_PLANE_Y] = GST_VIDEO_INFO_COMP_STRIDE (info, 0);
1723   image->stride[VPX_PLANE_U] = GST_VIDEO_INFO_COMP_STRIDE (info, 1);
1724   image->stride[VPX_PLANE_V] = GST_VIDEO_INFO_COMP_STRIDE (info, 2);
1725
1726   profile_str = g_strdup_printf ("%d", encoder->cfg.g_profile);
1727   caps = gst_caps_new_simple ("video/x-vp9",
1728       "profile", G_TYPE_STRING, profile_str, NULL);
1729   g_free (profile_str);
1730
1731   g_mutex_unlock (&encoder->encoder_lock);
1732
1733   output_state =
1734       gst_video_encoder_set_output_state (video_encoder, caps, state);
1735   gst_video_codec_state_unref (output_state);
1736
1737   gst_video_encoder_negotiate (GST_VIDEO_ENCODER (encoder));
1738
1739   return ret;
1740 }
1741
1742 static GstFlowReturn
1743 gst_vp9_enc_process (GstVP9Enc * encoder)
1744 {
1745   vpx_codec_iter_t iter = NULL;
1746   const vpx_codec_cx_pkt_t *pkt;
1747   GstVideoEncoder *video_encoder;
1748   GstVideoCodecFrame *frame;
1749   GstFlowReturn ret = GST_FLOW_OK;
1750
1751   video_encoder = GST_VIDEO_ENCODER (encoder);
1752
1753   g_mutex_lock (&encoder->encoder_lock);
1754   pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1755   while (pkt != NULL) {
1756     GstBuffer *buffer;
1757     gboolean invisible;
1758
1759     GST_DEBUG_OBJECT (encoder, "packet %u type %d", (guint) pkt->data.frame.sz,
1760         pkt->kind);
1761
1762     if (pkt->kind == VPX_CODEC_STATS_PKT
1763         && encoder->cfg.g_pass == VPX_RC_FIRST_PASS) {
1764       GST_LOG_OBJECT (encoder, "handling STATS packet");
1765
1766       g_byte_array_append (encoder->first_pass_cache_content,
1767           pkt->data.twopass_stats.buf, pkt->data.twopass_stats.sz);
1768
1769       frame = gst_video_encoder_get_oldest_frame (video_encoder);
1770       if (frame != NULL) {
1771         buffer = gst_buffer_new ();
1772         GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_LIVE);
1773         frame->output_buffer = buffer;
1774         g_mutex_unlock (&encoder->encoder_lock);
1775         ret = gst_video_encoder_finish_frame (video_encoder, frame);
1776         g_mutex_lock (&encoder->encoder_lock);
1777       }
1778
1779       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1780       continue;
1781     } else if (pkt->kind != VPX_CODEC_CX_FRAME_PKT) {
1782       GST_LOG_OBJECT (encoder, "non frame pkt: %d", pkt->kind);
1783       pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1784       continue;
1785     }
1786
1787     invisible = (pkt->data.frame.flags & VPX_FRAME_IS_INVISIBLE) != 0;
1788     frame = gst_video_encoder_get_oldest_frame (video_encoder);
1789     g_assert (frame != NULL);
1790     if ((pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0)
1791       GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
1792     else
1793       GST_VIDEO_CODEC_FRAME_UNSET_SYNC_POINT (frame);
1794
1795     /* FIXME : It would be nice to avoid the memory copy ... */
1796     buffer =
1797         gst_buffer_new_wrapped (g_memdup (pkt->data.frame.buf,
1798             pkt->data.frame.sz), pkt->data.frame.sz);
1799
1800     if (invisible) {
1801       g_mutex_unlock (&encoder->encoder_lock);
1802       ret = gst_pad_push (GST_VIDEO_ENCODER_SRC_PAD (encoder), buffer);
1803       g_mutex_lock (&encoder->encoder_lock);
1804       gst_video_codec_frame_unref (frame);
1805     } else {
1806       frame->output_buffer = buffer;
1807       g_mutex_unlock (&encoder->encoder_lock);
1808       ret = gst_video_encoder_finish_frame (video_encoder, frame);
1809       g_mutex_lock (&encoder->encoder_lock);
1810     }
1811
1812     pkt = vpx_codec_get_cx_data (&encoder->encoder, &iter);
1813   }
1814   g_mutex_unlock (&encoder->encoder_lock);
1815
1816   return ret;
1817 }
1818
1819 /* This function should be called holding then stream lock*/
1820 static GstFlowReturn
1821 gst_vp9_enc_drain (GstVideoEncoder * video_encoder)
1822 {
1823   GstVP9Enc *encoder;
1824   int flags = 0;
1825   vpx_codec_err_t status;
1826   gint64 deadline;
1827   vpx_codec_pts_t pts;
1828
1829   encoder = GST_VP9_ENC (video_encoder);
1830
1831   g_mutex_lock (&encoder->encoder_lock);
1832   deadline = encoder->deadline;
1833
1834   pts =
1835       gst_util_uint64_scale (encoder->last_pts,
1836       encoder->cfg.g_timebase.num * (GstClockTime) GST_SECOND,
1837       encoder->cfg.g_timebase.den);
1838
1839   status = vpx_codec_encode (&encoder->encoder, NULL, pts, 0, flags, deadline);
1840   g_mutex_unlock (&encoder->encoder_lock);
1841
1842   if (status != 0) {
1843     GST_ERROR_OBJECT (encoder, "encode returned %d %s", status,
1844         gst_vpx_error_name (status));
1845     return GST_FLOW_ERROR;
1846   }
1847
1848   /* dispatch remaining frames */
1849   gst_vp9_enc_process (encoder);
1850
1851   g_mutex_lock (&encoder->encoder_lock);
1852   if (encoder->cfg.g_pass == VPX_RC_FIRST_PASS && encoder->multipass_cache_file) {
1853     GError *err = NULL;
1854
1855     if (!g_file_set_contents (encoder->multipass_cache_file,
1856             (const gchar *) encoder->first_pass_cache_content->data,
1857             encoder->first_pass_cache_content->len, &err)) {
1858       GST_ELEMENT_ERROR (encoder, RESOURCE, WRITE, (NULL),
1859           ("Failed to write multipass cache file: %s", err->message));
1860       g_error_free (err);
1861     }
1862   }
1863   g_mutex_unlock (&encoder->encoder_lock);
1864
1865   return GST_FLOW_OK;
1866 }
1867
1868 static gboolean
1869 gst_vp9_enc_flush (GstVideoEncoder * video_encoder)
1870 {
1871   GstVP9Enc *encoder;
1872
1873   GST_DEBUG_OBJECT (video_encoder, "flush");
1874
1875   encoder = GST_VP9_ENC (video_encoder);
1876
1877   gst_vp9_enc_destroy_encoder (encoder);
1878   if (encoder->input_state) {
1879     gst_video_codec_state_ref (encoder->input_state);
1880     gst_vp9_enc_set_format (video_encoder, encoder->input_state);
1881     gst_video_codec_state_unref (encoder->input_state);
1882   }
1883
1884   return TRUE;
1885 }
1886
1887 static GstFlowReturn
1888 gst_vp9_enc_finish (GstVideoEncoder * video_encoder)
1889 {
1890   GstVP9Enc *encoder;
1891   GstFlowReturn ret;
1892
1893   GST_DEBUG_OBJECT (video_encoder, "finish");
1894
1895   encoder = GST_VP9_ENC (video_encoder);
1896
1897   if (encoder->inited) {
1898     ret = gst_vp9_enc_drain (video_encoder);
1899   } else {
1900     ret = GST_FLOW_OK;
1901   }
1902
1903   return ret;
1904 }
1905
1906 static vpx_image_t *
1907 gst_vp9_enc_buffer_to_image (GstVP9Enc * enc, GstVideoFrame * frame)
1908 {
1909   vpx_image_t *image = g_slice_new (vpx_image_t);
1910
1911   memcpy (image, &enc->image, sizeof (*image));
1912
1913   image->planes[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_DATA (frame, 0);
1914   image->planes[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_DATA (frame, 1);
1915   image->planes[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_DATA (frame, 2);
1916
1917   image->stride[VPX_PLANE_Y] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 0);
1918   image->stride[VPX_PLANE_U] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 1);
1919   image->stride[VPX_PLANE_V] = GST_VIDEO_FRAME_COMP_STRIDE (frame, 2);
1920
1921   return image;
1922 }
1923
1924 static GstFlowReturn
1925 gst_vp9_enc_handle_frame (GstVideoEncoder * video_encoder,
1926     GstVideoCodecFrame * frame)
1927 {
1928   GstVP9Enc *encoder;
1929   vpx_codec_err_t status;
1930   int flags = 0;
1931   vpx_image_t *image;
1932   GstVideoFrame vframe;
1933   vpx_codec_pts_t pts;
1934   unsigned long duration;
1935
1936   GST_DEBUG_OBJECT (video_encoder, "handle_frame");
1937
1938   encoder = GST_VP9_ENC (video_encoder);
1939
1940   GST_DEBUG_OBJECT (video_encoder, "size %d %d",
1941       GST_VIDEO_INFO_WIDTH (&encoder->input_state->info),
1942       GST_VIDEO_INFO_HEIGHT (&encoder->input_state->info));
1943
1944   gst_video_frame_map (&vframe, &encoder->input_state->info,
1945       frame->input_buffer, GST_MAP_READ);
1946   image = gst_vp9_enc_buffer_to_image (encoder, &vframe);
1947
1948   if (GST_VIDEO_CODEC_FRAME_IS_FORCE_KEYFRAME (frame)) {
1949     flags |= VPX_EFLAG_FORCE_KF;
1950   }
1951
1952   g_mutex_lock (&encoder->encoder_lock);
1953   pts =
1954       gst_util_uint64_scale (frame->pts,
1955       encoder->cfg.g_timebase.num * (GstClockTime) GST_SECOND,
1956       encoder->cfg.g_timebase.den);
1957   encoder->last_pts = frame->pts;
1958
1959   if (frame->duration != GST_CLOCK_TIME_NONE) {
1960     duration =
1961         gst_util_uint64_scale (frame->duration,
1962         encoder->cfg.g_timebase.num * (GstClockTime) GST_SECOND,
1963         encoder->cfg.g_timebase.den);
1964     encoder->last_pts += frame->duration;
1965   } else {
1966     duration = 1;
1967   }
1968
1969   status = vpx_codec_encode (&encoder->encoder, image,
1970       pts, duration, flags, encoder->deadline);
1971
1972   g_mutex_unlock (&encoder->encoder_lock);
1973   gst_video_frame_unmap (&vframe);
1974
1975   if (status != 0) {
1976     GST_ELEMENT_ERROR (encoder, LIBRARY, ENCODE,
1977         ("Failed to encode frame"), ("%s", gst_vpx_error_name (status)));
1978     return FALSE;
1979   }
1980   gst_video_codec_frame_unref (frame);
1981   return gst_vp9_enc_process (encoder);
1982 }
1983
1984 static gboolean
1985 gst_vp9_enc_sink_event (GstVideoEncoder * benc, GstEvent * event)
1986 {
1987   GstVP9Enc *enc = GST_VP9_ENC (benc);
1988
1989   /* FIXME : Move this to base encoder class */
1990
1991   if (GST_EVENT_TYPE (event) == GST_EVENT_TAG) {
1992     GstTagList *list;
1993     GstTagSetter *setter = GST_TAG_SETTER (enc);
1994     const GstTagMergeMode mode = gst_tag_setter_get_tag_merge_mode (setter);
1995
1996     gst_event_parse_tag (event, &list);
1997     gst_tag_setter_merge_tags (setter, list, mode);
1998   }
1999
2000   /* just peeked, baseclass handles the rest */
2001   return GST_VIDEO_ENCODER_CLASS (parent_class)->sink_event (benc, event);
2002 }
2003
2004 static gboolean
2005 gst_vp9_enc_propose_allocation (GstVideoEncoder * encoder, GstQuery * query)
2006 {
2007   gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
2008
2009   return GST_VIDEO_ENCODER_CLASS (parent_class)->propose_allocation (encoder,
2010       query);
2011 }
2012
2013 #endif /* HAVE_VP9_ENCODER */