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 <core/exception.h>
24 #include <fvfilters/median.h>
25 
26 #ifdef HAVE_IPP
27 # include <ippi.h>
28 #elif defined(HAVE_OPENCV)
29 # include <opencv2/opencv.hpp>
30 #else
31 # error "Neither IPP nor OpenCV available"
32 #endif
33 
34 namespace firevision {
35 
36 /** @class FilterMedian <fvfilters/median.h>
37  * Median filter.
38  * @author Tim Niemueller
39  */
40 
41 /** Constructor.
42  * @param mask_size size of median mask
43  */
44 FilterMedian::FilterMedian(unsigned int mask_size) : Filter("FilterMedian")
45 {
46  this->mask_size = mask_size;
47 }
48 
49 void
51 {
52 #if defined(HAVE_IPP)
53  IppiSize size;
54  size.width = src_roi[0]->width - mask_size;
55  size.height = src_roi[0]->height - mask_size;
56 
57  IppiSize mask = {mask_size, mask_size};
58  IppiPoint anchor = {(mask_size + 1) / 2, (mask_size + 1) / 2};
59 
60  IppStatus status;
61 
62  // base + number of bytes to line y + pixel bytes
63  status = ippiFilterMedian_8u_C1R(
64  src[0] + ((src_roi[0]->start.y + (mask_size + 1) / 2) * src_roi[0]->line_step)
65  + ((src_roi[0]->start.x + (mask_size + 1) / 2) * src_roi[0]->pixel_step),
66  src_roi[0]->line_step,
67  dst + ((dst_roi->start.y + (mask_size + 1) / 2) * dst_roi->line_step)
68  + ((dst_roi->start.x + (mask_size + 1) / 2) * dst_roi->pixel_step),
70  size,
71  mask,
72  anchor);
73 
74  if (status != ippStsNoErr) {
75  throw fawkes::Exception("Median filter failed with %i\n", status);
76  }
77 #elif defined(HAVE_OPENCV)
78  cv::Mat srcm(src_roi[0]->height,
79  src_roi[0]->width,
80  CV_8UC1,
81  src[0] + (src_roi[0]->start.y * src_roi[0]->line_step)
82  + (src_roi[0]->start.x * src_roi[0]->pixel_step),
83  src_roi[0]->line_step);
84 
85  if (dst == NULL) {
86  dst = src[0];
87  dst_roi = src_roi[0];
88  }
89 
90  cv::Mat dstm(dst_roi->height,
91  dst_roi->width,
92  CV_8UC1,
96 
97  cv::medianBlur(srcm, dstm, mask_size);
98 #endif
99 }
100 
101 } // end namespace firevision
Base class for exceptions in Fawkes.
Definition: exception.h:36
virtual void apply()
Apply the filter.
Definition: median.cpp:50
FilterMedian(unsigned int mask_size)
Constructor.
Definition: median.cpp:44
Filter interface.
Definition: filter.h:33
ROI ** src_roi
Source ROIs, dynamically allocated by Filter ctor.
Definition: filter.h:66
unsigned char ** src
Source buffers, dynamically allocated by Filter ctor.
Definition: filter.h:61
unsigned char * dst
Destination buffer.
Definition: filter.h:63
ROI * dst_roi
Destination ROI.
Definition: filter.h:68
unsigned int height
ROI height.
Definition: roi.h:119
fawkes::upoint_t start
ROI start.
Definition: roi.h:115
unsigned int line_step
line step
Definition: roi.h:125
unsigned int width
ROI width.
Definition: roi.h:117
unsigned int pixel_step
pixel step
Definition: roi.h:127
unsigned int x
x coordinate
Definition: types.h:36
unsigned int y
y coordinate
Definition: types.h:37