libmpcodecs/mp_image.h

   1 #ifndef MPLAYER_MP_IMAGE_H
   2 #define MPLAYER_MP_IMAGE_H
   3
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include "mp_msg.h"
   8
   9 //--------- codec's requirements (filled by the codec/vf) ---------
  10
  11 //--- buffer content restrictions:
  12 // set if buffer content shouldn't be modified:
  13 #define MP_IMGFLAG_PRESERVE 0x01
  14 // set if buffer content will be READ for next frame's MC: (I/P mpeg frames)
  15 #define MP_IMGFLAG_READABLE 0x02
  16
  17 //--- buffer width/stride/plane restrictions: (used for direct rendering)
  18 // stride _have_to_ be aligned to MB boundary:  [for DR restrictions]
  19 #define MP_IMGFLAG_ACCEPT_ALIGNED_STRIDE 0x4
  20 // stride should be aligned to MB boundary:     [for buffer allocation]
  21 #define MP_IMGFLAG_PREFER_ALIGNED_STRIDE 0x8
  22 // codec accept any stride (>=width):
  23 #define MP_IMGFLAG_ACCEPT_STRIDE 0x10
  24 // codec accept any width (width*bpp=stride -> stride%bpp==0) (>=width):
  25 #define MP_IMGFLAG_ACCEPT_WIDTH 0x20
  26 //--- for planar formats only:
  27 // uses only stride[0], and stride[1]=stride[2]=stride[0]>>mpi->chroma_x_shift
  28 #define MP_IMGFLAG_COMMON_STRIDE 0x40
  29 // uses only planes[0], and calculates planes[1,2] from width,height,imgfmt
  30 #define MP_IMGFLAG_COMMON_PLANE 0x80
  31
  32 #define MP_IMGFLAGMASK_RESTRICTIONS 0xFF
  33
  34 //--------- color info (filled by mp_image_setfmt() ) -----------
  35 // set if number of planes > 1
  36 #define MP_IMGFLAG_PLANAR 0x100
  37 // set if it's YUV colorspace
  38 #define MP_IMGFLAG_YUV 0x200
  39 // set if it's swapped (BGR or YVU) plane/byteorder
  40 #define MP_IMGFLAG_SWAPPED 0x400
  41 // set if you want memory for palette allocated and managed by vf_get_image etc.
  42 #define MP_IMGFLAG_RGB_PALETTE 0x800
  43
  44 #define MP_IMGFLAGMASK_COLORS 0xF00
  45
  46 // codec uses drawing/rendering callbacks (draw_slice()-like thing, DR method 2)
  47 // [the codec will set this flag if it supports callbacks, and the vo _may_
  48 //  clear it in get_image() if draw_slice() not implemented]
  49 #define MP_IMGFLAG_DRAW_CALLBACK 0x1000
  50 // set if it's in video buffer/memory: [set by vo/vf's get_image() !!!]
  51 #define MP_IMGFLAG_DIRECT 0x2000
  52 // set if buffer is allocated (used in destination images):
  53 #define MP_IMGFLAG_ALLOCATED 0x4000
  54
  55 // buffer type was printed (do NOT set this flag - it's for INTERNAL USE!!!)
  56 #define MP_IMGFLAG_TYPE_DISPLAYED 0x8000
  57
  58 // codec doesn't support any form of direct rendering - it has own buffer
  59 // allocation. so we just export its buffer pointers:
  60 #define MP_IMGTYPE_EXPORT 0
  61 // codec requires a static WO buffer, but it does only partial updates later:
  62 #define MP_IMGTYPE_STATIC 1
  63 // codec just needs some WO memory, where it writes/copies the whole frame to:
  64 #define MP_IMGTYPE_TEMP 2
  65 // I+P type, requires 2+ independent static R/W buffers
  66 #define MP_IMGTYPE_IP 3
  67 // I+P+B type, requires 2+ independent static R/W and 1+ temp WO buffers
  68 #define MP_IMGTYPE_IPB 4
  69 // Upper 16 bits give desired buffer number, -1 means get next available
  70 #define MP_IMGTYPE_NUMBERED 5
  71
  72 #define MP_MAX_PLANES   4
  73
  74 #define MP_IMGFIELD_ORDERED 0x01
  75 #define MP_IMGFIELD_TOP_FIRST 0x02
  76 #define MP_IMGFIELD_REPEAT_FIRST 0x04
  77 #define MP_IMGFIELD_TOP 0x08
  78 #define MP_IMGFIELD_BOTTOM 0x10
  79 #define MP_IMGFIELD_INTERLACED 0x20
  80
  81 typedef struct mp_image {
  82     unsigned int flags;
  83     unsigned char type;
  84     int number;
  85     unsigned char bpp;  // bits/pixel. NOT depth! for RGB it will be n*8
  86     unsigned int imgfmt;
  87     int width,height;  // stored dimensions
  88     int x,y,w,h;  // visible dimensions
  89     unsigned char* planes[MP_MAX_PLANES];
  90     int stride[MP_MAX_PLANES];
  91     char * qscale;
  92     int qstride;
  93     int pict_type; // 0->unknown, 1->I, 2->P, 3->B
  94     int fields;
  95     int qscale_type; // 0->mpeg1/4/h263, 1->mpeg2
  96     int num_planes;
  97     /* these are only used by planar formats Y,U(Cb),V(Cr) */
  98     int chroma_width;
  99     int chroma_height;
 100     int chroma_x_shift; // horizontal
 101     int chroma_y_shift; // vertical
 102     int usage_count;
 103     /* for private use by filter or vo driver (to store buffer id or dmpi) */
 104     void* priv;
 105 } mp_image_t;
 106
 107 #ifdef IMGFMT_YUY2
 108 static inline void mp_image_setfmt(mp_image_t* mpi,unsigned int out_fmt){
 109     mpi->flags&=~(MP_IMGFLAG_PLANAR|MP_IMGFLAG_YUV|MP_IMGFLAG_SWAPPED);
 110     mpi->imgfmt=out_fmt;
 111     // compressed formats
 112     if(out_fmt == IMGFMT_MPEGPES ||
 113        out_fmt == IMGFMT_ZRMJPEGNI || out_fmt == IMGFMT_ZRMJPEGIT || out_fmt == IMGFMT_ZRMJPEGIB ||
 114        IMGFMT_IS_VDPAU(out_fmt) || IMGFMT_IS_XVMC(out_fmt)){
 115         mpi->bpp=0;
 116         return;
 117     }
 118     mpi->num_planes=1;
 119     if (IMGFMT_IS_RGB(out_fmt)) {
 120         if (IMGFMT_RGB_DEPTH(out_fmt) < 8 && !(out_fmt&128))
 121             mpi->bpp = IMGFMT_RGB_DEPTH(out_fmt);
 122         else
 123             mpi->bpp=(IMGFMT_RGB_DEPTH(out_fmt)+7)&(~7);
 124         return;
 125     }
 126     if (IMGFMT_IS_BGR(out_fmt)) {
 127         if (IMGFMT_BGR_DEPTH(out_fmt) < 8 && !(out_fmt&128))
 128             mpi->bpp = IMGFMT_BGR_DEPTH(out_fmt);
 129         else
 130             mpi->bpp=(IMGFMT_BGR_DEPTH(out_fmt)+7)&(~7);
 131         mpi->flags|=MP_IMGFLAG_SWAPPED;
 132         return;
 133     }
 134     mpi->flags|=MP_IMGFLAG_YUV;
 135     mpi->num_planes=3;
 136     if (mp_get_chroma_shift(out_fmt, NULL, NULL)) {
 137         mpi->flags|=MP_IMGFLAG_PLANAR;
 138         mpi->bpp = mp_get_chroma_shift(out_fmt, &mpi->chroma_x_shift, &mpi->chroma_y_shift);
 139         mpi->chroma_width  = mpi->width  >> mpi->chroma_x_shift;
 140         mpi->chroma_height = mpi->height >> mpi->chroma_y_shift;
 141     }
 142     switch(out_fmt){
 143     case IMGFMT_I420:
 144     case IMGFMT_IYUV:
 145         mpi->flags|=MP_IMGFLAG_SWAPPED;
 146     case IMGFMT_YV12:
 147         return;
 148     case IMGFMT_420A:
 149     case IMGFMT_IF09:
 150         mpi->num_planes=4;
 151     case IMGFMT_YVU9:
 152     case IMGFMT_444P:
 153     case IMGFMT_422P:
 154     case IMGFMT_411P:
 155     case IMGFMT_440P:
 156     case IMGFMT_444P16_LE:
 157     case IMGFMT_444P16_BE:
 158     case IMGFMT_422P16_LE:
 159     case IMGFMT_422P16_BE:
 160     case IMGFMT_420P16_LE:
 161     case IMGFMT_420P16_BE:
 162         return;
 163     case IMGFMT_Y800:
 164     case IMGFMT_Y8:
 165         /* they're planar ones, but for easier handling use them as packed */
 166 //      mpi->flags|=MP_IMGFLAG_PLANAR;
 167         mpi->bpp=8;
 168         mpi->num_planes=1;
 169         return;
 170     case IMGFMT_UYVY:
 171         mpi->flags|=MP_IMGFLAG_SWAPPED;
 172     case IMGFMT_YUY2:
 173         mpi->bpp=16;
 174         mpi->num_planes=1;
 175         return;
 176     case IMGFMT_NV12:
 177         mpi->flags|=MP_IMGFLAG_SWAPPED;
 178     case IMGFMT_NV21:
 179         mpi->flags|=MP_IMGFLAG_PLANAR;
 180         mpi->bpp=12;
 181         mpi->num_planes=2;
 182         mpi->chroma_width=(mpi->width>>0);
 183         mpi->chroma_height=(mpi->height>>1);
 184         mpi->chroma_x_shift=0;
 185         mpi->chroma_y_shift=1;
 186         return;
 187     }
 188     mp_msg(MSGT_DECVIDEO,MSGL_WARN,"mp_image: unknown out_fmt: 0x%X\n",out_fmt);
 189     mpi->bpp=0;
 190 }
 191 #endif
 192
 193 static inline mp_image_t* new_mp_image(int w,int h){
 194     mp_image_t* mpi=(mp_image_t*)malloc(sizeof(mp_image_t));
 195     if(!mpi) return NULL; // error!
 196     memset(mpi,0,sizeof(mp_image_t));
 197     mpi->width=mpi->w=w;
 198     mpi->height=mpi->h=h;
 199     return mpi;
 200 }
 201
 202 static inline void free_mp_image(mp_image_t* mpi){
 203     if(!mpi) return;
 204     if(mpi->flags&MP_IMGFLAG_ALLOCATED){
 205         /* becouse we allocate the whole image in once */
 206         if(mpi->planes[0]) free(mpi->planes[0]);
 207         if (mpi->flags & MP_IMGFLAG_RGB_PALETTE)
 208             free(mpi->planes[1]);
 209     }
 210     free(mpi);
 211 }
 212
 213 mp_image_t* alloc_mpi(int w, int h, unsigned long int fmt);
 214 void mp_image_alloc_planes(mp_image_t *mpi);
 215 void copy_mpi(mp_image_t *dmpi, mp_image_t *mpi);
 216
 217 #endif /* MPLAYER_MP_IMAGE_H */