/*

 * Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code 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 General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

#import <dlfcn.h>

#import <JavaNativeFoundation/JavaNativeFoundation.h>

#include "jni_util.h"

#import "CMenuBar.h"

#import "InitIDs.h"

#import "LWCToolkit.h"

#import "ThreadUtilities.h"

#import "AWT_debug.h"

#import "CSystemColors.h"

#import "sun_lwawt_macosx_LWCToolkit.h"

int gNumberOfButtons;

jint* gButtonDownMasks;

@implementation AWTToolkit

static long eventCount;

+ (long) getEventCount{

    return eventCount;

}

+ (void) eventCountPlusPlus{

    eventCount++;

}

@end

@interface AWTRunLoopObject : NSObject {

    BOOL _shouldEndRunLoop;

}

@end

@implementation AWTRunLoopObject

- (id) init {

    self = [super init];

    if (self != nil) {

        _shouldEndRunLoop = NO;

    }

    return self;

}

- (BOOL) shouldEndRunLoop {

    return _shouldEndRunLoop;

}

- (void) endRunLoop {

    _shouldEndRunLoop = YES;

}

@end

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    nativeSyncQueue

 * Signature: (J)Z

 */

JNIEXPORT jboolean JNICALL Java_sun_lwawt_macosx_LWCToolkit_nativeSyncQueue

(JNIEnv *env, jobject self, jlong timeout)

{

    int currentEventNum = [AWTToolkit getEventCount];

    [JNFRunLoop performOnMainThreadWaiting:YES withBlock:^(){}];

    if (([AWTToolkit getEventCount] - currentEventNum) != 0) {

        return JNI_TRUE;

    }

    return JNI_FALSE;

}

static JNF_CLASS_CACHE(jc_Component, "java/awt/Component");

static JNF_MEMBER_CACHE(jf_Component_appContext, jc_Component, "appContext", "Lsun/awt/AppContext;");

static JNF_CLASS_CACHE(jc_MenuComponent, "java/awt/MenuComponent");

static JNF_MEMBER_CACHE(jf_MenuComponent_appContext, jc_MenuComponent, "appContext", "Lsun/awt/AppContext;");

/*

 * Class:     sun_awt_SunToolkit

 * Method:    getAppContext

 * Signature: (Ljava/awt/Object;)Lsun/awt/AppContext;

 */

JNIEXPORT jobject JNICALL

Java_sun_awt_SunToolkit_getAppContext

(JNIEnv *env, jclass cls, jobject obj)

{

    jobject appContext = NULL;

JNF_COCOA_ENTER(env);

    if (JNFIsInstanceOf(env, obj, &jc_Component)) {

        appContext = JNFGetObjectField(env, obj, jf_Component_appContext);

    } else if (JNFIsInstanceOf(env, obj, &jc_MenuComponent)) {

        appContext = JNFGetObjectField(env, obj, jf_MenuComponent_appContext);

    }

JNF_COCOA_EXIT(env);

    return appContext;

}

/*

 * Class:     sun_awt_SunToolkit

 * Method:    setAppContext

 * Signature: (Ljava/lang/Object;Lsun/awt/AppContext;)Z

 */

JNIEXPORT jboolean JNICALL

Java_sun_awt_SunToolkit_setAppContext

(JNIEnv *env, jclass cls, jobject obj, jobject appContext)

