/*

  Simple DirectMedia Layer

  Copyright (C) 1997-2012 Sam Lantinga <slouken@libsdl.org>

  This software is provided 'as-is', without any express or implied

  warranty.  In no event will the authors be held liable for any damages

  arising from the use of this software.

  Permission is granted to anyone to use this software for any purpose,

  including commercial applications, and to alter it and redistribute it

  freely, subject to the following restrictions:

  1. The origin of this software must not be misrepresented; you must not

     claim that you wrote the original software. If you use this software

     in a product, an acknowledgment in the product documentation would be

     appreciated but is not required.

  2. Altered source versions must be plainly marked as such, and must not be

     misrepresented as being the original software.

  3. This notice may not be removed or altered from any source distribution.

*/

#include "SDL_config.h"

#if SDL_VIDEO_RENDER_D3D11 && !SDL_RENDER_DISABLED

extern "C" {

#include "../../core/windows/SDL_windows.h"

//#include "SDL_hints.h"

//#include "SDL_loadso.h"

#include "SDL_system.h"

#include "SDL_syswm.h"

#include "../SDL_sysrender.h"

//#include "stdio.h"

}

#include <fstream>

#include <string>

#include <vector>

#include "SDL_render_d3d11_cpp.h"

using namespace DirectX;

using namespace Microsoft::WRL;

using namespace std;

#ifdef __WINRT__

using namespace Windows::Graphics::Display;

using namespace Windows::UI::Core;

#endif

/* Direct3D 11.1 renderer implementation */

static SDL_Renderer *D3D11_CreateRenderer(SDL_Window * window, Uint32 flags);

static void D3D11_WindowEvent(SDL_Renderer * renderer,

                            const SDL_WindowEvent *event);

static int D3D11_CreateTexture(SDL_Renderer * renderer, SDL_Texture * texture);

static int D3D11_UpdateTexture(SDL_Renderer * renderer, SDL_Texture * texture,

                             const SDL_Rect * rect, const void *pixels,

                             int pitch);

//static int D3D11_LockTexture(SDL_Renderer * renderer, SDL_Texture * texture,

//                           const SDL_Rect * rect, void **pixels, int *pitch);

//static void D3D11_UnlockTexture(SDL_Renderer * renderer, SDL_Texture * texture);

//static int D3D11_SetRenderTarget(SDL_Renderer * renderer, SDL_Texture * texture);

static int D3D11_UpdateViewport(SDL_Renderer * renderer);

static int D3D11_RenderClear(SDL_Renderer * renderer);

//static int D3D11_RenderDrawPoints(SDL_Renderer * renderer,

//                                const SDL_FPoint * points, int count);

//static int D3D11_RenderDrawLines(SDL_Renderer * renderer,

//                               const SDL_FPoint * points, int count);

//static int D3D11_RenderFillRects(SDL_Renderer * renderer,

//                               const SDL_FRect * rects, int count);

static int D3D11_RenderCopy(SDL_Renderer * renderer, SDL_Texture * texture,

                            const SDL_Rect * srcrect, const SDL_FRect * dstrect);

//static int D3D11_RenderCopyEx(SDL_Renderer * renderer, SDL_Texture * texture,

//                          const SDL_Rect * srcrect, const SDL_FRect * dstrect,

//                          const double angle, const SDL_FPoint * center, const SDL_RendererFlip flip);

//static int D3D11_RenderReadPixels(SDL_Renderer * renderer, const SDL_Rect * rect,

//                                Uint32 format, void * pixels, int pitch);

static void D3D11_RenderPresent(SDL_Renderer * renderer);

static void D3D11_DestroyTexture(SDL_Renderer * renderer,

                                 SDL_Texture * texture);

static void D3D11_DestroyRenderer(SDL_Renderer * renderer);

/* Direct3D 11.1 Internal Functions */

HRESULT D3D11_CreateDeviceResources(SDL_Renderer * renderer);

HRESULT D3D11_CreateWindowSizeDependentResources(SDL_Renderer * renderer);

HRESULT D3D11_UpdateForWindowSizeChange(SDL_Renderer * renderer);

HRESULT D3D11_HandleDeviceLost(SDL_Renderer * renderer);

extern "C" {

    SDL_RenderDriver D3D11_RenderDriver = {

    D3D11_CreateRenderer,

    {

     "direct3d 11.1",

     (SDL_RENDERER_ACCELERATED | SDL_RENDERER_PRESENTVSYNC | SDL_RENDERER_TARGETTEXTURE),

     1,

     {SDL_PIXELFORMAT_RGB888},

     0,

     0}

    };

}

//typedef struct

