1 /* VP8 2 * Copyright (C) 2006 David Schleef <ds@schleef.org> 3 * Copyright (C) 2010 Entropy Wave Inc 4 * Copyright (C) 2010 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 #ifndef __GST_VPX_ENC_H__ 23 #define __GST_VPX_ENC_H__ 24 25 #ifdef HAVE_CONFIG_H 26 #include "config.h" 27 #endif 28 29 #if defined(HAVE_VP8_ENCODER) || defined(HAVE_VP9_ENCODER) 30 31 #include <gst/gst.h> 32 #include <gst/video/gstvideoencoder.h> 33 34 /* FIXME: Undef HAVE_CONFIG_H because vpx_codec.h uses it, 35 * which causes compilation failures */ 36 #ifdef HAVE_CONFIG_H 37 #undef HAVE_CONFIG_H 38 #endif 39 40 #include <vpx/vpx_encoder.h> 41 #include <vpx/vp8cx.h> 42 43 G_BEGIN_DECLS 44 45 #define GST_TYPE_VPX_ENC \ 46 (gst_vpx_enc_get_type()) 47 #define GST_VPX_ENC(obj) \ 48 (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_VPX_ENC,GstVPXEnc)) 49 #define GST_VPX_ENC_CLASS(klass) \ 50 (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_VPX_ENC,GstVPXEncClass)) 51 #define GST_IS_VPX_ENC(obj) \ 52 (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_VPX_ENC)) 53 #define GST_IS_VPX_ENC_CLASS(obj) \ 54 (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_VPX_ENC)) 55 #define GST_VPX_ENC_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_VPX_ENC, GstVPXEncClass)) 56 57 typedef struct _GstVPXEnc GstVPXEnc; 58 typedef struct _GstVPXEncClass GstVPXEncClass; 59 60 struct _GstVPXEnc 61 { 62 GstVideoEncoder base_video_encoder; 63 64 /* < private > */ 65 vpx_codec_ctx_t encoder; 66 GMutex encoder_lock; 67 68 /* properties */ 69 vpx_codec_enc_cfg_t cfg; 70 gboolean have_default_config; 71 gboolean rc_target_bitrate_set; 72 gint n_ts_target_bitrate; 73 gint n_ts_rate_decimator; 74 gint n_ts_layer_id; 75 /* Global two-pass options */ 76 gchar *multipass_cache_file; 77 gchar *multipass_cache_prefix; 78 guint multipass_cache_idx; 79 GByteArray *first_pass_cache_content; 80 81 /* Encode parameter */ 82 gint64 deadline; 83 84 /* Controls */ 85 VPX_SCALING_MODE h_scaling_mode; 86 VPX_SCALING_MODE v_scaling_mode; 87 int cpu_used; 88 gboolean enable_auto_alt_ref; 89 unsigned int noise_sensitivity; 90 unsigned int sharpness; 91 unsigned int static_threshold; 92 vp8e_token_partitions token_partitions; 93 unsigned int arnr_maxframes; 94 unsigned int arnr_strength; 95 unsigned int arnr_type; 96 vp8e_tuning tuning; 97 unsigned int cq_level; 98 unsigned int max_intra_bitrate_pct; 99 /* Timebase - a value of 0 will use the framerate */ 100 unsigned int timebase_n; 101 unsigned int timebase_d; 102 103 /* state */ 104 gboolean inited; 105 106 vpx_image_t image; 107 108 GstClockTime last_pts; 109 110 GstVideoCodecState *input_state; 111 }; 112 113 struct _GstVPXEncClass 114 { 115 GstVideoEncoderClass base_video_encoder_class; 116 /*virtual function to get supported algo*/ 117 vpx_codec_iface_t* (*get_algo) (GstVPXEnc *enc); 118 /*enabled scaling*/ 119 gboolean (*enable_scaling) (GstVPXEnc *enc); 120 /*set image format info*/ 121 void (*set_image_format) (GstVPXEnc *enc, vpx_image_t *image); 122 /*get new simple caps*/ 123 GstCaps* (*get_new_vpx_caps) (GstVPXEnc *enc); 124 /*set stream info*/ 125 void (*set_stream_info) (GstVPXEnc *enc, GstCaps *caps, GstVideoInfo *info); 126 /*process user data*/ 127 void* (*process_frame_user_data) (GstVPXEnc *enc, GstVideoCodecFrame* frame); 128 /*set frame user data*/ 129 void (*set_frame_user_data) (GstVPXEnc *enc, GstVideoCodecFrame* frame, vpx_image_t *image); 130 /*Handle invisible frame*/ 131 GstFlowReturn (*handle_invisible_frame_buffer) (GstVPXEnc *enc, void* user_data, GstBuffer* buffer); 132 }; 133 134 GType gst_vpx_enc_get_type (void); 135 136 G_END_DECLS 137 138 #endif 139 140 #endif /* __GST_VPX_ENC_H__ */ 141