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