Update libtheora wrapper to use the 1.0 API

Originally committed as revision 20580 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
David Conrad 2009-11-22 21:08:37 +00:00
parent 9c5a9e6b48
commit bdc8c48893
2 changed files with 42 additions and 66 deletions

2
configure vendored
View File

@ -2375,7 +2375,7 @@ enabled libopenjpeg && require libopenjpeg openjpeg.h opj_version -lopenjpeg
enabled libschroedinger && add_cflags $(pkg-config --cflags schroedinger-1.0) && enabled libschroedinger && add_cflags $(pkg-config --cflags schroedinger-1.0) &&
require libschroedinger schroedinger/schro.h schro_init $(pkg-config --libs schroedinger-1.0) require libschroedinger schroedinger/schro.h schro_init $(pkg-config --libs schroedinger-1.0)
enabled libspeex && require libspeex speex/speex.h speex_decoder_init -lspeex enabled libspeex && require libspeex speex/speex.h speex_decoder_init -lspeex
enabled libtheora && require libtheora theora/theora.h theora_info_init -ltheora -logg enabled libtheora && require libtheora theora/theoraenc.h th_info_init -ltheora -logg
enabled libvorbis && require libvorbis vorbis/vorbisenc.h vorbis_info_init -lvorbisenc -lvorbis -logg enabled libvorbis && require libvorbis vorbis/vorbisenc.h vorbis_info_init -lvorbisenc -lvorbis -logg
enabled libx264 && require libx264 x264.h x264_encoder_encode -lx264 -lm && enabled libx264 && require libx264 x264.h x264_encoder_encode -lx264 -lm &&
{ check_cpp_condition x264.h "X264_BUILD >= 78" || { check_cpp_condition x264.h "X264_BUILD >= 78" ||

View File

@ -36,10 +36,10 @@
#include "avcodec.h" #include "avcodec.h"
/* libtheora includes */ /* libtheora includes */
#include <theora/theora.h> #include <theora/theoraenc.h>
typedef struct TheoraContext { typedef struct TheoraContext {
theora_state t_state; th_enc_ctx *t_state;
} TheoraContext; } TheoraContext;
/*! /*!
@ -80,20 +80,21 @@ static int concatenate_packet(unsigned int* offset,
static av_cold int encode_init(AVCodecContext* avc_context) static av_cold int encode_init(AVCodecContext* avc_context)
{ {
theora_info t_info; th_info t_info;
theora_comment t_comment; th_comment t_comment;
ogg_packet o_packet; ogg_packet o_packet;
unsigned int offset; unsigned int offset;
TheoraContext *h = avc_context->priv_data; TheoraContext *h = avc_context->priv_data;
uint32_t gop_size = avc_context->gop_size;
/* Set up the theora_info struct */ /* Set up the theora_info struct */
theora_info_init(&t_info); th_info_init(&t_info);
t_info.width = FFALIGN(avc_context->width, 16); t_info.frame_width = FFALIGN(avc_context->width, 16);
t_info.height = FFALIGN(avc_context->height, 16); t_info.frame_height = FFALIGN(avc_context->height, 16);
t_info.frame_width = avc_context->width; t_info.pic_width = avc_context->width;
t_info.frame_height = avc_context->height; t_info.pic_height = avc_context->height;
t_info.offset_x = 0; t_info.pic_x = 0;
t_info.offset_y = avc_context->height & 0xf; t_info.pic_y = 0;
/* Swap numerator and denominator as time_base in AVCodecContext gives the /* Swap numerator and denominator as time_base in AVCodecContext gives the
* time period between frames, but theora_info needs the framerate. */ * time period between frames, but theora_info needs the framerate. */
t_info.fps_numerator = avc_context->time_base.den; t_info.fps_numerator = avc_context->time_base.den;
@ -105,19 +106,8 @@ static av_cold int encode_init(AVCodecContext* avc_context)
t_info.aspect_numerator = 1; t_info.aspect_numerator = 1;
t_info.aspect_denominator = 1; t_info.aspect_denominator = 1;
} }
t_info.colorspace = OC_CS_UNSPECIFIED; t_info.colorspace = TH_CS_UNSPECIFIED;
t_info.pixelformat = OC_PF_420; t_info.pixel_fmt = TH_PF_420;
t_info.keyframe_frequency = avc_context->gop_size;
t_info.keyframe_frequency_force = avc_context->gop_size;
t_info.keyframe_mindistance = avc_context->keyint_min;
t_info.quick_p = 1;
t_info.dropframes_p = 0;
t_info.keyframe_auto_p = 1;
t_info.keyframe_data_target_bitrate = t_info.target_bitrate * 1.5;
t_info.keyframe_auto_threshold = 80;
t_info.noise_sensitivity = 1;
t_info.sharpness = 0;
if (avc_context->flags & CODEC_FLAG_QSCALE) { if (avc_context->flags & CODEC_FLAG_QSCALE) {
/* to be constant with the libvorbis implementation, clip global_quality to 0 - 10 /* to be constant with the libvorbis implementation, clip global_quality to 0 - 10
@ -133,13 +123,20 @@ static av_cold int encode_init(AVCodecContext* avc_context)
} }
/* Now initialise libtheora */ /* Now initialise libtheora */
if (theora_encode_init(&(h->t_state), &t_info)) { h->t_state = th_encode_alloc(&t_info);
if (!h->t_state) {
av_log(avc_context, AV_LOG_ERROR, "theora_encode_init failed\n"); av_log(avc_context, AV_LOG_ERROR, "theora_encode_init failed\n");
return -1; return -1;
} }
/* Clear up theora_info struct */ /* Clear up theora_info struct */
theora_info_clear(&t_info); th_info_clear(&t_info);
if (th_encode_ctl(h->t_state, TH_ENCCTL_SET_KEYFRAME_FREQUENCY_FORCE,
&gop_size, sizeof(gop_size))) {
av_log(avc_context, AV_LOG_ERROR, "Error setting GOP size\n");
return -1;
}
/* /*
Output first header packet consisting of theora Output first header packet consisting of theora
@ -150,26 +147,14 @@ static av_cold int encode_init(AVCodecContext* avc_context)
*/ */
offset = 0; offset = 0;
/* Header */ /* Headers */
theora_encode_header(&(h->t_state), &o_packet); th_comment_init(&t_comment);
if (concatenate_packet(&offset, avc_context, &o_packet))
return -1;
/* Comment */ while (th_encode_flushheader(h->t_state, &t_comment, &o_packet))
theora_comment_init(&t_comment); if (concatenate_packet(&offset, avc_context, &o_packet))
theora_encode_comment(&t_comment, &o_packet); return -1;
if (concatenate_packet(&offset, avc_context, &o_packet))
return -1;
/* Clear up theora_comment struct before we reset the packet */
theora_comment_clear(&t_comment);
/* And despite documentation to the contrary, theora_comment_clear
* does not release the packet */
ogg_packet_clear(&o_packet);
/* Tables */ th_comment_clear(&t_comment);
theora_encode_tables(&(h->t_state), &o_packet);
if (concatenate_packet(&offset, avc_context, &o_packet))
return -1;
/* Set up the output AVFrame */ /* Set up the output AVFrame */
avc_context->coded_frame= avcodec_alloc_frame(); avc_context->coded_frame= avcodec_alloc_frame();
@ -180,40 +165,31 @@ static av_cold int encode_init(AVCodecContext* avc_context)
static int encode_frame(AVCodecContext* avc_context, uint8_t *outbuf, static int encode_frame(AVCodecContext* avc_context, uint8_t *outbuf,
int buf_size, void *data) int buf_size, void *data)
{ {
yuv_buffer t_yuv_buffer; th_ycbcr_buffer t_yuv_buffer;
TheoraContext *h = avc_context->priv_data; TheoraContext *h = avc_context->priv_data;
AVFrame *frame = data; AVFrame *frame = data;
ogg_packet o_packet; ogg_packet o_packet;
int result; int result, i;
assert(avc_context->pix_fmt == PIX_FMT_YUV420P); assert(avc_context->pix_fmt == PIX_FMT_YUV420P);
/* Copy planes to the theora yuv_buffer */ /* Copy planes to the theora yuv_buffer */
if (frame->linesize[1] != frame->linesize[2]) { for (i = 0; i < 3; i++) {
av_log(avc_context, AV_LOG_ERROR, "U and V stride differ\n"); t_yuv_buffer[i].width = FFALIGN(avc_context->width, 16) >> !!i;
return -1; t_yuv_buffer[i].height = FFALIGN(avc_context->height, 16) >> !!i;
t_yuv_buffer[i].stride = frame->linesize[i];
t_yuv_buffer[i].data = frame->data[i];
} }
t_yuv_buffer.y_width = FFALIGN(avc_context->width, 16);
t_yuv_buffer.y_height = FFALIGN(avc_context->height, 16);
t_yuv_buffer.y_stride = frame->linesize[0];
t_yuv_buffer.uv_width = t_yuv_buffer.y_width / 2;
t_yuv_buffer.uv_height = t_yuv_buffer.y_height / 2;
t_yuv_buffer.uv_stride = frame->linesize[1];
t_yuv_buffer.y = frame->data[0];
t_yuv_buffer.u = frame->data[1];
t_yuv_buffer.v = frame->data[2];
/* Now call into theora_encode_YUVin */ /* Now call into theora_encode_YUVin */
result = theora_encode_YUVin(&(h->t_state), &t_yuv_buffer); result = th_encode_ycbcr_in(h->t_state, t_yuv_buffer);
if (result) { if (result) {
const char* message; const char* message;
switch (result) { switch (result) {
case -1: case -1:
message = "differing frame sizes"; message = "differing frame sizes";
break; break;
case OC_EINVAL: case TH_EINVAL:
message = "encoder is not ready or is finished"; message = "encoder is not ready or is finished";
break; break;
default: default:
@ -225,7 +201,7 @@ static int encode_frame(AVCodecContext* avc_context, uint8_t *outbuf,
} }
/* Pick up returned ogg_packet */ /* Pick up returned ogg_packet */
result = theora_encode_packetout(&(h->t_state), 0, &o_packet); result = th_encode_packetout(h->t_state, 0, &o_packet);
switch (result) { switch (result) {
case 0: case 0:
/* No packet is ready */ /* No packet is ready */
@ -258,8 +234,8 @@ static av_cold int encode_close(AVCodecContext* avc_context)
int result; int result;
const char* message; const char* message;
result = theora_encode_packetout(&(h->t_state), 1, &o_packet); result = th_encode_packetout(h->t_state, 1, &o_packet);
theora_clear(&(h->t_state)); th_encode_free(h->t_state);
av_freep(&avc_context->coded_frame); av_freep(&avc_context->coded_frame);
av_freep(&avc_context->extradata); av_freep(&avc_context->extradata);
avc_context->extradata_size = 0; avc_context->extradata_size = 0;