annotate libtheoraenc.c @ 7916:2a4c42e6aecb libavcodec

Cosmetics: move convolve() together with the other DSP functions
author vitor
date Wed, 24 Sep 2008 20:05:22 +0000
parents e943e1409077
children d679fd3a5359
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
1 /*
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
2 * Copyright (c) 2006 Paul Richards <paul.richards@gmail.com>
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
3 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
4 * This file is part of FFmpeg.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
5 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
6 * FFmpeg is free software; you can redistribute it and/or
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
7 * modify it under the terms of the GNU Lesser General Public
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
8 * License as published by the Free Software Foundation; either
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
9 * version 2.1 of the License, or (at your option) any later version.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
10 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
11 * FFmpeg is distributed in the hope that it will be useful,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
14 * Lesser General Public License for more details.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
15 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
16 * You should have received a copy of the GNU Lesser General Public
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
17 * License along with FFmpeg; if not, write to the Free Software
5215
2b72f9bc4f06 license header consistency cosmetics
diego
parents: 5089
diff changeset
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
19 */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
20
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
21 /*!
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
22 * \file theoraenc.c
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
23 * \brief Theora encoder using libtheora.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
24 * \author Paul Richards <paul.richards@gmail.com>
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
25 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
26 * A lot of this is copy / paste from other output codecs in
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
27 * libavcodec or pure guesswork (or both).
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
28 *
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
29 * I have used t_ prefixes on variables which are libtheora types
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
30 * and o_ prefixes on variables which are libogg types.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
31 */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
32
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
33 /* FFmpeg includes */
6763
f7cbb7733146 Use full path for #includes from another directory.
diego
parents: 6712
diff changeset
34 #include "libavutil/log.h"
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
35 #include "avcodec.h"
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
36
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
37 /* libtheora includes */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
38 #include <theora/theora.h>
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
39
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
40 typedef struct TheoraContext{
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
41 theora_state t_state;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
42 } TheoraContext;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
43
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
44 /*!
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
45 Concatenates an ogg_packet into the extradata.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
46 */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
47 static int concatenate_packet(unsigned int* offset, AVCodecContext* avc_context, const ogg_packet* packet)
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
48 {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
49 char* message = NULL;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
50 uint8_t* newdata = NULL;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
51 int newsize = avc_context->extradata_size + 2 + packet->bytes;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
52
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
53 if (packet->bytes < 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
54 message = "ogg_packet has negative size";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
55 } else if (packet->bytes > 0xffff) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
56 message = "ogg_packet is larger than 65535 bytes";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
57 } else if (newsize < avc_context->extradata_size) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
58 message = "extradata_size would overflow";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
59 } else {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
60 newdata = av_realloc(avc_context->extradata, newsize);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
61 if (newdata == NULL) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
62 message = "av_realloc failed";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
63 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
64 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
65 if (message != NULL) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
66 av_log(avc_context, AV_LOG_ERROR, "concatenate_packet failed: %s\n", message);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
67 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
68 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
69
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
70 avc_context->extradata = newdata;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
71 avc_context->extradata_size = newsize;
5089
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4496
diff changeset
72 AV_WB16(avc_context->extradata + (*offset), packet->bytes);
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4496
diff changeset
73 *offset += 2;
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
74 memcpy( avc_context->extradata + (*offset), packet->packet, packet->bytes );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
75 (*offset) += packet->bytes;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
76 return 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
77 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
78
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
79 static int encode_init(AVCodecContext* avc_context)
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
80 {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
81 theora_info t_info;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
82 theora_comment t_comment;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
83 ogg_packet o_packet;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
84 unsigned int offset;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
85 TheoraContext *h = avc_context->priv_data;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
86
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
87 /* Set up the theora_info struct */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
88 theora_info_init( &t_info );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
89 t_info.width = avc_context->width;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
90 t_info.height = avc_context->height;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
91 t_info.frame_width = avc_context->width;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
92 t_info.frame_height = avc_context->height;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
93 t_info.offset_x = 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
94 t_info.offset_y = 0;
4496
a02a0d06e99b Add a comment about swapped numerator and denominator.
diego
parents: 4403
diff changeset
95 /* Swap numerator and denominator as time_base in AVCodecContext gives the
a02a0d06e99b Add a comment about swapped numerator and denominator.
diego
parents: 4403
diff changeset
96 * time period between frames, but theora_info needs the framerate. */
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
97 t_info.fps_numerator = avc_context->time_base.den;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
98 t_info.fps_denominator = avc_context->time_base.num;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
99 if (avc_context->sample_aspect_ratio.num != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
100 t_info.aspect_numerator = avc_context->sample_aspect_ratio.num;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
101 t_info.aspect_denominator = avc_context->sample_aspect_ratio.den;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
102 } else {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
103 t_info.aspect_numerator = 1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
104 t_info.aspect_denominator = 1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
105 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
106 t_info.colorspace = OC_CS_UNSPECIFIED;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
107 t_info.pixelformat = OC_PF_420;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
108 t_info.target_bitrate = avc_context->bit_rate;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
109 t_info.keyframe_frequency = avc_context->gop_size;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
110 t_info.keyframe_frequency_force = avc_context->gop_size;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
111 t_info.keyframe_mindistance = avc_context->keyint_min;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
112 t_info.quality = 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
113
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
114 t_info.quick_p = 1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
115 t_info.dropframes_p = 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
116 t_info.keyframe_auto_p = 1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
117 t_info.keyframe_data_target_bitrate = t_info.target_bitrate * 1.5;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
118 t_info.keyframe_auto_threshold = 80;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
119 t_info.noise_sensitivity = 1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
120 t_info.sharpness = 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
121
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
122 /* Now initialise libtheora */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
123 if (theora_encode_init( &(h->t_state), &t_info ) != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
124 av_log(avc_context, AV_LOG_ERROR, "theora_encode_init failed\n");
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
125 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
126 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
127
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
128 /* Clear up theora_info struct */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
129 theora_info_clear( &t_info );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
130
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
131 /*
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
132 Output first header packet consisting of theora
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
133 header, comment, and tables.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
134
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
135 Each one is prefixed with a 16bit size, then they
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
136 are concatenated together into ffmpeg's extradata.
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
137 */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
138 offset = 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
139
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
140 /* Header */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
141 theora_encode_header( &(h->t_state), &o_packet );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
142 if (concatenate_packet( &offset, avc_context, &o_packet ) != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
143 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
144 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
145
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
146 /* Comment */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
147 theora_comment_init( &t_comment );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
148 theora_encode_comment( &t_comment, &o_packet );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
149 if (concatenate_packet( &offset, avc_context, &o_packet ) != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
150 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
151 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
152
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
153 /* Tables */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
154 theora_encode_tables( &(h->t_state), &o_packet );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
155 if (concatenate_packet( &offset, avc_context, &o_packet ) != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
156 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
157 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
158
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
159 /* Clear up theora_comment struct */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
160 theora_comment_clear( &t_comment );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
161
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
162 /* Set up the output AVFrame */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
163 avc_context->coded_frame= avcodec_alloc_frame();
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
164
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
165 return 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
166 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
167
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
168 static int encode_frame(
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
169 AVCodecContext* avc_context,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
170 uint8_t *outbuf,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
171 int buf_size,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
172 void *data)
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
173 {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
174 yuv_buffer t_yuv_buffer;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
175 TheoraContext *h = avc_context->priv_data;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
176 AVFrame *frame = data;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
177 ogg_packet o_packet;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
178 int result;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
179
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
180 assert(avc_context->pix_fmt == PIX_FMT_YUV420P);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
181
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
182 /* Copy planes to the theora yuv_buffer */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
183 if (frame->linesize[1] != frame->linesize[2]) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
184 av_log(avc_context, AV_LOG_ERROR, "U and V stride differ\n");
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
185 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
186 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
187
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
188 t_yuv_buffer.y_width = avc_context->width;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
189 t_yuv_buffer.y_height = avc_context->height;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
190 t_yuv_buffer.y_stride = frame->linesize[0];
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
191 t_yuv_buffer.uv_width = t_yuv_buffer.y_width / 2;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
192 t_yuv_buffer.uv_height = t_yuv_buffer.y_height / 2;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
193 t_yuv_buffer.uv_stride = frame->linesize[1];
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
194
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
195 t_yuv_buffer.y = frame->data[0];
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
196 t_yuv_buffer.u = frame->data[1];
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
197 t_yuv_buffer.v = frame->data[2];
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
198
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
199 /* Now call into theora_encode_YUVin */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
200 result = theora_encode_YUVin( &(h->t_state), &t_yuv_buffer );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
201 if (result != 0) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
202 const char* message;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
203 switch (result) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
204 case -1:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
205 message = "differing frame sizes";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
206 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
207 case OC_EINVAL:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
208 message = "encoder is not ready or is finished";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
209 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
210 default:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
211 message = "unknown reason";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
212 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
213 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
214 av_log(avc_context, AV_LOG_ERROR, "theora_encode_YUVin failed (%s) [%d]\n", message, result);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
215 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
216 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
217
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
218 /* Pick up returned ogg_packet */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
219 result = theora_encode_packetout( &(h->t_state), 0, &o_packet );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
220 switch (result) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
221 case 0:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
222 /* No packet is ready */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
223 return 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
224 case 1:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
225 /* Success, we have a packet */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
226 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
227 default:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
228 av_log(avc_context, AV_LOG_ERROR, "theora_encode_packetout failed [%d]\n", result);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
229 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
230 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
231
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
232 /* Copy ogg_packet content out to buffer */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
233 if (buf_size < o_packet.bytes) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
234 av_log(avc_context, AV_LOG_ERROR, "encoded frame too large\n");
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
235 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
236 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
237 memcpy(outbuf, o_packet.packet, o_packet.bytes);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
238
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
239 return o_packet.bytes;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
240 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
241
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
242 static int encode_close(AVCodecContext* avc_context)
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
243 {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
244 ogg_packet o_packet;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
245 TheoraContext *h = avc_context->priv_data;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
246 int result;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
247 const char* message;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
248
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
249 result = theora_encode_packetout( &(h->t_state), 1, &o_packet );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
250 theora_clear( &(h->t_state) );
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
251 switch (result) {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
252 case 0:/* No packet is ready */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
253 case -1:/* Encoding finished */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
254 return 0;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
255 case 1:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
256 /* We have a packet */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
257 message = "gave us a packet";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
258 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
259 default:
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
260 message = "unknown reason";
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
261 break;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
262 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
263 av_log(avc_context, AV_LOG_ERROR, "theora_encode_packetout failed (%s) [%d]\n", message, result);
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
264 return -1;
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
265 }
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
266
6788
e1302edb0f69 Replace some occurrences of -1 with PIX_FMT_NONE.
cehoyos
parents: 6763
diff changeset
267 static const enum PixelFormat supported_pixel_formats[] = { PIX_FMT_YUV420P, PIX_FMT_NONE };
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
268
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
269 /*! AVCodec struct exposed to libavcodec */
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
270 AVCodec libtheora_encoder =
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
271 {
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
272 .name = "libtheora",
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
273 .type = CODEC_TYPE_VIDEO,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
274 .id = CODEC_ID_THEORA,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
275 .priv_data_size = sizeof(TheoraContext),
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
276 .init = encode_init,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
277 .close = encode_close,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
278 .encode = encode_frame,
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
279 .pix_fmts = supported_pixel_formats,
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 6788
diff changeset
280 .long_name = NULL_IF_CONFIG_SMALL("libtheora Theora"),
4403
432d332b7def Theora encoding via libtheora.
diego
parents:
diff changeset
281 };