/*

 * 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.

 */

#include "awt.h"

#include "awt_MenuItem.h"

#include "awt_Menu.h"

#include "awt_MenuBar.h"

#include "awt_DesktopProperties.h"

#include <sun_awt_windows_WCheckboxMenuItemPeer.h>

// Begin -- Win32 SDK include files

#include <tchar.h>

#include <imm.h>

#include <ime.h>

// End -- Win32 SDK include files

//add for multifont menuitem

#include <java_awt_CheckboxMenuItem.h>

#include <java_awt_Toolkit.h>

#include <java_awt_event_InputEvent.h>

/* IMPORTANT! Read the README.JNI file for notes on JNI converted AWT code.

 */

/***********************************************************************/

// struct for _SetLabel() method

struct SetLabelStruct {

    jobject menuitem;

    jstring label;

};

/************************************************************************

 * AwtMenuItem fields

 */

HBITMAP AwtMenuItem::bmpCheck;

jobject AwtMenuItem::systemFont;

jfieldID AwtMenuItem::labelID;

jfieldID AwtMenuItem::enabledID;

jfieldID AwtMenuItem::fontID;

jfieldID AwtMenuItem::appContextID;

jfieldID AwtMenuItem::shortcutLabelID;

jfieldID AwtMenuItem::isCheckboxID;

jfieldID AwtMenuItem::stateID;

jmethodID AwtMenuItem::getDefaultFontMID;

// Added by waleed to initialize the RTL Flags

LANGID AwtMenuItem::m_idLang = LOWORD(GetKeyboardLayout(0));

UINT AwtMenuItem::m_CodePage =

    AwtMenuItem::LangToCodePage(AwtMenuItem::m_idLang);

BOOL AwtMenuItem::sm_rtl = PRIMARYLANGID(GetInputLanguage()) == LANG_ARABIC ||

                           PRIMARYLANGID(GetInputLanguage()) == LANG_HEBREW;

BOOL AwtMenuItem::sm_rtlReadingOrder =

    PRIMARYLANGID(GetInputLanguage()) == LANG_ARABIC;

/*

 * This constant holds width of the default menu

 * check-mark bitmap for default settings on XP/Vista,

 * in pixels

 */

static const int SM_CXMENUCHECK_DEFAULT_ON_XP = 13;

static const int SM_CXMENUCHECK_DEFAULT_ON_VISTA = 15;

/************************************************************************

 * AwtMenuItem methods

 */

AwtMenuItem::AwtMenuItem() {

    m_peerObject = NULL;

    m_menuContainer = NULL;

    m_Id = (UINT)-1;

    m_freeId = FALSE;

    m_isCheckbox = FALSE;

}

AwtMenuItem::~AwtMenuItem()

{

}

void AwtMenuItem::RemoveCmdID()

{

    if (m_freeId) {

        AwtToolkit::GetInstance().RemoveCmdID( GetID() );

    }

}

void AwtMenuItem::Dispose()

{

    RemoveCmdID();

    JNIEnv *env = (JNIEnv *)JNU_GetEnv(jvm, JNI_VERSION_1_2);

    if (m_peerObject != NULL) {

        JNI_SET_PDATA(m_peerObject, NULL);

        env->DeleteGlobalRef(m_peerObject);

        m_peerObject = NULL;

    }

    AwtObject::Dispose();

}

LPCTSTR AwtMenuItem::GetClassName() {

  return TEXT("SunAwtMenuItem");

}

// Convert Language ID to CodePage

UINT AwtMenuItem::LangToCodePage(LANGID idLang)

{

    TCHAR strCodePage[MAX_ACP_STR_LEN];

    // use the LANGID to create a LCID

    LCID idLocale = MAKELCID(idLang, SORT_DEFAULT);

    // get the ANSI code page associated with this locale

    if (GetLocaleInfo(idLocale, LOCALE_IDEFAULTANSICODEPAGE, strCodePage, sizeof(strCodePage)/sizeof(TCHAR)) > 0 )

        return _ttoi(strCodePage);

    else

        return GetACP();

}

BOOL AwtMenuItem::CheckMenuCreation(JNIEnv *env, jobject self, HMENU hMenu)

