/*

    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"

#include "SDL_syswm.h"

#include "../SDL_sysvideo.h"

#include "../../events/SDL_keyboard_c.h"

#include "../../events/SDL_mouse_c.h"

#include "../../events/SDL_touch_c.h"

#include "../../events/SDL_windowevents_c.h"

#include "SDL_cocoavideo.h"

#include "SDL_cocoashape.h"

#include "SDL_cocoamouse.h"

static __inline__ void ConvertNSRect(NSRect *r)

{

    r->origin.y = CGDisplayPixelsHigh(kCGDirectMainDisplay) - r->origin.y - r->size.height;

}

@implementation Cocoa_WindowListener

- (void)listen:(SDL_WindowData *)data

{

    NSNotificationCenter *center;

    NSWindow *window = data->nswindow;

    NSView *view = [window contentView];

    _data = data;

    center = [NSNotificationCenter defaultCenter];

    if ([window delegate] != nil) {

        [center addObserver:self selector:@selector(windowDidExpose:) name:NSWindowDidExposeNotification object:window];

        [center addObserver:self selector:@selector(windowDidMove:) name:NSWindowDidMoveNotification object:window];

        [center addObserver:self selector:@selector(windowDidResize:) name:NSWindowDidResizeNotification object:window];

        [center addObserver:self selector:@selector(windowDidMiniaturize:) name:NSWindowDidMiniaturizeNotification object:window];

        [center addObserver:self selector:@selector(windowDidDeminiaturize:) name:NSWindowDidDeminiaturizeNotification object:window];

        [center addObserver:self selector:@selector(windowDidBecomeKey:) name:NSWindowDidBecomeKeyNotification object:window];

        [center addObserver:self selector:@selector(windowDidResignKey:) name:NSWindowDidResignKeyNotification object:window];

    } else {

        [window setDelegate:self];

    }

    [center addObserver:self selector:@selector(windowDidHide:) name:NSApplicationDidHideNotification object:NSApp];

    [center addObserver:self selector:@selector(windowDidUnhide:) name:NSApplicationDidUnhideNotification object:NSApp];

    [window setNextResponder:self];

    [window setAcceptsMouseMovedEvents:YES];

    [view setNextResponder:self];

    [view addTrackingRect:[view visibleRect] owner:self userData:nil assumeInside:NO];

#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_6

    [view setAcceptsTouchEvents:YES];

#endif

}

- (void)close

{

    NSNotificationCenter *center;

    NSWindow *window = _data->nswindow;

    NSView *view = [window contentView];

    center = [NSNotificationCenter defaultCenter];

    if ([window delegate] != self) {

        [center removeObserver:self name:NSWindowDidExposeNotification object:window];

        [center removeObserver:self name:NSWindowDidMoveNotification object:window];

        [center removeObserver:self name:NSWindowDidResizeNotification object:window];

        [center removeObserver:self name:NSWindowDidMiniaturizeNotification object:window];

        [center removeObserver:self name:NSWindowDidDeminiaturizeNotification object:window];

        [center removeObserver:self name:NSWindowDidBecomeKeyNotification object:window];

        [center removeObserver:self name:NSWindowDidResignKeyNotification object:window];

    } else {

        [window setDelegate:nil];

    }

    [center removeObserver:self name:NSApplicationDidHideNotification object:NSApp];

    [center removeObserver:self name:NSApplicationDidUnhideNotification object:NSApp];

    if ([window nextResponder] == self) {

        [window setNextResponder:nil];

    }

    if ([view nextResponder] == self) {

        [view setNextResponder:nil];

    }

}

- (BOOL)windowShouldClose:(id)sender

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_CLOSE, 0, 0);

    return NO;

}

- (void)windowDidExpose:(NSNotification *)aNotification

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_EXPOSED, 0, 0);

}

- (void)windowDidMove:(NSNotification *)aNotification

{

    int x, y;

    NSRect rect = [_data->nswindow contentRectForFrameRect:[_data->nswindow frame]];

    ConvertNSRect(&rect);

    x = (int)rect.origin.x;

    y = (int)rect.origin.y;

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_MOVED, x, y);

}

- (void)windowDidResize:(NSNotification *)aNotification

{

    int w, h;

    NSRect rect = [_data->nswindow contentRectForFrameRect:[_data->nswindow frame]];

    w = (int)rect.size.width;

    h = (int)rect.size.height;

    if (SDL_IsShapedWindow(_data->window))

        Cocoa_ResizeWindowShape(_data->window);

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_RESIZED, w, h);

}

- (void)windowDidMiniaturize:(NSNotification *)aNotification

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_MINIMIZED, 0, 0);

}

- (void)windowDidDeminiaturize:(NSNotification *)aNotification

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_RESTORED, 0, 0);

}

- (void)windowDidBecomeKey:(NSNotification *)aNotification

{

    SDL_Window *window = _data->window;

    /* We're going to get keyboard events, since we're key. */

    SDL_SetKeyboardFocus(window);

    /* If we just gained focus we need the updated mouse position */

    if (SDL_GetMouseFocus() == window) {

        NSPoint point;

        point = [NSEvent mouseLocation];

        point = [_data->nswindow convertScreenToBase:point];

        point.y = window->h - point.y;

        SDL_SendMouseMotion(window, 0, (int)point.x, (int)point.y);

    }

    /* Check to see if someone updated the clipboard */

    Cocoa_CheckClipboardUpdate(_data->videodata);

}

