/*

    SDL - Simple DirectMedia Layer

    Copyright (C) 1997-2009 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

#include "SDL_win32video.h"

#include "../SDL_yuv_sw_c.h"

/* Direct3D renderer implementation */

#if 1                           /* This takes more memory but you won't lose your texture data */

#define D3DPOOL_SDL    D3DPOOL_MANAGED

#define SDL_MEMORY_POOL_MANAGED

#else

#define D3DPOOL_SDL    D3DPOOL_DEFAULT

#define SDL_MEMORY_POOL_DEFAULT

#endif

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

static int D3D_DisplayModeChanged(SDL_Renderer * renderer);

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

static int D3D_QueryTexturePixels(SDL_Renderer * renderer,

                                  SDL_Texture * texture, void **pixels,

                                  int *pitch);

static int D3D_SetTexturePalette(SDL_Renderer * renderer,

                                 SDL_Texture * texture,

                                 const SDL_Color * colors, int firstcolor,

                                 int ncolors);

static int D3D_GetTexturePalette(SDL_Renderer * renderer,

                                 SDL_Texture * texture, SDL_Color * colors,

                                 int firstcolor, int ncolors);

static int D3D_SetTextureColorMod(SDL_Renderer * renderer,

                                  SDL_Texture * texture);

static int D3D_SetTextureAlphaMod(SDL_Renderer * renderer,

                                  SDL_Texture * texture);

static int D3D_SetTextureBlendMode(SDL_Renderer * renderer,

                                   SDL_Texture * texture);

static int D3D_SetTextureScaleMode(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, int markDirty,

                           void **pixels, int *pitch);

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

static void D3D_DirtyTexture(SDL_Renderer * renderer, SDL_Texture * texture,

                             int numrects, const SDL_Rect * rects);

static int D3D_RenderPoints(SDL_Renderer * renderer, const SDL_Point * points,

                            int count);

static int D3D_RenderLines(SDL_Renderer * renderer, const SDL_Point * points,

                           int count);

static int D3D_RenderRects(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 int D3D_RenderWritePixels(SDL_Renderer * renderer, const SDL_Rect * rect,

                                 Uint32 format, const 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,

    {

     "d3d",

     (SDL_RENDERER_SINGLEBUFFER | SDL_RENDERER_PRESENTCOPY |

      SDL_RENDERER_PRESENTFLIP2 | SDL_RENDERER_PRESENTFLIP3 |

      SDL_RENDERER_PRESENTDISCARD | SDL_RENDERER_PRESENTVSYNC |

      SDL_RENDERER_ACCELERATED),

     (SDL_TEXTUREMODULATE_NONE | SDL_TEXTUREMODULATE_COLOR |

      SDL_TEXTUREMODULATE_ALPHA),

     (SDL_BLENDMODE_NONE | SDL_BLENDMODE_MASK |

      SDL_BLENDMODE_BLEND | SDL_BLENDMODE_ADD | SDL_BLENDMODE_MOD),

     (SDL_TEXTURESCALEMODE_NONE | SDL_TEXTURESCALEMODE_FAST |

      SDL_TEXTURESCALEMODE_SLOW | SDL_TEXTURESCALEMODE_BEST),

     0,

     {0},

     0,

     0}

};

typedef struct

{

    IDirect3D9 *d3d;

    IDirect3DDevice9 *device;

    UINT adapter;

    D3DPRESENT_PARAMETERS pparams;

    SDL_bool beginScene;

} D3D_RenderData;

typedef struct

{

    SDL_SW_YUVTexture *yuv;

    Uint32 format;

    IDirect3DTexture9 *texture;

} D3D_TextureData;

typedef struct

{

    float x, y, z;

    float rhw;

    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_INDEX8:

        return D3DFMT_P8;

    case SDL_PIXELFORMAT_RGB332:

        return D3DFMT_R3G3B2;

    case SDL_PIXELFORMAT_RGB444:

        return D3DFMT_X4R4G4B4;

    case SDL_PIXELFORMAT_RGB555:

        return D3DFMT_X1R5G5B5;

    case SDL_PIXELFORMAT_ARGB4444:

        return D3DFMT_A4R4G4B4;

    case SDL_PIXELFORMAT_ARGB1555:

        return D3DFMT_A1R5G5B5;

    case SDL_PIXELFORMAT_RGB565:

        return D3DFMT_R5G6B5;

    case SDL_PIXELFORMAT_RGB888:

        return D3DFMT_X8R8G8B8;

    case SDL_PIXELFORMAT_ARGB8888:

        return D3DFMT_A8R8G8B8;

    case SDL_PIXELFORMAT_ARGB2101010:

        return D3DFMT_A2R10G10B10;

    case SDL_PIXELFORMAT_YV12:

        return MAKEFOURCC('Y','V','1','2');

    case SDL_PIXELFORMAT_IYUV:

        return MAKEFOURCC('I','4','2','0');

    case SDL_PIXELFORMAT_UYVY:

        return D3DFMT_UYVY;

    case SDL_PIXELFORMAT_YUY2:

        return D3DFMT_YUY2;

    default:

        return D3DFMT_UNKNOWN;

    }

}

