/****************************************************************************
Copyright (c) 2008      Apple Inc. All Rights Reserved.
Copyright (c) 2010-2012 cocos2d-x.org
Copyright (c) 2013-2016 Chukong Technologies Inc.
Copyright (c) 2017-2018 Xiamen Yaji Software Co., Ltd.
Copyright (c) 2019-present Axmol Engine contributors (see AUTHORS.md).

https://axmolengine.github.io/

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
****************************************************************************/

/*
 * Support for RGBA_4_4_4_4 and RGBA_5_5_5_1 was copied from:
 * https://devforums.apple.com/message/37855#37855 by a1studmuffin
 */

#include "renderer/Texture2D.h"
#include "platform/Image.h"
#include "platform/GL.h"
#include "base/Utils.h"
#include "platform/Device.h"
#include "base/Config.h"
#include "base/Macros.h"
#include "base/UTF8.h"
#include "base/Configuration.h"
#include "platform/PlatformMacros.h"
#include "base/Director.h"
#include "base/NinePatchImageParser.h"
#include "renderer/backend/DriverBase.h"
#include "renderer/backend/ProgramState.h"
#include "renderer/Shaders.h"
#include "renderer/backend/PixelFormatUtils.h"
#include "renderer/Renderer.h"

#if AX_ENABLE_CACHE_TEXTURE_DATA
#    include "renderer/TextureCache.h"
#endif

NS_AX_BEGIN

// CLASS IMPLEMENTATIONS:

// If the image has alpha, you can create RGBA8 (32-bit) or RGBA4 (16-bit) or RGB5A1 (16-bit)
// Default is: RGBA8888 (32-bit textures)
static backend::PixelFormat g_defaultAlphaPixelFormat = backend::PixelFormat::RGBA8;

Texture2D::Texture2D()
    : _pixelFormat(backend::PixelFormat::NONE)
    , _pixelsWide(0)
    , _pixelsHigh(0)
    , _maxS(0.0)
    , _maxT(0.0)
    , _flags(TextureFlag::ANTIALIAS_ENABLED)
    , _samplerFlags(TextureSamplerFlag::DEFAULT)
    , _ninePatchInfo(nullptr)
    , _valid(true)
{
    backend::TextureDescriptor textureDescriptor;
    textureDescriptor.textureFormat = PixelFormat::NONE;
    _texture = static_cast<backend::Texture2DBackend*>(backend::DriverBase::getInstance()->newTexture(textureDescriptor));
}

Texture2D::~Texture2D()
{
#if AX_ENABLE_CACHE_TEXTURE_DATA
    VolatileTextureMgr::removeTexture(this);
#endif
    AXLOGINFO("deallocing Texture2D: %p - id=%u", this, _name);

    AX_SAFE_DELETE(_ninePatchInfo);

    AX_SAFE_RELEASE(_texture);
    AX_SAFE_RELEASE(_programState);
}

backend::PixelFormat Texture2D::getPixelFormat() const
{
    return _pixelFormat;
}

int Texture2D::getPixelsWide() const
{
    return _pixelsWide;
}

int Texture2D::getPixelsHigh() const
{
    return _pixelsHigh;
}

backend::TextureBackend* Texture2D::getBackendTexture() const
{
    return _texture;
}

Vec2 Texture2D::getContentSize() const
{
    Vec2 ret;
    ret.width  = _contentSize.width / AX_CONTENT_SCALE_FACTOR();
    ret.height = _contentSize.height / AX_CONTENT_SCALE_FACTOR();

    return ret;
}

const Vec2& Texture2D::getContentSizeInPixels()
{
    return _contentSize;
}

float Texture2D::getMaxS() const
{
    return _maxS;
}

void Texture2D::setMaxS(float maxS)
{
    _maxS = maxS;
}

float Texture2D::getMaxT() const
{
    return _maxT;
}

void Texture2D::setMaxT(float maxT)
{
    _maxT = maxT;
}

bool Texture2D::hasPremultipliedAlpha() const
{
    return _flags & TextureFlag::PREMULTIPLIEDALPHA;
}

