libmpcodecs/vf_rgbtest.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 <string.h>
  22 #include <inttypes.h>
  23
  24 #include "config.h"
  25 #include "mp_msg.h"
  26
  27 #include "img_format.h"
  28 #include "mp_image.h"
  29 #include "vf.h"
  30
  31 //===========================================================================//
  32
  33 struct vf_priv_s {
  34     unsigned int fmt;
  35     int w, h;
  36 };
  37
  38 static unsigned int getfmt(unsigned int outfmt){
  39     switch(outfmt){
  40     case IMGFMT_RGB15:
  41     case IMGFMT_RGB16:
  42     case IMGFMT_RGB24:
  43     case IMGFMT_RGBA:
  44     case IMGFMT_ARGB:
  45     case IMGFMT_BGR15:
  46     case IMGFMT_BGR16:
  47     case IMGFMT_BGR24:
  48     case IMGFMT_BGRA:
  49     case IMGFMT_ABGR:
  50         return outfmt;
  51     }
  52     return 0;
  53 }
  54
  55 static void put_pixel(uint8_t *buf, int x, int y, int stride, int r, int g, int b, int fmt){
  56     switch(fmt){
  57     case IMGFMT_BGR15: ((uint16_t*)(buf + y*stride))[x]= ((r>>3)<<10) | ((g>>3)<<5) | (b>>3);
  58     break;
  59     case IMGFMT_RGB15: ((uint16_t*)(buf + y*stride))[x]= ((b>>3)<<10) | ((g>>3)<<5) | (r>>3);
  60     break;
  61     case IMGFMT_BGR16: ((uint16_t*)(buf + y*stride))[x]= ((r>>3)<<11) | ((g>>2)<<5) | (b>>3);
  62     break;
  63     case IMGFMT_RGB16: ((uint16_t*)(buf + y*stride))[x]= ((b>>3)<<11) | ((g>>2)<<5) | (r>>3);
  64     break;
  65     case IMGFMT_RGB24:
  66         buf[3*x + y*stride + 0]= r;
  67         buf[3*x + y*stride + 1]= g;
  68         buf[3*x + y*stride + 2]= b;
  69     break;
  70     case IMGFMT_BGR24:
  71         buf[3*x + y*stride + 0]= b;
  72         buf[3*x + y*stride + 1]= g;
  73         buf[3*x + y*stride + 2]= r;
  74     break;
  75     case IMGFMT_RGBA:
  76         buf[4*x + y*stride + 0]= r;
  77         buf[4*x + y*stride + 1]= g;
  78         buf[4*x + y*stride + 2]= b;
  79     break;
  80     case IMGFMT_BGRA:
  81         buf[4*x + y*stride + 0]= b;
  82         buf[4*x + y*stride + 1]= g;
  83         buf[4*x + y*stride + 2]= r;
  84     break;
  85     case IMGFMT_ARGB:
  86         buf[4*x + y*stride + 1]= r;
  87         buf[4*x + y*stride + 2]= g;
  88         buf[4*x + y*stride + 3]= b;
  89     break;
  90     case IMGFMT_ABGR:
  91         buf[4*x + y*stride + 1]= b;
  92         buf[4*x + y*stride + 2]= g;
  93         buf[4*x + y*stride + 3]= r;
  94     break;
  95     }
  96 }
  97
  98 static int config(struct vf_instance* vf,
  99         int width, int height, int d_width, int d_height,
 100         unsigned int flags, unsigned int outfmt){
 101     if (vf->priv->w > 0) { d_width  = width  = vf->priv->w; }
 102     if (vf->priv->h > 0) { d_height = height = vf->priv->h; }
 103     vf->priv->fmt=getfmt(outfmt);
 104     mp_msg(MSGT_VFILTER,MSGL_V,"rgb test format:%s\n", vo_format_name(outfmt));
 105     return vf_next_config(vf,width,height,d_width,d_height,flags,vf->priv->fmt);
 106 }
 107
 108 static int put_image(struct vf_instance* vf, mp_image_t *mpi, double pts){
 109     mp_image_t *dmpi;
 110     int x, y;
 111     int w = vf->priv->w > 0 ? vf->priv->w : mpi->w;
 112     int h = vf->priv->h > 0 ? vf->priv->h : mpi->h;
 113
 114     // hope we'll get DR buffer:
 115     dmpi=vf_get_image(vf->next,vf->priv->fmt,
 116         MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
 117         w, h);
 118
 119      for(y=0; y<h; y++){
 120          for(x=0; x<w; x++){
 121              int c= 256*x/w;
 122              int r=0,g=0,b=0;
 123
 124              if(3*y<h)        r=c;
 125              else if(3*y<2*h) g=c;
 126              else                  b=c;
 127
 128              put_pixel(dmpi->planes[0], x, y, dmpi->stride[0], r, g, b, vf->priv->fmt);
 129          }
 130      }
 131
 132     return vf_next_put_image(vf,dmpi, pts);
 133 }
 134
 135 //===========================================================================//
 136
 137 static int query_format(struct vf_instance* vf, unsigned int outfmt){
 138     unsigned int fmt=getfmt(outfmt);
 139     if(!fmt) return 0;
 140     return vf_next_query_format(vf,fmt) & (~VFCAP_CSP_SUPPORTED_BY_HW);
 141 }
 142
 143 static int vf_open(vf_instance_t *vf, char *args){
 144     vf->config=config;
 145     vf->put_image=put_image;
 146     vf->query_format=query_format;
 147     vf->priv=malloc(sizeof(struct vf_priv_s));
 148     vf->priv->w = vf->priv->h = 0;
 149     if (args)
 150         sscanf(args, "%d:%d", &vf->priv->w, &vf->priv->h);
 151     return 1;
 152 }
 153
 154 const vf_info_t vf_info_rgbtest = {
 155     "rgbtest",
 156     "rgbtest",
 157     "Michael Niedermayer",
 158     "",
 159     vf_open,
 160     NULL
 161 };
 162
 163 //===========================================================================//