static UINT D3D_FindAdapter(IDirect3D9 * d3d, SDL_VideoDisplay * display)

{

    SDL_DisplayData *displaydata = (SDL_DisplayData *) display->driverdata;

    UINT adapter, count;

    count = IDirect3D9_GetAdapterCount(d3d);

    for (adapter = 0; adapter < count; ++adapter) {

        HRESULT result;

        D3DADAPTER_IDENTIFIER9 info;

        char *name;

        result = IDirect3D9_GetAdapterIdentifier(d3d, adapter, 0, &info);

        if (FAILED(result)) {

            continue;

        }

        name = WIN_StringToUTF8(displaydata->DeviceName);

        if (SDL_strcmp(name, info.DeviceName) == 0) {

            SDL_free(name);

            return adapter;

        }

        SDL_free(name);

    }

    /* This should never happen, but just in case... */

    return D3DADAPTER_DEFAULT;

}

static SDL_bool

D3D_IsTextureFormatAvailable(IDirect3D9 * d3d, UINT adapter,

                             Uint32 display_format,

                             Uint32 texture_format)

{

    HRESULT result;

    result = IDirect3D9_CheckDeviceFormat(d3d, adapter,

                                          D3DDEVTYPE_HAL,

                                          PixelFormatToD3DFMT(display_format),

                                          0,

                                          D3DRTYPE_TEXTURE,

                                          PixelFormatToD3DFMT

                                          (texture_format));

    return FAILED(result) ? SDL_FALSE : SDL_TRUE;

}

static void

UpdateYUVTextureData(SDL_Texture * texture)

{

    D3D_TextureData *data = (D3D_TextureData *) texture->driverdata;

    SDL_Rect rect;

    RECT d3drect;

    D3DLOCKED_RECT locked;

    HRESULT result;

    d3drect.left = 0;

    d3drect.right = texture->w;

    d3drect.top = 0;

    d3drect.bottom = texture->h;

    result =

        IDirect3DTexture9_LockRect(data->texture, 0, &locked, &d3drect, 0);

    if (FAILED(result)) {

        return;

    }

    rect.x = 0;

    rect.y = 0;

    rect.w = texture->w;

    rect.h = texture->h;

    SDL_SW_CopyYUVToRGB(data->yuv, &rect, data->format, texture->w,

                        texture->h, locked.pBits, locked.Pitch);

    IDirect3DTexture9_UnlockRect(data->texture, 0);

}

void

D3D_AddRenderDriver(_THIS)

{

    SDL_VideoData *data = (SDL_VideoData *) _this->driverdata;

    SDL_RendererInfo *info = &D3D_RenderDriver.info;

    if (data->d3d) {

        int i, j;

        int formats[] = {

            SDL_PIXELFORMAT_INDEX8,

            SDL_PIXELFORMAT_RGB332,

            SDL_PIXELFORMAT_RGB444,

            SDL_PIXELFORMAT_RGB555,

            SDL_PIXELFORMAT_ARGB4444,

            SDL_PIXELFORMAT_ARGB1555,

            SDL_PIXELFORMAT_RGB565,

            SDL_PIXELFORMAT_RGB888,

            SDL_PIXELFORMAT_ARGB8888,

            SDL_PIXELFORMAT_ARGB2101010,

        };

        for (i = 0; i < _this->num_displays; ++i) {

            SDL_VideoDisplay *display = &_this->displays[i];

            SDL_DisplayMode *mode = &display->desktop_mode;

            UINT adapter = D3D_FindAdapter(data->d3d, display);

            /* Get the matching D3D adapter for this display */

            info->num_texture_formats = 0;

            for (j = 0; j < SDL_arraysize(formats); ++j) {

                if (D3D_IsTextureFormatAvailable

                    (data->d3d, adapter, mode->format, formats[j])) {

                    info->texture_formats[info->num_texture_formats++] =

                        formats[j];

                }

            }

            info->texture_formats[info->num_texture_formats++] =

                SDL_PIXELFORMAT_YV12;

            info->texture_formats[info->num_texture_formats++] =

                SDL_PIXELFORMAT_IYUV;

            info->texture_formats[info->num_texture_formats++] =

                SDL_PIXELFORMAT_YUY2;

            info->texture_formats[info->num_texture_formats++] =

                SDL_PIXELFORMAT_UYVY;

            info->texture_formats[info->num_texture_formats++] =

                SDL_PIXELFORMAT_YVYU;

            SDL_AddRenderDriver(display, &D3D_RenderDriver);

        }

    }

}

