#include "ImageFormat.h"

ImageFormat::ImageFormat(const ImageFormatId id, const oat::D3DFORMAT d3dFormat, const oat::DXGI_FORMAT dxgiFormat)
    : m_id(id),
      m_d3d_format(d3dFormat),
      m_dxgi_format(dxgiFormat)
{
}

ImageFormatId ImageFormat::GetId() const
{
    return m_id;
}

oat::D3DFORMAT ImageFormat::GetD3DFormat() const
{
    return m_d3d_format;
}

oat::DXGI_FORMAT ImageFormat::GetDxgiFormat() const
{
    return m_dxgi_format;
}

ImageFormatUnsigned::ImageFormatUnsigned(const ImageFormatId id,
                                         const oat::D3DFORMAT d3dFormat,
                                         const oat::DXGI_FORMAT dxgiFormat,
                                         const unsigned bitsPerPixel,
                                         const unsigned rOffset,
                                         const unsigned rSize,
                                         const unsigned gOffset,
                                         const unsigned gSize,
                                         const unsigned bOffset,
                                         const unsigned bSize,
                                         const unsigned aOffset,
                                         const unsigned aSize)
    : ImageFormat(id, d3dFormat, dxgiFormat),
      m_bits_per_pixel(bitsPerPixel),
      m_r_offset(rOffset),
      m_r_size(rSize),
      m_g_offset(gOffset),
      m_g_size(gSize),
      m_b_offset(bOffset),
      m_b_size(bSize),
      m_a_offset(aOffset),
      m_a_size(aSize)
{
}

ImageFormatType ImageFormatUnsigned::GetType() const
{
    return ImageFormatType::UNSIGNED;
}

size_t ImageFormatUnsigned::GetPitch(const unsigned mipLevel, const unsigned width) const
{
    unsigned mipWidth = width >> mipLevel;
    if (mipWidth == 0)
        mipWidth = 1;

    return mipWidth * m_bits_per_pixel / 8;
}

size_t ImageFormatUnsigned::GetSizeOfMipLevel(const unsigned mipLevel, const unsigned width, const unsigned height, const unsigned depth) const
{
    unsigned mipWidth = width >> mipLevel;
    unsigned mipHeight = height >> mipLevel;
    unsigned mipDepth = depth >> mipLevel;

    if (mipWidth == 0)
        mipWidth = 1;
    if (mipHeight == 0)
        mipHeight = 1;
    if (mipDepth == 0)
        mipDepth = 1;

    return mipWidth * mipHeight * mipDepth * m_bits_per_pixel / 8;
}

ImageFormatBlockCompressed::ImageFormatBlockCompressed(
    const ImageFormatId id, const oat::D3DFORMAT d3dFormat, const oat::DXGI_FORMAT dxgiFormat, const unsigned blockSize, const unsigned bitsPerBlock)
    : ImageFormat(id, d3dFormat, dxgiFormat),
      m_block_size(blockSize),
      m_bits_per_block(bitsPerBlock)
{
}

ImageFormatType ImageFormatBlockCompressed::GetType() const
{
    return ImageFormatType::BLOCK_COMPRESSED;
}

size_t ImageFormatBlockCompressed::GetPitch(const unsigned mipLevel, const unsigned width) const
{
    unsigned mipWidth = width >> mipLevel;

    if (mipWidth == 0)
        mipWidth = 1;

    const unsigned blockCount = (mipWidth + m_block_size - 1) / m_block_size;

    return blockCount * m_bits_per_block / 8;
}

size_t ImageFormatBlockCompressed::GetSizeOfMipLevel(const unsigned mipLevel, const unsigned width, const unsigned height, const unsigned depth) const
{
    unsigned mipWidth = width >> mipLevel;
    unsigned mipHeight = height >> mipLevel;
    unsigned mipDepth = depth >> mipLevel;

    if (mipWidth == 0)
        mipWidth = 1;
    if (mipHeight == 0)
        mipHeight = 1;
    if (mipDepth == 0)
        mipDepth = 1;

    const unsigned blockCount = ((mipWidth + m_block_size - 1) / m_block_size) * ((mipHeight + m_block_size - 1) / m_block_size) * mipDepth;

    return blockCount * m_bits_per_block / 8;
}

bool ImageFormatUnsigned::HasR() const
{
    return m_r_size > 0;
}

bool ImageFormatUnsigned::HasG() const
{
    return m_g_size > 0;
}

