noise generating filter

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@6425 b3059339-0415-0410-9bf9-f77b7e298cf2

3 files changed, 299 insertions, 1 deletions

diff --git a/libmpcodecs/Makefile b/libmpcodecs/Makefile
index 6216a76810..9a877aa06f 100644
--- a/libmpcodecs/Makefile
+++ b/libmpcodecs/Makefile
@@ -6,7 +6,7 @@ LIBNAME2 = libmpencoders.a
 
 AUDIO_SRCS=dec_audio.c ad.c ad_a52.c ad_acm.c ad_alaw.c ad_dk3adpcm.c ad_dk4adpcm.c ad_dshow.c ad_dvdpcm.c ad_ffmpeg.c ad_hwac3.c ad_imaadpcm.c ad_mp3.c ad_msadpcm.c ad_pcm.c ad_roqaudio.c ad_msgsm.c ad_faad.c ad_vorbis.c ad_libmad.c ad_real.c
 VIDEO_SRCS=dec_video.c vd.c vd_null.c vd_real.c vd_cinepak.c vd_qtrpza.c vd_ffmpeg.c vd_dshow.c vd_vfw.c vd_odivx.c vd_divx4.c vd_raw.c vd_xanim.c vd_msvidc.c vd_fli.c vd_qtrle.c vd_qtsmc.c vd_roqvideo.c vd_cyuv.c vd_nuv.c vd_libmpeg2.c vd_msrle.c vd_huffyuv.c vd_zlib.c vd_mpegpes.c
-VFILTER_SRCS=vf.c vf_vo.c vf_crop.c vf_expand.c vf_pp.c vf_scale.c vf_format.c vf_yuy2.c vf_flip.c vf_rgb2bgr.c vf_rotate.c vf_mirror.c vf_palette.c vf_lavc.c vf_dvbscale.c vf_cropdetect.c vf_test.c
+VFILTER_SRCS=vf.c vf_vo.c vf_crop.c vf_expand.c vf_pp.c vf_scale.c vf_format.c vf_yuy2.c vf_flip.c vf_rgb2bgr.c vf_rotate.c vf_mirror.c vf_palette.c vf_lavc.c vf_dvbscale.c vf_cropdetect.c vf_test.c vf_noise.c
 ENCODER_SRCS=ve.c ve_divx4.c ve_lavc.c ve_vfw.c ve_rawrgb.c ve_libdv.c
 NATIVE_SRCS=native/RTjpegN.c native/cinepak.c native/cyuv.c native/fli.c native/minilzo.c native/msvidc.c native/nuppelvideo.c native/qtrle.c native/qtrpza.c native/qtsmc.c native/roqav.c native/xa_gsm.c
 
diff --git a/libmpcodecs/vf.c b/libmpcodecs/vf.c
index 4066d60ecc..bbb2ceef02 100644
--- a/libmpcodecs/vf.c
+++ b/libmpcodecs/vf.c
@@ -32,6 +32,7 @@ extern vf_info_t vf_info_lavc;
 extern vf_info_t vf_info_dvbscale;
 extern vf_info_t vf_info_cropdetect;
 extern vf_info_t vf_info_test;
+extern vf_info_t vf_info_noise;
 
 char** vo_plugin_args=(char**) NULL;
 
@@ -59,6 +60,7 @@ static vf_info_t* filter_list[]={
     &vf_info_dvbscale,
     &vf_info_cropdetect,
     &vf_info_test,
+    &vf_info_noise,
     NULL
 };
 
