/*

    SDL - Simple DirectMedia Layer

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

#include "SDL_win32video.h"

#include "../SDL_rect_c.h"

#include "../SDL_yuv_sw_c.h"

/* GDI renderer implementation */

static SDL_Renderer *SDL_GDI_CreateRenderer(SDL_Window * window,

                                            Uint32 flags);

static int SDL_GDI_CreateTexture(SDL_Renderer * renderer,

                                 SDL_Texture * texture);

static int SDL_GDI_QueryTexturePixels(SDL_Renderer * renderer,

                                      SDL_Texture * texture, void **pixels,

                                      int *pitch);

static int SDL_GDI_SetTexturePalette(SDL_Renderer * renderer,

                                     SDL_Texture * texture,

                                     const SDL_Color * colors, int firstcolor,

                                     int ncolors);

static int SDL_GDI_GetTexturePalette(SDL_Renderer * renderer,

                                     SDL_Texture * texture,

                                     SDL_Color * colors, int firstcolor,

                                     int ncolors);

static int SDL_GDI_UpdateTexture(SDL_Renderer * renderer,

                                 SDL_Texture * texture, const SDL_Rect * rect,

                                 const void *pixels, int pitch);

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

                               const SDL_Rect * rect, int markDirty,

                               void **pixels, int *pitch);

static void SDL_GDI_UnlockTexture(SDL_Renderer * renderer,

                                  SDL_Texture * texture);

static void SDL_GDI_DirtyTexture(SDL_Renderer * renderer,

                                 SDL_Texture * texture, int numrects,

                                 const SDL_Rect * rects);

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

                              Uint32 color);

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

                              const SDL_Rect * srcrect,

                              const SDL_Rect * dstrect, int blendMode,

                              int scaleMode);

static void SDL_GDI_RenderPresent(SDL_Renderer * renderer);

static void SDL_GDI_DestroyTexture(SDL_Renderer * renderer,

                                   SDL_Texture * texture);

static void SDL_GDI_DestroyRenderer(SDL_Renderer * renderer);

SDL_RenderDriver SDL_GDI_RenderDriver = {

    SDL_GDI_CreateRenderer,

    {

     "gdi",

     (SDL_Renderer_SingleBuffer | SDL_Renderer_PresentCopy |

      SDL_Renderer_PresentFlip2 | SDL_Renderer_PresentFlip3 |

     (SDL_TextureBlendMode_None | SDL_TextureBlendMode_Mask |

      SDL_TextureBlendMode_Blend),

     (SDL_TextureScaleMode_None | SDL_TextureScaleMode_Fast),

     11,

     {

      SDL_PixelFormat_Index8,

      SDL_PixelFormat_RGB555,

      SDL_PixelFormat_RGB565,

      SDL_PixelFormat_RGB888,

      SDL_PixelFormat_BGR888,

      SDL_PixelFormat_ARGB8888,

      SDL_PixelFormat_RGBA8888,

      SDL_PixelFormat_ABGR8888,

      SDL_PixelFormat_BGRA8888,

      SDL_PixelFormat_YUY2,

      SDL_PixelFormat_UYVY},

     0,

     0}

};

typedef struct

{

    HWND hwnd;

    HDC window_hdc;

    HDC render_hdc;

    HDC memory_hdc;

    HDC current_hdc;

    LPBITMAPINFO bmi;

    HBITMAP hbm[3];

    int current_hbm;

    SDL_DirtyRectList dirty;

    SDL_bool makedirty;

} SDL_GDI_RenderData;

typedef struct

{

    SDL_SW_YUVTexture *yuv;

    Uint32 format;

    HPALETTE hpal;

    HBITMAP hbm;

    void *pixels;

    int pitch;

} SDL_GDI_TextureData;

static void

UpdateYUVTextureData(SDL_Texture * texture)

{

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

    SDL_Rect rect;

    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, data->pixels, data->pitch);

}

void

GDI_AddRenderDriver(_THIS)

{

    SDL_AddRenderDriver(0, &SDL_GDI_RenderDriver);

}

SDL_Renderer *

SDL_GDI_CreateRenderer(SDL_Window * window, Uint32 flags)