void Texture2D::setPremultipliedAlpha(bool premultipliedAlpha)
{
    if (premultipliedAlpha)
        _flags |= TextureFlag::PREMULTIPLIEDALPHA;
    else
        _flags &= ~TextureFlag::PREMULTIPLIEDALPHA;
}

bool Texture2D::initWithData(const void* data,
                             ssize_t dataLen,
                             backend::PixelFormat pixelFormat,
                             backend::PixelFormat renderFormat,
                             int pixelsWide,
                             int pixelsHigh,
                             bool preMultipliedAlpha)
{
    AXASSERT(dataLen > 0 && pixelsWide > 0 && pixelsHigh > 0, "Invalid size");

    // if data has no mipmaps, we will consider it has only one mipmap
    MipmapInfo mipmap;
    mipmap.address = (unsigned char*)data;
    mipmap.len     = static_cast<int>(dataLen);
    return initWithMipmaps(&mipmap, 1, pixelFormat, renderFormat, pixelsWide, pixelsHigh, preMultipliedAlpha);
}

bool Texture2D::initWithMipmaps(MipmapInfo* mipmaps,
                                int mipmapsNum,
                                backend::PixelFormat pixelFormat,
                                backend::PixelFormat renderFormat,
                                int pixelsWide,
                                int pixelsHigh,
                                bool preMultipliedAlpha)
{
    // the pixelFormat must be a certain value
    updateWithMipmaps(mipmaps, mipmapsNum, pixelFormat, renderFormat, pixelsWide, pixelsHigh, preMultipliedAlpha);

    return true;
}

bool Texture2D::updateWithImage(Image* image, backend::PixelFormat format, int index)
{
    if (image == nullptr)
    {
        AXLOG("axmol: Texture2D. Can't create Texture. UIImage is nil");
        return false;
    }

    if (this->_filePath.empty())
        this->_filePath = image->getFilePath();

    int imageWidth  = image->getWidth();
    int imageHeight = image->getHeight();

    Configuration* conf = Configuration::getInstance();

    int maxTextureSize = conf->getMaxTextureSize();
    if (imageWidth > maxTextureSize || imageHeight > maxTextureSize)
    {
        AXLOG("axmol: WARNING: Image (%u x %u) is bigger than the supported %u x %u", imageWidth, imageHeight,
              maxTextureSize, maxTextureSize);
        return false;
    }

    unsigned char* tempData               = image->getData();
    // Vec2 imageSize                        = Vec2((float)imageWidth, (float)imageHeight);
    backend::PixelFormat renderFormat     = (PixelFormat::NONE == format) ? image->getPixelFormat() : format;
    backend::PixelFormat imagePixelFormat = image->getPixelFormat();
    size_t tempDataLen                    = image->getDataLen();

#ifdef AX_USE_METAL
    //! override renderFormat, since some render format is not supported by metal
    switch (renderFormat)
    {
#    if (AX_TARGET_PLATFORM != AX_PLATFORM_IOS || TARGET_OS_SIMULATOR)
    // packed 16 bits pixels only available on iOS
    case PixelFormat::RGB565:
    case PixelFormat::RGB5A1:
    case PixelFormat::RGBA4:
#    endif
    case PixelFormat::L8:
    case PixelFormat::LA8:
    case PixelFormat::RGB8:
        // Note: conversion to RGBA8 will happends
        renderFormat = PixelFormat::RGBA8;
        break;
    default:
        break;
    }
#elif !AX_GLES_PROFILE 
    // Non-GLES doesn't support follow render formats, needs convert PixelFormat::RGBA8
    // Note: axmol-1.1 deprecated A8, L8, LA8 as renderFormat, preferred R8, RG8
    switch (renderFormat)
    {
    case PixelFormat::A8:
    case PixelFormat::L8:
    case PixelFormat::LA8:
        // Note: conversion to RGBA8 will happends
        renderFormat = PixelFormat::RGBA8;
    }
#endif

    if (image->getNumberOfMipmaps() > 1)
    {
        if (renderFormat != image->getPixelFormat())
        {
            AXLOG("axmol: WARNING: This image has more than 1 mipmaps and we will not convert the data format");
        }

        // pixel format of data is not converted, renderFormat can be different from pixelFormat
        // it will be done later
        updateWithMipmaps(image->getMipmaps(), image->getNumberOfMipmaps(), image->getPixelFormat(), renderFormat, imageHeight, imageWidth, image->hasPremultipliedAlpha(), index);
    }
    else if (image->isCompressed())
    {  // !Only hardware support texture will be compression PixelFormat, otherwise, will convert to RGBA8 duraing image
       // load
        renderFormat = imagePixelFormat;
        updateWithData(tempData, tempDataLen, image->getPixelFormat(), image->getPixelFormat(), imageWidth, imageHeight, image->hasPremultipliedAlpha(), index);
    }
    else
    {
        // after conversion, renderFormat == pixelFormat of data
        updateWithData(tempData, tempDataLen, imagePixelFormat, renderFormat, imageWidth, imageHeight,
                       image->hasPremultipliedAlpha(), index);
    }

    return true;
}

