/*

    SDL - Simple DirectMedia Layer

    Copyright (C) 1997-2011 Sam Lantinga

    This library is free software; you can redistribute it and/or

    modify it under the terms of the GNU Lesser General Public

    License as published by the Free Software Foundation; either

    version 2.1 of the License, or (at your option) any later version.

    This library 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

    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public

    License along with this library; if not, write to the Free Software

    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    Sam Lantinga

    slouken@libsdl.org

*/

#include "SDL_config.h"

#if SDL_VIDEO_RENDER_D3D && !SDL_RENDER_DISABLED

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

#include "SDL_loadso.h"

#include "SDL_syswm.h"

#include "../SDL_sysrender.h"

#if SDL_VIDEO_RENDER_D3D

#define D3D_DEBUG_INFO

#include <d3d9.h>

#endif

#ifdef ASSEMBLE_SHADER

///////////////////////////////////////////////////////////////////////////

// ID3DXBuffer:

// ------------

// The buffer object is used by D3DX to return arbitrary size data.

//

// GetBufferPointer -

//    Returns a pointer to the beginning of the buffer.

//

// GetBufferSize -

//    Returns the size of the buffer, in bytes.

///////////////////////////////////////////////////////////////////////////

typedef interface ID3DXBuffer ID3DXBuffer;

typedef interface ID3DXBuffer *LPD3DXBUFFER;

// {8BA5FB08-5195-40e2-AC58-0D989C3A0102}

DEFINE_GUID(IID_ID3DXBuffer, 

0x8ba5fb08, 0x5195, 0x40e2, 0xac, 0x58, 0xd, 0x98, 0x9c, 0x3a, 0x1, 0x2);

#undef INTERFACE

#define INTERFACE ID3DXBuffer

typedef interface ID3DXBuffer {

    const struct ID3DXBufferVtbl FAR* lpVtbl;

} ID3DXBuffer;

typedef const struct ID3DXBufferVtbl ID3DXBufferVtbl;

const struct ID3DXBufferVtbl

{

    // IUnknown

    STDMETHOD(QueryInterface)(THIS_ REFIID iid, LPVOID *ppv) PURE;

    STDMETHOD_(ULONG, AddRef)(THIS) PURE;

    STDMETHOD_(ULONG, Release)(THIS) PURE;

    // ID3DXBuffer

    STDMETHOD_(LPVOID, GetBufferPointer)(THIS) PURE;

    STDMETHOD_(DWORD, GetBufferSize)(THIS) PURE;

};

HRESULT WINAPI

    D3DXAssembleShader(

        LPCSTR                          pSrcData,

        UINT                            SrcDataLen,

        CONST LPVOID*                   pDefines,

        LPVOID                          pInclude,

        DWORD                           Flags,

        LPD3DXBUFFER*                   ppShader,

        LPD3DXBUFFER*                   ppErrorMsgs);

#endif /* ASSEMBLE_SHADER */

/* Direct3D renderer implementation */

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

static void D3D_WindowEvent(SDL_Renderer * renderer,

                            const SDL_WindowEvent *event);

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

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

                             const SDL_Rect * rect, const void *pixels,

                             int pitch);

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

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

static void D3D_UnlockTexture(SDL_Renderer * renderer, SDL_Texture * texture);

static int D3D_UpdateViewport(SDL_Renderer * renderer);

static int D3D_RenderClear(SDL_Renderer * renderer);

static int D3D_RenderDrawPoints(SDL_Renderer * renderer,

                                const SDL_Point * points, int count);

static int D3D_RenderDrawLines(SDL_Renderer * renderer,

                               const SDL_Point * points, int count);

static int D3D_RenderFillRects(SDL_Renderer * renderer,

                               const SDL_Rect * rects, int count);

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

                          const SDL_Rect * srcrect, const SDL_Rect * dstrect);

static int D3D_RenderReadPixels(SDL_Renderer * renderer, const SDL_Rect * rect,

                                Uint32 format, void * pixels, int pitch);

static void D3D_RenderPresent(SDL_Renderer * renderer);

static void D3D_DestroyTexture(SDL_Renderer * renderer,

                               SDL_Texture * texture);

static void D3D_DestroyRenderer(SDL_Renderer * renderer);

SDL_RenderDriver D3D_RenderDriver = {

    D3D_CreateRenderer,

    {

     "direct3d",

     (SDL_RENDERER_ACCELERATED | SDL_RENDERER_PRESENTVSYNC),

     1,

     {SDL_PIXELFORMAT_ARGB8888},

     0,

     0}

};

typedef struct

