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>
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.
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.
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.
22 #ifndef __GST_VPX_ENC_H__
23 #define __GST_VPX_ENC_H__
29 #if defined(HAVE_VP8_ENCODER) || defined(HAVE_VP9_ENCODER)
32 #include <gst/video/gstvideoencoder.h>
34 /* FIXME: Undef HAVE_CONFIG_H because vpx_codec.h uses it,
35 * which causes compilation failures */
40 #include <vpx/vpx_encoder.h>
41 #include <vpx/vp8cx.h>
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))
57 typedef struct _GstVPXEnc GstVPXEnc;
58 typedef struct _GstVPXEncClass GstVPXEncClass;
62 GstVideoEncoder base_video_encoder;
65 vpx_codec_ctx_t encoder;
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;
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;
81 /* Encode parameter */
85 VPX_SCALING_MODE h_scaling_mode;
86 VPX_SCALING_MODE v_scaling_mode;
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;
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;
108 GstClockTime last_pts;
110 GstVideoCodecState *input_state;
113 struct _GstVPXEncClass
115 GstVideoEncoderClass base_video_encoder_class;
116 /*virtual function to get supported algo*/
117 vpx_codec_iface_t* (*get_algo) (GstVPXEnc *enc);
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);
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);
134 GType gst_vpx_enc_get_type (void);
140 #endif /* __GST_VPX_ENC_H__ */