bool Texture2D::updateWithData(const void* data,
                               ssize_t dataLen,
                               backend::PixelFormat pixelFormat,
                               backend::PixelFormat renderFormat,
                               int pixelsWide,
                               int pixelsHigh,
                               bool preMultipliedAlpha,
                               int index)
{
    AXASSERT(dataLen > 0 && pixelsWide > 0 && pixelsHigh > 0, "Invalid size");

    // if data has no mipmaps, we will consider it has only one mipmap
    MipmapInfo mipmap;
    mipmap.address = (unsigned char*)data;
    mipmap.len     = static_cast<int>(dataLen);
    return updateWithMipmaps(&mipmap, 1, pixelFormat, renderFormat, pixelsWide, pixelsHigh, preMultipliedAlpha, index);
}

bool Texture2D::updateWithMipmaps(MipmapInfo* mipmaps,
                                  int mipmapsNum,
                                  backend::PixelFormat pixelFormat,
                                  backend::PixelFormat renderFormat,
                                  int pixelsWide,
                                  int pixelsHigh,
                                  bool preMultipliedAlpha,
                                  int index)
{
    // the pixelFormat must be a certain value
    AXASSERT(pixelFormat != PixelFormat::NONE, "the \"pixelFormat\" param must be a certain value!");
    AXASSERT(pixelsWide > 0 && pixelsHigh > 0, "Invalid size");

    if (mipmapsNum <= 0)
    {
        AXLOG("axmol: WARNING: mipmap number is less than 1");
        return false;
    }

    auto& pfd = backend::PixelFormatUtils::getFormatDescriptor(pixelFormat);
    if (!pfd.bpp)
    {
        AXLOG("axmol: WARNING: unsupported pixelformat: %x", (uint32_t)pixelFormat);
#ifdef AX_USE_METAL
        AXASSERT(false, "pixeformat not found in _pixelFormatInfoTables, register required!");
#endif
        return false;
    }

    bool compressed = backend::PixelFormatUtils::isCompressed(pixelFormat);

    if (compressed && !Configuration::getInstance()->supportsPVRTC() && !Configuration::getInstance()->supportsETC1() &&
        !Configuration::getInstance()->supportsETC2() && !Configuration::getInstance()->supportsS3TC() &&
        !Configuration::getInstance()->supportsASTC() && !Configuration::getInstance()->supportsATITC())
    {
        AXLOG("axmol: WARNING: PVRTC/ETC images are not supported");
        return false;
    }

#if AX_ENABLE_CACHE_TEXTURE_DATA
    VolatileTextureMgr::findVolotileTexture(this);
#endif

    backend::TextureDescriptor textureDescriptor;
    textureDescriptor.width  = pixelsWide;
    textureDescriptor.height = pixelsHigh;

    textureDescriptor.samplerDescriptor.magFilter =
        (_flags & TextureFlag::ANTIALIAS_ENABLED) ? backend::SamplerFilter::LINEAR : backend::SamplerFilter::NEAREST;
    if (mipmapsNum == 1)
    {
        textureDescriptor.samplerDescriptor.minFilter = (_flags & TextureFlag::ANTIALIAS_ENABLED)
                                                            ? backend::SamplerFilter::LINEAR
                                                            : backend::SamplerFilter::NEAREST;
    }
    else
    {
        textureDescriptor.samplerDescriptor.minFilter = (_flags & TextureFlag::ANTIALIAS_ENABLED)
                                                            ? backend::SamplerFilter::LINEAR_MIPMAP_NEAREST
                                                            : backend::SamplerFilter::NEAREST_MIPMAP_NEAREST;
    }

    int width                           = pixelsWide;
    int height                          = pixelsHigh;
    backend::PixelFormat oriPixelFormat = pixelFormat;
    for (int i = 0; i < mipmapsNum; ++i)
    {
        unsigned char* data    = mipmaps[i].address;
        size_t dataLen         = mipmaps[i].len;
        unsigned char* outData = data;
        size_t outDataLen      = dataLen;

        if (renderFormat != oriPixelFormat && !compressed)  // need conversion
        {
            auto convertedFormat = backend::PixelFormatUtils::convertDataToFormat(data, dataLen, oriPixelFormat,
                                                                                  renderFormat, &outData, &outDataLen);
#ifdef AX_USE_METAL
            AXASSERT(convertedFormat == renderFormat, "PixelFormat convert failed!");
#endif
            if (convertedFormat == renderFormat)
                pixelFormat = renderFormat;
        }

        textureDescriptor.textureFormat = pixelFormat;
        AXASSERT(textureDescriptor.textureFormat != backend::PixelFormat::NONE, "PixelFormat should not be NONE");

        if (_texture->getTextureFormat() != textureDescriptor.textureFormat)
            _texture->updateTextureDescriptor(textureDescriptor, index);

        if (compressed)
        {
            _texture->updateCompressedData(data, width, height, dataLen, i, index);
        }
        else
        {
            _texture->updateData(outData, width, height, i, index);
        }

        if (outData && outData != data && outDataLen > 0)
        {
            free(outData);
            outData    = nullptr;
            outDataLen = 0;
        }

        if (i > 0 && (width != height || ccNextPOT(width) != width))
        {
            AXLOG(
                "axmol: Texture2D. WARNING. Mipmap level %u is not squared. Texture won't render correctly. width=%d "
                "!= height=%d",
                i, width, height);
        }

        width  = MAX(width >> 1, 1);
        height = MAX(height >> 1, 1);
    }

    if (index == 0)
    {
        _contentSize = Vec2((float)pixelsWide, (float)pixelsHigh);
        _pixelsWide  = pixelsWide;
        _pixelsHigh  = pixelsHigh;
        _pixelFormat = pixelFormat;
        _maxS        = 1;
        _maxT        = 1;

        setPremultipliedAlpha(preMultipliedAlpha);
    }

    // pitfall: we do merge RGB+A at at dual sampler shader, so must mark as _hasPremultipliedAlpha = true to makesure
    // alpha blend works well.
    if (index > 0)
    {
        setPremultipliedAlpha(Image::isCompressedImageHavePMA(Image::CompressedImagePMAFlag::DUAL_SAMPLER));
        _samplerFlags |= TextureSamplerFlag::DUAL_SAMPLER;
    }

    return true;
}

