File: itkAddConstantToImageFilter.h

package info (click to toggle)
ants 2.5.4%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 11,672 kB
  • sloc: cpp: 85,685; sh: 15,850; perl: 863; xml: 115; python: 111; makefile: 68
file content (155 lines) | stat: -rw-r--r-- 4,105 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
/*=========================================================================

  Program:   Advanced Normalization Tools

  Copyright (c) ConsortiumOfANTS. All rights reserved.
  See accompanying COPYING.txt or
 https://github.com/stnava/ANTs/blob/master/ANTSCopyright.txt for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __itkAddConstantToImageFilter_h
#define __itkAddConstantToImageFilter_h

#include "itkUnaryFunctorImageFilter.h"
#include "itkNumericTraits.h"

namespace itk
{
/** \class AddConstantToImageFilter
 *
 * \brief Add a constant to all input pixels.
 *
 * This filter is templated over the input image type
 * and the output image type.
 *
 * \author Tom Vercauteren, INRIA & Mauna Kea Technologies
 *
 * Based on filters from the Insight Journal paper:
 * http://hdl.handle.net/1926/510
 *
 * \ingroup IntensityImageFilters  Multithreaded
 * \sa UnaryFunctorImageFilter
 */
namespace Functor
{
template <typename TInput, typename TConstant, typename TOutput>
class AddConstantTo
{
public:
  AddConstantTo()
    : m_Constant(NumericTraits<TConstant>::OneValue()){};
  ~AddConstantTo() = default;
  bool
  operator!=(const AddConstantTo & other) const
  {
    return !(*this == other);
  }

  bool
  operator==(const AddConstantTo & other) const
  {
    return other.m_Constant == m_Constant;
  }

  inline TOutput
  operator()(const TInput & A) const
  {
    // Because the user has to specify the constant we don't
    // check if the cte is not 0;
    return static_cast<TOutput>(A + m_Constant);
  }

  void
  SetConstant(TConstant ct)
  {
    this->m_Constant = ct;
  }

  const TConstant &
  GetConstant() const
  {
    return m_Constant;
  }

  TConstant m_Constant;
};
} // namespace Functor

template <typename TInputImage, typename TConstant, typename TOutputImage>
class AddConstantToImageFilter
  : public UnaryFunctorImageFilter<
      TInputImage,
      TOutputImage,
      Functor::AddConstantTo<typename TInputImage::PixelType, TConstant, typename TOutputImage::PixelType>>
{
public:
  /** Standard class typedefs. */
  typedef AddConstantToImageFilter Self;
  typedef UnaryFunctorImageFilter<
    TInputImage,
    TOutputImage,
    Functor::AddConstantTo<typename TInputImage::PixelType, TConstant, typename TOutputImage::PixelType>>
    Superclass;

  typedef SmartPointer<Self>       Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  /** Method for creation through the object factory. */
  itkNewMacro(Self);

  /** Run-time type information (and related methods). */
  itkTypeMacro(AddConstantToImageFilter, UnaryFunctorImageFilter);

  /** Set the constant that will be used to multiply all the image
   * pixels */
  void
  SetConstant(TConstant ct)
  {
    if (ct != this->GetFunctor().GetConstant())
    {
      this->GetFunctor().SetConstant(ct);
      this->Modified();
    }
  }

  const TConstant &
  GetConstant() const
  {
    return this->GetFunctor().GetConstant();
  }

#ifdef ITK_USE_CONCEPT_CHECKING
  /** Begin concept checking */
  itkConceptMacro(InputConvertibleToOutputCheck,
                  (Concept::Convertible<typename TInputImage::PixelType, typename TOutputImage::PixelType>));
  itkConceptMacro(
    Input1Input2OutputAddOperatorCheck,
    (Concept::AdditiveOperators<typename TInputImage::PixelType, TConstant, typename TOutputImage::PixelType>));
  /** End concept checking */
#endif
protected:
  AddConstantToImageFilter() = default;
  virtual ~AddConstantToImageFilter() = default;

  void
  PrintSelf(std::ostream & os, Indent indent) const
  {
    Superclass::PrintSelf(os, indent);

    os << indent << "Constant: " << static_cast<typename NumericTraits<TConstant>::PrintType>(this->GetConstant())
       << std::endl;
  }

private:
  AddConstantToImageFilter(const Self &) = delete;
  void
  operator=(const Self &) = delete;
};
} // end namespace itk

#endif