{

    void* d3dDLL;

    IDirect3D9 *d3d;

    IDirect3DDevice9 *device;

    UINT adapter;

    D3DPRESENT_PARAMETERS pparams;

    SDL_bool updateSize;

    SDL_bool beginScene;

} D3D_RenderData;

typedef struct

{

    IDirect3DTexture9 *texture;

} D3D_TextureData;

typedef struct

{

    float x, y, z;

    DWORD color;

    float u, v;

} Vertex;

static void

D3D_SetError(const char *prefix, HRESULT result)

{

    const char *error;

    switch (result) {

    case D3DERR_WRONGTEXTUREFORMAT:

        error = "WRONGTEXTUREFORMAT";

        break;

    case D3DERR_UNSUPPORTEDCOLOROPERATION:

        error = "UNSUPPORTEDCOLOROPERATION";

        break;

    case D3DERR_UNSUPPORTEDCOLORARG:

        error = "UNSUPPORTEDCOLORARG";

        break;

    case D3DERR_UNSUPPORTEDALPHAOPERATION:

        error = "UNSUPPORTEDALPHAOPERATION";

        break;

    case D3DERR_UNSUPPORTEDALPHAARG:

        error = "UNSUPPORTEDALPHAARG";

        break;

    case D3DERR_TOOMANYOPERATIONS:

        error = "TOOMANYOPERATIONS";

        break;

    case D3DERR_CONFLICTINGTEXTUREFILTER:

        error = "CONFLICTINGTEXTUREFILTER";

        break;

    case D3DERR_UNSUPPORTEDFACTORVALUE:

        error = "UNSUPPORTEDFACTORVALUE";

        break;

    case D3DERR_CONFLICTINGRENDERSTATE:

        error = "CONFLICTINGRENDERSTATE";

        break;

    case D3DERR_UNSUPPORTEDTEXTUREFILTER:

        error = "UNSUPPORTEDTEXTUREFILTER";

        break;

    case D3DERR_CONFLICTINGTEXTUREPALETTE:

        error = "CONFLICTINGTEXTUREPALETTE";

        break;

    case D3DERR_DRIVERINTERNALERROR:

        error = "DRIVERINTERNALERROR";

        break;

    case D3DERR_NOTFOUND:

        error = "NOTFOUND";

        break;

    case D3DERR_MOREDATA:

        error = "MOREDATA";

        break;

    case D3DERR_DEVICELOST:

        error = "DEVICELOST";

        break;

    case D3DERR_DEVICENOTRESET:

        error = "DEVICENOTRESET";

        break;

    case D3DERR_NOTAVAILABLE:

        error = "NOTAVAILABLE";

        break;

    case D3DERR_OUTOFVIDEOMEMORY:

        error = "OUTOFVIDEOMEMORY";

        break;

    case D3DERR_INVALIDDEVICE:

        error = "INVALIDDEVICE";

        break;

    case D3DERR_INVALIDCALL:

        error = "INVALIDCALL";

        break;

    case D3DERR_DRIVERINVALIDCALL:

        error = "DRIVERINVALIDCALL";

        break;

    case D3DERR_WASSTILLDRAWING:

        error = "WASSTILLDRAWING";

        break;

    default:

        error = "UNKNOWN";

        break;

    }

    SDL_SetError("%s: %s", prefix, error);

}

static D3DFORMAT

PixelFormatToD3DFMT(Uint32 format)

{

    switch (format) {

    case SDL_PIXELFORMAT_RGB565:

        return D3DFMT_R5G6B5;

    case SDL_PIXELFORMAT_RGB888:

        return D3DFMT_X8R8G8B8;

    case SDL_PIXELFORMAT_ARGB8888:

        return D3DFMT_A8R8G8B8;

    default:

        return D3DFMT_UNKNOWN;

    }

}

static Uint32

D3DFMTToPixelFormat(D3DFORMAT format)

{

    switch (format) {

    case D3DFMT_R5G6B5:

        return SDL_PIXELFORMAT_RGB565;

    case D3DFMT_X8R8G8B8:

        return SDL_PIXELFORMAT_RGB888;

    case D3DFMT_A8R8G8B8:

        return SDL_PIXELFORMAT_ARGB8888;

    default:

        return SDL_PIXELFORMAT_UNKNOWN;

    }

}

static int

D3D_Reset(SDL_Renderer * renderer)

