git.sesse.net Git - ffmpeg/blob - 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 };