itkOffset.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 itkOffset_h
#define itkOffset_h
 
#include "itkSize.h"
#include "itkMath.h"
 
#include <cstddef> // For ptrdiff_t.
 
namespace itk
{

 
template <unsigned int VDimension = 2>
struct ITK_TEMPLATE_EXPORT Offset final
{
public:
  // Using the `rule of zero` to this aggregate type
  // C++20 changes the definition of aggregate such that classes with any user-declared ctors are no longer aggregates.

  using Self = Offset;

  using OffsetType = Offset<VDimension>;
  using OffsetValueType = itk::OffsetValueType;

  static constexpr unsigned int Dimension = VDimension;

  static constexpr unsigned int
  GetOffsetDimension()
  {
    return VDimension;
  }
 

  const Self
  operator+(const Self & vec) const
  {
    Self result;
 
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      result[i] = m_InternalArray[i] + vec.m_InternalArray[i];
    }
    return result;
  }

  const Self
  operator+(const Size<VDimension> & sz) const
  {
    Self result;
 
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      result[i] = m_InternalArray[i] + sz[i];
    }
    return result;
  }

  const Self &
  operator+=(const Size<VDimension> & sz)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] += sz[i];
    }
    return *this;
  }

  const Self &
  operator-=(const Size<VDimension> & sz)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] -= sz[i];
    }
    return *this;
  }

  const Self
  operator-(const Self & vec) const
  {
    Self result;
 
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      result[i] = m_InternalArray[i] - vec.m_InternalArray[i];
    }
    return result;
  }

  const Self &
  operator+=(const Self & vec)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] += vec.m_InternalArray[i];
    }
    return *this;
  }

  const Self &
  operator-=(const Self & vec)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] -= vec.m_InternalArray[i];
    }
    return *this;
  }

  [[nodiscard]] const OffsetValueType *
  GetOffset() const
  {
    return m_InternalArray;
  }

  void
  SetOffset(const OffsetValueType val[VDimension])
  {
    std::copy_n(val, VDimension, m_InternalArray);
  }

  void
  SetElement(unsigned long element, OffsetValueType val)
  {
    m_InternalArray[element] = val;
  }

  [[nodiscard]] OffsetValueType
  GetElement(unsigned long element) const
  {
    return m_InternalArray[element];
  }

  void
  Fill(OffsetValueType value)
  {
    std::fill_n(begin(), size(), value);
  } // MATCH std::array assign, ITK Fill

  /*
   *  Ask the compiler to align a type to the maximum useful alignment for the target
   *  machine you are compiling for. Whenever you leave out the alignment factor in an
   *  aligned attribute specification, the compiler automatically sets the alignment
   *  for the type to the largest alignment that is ever used for any data type on
   *  the target machine you are compiling for. Doing this can often make copy
   *  operations more efficient, because the compiler can use whatever instructions
   *  copy the biggest chunks of memory when performing copies to or from the variables
   *  that have types that you have aligned this way.
   */
  static_assert(VDimension > 0, "Error: Only positive value sized VDimension allowed");
  alignas(OffsetValueType) OffsetValueType m_InternalArray[VDimension];

  template <typename TCoordinate>
  inline void
  CopyWithRound(const FixedArray<TCoordinate, VDimension> & point)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] = Math::Round<OffsetValueType>(point[i]);
    }
  }

  template <typename TCoordinate>
  inline void
  CopyWithCast(const FixedArray<TCoordinate, VDimension> & point)
  {
    for (unsigned int i = 0; i < VDimension; ++i)
    {
      m_InternalArray[i] = static_cast<OffsetValueType>(point[i]);
    }
  }

  static Self
  GetBasisOffset(unsigned int dim);
 
 
  // ======================= Mirror the access pattern behavior of the std::array class
  using value_type = itk::OffsetValueType;
  using reference = value_type &;
  using const_reference = const value_type &;
  using iterator = value_type *;
  using const_iterator = const value_type *;
  using size_type = unsigned int;
  using difference_type = ptrdiff_t;
  using reverse_iterator = std::reverse_iterator<iterator>;
  using const_reverse_iterator = std::reverse_iterator<const_iterator>;

  void
  assign(const value_type & newValue)
  {
    std::fill_n(begin(), size(), newValue);
  }
 
  void
  swap(Offset & other) noexcept
  {
    std::swap(m_InternalArray, other.m_InternalArray);
  }
 
  [[nodiscard]] constexpr const_iterator
  cbegin() const
  {
    return &m_InternalArray[0];
  }
 
  constexpr iterator
  begin()
  {
    return &m_InternalArray[0];
  }
 
  [[nodiscard]] constexpr const_iterator
  begin() const
  {
    return &m_InternalArray[0];
  }
 
  [[nodiscard]] constexpr const_iterator
  cend() const
  {
    return &m_InternalArray[VDimension];
  }
 
  constexpr iterator
  end()
  {
    return &m_InternalArray[VDimension];
  }
 
  [[nodiscard]] constexpr const_iterator
  end() const
  {
    return &m_InternalArray[VDimension];
  }
 
  reverse_iterator
  rbegin()
  {
    return reverse_iterator(end());
  }
 
  [[nodiscard]] const_reverse_iterator
  rbegin() const
  {
    return const_reverse_iterator(end());
  }
 
  reverse_iterator
  rend()
  {
    return reverse_iterator(begin());
  }
 
  [[nodiscard]] const_reverse_iterator
  rend() const
  {
    return const_reverse_iterator(begin());
  }
 
  [[nodiscard]] constexpr size_type
  size() const
  {
    return VDimension;
  }
 
  [[nodiscard]] constexpr size_type
  max_size() const
  {
    return VDimension;
  }
 
  [[nodiscard]] constexpr bool
  empty() const
  {
    return false;
  }
 
  reference
  operator[](size_type pos)
  {
    return m_InternalArray[pos];
  }
 
  const_reference
  operator[](size_type pos) const
  {
    return m_InternalArray[pos];
  }
 
  reference
  at(size_type pos)
  {
    ExceptionThrowingBoundsCheck(pos);
    return m_InternalArray[pos];
  }
 
  [[nodiscard]] const_reference
  at(size_type pos) const
  {
    ExceptionThrowingBoundsCheck(pos);
    return m_InternalArray[pos];
  }
 
  constexpr reference
  front()
  {
    return *begin();
  }
 
  [[nodiscard]] constexpr const_reference
  front() const
  {
    return *begin();
  }
 
  constexpr reference
  back()
  {
    return VDimension ? *(end() - 1) : *end();
  }
 
  [[nodiscard]] constexpr const_reference
  back() const
  {
    return VDimension ? *(end() - 1) : *end();
  }
 
  OffsetValueType *
  data()
  {
    return &m_InternalArray[0];
  }
 
  [[nodiscard]] const OffsetValueType *
  data() const
  {
    return &m_InternalArray[0];
  }
 