diff --git a/libmpcodecs/vf_noise.c b/libmpcodecs/vf_noise.c
new file mode 100644
index 0000000000..b8b573053e
--- /dev/null
+++ b/libmpcodecs/vf_noise.c
@@ -0,0 +1,296 @@
+/*
+    Copyright (C) 2002 Michael Niedermayer <michaelni@gmx.at>
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <inttypes.h>
+#include <math.h>
+
+#include "../config.h"
+#include "../mp_msg.h"
+#include "../cpudetect.h"
+
+#ifdef HAVE_MALLOC_H
+#include <malloc.h>
+#endif
+
+#include "img_format.h"
+#include "mp_image.h"
+#include "vf.h"
+#include "../libvo/fastmemcpy.h"
+
+#define MAX_NOISE 4096
+#define MAX_SHIFT 1024
+#define MAX_RES (MAX_NOISE-MAX_SHIFT)
+
+//===========================================================================//
+
+static inline void lineNoise_C(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift);
+
+static void (*lineNoise)(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift)= lineNoise_C;
+
+typedef struct FilterParam{
+	int strength;
+	int uniform;
+	int temporal;
+	int8_t *noise;
+}FilterParam;
+
+struct vf_priv_s {
+	FilterParam lumaParam;
+	FilterParam chromaParam;
+};
+
+static int nonTempRandShift[MAX_RES]= {-1};
+
+static int8_t *initNoise(FilterParam *fp){
+	int strength= fp->strength;
+	int uniform= fp->uniform;
+	int8_t *noise= memalign(16, MAX_NOISE*sizeof(int8_t)); //FIXME deallocate
+	int i;
+
+	srand(123457);
+
+	for(i=0; i<MAX_NOISE; i++)
+	{
+		if(uniform)
+			noise[i]= ((rand()/11)%strength) - strength/2;
+		else
+		{
+			double x1, x2, w, y1;
+			do {
+				x1 = 2.0 * rand()/(float)RAND_MAX - 1.0;
+				x2 = 2.0 * rand()/(float)RAND_MAX - 1.0;
+				w = x1 * x1 + x2 * x2;
+			} while ( w >= 1.0 );
+
+			w = sqrt( (-2.0 * log( w ) ) / w );
+			y1= x1 * w;
+
+			y1*= strength / sqrt(3.0); 
+			if     (y1<-128) y1=-128;
+			else if(y1> 127) y1= 127;
+			noise[i]= (int)y1;
+		}
+	}
+
+	if(nonTempRandShift[0]==-1){
+		for(i=0; i<MAX_RES; i++){
+			nonTempRandShift[i]= rand()&(MAX_SHIFT-1);
+		}
+	}
+
+	fp->noise= noise;
+	return noise;
+}
+
+#ifdef HAVE_MMX
+static inline void lineNoise_MMX(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
+	int mmx_len= len&(~7);
+	shift&= ~7;
+	noise+=shift;
+
+	asm volatile(
+		"movl %3, %%eax			\n\t"
+		"pcmpeqb %%mm7, %%mm7		\n\t"
+		"psllw $15, %%mm7		\n\t"
+		"packsswb %%mm7, %%mm7		\n\t"
+		".balign 16			\n\t"
+		"1:				\n\t"
+		"movq (%0, %%eax), %%mm0	\n\t"
+		"movq (%1, %%eax), %%mm1	\n\t"
+		"pxor %%mm7, %%mm0		\n\t"
+		"paddsb %%mm1, %%mm0		\n\t"
+		"pxor %%mm7, %%mm0		\n\t"
+		"movq %%mm0, (%2, %%eax)	\n\t"
+		"addl $8, %%eax			\n\t"
+		" js 1b				\n\t"
+		:: "r" (src+mmx_len), "r" (noise+mmx_len), "r" (dst+mmx_len), "g" (-mmx_len)
+		: "%eax"
+	);
+	if(mmx_len!=len)
+		lineNoise_C(dst+mmx_len, src+mmx_len, noise+mmx_len, len-mmx_len, 0);
+}
+#endif
+
+//duplicate of previous except movntq
+#ifdef HAVE_MMX2
+static inline void lineNoise_MMX2(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
+	int mmx_len= len&(~7);
+	shift&= ~7;
+	noise+=shift;
+
+	asm volatile(
+		"movl %3, %%eax			\n\t"
+		"pcmpeqb %%mm7, %%mm7		\n\t"
+		"psllw $15, %%mm7		\n\t"
+		"packsswb %%mm7, %%mm7		\n\t"
+		".balign 16			\n\t"
+		"1:				\n\t"
+		"movq (%0, %%eax), %%mm0	\n\t"
+		"movq (%1, %%eax), %%mm1	\n\t"
+		"pxor %%mm7, %%mm0		\n\t"
+		"paddsb %%mm1, %%mm0		\n\t"
+		"pxor %%mm7, %%mm0		\n\t"
+		"movntq %%mm0, (%2, %%eax)	\n\t"
+		"addl $8, %%eax			\n\t"
+		" js 1b				\n\t"
+		:: "r" (src+mmx_len), "r" (noise+mmx_len), "r" (dst+mmx_len), "g" (-mmx_len)
+		: "%eax"
+	);
+	if(mmx_len!=len)
+		lineNoise_C(dst+mmx_len, src+mmx_len, noise+mmx_len, len-mmx_len, 0);
+}
+#endif
+
+static inline void lineNoise_C(uint8_t *dst, uint8_t *src, int8_t *noise, int len, int shift){
+	int i;
+	noise+= shift;
+	for(i=0; i<len; i++)
+	{
+		int v= src[i]+ noise[i];
+		if(v>255) 	dst[i]=255; //FIXME optimize
+		else if(v<0) 	dst[i]=0;
+		else		dst[i]=v;
+	}
+}
+
+static void noise(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int width, int height, FilterParam *fp){
+	int8_t *noise= fp->noise;
+	int y;
+	int shift=0;
+
+	if(!noise)
+	{
+		if(dstStride==srcStride) memcpy(dst, src, srcStride*height);
+		else
+		{
+			for(y=0; y<height; y++)
+			{
+				memcpy(dst, src, width);
+				dst+= dstStride;
+				src+= srcStride;	
+			}
+		}
+		return;
+	}
+
+	for(y=0; y<height; y++)
+	{
+		if(fp->temporal)	shift=  rand()&(MAX_SHIFT  -1);
+		else			shift= nonTempRandShift[y];
+
+		lineNoise(dst, src, noise, width, shift);
+		dst+= dstStride;
+		src+= srcStride;
+	}
+}
+
+static int config(struct vf_instance_s* vf,
+        int width, int height, int d_width, int d_height,
+	unsigned int flags, unsigned int outfmt){
+    
+	return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
+}
+
+static void put_image(struct vf_instance_s* vf, mp_image_t *mpi){
+	mp_image_t *dmpi;
+
+	// hope we'll get DR buffer:
+	dmpi=vf_get_image(vf->next,mpi->imgfmt,
+		MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
+		mpi->w, mpi->h);
+
+	noise(dmpi->planes[0], mpi->planes[0], dmpi->stride[0], mpi->stride[0], mpi->w, mpi->h, &vf->priv->lumaParam);
+	noise(dmpi->planes[1], mpi->planes[1], dmpi->stride[1], mpi->stride[1], mpi->w/2, mpi->h/2, &vf->priv->chromaParam);
+	noise(dmpi->planes[2], mpi->planes[2], dmpi->stride[2], mpi->stride[2], mpi->w/2, mpi->h/2, &vf->priv->chromaParam);
+
+	dmpi->qscale=mpi->qscale;
+	dmpi->qstride=mpi->qstride;
+
+#ifdef HAVE_MMX
+	if(gCpuCaps.hasMMX) asm volatile ("emms\n\t");
+#endif
+#ifdef HAVE_MMX2
+	if(gCpuCaps.hasMMX2) asm volatile ("sfence\n\t");
+#endif
+
+	vf_next_put_image(vf,dmpi);
+}
+
+//===========================================================================//
+
+static int query_format(struct vf_instance_s* vf, unsigned int fmt){
+	switch(fmt)
+	{
+	case IMGFMT_YV12:
+	case IMGFMT_I420:
+	case IMGFMT_IYUV:
+		return vf_next_query_format(vf,fmt);
+	}
+	return 0;
+}
+
+static void parse(FilterParam *fp, char* args){
+	char *pos;
+	char *max= strchr(args, ':');
+
+	if(!max) max= args + strlen(args);
+
+	fp->strength= atoi(args);
+	pos= strchr(args, 'u');
+	if(pos && pos<max) fp->uniform=1;
+	pos= strchr(args, 't');
+	if(pos && pos<max) fp->temporal=1;
+
+	if(fp->strength) initNoise(fp);
+}
+
+static int open(vf_instance_t *vf, char* args){
+    vf->config=config;
+    vf->put_image=put_image;
+    vf->query_format=query_format;
+    vf->priv=malloc(sizeof(struct vf_priv_s));
+    memset(vf->priv, 0, sizeof(struct vf_priv_s));
+    if(args)
+    {
+	char *arg2= strchr(args,':');
+	if(arg2) parse(&vf->priv->chromaParam, arg2+1);
+	parse(&vf->priv->lumaParam, args);
+    }
+ 
+#ifdef HAVE_MMX
+    if(gCpuCaps.hasMMX) lineNoise= lineNoise_MMX;
+#endif
+#ifdef HAVE_MMX2
+    if(gCpuCaps.hasMMX2) lineNoise= lineNoise_MMX2;
+#endif
+    
+    return 1;
+}
+
+vf_info_t vf_info_noise = {
+    "noise genenerator",
+    "noise",
+    "Michael Niedermayer",
+    "",
+    open
+};
+
+//===========================================================================//