/****************************************************************************
 * Copyright (C) 2015 Dimok
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ****************************************************************************/
#include <malloc.h>
#include <string.h>
#include "Texture2DShader.h"

static const u32 cpVertexShaderProgram[] =
{
    0x00000000,0x00008009,0x20000000,0x000080a0,
    0x3c200100,0x88060094,0x00400000,0x88042014,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x02290001,0x80000000,0x02041f00,0x900c0020,
    0x00a11f00,0xfc00624f,0xfd041f00,0x900c4060,
    0x02001f80,0x900c0000,0x83f9223e,0x0000803f,
    0xfe081f00,0x00080020,0xfe202081,0x10004040,
    0xfea49f80,0xfd00620f,0xdb0f49c0,0xdb0fc940,
    0xfea01f80,0x9000e06f,0x83f9223e,0x00000000,
    0xfe0c1f80,0x00370000,0xffa01f00,0x80000040,
    0xff101f00,0x800c0060,0x7f0c1f80,0x80370040,
    0x0000103f,0x00000000,0xffa01f00,0x80000000,
    0xff001f00,0x800c0020,0x02c51f01,0x80000040,
    0xfeac9f80,0x80000060,0x0000103f,0x398ee33f,
    0xfea01f00,0x80000000,0x02c19f01,0x9000e02f,
    0x01c41f01,0x9000e04f,0x02c59f80,0x80000060,
    0x398ee33f,0x00000000,0x01c49f01,0x80000020,
    0x02c11f80,0x80000040,0x01e08f00,0xfe04624f,
    0x01c01f81,0x7f08626f,0xfe2c2000,0x10004000,
    0xfe28a080,0x10004020,0xeb825790,0xb6f711be,
    0x7c0e2df2,0x81173cfa
};

static const u32 cpVertexShaderRegs[] = {
    0x00000103,0x00000000,0x00000000,0x00000001,
    0xffffff00,0xffffffff,0xffffffff,0xffffffff,
    0xffffffff,0xffffffff,0xffffffff,0xffffffff,
    0xffffffff,0xffffffff,0x00000000,0xfffffffc,
    0x00000002,0x00000000,0x00000001,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x000000ff,0x000000ff,0x000000ff,
    0x000000ff,0x00000000,0x0000000e,0x00000010
};

