ext/vpx/gstvp8enc.c

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