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

#pragma once
#ifndef NV_IMAGE_COLORBLOCK_H
#define NV_IMAGE_COLORBLOCK_H

#include "nvmath/Color.h"

namespace nv
{
    class Image;
    class FloatImage;

    /// 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 init(uint w, uint h, const uint * data, uint x, uint y);
        void init(uint w, uint h, const float * data, uint x, uint y);

        void swizzle(uint x, uint y, uint z, uint w); // 0=r, 1=g, 2=b, 3=a, 4=0xFF, 5=0

        bool isSingleColor() const;
        //bool isSingleColorNoAlpha() 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];
    }


    struct FloatColorBlock
    {
        FloatColorBlock() : w(4), h(4) {}
        FloatColorBlock(uint w, uint h) : w(w), h(h) {}

        void init(const Image * img, uint x, uint y);
        void init(const FloatImage * img, uint x, uint y);
        void init(const uint * data, uint w, uint h, uint x, uint y);
        void init(const float * data, uint w, uint h, uint x, uint y);

        Vector4 color(uint x, uint y) const { 	nvDebugCheck(x < w && y < h); return colors[y * 4 + x]; }
        Vector4 & color(uint x, uint y) { nvDebugCheck(x < w && y < h); return colors[y * 4 + x]; }


        Vector4 colors[16];
        uint w, h;
    };

} // nv namespace

#endif // NV_IMAGE_COLORBLOCK_H