#include "Mesh.h"
#include "Utilities.h"
#include "Vertex.h"

#include <DirectXMath.h>

using namespace DirectX;

UINT Mesh::init(IDirect3DDevice9* pD3DDevice)
{
    UINT error = 0;

    error = createVertexBuffer(pD3DDevice);
    CheckError(error);

    error = createIndexBuffer(pD3DDevice);
    CheckError(error);

    // identity matrix
    XMMATRIX identityMatrix = XMMatrixIdentity();
    XMFLOAT4X4 worldMatrix;
    XMStoreFloat4x4(&worldMatrix, identityMatrix);
    D3DMATRIX* pWorldMatrix = reinterpret_cast<D3DMATRIX*>(&worldMatrix);
    _worldMatrix = *pWorldMatrix;

    return 0;
}

void Mesh::update(FLOAT dt)
{
    static FLOAT x = 0.0f;
    static FLOAT y = 0.0f;
    static FLOAT z = 0.0f;
    static FLOAT pitch = XM_PIDIV4;
    static FLOAT yaw = 0.0f;
    static FLOAT roll = XM_PIDIV4;
    static FLOAT scale = 1.0f;

    yaw += 1.0f * dt;

    FLOAT move = 1.0f * dt;

    if (GetAsyncKeyState(VK_LEFT) & 0x8000) x -= move;
    if (GetAsyncKeyState(VK_RIGHT) & 0x8000) x += move;

    if (GetAsyncKeyState(VK_DOWN) & 0x8000) y -= move;
    if (GetAsyncKeyState(VK_UP) & 0x8000) y += move;

    if (GetAsyncKeyState('Q') & 0x8000) z -= move;
    if (GetAsyncKeyState('E') & 0x8000) z += move;

    XMMATRIX translationMatrix = XMMatrixTranslation(x, y, z);
    XMMATRIX rotationMatrix = XMMatrixRotationRollPitchYaw(pitch, yaw, roll);
    XMMATRIX scaleMatrix = XMMatrixScaling(scale, scale, scale);

    XMStoreFloat4x4(reinterpret_cast<XMFLOAT4X4*>(&_worldMatrix), scaleMatrix * rotationMatrix * translationMatrix);
}

void Mesh::render(IDirect3DDevice9* pD3DDevice)
{
    // 1. set mesh data
    pD3DDevice->SetStreamSource(0, _pVertexBuffer, 0, _vertexStride);
    pD3DDevice->SetFVF(FVF);
    pD3DDevice->SetIndices(_pIndexBuffer);
    pD3DDevice->SetTransform(D3DTS_WORLD, &_worldMatrix);

    // 2. draw call
    //pD3DDevice->DrawPrimitive(D3DPT_TRIANGLELIST, 0, _vertexCount / 3);
    //pD3DDevice->DrawPrimitive(D3DPT_TRIANGLESTRIP, 0, _vertexCount - 2);
    //pD3DDevice->DrawPrimitive(D3DPT_TRIANGLEFAN, 0, _vertexCount - 2);
    pD3DDevice->DrawIndexedPrimitive(D3DPT_TRIANGLELIST, 0, 0, _vertexCount, 0, _indexCount / 3);
}

void Mesh::deInit()
{
    safeRelease(_pIndexBuffer);
    safeRelease(_pVertexBuffer);
}

