libmpcodecs/vf_palette.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <inttypes.h>
   5
   6 #include "config.h"
   7 #include "mp_msg.h"
   8 #include "help_mp.h"
   9
  10 #include "img_format.h"
  11 #include "mp_image.h"
  12 #include "vf.h"
  13
  14 #include "libswscale/rgb2rgb.h"
  15
  16 //===========================================================================//
  17
  18 // commented out 16 and 15 bit output support, because the conversion
  19 // routines are incorrrect.  they assume the palette to be of the same
  20 // depth as the output, which is incorrect. --Joey
  21
  22 static const unsigned int bgr_list[]={
  23     IMGFMT_BGR32,
  24     IMGFMT_BGR24,
  25 //    IMGFMT_BGR16,
  26 //    IMGFMT_BGR15,
  27     0
  28 };
  29 static const unsigned int rgb_list[]={
  30     IMGFMT_RGB32,
  31     IMGFMT_RGB24,
  32 //    IMGFMT_RGB16,
  33 //    IMGFMT_RGB15,
  34     0
  35 };
  36
  37 static unsigned int gray_pal[256];
  38
  39 static unsigned int find_best(struct vf_instance* vf, unsigned int fmt){
  40     unsigned int best=0;
  41     int ret;
  42     const unsigned int* p;
  43     if(fmt==IMGFMT_BGR8) p=bgr_list;
  44     else if(fmt==IMGFMT_RGB8) p=rgb_list;
  45     else return 0;
  46     while(*p){
  47         ret=vf->next->query_format(vf->next,*p);
  48         mp_msg(MSGT_VFILTER,MSGL_DBG2,"[%s] query(%s) -> %d\n",vf->info->name,vo_format_name(*p),ret&3);
  49         if(ret&VFCAP_CSP_SUPPORTED_BY_HW){ best=*p; break;} // no conversion -> bingo!
  50         if(ret&VFCAP_CSP_SUPPORTED && !best) best=*p; // best with conversion
  51         ++p;
  52     }
  53     return best;
  54 }
  55
  56 //===========================================================================//
  57
  58 struct vf_priv_s {
  59     unsigned int fmt;
  60     int pal_msg;
  61 };
  62
  63 static int config(struct vf_instance* vf,
  64         int width, int height, int d_width, int d_height,
  65         unsigned int flags, unsigned int outfmt){
  66     if (!vf->priv->fmt)
  67         vf->priv->fmt=find_best(vf,outfmt);
  68     if(!vf->priv->fmt){
  69         // no matching fmt, so force one...
  70         if(outfmt==IMGFMT_RGB8) vf->priv->fmt=IMGFMT_RGB32;
  71         else if(outfmt==IMGFMT_BGR8) vf->priv->fmt=IMGFMT_BGR32;
  72         else return 0;
  73     }
  74     return vf_next_config(vf,width,height,d_width,d_height,flags,vf->priv->fmt);
  75 }
  76
  77 static int put_image(struct vf_instance* vf, mp_image_t *mpi, double pts){
  78     mp_image_t *dmpi;
  79     uint8_t *old_palette = mpi->planes[1];
  80
  81     // hope we'll get DR buffer:
  82     dmpi=vf_get_image(vf->next,vf->priv->fmt,
  83         MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
  84         mpi->w, mpi->h);
  85
  86     if (!mpi->planes[1])
  87     {
  88         if(!vf->priv->pal_msg){
  89             mp_msg(MSGT_VFILTER,MSGL_V,"[%s] no palette given, assuming builtin grayscale one\n",vf->info->name);
  90             vf->priv->pal_msg=1;
  91         }
  92         mpi->planes[1] = (unsigned char*)gray_pal;
  93     }
  94
  95     if(mpi->w==mpi->stride[0] && dmpi->w*(dmpi->bpp>>3)==dmpi->stride[0]){
  96         // no stride conversion needed
  97         switch(IMGFMT_RGB_DEPTH(dmpi->imgfmt)){
  98         case 15:
  99             if (IMGFMT_IS_BGR(dmpi->imgfmt))
 100                 palette8tobgr15(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 101             else
 102                 palette8torgb15(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 103             break;
 104         case 16:
 105             if (IMGFMT_IS_BGR(dmpi->imgfmt))
 106                 palette8tobgr16(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 107             else
 108                 palette8torgb16(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 109             break;
 110         case 24:
 111             if (IMGFMT_IS_BGR(dmpi->imgfmt))
 112                 palette8topacked24(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 113             else
 114                 palette8topacked24(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 115             break;
 116         case 32:
 117             if (IMGFMT_IS_BGR(dmpi->imgfmt))
 118                 palette8topacked32(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 119             else
 120                 palette8topacked32(mpi->planes[0],dmpi->planes[0],mpi->h*mpi->w,mpi->planes[1]);
 121             break;
 122         }
 123     } else {
 124         int y;
 125         for(y=0;y<mpi->h;y++){
 126             unsigned char* src=mpi->planes[0]+y*mpi->stride[0];
 127             unsigned char* dst=dmpi->planes[0]+y*dmpi->stride[0];
 128             switch(IMGFMT_RGB_DEPTH(dmpi->imgfmt)){
 129             case 15:
 130                 if (IMGFMT_IS_BGR(dmpi->imgfmt))
 131                     palette8tobgr15(src,dst,mpi->w,mpi->planes[1]);
 132                 else
 133                     palette8torgb15(src,dst,mpi->w,mpi->planes[1]);
 134                 break;
 135             case 16:
 136                 if (IMGFMT_IS_BGR(dmpi->imgfmt))
 137                     palette8tobgr16(src,dst,mpi->w,mpi->planes[1]);
 138                 else
 139                     palette8torgb16(src,dst,mpi->w,mpi->planes[1]);
 140                 break;
 141             case 24:
 142                 if (IMGFMT_IS_BGR(dmpi->imgfmt))
 143                     palette8topacked24(src,dst,mpi->w,mpi->planes[1]);
 144                 else
 145                     palette8topacked24(src,dst,mpi->w,mpi->planes[1]);
 146                 break;
 147             case 32:
 148                 if (IMGFMT_IS_BGR(dmpi->imgfmt))
 149                     palette8topacked32(src,dst,mpi->w,mpi->planes[1]);
 150                 else
 151                     palette8topacked32(src,dst,mpi->w,mpi->planes[1]);
 152                 break;
 153             }
 154         }
 155     }
 156     mpi->planes[1] = old_palette;
 157
 158     return vf_next_put_image(vf,dmpi, pts);
 159 }
 160
 161 //===========================================================================//
 162
 163 static int query_format(struct vf_instance* vf, unsigned int fmt){
 164     int best=find_best(vf,fmt);
 165     if(!best) return 0; // no match
 166     return vf->next->query_format(vf->next,best);
 167 }
 168
 169 static void uninit(vf_instance_t *vf) {
 170   free(vf->priv);
 171 }
 172
 173 static int open(vf_instance_t *vf, char* args){
 174     unsigned int i;
 175     vf->config=config;
 176     vf->uninit=uninit;
 177     vf->put_image=put_image;
 178     vf->query_format=query_format;
 179     vf->priv=malloc(sizeof(struct vf_priv_s));
 180     memset(vf->priv, 0, sizeof(struct vf_priv_s));
 181     for(i=0;i<256;i++) gray_pal[i]=0x01010101*i;
 182     if (args)
 183     {
 184         if (!strcasecmp(args,"rgb15")) vf->priv->fmt=IMGFMT_RGB15; else
 185         if (!strcasecmp(args,"rgb16")) vf->priv->fmt=IMGFMT_RGB16; else
 186         if (!strcasecmp(args,"rgb24")) vf->priv->fmt=IMGFMT_RGB24; else
 187         if (!strcasecmp(args,"rgb32")) vf->priv->fmt=IMGFMT_RGB32; else
 188         if (!strcasecmp(args,"bgr15")) vf->priv->fmt=IMGFMT_BGR15; else
 189         if (!strcasecmp(args,"bgr16")) vf->priv->fmt=IMGFMT_BGR16; else
 190         if (!strcasecmp(args,"bgr24")) vf->priv->fmt=IMGFMT_BGR24; else
 191         if (!strcasecmp(args,"bgr32")) vf->priv->fmt=IMGFMT_BGR32; else
 192         {
 193             mp_tmsg(MSGT_VFILTER, MSGL_WARN, "[VF_FORMAT] Unknown format name: '%s'.\n", args);
 194             return 0;
 195         }
 196     }
 197     return 1;
 198 }
 199
 200 const vf_info_t vf_info_palette = {
 201     "8bpp indexed (using palette) -> BGR 15/16/24/32 conversion",
 202     "palette",
 203     "A'rpi & Alex",
 204     "",
 205     open,
 206     NULL
 207 };
 208
 209 //===========================================================================//