#include "CUVElement.hpp"

/* Documentation at: http://www.metroid2002.com/retromodding/wiki/Particle_Script#UV_Elements */

namespace urde
{

CUVEAnimTexture::CUVEAnimTexture(TToken<CTexture>&& tex, std::unique_ptr<CIntElement>&& tileW,
                                 std::unique_ptr<CIntElement>&& tileH, std::unique_ptr<CIntElement>&& strideW,
                                 std::unique_ptr<CIntElement>&& strideH, std::unique_ptr<CIntElement>&& cycleFrames,
                                 bool loop)
: x4_tex(std::move(tex)), x24_loop(loop), x28_cycleFrames(std::move(cycleFrames))
{
    tileW->GetValue(0, x10_tileW);
    tileH->GetValue(0, x14_tileH);
    strideW->GetValue(0, x18_strideW);
    strideH->GetValue(0, x1c_strideH);

    int width = x4_tex.GetObj()->GetWidth();
    int height = x4_tex.GetObj()->GetHeight();
    float widthF = width;
    float heightF = height;
    int xTiles = std::max(1, width / x18_strideW);
    int yTiles = std::max(1, height / x1c_strideH);
    x20_tiles = xTiles * yTiles;
    x2c_uvElems.reserve(x20_tiles);
    for (int y=yTiles-1 ; y>=0 ; --y)
    {
        for (int x=0 ; x<xTiles ; ++x)
        {
            int px = x18_strideW * x;
            int px2 = px + x10_tileW;
            int py = x1c_strideH * y;
            int py2 = py + x14_tileH;
            x2c_uvElems.push_back({px / widthF, py / heightF, px2 / widthF, py2 / heightF});
        }
    }
}

void CUVEAnimTexture::GetValueUV(int frame, SUVElementSet& valOut) const
{
    int cv;
    x28_cycleFrames->GetValue(frame, cv);
    float cvf = cv / float(x20_tiles);
    cvf = frame / cvf;

    int tile = cvf;
    if (x24_loop)
    {
        if (cvf >= x20_tiles)
            tile = int(cvf) % x20_tiles;
    }
    else
    {
        if (cvf >= x20_tiles)
            tile = x20_tiles - 1;
    }

    valOut = x2c_uvElems[tile];
}

}