X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fvideo_filter%2Fgaussianblur.c;h=aa3c29da9e41a3e38e8300702d77dba15d596a41;hb=ce1b1bda347a6e4540b8c418b70f2b46f34726c7;hp=8a6536d6b748d4d7eca23c16e7a1b32b05e532f5;hpb=3561b9b28f58eb7a4183e158a8fd973800d31ceb;p=vlc diff --git a/modules/video_filter/gaussianblur.c b/modules/video_filter/gaussianblur.c index 8a6536d6b7..aa3c29da9e 100644 --- a/modules/video_filter/gaussianblur.c +++ b/modules/video_filter/gaussianblur.c @@ -31,42 +31,46 @@ #include #include -#include +#include -#include "vlc_filter.h" +#include +#include "filter_picture.h" #include /* exp(), sqrt() */ /***************************************************************************** - * Local prototypes + * Module descriptor *****************************************************************************/ static int Create ( vlc_object_t * ); static void Destroy ( vlc_object_t * ); -static picture_t *Filter( filter_t *, picture_t * ); - #define SIGMA_TEXT N_("Gaussian's std deviation") #define SIGMA_LONGTEXT N_( \ "Gaussian's standard deviation. The bluring will take " \ "into account pixels up to 3*sigma away in any direction.") +#define GAUSSIAN_HELP N_("Add a blurring effect") + #define FILTER_PREFIX "gaussianblur-" -/***************************************************************************** - * Module descriptor - *****************************************************************************/ -vlc_module_begin(); - set_description( N_("Gaussian blur video filter") ); - set_shortname( N_( "Gaussian Blur" )); - set_capability( "video filter2", 0 ); - set_category( CAT_VIDEO ); - set_subcategory( SUBCAT_VIDEO_VFILTER ); +vlc_module_begin () + set_description( N_("Gaussian blur video filter") ) + set_shortname( N_( "Gaussian Blur" )) + set_help(GAUSSIAN_HELP) + set_capability( "video filter2", 0 ) + set_category( CAT_VIDEO ) + set_subcategory( SUBCAT_VIDEO_VFILTER ) add_float( FILTER_PREFIX "sigma", 2., NULL, SIGMA_TEXT, SIGMA_LONGTEXT, - false ); + false ) + + set_callbacks( Create, Destroy ) +vlc_module_end () - set_callbacks( Create, Destroy ); -vlc_module_end(); +/***************************************************************************** + * Local prototypes + *****************************************************************************/ +static picture_t *Filter( filter_t *, picture_t * ); static const char *const ppsz_filter_options[] = { "sigma", NULL @@ -78,67 +82,60 @@ static const char *const ppsz_filter_options[] = { * For sigma = 4 floats are faster */ #define DONT_USE_FLOATS + +#ifdef DONT_USE_FLOATS +# define type_t int +#else +# define type_t float +#endif + struct filter_sys_t { double f_sigma; int i_dim; -#ifdef DONT_USE_FLOATS - int *pi_distribution; - int *pi_buffer; - int *pi_scale; -#else - float *pf_distribution; - float *pf_buffer; - float *pf_scale; -#endif + + type_t *pt_distribution; + type_t *pt_buffer; + type_t *pt_scale; }; static void gaussianblur_InitDistribution( filter_sys_t *p_sys ) { double f_sigma = p_sys->f_sigma; int i_dim = (int)(3.*f_sigma); -#ifdef DONT_USE_FLOATS - int *pi_distribution = (int*)malloc( (2*i_dim+1) * sizeof( int ) ); -#else - float *pf_distribution = (float*)malloc( (2*i_dim+1) * sizeof( float ) ); -#endif + type_t *pt_distribution = xmalloc( (2*i_dim+1) * sizeof( type_t ) ); int x; + for( x = -i_dim; x <= i_dim; x++ ) { + const float f_distribution = sqrt( exp(-(x*x)/(f_sigma*f_sigma) ) / (2.*M_PI*f_sigma*f_sigma) ); #ifdef DONT_USE_FLOATS - pi_distribution[i_dim+x] = - (int)( sqrt( exp(-(x*x)/(f_sigma*f_sigma) ) - / (2.*M_PI*f_sigma*f_sigma) ) * (double)(1<<8) ); - printf("%d\n",pi_distribution[i_dim+x]); + const float f_factor = 1 << 8; #else - pf_distribution[i_dim+x] = (float) - sqrt( exp(-(x*x)/(f_sigma*f_sigma) ) / (2.*M_PI*f_sigma*f_sigma) ); - printf("%f\n",pf_distribution[i_dim+x]); + const float f_factor = 1; #endif + + pt_distribution[i_dim+x] = (type_t)( f_distribution * f_factor ); + //printf("%f\n",(float)pt_distribution[i_dim+x]); } p_sys->i_dim = i_dim; -#ifdef DONT_USE_FLOATS - p_sys->pi_distribution = pi_distribution; -#else - p_sys->pf_distribution = pf_distribution; -#endif + p_sys->pt_distribution = pt_distribution; } static int Create( vlc_object_t *p_this ) { filter_t *p_filter = (filter_t *)p_this; - if( p_filter->fmt_in.video.i_chroma != VLC_FOURCC('I','4','2','0') - && p_filter->fmt_in.video.i_chroma != VLC_FOURCC('I','Y','U','V') - && p_filter->fmt_in.video.i_chroma != VLC_FOURCC('J','4','2','0') - && p_filter->fmt_in.video.i_chroma != VLC_FOURCC('Y','V','1','2') + if( p_filter->fmt_in.video.i_chroma != VLC_CODEC_I420 + && p_filter->fmt_in.video.i_chroma != VLC_CODEC_J420 + && p_filter->fmt_in.video.i_chroma != VLC_CODEC_YV12 - && p_filter->fmt_in.video.i_chroma != VLC_FOURCC('I','4','2','2') - && p_filter->fmt_in.video.i_chroma != VLC_FOURCC('J','4','2','2') + && p_filter->fmt_in.video.i_chroma != VLC_CODEC_I422 + && p_filter->fmt_in.video.i_chroma != VLC_CODEC_J422 ) { /* We only want planar YUV 4:2:0 or 4:2:2 */ - msg_Err( p_filter, "Unsupported input chroma (%4s)", + msg_Err( p_filter, "Unsupported input chroma (%4.4s)", (char*)&(p_filter->fmt_in.video.i_chroma) ); return VLC_EGENERIC; } @@ -151,10 +148,7 @@ static int Create( vlc_object_t *p_this ) p_filter->p_sys = malloc( sizeof( filter_sys_t ) ); if( p_filter->p_sys == NULL ) - { - msg_Err( p_filter, "out of memory" ); return VLC_ENOMEM; - } config_ChainParse( p_filter, FILTER_PREFIX, ppsz_filter_options, p_filter->p_cfg ); @@ -171,13 +165,9 @@ static int Create( vlc_object_t *p_this ) gaussianblur_InitDistribution( p_filter->p_sys ); msg_Dbg( p_filter, "gaussian distribution is %d pixels wide", p_filter->p_sys->i_dim*2+1 ); -#ifdef DONT_USE_FLOATS - p_filter->p_sys->pi_buffer = NULL; - p_filter->p_sys->pi_scale = NULL; -#else - p_filter->p_sys->pf_buffer = NULL; - p_filter->p_sys->pf_scale = NULL; -#endif + + p_filter->p_sys->pt_buffer = NULL; + p_filter->p_sys->pt_scale = NULL; return VLC_SUCCESS; } @@ -185,15 +175,11 @@ static int Create( vlc_object_t *p_this ) static void Destroy( vlc_object_t *p_this ) { filter_t *p_filter = (filter_t *)p_this; -#ifdef DONT_USE_FLOATS - free( p_filter->p_sys->pi_distribution ); - free( p_filter->p_sys->pi_buffer ); - free( p_filter->p_sys->pi_scale ); -#else - free( p_filter->p_sys->pf_distribution ); - free( p_filter->p_sys->pf_buffer ); - free( p_filter->p_sys->pf_scale ); -#endif + + free( p_filter->p_sys->pt_distribution ); + free( p_filter->p_sys->pt_buffer ); + free( p_filter->p_sys->pt_scale ); + free( p_filter->p_sys ); } @@ -203,73 +189,43 @@ static picture_t *Filter( filter_t *p_filter, picture_t *p_pic ) filter_sys_t *p_sys = p_filter->p_sys; int i_plane; const int i_dim = p_sys->i_dim; -#ifdef DONT_USE_FLOATS - int *pi_buffer; - int *pi_scale; - const int *pi_distribution = p_sys->pi_distribution; -#else - float *pf_buffer; - float *pf_scale; - const float *pf_distribution = p_sys->pf_distribution; -#endif + type_t *pt_buffer; + type_t *pt_scale; + const type_t *pt_distribution = p_sys->pt_distribution; + if( !p_pic ) return NULL; - p_outpic = p_filter->pf_vout_buffer_new( p_filter ); + p_outpic = filter_NewPicture( p_filter ); if( !p_outpic ) { - msg_Warn( p_filter, "can't get output picture" ); - if( p_pic->pf_release ) - p_pic->pf_release( p_pic ); + picture_Release( p_pic ); return NULL; } -#ifdef DONT_USE_FLOATS - if( !p_sys->pi_buffer ) + if( !p_sys->pt_buffer ) { - p_sys->pi_buffer = (int*)realloc( p_sys->pi_buffer, - p_pic->p[Y_PLANE].i_visible_lines - * p_pic->p[Y_PLANE].i_pitch - * sizeof( int ) ); + p_sys->pt_buffer = realloc_or_free( p_sys->pt_buffer, + p_pic->p[Y_PLANE].i_visible_lines * + p_pic->p[Y_PLANE].i_pitch * sizeof( type_t ) ); } - pi_buffer = p_sys->pi_buffer; -#else - if( !p_sys->pf_buffer ) - { - p_sys->pf_buffer = (float*)realloc( p_sys->pf_buffer, - p_pic->p[Y_PLANE].i_visible_lines - * p_pic->p[Y_PLANE].i_pitch - * sizeof( float ) ); - } - pf_buffer = p_sys->pf_buffer; -#endif -#ifdef DONT_USE_FLOATS - if( !p_sys->pi_scale ) -#else - if( !p_sys->pf_scale ) -#endif + + pt_buffer = p_sys->pt_buffer; + if( !p_sys->pt_scale ) { const int i_visible_lines = p_pic->p[Y_PLANE].i_visible_lines; const int i_visible_pitch = p_pic->p[Y_PLANE].i_visible_pitch; const int i_pitch = p_pic->p[Y_PLANE].i_pitch; int i_col, i_line; -#ifdef DONT_USE_FLOATS - p_sys->pi_scale = (int*)malloc( i_visible_lines * i_pitch - * sizeof( int ) ); - pi_scale = p_sys->pi_scale; -#else - p_sys->pf_scale = (float*)malloc( i_visible_lines * i_pitch - * sizeof( float ) ); - pf_scale = p_sys->pf_scale; -#endif + + p_sys->pt_scale = xmalloc( i_visible_lines * i_pitch * sizeof( type_t ) ); + pt_scale = p_sys->pt_scale; + for( i_line = 0 ; i_line < i_visible_lines ; i_line++ ) { for( i_col = 0; i_col < i_visible_pitch ; i_col++ ) { int x, y; -#ifdef DONT_USE_FLOATS - int value = 0; -#else - double value = 0.; -#endif + type_t t_value = 0; + for( y = __MAX( -i_dim, -i_line ); y <= __MIN( i_dim, i_visible_lines - i_line - 1 ); y++ ) @@ -278,29 +234,16 @@ static picture_t *Filter( filter_t *p_filter, picture_t *p_pic ) x <= __MIN( i_dim, i_visible_pitch - i_col + 1 ); x++ ) { -#ifdef DONT_USE_FLOATS - value += pi_distribution[y+i_dim] - * pi_distribution[x+i_dim]; -#else - value += ((double)pf_distribution[y+i_dim]) - * ((double)pf_distribution[x+i_dim]); -#endif + t_value += pt_distribution[y+i_dim] * + pt_distribution[x+i_dim]; } } -#ifdef DONT_USE_FLOATS - pi_scale[i_line*i_pitch+i_col] = value; -#else - pf_scale[i_line*i_pitch+i_col] = (float)(1./value); -#endif + pt_scale[i_line*i_pitch+i_col] = t_value; } } } -#ifdef DONT_USE_FLOATS - pi_scale = p_sys->pi_scale; -#else - pf_scale = p_sys->pf_scale; -#endif + pt_scale = p_sys->pt_scale; for( i_plane = 0 ; i_plane < p_pic->i_planes ; i_plane++ ) { @@ -319,72 +262,39 @@ static picture_t *Filter( filter_t *p_filter, picture_t *p_pic ) { for( i_col = 0; i_col < i_visible_pitch ; i_col++ ) { -#ifdef DONT_USE_FLOATS - int value = 0; -#else - float value = 0.; -#endif + type_t t_value = 0; int x; const int c = i_line*i_pitch+i_col; for( x = __MAX( -i_dim, -i_col*(x_factor+1) ); x <= __MIN( i_dim, (i_visible_pitch - i_col)*(x_factor+1) + 1 ); x++ ) { -#ifdef DONT_USE_FLOATS - value += pi_distribution[x+i_dim] - * p_in[c+(x>>x_factor)]; -#else - value += pf_distribution[x+i_dim] - * (float)p_in[c+(x>>x_factor)]; -#endif + t_value += pt_distribution[x+i_dim] * + p_in[c+(x>>x_factor)]; } -#ifdef DONT_USE_FLOATS - pi_buffer[c] = value; -#else - pf_buffer[c] = value; -#endif + pt_buffer[c] = t_value; } } for( i_line = 0 ; i_line < i_visible_lines ; i_line++ ) { for( i_col = 0; i_col < i_visible_pitch ; i_col++ ) { -#ifdef DONT_USE_FLOATS - int value = 0; -#else - float value = 0.; -#endif + type_t t_value = 0; int y; const int c = i_line*i_pitch+i_col; for( y = __MAX( -i_dim, (-i_line)*(y_factor+1) ); y <= __MIN( i_dim, (i_visible_lines - i_line)*(y_factor+1) - 1 ); y++ ) { -#ifdef DONT_USE_FLOATS - value += pi_distribution[y+i_dim] - * pi_buffer[c+(y>>y_factor)*i_pitch]; -#else - value += pf_distribution[y+i_dim] - * pf_buffer[c+(y>>y_factor)*i_pitch]; -#endif + t_value += pt_distribution[y+i_dim] * + pt_buffer[c+(y>>y_factor)*i_pitch]; } -#ifdef DONT_USE_FLOATS - p_out[c] = (uint8_t)(value/pi_scale[(i_line<date = p_pic->date; - p_outpic->b_force = p_pic->b_force; - p_outpic->i_nb_fields = p_pic->i_nb_fields; - p_outpic->b_progressive = p_pic->b_progressive; - p_outpic->b_top_field_first = p_pic->b_top_field_first; - - if( p_pic->pf_release ) - p_pic->pf_release( p_pic ); - - return p_outpic; + return CopyInfoAndRelease( p_outpic, p_pic ); }