ITK  5.2.0
Insight Toolkit
itkMultivariateLegendrePolynomial.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright NumFOCUS
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 #ifndef itkMultivariateLegendrePolynomial_h
19 #define itkMultivariateLegendrePolynomial_h
20 
21 #include "itkIntTypes.h"
22 #include "itkIndent.h"
23 #include <vector>
24 #include "itkArray.h"
25 #include "ITKPolynomialsExport.h"
26 
27 namespace itk
28 {
76 
77 class ITKPolynomials_EXPORT MultivariateLegendrePolynomial
78 {
79 public:
81 
82  using DoubleArrayType = std::vector<double>;
83  using ULongArrayType = std::vector<unsigned long>;
84  using LongArrayType = std::vector<long>;
85 
88 
92 
96 
98  MultivariateLegendrePolynomial(unsigned int dimension, unsigned int degree, const DomainSizeType & domainSize);
99 
102 
104  unsigned int
105  GetDimension() const
106  {
107  return m_Dimension;
108  }
109 
111  unsigned int
112  GetDegree() const
113  {
114  return m_Degree;
115  }
116 
123  unsigned int
125  {
126  return m_NumberOfCoefficients;
127  }
128 
130  const DomainSizeType &
132  {
133  return m_DomainSize;
134  }
135 
142  {
143  public:
144  CoefficientVectorSizeMismatch(const std::size_t given, const std::size_t required)
145  : m_Required{ required }
146  , m_Given{ given }
147  {}
148 
149  std::size_t m_Required;
150  std::size_t m_Given;
151  };
152 
157  void
158  SetCoefficients(const CoefficientArrayType & coefficients);
159 
160  void
161  SetCoefficients(const ParametersType & coefficients);
162 
164  const CoefficientArrayType &
165  GetCoefficients() const;
166 
169  double
171  {
172  if (m_Dimension == 2)
173  {
174  if (index[1] != m_PrevY)
175  {
176  // normalized y [-1, 1]
177  double norm_y = m_NormFactor[1] * static_cast<double>(index[1] - 1);
178  this->CalculateXCoef(norm_y, m_CoefficientArray);
179  m_PrevY = index[1];
180  }
182 
183  // normalized x [-1, 1]
184  double norm_x = m_NormFactor[0] * static_cast<double>(index[0] - 1);
185 
186  return LegendreSum(norm_x, m_Degree, m_CachedXCoef);
187  }
188  else if (m_Dimension == 3)
189  {
190  if (index[2] != m_PrevZ)
191  {
192  // normalized z [-1, 1]
193  double norm_z = m_NormFactor[2] * static_cast<double>(index[2] - 1);
194  this->CalculateYCoef(norm_z, m_CoefficientArray);
195  m_PrevZ = index[2];
196  }
197 
198  if (index[1] != m_PrevY)
199  {
200  // normalized y [-1, 1]
201  double norm_y = m_NormFactor[1] * static_cast<double>(index[1] - 1);
202  this->CalculateXCoef(norm_y, m_CachedYCoef);
203  m_PrevY = index[1];
204  }
205 
206  // normalized x [-1, 1]
207  double norm_x = m_NormFactor[0] * static_cast<double>(index[0] - 1);
208  return this->LegendreSum(norm_x, m_Degree, m_CachedXCoef);
209  }
210  return 0;
211  }
212 
214  unsigned int
215  GetNumberOfCoefficients(unsigned int dimension, unsigned int degree);
216 
225  {
226  public:
228  : m_MultivariateLegendrePolynomial(polynomial)
229  , m_Dimension(m_MultivariateLegendrePolynomial->GetDimension())
230  , m_DomainSize(m_MultivariateLegendrePolynomial->GetDomainSize())
231  , m_IsAtEnd(false)
232  {
233  m_Index.resize(m_Dimension);
234  std::fill(m_Index.begin(), m_Index.end(), 0);
235  }
237 
238  void
240  {
241  m_IsAtEnd = false;
242  for (unsigned int dim = 0; dim < m_Dimension; dim++)
243  {
244  m_Index[dim] = 0;
245  }
246  }
247 
248  bool
249  IsAtEnd() const
250  {
251  return m_IsAtEnd;
252  }
253 
256  {
257  for (unsigned int dim = 0; dim < m_Dimension; dim++)
258  {
259  if (m_Index[dim] < static_cast<int>(m_DomainSize[dim] - 1))
260  {
261  m_Index[dim] += 1;
262  return *this;
263  }
264  else
265  {
266  if (dim == m_Dimension - 1)
267  {
268  m_IsAtEnd = true;
269  break;
270  }
271  else
272  {
273  m_Index[dim] = 0;
274  }
275  }
276  }
277  return *this;
278  }
279 
280  double
281  Get()
282  {
283  return m_MultivariateLegendrePolynomial->Evaluate(m_Index);
284  }
285 
286  private:
288  unsigned int m_Dimension;
291  bool m_IsAtEnd;
292  }; // end of class Iterator
293 
294  void
295  Print(std::ostream & os) const;
296 
297 protected:
298  void
299  PrintSelf(std::ostream & os, Indent indent) const;
300 
301  double
302  LegendreSum(const double x, int n, const CoefficientArrayType & coef, int offset = 0);
303 
304  void
305  CalculateXCoef(double norm_y, const CoefficientArrayType & coef);
306 
307  void
308  CalculateYCoef(double norm_z, const CoefficientArrayType & coef);
309 
310 private:
312  unsigned int m_Dimension;
313  unsigned int m_Degree;
315 
320 
324 }; // end of class
325 
326 ITKPolynomials_EXPORT std::ostream &
327  operator<<(std::ostream & os, const MultivariateLegendrePolynomial & poly);
328 } // end of namespace itk
329 #endif
itk::MultivariateLegendrePolynomial::CoefficientArrayType
DoubleArrayType CoefficientArrayType
Definition: itkMultivariateLegendrePolynomial.h:87
itk::MultivariateLegendrePolynomial::m_CachedZCoef
CoefficientArrayType m_CachedZCoef
Definition: itkMultivariateLegendrePolynomial.h:319
itk::MultivariateLegendrePolynomial::m_CachedYCoef
CoefficientArrayType m_CachedYCoef
Definition: itkMultivariateLegendrePolynomial.h:318
itk::MultivariateLegendrePolynomial::CoefficientVectorSizeMismatch
Exception object.
Definition: itkMultivariateLegendrePolynomial.h:141
itk::MultivariateLegendrePolynomial::GetDegree
unsigned int GetDegree() const
Definition: itkMultivariateLegendrePolynomial.h:112
itk::MultivariateLegendrePolynomial::GetDomainSize
const DomainSizeType & GetDomainSize() const
Definition: itkMultivariateLegendrePolynomial.h:131
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::Get
double Get()
Definition: itkMultivariateLegendrePolynomial.h:281
itk::MultivariateLegendrePolynomial
2D and 3D multivariate Legendre Polynomial
Definition: itkMultivariateLegendrePolynomial.h:77
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::m_Index
IndexType m_Index
Definition: itkMultivariateLegendrePolynomial.h:290
itk::operator<<
std::ostream & operator<<(std::ostream &os, const Array< TValue > &arr)
Definition: itkArray.h:218
itk::MultivariateLegendrePolynomial::m_CoefficientArray
CoefficientArrayType m_CoefficientArray
Definition: itkMultivariateLegendrePolynomial.h:316
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::operator++
SimpleForwardIterator & operator++()
Definition: itkMultivariateLegendrePolynomial.h:255
itk::MultivariateLegendrePolynomial::m_Degree
unsigned int m_Degree
Definition: itkMultivariateLegendrePolynomial.h:313
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::m_IsAtEnd
bool m_IsAtEnd
Definition: itkMultivariateLegendrePolynomial.h:291
itk::MultivariateLegendrePolynomial::m_PrevY
IndexValueType m_PrevY
Definition: itkMultivariateLegendrePolynomial.h:322
itk::Indent
Control indentation during Print() invocation.
Definition: itkIndent.h:49
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::m_MultivariateLegendrePolynomial
MultivariateLegendrePolynomial * m_MultivariateLegendrePolynomial
Definition: itkMultivariateLegendrePolynomial.h:287
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::m_DomainSize
DomainSizeType m_DomainSize
Definition: itkMultivariateLegendrePolynomial.h:289
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::IsAtEnd
bool IsAtEnd() const
Definition: itkMultivariateLegendrePolynomial.h:249
itk::MultivariateLegendrePolynomial::CoefficientVectorSizeMismatch::m_Required
std::vcl_size_t m_Required
Definition: itkMultivariateLegendrePolynomial.h:149
itk::MultivariateLegendrePolynomial::DoubleArrayType
std::vector< double > DoubleArrayType
Definition: itkMultivariateLegendrePolynomial.h:82
itk::MultivariateLegendrePolynomial::DomainSizeType
ULongArrayType DomainSizeType
Definition: itkMultivariateLegendrePolynomial.h:94
itk::MultivariateLegendrePolynomial::ULongArrayType
std::vector< unsigned long > ULongArrayType
Definition: itkMultivariateLegendrePolynomial.h:83
itkIndent.h
itk::MultivariateLegendrePolynomial::IndexType
LongArrayType IndexType
Definition: itkMultivariateLegendrePolynomial.h:95
itk::MultivariateLegendrePolynomial::GetNumberOfCoefficients
unsigned int GetNumberOfCoefficients() const
Definition: itkMultivariateLegendrePolynomial.h:124
itk::MultivariateLegendrePolynomial::m_CachedXCoef
CoefficientArrayType m_CachedXCoef
Definition: itkMultivariateLegendrePolynomial.h:317
itk::MultivariateLegendrePolynomial::m_NumberOfCoefficients
unsigned int m_NumberOfCoefficients
Definition: itkMultivariateLegendrePolynomial.h:314
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::SimpleForwardIterator
SimpleForwardIterator(MultivariateLegendrePolynomial *polynomial)
Definition: itkMultivariateLegendrePolynomial.h:227
itk::MultivariateLegendrePolynomial::CoefficientVectorSizeMismatch::m_Given
std::vcl_size_t m_Given
Definition: itkMultivariateLegendrePolynomial.h:150
itkIntTypes.h
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::m_Dimension
unsigned int m_Dimension
Definition: itkMultivariateLegendrePolynomial.h:288
itk::MultivariateLegendrePolynomial::SimpleForwardIterator
Iterator which only supports forward iteration and Begin(), IsAtEnd(), and Get() method which work ju...
Definition: itkMultivariateLegendrePolynomial.h:224
itk::MultivariateLegendrePolynomial::m_NormFactor
DoubleArrayType m_NormFactor
Definition: itkMultivariateLegendrePolynomial.h:321
itkArray.h
itk::MultivariateLegendrePolynomial::Evaluate
double Evaluate(IndexType &index)
Definition: itkMultivariateLegendrePolynomial.h:170
itk
The "itk" namespace contains all Insight Segmentation and Registration Toolkit (ITK) classes....
Definition: itkAnnulusOperator.h:24
itk::MultivariateLegendrePolynomial::m_Dimension
unsigned int m_Dimension
Definition: itkMultivariateLegendrePolynomial.h:312
itk::IndexValueType
signed long IndexValueType
Definition: itkIntTypes.h:90
itk::Array< double >
itk::MultivariateLegendrePolynomial::m_PrevZ
IndexValueType m_PrevZ
Definition: itkMultivariateLegendrePolynomial.h:323
itk::MultivariateLegendrePolynomial::CoefficientVectorSizeMismatch::CoefficientVectorSizeMismatch
CoefficientVectorSizeMismatch(const std::vcl_size_t given, const std::vcl_size_t required)
Definition: itkMultivariateLegendrePolynomial.h:144
itk::MultivariateLegendrePolynomial::LongArrayType
std::vector< long > LongArrayType
Definition: itkMultivariateLegendrePolynomial.h:84
itk::MultivariateLegendrePolynomial::SimpleForwardIterator::Begin
void Begin()
Definition: itkMultivariateLegendrePolynomial.h:239
itk::MultivariateLegendrePolynomial::m_DomainSize
DomainSizeType m_DomainSize
Definition: itkMultivariateLegendrePolynomial.h:311
itk::MultivariateLegendrePolynomial::GetDimension
unsigned int GetDimension() const
Definition: itkMultivariateLegendrePolynomial.h:105