{

    // fix for 5088782

    // check if CreateMenu() returns not null value and if it does -

    //   create an InternalError or OutOfMemoryError based on GetLastError().

    //   This error is set to createError field of WObjectPeer and then

    //   checked and thrown in WMenuPeer or WMenuItemPeer constructor. We

    //   can't throw an error here because this code is invoked on Toolkit thread

    // return TRUE if menu is created successfully, FALSE otherwise

    if (hMenu == NULL)

    {

        DWORD dw = GetLastError();

        jobject createError = NULL;

        if (dw == ERROR_OUTOFMEMORY)

        {

            jstring errorMsg = JNU_NewStringPlatform(env, L"too many menu handles");

            createError = JNU_NewObjectByName(env, "java/lang/OutOfMemoryError",

                                                   "(Ljava/lang/String;)V",

                                                   errorMsg);

            env->DeleteLocalRef(errorMsg);

        }

        else

        {

            TCHAR *buf;

            FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,

                NULL, dw, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),

                (LPTSTR)&buf, 0, NULL);

            jstring s = JNU_NewStringPlatform(env, buf);

            createError = JNU_NewObjectByName(env, "java/lang/InternalError",

                                                   "(Ljava/lang/String;)V", s);

            LocalFree(buf);

            env->DeleteLocalRef(s);

        }

        env->SetObjectField(self, AwtObject::createErrorID, createError);

        if (createError != NULL)

        {

            env->DeleteLocalRef(createError);

        }

        return FALSE;

    }

    return TRUE;

}

/*

 * Link the C++, Java peer together

 */

void AwtMenuItem::LinkObjects(JNIEnv *env, jobject peer)

{

    m_peerObject = env->NewGlobalRef(peer);

    JNI_SET_PDATA(peer, this);

}

AwtMenuItem* AwtMenuItem::Create(jobject peer, jobject menuPeer)

{

    JNIEnv *env = (JNIEnv *)JNU_GetEnv(jvm, JNI_VERSION_1_2);

    jobject target = NULL;

    AwtMenuItem* item = NULL;

    try {

        if (env->EnsureLocalCapacity(1) < 0) {

            return NULL;

        }

        PDATA pData;

        JNI_CHECK_PEER_RETURN_NULL(menuPeer);

        /* target is a java.awt.MenuItem  */

        target = env->GetObjectField(peer, AwtObject::targetID);

        AwtMenu* menu = (AwtMenu *)pData;

        item = new AwtMenuItem();

        jboolean isCheckbox =

            (jboolean)env->GetBooleanField(peer, AwtMenuItem::isCheckboxID);

        if (isCheckbox) {

            item->SetCheckbox();

        }

        item->LinkObjects(env, peer);

        item->SetMenuContainer(menu);

        item->SetNewID();

        menu->AddItem(item);

    } catch (...) {

        env->DeleteLocalRef(target);

        throw;

    }

    env->DeleteLocalRef(target);

    return item;

}

MsgRouting AwtMenuItem::WmNotify(UINT notifyCode)

{

    return mrDoDefault;

}

// This function returns a local reference

jobject

AwtMenuItem::GetFont(JNIEnv *env)

{

    jobject self = GetPeer(env);

    jobject target = env->GetObjectField(self, AwtObject::targetID);

    jobject font = JNU_CallMethodByName(env, 0, target, "getFont_NoClientCode", "()Ljava/awt/Font;").l;

    if (font == NULL) {

        font = env->NewLocalRef(GetDefaultFont(env));

    }

    env->DeleteLocalRef(target);

    return font;

}

jobject

AwtMenuItem::GetDefaultFont(JNIEnv *env) {

    if (AwtMenuItem::systemFont == NULL) {

        jclass cls = env->FindClass("sun/awt/windows/WMenuItemPeer");

        DASSERT(cls != NULL);

        AwtMenuItem::systemFont =

            env->CallStaticObjectMethod(cls, AwtMenuItem::getDefaultFontMID);

        DASSERT(AwtMenuItem::systemFont);

        AwtMenuItem::systemFont = env->NewGlobalRef(AwtMenuItem::systemFont);

    }

    return AwtMenuItem::systemFont;

}

void

AwtMenuItem::DrawSelf(DRAWITEMSTRUCT& drawInfo)

