1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
/* Copyright (c) 2008-2022 the MRtrix3 contributors.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
* Covered Software is provided under this License on an "as is"
* basis, without warranty of any kind, either expressed, implied, or
* statutory, including, without limitation, warranties that the
* Covered Software is free of defects, merchantable, fit for a
* particular purpose or non-infringing.
* See the Mozilla Public License v. 2.0 for more details.
*
* For more details, see http://www.mrtrix.org/.
*/
#ifndef __image_adapter_median_h__
#define __image_adapter_median_h__
#include "math/median.h"
#include "adapter/base.h"
namespace MR
{
namespace Adapter
{
template <class ImageType>
class Median :
public Base<Median<ImageType>,ImageType>
{ MEMALIGN(Median<ImageType>)
public:
using base_type = Base<Median<ImageType>, ImageType>;
using value_type = typename ImageType::value_type;
using voxel_type = Median;
using base_type::name;
using base_type::size;
using base_type::index;
Median (const ImageType& parent) :
base_type (parent) {
set_extent (vector<int>(1,3));
}
Median (const ImageType& parent, const vector<uint32_t>& extent) :
base_type (parent) {
set_extent (extent);
}
void set_extent (const vector<uint32_t>& ext)
{
for (size_t i = 0; i < ext.size(); ++i)
if (! (ext[i] & uint32_t(1)))
throw Exception ("expected odd number for extent");
if (ext.size() != 1 && ext.size() != 3)
throw Exception ("unexpected number of elements specified in extent");
if (ext.size() == 1)
extent = vector<uint32_t> (3, ext[0]);
else
extent = ext;
DEBUG ("median adapter for image \"" + name() + "\" initialised with extent " + str(extent));
for (size_t i = 0; i < 3; ++i)
extent[i] = (extent[i]-1)/2;
}
value_type value ()
{
const ssize_t old_pos [3] = { index(0), index(1), index(2) };
const ssize_t from[3] = {
index(0) < extent[0] ? 0 : index(0) - extent[0],
index(1) < extent[1] ? 0 : index(1) - extent[1],
index(2) < extent[2] ? 0 : index(2) - extent[2]
};
const ssize_t to[3] = {
index(0) >= size(0)-extent[0] ? size(0) : index(0)+extent[0]+1,
index(1) >= size(1)-extent[1] ? size(1) : index(1)+extent[1]+1,
index(2) >= size(2)-extent[2] ? size(2) : index(2)+extent[2]+1
};
values.clear();
for (index(2) = from[2]; index(2) < to[2]; ++index(2))
for (index(1) = from[1]; index(1) < to[1]; ++index(1))
for (index(0) = from[0]; index(0) < to[0]; ++index(0))
values.push_back (base_type::value());
index(0) = old_pos[0];
index(1) = old_pos[1];
index(2) = old_pos[2];
return Math::median (values);
}
protected:
vector<uint32_t> extent;
vector<value_type> values;
};
}
}
#endif
|