{

    jboolean isComponent;

JNF_COCOA_ENTER(env);

    if (JNFIsInstanceOf(env, obj, &jc_Component)) {

        JNFSetObjectField(env, obj, jf_Component_appContext, appContext);

        isComponent = JNI_TRUE;

    } else if (JNFIsInstanceOf(env, obj, &jc_MenuComponent)) {

        JNFSetObjectField(env, obj, jf_MenuComponent_appContext, appContext);

        isComponent = JNI_FALSE;

    }

JNF_COCOA_EXIT(env);

    return isComponent;

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    beep

 * Signature: ()V

 */

JNIEXPORT void JNICALL

Java_sun_lwawt_macosx_LWCToolkit_beep

(JNIEnv *env, jobject self)

{

    NSBeep(); // produces both sound and visual flash, if configured in System Preferences

}

CGDirectDisplayID

FindCGDirectDisplayIDForScreenIndex(jint screenIndex)

{

    // most common case - just one monitor

    CGDirectDisplayID screenID = CGMainDisplayID();

    CGDisplayCount displayCount = 0;

    CGGetOnlineDisplayList(0, NULL, &displayCount);

    if ((displayCount > 1) &&

        (screenIndex >= 0) &&

        (screenIndex < (jint)displayCount))

    {

        if (displayCount < 10) {

            // stack allocated optimization for less than 10 monitors

            CGDirectDisplayID onlineDisplays[displayCount];

            CGGetOnlineDisplayList(displayCount, onlineDisplays, &displayCount);

            screenID = (CGDirectDisplayID)onlineDisplays[screenIndex];

        } else {

            CGDirectDisplayID *onlineDisplays =

            malloc(displayCount*sizeof(CGDirectDisplayID));

            if (onlineDisplays != NULL) {

                CGGetOnlineDisplayList(displayCount, onlineDisplays,

                                       &displayCount);

                screenID = (CGDirectDisplayID)onlineDisplays[screenIndex];

                free(onlineDisplays);

            }

        }

    }

    return screenID;

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    initIDs

 * Signature: ()V

 */

JNIEXPORT void JNICALL

Java_sun_lwawt_macosx_LWCToolkit_initIDs

(JNIEnv *env, jclass klass) {

    // set thread names

    dispatch_async(dispatch_get_main_queue(), ^(void){

        [[NSThread currentThread] setName:@"AppKit Thread"];

        JNIEnv *env = [ThreadUtilities getJNIEnv];

        static JNF_CLASS_CACHE(jc_LWCToolkit, "sun/lwawt/macosx/LWCToolkit");

        static JNF_STATIC_MEMBER_CACHE(jsm_installToolkitThreadNameInJava, jc_LWCToolkit, "installToolkitThreadNameInJava", "()V");

        JNFCallStaticVoidMethod(env, jsm_installToolkitThreadNameInJava);

    });

    gNumberOfButtons = sun_lwawt_macosx_LWCToolkit_BUTTONS;

    jclass inputEventClazz = (*env)->FindClass(env, "java/awt/event/InputEvent");

    jmethodID getButtonDownMasksID = (*env)->GetStaticMethodID(env, inputEventClazz, "getButtonDownMasks", "()[I");

    jintArray obj = (jintArray)(*env)->CallStaticObjectMethod(env, inputEventClazz, getButtonDownMasksID);

    jint * tmp = (*env)->GetIntArrayElements(env, obj, JNI_FALSE);

    gButtonDownMasks = (jint*)malloc(sizeof(jint) * gNumberOfButtons);

    if (gButtonDownMasks == NULL) {

        gNumberOfButtons = 0;

        JNU_ThrowOutOfMemoryError(env, NULL);

        return;

    }

    int i;

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

        gButtonDownMasks[i] = tmp[i];

    }

    (*env)->ReleaseIntArrayElements(env, obj, tmp, 0);

    (*env)->DeleteLocalRef(env, obj);

}

static UInt32 RGB(NSColor *c) {

    c = [c colorUsingColorSpaceName:NSCalibratedRGBColorSpace];

    if (c == nil)

    {

        return -1; // opaque white

    }

    CGFloat r, g, b, a;

    [c getRed:&r green:&g blue:&b alpha:&a];

    UInt32 ir = (UInt32) (r*255+0.5),

    ig = (UInt32) (g*255+0.5),

    ib = (UInt32) (b*255+0.5),

    ia = (UInt32) (a*255+0.5);

    //    NSLog(@"%@ %d, %d, %d", c, ir, ig, ib);

    return ((ia & 0xFF) << 24) | ((ir & 0xFF) << 16) | ((ig & 0xFF) << 8) | ((ib & 0xFF) << 0);

}

void doLoadNativeColors(JNIEnv *env, jintArray jColors, BOOL useAppleColors) {

    jint len = (*env)->GetArrayLength(env, jColors);

    UInt32 colorsArray[len];

    UInt32 *colors = colorsArray;

    [JNFRunLoop performOnMainThreadWaiting:YES withBlock:^(){

        NSUInteger i;

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

            colors[i] = RGB([CSystemColors getColor:i useAppleColor:useAppleColors]);

        }

    }];

    jint *_colors = (*env)->GetPrimitiveArrayCritical(env, jColors, 0);

    memcpy(_colors, colors, len * sizeof(UInt32));

    (*env)->ReleasePrimitiveArrayCritical(env, jColors, _colors, 0);

}

/**

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    loadNativeColors

 * Signature: ([I[I)V

 */

JNIEXPORT void JNICALL Java_sun_lwawt_macosx_LWCToolkit_loadNativeColors

(JNIEnv *env, jobject peer, jintArray jSystemColors, jintArray jAppleColors)

