itkRGBAPixel.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  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 itkRGBAPixel_h
#define itkRGBAPixel_h

// Undefine an eventual RGBAPixel macro
#ifdef RGBAPixel
#undef RGBAPixel
#endif

#include "itkIndent.h"
#include "itkFixedArray.h"
#include "itkMath.h"

namespace itk
{
template< typename TComponent = unsigned short >
class ITK_TEMPLATE_EXPORT RGBAPixel:public FixedArray< TComponent, 4 >
{
public:
  using Self = RGBAPixel;
  using Superclass = FixedArray< TComponent, 4 >;

  static constexpr unsigned int Dimension = 4;

  static constexpr unsigned int Length = 4;

  using BaseArray = FixedArray< TComponent, 4 >;

  using ComponentType = TComponent;
  using LuminanceType = typename NumericTraits< ComponentType >::RealType;

  RGBAPixel() { this->Fill(0); }
  RGBAPixel(const RGBAPixel &) = default;
  RGBAPixel & operator=(const RGBAPixel &) = default;
  RGBAPixel( RGBAPixel &&) = default;
  RGBAPixel & operator=(RGBAPixel &&) = default;
  ~RGBAPixel() = default;

  template< typename TRGBAPixelValueType >
  RGBAPixel(const RGBAPixel< TRGBAPixelValueType > & r):BaseArray(r) {}
  RGBAPixel(const ComponentType r[4]):BaseArray(r) {}
  RGBAPixel(const ComponentType & r) { this->Fill(r); }

  RGBAPixel & operator=(const ComponentType r[4]);

  Self operator+(const Self & vec) const;
  Self operator-(const Self & vec) const;
  Self operator *(const ComponentType & f) const;
  Self operator /(const ComponentType & f) const;

  const Self & operator+=(const Self & vec);
  const Self & operator-=(const Self & vec);
  const Self & operator*=(const ComponentType & f);
  const Self & operator/=(const ComponentType & f);

  bool operator<(const Self & vec) const;

  bool operator==(const Self & vec) const;

  static unsigned int GetNumberOfComponents() { return 4; }

  ComponentType GetNthComponent(int c) const { return this->operator[](c); }

  ComponentType GetScalarValue() const
  {
    return static_cast< ComponentType >( std::sqrt(
                                           static_cast< double >( this->operator[](0) )
                                           * static_cast< double >( this->operator[](0) )
                                           + static_cast< double >( this->operator[](1) )
                                           * static_cast< double >( this->operator[](1) )
                                           + static_cast< double >( this->operator[](2) )
                                           * static_cast< double >( this->operator[](2) ) ) );
  }

  void SetNthComponent(int c, const ComponentType & v) {  this->operator[](c) = v; }

  void SetRed(ComponentType red) { this->operator[](0) = red; }

  void SetGreen(ComponentType green) { this->operator[](1) = green; }

  void SetBlue(ComponentType blue) { this->operator[](2) = blue; }

  void SetAlpha(ComponentType alpha) { this->operator[](3) = alpha; }

  void Set(ComponentType red, ComponentType green, ComponentType blue, ComponentType alpha)
  {
    this->operator[](0) = red;
    this->operator[](1) = green;
    this->operator[](2) = blue;
    this->operator[](3) = alpha;
  }

  const ComponentType & GetRed() const { return this->operator[](0); }

  const ComponentType & GetGreen() const { return this->operator[](1); }

  const ComponentType & GetBlue() const { return this->operator[](2); }

  const ComponentType & GetAlpha() const { return this->operator[](3); }

  LuminanceType GetLuminance() const;
};

template< typename TComponent  >
std::ostream & operator<<(std::ostream & os,
                                     const RGBAPixel< TComponent > & c);

template< typename TComponent  >
std::istream & operator>>(std::istream & is,
                                     RGBAPixel< TComponent > & c);

template<typename T>
inline void swap( RGBAPixel<T> &a, RGBAPixel<T> &b )
{
  a.swap(b);
}

} // end namespace itk

//
// Numeric traits must be included after (optionally) including the explicit
// instantiations control of this class, in case the implicit instantiation
// needs to be disabled.
//
// NumericTraits must be included before (optionally) including the .hxx file,
// in case the .hxx requires to use NumericTraits.
//
#include "itkNumericTraitsRGBAPixel.h"

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkRGBAPixel.hxx"
#endif

#endif