bool ImageFormatUnsigned::HasB() const
{
    return m_b_size > 0;
}

bool ImageFormatUnsigned::HasA() const
{
    return m_a_size > 0;
}

const ImageFormatUnsigned ImageFormat::FORMAT_R8_G8_B8(ImageFormatId::R8_G8_B8, oat::D3DFMT_R8G8B8, oat::DXGI_FORMAT_UNKNOWN, 24, 0, 8, 8, 8, 16, 8, 0, 0);
const ImageFormatUnsigned
    ImageFormat::FORMAT_B8_G8_R8_X8(ImageFormatId::B8_G8_R8_X8, oat::D3DFMT_X8R8G8B8, oat::DXGI_FORMAT_B8G8R8X8_UNORM, 32, 16, 8, 8, 8, 0, 8, 0, 0);
const ImageFormatUnsigned
    ImageFormat::FORMAT_R8_G8_B8_A8(ImageFormatId::R8_G8_B8_A8, oat::D3DFMT_A8B8G8R8, oat::DXGI_FORMAT_R8G8B8A8_UNORM, 32, 0, 8, 8, 8, 16, 8, 24, 8);
const ImageFormatUnsigned
    ImageFormat::FORMAT_B8_G8_R8_A8(ImageFormatId::B8_G8_R8_A8, oat::D3DFMT_A8R8G8B8, oat::DXGI_FORMAT_B8G8R8A8_UNORM, 32, 16, 8, 8, 8, 0, 8, 24, 8);
const ImageFormatUnsigned ImageFormat::FORMAT_A8(ImageFormatId::A8, oat::D3DFMT_A8, oat::DXGI_FORMAT_A8_UNORM, 8, 0, 0, 0, 0, 0, 0, 0, 8);
const ImageFormatUnsigned ImageFormat::FORMAT_R16_G16_B16_A16_FLOAT(
    ImageFormatId::R16_G16_B16_A16_FLOAT, oat::D3DFMT_A16B16G16R16F, oat::DXGI_FORMAT_R16G16B16A16_FLOAT, 128, 0, 0, 0, 0, 0, 0, 0, 8);
const ImageFormatUnsigned ImageFormat::FORMAT_R8(ImageFormatId::R8, oat::D3DFMT_L8, oat::DXGI_FORMAT_R8_UNORM, 8, 0, 8, 0, 0, 0, 0, 0, 0);
const ImageFormatUnsigned ImageFormat::FORMAT_R8_A8(ImageFormatId::R8_A8, oat::D3DFMT_A8L8, oat::DXGI_FORMAT_UNKNOWN, 16, 0, 8, 0, 0, 0, 0, 8, 8);
const ImageFormatBlockCompressed ImageFormat::FORMAT_BC1(ImageFormatId::BC1, oat::D3DFMT_DXT1, oat::DXGI_FORMAT_BC1_UNORM, 4, 64);
const ImageFormatBlockCompressed ImageFormat::FORMAT_BC2(ImageFormatId::BC2, oat::D3DFMT_DXT3, oat::DXGI_FORMAT_BC2_UNORM, 4, 128);
const ImageFormatBlockCompressed ImageFormat::FORMAT_BC3(ImageFormatId::BC3, oat::D3DFMT_DXT5, oat::DXGI_FORMAT_BC3_UNORM, 4, 128);
const ImageFormatBlockCompressed ImageFormat::FORMAT_BC4(ImageFormatId::BC4, oat::D3DFMT_UNKNOWN, oat::DXGI_FORMAT_BC4_UNORM, 4, 64);
const ImageFormatBlockCompressed ImageFormat::FORMAT_BC5(ImageFormatId::BC5, oat::D3DFMT_UNKNOWN, oat::DXGI_FORMAT_BC5_UNORM, 4, 128);

const ImageFormat* const ImageFormat::ALL_FORMATS[static_cast<unsigned>(ImageFormatId::MAX)]{
    &FORMAT_R8_G8_B8,
    &FORMAT_B8_G8_R8_X8,
    &FORMAT_R8_G8_B8_A8,
    &FORMAT_B8_G8_R8_A8,
    &FORMAT_A8,
    &FORMAT_R16_G16_B16_A16_FLOAT,
    &FORMAT_R8,
    &FORMAT_R8_A8,
    &FORMAT_BC1,
    &FORMAT_BC2,
    &FORMAT_BC3,
    &FORMAT_BC4,
    &FORMAT_BC5,
};