{

JNF_COCOA_ENTER(env);

    doLoadNativeColors(env, jSystemColors, NO);

    doLoadNativeColors(env, jAppleColors, YES);

JNF_COCOA_EXIT(env);

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    createAWTRunLoopMediator

 * Signature: ()J

 */

JNIEXPORT jlong JNICALL Java_sun_lwawt_macosx_LWCToolkit_createAWTRunLoopMediator

(JNIEnv *env, jclass clz)

{

AWT_ASSERT_APPKIT_THREAD;

    AWTRunLoopObject *o = nil;

    // We double retain because this object is owned by both main thread and "other" thread

    // We release in both doAWTRunLoop and stopAWTRunLoop

    o = [[AWTRunLoopObject alloc] init];

    if (o) {

        CFRetain(o); // GC

        CFRetain(o); // GC

        [o release];

    }

    return ptr_to_jlong(o);

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    doAWTRunLoop

 * Signature: (JZZ)V

 */

JNIEXPORT void JNICALL Java_sun_lwawt_macosx_LWCToolkit_doAWTRunLoop

(JNIEnv *env, jclass clz, jlong mediator, jboolean awtMode, jboolean detectDeadlocks)

{

AWT_ASSERT_APPKIT_THREAD;

JNF_COCOA_ENTER(env);

    AWTRunLoopObject* mediatorObject = (AWTRunLoopObject*)jlong_to_ptr(mediator);

    if (mediatorObject == nil) return;

    if (!sInPerformFromJava || !detectDeadlocks) {

        NSRunLoop *currentRunLoop = [NSRunLoop currentRunLoop];

        NSDate *distantFuture = [NSDate distantFuture];

        NSString *mode = (awtMode) ? [JNFRunLoop javaRunLoopMode] : NSDefaultRunLoopMode;

        BOOL isRunning = YES;

        while (isRunning && ![mediatorObject shouldEndRunLoop]) {

            // Don't use acceptInputForMode because that doesn't setup autorelease pools properly

            isRunning = [currentRunLoop runMode:mode beforeDate:distantFuture];

        }

    }

#ifndef PRODUCT_BUILD

    if (sInPerformFromJava) {

        NSLog(@"Apple AWT: Short-circuiting CToolkit.invokeAndWait trampoline deadlock!!!!!");

        NSLog(@"\tPlease file a bug report with this message and a reproducible test case.");

    }

#endif

    CFRelease(mediatorObject);

JNF_COCOA_EXIT(env);

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    stopAWTRunLoop

 * Signature: (J)V

 */

JNIEXPORT void JNICALL Java_sun_lwawt_macosx_LWCToolkit_stopAWTRunLoop

(JNIEnv *env, jclass clz, jlong mediator)

{

AWT_ASSERT_NOT_APPKIT_THREAD;

JNF_COCOA_ENTER(env);

    AWTRunLoopObject* mediatorObject = (AWTRunLoopObject*)jlong_to_ptr(mediator);

    [ThreadUtilities performOnMainThread:@selector(endRunLoop) onObject:mediatorObject withObject:nil waitUntilDone:NO awtMode:YES];

    CFRelease(mediatorObject);

JNF_COCOA_EXIT(env);

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    isCapsLockOn

 * Signature: ()Z

 */

JNIEXPORT jboolean JNICALL Java_sun_lwawt_macosx_LWCToolkit_isCapsLockOn

(JNIEnv *env, jobject self)

{

    __block jboolean isOn = JNI_FALSE;

    [JNFRunLoop performOnMainThreadWaiting:YES withBlock:^(){

        NSUInteger modifiers = [NSEvent modifierFlags];

        isOn = (modifiers & NSAlphaShiftKeyMask) != 0;

    }];

    return isOn;

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Method:    isApplicationActive

 * Signature: ()Z

 */

JNIEXPORT jboolean JNICALL Java_sun_lwawt_macosx_LWCToolkit_isApplicationActive

(JNIEnv *env, jclass clazz)

{

    __block jboolean active = JNI_FALSE;

JNF_COCOA_ENTER(env);

    if ([NSThread isMainThread]) {

        active = (jboolean)[NSRunningApplication currentApplication].active;

    } else {

        [JNFRunLoop performOnMainThreadWaiting:YES withBlock:^() {

            active = (jboolean)[NSRunningApplication currentApplication].active;

        }];

    }

JNF_COCOA_EXIT(env);

    return active;

}

/*

 * Class:     sun_awt_SunToolkit

 * Method:    closeSplashScreen

 * Signature: ()V

 */

JNIEXPORT void JNICALL

Java_sun_awt_SunToolkit_closeSplashScreen(JNIEnv *env, jclass cls)

{

    void *hSplashLib = dlopen(0, RTLD_LAZY);

    if (!hSplashLib) return;

    void (*splashClose)() = dlsym(hSplashLib, "SplashClose");

    if (splashClose) {

        splashClose();

    }

    dlclose(hSplashLib);

}

// TODO: definitely doesn't belong here (copied from fontpath.c in the

// solaris tree)...

JNIEXPORT jstring JNICALL

Java_sun_font_FontManager_getFontPath

(JNIEnv *env, jclass obj, jboolean noType1)

{

    return JNFNSToJavaString(env, @"/Library/Fonts");

}

// This isn't yet used on unix, the implementation is added since shared

// code calls this method in preparation for future use.

JNIEXPORT void JNICALL

Java_sun_font_FontManager_populateFontFileNameMap

(JNIEnv *env, jclass obj, jobject fontToFileMap, jobject fontToFamilyMap, jobject familyToFontListMap, jobject locale)

{

}

/*

 * Class:     sun_lwawt_macosx_LWCToolkit

 * Signature: ()V

 */

(JNIEnv *env, jclass cls)

{

    }

}