File: itkScaledSingleValuedCostFunction.h

package info (click to toggle)
elastix 5.2.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 42,480 kB
  • sloc: cpp: 68,403; lisp: 4,118; python: 1,013; xml: 182; sh: 177; makefile: 33
file content (156 lines) | stat: -rw-r--r-- 5,553 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
/*=========================================================================
 *
 *  Copyright UMC Utrecht and contributors
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/

#ifndef itkScaledSingleValuedCostFunction_h
#define itkScaledSingleValuedCostFunction_h

#include "itkSingleValuedCostFunction.h"
#include "itkIntTypes.h" //temp, needed for IdentifierType

namespace itk
{
/**
 * \class ScaledSingleValuedCostFunction
 * \brief A cost function that applies a scaling to another cost function
 *
 * This class can be used to adapt an existing, badly scaled, cost function.
 *
 * By default it does not apply any scaling. Use the method SetUseScales(true)
 * to enable the use of scales.
 *
 * \ingroup Numerics
 */

class ScaledSingleValuedCostFunction : public SingleValuedCostFunction
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(ScaledSingleValuedCostFunction);

  /** Standard ITK-stuff. */
  using Self = ScaledSingleValuedCostFunction;
  using Superclass = SingleValuedCostFunction;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

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

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

  /** Typedefs inherited from the superclass. */
  using Superclass::MeasureType;
  using Superclass::DerivativeType;
  using Superclass::ParametersType;
  // temporary, untill it is fixed in the ITK4
  // typedef IdentifierType NumberOfParametersType; // temp, copied from itk::TransformBase
  using NumberOfParametersType = unsigned int; // temp, copied from itk::CostFunction
  using SingleValuedCostFunctionPointer = Superclass::Pointer;

  using ScalesType = Array<double>;

  /** Divide the parameters by the scales and call the GetValue routine
   * of the unscaled cost function.
   */
  MeasureType
  GetValue(const ParametersType & parameters) const override;

  /** Divide the parameters by the scales, call the GetDerivative routine
   * of the unscaled cost function and divide the resulting derivative by
   * the scales.
   */
  void
  GetDerivative(const ParametersType & parameters, DerivativeType & derivative) const override;

  /** Same procedure as in GetValue and GetDerivative. */
  void
  GetValueAndDerivative(const ParametersType & parameters,
                        MeasureType &          value,
                        DerivativeType &       derivative) const override;

  /** Ask the UnscaledCostFunction how many parameters it has. */
  NumberOfParametersType
  GetNumberOfParameters() const override;

  /** Set the cost function that needs scaling. */
  itkSetObjectMacro(UnscaledCostFunction, Superclass);
  /** Get the cost function that needs scaling. */
  itkGetModifiableObjectMacro(UnscaledCostFunction, Superclass);

  /** Set the scales. Also computes the squared scales, just in case users
   * call GetSquaredScales (for compatibility with the ITK convention). */
  virtual void
  SetScales(const ScalesType & scales);

  /** Get the scales. */
  itkGetConstReferenceMacro(Scales, ScalesType);

  /** The ITK convention is to use the squared scales. This function
   * takes the square root of the input scales and sets them as the
   * the actual scales */
  virtual void
  SetSquaredScales(const ScalesType & squaredScales);

  /** The ITK convention is to use the squared scales. This function
   * returns the squared actual scales. */
  itkGetConstReferenceMacro(SquaredScales, ScalesType);

  /** Set the flag to use scales or not. */
  itkSetMacro(UseScales, bool);

  /** Get the flag to use scales or not. */
  itkGetConstMacro(UseScales, bool);

  /** Set the flag to negate the cost function or not. */
  itkBooleanMacro(NegateCostFunction);

  /** Set the flag to negate the cost function or not. */
  itkSetMacro(NegateCostFunction, bool);
  /** Get the flag to negate the cost function or not. */
  itkGetConstMacro(NegateCostFunction, bool);

  /** Convert the parameters from scaled to unscaled: x = y/s. */
  virtual void
  ConvertScaledToUnscaledParameters(ParametersType & parameters) const;

  /** Convert the parameters from unscaled to scaled: y = x*s. */
  virtual void
  ConvertUnscaledToScaledParameters(ParametersType & parameters) const;

protected:
  /** The constructor. */
  ScaledSingleValuedCostFunction();
  /** The destructor. */
  ~ScaledSingleValuedCostFunction() override = default;

  /** PrintSelf. */
  void
  PrintSelf(std::ostream & os, Indent indent) const override;

private:
  /** Member variables. */
  ScalesType                      m_Scales{};
  ScalesType                      m_SquaredScales{};
  SingleValuedCostFunctionPointer m_UnscaledCostFunction{ nullptr };
  bool                            m_UseScales{ false };
  bool                            m_NegateCostFunction{ false };
};

} // end namespace itk

#endif // #ifndef itkScaledSingleValuedCostFunction_h