+
+/*****************************************************************************
+ * encoder_sys_t : theora encoder descriptor
+ *****************************************************************************/
+struct encoder_sys_t
+{
+ /*
+ * Input properties
+ */
+ vlc_bool_t b_headers;
+
+ /*
+ * Theora properties
+ */
+ theora_info ti; /* theora bitstream settings */
+ theora_comment tc; /* theora comment header */
+ theora_state td; /* theora bitstream user comments */
+
+ int i_width, i_height;
+};
+
+/*****************************************************************************
+ * OpenEncoder: probe the encoder and return score
+ *****************************************************************************/
+static int OpenEncoder( vlc_object_t *p_this )
+{
+ encoder_t *p_enc = (encoder_t *)p_this;
+ encoder_sys_t *p_sys = p_enc->p_sys;
+ ogg_packet header;
+ uint8_t *p_extra;
+ vlc_value_t val;
+ int i_quality, i;
+
+ if( p_enc->fmt_out.i_codec != VLC_FOURCC('t','h','e','o') &&
+ !p_enc->b_force )
+ {
+ return VLC_EGENERIC;
+ }
+
+ /* Allocate the memory needed to store the decoder's structure */
+ if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
+ {
+ msg_Err( p_enc, "out of memory" );
+ return VLC_EGENERIC;
+ }
+ p_enc->p_sys = p_sys;
+
+ p_enc->pf_encode_video = Encode;
+ p_enc->fmt_in.i_codec = VLC_FOURCC('I','4','2','0');
+ p_enc->fmt_out.i_codec = VLC_FOURCC('t','h','e','o');
+
+ sout_CfgParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
+
+ var_Get( p_enc, ENC_CFG_PREFIX "quality", &val );
+ i_quality = val.i_int;
+ if( i_quality > 10 ) i_quality = 10;
+ if( i_quality < 0 ) i_quality = 0;
+
+ theora_info_init( &p_sys->ti );
+
+ p_sys->ti.width = p_enc->fmt_in.video.i_width;
+ p_sys->ti.height = p_enc->fmt_in.video.i_height;
+
+ if( p_sys->ti.width % 16 || p_sys->ti.height % 16 )
+ {
+ /* Pictures from the transcoder should always have a pitch
+ * which is a multiple of 16 */
+ p_sys->ti.width = (p_sys->ti.width + 15) >> 4 << 4;
+ p_sys->ti.height = (p_sys->ti.height + 15) >> 4 << 4;
+
+ msg_Dbg( p_enc, "padding video from %dx%d to %dx%d",
+ p_enc->fmt_in.video.i_width, p_enc->fmt_in.video.i_height,
+ p_sys->ti.width, p_sys->ti.height );
+ }
+
+ p_sys->ti.frame_width = p_enc->fmt_in.video.i_width;
+ p_sys->ti.frame_height = p_enc->fmt_in.video.i_height;
+ p_sys->ti.offset_x = 0 /*frame_x_offset*/;
+ p_sys->ti.offset_y = 0 /*frame_y_offset*/;
+
+ p_sys->i_width = p_sys->ti.width;
+ p_sys->i_height = p_sys->ti.height;
+
+ if( !p_enc->fmt_in.video.i_frame_rate ||
+ !p_enc->fmt_in.video.i_frame_rate_base )
+ {
+ p_sys->ti.fps_numerator = 25;
+ p_sys->ti.fps_denominator = 1;
+ }
+ else
+ {
+ p_sys->ti.fps_numerator = p_enc->fmt_in.video.i_frame_rate;
+ p_sys->ti.fps_denominator = p_enc->fmt_in.video.i_frame_rate_base;
+ }
+
+ if( p_enc->fmt_in.video.i_aspect )
+ {
+ uint64_t i_num, i_den;
+ unsigned i_dst_num, i_dst_den;
+
+ i_num = p_enc->fmt_in.video.i_aspect * (int64_t)p_sys->ti.height;
+ i_den = VOUT_ASPECT_FACTOR * p_sys->ti.width;
+ vlc_ureduce( &i_dst_num, &i_dst_den, i_num, i_den, 0 );
+ p_sys->ti.aspect_numerator = i_dst_num;
+ p_sys->ti.aspect_denominator = i_dst_den;
+ }
+ else
+ {
+ p_sys->ti.aspect_numerator = 4;
+ p_sys->ti.aspect_denominator = 3;
+ }
+
+ p_sys->ti.target_bitrate = p_enc->fmt_out.i_bitrate;
+ p_sys->ti.quality = ((float)i_quality) * 6.3;
+
+ p_sys->ti.dropframes_p = 0;
+ p_sys->ti.quick_p = 1;
+ p_sys->ti.keyframe_auto_p = 1;
+ p_sys->ti.keyframe_frequency = 64;
+ p_sys->ti.keyframe_frequency_force = 64;
+ p_sys->ti.keyframe_data_target_bitrate = p_enc->fmt_out.i_bitrate * 1.5;
+ p_sys->ti.keyframe_auto_threshold = 80;
+ p_sys->ti.keyframe_mindistance = 8;
+ p_sys->ti.noise_sensitivity = 1;
+
+ theora_encode_init( &p_sys->td, &p_sys->ti );
+ theora_info_clear( &p_sys->ti );
+ theora_comment_init( &p_sys->tc );
+
+ /* Create and store headers */
+ p_enc->fmt_out.i_extra = 3 * 2;
+ for( i = 0; i < 3; i++ )
+ {
+ if( i == 0 ) theora_encode_header( &p_sys->td, &header );
+ else if( i == 1 ) theora_encode_comment( &p_sys->tc, &header );
+ else if( i == 2 ) theora_encode_tables( &p_sys->td, &header );
+
+ p_enc->fmt_out.p_extra =
+ realloc( p_enc->fmt_out.p_extra,
+ p_enc->fmt_out.i_extra + header.bytes );
+ p_extra = p_enc->fmt_out.p_extra;
+ p_extra += p_enc->fmt_out.i_extra + (i-3)*2;
+ p_enc->fmt_out.i_extra += header.bytes;
+
+ *(p_extra++) = header.bytes >> 8;
+ *(p_extra++) = header.bytes & 0xFF;
+
+ memcpy( p_extra, header.packet, header.bytes );
+ }
+
+ return VLC_SUCCESS;
+}
+
+/****************************************************************************
+ * Encode: the whole thing
+ ****************************************************************************
+ * This function spits out ogg packets.
+ ****************************************************************************/
+static block_t *Encode( encoder_t *p_enc, picture_t *p_pict )
+{
+ encoder_sys_t *p_sys = p_enc->p_sys;
+ ogg_packet oggpacket;
+ block_t *p_block;
+ yuv_buffer yuv;
+ int i;
+
+ /* Sanity check */
+ if( p_pict->p[0].i_pitch < (int)p_sys->i_width ||
+ p_pict->p[0].i_lines < (int)p_sys->i_height )
+ {
+ msg_Warn( p_enc, "frame is smaller than encoding size"
+ "(%ix%i->%ix%i) -> dropping frame",
+ p_pict->p[0].i_pitch, p_pict->p[0].i_lines,
+ p_sys->i_width, p_sys->i_height );
+ return NULL;
+ }
+
+ /* Fill padding */
+ if( p_pict->p[0].i_visible_pitch < (int)p_sys->i_width )
+ {
+ for( i = 0; i < p_sys->i_height; i++ )
+ {
+ memset( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch +
+ p_pict->p[0].i_visible_pitch,
+ *( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch +
+ p_pict->p[0].i_visible_pitch - 1 ),
+ p_sys->i_width - p_pict->p[0].i_visible_pitch );
+ }
+ for( i = 0; i < p_sys->i_height / 2; i++ )
+ {
+ memset( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch +
+ p_pict->p[1].i_visible_pitch,
+ *( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch +
+ p_pict->p[1].i_visible_pitch - 1 ),
+ p_sys->i_width / 2 - p_pict->p[1].i_visible_pitch );
+ memset( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch +
+ p_pict->p[2].i_visible_pitch,
+ *( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch +
+ p_pict->p[2].i_visible_pitch - 1 ),
+ p_sys->i_width / 2 - p_pict->p[2].i_visible_pitch );
+ }
+ }
+
+ if( p_pict->p[0].i_visible_lines < (int)p_sys->i_height )
+ {
+ for( i = p_pict->p[0].i_visible_lines; i < p_sys->i_height; i++ )
+ {
+ memset( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch, 0,
+ p_sys->i_width );
+ }
+ for( i = p_pict->p[1].i_visible_lines; i < p_sys->i_height / 2; i++ )
+ {
+ memset( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch, 0x80,
+ p_sys->i_width / 2 );
+ memset( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch, 0x80,
+ p_sys->i_width / 2 );
+ }
+ }
+
+ /* Theora is a one-frame-in, one-frame-out system. Submit a frame
+ * for compression and pull out the packet. */
+
+ yuv.y_width = p_sys->i_width;
+ yuv.y_height = p_sys->i_height;
+ yuv.y_stride = p_pict->p[0].i_pitch;
+
+ yuv.uv_width = p_sys->i_width / 2;
+ yuv.uv_height = p_sys->i_height / 2;
+ yuv.uv_stride = p_pict->p[1].i_pitch;
+
+ yuv.y = p_pict->p[0].p_pixels;
+ yuv.u = p_pict->p[1].p_pixels;
+ yuv.v = p_pict->p[2].p_pixels;
+
+ if( theora_encode_YUVin( &p_sys->td, &yuv ) < 0 )
+ {
+ msg_Warn( p_enc, "failed encoding a frame" );
+ return NULL;
+ }
+
+ theora_encode_packetout( &p_sys->td, 0, &oggpacket );
+
+ /* Ogg packet to block */
+ p_block = block_New( p_enc, oggpacket.bytes );
+ memcpy( p_block->p_buffer, oggpacket.packet, oggpacket.bytes );
+ p_block->i_dts = p_block->i_pts = p_pict->date;
+
+ return p_block;
+}
+
+/*****************************************************************************
+ * CloseEncoder: theora encoder destruction
+ *****************************************************************************/
+static void CloseEncoder( vlc_object_t *p_this )
+{
+ encoder_t *p_enc = (encoder_t *)p_this;
+ encoder_sys_t *p_sys = p_enc->p_sys;
+
+ theora_info_clear( &p_sys->ti );
+ theora_comment_clear( &p_sys->tc );
+
+ free( p_sys );
+}