// This code is in the public domain -- castanyo@yahoo.es

#ifndef NV_IMAGE_COLORBLOCK_H
#define NV_IMAGE_COLORBLOCK_H

#include <nvmath/Color.h>

namespace nv
{
	class Image;

	/// Uncompressed 4x4 color block.
	struct ColorBlock
	{
		ColorBlock();
		ColorBlock(const uint * linearImage);
		ColorBlock(const ColorBlock & block);
		ColorBlock(const Image * img, uint x, uint y);

		void init(const Image * img, uint x, uint y);

		void swizzleDXT5n();
		void splatX();
		void splatY();

		bool isSingleColor() const;
		uint countUniqueColors() const;
		Color32 averageColor() const;
		bool hasAlpha() const;

		void diameterRange(Color32 * start, Color32 * end) const;
		void luminanceRange(Color32 * start, Color32 * end) const;
		void boundsRange(Color32 * start, Color32 * end) const;
		void boundsRangeAlpha(Color32 * start, Color32 * end) const;

		void sortColorsByAbsoluteValue();

		void computeRange(const Vector3 & axis, Color32 * start, Color32 * end) const;
		void sortColors(const Vector3 & axis);

		float volume() const;

		// Accessors
		const Color32 * colors() const;

		Color32 color(uint i) const;
		Color32 & color(uint i);

		Color32 color(uint x, uint y) const;
		Color32 & color(uint x, uint y);

	private:

		Color32 m_color[4*4];

	};


	/// Get pointer to block colors.
	inline const Color32 * ColorBlock::colors() const
	{
		return m_color;
	}

	/// Get block color.
	inline Color32 ColorBlock::color(uint i) const
	{
		nvDebugCheck(i < 16);
		return m_color[i];
	}

	/// Get block color.
	inline Color32 & ColorBlock::color(uint i)
	{
		nvDebugCheck(i < 16);
		return m_color[i];
	}

	/// Get block color.
	inline Color32 ColorBlock::color(uint x, uint y) const
	{
		nvDebugCheck(x < 4 && y < 4);
		return m_color[y * 4 + x];
	}

	/// Get block color.
	inline Color32 & ColorBlock::color(uint x, uint y)
	{
		nvDebugCheck(x < 4 && y < 4);
		return m_color[y * 4 + x];
	}

} // nv namespace

#endif // NV_IMAGE_COLORBLOCK_H
