Add conv5x5 kernel

4 files changed, 32 insertions, 2 deletions

diff --git a/src/gstdsp66videokernel.c b/src/gstdsp66videokernel.c
index 525c768..48c7764 100644
--- a/src/gstdsp66videokernel.c
+++ b/src/gstdsp66videokernel.c
@@ -61,11 +61,13 @@ enum
 static const GEnumValue dsp66_video_kernel_filtersizes[] = {
   {GST_DSP66_VIDEO_KERNEL_FILTERSIZE_5, "Kernel of 5 neighbour pixels", "5"},
   {GST_DSP66_VIDEO_KERNEL_FILTERSIZE_9, "Kernel of 9 neighbour pixels", "9"},
+  {GST_DSP66_VIDEO_KERNEL_FILTERSIZE_25, "Kernel of 25 neighbour pixels", "25"},
   {0, NULL, NULL},
 };
 static const GEnumValue dsp66_video_kerneltype[] = {
   {GST_DSP66_VIDEO_KERNELTYPE_MEDIAN, "Kernel median", "0"},
   {GST_DSP66_VIDEO_KERNELTYPE_SOBEL, "Kernel sobel", "1"},
+  {GST_DSP66_VIDEO_KERNELTYPE_CONV,  "Kernel conv",  "2"},
   {0, NULL, NULL},
 };
 
diff --git a/src/gstdsp66videokernel.h b/src/gstdsp66videokernel.h
index 7cb2c72..9d79b15 100644
--- a/src/gstdsp66videokernel.h
+++ b/src/gstdsp66videokernel.h
@@ -46,11 +46,13 @@ typedef enum
 {
   GST_DSP66_VIDEO_KERNEL_FILTERSIZE_5  = 5,
   GST_DSP66_VIDEO_KERNEL_FILTERSIZE_9  = 9,
+  GST_DSP66_VIDEO_KERNEL_FILTERSIZE_25 = 25
 } GstDsp66VideoKernelFilterSize;
 
 typedef enum {
   GST_DSP66_VIDEO_KERNELTYPE_MEDIAN = 0,
-  GST_DSP66_VIDEO_KERNELTYPE_SOBEL
+  GST_DSP66_VIDEO_KERNELTYPE_SOBEL  = 1,
+  GST_DSP66_VIDEO_KERNELTYPE_CONV   = 2
 } GstDsp66VideoKernelType;
 
 struct _GstDsp66VideoKernel {
diff --git a/src/kernels/oclconv/conv.cl b/src/kernels/oclconv/conv.cl
index bc9aced..6c86e52 100644
--- a/src/kernels/oclconv/conv.cl
+++ b/src/kernels/oclconv/conv.cl
@@ -29,7 +29,8 @@
 #define PIX_SWAP(a,b) { unsigned char temp=(a);(a)=(b);(b)=temp; }
 
 void IMG_median_3x3_8 (const unsigned char *restrict in_data, int cols, unsigned char * restrict out_data);
-void IMG_sobel_3x3_8  (const unsigned char *restrict in_data, unsigned char * restrict out_data, int rows, int cols);
+void IMG_sobel_3x3_8  (const unsigned char *restrict in_data, unsigned char *restrict out_data, int rows, int cols);
+void IMG_conv_3x3_i8_c8s (const unsigned char *restrict in_data, unsigned char *restrict out_data, int cols, const char *restrict mask, int shift);
 
 kernel void Median3x3(global const uchar* src, global uchar *dest,
                       const int width, const int height,
@@ -52,6 +53,25 @@ kernel void Sobel3x3(global const uchar* src, global uchar *dest,
   IMG_sobel_3x3_8((const unsigned char *)src, (const unsigned char *)dest, width, height);
 }
 
+kernel void Conv5x5(global const uchar* src, global uchar *dest,
+                      const int width, const int height,
+                      const int dstride, const int sstride)
+{
+  int i;
+  const char conv_mask5x5[25] = {
+     1, 4, 6, 4, 1,
+     4,16,24,16, 4,
+     6,24,36,24, 6,
+     4,16,24,16, 4,
+     1, 4, 6, 4, 1
+  };
+    for (i = 0; i < height; i++) {
+      IMG_conv_5x5_i8_c8s ((const unsigned char *)src, (unsigned char *)dest, width, sstride, conv_mask5x5, 8); 
+      src  += sstride;
+      dest += dstride; 
+    }
+}
+
 kernel void Median2x2 (global const uchar* src, global uchar *dest,
                       const int width, const int height,
                       const int dstride, const int sstride)
diff --git a/src/kernels/oclconv/oclconv.cpp b/src/kernels/oclconv/oclconv.cpp
index 601ea32..c7b722c 100644
--- a/src/kernels/oclconv/oclconv.cpp
+++ b/src/kernels/oclconv/oclconv.cpp
@@ -108,6 +108,12 @@ int oclconv_kernel(int kernel_type, int filter_size,
         retval = oclconv_imgproc("Sobel3x3", data_in, data_out, width, height, sstride, dstride);
       }
       break;
+    case 2: /* conv */
+      if(filter_size == 25) { 
+        retval = oclconv_imgproc("Conv5x5", data_in, data_out, width, height, sstride, dstride);
+        return 0;
+      }
+      break;
     default:
       break;
   }