libmpcodecs/vd_theora.c

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <stdarg.h>
  22 #include <assert.h>
  23
  24 #include "config.h"
  25 #include "mp_msg.h"
  26 #include "help_mp.h"
  27
  28 #include "vd_internal.h"
  29
  30 static const vd_info_t info = {
  31    "Theora/VP3",
  32    "theora",
  33    "David Kuehling",
  34    "www.theora.org",
  35    "Theora project's VP3 codec"
  36 };
  37
  38 LIBVD_EXTERN(theora)
  39
  40 #include <theora/theora.h>
  41
  42 #define THEORA_NUM_HEADER_PACKETS 3
  43
  44 typedef struct theora_struct_st {
  45     theora_state st;
  46     theora_comment cc;
  47     theora_info inf;
  48 } theora_struct_t;
  49
  50 /** Convert Theora pixelformat to the corresponding IMGFMT_ */
  51 static uint32_t theora_pixelformat2imgfmt(theora_pixelformat fmt){
  52     switch(fmt) {
  53        case OC_PF_420: return IMGFMT_YV12;
  54        case OC_PF_422: return IMGFMT_422P;
  55        case OC_PF_444: return IMGFMT_444P;
  56     }
  57     return 0;
  58 }
  59
  60 // to set/get/query special features/parameters
  61 static int control(sh_video_t *sh,int cmd,void* arg,...){
  62     theora_struct_t *context = sh->context;
  63     switch(cmd) {
  64     case VDCTRL_QUERY_FORMAT:
  65         if (*(int*)arg == theora_pixelformat2imgfmt(context->inf.pixelformat))
  66             return CONTROL_TRUE;
  67         return CONTROL_FALSE;
  68     }
  69
  70     return CONTROL_UNKNOWN;
  71 }
  72
  73 /*
  74  * init driver
  75  */
  76 static int init(sh_video_t *sh){
  77     theora_struct_t *context = NULL;
  78     int errorCode = 0;
  79     ogg_packet op;
  80     int i;
  81
  82     context = calloc (sizeof (theora_struct_t), 1);
  83     sh->context = context;
  84     if (!context)
  85         goto err_out;
  86
  87     theora_info_init(&context->inf);
  88     theora_comment_init(&context->cc);
  89
  90     /* Read all header packets, pass them to theora_decode_header. */
  91     for (i = 0; i < THEORA_NUM_HEADER_PACKETS; i++)
  92     {
  93         op.bytes = ds_get_packet (sh->ds, &op.packet);
  94         op.b_o_s = 1;
  95         if ( (errorCode = theora_decode_header (&context->inf, &context->cc, &op)) )
  96         {
  97             mp_msg(MSGT_DECAUDIO, MSGL_ERR, "Broken Theora header; errorCode=%i!\n", errorCode);
  98             goto err_out;
  99         }
 100     }
 101
 102     /* now init codec */
 103     errorCode = theora_decode_init (&context->st, &context->inf);
 104     if (errorCode)
 105     {
 106         mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Theora decode init failed: %i \n", errorCode);
 107         goto err_out;
 108     }
 109
 110     if(sh->aspect==0.0 && context->inf.aspect_denominator!=0)
 111     {
 112        sh->aspect = ((double)context->inf.aspect_numerator * context->inf.frame_width)/
 113           ((double)context->inf.aspect_denominator * context->inf.frame_height);
 114     }
 115
 116     mp_msg(MSGT_DECVIDEO,MSGL_V,"INFO: Theora video init ok!\n");
 117
 118     return mpcodecs_config_vo (sh,context->inf.frame_width,context->inf.frame_height,theora_pixelformat2imgfmt(context->inf.pixelformat));
 119
 120 err_out:
 121     free(context);
 122     sh->context = NULL;
 123     return 0;
 124 }
 125
 126 /*
 127  * uninit driver
 128  */
 129 static void uninit(sh_video_t *sh)
 130 {
 131    theora_struct_t *context = sh->context;
 132
 133    if (context)
 134    {
 135       theora_info_clear(&context->inf);
 136       theora_comment_clear(&context->cc);
 137       theora_clear (&context->st);
 138       free (context);
 139    }
 140 }
 141
 142 /*
 143  * decode frame
 144  */
 145 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags)
 146 {
 147    theora_struct_t *context = sh->context;
 148    int errorCode = 0;
 149    ogg_packet op;
 150    yuv_buffer yuv;
 151    mp_image_t* mpi;
 152
 153    memset (&op, 0, sizeof (op));
 154    op.bytes = len;
 155    op.packet = data;
 156    op.granulepos = -1;
 157
 158    errorCode = theora_decode_packetin (&context->st, &op);
 159    if (errorCode)
 160    {
 161       mp_msg(MSGT_DECVIDEO,MSGL_ERR,"Theora decode packetin failed: %i \n",
 162              errorCode);
 163       return NULL;
 164    }
 165
 166    errorCode = theora_decode_YUVout (&context->st, &yuv);
 167    if (errorCode)
 168    {
 169       mp_msg(MSGT_DEMUX,MSGL_ERR,"Theora decode YUVout failed: %i \n",
 170              errorCode);
 171       return NULL;
 172    }
 173
 174     mpi = mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, 0, yuv.y_width, yuv.y_height);
 175     if(!mpi) return NULL;
 176
 177     mpi->planes[0]=yuv.y;
 178     mpi->stride[0]=yuv.y_stride;
 179     mpi->planes[1]=yuv.u;
 180     mpi->stride[1]=yuv.uv_stride;
 181     mpi->planes[2]=yuv.v;
 182     mpi->stride[2]=yuv.uv_stride;
 183
 184     return mpi;
 185 }