bool Texture2D::updateWithSubData(void* data, int offsetX, int offsetY, int width, int height, int index)
{
    if (_texture && width > 0 && height > 0)
    {
        uint8_t* textureData = static_cast<uint8_t*>(data);
        _texture->updateSubData(offsetX, offsetY, width, height, 0, textureData, index);
        return true;
    }
    return false;
}

// implementation Texture2D (Image)
bool Texture2D::initWithImage(Image* image)
{
    return initWithImage(image, g_defaultAlphaPixelFormat);
}

bool Texture2D::initWithImage(Image* image, backend::PixelFormat format)
{
    if (image == nullptr)
    {
        AXLOG("axmol: Texture2D. Can't create Texture. UIImage is nil");
        return false;
    }

    this->_filePath = image->getFilePath();

    return updateWithImage(image, format);
}

// implementation Texture2D (Text)
bool Texture2D::initWithString(std::string_view text,
                               std::string_view fontName,
                               float fontSize,
                               const Vec2& dimensions /* = Vec2(0, 0)*/,
                               TextHAlignment hAlignment /* =  TextHAlignment::CENTER */,
                               TextVAlignment vAlignment /* =  TextVAlignment::TOP */,
                               bool enableWrap /* = false */,
                               int overflow /* = 0 */)
{
    FontDefinition tempDef;

    tempDef._shadow._shadowEnabled = false;
    tempDef._stroke._strokeEnabled = false;

    tempDef._fontName      = fontName;
    tempDef._fontSize      = fontSize;
    tempDef._dimensions    = dimensions;
    tempDef._alignment     = hAlignment;
    tempDef._vertAlignment = vAlignment;
    tempDef._fontFillColor = Color3B::WHITE;
    tempDef._enableWrap    = enableWrap;
    tempDef._overflow      = overflow;

    return initWithString(text, tempDef);
}

