libmpcodecs/vd_libmpeg2.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
  22 #include "config.h"
  23 #include "options.h"
  24 #include "mp_msg.h"
  25
  26 #include "vd_internal.h"
  27
  28 //#undef MPEG12_POSTPROC
  29
  30 static const vd_info_t info =
  31 {
  32         "libmpeg2 MPEG 1/2 Video decoder",
  33         "libmpeg2",
  34         "A'rpi & Fabian Franz",
  35         "Aaron & Walken",
  36         "native"
  37 };
  38
  39 LIBVD_EXTERN(libmpeg2)
  40
  41 //#include "libvo/video_out.h"  // FIXME!!!
  42
  43 #include "libmpeg2/mpeg2.h"
  44 #include "libmpeg2/attributes.h"
  45 #include "libmpeg2/mpeg2_internal.h"
  46
  47 #include "cpudetect.h"
  48
  49 typedef struct {
  50     mpeg2dec_t *mpeg2dec;
  51     int quant_store_idx;
  52     char *quant_store[3];
  53     int imgfmt;
  54     int width;
  55     int height;
  56     double aspect;
  57 } vd_libmpeg2_ctx_t;
  58
  59 // to set/get/query special features/parameters
  60 static int control(sh_video_t *sh,int cmd,void* arg,...){
  61     vd_libmpeg2_ctx_t *context = sh->context;
  62     mpeg2dec_t * mpeg2dec = context->mpeg2dec;
  63     const mpeg2_info_t * info = mpeg2_info (mpeg2dec);
  64
  65     switch(cmd) {
  66     case VDCTRL_QUERY_FORMAT:
  67         if (info->sequence->width >> 1 == info->sequence->chroma_width &&
  68             info->sequence->height >> 1 == info->sequence->chroma_height &&
  69             (*((int*)arg)) == IMGFMT_YV12)
  70             return CONTROL_TRUE;
  71         if (info->sequence->width >> 1 == info->sequence->chroma_width &&
  72                 info->sequence->height == info->sequence->chroma_height &&
  73             (*((int*)arg)) == IMGFMT_422P)
  74             return CONTROL_TRUE;
  75         return CONTROL_FALSE;
  76     }
  77
  78     return CONTROL_UNKNOWN;
  79 }
  80
  81 // init driver
  82 static int init(sh_video_t *sh){
  83     vd_libmpeg2_ctx_t *context;
  84     mpeg2dec_t * mpeg2dec;
  85 //    const mpeg2_info_t * info;
  86     int accel;
  87
  88     accel = 0;
  89     if(gCpuCaps.hasMMX)
  90        accel |= MPEG2_ACCEL_X86_MMX;
  91     if(gCpuCaps.hasMMX2)
  92        accel |= MPEG2_ACCEL_X86_MMXEXT;
  93     if(gCpuCaps.has3DNow)
  94        accel |= MPEG2_ACCEL_X86_3DNOW;
  95     if(gCpuCaps.hasSSE2)
  96        accel |= MPEG2_ACCEL_X86_SSE2;
  97     if(gCpuCaps.hasAltiVec)
  98        accel |= MPEG2_ACCEL_PPC_ALTIVEC;
  99     #if ARCH_ALPHA
 100        accel |= MPEG2_ACCEL_ALPHA;
 101     #elif ARCH_ARM
 102        accel |= MPEG2_ACCEL_ARM;
 103     #endif
 104     #if HAVE_MVI
 105        accel |= MPEG2_ACCEL_ALPHA_MVI;
 106     #elif HAVE_VIS
 107        accel |= MPEG2_ACCEL_SPARC_VIS;
 108     #endif
 109     mpeg2_accel(accel);
 110
 111     mpeg2dec = mpeg2_init ();
 112
 113     if(!mpeg2dec) return 0;
 114
 115     mpeg2_custom_fbuf(mpeg2dec,1); // enable DR1
 116
 117     context = calloc(1, sizeof(vd_libmpeg2_ctx_t));
 118     context->mpeg2dec = mpeg2dec;
 119     sh->context = context;
 120
 121     mpeg2dec->pending_buffer = 0;
 122     mpeg2dec->pending_length = 0;
 123
 124     return 1;
 125 }
 126
 127 // uninit driver
 128 static void uninit(sh_video_t *sh){
 129     int i;
 130     vd_libmpeg2_ctx_t *context = sh->context;
 131     mpeg2dec_t * mpeg2dec = context->mpeg2dec;
 132     if (mpeg2dec->pending_buffer) free(mpeg2dec->pending_buffer);
 133     mpeg2dec->decoder.convert=NULL;
 134     mpeg2dec->decoder.convert_id=NULL;
 135     mpeg2_close (mpeg2dec);
 136     for (i=0; i < 3; i++)
 137         free(context->quant_store[i]);
 138     free(sh->context);
 139 }
 140
 141 static void draw_slice (void * _sh, uint8_t * const * src, unsigned int y){
 142     sh_video_t* sh = (sh_video_t*) _sh;
 143     vd_libmpeg2_ctx_t *context = sh->context;
 144     mpeg2dec_t* mpeg2dec = context->mpeg2dec;
 145     const mpeg2_info_t * info = mpeg2_info (mpeg2dec);
 146     int stride[MP_MAX_PLANES] = {mpeg2dec->decoder.stride, mpeg2dec->decoder.uv_stride, mpeg2dec->decoder.uv_stride};
 147     uint8_t *srcs[MP_MAX_PLANES] = {src[0], src[1], src[2]};
 148
 149 //  printf("draw_slice() y=%d  \n",y);
 150
 151     mpcodecs_draw_slice(sh, srcs,
 152                 stride, info->sequence->picture_width,
 153                 (y+16<=info->sequence->picture_height) ? 16 :
 154                     info->sequence->picture_height-y,
 155                 0, y);
 156 }
 157
 158 // decode a frame
 159 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){
 160     vd_libmpeg2_ctx_t *context = sh->context;
 161     mpeg2dec_t * mpeg2dec = context->mpeg2dec;
 162     const mpeg2_info_t * info = mpeg2_info (mpeg2dec);
 163     int drop_frame, framedrop=flags&3;
 164
 165     // MPlayer registers its own draw_slice callback, prevent libmpeg2 from freeing the context
 166     mpeg2dec->decoder.convert=NULL;
 167     mpeg2dec->decoder.convert_id=NULL;
 168
 169     if(len<=0) return NULL; // skipped null frame
 170
 171     // append extra 'end of frame' code:
 172     ((char*)data+len)[0]=0;
 173     ((char*)data+len)[1]=0;
 174     ((char*)data+len)[2]=1;
 175     ((char*)data+len)[3]=0xff;
 176     len+=4;
 177
 178     if (mpeg2dec->pending_length) {
 179         mpeg2_buffer (mpeg2dec, mpeg2dec->pending_buffer, mpeg2dec->pending_buffer + mpeg2dec->pending_length);
 180     } else {
 181         mpeg2_buffer (mpeg2dec, data, (uint8_t *)data+len);
 182     }
 183
 184     while(1){
 185         int state=mpeg2_parse (mpeg2dec);
 186         int type, use_callback;
 187         mp_image_t* mpi_new;
 188         unsigned long pw, ph;
 189         int imgfmt;
 190
 191         switch(state){
 192         case STATE_BUFFER:
 193             if (mpeg2dec->pending_length) {
 194                 // just finished the pending data, continue with processing of the passed buffer
 195                 mpeg2dec->pending_length = 0;
 196                 mpeg2_buffer (mpeg2dec, data, (uint8_t *)data+len);
 197             } else {
 198                 // parsing of the passed buffer finished, return.
 199                 return 0;
 200             }
 201             break;
 202         case STATE_SEQUENCE:
 203             pw = info->sequence->display_width * info->sequence->pixel_width;
 204             ph = info->sequence->display_height * info->sequence->pixel_height;
 205             if(ph) sh->aspect = (float) pw / (float) ph;
 206             // video parameters initialized/changed, (re)init libvo:
 207             if (info->sequence->width >> 1 == info->sequence->chroma_width &&
 208                 info->sequence->height >> 1 == info->sequence->chroma_height) {
 209                 imgfmt = IMGFMT_YV12;
 210             } else if (info->sequence->width >> 1 == info->sequence->chroma_width &&
 211                 info->sequence->height == info->sequence->chroma_height) {
 212                 imgfmt = IMGFMT_422P;
 213             } else return 0;
 214             if (imgfmt == context->imgfmt &&
 215                 info->sequence->picture_width == context->width &&
 216                 info->sequence->picture_height == context->height &&
 217                 sh->aspect == context->aspect)
 218                 break;
 219             if(!mpcodecs_config_vo(sh,
 220                   info->sequence->picture_width,
 221                   info->sequence->picture_height, imgfmt))
 222                 return 0;
 223             context->imgfmt = imgfmt;
 224             context->width = info->sequence->picture_width;
 225             context->height = info->sequence->picture_height;
 226             context->aspect = sh->aspect;
 227             break;
 228         case STATE_PICTURE:
 229             type=info->current_picture->flags&PIC_MASK_CODING_TYPE;
 230
 231             drop_frame = framedrop && (mpeg2dec->decoder.coding_type == B_TYPE);
 232             drop_frame |= framedrop>=2; // hard drop
 233             if (drop_frame) {
 234                mpeg2_skip(mpeg2dec, 1);
 235                //printf("Dropping Frame ...\n");
 236                break;
 237             }
 238             mpeg2_skip(mpeg2dec, 0); //mpeg2skip skips frames until set again to 0
 239
 240             use_callback = (!framedrop && sh->opts->vd_use_slices &&
 241                             (info->current_picture->flags&PIC_FLAG_PROGRESSIVE_FRAME)) ?
 242                             MP_IMGFLAG_DRAW_CALLBACK:0;
 243
 244             // get_buffer "callback":
 245             mpi_new=mpcodecs_get_image(sh,MP_IMGTYPE_IPB,
 246                                        (type==PIC_FLAG_CODING_TYPE_B) ?
 247                                         use_callback : (MP_IMGFLAG_PRESERVE|MP_IMGFLAG_READABLE),
 248                                        info->sequence->width,
 249                                        info->sequence->height);
 250
 251             if(!mpi_new) return 0; // VO ERROR!!!!!!!!
 252             mpeg2_set_buf(mpeg2dec, mpi_new->planes, mpi_new);
 253             mpi_new->stride[0] = info->sequence->width;
 254             mpi_new->stride[1] = info->sequence->chroma_width;
 255             mpi_new->stride[2] = info->sequence->chroma_width;
 256             if (info->current_picture->flags&PIC_FLAG_TOP_FIELD_FIRST)
 257                 mpi_new->fields |= MP_IMGFIELD_TOP_FIRST;
 258             else mpi_new->fields &= ~MP_IMGFIELD_TOP_FIRST;
 259             if (info->current_picture->flags&PIC_FLAG_REPEAT_FIRST_FIELD)
 260                 mpi_new->fields |= MP_IMGFIELD_REPEAT_FIRST;
 261             else mpi_new->fields &= ~MP_IMGFIELD_REPEAT_FIRST;
 262             mpi_new->fields |= MP_IMGFIELD_ORDERED;
 263             if (!(info->current_picture->flags&PIC_FLAG_PROGRESSIVE_FRAME))
 264                 mpi_new->fields |= MP_IMGFIELD_INTERLACED;
 265
 266 #ifdef MPEG12_POSTPROC
 267             mpi_new->qstride=info->sequence->width>>4;
 268             {
 269             char **p = &context->quant_store[type==PIC_FLAG_CODING_TYPE_B ?
 270                                         2 : (context->quant_store_idx ^= 1)];
 271             *p = realloc(*p, mpi_new->qstride*(info->sequence->height>>4));
 272             mpi_new->qscale = *p;
 273             }
 274             mpeg2dec->decoder.quant_store=mpi_new->qscale;
 275             mpeg2dec->decoder.quant_stride=mpi_new->qstride;
 276             mpi_new->pict_type=type; // 1->I, 2->P, 3->B
 277             mpi_new->qscale_type= 1;
 278 #endif
 279
 280             if (mpi_new->flags&MP_IMGFLAG_DRAW_CALLBACK
 281                 && !(mpi_new->flags&MP_IMGFLAG_DIRECT)) {
 282                 // nice, filter/vo likes draw_callback :)
 283                 mpeg2dec->decoder.convert=draw_slice;
 284                 mpeg2dec->decoder.convert_id=sh;
 285             } else {
 286                 mpeg2dec->decoder.convert=NULL;
 287                 mpeg2dec->decoder.convert_id=NULL;
 288             }
 289
 290             break;
 291         case STATE_SLICE:
 292         case STATE_END:
 293         case STATE_INVALID_END:
 294             // decoding done:
 295             if(info->display_fbuf) {
 296                 mp_image_t* mpi = info->display_fbuf->id;
 297                 if (mpeg2dec->pending_length == 0) {
 298                     mpeg2dec->pending_length = mpeg2dec->buf_end - mpeg2dec->buf_start;
 299                     mpeg2dec->pending_buffer = realloc(mpeg2dec->pending_buffer, mpeg2dec->pending_length);
 300                     memcpy(mpeg2dec->pending_buffer, mpeg2dec->buf_start, mpeg2dec->pending_length);
 301                 } else {
 302                     // still some data in the pending buffer, shouldn't happen
 303                     mpeg2dec->pending_length = mpeg2dec->buf_end - mpeg2dec->buf_start;
 304                     memmove(mpeg2dec->pending_buffer, mpeg2dec->buf_start, mpeg2dec->pending_length);
 305                     mpeg2dec->pending_buffer = realloc(mpeg2dec->pending_buffer, mpeg2dec->pending_length + len);
 306                     memcpy(mpeg2dec->pending_buffer+mpeg2dec->pending_length, data, len);
 307                     mpeg2dec->pending_length += len;
 308                 }
 309 //              fprintf(stderr, "pending = %d\n", mpeg2dec->pending_length);
 310                 return mpi;
 311             }
 312         }
 313     }
 314 }