{

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

    SDL_Renderer *renderer;

    SDL_GDI_RenderData *data;

    int bmi_size;

    HBITMAP hbm;

    int i, n;

    renderer = (SDL_Renderer *) SDL_malloc(sizeof(*renderer));

    if (!renderer) {

        SDL_OutOfMemory();

        return NULL;

    }

    SDL_zerop(renderer);

    data = (SDL_GDI_RenderData *) SDL_malloc(sizeof(*data));

    if (!data) {

        SDL_GDI_DestroyRenderer(renderer);

        SDL_OutOfMemory();

        return NULL;

    }

    SDL_zerop(data);

    renderer->CreateTexture = SDL_GDI_CreateTexture;

    renderer->QueryTexturePixels = SDL_GDI_QueryTexturePixels;

    renderer->SetTexturePalette = SDL_GDI_SetTexturePalette;

    renderer->GetTexturePalette = SDL_GDI_GetTexturePalette;

    renderer->UpdateTexture = SDL_GDI_UpdateTexture;

    renderer->LockTexture = SDL_GDI_LockTexture;

    renderer->UnlockTexture = SDL_GDI_UnlockTexture;

    renderer->DirtyTexture = SDL_GDI_DirtyTexture;

    renderer->RenderFill = SDL_GDI_RenderFill;

    renderer->RenderCopy = SDL_GDI_RenderCopy;

    renderer->RenderPresent = SDL_GDI_RenderPresent;

    renderer->DestroyTexture = SDL_GDI_DestroyTexture;

    renderer->DestroyRenderer = SDL_GDI_DestroyRenderer;

    renderer->info = SDL_GDI_RenderDriver.info;

    renderer->window = window->id;

    renderer->driverdata = data;

    data->hwnd = windowdata->hwnd;

    data->window_hdc = GetDC(data->hwnd);

    data->render_hdc = CreateCompatibleDC(data->window_hdc);

    data->memory_hdc = CreateCompatibleDC(data->window_hdc);

    /* Fill in the compatible bitmap info */

    bmi_size = sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD);

    data->bmi = (LPBITMAPINFO) SDL_malloc(bmi_size);

    if (!data->bmi) {

        SDL_GDI_DestroyRenderer(renderer);

        SDL_OutOfMemory();

        return NULL;

    }

    SDL_memset(data->bmi, 0, bmi_size);

    data->bmi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

    hbm = CreateCompatibleBitmap(data->window_hdc, 1, 1);

    GetDIBits(data->window_hdc, hbm, 0, 1, NULL, data->bmi, DIB_RGB_COLORS);

    GetDIBits(data->window_hdc, hbm, 0, 1, NULL, data->bmi, DIB_RGB_COLORS);

    DeleteObject(hbm);

    if (flags & SDL_Renderer_SingleBuffer) {

        n = 0;

    } else if (flags & SDL_Renderer_PresentFlip2) {

        renderer->info.flags |= SDL_Renderer_PresentFlip2;

        n = 2;

    } else if (flags & SDL_Renderer_PresentFlip3) {

        renderer->info.flags |= SDL_Renderer_PresentFlip3;

        n = 3;

    } else {

        renderer->info.flags |= SDL_Renderer_PresentCopy;

        n = 1;

    }

    for (i = 0; i < n; ++i) {

        data->hbm[i] =

            CreateCompatibleBitmap(data->window_hdc, window->w, window->h);

        if (!data->hbm[i]) {

            SDL_GDI_DestroyRenderer(renderer);

            WIN_SetError("CreateCompatibleBitmap()");

            return NULL;

        }

    }

    if (n > 0) {

        SelectObject(data->render_hdc, data->hbm[0]);

        data->current_hdc = data->render_hdc;

        data->makedirty = SDL_TRUE;

    } else {

        data->current_hdc = data->window_hdc;

        data->makedirty = SDL_FALSE;

    }

    data->current_hbm = 0;

    return renderer;

}

static int

SDL_GDI_CreateTexture(SDL_Renderer * renderer, SDL_Texture * texture)

