1 /*
2 * Copyright (C) 2002 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * This file is part of MPlayer.
5 *
6 * MPlayer is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * MPlayer is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <inttypes.h>
25
26 #include "config.h"
27 #include "mp_msg.h"
28 #include "help_mp.h"
29
30 #include "img_format.h"
31 #include "mp_image.h"
32 #include "vf.h"
33
34 //===========================================================================//
35
36 #include <inttypes.h>
37 #include <math.h>
38
39 #define MAX(a,b) ((a) > (b) ? (a) : (b))
40 #define MIN(a,b) ((a) < (b) ? (a) : (b))
41 #define ABS(a,b) ((a) > 0 ? (a) : -(a))
42
43 #define WIDTH 512
44 #define HEIGHT 512
45
46 struct vf_priv_s {
47 int frame_num;
48 };
49
config(struct vf_instance * vf,int width,int height,int d_width,int d_height,unsigned int flags,unsigned int outfmt)50 static int config(struct vf_instance *vf,
51 int width, int height, int d_width, int d_height,
52 unsigned int flags, unsigned int outfmt){
53
54 if(vf_next_query_format(vf,IMGFMT_YV12)<=0){
55 mp_msg(MSGT_VFILTER, MSGL_WARN, MSGTR_MPCODECS_WarnNextFilterDoesntSupport, "YV12");
56 return 0;
57 }
58
59 //hmm whats the meaning of these ... ;)
60 d_width= width= WIDTH;
61 d_height= height= HEIGHT;
62
63 return vf_next_config(vf,width,height,d_width,d_height,flags,IMGFMT_YV12);
64 }
65
66 static double c[64];
67
initIdct(void)68 static void initIdct(void)
69 {
70 int i;
71
72 for (i=0; i<8; i++)
73 {
74 double s= i==0 ? sqrt(0.125) : 0.5;
75 int j;
76
77 for(j=0; j<8; j++)
78 c[i*8+j]= s*cos((3.141592654/8.0)*i*(j+0.5));
79 }
80 }
81
82
idct(uint8_t * dst,int dstStride,int src[64])83 static void idct(uint8_t *dst, int dstStride, int src[64])
84 {
85 int i, j, k;
86 double tmp[64];
87
88 for(i=0; i<8; i++)
89 {
90 for(j=0; j<8; j++)
91 {
92 double sum= 0.0;
93
94 for(k=0; k<8; k++)
95 sum+= c[k*8+j]*src[8*i+k];
96
97 tmp[8*i+j]= sum;
98 }
99 }
100
101 for(j=0; j<8; j++)
102 {
103 for(i=0; i<8; i++)
104 {
105 int v;
106 double sum= 0.0;
107
108 for(k=0; k<8; k++)
109 sum+= c[k*8+i]*tmp[8*k+j];
110
111 v= (int)floor(sum+0.5);
112 if(v<0) v=0;
113 else if(v>255) v=255;
114
115 dst[dstStride*i + j] = v;
116 }
117 }
118 }
119
drawDc(uint8_t * dst,int stride,int color,int w,int h)120 static void drawDc(uint8_t *dst, int stride, int color, int w, int h)
121 {
122 int y;
123 for(y=0; y<h; y++)
124 {
125 int x;
126 for(x=0; x<w; x++)
127 {
128 dst[x + y*stride]= color;
129 }
130 }
131 }
132
drawBasis(uint8_t * dst,int stride,int amp,int freq,int dc)133 static void drawBasis(uint8_t *dst, int stride, int amp, int freq, int dc)
134 {
135 int src[64];
136
137 memset(src, 0, 64*sizeof(int));
138 src[0]= dc;
139 if(amp) src[freq]= amp;
140 idct(dst, stride, src);
141 }
142
drawCbp(uint8_t * dst[3],int stride[3],int cbp,int amp,int dc)143 static void drawCbp(uint8_t *dst[3], int stride[3], int cbp, int amp, int dc)
144 {
145 if(cbp&1) drawBasis(dst[0] , stride[0], amp, 1, dc);
146 if(cbp&2) drawBasis(dst[0]+8 , stride[0], amp, 1, dc);
147 if(cbp&4) drawBasis(dst[0]+ 8*stride[0], stride[0], amp, 1, dc);
148 if(cbp&8) drawBasis(dst[0]+8+8*stride[0], stride[0], amp, 1, dc);
149 if(cbp&16)drawBasis(dst[1] , stride[1], amp, 1, dc);
150 if(cbp&32)drawBasis(dst[2] , stride[2], amp, 1, dc);
151 }
152
dc1Test(uint8_t * dst,int stride,int w,int h,int off)153 static void dc1Test(uint8_t *dst, int stride, int w, int h, int off)
154 {
155 const int step= MAX(256/(w*h/256), 1);
156 int y;
157 int color=off;
158 for(y=0; y<h; y+=16)
159 {
160 int x;
161 for(x=0; x<w; x+=16)
162 {
163 drawDc(dst + x + y*stride, stride, color, 8, 8);
164 color+=step;
165 }
166 }
167 }
168
freq1Test(uint8_t * dst,int stride,int off)169 static void freq1Test(uint8_t *dst, int stride, int off)
170 {
171 int y;
172 int freq=0;
173 for(y=0; y<8*16; y+=16)
174 {
175 int x;
176 for(x=0; x<8*16; x+=16)
177 {
178 drawBasis(dst + x + y*stride, stride, 4*(96+off), freq, 128*8);
179 freq++;
180 }
181 }
182 }
183
amp1Test(uint8_t * dst,int stride,int off)184 static void amp1Test(uint8_t *dst, int stride, int off)
185 {
186 int y;
187 int amp=off;
188 for(y=0; y<16*16; y+=16)
189 {
190 int x;
191 for(x=0; x<16*16; x+=16)
192 {
193 drawBasis(dst + x + y*stride, stride, 4*(amp), 1, 128*8);
194 amp++;
195 }
196 }
197 }
198
cbp1Test(uint8_t * dst[3],int stride[3],int off)199 static void cbp1Test(uint8_t *dst[3], int stride[3], int off)
200 {
201 int y;
202 int cbp=0;
203 for(y=0; y<16*8; y+=16)
204 {
205 int x;
206 for(x=0; x<16*8; x+=16)
207 {
208 uint8_t *dst1[3];
209 dst1[0]= dst[0] + x*2 + y*2*stride[0];
210 dst1[1]= dst[1] + x + y*stride[1];
211 dst1[2]= dst[2] + x + y*stride[2];
212
213 drawCbp(dst1, stride, cbp, (64+off)*4, 128*8);
214 cbp++;
215 }
216 }
217 }
218
mv1Test(uint8_t * dst,int stride,int off)219 static void mv1Test(uint8_t *dst, int stride, int off)
220 {
221 int y;
222 for(y=0; y<16*16; y++)
223 {
224 int x;
225 if(y&16) continue;
226 for(x=0; x<16*16; x++)
227 {
228 dst[x + y*stride]= x + off*8/(y/32+1);
229 }
230 }
231 }
232
ring1Test(uint8_t * dst,int stride,int off)233 static void ring1Test(uint8_t *dst, int stride, int off)
234 {
235 int y;
236 int color=0;
237 for(y=off; y<16*16; y+=16)
238 {
239 int x;
240 for(x=off; x<16*16; x+=16)
241 {
242 drawDc(dst + x + y*stride, stride, ((x+y)&16) ? color : -color, 16, 16);
243 // dst[x + y*stride]= 255 + (off&1);
244 color++;
245 }
246 }
247 }
248
ring2Test(uint8_t * dst,int stride,int off)249 static void ring2Test(uint8_t *dst, int stride, int off)
250 {
251 int y;
252 for(y=0; y<16*16; y++)
253 {
254 int x;
255 for(x=0; x<16*16; x++)
256 {
257 double d= sqrt((x-8*16)*(x-8*16) + (y-8*16)*(y-8*16));
258 double r= d/20 - (int)(d/20);
259 if(r<off/30.0)
260 {
261 dst[x + y*stride]= 255;
262 dst[x + y*stride+256]= 0;
263 }
264 else{
265 dst[x + y*stride]= x;
266 dst[x + y*stride+256]= x;
267 }
268 }
269 }
270 }
271
put_image(struct vf_instance * vf,mp_image_t * mpi,double pts,double endpts)272 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts, double endpts){
273 mp_image_t *dmpi;
274 int frame= vf->priv->frame_num;
275
276 // hope we'll get DR buffer:
277 dmpi=vf_get_image(vf->next,IMGFMT_YV12,
278 MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
279 WIDTH, HEIGHT);
280
281 // clean
282 memset(dmpi->planes[0], 0, dmpi->stride[0]*dmpi->h);
283 memset(dmpi->planes[1], 128, dmpi->stride[1]*dmpi->h>>dmpi->chroma_y_shift);
284 memset(dmpi->planes[2], 128, dmpi->stride[2]*dmpi->h>>dmpi->chroma_y_shift);
285
286 if(frame%30)
287 {
288 switch(frame/30)
289 {
290 case 0: dc1Test(dmpi->planes[0], dmpi->stride[0], 256, 256, frame%30); break;
291 case 1: dc1Test(dmpi->planes[1], dmpi->stride[1], 256, 256, frame%30); break;
292 case 2: freq1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
293 case 3: freq1Test(dmpi->planes[1], dmpi->stride[1], frame%30); break;
294 case 4: amp1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
295 case 5: amp1Test(dmpi->planes[1], dmpi->stride[1], frame%30); break;
296 case 6: cbp1Test(dmpi->planes , dmpi->stride , frame%30); break;
297 case 7: mv1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
298 case 8: ring1Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
299 case 9: ring2Test(dmpi->planes[0], dmpi->stride[0], frame%30); break;
300 }
301 }
302
303 frame++;
304 vf->priv->frame_num= frame;
305 return vf_next_put_image(vf, dmpi, pts, endpts);
306 }
307
308 //===========================================================================//
309
query_format(struct vf_instance * vf,unsigned int fmt)310 static int query_format(struct vf_instance *vf, unsigned int fmt){
311 return vf_next_query_format(vf,IMGFMT_YV12) & (~VFCAP_CSP_SUPPORTED_BY_HW);
312 }
313
vf_open(vf_instance_t * vf,char * args)314 static int vf_open(vf_instance_t *vf, char *args){
315 vf->config=config;
316 vf->put_image=put_image;
317 vf->query_format=query_format;
318 vf->priv=malloc(sizeof(struct vf_priv_s));
319 vf->priv->frame_num= args ? atoi(args) : 0;
320 initIdct();
321 return 1;
322 }
323
324 const vf_info_t vf_info_test = {
325 "test pattern generator",
326 "test",
327 "Michael Niedermayer",
328 "",
329 vf_open,
330 NULL
331 };
332
333 //===========================================================================//
334