/*

    SDL - Simple DirectMedia Layer

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

    _data = data;

    center = [NSNotificationCenter defaultCenter];

    [_data->nswindow setNextResponder:self];

    if ([_data->nswindow delegate] != nil) {

        [center addObserver:self selector:@selector(windowDisExpose:) name:NSWindowDidExposeNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidMove:) name:NSWindowDidMoveNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidResize:) name:NSWindowDidResizeNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidMiniaturize:) name:NSWindowDidMiniaturizeNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidDeminiaturize:) name:NSWindowDidDeminiaturizeNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidBecomeKey:) name:NSWindowDidBecomeKeyNotification object:_data->nswindow];

        [center addObserver:self selector:@selector(windowDidResignKey:) name:NSWindowDidResignKeyNotification object:_data->nswindow];

    } else {

        [_data->nswindow setDelegate:self];

    }

// FIXME: Why doesn't this work?

//    [center addObserver:self selector:@selector(rightMouseDown:) name:[NSString stringWithCString:"rightMouseDown" encoding:NSUTF8StringEncoding] object:[_data->nswindow contentView]];

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

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

    [_data->nswindow setAcceptsMouseMovedEvents:YES];

#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_6

    [[_data->nswindow contentView] setAcceptsTouchEvents:YES];

#endif

}

- (void)close

{

    NSNotificationCenter *center;

    center = [NSNotificationCenter defaultCenter];

    [_data->nswindow setNextResponder:nil];

    if ([_data->nswindow delegate] != self) {

        [center removeObserver:self name:NSWindowDidExposeNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidMoveNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidResizeNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidMiniaturizeNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidDeminiaturizeNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidBecomeKeyNotification object:_data->nswindow];

        [center removeObserver:self name:NSWindowDidResignKeyNotification object:_data->nswindow];

    } else {

        [_data->nswindow setDelegate:nil];

    }

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

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

}

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

{

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

    SDL_SetKeyboardFocus(_data->window);

    /* 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)mouseMoved:(NSEvent *)theEvent

{

    SDL_Window *window = _data->window;

    NSPoint point;

    if (window->flags & SDL_WINDOW_FULLSCREEN)

        return;

    point = [theEvent locationInWindow];

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

    if ( point.x < 0 || point.x >= window->w ||

         point.y < 0 || point.y >= window->h ) {

        if (SDL_GetMouseFocus() == window) {

            SDL_SetMouseFocus(NULL);

        }

    } else {

        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 {

    Cocoa_WindowListener *listener;

}

@end

@implementation SDLView

- (id) initWithFrame: (NSRect) rect

            listener: (Cocoa_WindowListener *) theListener

{

    if (self = [super initWithFrame:rect]) {

        listener = theListener;

    }

    return self;

}

- (void)rightMouseDown:(NSEvent *)theEvent

{

    [listener mouseDown:theEvent];

}

@end

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_malloc(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];

    [data->listener listen:data];

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

    {

        SDL_Rect bounds;

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

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

                                                    listener: data->listener];

#if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_6

        [contentView setAcceptsTouchEvents:YES];

#endif

        [nswindow setContentView: contentView];

        [contentView release];

        ConvertNSRect(&rect);

        window->w = (int)rect.size.width;

        window->h = (int)rect.size.height;

    }

    if ([nswindow isVisible]) {

        window->flags |= SDL_WINDOW_SHOWN;

    } else {

        window->flags &= ~SDL_WINDOW_SHOWN;

    }

    {

        unsigned int style = [nswindow styleMask];

        if ((style & ~NSResizableWindowMask) == NSBorderlessWindowMask) {

            window->flags |= SDL_WINDOW_BORDERLESS;

        } else {

            window->flags &= ~SDL_WINDOW_BORDERLESS;

        }

        if (style & NSResizableWindowMask) {

            window->flags |= SDL_WINDOW_RESIZABLE;

        } else {

            window->flags &= ~SDL_WINDOW_RESIZABLE;

        }

    }

    if ([nswindow isZoomed]) {

        window->flags |= SDL_WINDOW_MAXIMIZED;

    } else {

        window->flags &= ~SDL_WINDOW_MAXIMIZED;

    }

    if ([nswindow isMiniaturized]) {

        window->flags |= SDL_WINDOW_MINIMIZED;

    } else {

        window->flags &= ~SDL_WINDOW_MINIMIZED;

    }

    if ([nswindow isKeyWindow]) {

        window->flags |= SDL_WINDOW_INPUT_FOCUS;

        SDL_SetKeyboardFocus(data->window);

        if (window->flags & SDL_WINDOW_INPUT_GRABBED) {

            /* FIXME */

        }

    }

    /* All done! */

    [pool release];

    window->driverdata = data;

    return 0;

}

int

Cocoa_CreateWindow(_THIS, SDL_Window * window)

{

    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];

    NSWindow *nswindow;

    NSRect rect;

    SDL_Rect bounds;

    unsigned int style;

    Cocoa_GetDisplayBounds(_this, display, &bounds);

    if ((window->flags & SDL_WINDOW_FULLSCREEN)

        rect.origin.x = bounds.x + (bounds.w - window->w) / 2;