{

    SDL_GDI_RenderData *renderdata =

        (SDL_GDI_RenderData *) renderer->driverdata;

    SDL_Window *window = SDL_GetWindowFromID(renderer->window);

    SDL_VideoDisplay *display = SDL_GetDisplayFromWindow(window);

    SDL_GDI_TextureData *data;

    data = (SDL_GDI_TextureData *) SDL_malloc(sizeof(*data));

    if (!data) {

        SDL_OutOfMemory();

        return -1;

    }

    SDL_zerop(data);

    texture->driverdata = data;

    if (SDL_ISPIXELFORMAT_FOURCC(texture->format)) {

        data->yuv = SDL_SW_CreateYUVTexture(texture);

        if (!data->yuv) {

            SDL_GDI_DestroyTexture(renderer, texture);

            return -1;

        }

        data->format = display->current_mode.format;

    } else {

        data->format = texture->format;

    }

    data->pitch = (texture->w * SDL_BYTESPERPIXEL(data->format));

    if (data->yuv || texture->access == SDL_TextureAccess_Local

        || texture->format != SDL_GetCurrentDisplayMode()->format) {

        int bmi_size;

        LPBITMAPINFO bmi;

        bmi_size = sizeof(BITMAPINFOHEADER) + 256 * sizeof(RGBQUAD);

        bmi = (LPBITMAPINFO) SDL_malloc(bmi_size);

        if (!bmi) {

            SDL_GDI_DestroyTexture(renderer, texture);

            SDL_OutOfMemory();

            return -1;

        }

        SDL_memset(bmi, 0, bmi_size);

        bmi->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);

        bmi->bmiHeader.biWidth = texture->w;

        bmi->bmiHeader.biHeight = -texture->h;  /* topdown bitmap */

        bmi->bmiHeader.biPlanes = 1;

        bmi->bmiHeader.biSizeImage = texture->h * data->pitch;

        bmi->bmiHeader.biXPelsPerMeter = 0;

        bmi->bmiHeader.biYPelsPerMeter = 0;

        bmi->bmiHeader.biClrUsed = 0;

        bmi->bmiHeader.biClrImportant = 0;

        bmi->bmiHeader.biBitCount = SDL_BYTESPERPIXEL(data->format) * 8;

        if (SDL_ISPIXELFORMAT_INDEXED(data->format)) {

            int i, ncolors;

            LOGPALETTE *palette;

            bmi->bmiHeader.biCompression = BI_RGB;

            ncolors = (1 << SDL_BITSPERPIXEL(data->format));

            palette =

                (LOGPALETTE *) SDL_malloc(sizeof(*palette) +

                                          ncolors * sizeof(PALETTEENTRY));

            if (!palette) {

                SDL_free(bmi);

                SDL_GDI_DestroyTexture(renderer, texture);

                SDL_OutOfMemory();

                return -1;

            }

            palette->palVersion = 0x300;

            palette->palNumEntries = ncolors;

            for (i = 0; i < ncolors; ++i) {

                palette->palPalEntry[i].peRed = 0xFF;

                palette->palPalEntry[i].peGreen = 0xFF;

                palette->palPalEntry[i].peBlue = 0xFF;

                palette->palPalEntry[i].peFlags = 0;

            }

            data->hpal = CreatePalette(palette);

            SDL_free(palette);

        } else {

            int bpp;

            Uint32 Rmask, Gmask, Bmask, Amask;

            bmi->bmiHeader.biCompression = BI_BITFIELDS;

            SDL_PixelFormatEnumToMasks(data->format, &bpp, &Rmask, &Gmask,

                                       &Bmask, &Amask);

            ((Uint32 *) bmi->bmiColors)[0] = Rmask;

            ((Uint32 *) bmi->bmiColors)[1] = Gmask;

            ((Uint32 *) bmi->bmiColors)[2] = Bmask;

            data->hpal = NULL;

        }

        data->hbm =

            CreateDIBSection(renderdata->memory_hdc, bmi, DIB_RGB_COLORS,

                             &data->pixels, NULL, 0);

    } else {

        data->hbm =

            CreateCompatibleBitmap(renderdata->window_hdc, texture->w,

                                   texture->h);

        data->pixels = NULL;

    }

    if (!data->hbm) {

        SDL_GDI_DestroyTexture(renderer, texture);

        WIN_SetError("Couldn't create bitmap");

        return -1;

    }

    return 0;

}

static int

SDL_GDI_QueryTexturePixels(SDL_Renderer * renderer, SDL_Texture * texture,

                           void **pixels, int *pitch)

{

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

    if (data->yuv) {

        return SDL_SW_QueryYUVTexturePixels(data->yuv, pixels, pitch);

    } else {

        *pixels = data->pixels;

        *pitch = data->pitch;

        return 0;

    }

}

static int

SDL_GDI_SetTexturePalette(SDL_Renderer * renderer, SDL_Texture * texture,

                          const SDL_Color * colors, int firstcolor,

                          int ncolors)

