libmpcodecs/vd_theora.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <stdarg.h>
   4 #include <assert.h>
   5
   6 #include "config.h"
   7 #include "mp_msg.h"
   8 #include "help_mp.h"
   9
  10 #ifdef HAVE_OGGTHEORA
  11
  12 #include "vd_internal.h"
  13
  14 static vd_info_t info = {
  15    "Theora/VP3",
  16    "theora",
  17    "David Kuehling",
  18    "www.theora.org",
  19    "Theora project's VP3 codec"
  20 };
  21
  22 LIBVD_EXTERN(theora)
  23
  24 #include <theora/theora.h>
  25
  26 #define THEORA_NUM_HEADER_PACKETS 3
  27
  28 // to set/get/query special features/parameters
  29 static int control(sh_video_t *sh,int cmd,void* arg,...){
  30     switch(cmd) {
  31     case VDCTRL_QUERY_FORMAT:
  32         if ((*((int*)arg)) == IMGFMT_YV12)
  33             return CONTROL_TRUE;
  34         return CONTROL_FALSE;
  35     }
  36
  37     return CONTROL_UNKNOWN;
  38 }
  39
  40 typedef struct theora_struct_st {
  41     theora_state st;
  42     theora_comment cc;
  43     theora_info inf;
  44 } theora_struct_t;
  45
  46 /*
  47  * init driver
  48  */
  49 static int init(sh_video_t *sh){
  50     theora_struct_t *context = NULL;
  51     int failed = 1;
  52     int errorCode = 0;
  53     ogg_packet op;
  54     int i;
  55
  56     /* check whether video output format is supported */
  57     switch(sh->codec->outfmt[sh->outfmtidx])
  58     {
  59        case IMGFMT_YV12: /* well, this should work... */ break;
  60        default:
  61           mp_msg (MSGT_DECVIDEO,MSGL_ERR,"Unsupported out_fmt: 0x%X\n",
  62                   sh->codec->outfmt[sh->outfmtidx]);
  63           return 0;
  64     }
  65
  66     /* this is not a loop, just a context, from which we can break on error */
  67     do
  68     {
  69        context = (theora_struct_t *)calloc (sizeof (theora_struct_t), 1);
  70        sh->context = context;
  71        if (!context)
  72           break;
  73
  74        theora_info_init(&context->inf);
  75        theora_comment_init(&context->cc);
  76
  77        /* Read all header packets, pass them to theora_decode_header. */
  78        for (i = 0; i < THEORA_NUM_HEADER_PACKETS; i++)
  79        {
  80           op.bytes = ds_get_packet (sh->ds, &op.packet);
  81           op.b_o_s = 1;
  82           if ( (errorCode = theora_decode_header (&context->inf, &context->cc, &op)) )
  83           {
  84             mp_msg(MSGT_DECAUDIO, MSGL_ERR, "Broken Theora header; errorCode=%i!\n", errorCode);
  85             break;
  86           }
  87        }
  88        if (errorCode)
  89           break;
  90
  91        /* now init codec */
  92        errorCode = theora_decode_init (&context->st, &context->inf);
  93        if (errorCode)
  94        {
  95           mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Theora decode init failed: %i \n",
  96                  errorCode);
  97           break;
  98        }
  99        failed = 0;
 100     } while (0);
 101
 102     if (failed)
 103     {
 104        if (context)
 105        {
 106           free (context);
 107           sh->context = NULL;
 108        }
 109        return 0;
 110     }
 111
 112     if(sh->aspect==0.0 && context->inf.aspect_denominator!=0)
 113     {
 114        sh->aspect = (float)(context->inf.aspect_numerator * context->inf.frame_width)/
 115           (context->inf.aspect_denominator * context->inf.frame_height);
 116     }
 117
 118     mp_msg(MSGT_DECVIDEO,MSGL_V,"INFO: Theora video init ok!\n");
 119
 120     return mpcodecs_config_vo (sh,context->inf.frame_width,context->inf.frame_height,IMGFMT_YV12);
 121 }
 122
 123 /*
 124  * uninit driver
 125  */
 126 static void uninit(sh_video_t *sh)
 127 {
 128    theora_struct_t *context = (theora_struct_t *)sh->context;
 129
 130    if (context)
 131    {
 132       theora_clear (&context->st);
 133       free (context);
 134    }
 135 }
 136
 137 /*
 138  * decode frame
 139  */
 140 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags)
 141 {
 142    theora_struct_t *context = (theora_struct_t *)sh->context;
 143    int errorCode = 0;
 144    ogg_packet op;
 145    yuv_buffer yuv;
 146    mp_image_t* mpi;
 147    int i;
 148
 149    memset (&op, 0, sizeof (op));
 150    op.bytes = len;
 151    op.packet = data;
 152    op.granulepos = -1;
 153
 154    errorCode = theora_decode_packetin (&context->st, &op);
 155    if (errorCode)
 156    {
 157       mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Theora decode packetin failed: %i \n",
 158              errorCode);
 159       return NULL;
 160    }
 161
 162    errorCode = theora_decode_YUVout (&context->st, &yuv);
 163    if (errorCode)
 164    {
 165       mp_msg(MSGT_DEMUX,MSGL_ERR,"Theora decode YUVout failed: %i \n",
 166              errorCode);
 167       return 0;
 168    }
 169
 170     mpi = mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, 0, yuv.y_width, yuv.y_height);
 171     if(!mpi) return NULL;
 172
 173     mpi->planes[0]=yuv.y;
 174     mpi->stride[0]=yuv.y_stride;
 175     mpi->planes[1]=yuv.u;
 176     mpi->stride[1]=yuv.uv_stride;
 177     mpi->planes[2]=yuv.v;
 178     mpi->stride[2]=yuv.uv_stride;
 179
 180     return mpi;
 181 }
 182
 183 #endif