SDL_Renderer *

D3D_CreateRenderer(SDL_Window * window, Uint32 flags)

{

    SDL_VideoDisplay *display = SDL_GetDisplayFromWindow(window);

    SDL_VideoData *videodata = (SDL_VideoData *) display->device->driverdata;

    SDL_WindowData *windowdata = (SDL_WindowData *) window->driverdata;

    SDL_Renderer *renderer;

    D3D_RenderData *data;

    HRESULT result;

    D3DPRESENT_PARAMETERS pparams;

    IDirect3DSwapChain9 *chain;

    D3DCAPS9 caps;

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

    if (!renderer) {

        SDL_OutOfMemory();

        return NULL;

    }

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

    if (!data) {

        D3D_DestroyRenderer(renderer);

        SDL_OutOfMemory();

        return NULL;

    }

    data->d3d = videodata->d3d;

    renderer->DisplayModeChanged = D3D_DisplayModeChanged;

    renderer->CreateTexture = D3D_CreateTexture;

    renderer->QueryTexturePixels = D3D_QueryTexturePixels;

    renderer->SetTexturePalette = D3D_SetTexturePalette;

    renderer->GetTexturePalette = D3D_GetTexturePalette;

    renderer->SetTextureColorMod = D3D_SetTextureColorMod;

    renderer->SetTextureAlphaMod = D3D_SetTextureAlphaMod;

    renderer->SetTextureBlendMode = D3D_SetTextureBlendMode;

    renderer->SetTextureScaleMode = D3D_SetTextureScaleMode;

    renderer->UpdateTexture = D3D_UpdateTexture;

    renderer->LockTexture = D3D_LockTexture;

    renderer->UnlockTexture = D3D_UnlockTexture;

    renderer->DirtyTexture = D3D_DirtyTexture;

    renderer->RenderPoints = D3D_RenderPoints;

    renderer->RenderLines = D3D_RenderLines;

    renderer->RenderRects = D3D_RenderRects;

    renderer->RenderCopy = D3D_RenderCopy;

    renderer->RenderReadPixels = D3D_RenderReadPixels;

    renderer->RenderWritePixels = D3D_RenderWritePixels;

    renderer->RenderPresent = D3D_RenderPresent;

    renderer->DestroyTexture = D3D_DestroyTexture;

    renderer->DestroyRenderer = D3D_DestroyRenderer;

    renderer->info = D3D_RenderDriver.info;

    renderer->window = window->id;

    renderer->driverdata = data;

    renderer->info.flags = SDL_RENDERER_ACCELERATED;

    SDL_zero(pparams);

    pparams.BackBufferWidth = window->w;

    pparams.BackBufferHeight = window->h;

    if (window->flags & SDL_WINDOW_FULLSCREEN) {

        pparams.BackBufferFormat =

            PixelFormatToD3DFMT(window->fullscreen_mode.format);

    } else {

        pparams.BackBufferFormat = D3DFMT_UNKNOWN;

    }

    if (flags & SDL_RENDERER_PRESENTFLIP2) {

        pparams.BackBufferCount = 2;

        pparams.SwapEffect = D3DSWAPEFFECT_FLIP;

    } else if (flags & SDL_RENDERER_PRESENTFLIP3) {

        pparams.BackBufferCount = 3;

        pparams.SwapEffect = D3DSWAPEFFECT_FLIP;

    } else if (flags & SDL_RENDERER_PRESENTCOPY) {

        pparams.BackBufferCount = 1;

        pparams.SwapEffect = D3DSWAPEFFECT_COPY;

    } else {

        pparams.BackBufferCount = 1;

        pparams.SwapEffect = D3DSWAPEFFECT_DISCARD;

    }

    if (window->flags & SDL_WINDOW_FULLSCREEN) {

        pparams.Windowed = FALSE;

        pparams.FullScreen_RefreshRateInHz =

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

    }

    data->adapter = D3D_FindAdapter(videodata->d3d, display);

    IDirect3D9_GetDeviceCaps(videodata->d3d, data->adapter,

                             D3DDEVTYPE_HAL, &caps);

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

                                     D3DDEVTYPE_HAL,

                                     windowdata->hwnd,

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