{

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

    HRESULT result;

    result = IDirect3DDevice9_Reset(data->device, &data->pparams);

    if (FAILED(result)) {

        if (result == D3DERR_DEVICELOST) {

            /* Don't worry about it, we'll reset later... */

            return 0;

        } else {

            D3D_SetError("Reset()", result);

            return -1;

        }

    }

    IDirect3DDevice9_SetVertexShader(data->device, NULL);

    IDirect3DDevice9_SetFVF(data->device,

                            D3DFVF_XYZ | D3DFVF_DIFFUSE | D3DFVF_TEX1);

    IDirect3DDevice9_SetRenderState(data->device, D3DRS_CULLMODE,

                                    D3DCULL_NONE);

    IDirect3DDevice9_SetRenderState(data->device, D3DRS_LIGHTING, FALSE);

    return 0;

}

static int

D3D_ActivateRenderer(SDL_Renderer * renderer)

{

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

    HRESULT result;

    if (data->updateSize) {

        SDL_Window *window = renderer->window;

        int w, h;

        SDL_GetWindowSize(window, &w, &h);

        data->pparams.BackBufferWidth = w;

        data->pparams.BackBufferHeight = h;

        if (SDL_GetWindowFlags(window) & SDL_WINDOW_FULLSCREEN) {

            data->pparams.BackBufferFormat =

                PixelFormatToD3DFMT(SDL_GetWindowPixelFormat(window));

        } else {

            data->pparams.BackBufferFormat = D3DFMT_UNKNOWN;

        }

        if (D3D_Reset(renderer) < 0) {

            return -1;

        }

        D3D_UpdateViewport(renderer);

        data->updateSize = SDL_FALSE;

    }

    if (data->beginScene) {

        result = IDirect3DDevice9_BeginScene(data->device);

        if (result == D3DERR_DEVICELOST) {

            if (D3D_Reset(renderer) < 0) {

                return -1;

            }

            result = IDirect3DDevice9_BeginScene(data->device);

        }

        if (FAILED(result)) {

            D3D_SetError("BeginScene()", result);

            return -1;

        }

        data->beginScene = SDL_FALSE;

    }

    return 0;

}

SDL_Renderer *

D3D_CreateRenderer(SDL_Window * window, Uint32 flags)

