]> git.sesse.net Git - vlc/blobdiff - modules/video_filter/gaussianblur.c
Use var_InheritString for --decklink-video-connection.
[vlc] / modules / video_filter / gaussianblur.c
index 948776836652727f2d8970b4433acce83299f36b..c736c8c213bc5a552f3d24480e5af8f203e3f5e8 100644 (file)
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
-#include <stdlib.h>                                      /* malloc(), free() */
-#include <string.h>
 
-#include <vlc/vlc.h>
-#include <vlc_vout.h>
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
 
-#include "vlc_filter.h"
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_memory.h>
 
-#include <math.h>                                                  /* exp() */
+#include <vlc_filter.h>
+#include "filter_picture.h"
+
+#include <math.h>                                          /* 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( _("Gaussian blur video filter") );
-    set_shortname( _( "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,
-               VLC_FALSE );
+               false )
 
-    set_callbacks( Create, Destroy );
-vlc_module_end();
+    set_callbacks( Create, Destroy )
+vlc_module_end ()
 
-static const char *ppsz_filter_options[] = {
+/*****************************************************************************
+ * Local prototypes
+ *****************************************************************************/
+static picture_t *Filter( filter_t *, picture_t * );
+
+static const char *const ppsz_filter_options[] = {
     "sigma", NULL
 };
 
+/* Comment this to use floats instead of integers (faster for bigger sigma
+ * values)
+ * For sigma = 2 ints are faster
+ * 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 *pi_distribution;
     int i_dim;
+
+    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);
-    int *pi_distribution = (int*)malloc( (2*i_dim+1) * (2*i_dim+1) * sizeof( int ) );
-    int x, y;
+    type_t *pt_distribution = xmalloc( (2*i_dim+1) * sizeof( type_t ) );
+    int x;
+
     for( x = -i_dim; x <= i_dim; x++ )
-        for( y = -i_dim; y <= i_dim; y++ )
-            pi_distribution[(i_dim+y)*(2*i_dim+1)+(i_dim+x)] =
-                (int)( exp(-(x*x+y*y)/(2.*f_sigma*f_sigma))
-                       / (2.*M_PI*f_sigma*f_sigma) * (double)(1<<16) );
+    {
+        const float f_distribution = sqrt( exp(-(x*x)/(f_sigma*f_sigma) ) / (2.*M_PI*f_sigma*f_sigma) );
+#ifdef DONT_USE_FLOATS
+        const float f_factor = 1 << 8;
+#else
+        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;
-    p_sys->pi_distribution = pi_distribution;
+    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_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_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 (%4.4s)",
+                 (char*)&(p_filter->fmt_in.video.i_chroma) );
+        return VLC_EGENERIC;
+    }
+
+    if( p_filter->fmt_in.video.i_chroma != p_filter->fmt_out.video.i_chroma )
+    {
+        msg_Err( p_filter, "Input and output chromas don't match" );
+        return VLC_EGENERIC;
+    }
+
     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 );
@@ -118,13 +166,20 @@ static int Create( vlc_object_t *p_this )
     msg_Dbg( p_filter, "gaussian distribution is %d pixels wide",
              p_filter->p_sys->i_dim*2+1 );
 
+    p_filter->p_sys->pt_buffer = NULL;
+    p_filter->p_sys->pt_scale = NULL;
+
     return VLC_SUCCESS;
 }
 
 static void Destroy( vlc_object_t *p_this )
 {
     filter_t *p_filter = (filter_t *)p_this;
-    free( p_filter->p_sys->pi_distribution );
+
+    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 );
 }
 
@@ -133,71 +188,113 @@ static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
     picture_t *p_outpic;
     filter_sys_t *p_sys = p_filter->p_sys;
     int i_plane;
+    const int i_dim = p_sys->i_dim;
+    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;
     }
-
-    for( i_plane = 0 ; i_plane < p_pic->i_planes ; i_plane++ )
+    if( !p_sys->pt_buffer )
     {
+        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 ) );
+    }
 
-        uint8_t *p_in = p_pic->p[i_plane].p_pixels;
-        uint8_t *p_out = p_outpic->p[i_plane].p_pixels;
-
-        const int i_visible_lines = p_pic->p[i_plane].i_visible_lines;
-        const int i_visible_pitch = p_pic->p[i_plane].i_visible_pitch;
-        const int i_pitch = p_pic->p[i_plane].i_pitch;
-
-        const int i_dim = p_sys->i_dim;
-        const int i_dim2 = 2*i_dim+1;
-        const int *pi_distribution = p_sys->pi_distribution;
+    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;
 
-        int i_line, i_col;
-        const int factor = i_plane ? 1 : 0;
+        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++ )
         {
-            uint8_t *p_o = &p_out[i_line*i_pitch];
             for( i_col = 0; i_col < i_visible_pitch ; i_col++ )
             {
-                int value = 0;
-                int scale = 0;
                 int x, y;
+                type_t t_value = 0;
+
                 for( y = __MAX( -i_dim, -i_line );
                      y <= __MIN( i_dim, i_visible_lines - i_line - 1 );
                      y++ )
                 {
-                    const int *pi_d = &pi_distribution[(y+i_dim)*i_dim2+i_dim];
-                    const uint8_t *p_i = &p_in[(i_line+(y>>factor))*i_pitch+(i_col)];
                     for( x = __MAX( -i_dim, -i_col );
                          x <= __MIN( i_dim, i_visible_pitch - i_col + 1 );
                          x++ )
                     {
-                         const int weight = pi_d[x];
-                         value += weight * p_i[x>>factor];
-                         scale += weight;
+                        t_value += pt_distribution[y+i_dim] *
+                                   pt_distribution[x+i_dim];
                     }
                 }
-                p_o[i_col] = value / scale;
+                pt_scale[i_line*i_pitch+i_col] = t_value;
             }
         }
     }
 
-    p_outpic->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;
+    pt_scale = p_sys->pt_scale;
+    for( i_plane = 0 ; i_plane < p_pic->i_planes ; i_plane++ )
+    {
 
-    if( p_pic->pf_release )
-        p_pic->pf_release( p_pic );
+        uint8_t *p_in = p_pic->p[i_plane].p_pixels;
+        uint8_t *p_out = p_outpic->p[i_plane].p_pixels;
+
+        const int i_visible_lines = p_pic->p[i_plane].i_visible_lines;
+        const int i_visible_pitch = p_pic->p[i_plane].i_visible_pitch;
+        const int i_in_pitch = p_pic->p[i_plane].i_pitch;
+
+        int i_line, i_col;
+        const int x_factor = p_pic->p[Y_PLANE].i_visible_pitch/i_visible_pitch-1;
+        const int y_factor = p_pic->p[Y_PLANE].i_visible_lines/i_visible_lines-1;
+
+        for( i_line = 0 ; i_line < i_visible_lines ; i_line++ )
+        {
+            for( i_col = 0; i_col < i_visible_pitch ; i_col++ )
+            {
+                type_t t_value = 0;
+                int x;
+                const int c = i_line*i_in_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++ )
+                {
+                    t_value += pt_distribution[x+i_dim] *
+                               p_in[c+(x>>x_factor)];
+                }
+                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++ )
+            {
+                type_t t_value = 0;
+                int y;
+                const int c = i_line*i_in_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++ )
+                {
+                    t_value += pt_distribution[y+i_dim] *
+                               pt_buffer[c+(y>>y_factor)*i_in_pitch];
+                }
+
+                const type_t t_scale = pt_scale[(i_line<<y_factor)*(i_in_pitch<<x_factor)+(i_col<<x_factor)];
+                p_out[i_line * p_outpic->p[i_plane].i_pitch + i_col] = (uint8_t)(t_value / t_scale); // FIXME wouldn't it be better to round instead of trunc ?
+            }
+        }
+    }
 
-    return p_outpic;
+    return CopyInfoAndRelease( p_outpic, p_pic );
 }