bool Texture2D::initWithString(std::string_view text, const FontDefinition& textDefinition)
{
    if (text.empty())
        return false;

#if AX_ENABLE_CACHE_TEXTURE_DATA
    // cache the texture data
    VolatileTextureMgr::addStringTexture(this, text, textDefinition);
#endif

    bool ret = false;
    Device::TextAlign align;

    if (TextVAlignment::TOP == textDefinition._vertAlignment)
    {
        align = (TextHAlignment::CENTER == textDefinition._alignment) ? Device::TextAlign::TOP
                : (TextHAlignment::LEFT == textDefinition._alignment) ? Device::TextAlign::TOP_LEFT
                                                                      : Device::TextAlign::TOP_RIGHT;
    }
    else if (TextVAlignment::CENTER == textDefinition._vertAlignment)
    {
        align = (TextHAlignment::CENTER == textDefinition._alignment) ? Device::TextAlign::CENTER
                : (TextHAlignment::LEFT == textDefinition._alignment) ? Device::TextAlign::LEFT
                                                                      : Device::TextAlign::RIGHT;
    }
    else if (TextVAlignment::BOTTOM == textDefinition._vertAlignment)
    {
        align = (TextHAlignment::CENTER == textDefinition._alignment) ? Device::TextAlign::BOTTOM
                : (TextHAlignment::LEFT == textDefinition._alignment) ? Device::TextAlign::BOTTOM_LEFT
                                                                      : Device::TextAlign::BOTTOM_RIGHT;
    }
    else
    {
        AXASSERT(false, "Not supported alignment format!");
        return false;
    }

#if (AX_TARGET_PLATFORM != AX_PLATFORM_ANDROID) && (AX_TARGET_PLATFORM != AX_PLATFORM_IOS)
    AXASSERT(textDefinition._stroke._strokeEnabled == false, "Currently stroke only supported on iOS and Android!");
#endif

    PixelFormat pixelFormat    = g_defaultAlphaPixelFormat;
    unsigned char* outTempData = nullptr;
    size_t outTempDataLen      = 0;

    int imageWidth;
    int imageHeight;
    auto textDef            = textDefinition;
    auto contentScaleFactor = AX_CONTENT_SCALE_FACTOR();
    textDef._fontSize *= contentScaleFactor;
    textDef._dimensions.width *= contentScaleFactor;
    textDef._dimensions.height *= contentScaleFactor;
    textDef._stroke._strokeSize *= contentScaleFactor;
    textDef._shadow._shadowEnabled = false;

    bool hasPremultipliedAlpha;
    Data outData = Device::getTextureDataForText(text, textDef, align, imageWidth, imageHeight, hasPremultipliedAlpha);
    if (outData.isNull())
    {
        return false;
    }

    Vec2 imageSize = Vec2((float)imageWidth, (float)imageHeight);
    pixelFormat =
        backend::PixelFormatUtils::convertDataToFormat(outData.getBytes(), imageWidth * imageHeight * 4,
                                                       PixelFormat::RGBA8, pixelFormat, &outTempData, &outTempDataLen);

    ret = initWithData(outTempData, outTempDataLen, pixelFormat, imageWidth, imageHeight);

    if (outTempData != nullptr && outTempData != outData.getBytes())
    {
        free(outTempData);
    }
    setPremultipliedAlpha(hasPremultipliedAlpha);

    return ret;
}