UINT Mesh::createVertexBuffer(IDirect3DDevice9* pD3DDevice)
{
    // 1. define mesh structure
    _vertexCount = 8;
    _vertexStride = sizeof(Vertex);

    // 2. create vertex buffer WITHOUT data
    HRESULT hr = pD3DDevice->CreateVertexBuffer(
        _vertexCount * _vertexStride, // vertex buffer byte length
        D3DUSAGE_WRITEONLY, // vertex buffer access mode
        FVF, // FVF - Flexible Vertex Format
        D3DPOOL_MANAGED, // video memory management
        &_pVertexBuffer,
        nullptr // not used anymore
    );
    CheckFailed(hr, 300);

    // 3. fill vertex buffer with data
    Vertex* pVertices = nullptr;
    hr = _pVertexBuffer->Lock(0, 0, reinterpret_cast<void**>(&pVertices), 0);
    CheckFailed(hr, 310);

    // triangle
    //pVertices[0] = Vertex(0.0f, 0.5f, 0.0f);
    //pVertices[1] = Vertex(0.5f, -0.5f, 0.0f);
    //pVertices[2] = Vertex(-0.5f, -0.5f, 0.0f);

    // quad with trianglelist WITHOUT index buffer
    //// triangle 1
    //pVertices[0] = Vertex(-0.5f, 0.5f, 0.0f);
    //pVertices[1] = Vertex(0.5f, 0.5f, 0.0f);
    //pVertices[2] = Vertex(0.5f, -0.5f, 0.0f);

    //// triangle 2
    //pVertices[3] = Vertex(-0.5f, 0.5f, 0.0f);
    //pVertices[4] = Vertex(0.5f, -0.5f, 0.0f);
    //pVertices[5] = Vertex(-0.5f, -0.5f, 0.0f);

    // quad with trianglestrip
    //pVertices[0] = Vertex(-0.5f, 0.5f, 0.0f);
    //pVertices[1] = Vertex(0.5f, 0.5f, 0.0f);
    //pVertices[2] = Vertex(-0.5f, -0.5f, 0.0f);
    //pVertices[3] = Vertex(0.5f, -0.5f, 0.0f);

    //// quad with trianglefan
    //// quad with trianglelist & index buffer
    //pVertices[0] = Vertex(-0.5f, 0.5f, 0.0f);
    //pVertices[1] = Vertex(0.5f, 0.5f, 0.0f);
    //pVertices[2] = Vertex(0.5f, -0.5f, 0.0f);
    //pVertices[3] = Vertex(-0.5f, -0.5f, 0.0f);

    // cube
    // front
    pVertices[0] = Vertex(-0.5f, 0.5f, -0.5f, 255, 0, 0);
    pVertices[1] = Vertex(0.5f, 0.5f, -0.5f, 0, 255, 0);
    pVertices[2] = Vertex(0.5f, -0.5f, -0.5f, 0, 0, 255);
    pVertices[3] = Vertex(-0.5f, -0.5f, -0.5f, 255, 0, 255);

    // back
    pVertices[4] = Vertex(0.5f, 0.5f, 0.5f, 255, 255, 0);
    pVertices[5] = Vertex(-0.5f, 0.5f, 0.5f, 0, 255, 255);
    pVertices[6] = Vertex(-0.5f, -0.5f, 0.5f, 0, 0, 0);
    pVertices[7] = Vertex(0.5f, -0.5f, 0.5f, 200, 200, 200);

    hr = _pVertexBuffer->Unlock();
    CheckFailed(hr, 320);

    pVertices = nullptr;

    return 0;
}

UINT Mesh::createIndexBuffer(IDirect3DDevice9* pD3DDevice)
{
    _indexCount = 36; // triangle * 3

    HRESULT hr = pD3DDevice->CreateIndexBuffer(
        _indexCount * sizeof(WORD),
        D3DUSAGE_WRITEONLY,
        D3DFMT_INDEX16,
        D3DPOOL_MANAGED,
        &_pIndexBuffer,
        nullptr
    );
    CheckFailed(hr, 330);

    WORD* pIndices = nullptr;
    hr = _pIndexBuffer->Lock(0, 0, reinterpret_cast<void**>(&pIndices), 0);
    CheckFailed(hr, 340);

    //// quad
    //// triangle 1
    //pIndices[0] = 0; pIndices[1] = 1; pIndices[2] = 2;

    //// triangle 2
    //pIndices[3] = 0; pIndices[4] = 2; pIndices[5] = 3;

    // cube
    // front
    pIndices[0] = 0; pIndices[1] = 1; pIndices[2] = 2;
    pIndices[3] = 0; pIndices[4] = 2; pIndices[5] = 3;

    // back
    pIndices[6] = 4; pIndices[7] = 5; pIndices[8] = 6;
    pIndices[9] = 4; pIndices[10] = 6; pIndices[11] = 7;

    // left
    pIndices[12] = 5; pIndices[13] = 0; pIndices[14] = 3;
    pIndices[15] = 5; pIndices[16] = 3; pIndices[17] = 6;

    // right
    pIndices[18] = 1; pIndices[19] = 4; pIndices[20] = 7;
    pIndices[21] = 1; pIndices[22] = 7; pIndices[23] = 2;

    // top
    pIndices[24] = 5; pIndices[25] = 4; pIndices[26] = 1;
    pIndices[27] = 5; pIndices[28] = 1; pIndices[29] = 0;

    // bottom
    pIndices[30] = 3; pIndices[31] = 2; pIndices[32] = 7;
    pIndices[33] = 3; pIndices[34] = 7; pIndices[35] = 6;

    hr = _pIndexBuffer->Unlock();
    CheckFailed(hr, 350);

    pIndices = nullptr;

    return 0;
}