itkNumericTraitsArrayPixel.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  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
 *
 *         https://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 itkNumericTraitsArrayPixel_h
#define itkNumericTraitsArrayPixel_h
 
#include "itkNumericTraits.h"
#include "itkArray.h"
 
namespace itk
{
template <typename T>
class NumericTraits<Array<T>>
{
private:
  using ElementAbsType = typename NumericTraits<T>::AbsType;
  using ElementAccumulateType = typename NumericTraits<T>::AccumulateType;
  using ElementFloatType = typename NumericTraits<T>::FloatType;
  using ElementPrintType = typename NumericTraits<T>::PrintType;
  using ElementRealType = typename NumericTraits<T>::RealType;
 
public:
  using ValueType = T;
  using Self = Array<T>;

  using AbsType = Array<ElementAbsType>;

  using AccumulateType = Array<ElementAccumulateType>;

  using FloatType = Array<ElementFloatType>;

  using PrintType = Array<ElementPrintType>;

  using RealType = Array<ElementRealType>;

  using ScalarRealType = ElementRealType;

  using MeasurementVectorType = Self;

  static const Self
  max(const Self & a)
  {
    Self b(a.Size());
 
    b.Fill(NumericTraits<T>::max());
    return b;
  }
 
  static const Self
  min(const Self & a)
  {
    Self b(a.Size());
 
    b.Fill(NumericTraits<T>::min());
    return b;
  }
 
  static const Self
  ZeroValue(const Self & a)
  {
    Self b(a.Size());
 
    b.Fill(T{});
    return b;
  }
 
  static const Self
  OneValue(const Self & a)
  {
    Self b(a.Size());
 
    b.Fill(NumericTraits<T>::OneValue());
    return b;
  }
 
  static const Self
  NonpositiveMin(const Self & a)
  {
    Self b(a.Size());
    b.Fill(NumericTraits<T>::NonpositiveMin());
    return b;
  }
 
  static constexpr bool IsSigned = std::is_signed_v<ValueType>;
  static constexpr bool IsInteger = std::is_integral_v<ValueType>;
  static constexpr bool IsComplex = NumericTraits<ValueType>::IsComplex;

  static void
  SetLength(Array<T> & m, const unsigned int s)
  {
    m.SetSize(s);
    m.Fill(T{});
  }


  static size_t
  GetLength(const Array<T> & m)
  {
    return m.GetSize();
  }
 
  static void
  AssignToArray(const Self & v, MeasurementVectorType & mv)
  {
    mv = v;
  }
 
  template <typename TArray>
  static void
  AssignToArray(const Self & v, TArray & mv)
  {
    for (unsigned int i = 0; i < GetLength(v); ++i)
    {
      mv[i] = v[i];
    }
  }
};
} // end namespace itk
 
#endif // itkNumericTraitsArrayPixel_h