static const u32 cPixelShaderProgram[] =
{
    0x20000000,0x00000ca4,0x0b000000,0x00000085,
    0x24000000,0x000050a0,0xb0000000,0x000cc080,
    0x39000000,0x00005ca0,0xb8000000,0x000cc080,
    0x51000000,0x000078a0,0xc0000000,0x000cc080,
    0x70000000,0x000064a0,0xc8000000,0x0008c080,
    0x8a000000,0x00005ca0,0x0e000000,0x01008086,
    0xce000000,0x0000c080,0xa2000000,0x00000ca8,
    0x00800000,0x88062094,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00051f00,0x80060000,0x00011f80,0x80060040,
    0xfec81f80,0xfb802320,0x01041f80,0x8c220000,
    0x00a41f00,0xfc10620f,0x010d1f00,0x900c0021,
    0x00091f00,0x80060040,0x00a01f80,0xfc10626f,
    0x00000040,0x00000000,0xfe080000,0xfe8cc300,
    0xfe088080,0xfe80c320,0x00a11f00,0xfe000200,
    0x00a51f00,0xfe040220,0x00a19f00,0xfe000240,
    0x00a59f00,0xfe040260,0x00a11f81,0xfe002600,
    0x4260e5bc,0xa69bc4bc,0x0ad7a3bc,0x00000000,
    0x00a11f00,0x06004200,0x00a59f00,0x06042220,
    0x00a51f00,0x06044240,0x00a11f01,0x06008260,
    0x00a51f81,0x06048620,0x6f1283bc,0x0ad7a3bc,
    0xa69b44bc,0x00000000,0x00a41f00,0x80000000,
    0x00a01f00,0x80000020,0x00ac1f00,0x80000040,
    0x00a81f00,0x80000060,0x00a19f80,0x06000600,
    0xcac3123c,0x6f1203bc,0x03a41f00,0xfe00620f,
    0x03a01f00,0xfe04622f,0x03ac1f00,0xfe08624f,
    0x03a81f00,0xfe0c626f,0x00a59f80,0x06040620,
    0xcc28913b,0x6f1203bc,0x01a41f00,0xfe00620f,
    0x01a01f00,0xfe04622f,0x01ac1f00,0xfe08624f,
    0x01a81f00,0xfe0c626f,0x00a19f80,0x06002600,
    0xe8eab03c,0x6f1283bb,0x02ac1f00,0xfe084200,
    0x02a81f00,0xfe0c4220,0x02a41f00,0xfe004240,
    0x02a01f00,0xfe044260,0x00a59f80,0x06042620,
    0x92bb353d,0x6f1283bb,0x04a81f00,0x0204620f,
    0x04ac1f00,0x0200662f,0x04a41f00,0x0208624f,
    0x04a01f00,0x020c626f,0x00a19f80,0x06004600,
    0xc4139f3d,0x6f12833b,0x00a41f00,0xfe08620f,
    0x00a01f00,0xfe0c622f,0x00ac1f00,0xfe04624f,
    0x00a81f00,0xfe00626f,0x00a59f80,0x06044620,
    0xb950ed3d,0x6f12833b,0x01a41f00,0xfe00620f,
    0x01a01f00,0xfe04622f,0x01ac1f00,0xfe08624f,
    0x01a81f00,0xfe0c626f,0x00a19f80,0x06002600,
    0xecd7163e,0x6f12033c,0x03a41f00,0xfe000200,
    0x03a01f00,0xfe040220,0x03ac1f00,0xfe082240,
    0x03a81f00,0xfe0c2260,0x00a59f80,0x06042620,
    0x2168233e,0x6f12033c,0x00a11f00,0x06006200,
    0x00a51f00,0x06046220,0x00a19f00,0x06006240,
    0x00a59f00,0x06046260,0x00a11f81,0x0600e600,
    0xa69b443c,0x6f12833c,0x0ad7a33c,0x00000000,
    0x02ac1f00,0x0108620f,0x02a81f00,0x010c622f,
    0x02a41f00,0x0000624f,0x02a01f00,0x0004666f,
    0x00a59f80,0x0604e620,0xecd7163e,0x0ad7a33c,
    0x04a81f00,0xfe04620f,0x04ac1f00,0xfe00622f,
    0x04a41f00,0xfe08624f,0x04a01f00,0xfe0c626f,
    0x00a19f80,0x06008600,0xb950ed3d,0xa69bc43c,
    0x05a41f00,0xfe08620f,0x05a01f00,0xfe0c622f,
    0x05ac1f00,0xfe04624f,0x05a81f00,0xfe00626f,
    0x00a59f80,0x06048620,0xc4139f3d,0xa69bc43c,
    0x03a41f00,0xfe00a200,0x03a01f00,0xfe04a220,
    0x03ac1f00,0xfe086240,0x03a81f00,0xfe0c6260,
    0x00a19f80,0x06006600,0x92bb353d,0x4260e53c,
    0x00a51f80,0x06046220,0x4260e53c,0x00000000,
    0x07ac1f00,0x0308620f,0x07a81f00,0x030c622f,
    0x07a41f00,0x0500624f,0x07a01f80,0x0504626f,
    0xe8eab03c,0x00000000,0x04a81f00,0xfe04620f,
    0x04ac1f00,0xfe00622f,0x04a41f00,0xfe08624f,
    0x04a01f80,0xfe0c626f,0xcac3123c,0x00000000,
    0x06a41f00,0xfe08620f,0x06a01f00,0xfe0c622f,
    0x06ac1f00,0xfe04624f,0x06a81f80,0xfe00626f,
    0xcc28913b,0x00000000,0xfe20a000,0x9000e00f,
    0xfe242000,0x9000e02f,0xfe28a001,0x9000e04f,
    0xfe2c2081,0x9000e06f,0xfe28a081,0x80060020,
    0xfee48f00,0x7f842300,0xfee40f00,0x7f802320,
    0xfee48f01,0x7f8c2340,0xfee40f81,0x08842b60,
    0x00202000,0x90002000,0x0024a000,0x90002020,
    0x00282001,0x90002040,0x002ca081,0x90002060,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x10000000,0x03100df0,0x00008010,0xecdfea0d,
    0x10000000,0x00100df0,0x0000a051,0xecdfea0d,
    0x10000100,0x01100df0,0x00008010,0xecdfea0d,
    0x10000200,0x02100df0,0x00000011,0xecdfea0d,
    0x10000400,0x04100df0,0x0000b070,0xecdfea0d,
    0x10000000,0x00100df0,0x00008010,0xecdfea0d,
    0x10000100,0x01100df0,0x00008010,0xecdfea0d,
    0x10000600,0x03100df0,0x00008010,0xecdfea0d,
    0x10000200,0x02100df0,0x00008010,0xecdfea0d,
    0x10000100,0x04100df0,0x00008010,0xecdfea0d,
    0x10000300,0x05100df0,0x00008010,0xecdfea0d,
    0x10000300,0x03100df0,0x0000a051,0xecdfea0d,
    0x10000700,0x07100df0,0x00008010,0xecdfea0d,
    0x10000400,0x04100df0,0x00008010,0xecdfea0d,
    0x10000300,0x06100df0,0x00008010,0xecdfea0d,
    0x10000000,0x00100df0,0x00008010,0xecdfea0d,
    0xc8581837,0x22740275,0x281eddcc,0xfa8b9b65
};
static const u32 cPixelShaderRegs[] = {
    0x00000109,0x00000002,0x14000001,0x00000000,
    0x00000001,0x00000100,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x00000000,0x00000000,0x00000000,
    0x00000000,0x0000000f,0x00000001,0x00000010,
    0x00000000
};

