/*=========================================================================
 *
 *  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 itkNumericTraitsRGBAPixel_h
#define itkNumericTraitsRGBAPixel_h

#include "itkNumericTraits.h"
#include "itkRGBAPixel.h"

namespace itk
{
/**
 * \brief Define numeric traits for RGBAPixel.
 * \tparam T Component type of RBGAPixel
 *
 * We provide here a generic implementation based on creating types of
 * RGBAPixel whose components are the types of the NumericTraits from
 * the original RGBAPixel components. This implementation require
 * support for partial specializations, since it is based on the
 * concept that:
 *   NumericTraits<RGBAPixel< T > >  is defined piecewise by
 *   RGBAPixel< NumericTraits< T > >
 *
 * \sa NumericTraits
 * \ingroup DataRepresentation
 * \ingroup ITKCommon
 */
template <typename T>
class NumericTraits<RGBAPixel<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:
  /** Return the type of the native component type. */
  using ValueType = T;

  using Self = RGBAPixel<T>;

  /** Unsigned component type */
  using AbsType = RGBAPixel<ElementAbsType>;

  /** Accumulation of addition and multiplication. */
  using AccumulateType = RGBAPixel<ElementAccumulateType>;

  /** Typedef for operations that use floating point instead of real precision
   */
  using FloatType = RGBAPixel<ElementFloatType>;

  /** Return the type that can be printed. */
  using PrintType = RGBAPixel<ElementPrintType>;

  /** Type for real-valued scalar operations. */
  using RealType = RGBAPixel<ElementRealType>;

  /** Type for real-valued scalar operations. */
  using ScalarRealType = ElementRealType;

  /** Measurement vector type */
  using MeasurementVectorType = Self;

  /** Component wise defined elements
   *
   * \note minimum value for floating pointer types is defined as
   * minimum positive normalize value.
   */
  static const Self
  max(const Self &)
  {
    return MakeFilled<Self>(NumericTraits<T>::max());
  }

  static const Self
  min(const Self &)
  {
    return MakeFilled<Self>(NumericTraits<T>::min());
  }

  static const Self
  max()
  {
    return MakeFilled<Self>(NumericTraits<T>::max());
  }

  static const Self
  min()
  {
    return MakeFilled<Self>(NumericTraits<T>::min());
  }

  static const Self
  NonpositiveMin()
  {
    return MakeFilled<Self>(NumericTraits<T>::NonpositiveMin());
  }

  static const Self
  ZeroValue()
  {
    return Self{};
  }

  static const Self
  OneValue()
  {
    return MakeFilled<Self>(NumericTraits<T>::OneValue());
  }

  static const Self
  NonpositiveMin(const Self &)
  {
    return NonpositiveMin();
  }

  static const Self
  ZeroValue(const Self &)
  {
    return ZeroValue();
  }

  static const Self
  OneValue(const Self &)
  {
    return OneValue();
  }

  /** Is a given value positive? */
  static bool
  IsPositive(Self val)
  {
    return NumericTraits<typename Self::LuminanceType>::IsPositive(val.GetLuminance());
  }

  /** Is a given value nonpositive? */
  static bool
  IsNonpositive(Self val)
  {
    return NumericTraits<typename Self::LuminanceType>::IsNonpositive(val.GetLuminance());
  }

  /** Is a given value negative? */
  static bool
  IsNegative(Self val)
  {
    return NumericTraits<typename Self::LuminanceType>::IsNegative(val.GetLuminance());
  }

  /** Is a given value nonnegative? */
  static bool
  IsNonnegative(Self val)
  {
    return NumericTraits<typename Self::LuminanceType>::IsNonnegative(val.GetLuminance());
  }

  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;

  /** RGBA pixels must have 4 components, so the size cannot be
   *  set to anything besides 4.  If called with size of 4, this
   *  function will fill the pixel with zeros. */
  static void
  SetLength(RGBAPixel<T> & m, const unsigned int s)
  {
    if (s != 4)
    {
      itkGenericExceptionMacro("Cannot set the size of a RGBAPixel to anything other "
                               "than 4.");
    }
    m.Fill(T{});
  }

  /** Return the dimensionality of the pixel. Always returns 4. */
  static unsigned int
  GetLength(const RGBAPixel<T> &)
  {
    return 4;
  }

  /** Return the dimensionality of the pixel. Always returns 4. */
  static unsigned int
  GetLength()
  {
    return 4;
  }

  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 < 4; ++i)
    {
      mv[i] = v[i];
    }
  }

  /** \note: the functions are preferred over the member variables as
   * they are defined for all partial specialization
   */
  static const Self ITKCommon_EXPORT Zero;
  static const Self ITKCommon_EXPORT One;
};
} // end namespace itk

#endif // itkNumericTraitsRGBAPixel_h
