/*

  Simple DirectMedia Layer

  Copyright (C) 1997-2013 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_D3D && !SDL_RENDER_DISABLED

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

#include "SDL_hints.h"

#include "SDL_loadso.h"

#include "SDL_syswm.h"

#include "../SDL_sysrender.h"

#include "stdio.h"

#if SDL_VIDEO_RENDER_D3D

#define D3D_DEBUG_INFO

#include <d3d9.h>

#endif

typedef interface ID3DXMatrixStack *LPD3DXMATRIXSTACK;

typedef struct _D3DMATRIX D3DXMATRIX, *LPD3DXMATRIX;

typedef struct _D3DVECTOR D3DXVECTOR3, *LPD3DXVECTOR3;

DEFINE_GUID(IID_ID3DXMatrixStack,

0xc7885ba7, 0xf990, 0x4fe7, 0x92, 0x2d, 0x85, 0x15, 0xe4, 0x77, 0xdd, 0x85);

#undef INTERFACE

#define INTERFACE ID3DXMatrixStack

DECLARE_INTERFACE_(ID3DXMatrixStack, IUnknown)

{

    STDMETHOD(QueryInterface)(THIS_ REFIID riid, LPVOID * ppvObj) PURE;

    STDMETHOD_(ULONG,AddRef)(THIS) PURE;

    STDMETHOD_(ULONG,Release)(THIS) PURE;

    STDMETHOD(Pop)(THIS) PURE;

    STDMETHOD(Push)(THIS) PURE;

    STDMETHOD(LoadIdentity)(THIS) PURE;

    STDMETHOD(LoadMatrix)(THIS_ CONST D3DXMATRIX* pM ) PURE;

    STDMETHOD(MultMatrix)(THIS_ CONST D3DXMATRIX* pM ) PURE;

    STDMETHOD(MultMatrixLocal)(THIS_ CONST D3DXMATRIX* pM ) PURE;

    STDMETHOD(RotateAxis)(THIS_ CONST D3DXVECTOR3* pV, FLOAT Angle) PURE;

    STDMETHOD(RotateAxisLocal)(THIS_ CONST D3DXVECTOR3* pV, FLOAT Angle) PURE;

    STDMETHOD(RotateYawPitchRoll)(THIS_ FLOAT Yaw, FLOAT Pitch, FLOAT Roll) PURE;

    STDMETHOD(RotateYawPitchRollLocal)(THIS_ FLOAT Yaw, FLOAT Pitch, FLOAT Roll) PURE;

    STDMETHOD(Scale)(THIS_ FLOAT x, FLOAT y, FLOAT z) PURE;

    STDMETHOD(ScaleLocal)(THIS_ FLOAT x, FLOAT y, FLOAT z) PURE;

    STDMETHOD(Translate)(THIS_ FLOAT x, FLOAT y, FLOAT z ) PURE;

    STDMETHOD(TranslateLocal)(THIS_ FLOAT x, FLOAT y, FLOAT z) PURE;

    STDMETHOD_(D3DXMATRIX*, GetTop)(THIS) PURE;

};

#undef INTERFACE

#if !defined(__cplusplus) || defined(CINTERFACE)

#define ID3DXMatrixStack_QueryInterface(p,a,b)            (p)->lpVtbl->QueryInterface(p,a,b)

#define ID3DXMatrixStack_AddRef(p)                        (p)->lpVtbl->AddRef(p)

#define ID3DXMatrixStack_Release(p)                       (p)->lpVtbl->Release(p)

#define ID3DXMatrixStack_Pop(p)                           (p)->lpVtbl->Pop(p)

#define ID3DXMatrixStack_Push(p)                          (p)->lpVtbl->Push(p)

#define ID3DXMatrixStack_LoadIdentity(p)                  (p)->lpVtbl->LoadIdentity(p)

#define ID3DXMatrixStack_LoadMatrix(p,a)                  (p)->lpVtbl->LoadMatrix(p,a)

#define ID3DXMatrixStack_MultMatrix(p,a)                  (p)->lpVtbl->MultMatrix(p,a)

#define ID3DXMatrixStack_MultMatrixLocal(p,a)             (p)->lpVtbl->MultMatrixLocal(p,a)

#define ID3DXMatrixStack_RotateAxis(p,a,b)                (p)->lpVtbl->RotateAxis(p,a,b)

#define ID3DXMatrixStack_RotateAxisLocal(p,a,b)           (p)->lpVtbl->RotateAxisLocal(p,a,b)

#define ID3DXMatrixStack_RotateYawPitchRoll(p,a,b,c)      (p)->lpVtbl->RotateYawPitchRoll(p,a,b,c)

#define ID3DXMatrixStack_RotateYawPitchRollLocal(p,a,b,c) (p)->lpVtbl->RotateYawPitchRollLocal(p,a,b,c)

#define ID3DXMatrixStack_Scale(p,a,b,c)                   (p)->lpVtbl->Scale(p,a,b,c)

#define ID3DXMatrixStack_ScaleLocal(p,a,b,c)              (p)->lpVtbl->ScaleLocal(p,a,b,c)

#define ID3DXMatrixStack_Translate(p,a,b,c)               (p)->lpVtbl->Translate(p,a,b,c)

#define ID3DXMatrixStack_TranslateLocal(p,a,b,c)          (p)->lpVtbl->TranslateLocal(p,a,b,c)

#define ID3DXMatrixStack_GetTop(p)                        (p)->lpVtbl->GetTop(p)

#else

#define ID3DXMatrixStack_QueryInterface(p,a,b)            (p)->QueryInterface(a,b)

#define ID3DXMatrixStack_AddRef(p)                        (p)->AddRef()

#define ID3DXMatrixStack_Release(p)                       (p)->Release()

#define ID3DXMatrixStack_Pop(p)    (p)->Pop()

#define ID3DXMatrixStack_Push(p)    (p)->Push()

#define ID3DXMatrixStack_LoadIdentity(p)    (p)->LoadIdentity()

#define ID3DXMatrixStack_LoadMatrix(p,a)    (p)->LoadMatrix(a)

#define ID3DXMatrixStack_MultMatrix(p,a)    (p)->MultMatrix(a)

#define ID3DXMatrixStack_MultMatrixLocal(p,a)    (p)->MultMatrixLocal(a)

#define ID3DXMatrixStack_RotateAxis(p,a,b)    (p)->RotateAxis(a,b)

#define ID3DXMatrixStack_RotateAxisLocal(p,a,b)    (p)->RotateAxisLocal(a,b)

#define ID3DXMatrixStack_RotateYawPitchRoll(p,a,b,c)    (p)->RotateYawPitchRollLocal(a,b,c)

#define ID3DXMatrixStack_Scale(p,a,b,c)    (p)->Scale(a,b,c)

#define ID3DXMatrixStack_ScaleLocal(p,a,b,c)    (p)->ScaleLocal(a,b,c)

#define ID3DXMatrixStack_Translate(p,a,b,c)    (p)->Translate(a,b,c)

#define ID3DXMatrixStack_TranslateLocal(p,a,b,c)    (p)->TranslateLocal(a,b,c)

#define ID3DXMatrixStack_GetTop(p)    (p)->GetTop()

#endif

#ifdef __cplusplus

extern "C" {

#endif

HRESULT WINAPI D3DXCreateMatrixStack(DWORD flags, LPD3DXMATRIXSTACK* ppstack);

#ifdef __cplusplus

}

#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_SetRenderTarget(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_FPoint * points, int count);

static int D3D_RenderDrawLines(SDL_Renderer * renderer,

                               const SDL_FPoint * points, int count);

static int D3D_RenderFillRects(SDL_Renderer * renderer,

                               const SDL_FRect * rects, int count);

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

                          const SDL_Rect * srcrect, const SDL_FRect * dstrect);

static int D3D_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 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 | SDL_RENDERER_TARGETTEXTURE),

     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;

    D3DTEXTUREFILTERTYPE scaleMode;

    IDirect3DSurface9 *defaultRenderTarget;

    IDirect3DSurface9 *currentRenderTarget;

    void* d3dxDLL;

    ID3DXMatrixStack *matrixStack;

} D3D_RenderData;

typedef struct

{

    IDirect3DTexture9 *texture;

    D3DTEXTUREFILTERTYPE scaleMode;

} 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;

    /* Release the default render target before reset */

    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);

    IDirect3DDevice9_GetRenderTarget(data->device, 0, &data->defaultRenderTarget);

    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;

    int d3dxVersion;

	char d3dxDLLFile[50];

    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;

        }

        for (d3dxVersion=50;d3dxVersion>0;d3dxVersion--) {

            SDL_snprintf(d3dxDLLFile, 49, "D3DX9_%02d.dll", d3dxVersion);

            data->d3dxDLL = SDL_LoadObject(d3dxDLLFile);

            if (data->d3dxDLL) {

                HRESULT (WINAPI *D3DXCreateMatrixStack) (DWORD Flags, LPD3DXMATRIXSTACK*  ppStack);

                D3DXCreateMatrixStack = (HRESULT (WINAPI *) (DWORD, LPD3DXMATRIXSTACK*)) SDL_LoadFunction(data->d3dxDLL, "D3DXCreateMatrixStack");

                if (D3DXCreateMatrixStack) {

                    D3DXCreateMatrixStack(0, &data->matrixStack);

                    break;

                }

            }

        }

        if (!data->matrixStack) {