{

    SDL_GDI_RenderData *renderdata =

        (SDL_GDI_RenderData *) renderer->driverdata;

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

    if (data->yuv) {

        SDL_SetError("YUV textures don't have a palette");

        return -1;

    } else {

        PALETTEENTRY entries[256];

        int i;

        for (i = 0; i < ncolors; ++i) {

            entries[i].peRed = colors[i].r;

            entries[i].peGreen = colors[i].g;

            entries[i].peBlue = colors[i].b;

            entries[i].peFlags = 0;

        }

        if (!SetPaletteEntries(data->hpal, firstcolor, ncolors, entries)) {

            WIN_SetError("SetPaletteEntries()");

            return -1;

        }

        return 0;

    }

}

static int

SDL_GDI_GetTexturePalette(SDL_Renderer * renderer, SDL_Texture * texture,

                          SDL_Color * colors, int firstcolor, int ncolors)

{

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

    if (data->yuv) {

        SDL_SetError("YUV textures don't have a palette");

        return -1;

    } else {

        PALETTEENTRY entries[256];

        int i;

        if (!GetPaletteEntries(data->hpal, firstcolor, ncolors, entries)) {

            WIN_SetError("GetPaletteEntries()");

            return -1;

        }

        for (i = 0; i < ncolors; ++i) {

            colors[i].r = entries[i].peRed;

            colors[i].g = entries[i].peGreen;

            colors[i].b = entries[i].peBlue;

        }

        return 0;

    }

}

static int

SDL_GDI_UpdateTexture(SDL_Renderer * renderer, SDL_Texture * texture,

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

{

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

    if (data->yuv) {

        if (SDL_SW_UpdateYUVTexture(data->yuv, rect, pixels, pitch) < 0) {

            return -1;

        }

        UpdateYUVTextureData(texture);

        return 0;

    } else {

        SDL_GDI_RenderData *renderdata =

            (SDL_GDI_RenderData *) renderer->driverdata;

        if (data->pixels) {

            Uint8 *src, *dst;

            int row;

            size_t length;

            src = (Uint8 *) pixels;

            dst =

                (Uint8 *) data->pixels + rect->y * data->pitch +

                rect->x * SDL_BYTESPERPIXEL(texture->format);

            length = rect->w * SDL_BYTESPERPIXEL(texture->format);

            for (row = 0; row < rect->h; ++row) {

                SDL_memcpy(dst, src, length);

                src += pitch;

                dst += data->pitch;

            }

        } else if (rect->w == texture->w && pitch == data->pitch) {

            if (!SetDIBits

                (renderdata->window_hdc, data->hbm, rect->y, rect->h, pixels,

                 renderdata->bmi, DIB_RGB_COLORS)) {

                WIN_SetError("SetDIBits()");

                return -1;

            }

        } else {

            SDL_SetError

                ("FIXME: Need to allocate temporary memory and do GetDIBits() followed by SetDIBits(), since we can only set blocks of scanlines at a time");

            return -1;

        }

        return 0;

    }

}

static int

SDL_GDI_LockTexture(SDL_Renderer * renderer, SDL_Texture * texture,

                    const SDL_Rect * rect, int markDirty, void **pixels,

                    int *pitch)

{

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

    if (data->yuv) {

        return SDL_SW_LockYUVTexture(data->yuv, rect, markDirty, pixels,

                                     pitch);

    } else if (data->pixels) {

        GdiFlush();

        *pixels =

            (void *) ((Uint8 *) data->pixels + rect->y * data->pitch +

                      rect->x * SDL_BYTESPERPIXEL(texture->format));

        *pitch = data->pitch;

        return 0;

    } else {

        SDL_SetError("No pixels available");

        return -1;

    }

}

static void

SDL_GDI_UnlockTexture(SDL_Renderer * renderer, SDL_Texture * texture)

{

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

    if (data->yuv) {

        SDL_SW_UnlockYUVTexture(data->yuv);

        UpdateYUVTextureData(texture);

    }

}

static void

SDL_GDI_DirtyTexture(SDL_Renderer * renderer, SDL_Texture * texture,

                     int numrects, const SDL_Rect * rects)

{

}

static int

SDL_GDI_RenderFill(SDL_Renderer * renderer, const SDL_Rect * rect,

                   Uint32 color)

{

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

    Uint8 r, g, b;

    RECT rc;

    HBRUSH brush;

    int status;

    if (data->makedirty) {

        SDL_AddDirtyRect(&data->dirty, rect);

    }

    r = (Uint8) ((color >> 16) & 0xFF);

    g = (Uint8) ((color >> 8) & 0xFF);