//{

//    float x, y, z;

//    DWORD color;

//    float u, v;

//} Vertex;

SDL_Renderer *

D3D11_CreateRenderer(SDL_Window * window, Uint32 flags)

{

    SDL_Renderer *renderer;

    D3D11_RenderData *data;

    renderer = (SDL_Renderer *) SDL_calloc(1, sizeof(*renderer));

    if (!renderer) {

        SDL_OutOfMemory();

        return NULL;

    }

    SDL_zerop(renderer);

    data = new D3D11_RenderData;    // Use the C++ 'new' operator to make sure the struct's members initialize using C++ rules

    if (!data) {

        SDL_OutOfMemory();

        return NULL;

    }

    data->featureLevel = (D3D_FEATURE_LEVEL) 0;

    data->loadingComplete = false;

    data->windowSizeInDIPs = XMFLOAT2(0, 0);

    data->renderTargetSize = XMFLOAT2(0, 0);

    renderer->WindowEvent = D3D11_WindowEvent;

    renderer->CreateTexture = D3D11_CreateTexture;

    renderer->UpdateTexture = D3D11_UpdateTexture;

    //renderer->LockTexture = D3D11_LockTexture;

    //renderer->UnlockTexture = D3D11_UnlockTexture;

    //renderer->SetRenderTarget = D3D11_SetRenderTarget;

    renderer->UpdateViewport = D3D11_UpdateViewport;

    renderer->RenderClear = D3D11_RenderClear;

    //renderer->RenderDrawPoints = D3D11_RenderDrawPoints;

    //renderer->RenderDrawLines = D3D11_RenderDrawLines;

    //renderer->RenderFillRects = D3D11_RenderFillRects;

    renderer->RenderCopy = D3D11_RenderCopy;

    //renderer->RenderCopyEx = D3D11_RenderCopyEx;

    //renderer->RenderReadPixels = D3D11_RenderReadPixels;

    renderer->RenderPresent = D3D11_RenderPresent;

    renderer->DestroyTexture = D3D11_DestroyTexture;

    renderer->DestroyRenderer = D3D11_DestroyRenderer;

    renderer->info = D3D11_RenderDriver.info;

    renderer->info.flags = SDL_RENDERER_ACCELERATED;

    renderer->driverdata = data;

    // HACK: make sure the SDL_Renderer references the SDL_Window data now, in

    // order to give init functions access to the underlying window handle:

    renderer->window = window;

    /* Initialize Direct3D resources */

    if (FAILED(D3D11_CreateDeviceResources(renderer))) {

        D3D11_DestroyRenderer(renderer);

        return NULL;

    }

    if (FAILED(D3D11_CreateWindowSizeDependentResources(renderer))) {

        D3D11_DestroyRenderer(renderer);

        return NULL;

    }

    // TODO, WinRT: fill in renderer->info.texture_formats where appropriate

    return renderer;

}

static void

D3D11_DestroyRenderer(SDL_Renderer * renderer)

{

    D3D11_RenderData *data = (D3D11_RenderData *) renderer->driverdata;

    if (data) {

        delete data;

        data = NULL;

    }

}

static bool

D3D11_ReadFileContents(const wstring & fileName, vector<char> & out)

{

    ifstream in(fileName, ios::in | ios::binary);

    if (!in) {

        return false;

    }

    in.seekg(0, ios::end);

    out.resize((size_t) in.tellg());

    in.seekg(0, ios::beg);

    in.read(&out[0], out.size());

    return in.good();

}

static bool

D3D11_ReadShaderContents(const wstring & shaderName, vector<char> & out)

{

    wstring fileName;

#if WINAPI_FAMILY == WINAPI_FAMILY_APP

    fileName = SDL_WinRTGetFileSystemPath(SDL_WINRT_PATH_INSTALLED_LOCATION);

    fileName += L"\\SDL_VS2012_WinRT\\";

#elif WINAPI_FAMILY == WINAPI_PHONE_APP

    fileName = SDL_WinRTGetFileSystemPath(SDL_WINRT_PATH_INSTALLED_LOCATION);

    fileName += L"\\";

#endif

    // WinRT, TODO: test Direct3D 11.1 shader loading on Win32

    fileName += shaderName;

    return D3D11_ReadFileContents(fileName, out);

}

// Create resources that depend on the device.

HRESULT

D3D11_CreateDeviceResources(SDL_Renderer * renderer)

