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