libmpcodecs/vf_rgb2bgr.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
   9 #include "img_format.h"
  10 #include "mp_image.h"
  11 #include "vf.h"
  12
  13 #include "libswscale/rgb2rgb.h"
  14
  15 //===========================================================================//
  16
  17 struct vf_priv_s {
  18     unsigned int fmt;
  19     int forced;
  20 };
  21
  22 static unsigned int getfmt(unsigned int outfmt,int forced){
  23     if(forced) switch(outfmt){
  24     case IMGFMT_RGB24:
  25     case IMGFMT_RGB32:
  26     case IMGFMT_BGR24:
  27     case IMGFMT_BGR32:
  28         return outfmt;
  29     }
  30     switch(outfmt){
  31     case IMGFMT_RGB24: return IMGFMT_BGR24;
  32     case IMGFMT_RGB32: return IMGFMT_BGR32;
  33     case IMGFMT_BGR24: return IMGFMT_RGB24;
  34     case IMGFMT_BGR32: return IMGFMT_RGB32;
  35     }
  36     return 0;
  37 }
  38
  39 static int config(struct vf_instance* vf,
  40         int width, int height, int d_width, int d_height,
  41         unsigned int flags, unsigned int outfmt){
  42     vf->priv->fmt=getfmt(outfmt,vf->priv->forced);
  43     return vf_next_config(vf,width,height,d_width,d_height,flags,vf->priv->fmt);
  44 }
  45
  46 static int put_image(struct vf_instance* vf, mp_image_t *mpi, double pts){
  47     mp_image_t *dmpi;
  48
  49     // hope we'll get DR buffer:
  50     dmpi=vf_get_image(vf->next,vf->priv->fmt,
  51         MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
  52         mpi->w, mpi->h);
  53
  54     if(mpi->stride[0]!=dmpi->stride[0] || mpi->stride[0]!=mpi->w*(mpi->bpp/8)){
  55         int y;
  56         unsigned char* src=mpi->planes[0];
  57         unsigned char* dst=dmpi->planes[0];
  58         int srcsize=mpi->w*mpi->bpp/8;
  59         for(y=0;y<mpi->h;y++){
  60             if(mpi->bpp==32)
  61                 rgb32tobgr32(src,dst,srcsize);
  62             else
  63                 rgb24tobgr24(src,dst,srcsize);
  64             src+=mpi->stride[0];
  65             dst+=dmpi->stride[0];
  66         }
  67     } else {
  68         if(mpi->bpp==32)
  69             rgb32tobgr32(mpi->planes[0],dmpi->planes[0],mpi->w*mpi->h*4);
  70         else
  71             rgb24tobgr24(mpi->planes[0],dmpi->planes[0],mpi->w*mpi->h*3);
  72     }
  73
  74     return vf_next_put_image(vf,dmpi, pts);
  75 }
  76
  77 //===========================================================================//
  78
  79 static int query_format(struct vf_instance* vf, unsigned int outfmt){
  80     unsigned int fmt=getfmt(outfmt,vf->priv->forced);
  81     if(!fmt) return 0;
  82     return vf_next_query_format(vf,fmt) & (~VFCAP_CSP_SUPPORTED_BY_HW);
  83 }
  84
  85 static int open(vf_instance_t *vf, char* args){
  86     vf->config=config;
  87     vf->put_image=put_image;
  88     vf->query_format=query_format;
  89     vf->priv=malloc(sizeof(struct vf_priv_s));
  90     vf->priv->forced=args && !strcasecmp(args,"swap");
  91     return 1;
  92 }
  93
  94 const vf_info_t vf_info_rgb2bgr = {
  95     "fast 24/32bpp RGB<->BGR conversion",
  96     "rgb2bgr",
  97     "A'rpi",
  98     "",
  99     open,
 100     NULL
 101 };
 102
 103 //===========================================================================//