- (void)windowDidResignKey:(NSNotification *)aNotification

{

    /* Some other window will get mouse events, since we're not key. */

    if (SDL_GetMouseFocus() == _data->window) {

        SDL_SetMouseFocus(NULL);

    }

    /* Some other window will get keyboard events, since we're not key. */

    if (SDL_GetKeyboardFocus() == _data->window) {

        SDL_SetKeyboardFocus(NULL);

    }

}

- (void)windowDidHide:(NSNotification *)aNotification

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_HIDDEN, 0, 0);

}

- (void)windowDidUnhide:(NSNotification *)aNotification

{

    SDL_SendWindowEvent(_data->window, SDL_WINDOWEVENT_SHOWN, 0, 0);

}

- (void)mouseDown:(NSEvent *)theEvent

{

    int button;

    switch ([theEvent buttonNumber]) {

    case 0:

        button = SDL_BUTTON_LEFT;

        break;

    case 1:

        button = SDL_BUTTON_RIGHT;

        break;

    case 2:

        button = SDL_BUTTON_MIDDLE;

        break;

    default:

        button = [theEvent buttonNumber] + 1;

        break;

    }

    SDL_SendMouseButton(_data->window, SDL_PRESSED, button);

}

- (void)rightMouseDown:(NSEvent *)theEvent

{

    [self mouseDown:theEvent];

}

- (void)otherMouseDown:(NSEvent *)theEvent

{

    [self mouseDown:theEvent];

}

- (void)mouseUp:(NSEvent *)theEvent

{

    int button;

    switch ([theEvent buttonNumber]) {

    case 0:

        button = SDL_BUTTON_LEFT;

        break;

    case 1:

        button = SDL_BUTTON_RIGHT;

        break;

    case 2:

        button = SDL_BUTTON_MIDDLE;

        break;

    default:

        button = [theEvent buttonNumber] + 1;

        break;

    }

    SDL_SendMouseButton(_data->window, SDL_RELEASED, button);

}

- (void)rightMouseUp:(NSEvent *)theEvent

{

    [self mouseUp:theEvent];

}

- (void)otherMouseUp:(NSEvent *)theEvent

{

    [self mouseUp:theEvent];

}

- (void)mouseEntered:(NSEvent *)theEvent

{

    SDL_Mouse *mouse = SDL_GetMouse();

    SDL_SetMouseFocus(_data->window);

    if (!mouse->cursor_shown) {

        [NSCursor hide];

    }

}

- (void)mouseExited:(NSEvent *)theEvent

{

    SDL_Window *window = _data->window;

    SDL_Mouse *mouse = SDL_GetMouse();

    if (SDL_GetMouseFocus() == window) {

        if (window->flags & SDL_WINDOW_INPUT_GRABBED) {

            int x, y;

            NSPoint point;

            CGPoint cgpoint;

            point = [theEvent locationInWindow];

            point.y = window->h - point.y;

            SDL_SendMouseMotion(window, 0, (int)point.x, (int)point.y);

            SDL_GetMouseState(&x, &y);

            cgpoint.x = window->x + x;

            cgpoint.y = window->y + y;

            CGDisplayMoveCursorToPoint(kCGDirectMainDisplay, cgpoint);

        } else {

            SDL_SetMouseFocus(NULL);

        }

    }

    if (!mouse->cursor_shown) {

        [NSCursor unhide];

    }

}

- (void)mouseMoved:(NSEvent *)theEvent

{

    SDL_Window *window = _data->window;

#ifdef RELATIVE_MOTION

    if (window->flags & SDL_WINDOW_INPUT_GRABBED) {

        return;

    }

#endif

    if (SDL_GetMouseFocus() == window) {

        NSPoint point;

        point = [theEvent locationInWindow];

        point.y = window->h - point.y;

        SDL_SendMouseMotion(window, 0, (int)point.x, (int)point.y);

    }

}

- (void)mouseDragged:(NSEvent *)theEvent

{

    [self mouseMoved:theEvent];

}

- (void)rightMouseDragged:(NSEvent *)theEvent

{

    [self mouseMoved:theEvent];

}

- (void)otherMouseDragged:(NSEvent *)theEvent

{

    [self mouseMoved:theEvent];

}

- (void)scrollWheel:(NSEvent *)theEvent

{

    Cocoa_HandleMouseWheel(_data->window, theEvent);

}

- (void)touchesBeganWithEvent:(NSEvent *) theEvent

{

    [self handleTouches:COCOA_TOUCH_DOWN withEvent:theEvent];

}

