/*

    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"

#include "SDL_win32video.h"

#include "SDL_syswm.h"

#include "SDL_vkeys.h"

#include "../../events/SDL_events_c.h"

/*#define WMMSG_DEBUG*/

#ifdef WMMSG_DEBUG

#include <stdio.h>

#include "wmmsg.h"

#endif

/* Masks for processing the windows KEYDOWN and KEYUP messages */

#define REPEATED_KEYMASK    (1<<30)

#define EXTENDED_KEYMASK    (1<<24)

#define VK_ENTER    10          /* Keypad Enter ... no VKEY defined? */

/* Make sure XBUTTON stuff is defined that isn't in older Platform SDKs... */

#ifndef WM_XBUTTONDOWN

#define WM_XBUTTONDOWN 0x020B

#endif

#ifndef WM_XBUTTONUP

#define WM_XBUTTONUP 0x020C

#endif

#ifndef GET_XBUTTON_WPARAM

#define GET_XBUTTON_WPARAM(w) (HIWORD(w))

#endif

static WPARAM

RemapVKEY(WPARAM wParam, LPARAM lParam)

{

    int i;

    BYTE scancode = (BYTE) ((lParam >> 16) & 0xFF);

    /* Windows remaps alphabetic keys based on current layout.

       We try to provide USB scancodes, so undo this mapping.

     */

    if (wParam >= 'A' && wParam <= 'Z') {

        if (scancode != alpha_scancodes[wParam - 'A']) {

            for (i = 0; i < SDL_arraysize(alpha_scancodes); ++i) {

                if (scancode == alpha_scancodes[i]) {

                    wParam = 'A' + i;

                    break;

                }

            }

        }

    }

    /* Keypad keys are a little trickier, we always scan for them. */

    for (i = 0; i < SDL_arraysize(keypad_scancodes); ++i) {

        if (scancode == keypad_scancodes[i]) {

            wParam = VK_NUMPAD0 + i;

            break;

        }

    }

    return wParam;

}

LRESULT CALLBACK

WIN_WindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)