private:
  static void
  ExceptionThrowingBoundsCheck(size_type pos)
  {
    if (pos >= VDimension)
    {
      throw std::out_of_range("Out of range: `itk::Offset::at` argument `pos` (which is " + std::to_string(pos) +
                              ") should be less than `Dimension` (which is " + std::to_string(VDimension) + ")!");
    }
  }
 
}; //------------ End struct Offset
 
template <unsigned int VDimension>
Offset<VDimension>
Offset<VDimension>::GetBasisOffset(unsigned int dim)
{
  Self ind{};
  ind.m_InternalArray[dim] = 1;
  return ind;
}
 
template <unsigned int VDimension>
std::ostream &
operator<<(std::ostream & os, const Offset<VDimension> & ind)
{
  os << '[';
  const unsigned int dimlim = VDimension - 1;
  for (unsigned int i = 0; i < dimlim; ++i)
  {
    os << ind[i] << ", ";
  }
  if constexpr (VDimension >= 1)
  {
    os << ind[VDimension - 1];
  }
  os << ']';
  return os;
}
 
// ======================= Mirror the access pattern behavior of the std::array class
// Array comparisons.
template <unsigned int VDimension>
inline bool
operator==(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return std::equal(one.begin(), one.end(), two.begin());
}
 
template <unsigned int VDimension>
inline bool
operator!=(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return !(one == two);
}
 
template <unsigned int VDimension>
inline bool
operator<(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return std::lexicographical_compare(one.begin(), one.end(), two.begin(), two.end());
}
 
template <unsigned int VDimension>
inline bool
operator>(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return two < one;
}
 
template <unsigned int VDimension>
inline bool
operator<=(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return !(one > two);
}
 
template <unsigned int VDimension>
inline bool
operator>=(const Offset<VDimension> & one, const Offset<VDimension> & two)
{
  return !(one < two);
}
 
// Specialized algorithms [6.2.2.2].
template <unsigned int VDimension>
inline void
swap(Offset<VDimension> & one, Offset<VDimension> & two) noexcept
{
  std::swap(one.m_InternalArray, two.m_InternalArray);
}
 
} // end namespace itk
 
#endif