- (void)touchesMovedWithEvent:(NSEvent *) theEvent

{

    [self handleTouches:COCOA_TOUCH_MOVE withEvent:theEvent];

}

- (void)touchesEndedWithEvent:(NSEvent *) theEvent

{

    [self handleTouches:COCOA_TOUCH_UP withEvent:theEvent];

}

- (void)touchesCancelledWithEvent:(NSEvent *) theEvent

{

    [self handleTouches:COCOA_TOUCH_CANCELLED withEvent:theEvent];

}

- (void)handleTouches:(cocoaTouchType)type withEvent:(NSEvent *)event

{

#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_6

    NSSet *touches = 0;

    NSEnumerator *enumerator;

    NSTouch *touch;

    switch (type) {

        case COCOA_TOUCH_DOWN:

            touches = [event touchesMatchingPhase:NSTouchPhaseBegan inView:nil];

            break;

        case COCOA_TOUCH_UP:

        case COCOA_TOUCH_CANCELLED:

            touches = [event touchesMatchingPhase:NSTouchPhaseEnded inView:nil];

            break;

        case COCOA_TOUCH_MOVE:

            touches = [event touchesMatchingPhase:NSTouchPhaseMoved inView:nil];

            break;

    }

    enumerator = [touches objectEnumerator];

    touch = (NSTouch*)[enumerator nextObject];

    while (touch) {

        SDL_TouchID touchId = (SDL_TouchID)[touch device];

        if (!SDL_GetTouch(touchId)) {

            SDL_Touch touch;

            touch.id = touchId;

            touch.x_min = 0;

            touch.x_max = 1;

            touch.native_xres = touch.x_max - touch.x_min;

            touch.y_min = 0;

            touch.y_max = 1;

            touch.native_yres = touch.y_max - touch.y_min;

            touch.pressure_min = 0;

            touch.pressure_max = 1;

            touch.native_pressureres = touch.pressure_max - touch.pressure_min;

            if (SDL_AddTouch(&touch, "") < 0) {

                return;

            }

        } 

        SDL_FingerID fingerId = (SDL_FingerID)[touch identity];

        float x = [touch normalizedPosition].x;

        float y = [touch normalizedPosition].y;

        /* Make the origin the upper left instead of the lower left */

        y = 1.0f - y;

        switch (type) {

        case COCOA_TOUCH_DOWN:

            SDL_SendFingerDown(touchId, fingerId, SDL_TRUE, x, y, 1);

            break;

        case COCOA_TOUCH_UP:

        case COCOA_TOUCH_CANCELLED:

            SDL_SendFingerDown(touchId, fingerId, SDL_FALSE, x, y, 1);

            break;

        case COCOA_TOUCH_MOVE:

            SDL_SendTouchMotion(touchId, fingerId, SDL_FALSE, x, y, 1);

            break;

        }

        touch = (NSTouch*)[enumerator nextObject];

    }

#endif /* MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_6 */

}

@end

@interface SDLWindow : NSWindow

/* These are needed for borderless/fullscreen windows */

- (BOOL)canBecomeKeyWindow;

- (BOOL)canBecomeMainWindow;

@end

@implementation SDLWindow

- (BOOL)canBecomeKeyWindow

{

    return YES;

}

- (BOOL)canBecomeMainWindow

{

    return YES;

}

@end

@interface SDLView : NSView { }

@end

@implementation SDLView

- (void)rightMouseDown:(NSEvent *)theEvent

{

    [[self nextResponder] rightMouseDown:theEvent];

}

@end

static unsigned int

GetWindowStyle(SDL_Window * window)

{

    unsigned int style;

	if (window->flags & SDL_WINDOW_FULLSCREEN) {

        style = NSBorderlessWindowMask;

	} else {

		if (window->flags & SDL_WINDOW_BORDERLESS) {

			style = NSBorderlessWindowMask;

		} else {

			style = (NSTitledWindowMask|NSClosableWindowMask|NSMiniaturizableWindowMask);

		}

		if (window->flags & SDL_WINDOW_RESIZABLE) {

			style |= NSResizableWindowMask;

		}

	}

    return style;

}

static int

SetupWindowData(_THIS, SDL_Window * window, NSWindow *nswindow, SDL_bool created)

{

    NSAutoreleasePool *pool;

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

    SDL_WindowData *data;

    /* Allocate the window data */

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

    if (!data) {

        SDL_OutOfMemory();

        return -1;

    }

    data->window = window;

    data->nswindow = nswindow;

    data->created = created;

    data->videodata = videodata;

    pool = [[NSAutoreleasePool alloc] init];

    /* Create an event listener for the window */

    data->listener = [[Cocoa_WindowListener alloc] init];

    /* Fill in the SDL window with the window data */

    {

        NSRect rect = [nswindow contentRectForFrameRect:[nswindow frame]];

        NSView *contentView = [[SDLView alloc] initWithFrame:rect];

        [nswindow setContentView: contentView];