{

    JNIEnv *env = (JNIEnv *)JNU_GetEnv(jvm, JNI_VERSION_1_2);

    if (env->EnsureLocalCapacity(4) < 0) {

        return;

    }

    // self is sun.awt.windows.WMenuItemPeer

    jobject self = GetPeer(env);

    //  target is java.awt.MenuItem

    jobject target = env->GetObjectField(self, AwtObject::targetID);

    HDC hDC = drawInfo.hDC;

    RECT rect = drawInfo.rcItem;

    RECT textRect = rect;

    SIZE size;

    DWORD crBack,crText;

    HBRUSH hbrBack;

    jobject font = GetFont(env);

    jstring text = GetJavaString(env);

    size = AwtFont::getMFStringSize(hDC, font, text);

    /* 4700350: If the font size is taller than the menubar, change to the

     * default font.  Otherwise, menu text is painted over the title bar and

     * client area.  -bchristi

     */

    if (IsTopMenu() && size.cy > ::GetSystemMetrics(SM_CYMENU)) {

        env->DeleteLocalRef(font);

        font = env->NewLocalRef(GetDefaultFont(env));

        size = AwtFont::getMFStringSize(hDC, font, text);

    }

    /* Fix for bug 4257944 by ssi@sparc.spb.su

    * check state of the parent

    */

    AwtMenu* menu = GetMenuContainer();

    DASSERT(menu != NULL && GetID() >= 0);

    //Check whether the MenuItem is disabled.

    BOOL bEnabled = (jboolean)env->GetBooleanField(target,

                                                   AwtMenuItem::enabledID);

    if (menu != NULL) {

        bEnabled = bEnabled && !menu->IsDisabledAndPopup();

    }

    if ((drawInfo.itemState) & (ODS_SELECTED)) {

        // Set background and text colors for selected item

        crBack = ::GetSysColor (COLOR_HIGHLIGHT);

        // Disabled text must be drawn in gray.

        crText = ::GetSysColor(bEnabled? COLOR_HIGHLIGHTTEXT : COLOR_GRAYTEXT);

    } else {

        // COLOR_MENUBAR is only defined on WindowsXP. Our binaries are

        // built on NT, hence the below ifdef.

#ifndef COLOR_MENUBAR

#define COLOR_MENUBAR 30

#endif

        // Set background and text colors for unselected item

        if (IS_WINXP && IsTopMenu() && AwtDesktopProperties::IsXPStyle()) {

            crBack = ::GetSysColor (COLOR_MENUBAR);

        } else {

            crBack = ::GetSysColor (COLOR_MENU);

        }

        // Disabled text must be drawn in gray.

        crText = ::GetSysColor (bEnabled ? COLOR_MENUTEXT : COLOR_GRAYTEXT);

    }

    // Fill item rectangle with background color

    hbrBack = ::CreateSolidBrush (crBack);

    DASSERT(hbrBack);

    VERIFY(::FillRect (hDC, &rect, hbrBack));

    VERIFY(::DeleteObject (hbrBack));

    // Set current background and text colors

    ::SetBkColor (hDC, crBack);

    ::SetTextColor (hDC, crText);

    int nOldBkMode = ::SetBkMode(hDC, OPAQUE);

    DASSERT(nOldBkMode != 0);

    //draw check mark

    int checkWidth = ::GetSystemMetrics(SM_CXMENUCHECK);

    // Workaround for CR#6401956

    if (IS_WINVISTA) {

        AdjustCheckWidth(checkWidth);

    }

    if (IsCheckbox()) {

        // means that target is a java.awt.CheckboxMenuItem

        jboolean state =

            (jboolean)env->GetBooleanField(target, AwtMenuItem::stateID);

        if (state) {

            DASSERT(drawInfo.itemState & ODS_CHECKED);

            RECT checkRect;

            ::CopyRect(&checkRect, &textRect);

            if (GetRTL())

                checkRect.left = checkRect.right - checkWidth;

            else

                checkRect.right = checkRect.left + checkWidth;

            DrawCheck(hDC, checkRect);

        }

    }

    ::SetBkMode(hDC, TRANSPARENT);

    int x = 0;

    //draw string

    if (!IsTopMenu()){

        textRect.left += checkWidth;

        x = (GetRTL()) ? textRect.right - checkWidth - size.cx : textRect.left;

    } else {

        x = textRect.left = (textRect.left + textRect.right - size.cx) / 2;

    }

    int y = (textRect.top+textRect.bottom-size.cy)/2;

    // Text must be drawn in emboss if the Menu is disabled and not selected.

    BOOL bEmboss = !bEnabled && !(drawInfo.itemState & ODS_SELECTED);

    if (bEmboss) {

        ::SetTextColor(hDC, GetSysColor(COLOR_BTNHILIGHT));

        AwtFont::drawMFString(hDC, font, text, x + 1, y + 1, GetCodePage());

        ::SetTextColor(hDC, GetSysColor(COLOR_BTNSHADOW));

    }

    AwtFont::drawMFString(hDC, font, text, x, y, GetCodePage());

    jstring shortcutLabel =

        (jstring)env->GetObjectField(self, AwtMenuItem::shortcutLabelID);

    if (!IsTopMenu() && shortcutLabel != NULL) {

        UINT oldAlign = 0;

        if (GetRTL()){

            oldAlign = ::SetTextAlign(hDC, TA_LEFT);

            AwtFont::drawMFString(hDC, font, shortcutLabel, textRect.left, y,

                                  GetCodePage());

        } else {

            oldAlign = ::SetTextAlign(hDC, TA_RIGHT);

            AwtFont::drawMFString(hDC, font, shortcutLabel,

                                  textRect.right - checkWidth, y,

                                  GetCodePage());

        }

        ::SetTextAlign(hDC, oldAlign);

    }

    VERIFY(::SetBkMode(hDC,nOldBkMode));

    env->DeleteLocalRef(target);

    env->DeleteLocalRef(text);

    env->DeleteLocalRef(font);

    env->DeleteLocalRef(shortcutLabel);

}