{

    D3D11_RenderData *data = (D3D11_RenderData *) renderer->driverdata;

    // This flag adds support for surfaces with a different color channel ordering

    // than the API default. It is required for compatibility with Direct2D.

    UINT creationFlags = D3D11_CREATE_DEVICE_BGRA_SUPPORT;

#if defined(_DEBUG)

    // If the project is in a debug build, enable debugging via SDK Layers with this flag.

    creationFlags |= D3D11_CREATE_DEVICE_DEBUG;

#endif

    // This array defines the set of DirectX hardware feature levels this app will support.

    // Note the ordering should be preserved.

    // Don't forget to declare your application's minimum required feature level in its

    // description.  All applications are assumed to support 9.1 unless otherwise stated.

    D3D_FEATURE_LEVEL featureLevels[] = 

    {

        D3D_FEATURE_LEVEL_11_1,

        D3D_FEATURE_LEVEL_11_0,

        D3D_FEATURE_LEVEL_10_1,

        D3D_FEATURE_LEVEL_10_0,

        D3D_FEATURE_LEVEL_9_3,

        D3D_FEATURE_LEVEL_9_2,

        D3D_FEATURE_LEVEL_9_1

    };

    // Create the Direct3D 11 API device object and a corresponding context.

    ComPtr<ID3D11Device> device;

    ComPtr<ID3D11DeviceContext> context;

    HRESULT result = S_OK;

    result = D3D11CreateDevice(

        nullptr, // Specify nullptr to use the default adapter.

        D3D_DRIVER_TYPE_HARDWARE,

        nullptr,

        creationFlags, // Set set debug and Direct2D compatibility flags.

        featureLevels, // List of feature levels this app can support.

        ARRAYSIZE(featureLevels),

        D3D11_SDK_VERSION, // Always set this to D3D11_SDK_VERSION for Windows Store apps.

        &device, // Returns the Direct3D device created.

        &data->featureLevel, // Returns feature level of device created.

        &context // Returns the device immediate context.

        );

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    // Get the Direct3D 11.1 API device and context interfaces.

    Microsoft::WRL::ComPtr<ID3D11Device1> d3dDevice1;

    result = device.As(&(data->d3dDevice));

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    result = context.As(&data->d3dContext);

    if (FAILED(result)) {

        return result;

    }

    // Start loading GPU shaders:

    vector<char> fileData;

    //

    // Load in SDL's one and only vertex shader:

    //

    if (!D3D11_ReadShaderContents(L"SimpleVertexShader.cso", fileData)) {

        SDL_SetError("Unable to open SDL's vertex shader file.");

        return E_FAIL;

    }

    result = data->d3dDevice->CreateVertexShader(

        &fileData[0],

        fileData.size(),

        nullptr,

        &data->vertexShader

        );

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    //

    // Create an input layout for SDL's vertex shader:

    //

    const D3D11_INPUT_ELEMENT_DESC vertexDesc[] = 

    {

        { "POSITION", 0, DXGI_FORMAT_R32G32B32_FLOAT, 0, 0,  D3D11_INPUT_PER_VERTEX_DATA, 0 },

        { "TEXCOORD", 0, DXGI_FORMAT_R32G32_FLOAT, 0, 12, D3D11_INPUT_PER_VERTEX_DATA, 0 },

    };

    result = data->d3dDevice->CreateInputLayout(

        vertexDesc,

        ARRAYSIZE(vertexDesc),

        &fileData[0],

        fileData.size(),

        &data->inputLayout

        );

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    //

    // Load in SDL's one and only pixel shader (for now, more are likely to follow):

    //

    if (!D3D11_ReadShaderContents(L"SimplePixelShader.cso", fileData)) {

        SDL_SetError("Unable to open SDL's pixel shader file.");

        return E_FAIL;

    }

    result = data->d3dDevice->CreatePixelShader(

        &fileData[0],

        fileData.size(),

        nullptr,

        &data->pixelShader

        );

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    //

    // Setup space to hold vertex shader constants:

    //

    CD3D11_BUFFER_DESC constantBufferDesc(sizeof(SDL_VertexShaderConstants), D3D11_BIND_CONSTANT_BUFFER);

    result = data->d3dDevice->CreateBuffer(

		&constantBufferDesc,

		nullptr,

        &data->vertexShaderConstants

		);

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    //

    // Make sure that the vertex buffer, if already created, gets freed.

    // It will be recreated later.

    //

    data->vertexBuffer = nullptr;

    //

    // Create a sampler to use when drawing textures:

    //

    D3D11_SAMPLER_DESC samplerDesc;

    samplerDesc.Filter = D3D11_FILTER_MIN_MAG_MIP_POINT;

    samplerDesc.AddressU = D3D11_TEXTURE_ADDRESS_CLAMP;

    samplerDesc.AddressV = D3D11_TEXTURE_ADDRESS_CLAMP;

    samplerDesc.AddressW = D3D11_TEXTURE_ADDRESS_CLAMP;

    samplerDesc.MipLODBias = 0.0f;

    samplerDesc.MaxAnisotropy = 1;

    samplerDesc.ComparisonFunc = D3D11_COMPARISON_ALWAYS;

    samplerDesc.BorderColor[0] = 0.0f;

    samplerDesc.BorderColor[1] = 0.0f;

    samplerDesc.BorderColor[2] = 0.0f;

    samplerDesc.BorderColor[3] = 0.0f;

    samplerDesc.MinLOD = 0.0f;

    samplerDesc.MaxLOD = D3D11_FLOAT32_MAX;

    result = data->d3dDevice->CreateSamplerState(

        &samplerDesc,

        &data->mainSampler

        );

    if (FAILED(result)) {

        WIN_SetErrorFromHRESULT(__FUNCTION__, result);

        return result;

    }

    //

    // All done!

    //

    data->loadingComplete = true;       // This variable can probably be factored-out

    return S_OK;

}