Texture2DShader * Texture2DShader::shaderInstance = NULL;

Texture2DShader::Texture2DShader()
    : vertexShader(cuAttributeCount)
{
    //! create pixel shader
    pixelShader.setProgram(cPixelShaderProgram, sizeof(cPixelShaderProgram), cPixelShaderRegs, sizeof(cPixelShaderRegs));

    blurLocation = 0;
    colorIntensityLocation = 4;
    pixelShader.addUniformVar((GX2UniformVar){ "unf_blur_texture_direction", GX2_VAR_TYPE_VEC3, 1, blurLocation, 0xffffffff });
    pixelShader.addUniformVar((GX2UniformVar){ "unf_color_intensity", GX2_VAR_TYPE_VEC4, 1, colorIntensityLocation, 0xffffffff });

    samplerLocation = 0;
    pixelShader.addSamplerVar((GX2SamplerVar){ "sampl_texture", GX2_SAMPLER_TYPE_2D, samplerLocation });

    //! create vertex shader
    vertexShader.setProgram(cpVertexShaderProgram, sizeof(cpVertexShaderProgram), cpVertexShaderRegs, sizeof(cpVertexShaderRegs));

    angleLocation = 0;
    offsetLocation = 4;
    scaleLocation = 8;
    vertexShader.addUniformVar((GX2UniformVar){ "unf_angle", GX2_VAR_TYPE_FLOAT, 1, angleLocation, 0xffffffff });
    vertexShader.addUniformVar((GX2UniformVar){ "unf_offset", GX2_VAR_TYPE_VEC3, 1, offsetLocation, 0xffffffff });
    vertexShader.addUniformVar((GX2UniformVar){ "unf_scale", GX2_VAR_TYPE_VEC3, 1, scaleLocation, 0xffffffff });

    positionLocation = 0;
    texCoordLocation = 1;
    vertexShader.addAttribVar((GX2AttribVar){ "attr_position", GX2_VAR_TYPE_VEC3, 0, positionLocation });
    vertexShader.addAttribVar((GX2AttribVar){ "attr_texture_coord", GX2_VAR_TYPE_VEC2, 0, texCoordLocation });

    //! setup attribute streams
    GX2InitAttribStream(vertexShader.getAttributeBuffer(0), positionLocation, 0, 0, GX2_ATTRIB_FORMAT_32_32_32_FLOAT);
    GX2InitAttribStream(vertexShader.getAttributeBuffer(1), texCoordLocation, 1, 0, GX2_ATTRIB_FORMAT_32_32_FLOAT);

    //! create fetch shader
    fetchShader = new FetchShader(vertexShader.getAttributeBuffer(), vertexShader.getAttributesCount());

    //! model vertex has to be align and cannot be in unknown regions for GX2 like 0xBCAE1000
    posVtxs = (f32*)memalign(GX2_VERTEX_BUFFER_ALIGNMENT, ciPositionVtxsSize);
    texCoords = (f32*)memalign(GX2_VERTEX_BUFFER_ALIGNMENT, ciTexCoordsVtxsSize);

    //! defaults for normal square
    //! position vertex structure and texture coordinate vertex structure
    s32 i = 0;
    posVtxs[i++] = -1.0f; posVtxs[i++] = -1.0f; posVtxs[i++] = 0.0f;
    posVtxs[i++] =  1.0f; posVtxs[i++] = -1.0f; posVtxs[i++] = 0.0f;
    posVtxs[i++] =  1.0f; posVtxs[i++] =  1.0f; posVtxs[i++] = 0.0f;
    posVtxs[i++] = -1.0f; posVtxs[i++] =  1.0f; posVtxs[i++] = 0.0f;
    GX2Invalidate(GX2_INVALIDATE_CPU_ATTRIB_BUFFER, posVtxs, ciPositionVtxsSize);

    i = 0;
    texCoords[i++] = 0.0f; texCoords[i++] = 1.0f;
    texCoords[i++] = 1.0f; texCoords[i++] = 1.0f;
    texCoords[i++] = 1.0f; texCoords[i++] = 0.0f;
    texCoords[i++] = 0.0f; texCoords[i++] = 0.0f;
    GX2Invalidate(GX2_INVALIDATE_CPU_ATTRIB_BUFFER, texCoords, ciTexCoordsVtxsSize);
}

Texture2DShader::~Texture2DShader()
{
    if(posVtxs)
    {
        free(posVtxs);
        posVtxs = NULL;
    }
    if(texCoords)
    {
        free(texCoords);
        texCoords = NULL;
    }

    delete fetchShader;
    fetchShader = NULL;
}