/*

 * This function helps us to prevent check-mark's

 * distortion appeared due to changing of default

 * settings on Vista

 */

void AwtMenuItem::AdjustCheckWidth(int& checkWidth)

{

    if (checkWidth == SM_CXMENUCHECK_DEFAULT_ON_VISTA) {

        checkWidth = SM_CXMENUCHECK_DEFAULT_ON_XP;

    }

}

void AwtMenuItem::DrawItem(DRAWITEMSTRUCT& drawInfo)

{

    DASSERT(drawInfo.CtlType == ODT_MENU);

    if (drawInfo.itemID != m_Id)

        return;

    DrawSelf(drawInfo);

}

void AwtMenuItem::MeasureSelf(HDC hDC, MEASUREITEMSTRUCT& measureInfo)

{

    JNIEnv *env =(JNIEnv *)JNU_GetEnv(jvm, JNI_VERSION_1_2);

    if (env->EnsureLocalCapacity(4) < 0) {

        return;

    }

    /* self is a sun.awt.windows.WMenuItemPeer */

    jobject self = GetPeer(env);

    /* font is a java.awt.Font */

    jobject font = GetFont(env);

    jstring text = GetJavaString(env);

    SIZE size = AwtFont::getMFStringSize(hDC, font, text);

    /* 4700350: If the font size is taller than the menubar, change to the

     * default font.  Otherwise, menu text is painted over the title bar and

     * client area.  -bchristi

     */

    if (IsTopMenu() && size.cy > ::GetSystemMetrics(SM_CYMENU)) {

        jobject defFont = GetDefaultFont(env);

        env->DeleteLocalRef(font);

        font = env->NewLocalRef(defFont);

        size = AwtFont::getMFStringSize(hDC, font, text);

    }

    jstring fontName =

        (jstring)JNU_CallMethodByName(env, 0,font, "getName",

                                      "()Ljava/lang/String;").l;

    /* fontMetrics is a Hsun_awt_windows_WFontMetrics */

    jobject fontMetrics =  GetFontMetrics(env, font);

//     int height = env->GetIntField(fontMetrics, AwtFont::heightID);

    int height = (jint)JNU_CallMethodByName(env, 0, fontMetrics, "getHeight",

                                            "()I").i;

    measureInfo.itemHeight = height;

    measureInfo.itemHeight += measureInfo.itemHeight/3;

    // 3 is a heuristic number

    measureInfo.itemWidth = size.cx;

    if (!IsTopMenu()) {

        int checkWidth = ::GetSystemMetrics(SM_CXMENUCHECK);

        // Workaround for CR#6401956

        if (IS_WINVISTA) {

            AdjustCheckWidth(checkWidth);

        }

        measureInfo.itemWidth += checkWidth;

        // Add in shortcut width, if one exists.

        jstring shortcutLabel =

            (jstring)env->GetObjectField(self, AwtMenuItem::shortcutLabelID);

        if (shortcutLabel != NULL) {

            size = AwtFont::getMFStringSize(hDC, font, shortcutLabel);

            measureInfo.itemWidth += size.cx + checkWidth;

            env->DeleteLocalRef(shortcutLabel);

        }

    }

    env->DeleteLocalRef(text);

    env->DeleteLocalRef(font);

    env->DeleteLocalRef(fontName);

    env->DeleteLocalRef(fontMetrics);

}

void AwtMenuItem::MeasureItem(HDC hDC, MEASUREITEMSTRUCT& measureInfo)

{

    DASSERT(measureInfo.CtlType == ODT_MENU);

    if (measureInfo.itemID != m_Id)

        return;

    MeasureSelf(hDC, measureInfo);

}

jobject AwtMenuItem::GetFontMetrics(JNIEnv *env, jobject font)

{

    static jobject toolkit = NULL;

    if (toolkit == NULL) {

        if (env->PushLocalFrame(2) < 0)

            return NULL;

        jclass cls = env->FindClass("java/awt/Toolkit");

        jobject toolkitLocal =

            env->CallStaticObjectMethod(cls, AwtToolkit::getDefaultToolkitMID);

        toolkit = env->NewGlobalRef(toolkitLocal);

        DASSERT(!safe_ExceptionOccurred(env));

        env->PopLocalFrame(0);