Correct edge MC for chroma
[ffmpeg.git] / libavcodec / oggtheora.c
1 /**
2     Copyright (C) 2005  Måns Rullgård
3
4     Permission is hereby granted, free of charge, to any person
5     obtaining a copy of this software and associated documentation
6     files (the "Software"), to deal in the Software without
7     restriction, including without limitation the rights to use, copy,
8     modify, merge, publish, distribute, sublicense, and/or sell copies
9     of the Software, and to permit persons to whom the Software is
10     furnished to do so, subject to the following conditions:
11
12     The above copyright notice and this permission notice shall be
13     included in all copies or substantial portions of the Software.
14
15     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
19     HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
20     WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21     OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22     DEALINGS IN THE SOFTWARE.
23 **/
24
25 #include <stdlib.h>
26 #include <theora/theora.h>
27 #include "avcodec.h"
28
29 typedef struct TheoraContext {
30     theora_info info;
31     theora_state state;
32     theora_comment comment;
33     ogg_packet op;
34 } TheoraContext;
35
36 static int
37 Theora_decode_frame(AVCodecContext *ctx, void *outdata, int *outdata_size,
38                     uint8_t *buf, int buf_size)
39 {
40     TheoraContext *thc = ctx->priv_data;
41     AVFrame *frame = outdata;
42     yuv_buffer yuv;
43
44     thc->op.packet = buf;
45     thc->op.bytes = buf_size;
46
47     if(theora_decode_packetin(&thc->state, &thc->op))
48         return -1;
49
50     theora_decode_YUVout(&thc->state, &yuv);
51
52     frame->data[0] = yuv.y;
53     frame->data[1] = yuv.u;
54     frame->data[2] = yuv.v;
55     frame->linesize[0] = yuv.y_stride;
56     frame->linesize[1] = yuv.uv_stride;
57     frame->linesize[2] = yuv.uv_stride;
58
59     *outdata_size = sizeof(*frame);
60     return buf_size;
61 }
62
63 static int
64 Theora_decode_end(AVCodecContext *ctx)
65 {
66     TheoraContext *thc = ctx->priv_data;
67     theora_info_clear(&thc->info);
68     theora_comment_clear(&thc->comment);
69     return 0;
70 }
71
72 static int
73 Theora_decode_init(AVCodecContext *ctx)
74 {
75     TheoraContext *thc = ctx->priv_data;
76     int size, hs, i;
77     ogg_packet op;
78     uint8_t *cdp;
79
80     if(ctx->extradata_size < 6)
81         return -1;
82
83     theora_info_init(&thc->info);
84
85     memset(&op, 0, sizeof(op));
86     cdp = ctx->extradata;
87     size = ctx->extradata_size;
88
89     for(i = 0; i < 3; i++){
90         hs = *cdp++ << 8;
91         hs += *cdp++;
92         size -= 2;
93
94         if(hs > size){
95             av_log(ctx, AV_LOG_ERROR, "extradata too small: %i > %i\n",
96                    hs, size);
97             return -1;
98         }
99
100         op.packet = cdp;
101         op.bytes = hs;
102         op.b_o_s = !i;
103         if(theora_decode_header(&thc->info, &thc->comment, &op))
104             return -1;
105         op.packetno++;
106
107         cdp += hs;
108         size -= hs;
109     }
110
111     theora_decode_init(&thc->state, &thc->info);
112
113     ctx->width = thc->info.width;
114     ctx->height = thc->info.height;
115     ctx->time_base.num = thc->info.fps_denominator;
116     ctx->time_base.den = thc->info.fps_numerator;
117     ctx->pix_fmt = PIX_FMT_YUV420P; /* FIXME: others are possible */
118
119     return 0;
120 }
121
122 AVCodec oggtheora_decoder = {
123     "theora",
124     CODEC_TYPE_VIDEO,
125     CODEC_ID_THEORA,
126     sizeof(TheoraContext),
127     Theora_decode_init,
128     NULL,
129     Theora_decode_end,
130     Theora_decode_frame,
131     0,
132     NULL
133 };