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
34namespace 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 */
44FilterMedian::FilterMedian(unsigned int mask_size) : Filter("FilterMedian")
45{
46 this->mask_size = mask_size;
47}
48
49void
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),
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,
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