Fawkes API  Fawkes Development Version
median.cpp
1 
2 /***************************************************************************
3  * median.cpp - Implementation of a median filter
4  *
5  * Created: Mon Jun 05 15:02:36 2006
6  * Copyright 2005-2012 Tim Niemueller [www.niemueller.de]
7  ****************************************************************************/
8 
9 /* This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version. A runtime exception applies to
13  * this software (see LICENSE.GPL_WRE file mentioned below for details).
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU Library General Public License for more details.
19  *
20  * Read the full text in the LICENSE.GPL_WRE file in the doc directory.
21  */
22 
23 #include <fvfilters/median.h>
24 
25 #include <core/exception.h>
26 
27 #ifdef HAVE_IPP
28 # include <ippi.h>
29 #elif defined(HAVE_OPENCV)
30 # if CV_MAJOR_VERSION < 2 || (CV_MAJOR_VERSION == 2 && CV_MINOR_VERSION < 4)
31 # include <opencv/cv.h>
32 # endif
33 # include <opencv/cv.hpp>
34 #else
35 # error "Neither IPP nor OpenCV available"
36 #endif
37 
38 namespace firevision {
39 #if 0 /* just to make Emacs auto-indent happy */
40 }
41 #endif
42 
43 /** @class FilterMedian <fvfilters/median.h>
44  * Median filter.
45  * @author Tim Niemueller
46  */
47 
48 /** Constructor.
49  * @param mask_size size of median mask
50  */
51 FilterMedian::FilterMedian(unsigned int mask_size)
52  : Filter("FilterMedian")
53 {
54  this->mask_size = mask_size;
55 }
56 
57 
58 void
60 {
61 #if defined(HAVE_IPP)
62  IppiSize size;
63  size.width = src_roi[0]->width - mask_size;
64  size.height = src_roi[0]->height - mask_size;
65 
66  IppiSize mask = { mask_size, mask_size };
67  IppiPoint anchor = { (mask_size + 1) / 2, (mask_size + 1) / 2 };
68 
69  IppStatus status;
70 
71  // base + number of bytes to line y + pixel bytes
72  status = ippiFilterMedian_8u_C1R( src[0] + ((src_roi[0]->start.y + (mask_size + 1) / 2) * src_roi[0]->line_step) + ((src_roi[0]->start.x + ( mask_size + 1) / 2) * src_roi[0]->pixel_step), src_roi[0]->line_step,
73  dst + ((dst_roi->start.y + (mask_size + 1) / 2) * dst_roi->line_step) + ((dst_roi->start.x + ( mask_size + 1) / 2) * dst_roi->pixel_step), dst_roi->line_step,
74  size, mask, anchor );
75 
76  if ( status != ippStsNoErr ) {
77  throw fawkes::Exception("Median filter failed with %i\n", status);
78  }
79 #elif defined(HAVE_OPENCV)
80  cv::Mat srcm(src_roi[0]->height, src_roi[0]->width, CV_8UC1,
81  src[0] +
82  (src_roi[0]->start.y * src_roi[0]->line_step) +
83  (src_roi[0]->start.x * src_roi[0]->pixel_step),
84  src_roi[0]->line_step);
85 
86  if (dst == NULL) { dst = src[0]; dst_roi = src_roi[0]; }
87 
88  cv::Mat dstm(dst_roi->height, dst_roi->width, CV_8UC1,
89  dst +
93 
94  cv::medianBlur(srcm, dstm, mask_size);
95 #endif
96 }
97 
98 } // end namespace firevision
fawkes::upoint_t start
ROI start.
Definition: roi.h:119
unsigned int y
y coordinate
Definition: types.h:36
unsigned int x
x coordinate
Definition: types.h:35
unsigned int width
ROI width.
Definition: roi.h:121
virtual void apply()
Apply the filter.
Definition: median.cpp:59
unsigned char ** src
Source buffers, dynamically allocated by Filter ctor.
Definition: filter.h:65
Base class for exceptions in Fawkes.
Definition: exception.h:36
Filter interface.
Definition: filter.h:35
ROI ** src_roi
Source ROIs, dynamically allocated by Filter ctor.
Definition: filter.h:70
unsigned int height
ROI height.
Definition: roi.h:123
FilterMedian(unsigned int mask_size)
Constructor.
Definition: median.cpp:51
unsigned int line_step
line step
Definition: roi.h:129
unsigned char * dst
Destination buffer.
Definition: filter.h:67
unsigned int pixel_step
pixel step
Definition: roi.h:131
ROI * dst_roi
Destination ROI.
Definition: filter.h:72