vf_mp: remove boxblur wrapper
[ffmpeg.git] / libavfilter / libmpcodecs / vf_test.c
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
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
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
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
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
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
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
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
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
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
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
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
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
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
272 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
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);
306 }
307
308 //===========================================================================//
309
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
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 //===========================================================================//