bool Texture2D::updateTextureDescriptor(const backend::TextureDescriptor& descriptor, bool preMultipliedAlpha)
{
    AX_ASSERT(_texture);

    _texture->updateTextureDescriptor(descriptor);
    _pixelsWide = _contentSize.width = _texture->getWidth();
    _pixelsHigh = _contentSize.height = _texture->getHeight();
    setPremultipliedAlpha(preMultipliedAlpha);

    setRenderTarget(descriptor.textureUsage == TextureUsage::RENDER_TARGET);

    if (_pixelFormat == PixelFormat::NONE)
        _pixelFormat = descriptor.textureFormat;

    return true;
}

void Texture2D::setRenderTarget(bool renderTarget)
{
    if (renderTarget)
        _flags |= TextureFlag::RENDERTARGET;
    else
        _flags &= TextureFlag::RENDERTARGET;
}

bool Texture2D::hasMipmaps() const
{
    return _texture->hasMipmaps();
}

void Texture2D::setAliasTexParameters()
{

    if ((_flags & TextureFlag::ANTIALIAS_ENABLED) == 0)
    {
        return;
    }

    _flags &= ~TextureFlag::ANTIALIAS_ENABLED;

    backend::SamplerDescriptor descriptor(backend::SamplerFilter::NEAREST,  // magFilter
                                          (_texture->hasMipmaps()) ? backend::SamplerFilter::NEAREST_MIPMAP_NEAREST
                                                                   : backend::SamplerFilter::NEAREST,  // minFilter
                                          backend::SamplerAddressMode::DONT_CARE,                      // sAddressMode
                                          backend::SamplerAddressMode::DONT_CARE                       // tAddressMode
    );
    _texture->updateSamplerDescriptor(descriptor);
}

void Texture2D::setAntiAliasTexParameters()
{

    if (_flags & TextureFlag::ANTIALIAS_ENABLED)
    {
        return;
    }
    _flags |= TextureFlag::ANTIALIAS_ENABLED;

    backend::SamplerDescriptor descriptor(backend::SamplerFilter::LINEAR,  // magFilter
                                          (_texture->hasMipmaps()) ? backend::SamplerFilter::LINEAR_MIPMAP_NEAREST
                                                                   : backend::SamplerFilter::LINEAR,  // minFilter
                                          backend::SamplerAddressMode::DONT_CARE,                     // sAddressMode
                                          backend::SamplerAddressMode::DONT_CARE                      // tAddressMode
    );
    _texture->updateSamplerDescriptor(descriptor);
}

const char* Texture2D::getStringForFormat() const
{
    return backend::PixelFormatUtils::getFormatDescriptor(_pixelFormat).name;
}

//
// Texture options for images that contains alpha
//
// implementation Texture2D (PixelFormat)

void Texture2D::setDefaultAlphaPixelFormat(backend::PixelFormat format)
{
    g_defaultAlphaPixelFormat = format;
}

backend::PixelFormat Texture2D::getDefaultAlphaPixelFormat()
{
    return g_defaultAlphaPixelFormat;
}

unsigned int Texture2D::getBitsPerPixelForFormat(backend::PixelFormat format) const
{
    return backend::PixelFormatUtils::getFormatDescriptor(format).bpp;
}

unsigned int Texture2D::getBitsPerPixelForFormat() const
{
    return getBitsPerPixelForFormat(_pixelFormat);
}

void Texture2D::addSpriteFrameCapInset(SpriteFrame* spritframe, const Rect& capInsets)
{
    if (nullptr == _ninePatchInfo)
    {
        _ninePatchInfo = new NinePatchInfo;
    }
    if (nullptr == spritframe)
    {
        _ninePatchInfo->capInsetSize = capInsets;
    }
    else
    {
        _ninePatchInfo->capInsetMap[spritframe] = capInsets;
    }
}

bool Texture2D::isContain9PatchInfo() const
{
    return nullptr != _ninePatchInfo;
}

const Rect& Texture2D::getSpriteFrameCapInset(ax::SpriteFrame* spriteFrame) const
{
    AXASSERT(_ninePatchInfo != nullptr,
             "Can't get the sprite frame capInset when the texture contains no 9-patch info.");
    if (nullptr == spriteFrame)
    {
        return this->_ninePatchInfo->capInsetSize;
    }
    else
    {
        auto& capInsetMap = this->_ninePatchInfo->capInsetMap;
        if (capInsetMap.find(spriteFrame) != capInsetMap.end())
        {
            return capInsetMap.at(spriteFrame);
        }
        else
        {
            return this->_ninePatchInfo->capInsetSize;
        }
    }
}

