File: threshold.cpp

package info (click to toggle)
sight 25.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 43,252 kB
  • sloc: cpp: 310,629; xml: 17,622; ansic: 9,960; python: 1,379; sh: 144; makefile: 33
file content (188 lines) | stat: -rw-r--r-- 6,867 bytes parent folder | download
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
/************************************************************************
 *
 * Copyright (C) 2009-2024 IRCAD France
 * Copyright (C) 2012-2020 IHU Strasbourg
 *
 * This file is part of Sight.
 *
 * Sight is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Sight is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Sight. If not, see <https://www.gnu.org/licenses/>.
 *
 ***********************************************************************/

#include "module/filter/image/threshold.hpp"

#include <core/com/signal.hxx>
#include <core/tools/dispatcher.hpp>

#include <data/image.hpp>
#include <data/image_series.hpp>

namespace sight::module::filter::image
{

//-----------------------------------------------------------------------------

threshold::threshold() noexcept :
    filter(m_signals)
{
}

//-----------------------------------------------------------------------------

void threshold::starting()
{
}

//-----------------------------------------------------------------------------

void threshold::stopping()
{
    m_target.reset();
}

//-----------------------------------------------------------------------------

void threshold::configuring()
{
    const service::config_t& srv_config = this->get_config();

    SIGHT_ASSERT("You must have one <config/> element.", srv_config.count("config") == 1);

    const service::config_t& config = srv_config.get_child("config");

    SIGHT_ASSERT("You must have one <threshold/> element.", config.count("threshold") == 1);
    const service::config_t& threshold_cfg = config.get_child("threshold");
    m_threshold = threshold_cfg.get_value<double>();
}

//-----------------------------------------------------------------------------

/**
 * Functor to apply a threshold filter.
 *
 * The pixel with the value less than the threshold value will be set to 0, else the value is set to the maximum
 * value of the image pixel type.
 *
 * The functor provides a template method operator(param) to apply the filter
 */
struct threshold_filter
{
    struct parameter
    {
        double threshold_value {};   ///< threshold value.
        data::image::csptr image_in; ///< image source
        data::image::sptr image_out; ///< image target: contains the result of the filter
    };

    /**
     * @brief Applies the filter
     * @tparam PIXELTYPE image pixel type (uint8, uint16, int8, int16, float, double, ....)
     */
    template<class PIXELTYPE>
    void operator()(parameter& _param)
    {
        const auto threshold_value  = static_cast<PIXELTYPE>(_param.threshold_value);
        data::image::csptr image_in = _param.image_in;
        data::image::sptr image_out = _param.image_out;
        SIGHT_ASSERT("Sorry, image must be 3D", image_in->num_dimensions() == 3);

        image_out->copy_information(image_in); // Copy image size, type... without copying the buffer
        image_out->resize(image_out->size(), image_out->type(), image_out->pixel_format());
        const auto lockin   = image_in->dump_lock();
        const auto lock_out = image_out->dump_lock();

        // Get iterators on image buffers
        auto it1           = image_in->begin<PIXELTYPE>();
        const auto it1_end = image_in->end<PIXELTYPE>();
        auto it2           = image_out->begin<PIXELTYPE>();
        const auto it2_end = image_out->end<PIXELTYPE>();

        const PIXELTYPE max_value = std::numeric_limits<PIXELTYPE>::max();

        // Fill the target buffer considering the threshold
        for( ; it1 != it1_end && it2 != it2_end ; ++it1, ++it2)
        {
            *it2 = (*it1 < threshold_value) ? 0 : max_value;
        }
    }
};

//-----------------------------------------------------------------------------

void threshold::updating()
{
    threshold_filter::parameter param; // filter parameters: threshold value, image source, image target

    // retrieve the input object
    auto input = m_source.lock();

    // try to dynamic cast to an image and an ImageSeries to know which type of data we use
    data::image_series::csptr image_series_src =
        std::dynamic_pointer_cast<const data::image_series>(input.get_shared());
    data::image::csptr image_src = std::dynamic_pointer_cast<const data::image>(input.get_shared());
    data::object::sptr output;

    // Get source/target image
    if(image_series_src)
    {
        param.image_in = image_series_src;
        data::image_series::sptr image_series_dest = data::image_series::copy(image_series_src);
        // define the input image series as the reference
        image_series_dest->set_dicom_reference(image_series_src->get_dicom_reference());

        // create the output image
        data::image::sptr image_out = std::make_shared<data::image>();
        image_series_dest->image::shallow_copy(image_out);
        param.image_out = image_series_dest;
        output          = image_series_dest;
    }
    else if(image_src)
    {
        param.image_in = image_src;
        // create the output image
        data::image::sptr image_out = std::make_shared<data::image>();
        param.image_out = image_out;
        output          = image_out;
    }
    else
    {
        SIGHT_THROW("Wrong type: source type must be an ImageSeries or an image");
    }

    param.threshold_value = m_threshold;

    // get image type
    core::type type = param.image_in->type();

    /* The dispatcher allows to apply the filter on any type of image.
     * It invokes the template functor ThresholdFilter using the image type.
     * - template parameters:
     *   - core::tools::supported_dispatcher_types defined all the supported type of the functor, here all the type
     *     supported by core::type(std::int8_t, std::uint8_t, std::int16_t, std::uint16_t, std::int32_t,
     *     std::uint32_t, std::int64_t, std::uint64_t, float, double)
     *   - ThresholdFilter: functor struct or class
     * - parameters:
     *   - type: core::type of the image
     *   - param: struct containing the functor parameters (here the input and output images and the threshold value)
     */

    core::tools::dispatcher<core::tools::supported_dispatcher_types, threshold_filter>::invoke(type, param);

    // register the output image to be accesible by the other service from the XML configuration
    m_target = output;
}

//-----------------------------------------------------------------------------

} // namespace sight::module::filter::image