{

    SDL_Renderer *renderer;

    D3D_RenderData *data;

    SDL_SysWMinfo windowinfo;

    HRESULT result;

    D3DPRESENT_PARAMETERS pparams;

    IDirect3DSwapChain9 *chain;

    D3DCAPS9 caps;

    Uint32 window_flags;

    int w, h;

    SDL_DisplayMode fullscreen_mode;

    D3DMATRIX matrix;

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

    if (!renderer) {

        SDL_OutOfMemory();

        return NULL;

    }

    data = (D3D_RenderData *) SDL_calloc(1, sizeof(*data));

    if (!data) {

        SDL_free(renderer);

        SDL_OutOfMemory();

        return NULL;

    }

    data->d3dDLL = SDL_LoadObject("D3D9.DLL");

    if (data->d3dDLL) {

        IDirect3D9 *(WINAPI * D3DCreate) (UINT SDKVersion);

        D3DCreate =

            (IDirect3D9 * (WINAPI *) (UINT)) SDL_LoadFunction(data->d3dDLL,

                                                            "Direct3DCreate9");

        if (D3DCreate) {

            data->d3d = D3DCreate(D3D_SDK_VERSION);

        }

        if (!data->d3d) {

            SDL_UnloadObject(data->d3dDLL);

            data->d3dDLL = NULL;

        }

    }

    if (!data->d3d) {

        SDL_free(renderer);

        SDL_free(data);

        SDL_SetError("Unable to create Direct3D interface");

        return NULL;

    }

    renderer->WindowEvent = D3D_WindowEvent;

    renderer->CreateTexture = D3D_CreateTexture;

    renderer->UpdateTexture = D3D_UpdateTexture;

    renderer->LockTexture = D3D_LockTexture;

    renderer->UnlockTexture = D3D_UnlockTexture;

    renderer->UpdateViewport = D3D_UpdateViewport;

    renderer->RenderClear = D3D_RenderClear;

    renderer->RenderDrawPoints = D3D_RenderDrawPoints;

    renderer->RenderDrawLines = D3D_RenderDrawLines;

    renderer->RenderFillRects = D3D_RenderFillRects;

    renderer->RenderCopy = D3D_RenderCopy;

    renderer->RenderReadPixels = D3D_RenderReadPixels;

    renderer->RenderPresent = D3D_RenderPresent;

    renderer->DestroyTexture = D3D_DestroyTexture;

    renderer->DestroyRenderer = D3D_DestroyRenderer;

    renderer->info = D3D_RenderDriver.info;

    renderer->driverdata = data;

    renderer->info.flags = SDL_RENDERER_ACCELERATED;

    SDL_VERSION(&windowinfo.version);

    SDL_GetWindowWMInfo(window, &windowinfo);

    window_flags = SDL_GetWindowFlags(window);

    SDL_GetWindowSize(window, &w, &h);

    SDL_GetWindowDisplayMode(window, &fullscreen_mode);

    SDL_zero(pparams);

    pparams.hDeviceWindow = windowinfo.info.win.window;

    pparams.BackBufferWidth = w;

    pparams.BackBufferHeight = h;

    if (window_flags & SDL_WINDOW_FULLSCREEN) {

        pparams.BackBufferFormat =

            PixelFormatToD3DFMT(fullscreen_mode.format);

    } else {

        pparams.BackBufferFormat = D3DFMT_UNKNOWN;

    }

    pparams.BackBufferCount = 1;

    pparams.SwapEffect = D3DSWAPEFFECT_DISCARD;

    if (window_flags & SDL_WINDOW_FULLSCREEN) {

        pparams.Windowed = FALSE;

        pparams.FullScreen_RefreshRateInHz =

            fullscreen_mode.refresh_rate;

    } else {

        pparams.Windowed = TRUE;

        pparams.FullScreen_RefreshRateInHz = 0;

    }

    if (flags & SDL_RENDERER_PRESENTVSYNC) {

        pparams.PresentationInterval = D3DPRESENT_INTERVAL_ONE;

    } else {

        pparams.PresentationInterval = D3DPRESENT_INTERVAL_IMMEDIATE;

    }

    /* FIXME: Which adapter? */

    data->adapter = D3DADAPTER_DEFAULT;

    IDirect3D9_GetDeviceCaps(data->d3d, data->adapter, D3DDEVTYPE_HAL, &caps);

    result = IDirect3D9_CreateDevice(data->d3d, data->adapter,

                                     D3DDEVTYPE_HAL,

                                     pparams.hDeviceWindow,

                                     (caps.

                                      DevCaps &

                                      D3DDEVCAPS_HWTRANSFORMANDLIGHT) ?

                                     D3DCREATE_HARDWARE_VERTEXPROCESSING :

                                     D3DCREATE_SOFTWARE_VERTEXPROCESSING,

                                     &pparams, &data->device);

    if (FAILED(result)) {

        D3D_DestroyRenderer(renderer);

        D3D_SetError("CreateDevice()", result);

        return NULL;

    }

    data->beginScene = SDL_TRUE;

    /* Get presentation parameters to fill info */

    result = IDirect3DDevice9_GetSwapChain(data->device, 0, &chain);

    if (FAILED(result)) {

        D3D_DestroyRenderer(renderer);

        D3D_SetError("GetSwapChain()", result);

        return NULL;

    }

    result = IDirect3DSwapChain9_GetPresentParameters(chain, &pparams);

    if (FAILED(result)) {

        IDirect3DSwapChain9_Release(chain);

        D3D_DestroyRenderer(renderer);

        D3D_SetError("GetPresentParameters()", result);

        return NULL;

    }

    IDirect3DSwapChain9_Release(chain);

    if (pparams.PresentationInterval == D3DPRESENT_INTERVAL_ONE) {

        renderer->info.flags |= SDL_RENDERER_PRESENTVSYNC;

    }

    data->pparams = pparams;

    IDirect3DDevice9_GetDeviceCaps(data->device, &caps);

    renderer->info.max_texture_width = caps.MaxTextureWidth;

    renderer->info.max_texture_height = caps.MaxTextureHeight;

    /* Set up parameters for rendering */

    IDirect3DDevice9_SetVertexShader(data->device, NULL);

    IDirect3DDevice9_SetFVF(data->device,

                            D3DFVF_XYZ | D3DFVF_DIFFUSE | D3DFVF_TEX1);

    IDirect3DDevice9_SetRenderState(data->device, D3DRS_ZENABLE, D3DZB_FALSE);

    IDirect3DDevice9_SetRenderState(data->device, D3DRS_CULLMODE,

                                    D3DCULL_NONE);

    IDirect3DDevice9_SetRenderState(data->device, D3DRS_LIGHTING, FALSE);

    /* Enable color modulation by diffuse color */

    IDirect3DDevice9_SetTextureStageState(data->device, 0, D3DTSS_COLOROP,

                                          D3DTOP_MODULATE);

    IDirect3DDevice9_SetTextureStageState(data->device, 0, D3DTSS_COLORARG1,