#ifdef __WINRT__

static CoreWindow ^

D3D11_GetCoreWindowFromSDLRenderer(SDL_Renderer * renderer)

{

    SDL_Window * sdlWindow = renderer->window;

    if ( ! renderer->window ) {

        return nullptr;

    }

    SDL_SysWMinfo sdlWindowInfo;

    SDL_VERSION(&sdlWindowInfo.version);

    if ( ! SDL_GetWindowWMInfo(sdlWindow, &sdlWindowInfo) ) {

        return nullptr;

    }

    if (sdlWindowInfo.subsystem != SDL_SYSWM_WINDOWSRT) {

        return nullptr;

    }

    CoreWindow ^* coreWindowPointer = (CoreWindow ^*) sdlWindowInfo.info.winrt.window;

    if ( ! coreWindowPointer ) {

        return nullptr;

    }

    return *coreWindowPointer;

}

// Method to convert a length in device-independent pixels (DIPs) to a length in physical pixels.

static float

D3D11_ConvertDipsToPixels(float dips)

{

    static const float dipsPerInch = 96.0f;

    return floor(dips * DisplayProperties::LogicalDpi / dipsPerInch + 0.5f); // Round to nearest integer.

}

#endif

// Initialize all resources that change when the window's size changes.

// WinRT, TODO: get D3D11_CreateWindowSizeDependentResources working on Win32

HRESULT

D3D11_CreateWindowSizeDependentResources(SDL_Renderer * renderer)

{

    D3D11_RenderData *data = (D3D11_RenderData *) renderer->driverdata;

    HRESULT result = S_OK;

    Windows::UI::Core::CoreWindow ^ coreWindow = D3D11_GetCoreWindowFromSDLRenderer(renderer);

    // Store the window bounds so the next time we get a SizeChanged event we can

    // avoid rebuilding everything if the size is identical.

    data->windowSizeInDIPs.x = coreWindow->Bounds.Width;

    data->windowSizeInDIPs.y = coreWindow->Bounds.Height;

    // Calculate the necessary swap chain and render target size in pixels.

    float windowWidth = D3D11_ConvertDipsToPixels(data->windowSizeInDIPs.x);

    float windowHeight = D3D11_ConvertDipsToPixels(data->windowSizeInDIPs.y);

    // The width and height of the swap chain must be based on the window's

    // landscape-oriented width and height. If the window is in a portrait

    // orientation, the dimensions must be reversed.

    data->orientation = DisplayProperties::CurrentOrientation;

    bool swapDimensions =

        data->orientation == DisplayOrientations::Portrait ||

        data->orientation == DisplayOrientations::PortraitFlipped;

    data->renderTargetSize.x = swapDimensions ? windowHeight : windowWidth;

    data->renderTargetSize.y = swapDimensions ? windowWidth : windowHeight;

    if(data->swapChain != nullptr)

    {

        // If the swap chain already exists, resize it.

        result = data->swapChain->ResizeBuffers(

            2, // Double-buffered swap chain.

            static_cast<UINT>(data->renderTargetSize.x),

            static_cast<UINT>(data->renderTargetSize.y),

            DXGI_FORMAT_B8G8R8A8_UNORM,

            0

            );

        if (FAILED(result)) {

            WIN_SetErrorFromHRESULT(__FUNCTION__, result);

            return result;

        }

    }

    else

    {

        // Otherwise, create a new one using the same adapter as the existing Direct3D device.

        DXGI_SWAP_CHAIN_DESC1 swapChainDesc = {0};