void Texture2D::removeSpriteFrameCapInset(SpriteFrame* spriteFrame)
{
    if (nullptr != this->_ninePatchInfo)
    {
        auto capInsetMap = this->_ninePatchInfo->capInsetMap;
        if (capInsetMap.find(spriteFrame) != capInsetMap.end())
        {
            capInsetMap.erase(spriteFrame);
        }
    }
}

void Texture2D::setTexParameters(const Texture2D::TexParams& desc)
{
    _texture->updateSamplerDescriptor(desc);
}

void Texture2D::generateMipmap()
{
    AXASSERT(_pixelsWide == ccNextPOT(_pixelsWide) && _pixelsHigh == ccNextPOT(_pixelsHigh),
             "Mipmap texture only works in POT textures");

    _texture->generateMipmaps();
}

void Texture2D::initProgram()
{
    if (_programState != nullptr)
        return;

    auto& pipelineDescriptor = _customCommand.getPipelineDescriptor();
    // create program state
    auto* program      = backend::Program::getBuiltinProgram(backend::ProgramType::POSITION_TEXTURE);
    _programState      = new ax::backend::ProgramState(program);
    _mvpMatrixLocation = _programState->getUniformLocation("u_MVPMatrix");
    _textureLocation   = _programState->getUniformLocation("u_tex0");

    pipelineDescriptor.programState = _programState;

    // create vertex buffer
    _customCommand.setDrawType(CustomCommand::DrawType::ARRAY);
    _customCommand.setPrimitiveType(CustomCommand::PrimitiveType::TRIANGLE_STRIP);
    _customCommand.createVertexBuffer(4 * sizeof(float), 4, CustomCommand::BufferUsage::DYNAMIC);

    // setup blend state
    BlendFunc blendFunc;
    if (hasPremultipliedAlpha())
    {
        blendFunc = BlendFunc::ALPHA_PREMULTIPLIED;
    }
    else
    {
        blendFunc = BlendFunc::ALPHA_NON_PREMULTIPLIED;
    }

    auto& blendDescriptor                = pipelineDescriptor.blendDescriptor;
    blendDescriptor.blendEnabled         = true;
    blendDescriptor.sourceRGBBlendFactor = blendDescriptor.sourceAlphaBlendFactor = blendFunc.src;
    blendDescriptor.destinationRGBBlendFactor = blendDescriptor.destinationAlphaBlendFactor = blendFunc.dst;

    _programState->setTexture(_textureLocation, 0, _texture);
}

void Texture2D::drawAtPoint(const Vec2& point, float globalZOrder)
{
    float width  = (float)_pixelsWide * _maxS;
    float height = (float)_pixelsHigh * _maxT;
    Rect rect    = {point.x, point.y, width, height};
    drawInRect(rect, globalZOrder);
}

void Texture2D::drawInRect(const Rect& rect, float globalZOrder)
{
    initProgram();
    _customCommand.init(globalZOrder);
    auto director          = Director::getInstance();
    const auto& modelView  = director->getMatrix(MATRIX_STACK_TYPE::MATRIX_STACK_MODELVIEW);
    const auto& projection = director->getMatrix(MATRIX_STACK_TYPE::MATRIX_STACK_PROJECTION);

    Mat4 matrixMVP = projection * modelView;

    float vertexData[] = {rect.origin.x,
                          rect.origin.y,
                          0.0f,
                          _maxT,
                          rect.size.width + rect.origin.x,
                          rect.origin.y,
                          _maxS,
                          _maxT,
                          rect.origin.x,
                          rect.size.height + rect.origin.y,
                          0.0f,
                          0.0f,
                          rect.size.width + rect.origin.x,
                          rect.size.height + rect.origin.y,
                          _maxS,
                          0.0f};

    _programState->setUniform(_mvpMatrixLocation, matrixMVP.m, sizeof(matrixMVP.m));
    _customCommand.updateVertexBuffer(vertexData, sizeof(vertexData));
    Director::getInstance()->getRenderer()->addCommand(&_customCommand);
}

NS_AX_END