{

    SDL_WindowData *data;

    /* Send a SDL_SYSWMEVENT if the application wants them */

    if (SDL_ProcessEvents[SDL_SYSWMEVENT] == SDL_ENABLE) {

        SDL_SysWMmsg wmmsg;

        SDL_VERSION(&wmmsg.version);

        wmmsg.hwnd = hwnd;

        wmmsg.msg = msg;

        wmmsg.wParam = wParam;

        wmmsg.lParam = lParam;

        SDL_SendSysWMEvent(&wmmsg);

    }

    /* Get the window data for the window */

    data = (SDL_WindowData *) GetProp(hwnd, TEXT("SDL_WindowData"));

    if (!data) {

        return CallWindowProc(DefWindowProc, hwnd, msg, wParam, lParam);

    }

#ifdef WMMSG_DEBUG

    {

        FILE *log = fopen("wmmsg.txt", "a");

        fprintf(log, "Received windows message: %p ", hwnd);

        if (msg > MAX_WMMSG) {

            fprintf(log, "%d", msg);

        } else {

            fprintf(log, "%s", wmtab[msg]);

        }

        fprintf(log, " -- 0x%X, 0x%X\n", wParam, lParam);

        fclose(log);

    }

#endif

    switch (msg) {

    case WM_SHOWWINDOW:

        {

            if (wParam) {

                SDL_SendWindowEvent(data->windowID, SDL_WINDOWEVENT_SHOWN, 0,

                                    0);

            } else {

                SDL_SendWindowEvent(data->windowID, SDL_WINDOWEVENT_HIDDEN, 0,

                                    0);

            }

        }

        break;

    case WM_ACTIVATE:

        {

            int index;

            SDL_Keyboard *keyboard;

            BOOL minimized;

            minimized = HIWORD(wParam);

            index = data->videodata->keyboard;

            keyboard = SDL_GetKeyboard(index);

            if (!minimized && (LOWORD(wParam) != WA_INACTIVE)) {

                SDL_SendWindowEvent(data->windowID, SDL_WINDOWEVENT_SHOWN,

                                    0, 0);

                SDL_SendWindowEvent(data->windowID,

                                    SDL_WINDOWEVENT_RESTORED, 0, 0);

                if (IsZoomed(hwnd)) {

                    SDL_SendWindowEvent(data->windowID,

                                        SDL_WINDOWEVENT_MAXIMIZED, 0, 0);

                }

                if (keyboard && keyboard->focus != data->windowID) {

                    SDL_SetKeyboardFocus(index, data->windowID);

                }

                /* FIXME: Update keyboard state */

            } else {

                if (keyboard && keyboard->focus == data->windowID) {

                    SDL_SetKeyboardFocus(index, 0);

                }

                if (minimized) {

                    SDL_SendWindowEvent(data->windowID,

                                        SDL_WINDOWEVENT_MINIMIZED, 0, 0);

                }

            }

            return (0);

        }

        break;

    case WM_MOUSEMOVE:

        {

            int index;

            SDL_Mouse *mouse;

            int x, y;

            index = data->videodata->mouse;

            mouse = SDL_GetMouse(index);

            if (mouse->focus != data->windowID) {

                TRACKMOUSEEVENT tme;

                tme.cbSize = sizeof(tme);

                tme.dwFlags = TME_LEAVE;

                tme.hwndTrack = hwnd;

                TrackMouseEvent(&tme);

                SDL_SetMouseFocus(index, data->windowID);

            }

            /* mouse has moved within the window */

            x = LOWORD(lParam);

            y = HIWORD(lParam);

            if (mouse->relative_mode) {

                int w, h;

                POINT center;

                SDL_GetWindowSize(data->windowID, &w, &h);

                center.x = (w / 2);

                center.y = (h / 2);

                x -= center.x;

                y -= center.y;

                if (x || y) {

                    ClientToScreen(hwnd, &center);

                    SetCursorPos(center.x, center.y);

                    SDL_SendMouseMotion(index, 1, x, y);

                }

            } else {

                SDL_SendMouseMotion(index, 0, x, y);

            }

        }

        return (0);

    case WM_MOUSELEAVE:

        {

            int index;

            SDL_Mouse *mouse;

            index = data->videodata->mouse;

            mouse = SDL_GetMouse(index);

            if (mouse->focus == data->windowID) {

                SDL_SetMouseFocus(index, 0);

            }

        }

        return (0);

    case WM_LBUTTONDOWN:

    case WM_LBUTTONUP:

    case WM_MBUTTONDOWN:

    case WM_MBUTTONUP:

    case WM_RBUTTONDOWN:

    case WM_RBUTTONUP:

    case WM_XBUTTONDOWN:

    case WM_XBUTTONUP:

        {

            int xbuttonval = 0;

            int index;

            SDL_Mouse *mouse;

            Uint8 button, state;

            /* DJM:

               We want the SDL window to take focus so that

               it acts like a normal windows "component"

               (e.g. gains keyboard focus on a mouse click).

             */

            SetFocus(hwnd);

            index = data->videodata->mouse;

            mouse = SDL_GetMouse(index);

            /* Figure out which button to use */

            switch (msg) {

            case WM_LBUTTONDOWN:

                button = SDL_BUTTON_LEFT;

                state = SDL_PRESSED;

                break;

            case WM_LBUTTONUP:

                button = SDL_BUTTON_LEFT;

                state = SDL_RELEASED;

                break;

            case WM_MBUTTONDOWN:

                button = SDL_BUTTON_MIDDLE;

                state = SDL_PRESSED;

                break;

            case WM_MBUTTONUP:

                button = SDL_BUTTON_MIDDLE;

                state = SDL_RELEASED;

                break;

            case WM_RBUTTONDOWN:

                button = SDL_BUTTON_RIGHT;

                state = SDL_PRESSED;

                break;

            case WM_RBUTTONUP:

                button = SDL_BUTTON_RIGHT;

                state = SDL_RELEASED;

                break;

            case WM_XBUTTONDOWN:

                xbuttonval = GET_XBUTTON_WPARAM(wParam);

                button = SDL_BUTTON_X1 + xbuttonval - 1;

                state = SDL_PRESSED;

                break;

            case WM_XBUTTONUP:

                xbuttonval = GET_XBUTTON_WPARAM(wParam);

                button = SDL_BUTTON_X1 + xbuttonval - 1;

                state = SDL_RELEASED;

                break;

            default:

                /* Eh? Unknown button? */

                return (0);

            }

            if (state == SDL_PRESSED) {

                /* Grab mouse so we get up events */

                if (++data->mouse_pressed > 0) {

                    SetCapture(hwnd);

                }

            } else {

                /* Release mouse after all up events */

                if (--data->mouse_pressed <= 0) {

                    ReleaseCapture();

                    data->mouse_pressed = 0;

                }

            }

            if (!mouse->relative_mode) {

                int x, y;

                x = LOWORD(lParam);

                y = HIWORD(lParam);

                SDL_SendMouseMotion(index, 0, x, y);

            }

            SDL_SendMouseButton(index, state, button);

            /*

             * MSDN says:

             *  "Unlike the WM_LBUTTONUP, WM_MBUTTONUP, and WM_RBUTTONUP

             *   messages, an application should return TRUE from [an

             *   XBUTTON message] if it processes it. Doing so will allow

             *   software that simulates this message on Microsoft Windows

             *   systems earlier than Windows 2000 to determine whether

             *   the window procedure processed the message or called

             *   DefWindowProc to process it.

             */

            if (xbuttonval > 0) {

                return (TRUE);

            }

        }

        return (0);

    case WM_MOUSEWHEEL:

        {

            int index;

            int motion = (short) HIWORD(wParam);

            index = data->videodata->mouse;

            SDL_SendMouseWheel(index, 0, motion);

        }

        return (0);

    case WM_SYSKEYDOWN:

    case WM_KEYDOWN:

        {

            int index;

            /* Ignore repeated keys */

            if (lParam & REPEATED_KEYMASK) {

                return (0);

            }

            index = data->videodata->keyboard;

            wParam = RemapVKEY(wParam, lParam);

            switch (wParam) {

            case VK_CONTROL:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_RCONTROL;

                else

                    wParam = VK_LCONTROL;

                break;

            case VK_SHIFT:

                /* EXTENDED trick doesn't work here */

                {

                    Uint8 *state = SDL_GetKeyboardState(NULL);

                    if (state[SDL_SCANCODE_LSHIFT] == SDL_RELEASED

                        && (GetKeyState(VK_LSHIFT) & 0x8000)) {

                        wParam = VK_LSHIFT;

                    } else if (state[SDL_SCANCODE_RSHIFT] == SDL_RELEASED

                               && (GetKeyState(VK_RSHIFT) & 0x8000)) {

                        wParam = VK_RSHIFT;

                    } else {

                        /* Probably a key repeat */

                        return (0);

                    }

                }

                break;

            case VK_MENU:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_RMENU;

                else

                    wParam = VK_LMENU;

                break;

            case VK_RETURN:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_ENTER;

                break;

            }

            if (wParam < 256) {

                SDL_SendKeyboardKey(index, SDL_PRESSED,

                                    data->videodata->key_layout[wParam]);

            }

        }

        return (0);

    case WM_SYSKEYUP:

    case WM_KEYUP:

        {

            int index;

            index = data->videodata->keyboard;

            wParam = RemapVKEY(wParam, lParam);

            switch (wParam) {

            case VK_CONTROL:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_RCONTROL;

                else

                    wParam = VK_LCONTROL;

                break;

            case VK_SHIFT:

                /* EXTENDED trick doesn't work here */

                {

                    Uint8 *state = SDL_GetKeyboardState(NULL);

                    if (state[SDL_SCANCODE_LSHIFT] == SDL_PRESSED

                        && !(GetKeyState(VK_LSHIFT) & 0x8000)) {

                        wParam = VK_LSHIFT;

                    } else if (state[SDL_SCANCODE_RSHIFT] == SDL_PRESSED

                               && !(GetKeyState(VK_RSHIFT) & 0x8000)) {

                        wParam = VK_RSHIFT;

                    } else {

                        /* Probably a key repeat */

                        return (0);

                    }

                }

                break;

            case VK_MENU:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_RMENU;

                else

                    wParam = VK_LMENU;

                break;

            case VK_RETURN:

                if (lParam & EXTENDED_KEYMASK)

                    wParam = VK_ENTER;

                break;

            }

            /* Windows only reports keyup for print screen */

            if (wParam == VK_SNAPSHOT

                && SDL_GetKeyboardState(NULL)[SDL_SCANCODE_PRINTSCREEN] ==

                SDL_RELEASED) {

                SDL_SendKeyboardKey(index, SDL_PRESSED,

                                    data->videodata->key_layout[wParam]);

            }

            if (wParam < 256) {

                SDL_SendKeyboardKey(index, SDL_RELEASED,

                                    data->videodata->key_layout[wParam]);

            }

        }

        return (0);

    case WM_CHAR:

        {

            char text[4];

            /* Convert to UTF-8 and send it on... */

            if (wParam <= 0x7F) {

                text[0] = (char) wParam;

                text[1] = '\0';

            } else if (wParam <= 0x7FF) {

                text[0] = 0xC0 | (char) ((wParam >> 6) & 0x1F);

                text[1] = 0x80 | (char) (wParam & 0x3F);

                text[2] = '\0';

            } else {

                text[0] = 0xE0 | (char) ((wParam >> 12) & 0x0F);

                text[1] = 0x80 | (char) ((wParam >> 6) & 0x3F);

                text[2] = 0x80 | (char) (wParam & 0x3F);

                text[3] = '\0';

            }

            SDL_SendKeyboardText(data->videodata->keyboard, text);

        }

        return (0);

    case WM_INPUTLANGCHANGE:

        {

            WIN_UpdateKeymap(data->videodata->keyboard);

        }

        return (1);

    case WM_GETMINMAXINFO:

        {

            MINMAXINFO *info;

            RECT size;

            int x, y;

            int w, h;

            int style;

            /* If we allow resizing, let the resize happen naturally */

            if (SDL_GetWindowFlags(data->windowID) & SDL_WINDOW_RESIZABLE) {

                return (0);

            }

            /* Get the current position of our window */

            GetWindowRect(hwnd, &size);

            x = size.left;

            y = size.top;

            /* Calculate current size of our window */

            SDL_GetWindowSize(data->windowID, &w, &h);

            size.top = 0;

            size.left = 0;

            size.bottom = h;

            size.right = w;

            /* DJM - according to the docs for GetMenu(), the

               return value is undefined if hwnd is a child window.

               Aparently it's too difficult for MS to check

               inside their function, so I have to do it here.

             */

            style = GetWindowLong(hwnd, GWL_STYLE);

            AdjustWindowRect(&size,

                             style,

                             style & WS_CHILDWINDOW ? FALSE : GetMenu(hwnd) !=

                             NULL);

            w = size.right - size.left;

            h = size.bottom - size.top;

            /* Fix our size to the current size */

            info = (MINMAXINFO *) lParam;

            info->ptMaxSize.x = w;

            info->ptMaxSize.y = h;