?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Use proper length specifiers in mp_msg calls, fixes the warnings:
[mplayer/greg.git] / libmpcodecs / vf_noise.c
blob31ae7add09c656e94329ccc03e11f40b09eb762a
1 /*
2 Copyright (C) 2002 Michael Niedermayer <michaelni@gmx.at>
3
4 This program 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 This program 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
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, 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 #include <math.h>
24
25 #include "config.h"
26 #include "mp_msg.h"
27 #include "cpudetect.h"
28
29 #ifdef HAVE_MALLOC_H
30 #include <malloc.h>
31 #endif
32
33 #include "img_format.h"
34 #include "mp_image.h"
35 #include "vf.h"
36 #include "libvo/fastmemcpy.h"
37
38 #define MAX_NOISE 4096
39 #define MAX_SHIFT 1024
40 #define MAX_RES (MAX_NOISE-MAX_SHIFT)
41
42 //===========================================================================//
43
44 static inline void lineNoise_C(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift);
45 static inline void lineNoiseAvg_C(uint8_t *dst, uint8_t *src, int len, int8_t **shift);
46
47 static void (*lineNoise)(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift)= lineNoise_C;
48 static void (*lineNoiseAvg)(uint8_t *dst, uint8_t *src, int len, int8_t **shift)= lineNoiseAvg_C;
49
50 typedef struct FilterParam{
51 int strength;
52 int uniform;
53 int temporal;
54 int quality;
55 int averaged;
56 int pattern;
57 int shiftptr;
58 int8_t *noise;
59 int8_t *prev_shift[MAX_RES][3];
60 }FilterParam;
61
62 struct vf_priv_s {
63 FilterParam lumaParam;
64 FilterParam chromaParam;
65 unsigned int outfmt;
66 };
67
68 static int nonTempRandShift_init;
69 static int nonTempRandShift[MAX_RES];
70
71 static int patt[4] = {
72 -1,0,1,0
73 };
74
75 #define RAND_N(range) ((int) ((double)range*rand()/(RAND_MAX+1.0)))
76 static int8_t *initNoise(FilterParam *fp){
77 int strength= fp->strength;
78 int uniform= fp->uniform;
79 int averaged= fp->averaged;
80 int pattern= fp->pattern;
81 int8_t *noise= memalign(16, MAX_NOISE*sizeof(int8_t));
82 int i, j;
83
84 srand(123457);
85
86 for(i=0,j=0; i<MAX_NOISE; i++,j++)
87 {
88 if(uniform) {
89 if (averaged) {
90 if (pattern) {
91 noise[i]= (RAND_N(strength) - strength/2)/6
92 +patt[j%4]*strength*0.25/3;
93 } else {
94 noise[i]= (RAND_N(strength) - strength/2)/3;
95 }
96 } else {
97 if (pattern) {
98 noise[i]= (RAND_N(strength) - strength/2)/2
99 + patt[j%4]*strength*0.25;
100 } else {
101 noise[i]= RAND_N(strength) - strength/2;
102 }
103 }
104 } else {
105 double x1, x2, w, y1;
106 do {
107 x1 = 2.0 * rand()/(float)RAND_MAX - 1.0;
108 x2 = 2.0 * rand()/(float)RAND_MAX - 1.0;
109 w = x1 * x1 + x2 * x2;
110 } while ( w >= 1.0 );
111
112 w = sqrt( (-2.0 * log( w ) ) / w );
113 y1= x1 * w;
114 y1*= strength / sqrt(3.0);
115 if (pattern) {
116 y1 /= 2;
117 y1 += patt[j%4]*strength*0.35;
118 }
119 if (y1<-128) y1=-128;
120 else if(y1> 127) y1= 127;
121 if (averaged) y1 /= 3.0;
122 noise[i]= (int)y1;
123 }
124 if (RAND_N(6) == 0) j--;
125 }
126
127
128 for (i = 0; i < MAX_RES; i++)
129 for (j = 0; j < 3; j++)
130 fp->prev_shift[i][j] = noise + (rand()&(MAX_SHIFT-1));
131
132 if(!nonTempRandShift_init){
133 for(i=0; i<MAX_RES; i++){
134 nonTempRandShift[i]= rand()&(MAX_SHIFT-1);
135 }
136 nonTempRandShift_init = 1;
137 }
138
139 fp->noise= noise;
140 fp->shiftptr= 0;
141 return noise;
142 }
143
144 /***************************************************************************/
145
146 #ifdef HAVE_MMX
147 static inline void lineNoise_MMX(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
148 long mmx_len= len&(~7);
149 noise+=shift;
150
151 asm volatile(
152 "mov %3, %%"REG_a" \n\t"
153 "pcmpeqb %%mm7, %%mm7 \n\t"
154 "psllw $15, %%mm7 \n\t"
155 "packsswb %%mm7, %%mm7 \n\t"
156 ASMALIGN(4)
157 "1: \n\t"
158 "movq (%0, %%"REG_a"), %%mm0 \n\t"
159 "movq (%1, %%"REG_a"), %%mm1 \n\t"
160 "pxor %%mm7, %%mm0 \n\t"
161 "paddsb %%mm1, %%mm0 \n\t"
162 "pxor %%mm7, %%mm0 \n\t"
163 "movq %%mm0, (%2, %%"REG_a") \n\t"
164 "add $8, %%"REG_a" \n\t"
165 " js 1b \n\t"
166 :: "r" (src+mmx_len), "r" (noise+mmx_len), "r" (dst+mmx_len), "g" (-mmx_len)
167 : "%"REG_a
168 );
169 if(mmx_len!=len)
170 lineNoise_C(dst+mmx_len, src+mmx_len, noise+mmx_len, len-mmx_len, 0);
171 }
172 #endif
173
174 //duplicate of previous except movntq
175 #ifdef HAVE_MMX2
176 static inline void lineNoise_MMX2(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
177 long mmx_len= len&(~7);
178 noise+=shift;
179
180 asm volatile(
181 "mov %3, %%"REG_a" \n\t"
182 "pcmpeqb %%mm7, %%mm7 \n\t"
183 "psllw $15, %%mm7 \n\t"
184 "packsswb %%mm7, %%mm7 \n\t"
185 ASMALIGN(4)
186 "1: \n\t"
187 "movq (%0, %%"REG_a"), %%mm0 \n\t"
188 "movq (%1, %%"REG_a"), %%mm1 \n\t"
189 "pxor %%mm7, %%mm0 \n\t"
190 "paddsb %%mm1, %%mm0 \n\t"
191 "pxor %%mm7, %%mm0 \n\t"
192 "movntq %%mm0, (%2, %%"REG_a") \n\t"
193 "add $8, %%"REG_a" \n\t"
194 " js 1b \n\t"
195 :: "r" (src+mmx_len), "r" (noise+mmx_len), "r" (dst+mmx_len), "g" (-mmx_len)
196 : "%"REG_a
197 );
198 if(mmx_len!=len)
199 lineNoise_C(dst+mmx_len, src+mmx_len, noise+mmx_len, len-mmx_len, 0);
200 }
201 #endif
202
203 static inline void lineNoise_C(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
204 int i;
205 noise+= shift;
206 for(i=0; i<len; i++)
207 {
208 int v= src[i]+ noise[i];
209 if(v>255) dst[i]=255; //FIXME optimize
210 else if(v<0) dst[i]=0;
211 else dst[i]=v;
212 }
213 }
214
215 /***************************************************************************/
216
217 #ifdef HAVE_MMX
218 static inline void lineNoiseAvg_MMX(uint8_t *dst, uint8_t *src, int len, int8_t **shift){
219 long mmx_len= len&(~7);
220
221 asm volatile(
222 "mov %5, %%"REG_a" \n\t"
223 ASMALIGN(4)
224 "1: \n\t"
225 "movq (%1, %%"REG_a"), %%mm1 \n\t"
226 "movq (%0, %%"REG_a"), %%mm0 \n\t"
227 "paddb (%2, %%"REG_a"), %%mm1 \n\t"
228 "paddb (%3, %%"REG_a"), %%mm1 \n\t"
229 "movq %%mm0, %%mm2 \n\t"
230 "movq %%mm1, %%mm3 \n\t"
231 "punpcklbw %%mm0, %%mm0 \n\t"
232 "punpckhbw %%mm2, %%mm2 \n\t"
233 "punpcklbw %%mm1, %%mm1 \n\t"
234 "punpckhbw %%mm3, %%mm3 \n\t"
235 "pmulhw %%mm0, %%mm1 \n\t"
236 "pmulhw %%mm2, %%mm3 \n\t"
237 "paddw %%mm1, %%mm1 \n\t"
238 "paddw %%mm3, %%mm3 \n\t"
239 "paddw %%mm0, %%mm1 \n\t"
240 "paddw %%mm2, %%mm3 \n\t"
241 "psrlw $8, %%mm1 \n\t"
242 "psrlw $8, %%mm3 \n\t"
243 "packuswb %%mm3, %%mm1 \n\t"
244 "movq %%mm1, (%4, %%"REG_a") \n\t"
245 "add $8, %%"REG_a" \n\t"
246 " js 1b \n\t"
247 :: "r" (src+mmx_len), "r" (shift[0]+mmx_len), "r" (shift[1]+mmx_len), "r" (shift[2]+mmx_len),
248 "r" (dst+mmx_len), "g" (-mmx_len)
249 : "%"REG_a
250 );
251
252 if(mmx_len!=len){
253 int8_t *shift2[3]={shift[0]+mmx_len, shift[1]+mmx_len, shift[2]+mmx_len};
254 lineNoiseAvg_C(dst+mmx_len, src+mmx_len, len-mmx_len, shift2);
255 }
256 }
257 #endif
258
259 static inline void lineNoiseAvg_C(uint8_t *dst, uint8_t *src, int len, int8_t **shift){
260 int i;
261 int8_t *src2= (int8_t*)src;
262
263 for(i=0; i<len; i++)
264 {
265 const int n= shift[0][i] + shift[1][i] + shift[2][i];
266 dst[i]= src2[i]+((n*src2[i])>>7);
267 }
268 }
269
270 /***************************************************************************/
271
272 static void noise(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int width, int height, FilterParam *fp){
273 int8_t *noise= fp->noise;
274 int y;
275 int shift=0;
276
277 if(!noise)
278 {
279 if(src==dst) return;
280
281 if(dstStride==srcStride) fast_memcpy(dst, src, srcStride*height);
282 else
283 {
284 for(y=0; y<height; y++)
285 {
286 fast_memcpy(dst, src, width);
287 dst+= dstStride;
288 src+= srcStride;
289 }
290 }
291 return;
292 }
293
294 for(y=0; y<height; y++)
295 {
296 if(fp->temporal) shift= rand()&(MAX_SHIFT -1);
297 else shift= nonTempRandShift[y];
298
299 if(fp->quality==0) shift&= ~7;
300 if (fp->averaged) {
301 lineNoiseAvg(dst, src, width, fp->prev_shift[y]);
302 fp->prev_shift[y][fp->shiftptr] = noise + shift;
303 } else {
304 lineNoise(dst, src, noise, width, shift);
305 }
306 dst+= dstStride;
307 src+= srcStride;
308 }
309 fp->shiftptr++;
310 if (fp->shiftptr == 3) fp->shiftptr = 0;
311 }
312
313 static int config(struct vf_instance_s* vf,
314 int width, int height, int d_width, int d_height,
315 unsigned int flags, unsigned int outfmt){
316
317 return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
318 }
319
320 static void get_image(struct vf_instance_s* vf, mp_image_t *mpi){
321 if(mpi->flags&MP_IMGFLAG_PRESERVE) return; // don't change
322 if(mpi->imgfmt!=vf->priv->outfmt) return; // colorspace differ
323 // ok, we can do pp in-place (or pp disabled):
324 vf->dmpi=vf_get_image(vf->next,mpi->imgfmt,
325 mpi->type, mpi->flags, mpi->w, mpi->h);
326 mpi->planes[0]=vf->dmpi->planes[0];
327 mpi->stride[0]=vf->dmpi->stride[0];
328 mpi->width=vf->dmpi->width;
329 if(mpi->flags&MP_IMGFLAG_PLANAR){
330 mpi->planes[1]=vf->dmpi->planes[1];
331 mpi->planes[2]=vf->dmpi->planes[2];
332 mpi->stride[1]=vf->dmpi->stride[1];
333 mpi->stride[2]=vf->dmpi->stride[2];
334 }
335 mpi->flags|=MP_IMGFLAG_DIRECT;
336 }
337
338 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){
339 mp_image_t *dmpi;
340
341 if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
342 // no DR, so get a new image! hope we'll get DR buffer:
343 vf->dmpi=vf_get_image(vf->next,vf->priv->outfmt,
344 MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
345 mpi->w,mpi->h);
346 //printf("nodr\n");
347 }
348 //else printf("dr\n");
349 dmpi= vf->dmpi;
350
351 noise(dmpi->planes[0], mpi->planes[0], dmpi->stride[0], mpi->stride[0], mpi->w, mpi->h, &vf->priv->lumaParam);
352 noise(dmpi->planes[1], mpi->planes[1], dmpi->stride[1], mpi->stride[1], mpi->w/2, mpi->h/2, &vf->priv->chromaParam);
353 noise(dmpi->planes[2], mpi->planes[2], dmpi->stride[2], mpi->stride[2], mpi->w/2, mpi->h/2, &vf->priv->chromaParam);
354
355 vf_clone_mpi_attributes(dmpi, mpi);
356
357 #ifdef HAVE_MMX
358 if(gCpuCaps.hasMMX) asm volatile ("emms\n\t");
359 #endif
360 #ifdef HAVE_MMX2
361 if(gCpuCaps.hasMMX2) asm volatile ("sfence\n\t");
362 #endif
363
364 return vf_next_put_image(vf,dmpi, pts);
365 }
366
367 static void uninit(struct vf_instance_s* vf){
368 if(!vf->priv) return;
369
370 if(vf->priv->chromaParam.noise) free(vf->priv->chromaParam.noise);
371 vf->priv->chromaParam.noise= NULL;
372
373 if(vf->priv->lumaParam.noise) free(vf->priv->lumaParam.noise);
374 vf->priv->lumaParam.noise= NULL;
375
376 free(vf->priv);
377 vf->priv=NULL;
378 }
379
380 //===========================================================================//
381
382 static int query_format(struct vf_instance_s* vf, unsigned int fmt){
383 switch(fmt)
384 {
385 case IMGFMT_YV12:
386 case IMGFMT_I420:
387 case IMGFMT_IYUV:
388 return vf_next_query_format(vf,vf->priv->outfmt);
389 }
390 return 0;
391 }
392
393 static void parse(FilterParam *fp, char* args){
394 char *pos;
395 char *max= strchr(args, ':');
396
397 if(!max) max= args + strlen(args);
398
399 fp->strength= atoi(args);
400 pos= strchr(args, 'u');
401 if(pos && pos<max) fp->uniform=1;
402 pos= strchr(args, 't');
403 if(pos && pos<max) fp->temporal=1;
404 pos= strchr(args, 'h');
405 if(pos && pos<max) fp->quality=1;
406 pos= strchr(args, 'p');
407 if(pos && pos<max) fp->pattern=1;
408 pos= strchr(args, 'a');
409 if(pos && pos<max) {
410 fp->temporal=1;
411 fp->averaged=1;
412 }
413
414 if(fp->strength) initNoise(fp);
415 }
416
417 static unsigned int fmt_list[]={
418 IMGFMT_YV12,
419 IMGFMT_I420,
420 IMGFMT_IYUV,
421 0
422 };
423
424 static int open(vf_instance_t *vf, char* args){
425 vf->config=config;
426 vf->put_image=put_image;
427 vf->get_image=get_image;
428 vf->query_format=query_format;
429 vf->uninit=uninit;
430 vf->priv=malloc(sizeof(struct vf_priv_s));
431 memset(vf->priv, 0, sizeof(struct vf_priv_s));
432 if(args)
433 {
434 char *arg2= strchr(args,':');
435 if(arg2) parse(&vf->priv->chromaParam, arg2+1);
436 parse(&vf->priv->lumaParam, args);
437 }
438
439 // check csp:
440 vf->priv->outfmt=vf_match_csp(&vf->next,fmt_list,IMGFMT_YV12);
441 if(!vf->priv->outfmt)
442 {
443 uninit(vf);
444 return 0; // no csp match :(
445 }
446
447
448 #ifdef HAVE_MMX
449 if(gCpuCaps.hasMMX){
450 lineNoise= lineNoise_MMX;
451 lineNoiseAvg= lineNoiseAvg_MMX;
452 }
453 #endif
454 #ifdef HAVE_MMX2
455 if(gCpuCaps.hasMMX2) lineNoise= lineNoise_MMX2;
456 // if(gCpuCaps.hasMMX) lineNoiseAvg= lineNoiseAvg_MMX2;
457 #endif
458
459 return 1;
460 }
461
462 const vf_info_t vf_info_noise = {
463 "noise generator",
464 "noise",
465 "Michael Niedermayer",
466 "",
467 open,
468 NULL
469 };
470
471 //===========================================================================//