/** @file
  UI Common Controls

;******************************************************************************
;* Copyright (c) 2014 - 2020, Insyde Software Corp. All Rights Reserved.
;*
;* You may not reproduce, distribute, publish, display, perform, modify, adapt,
;* transmit, broadcast, present, recite, release, license or otherwise exploit
;* any part of this publication in any form, by any means, without the prior
;* written permission of Insyde Software Corporation.
;*
;******************************************************************************
*/

#include "H2OControls.h"
#include "H2ODisplayEngineLocalMetro.h"
#include "MetroUi.h"

STATIC H2O_FRONT_PAGE_ITEM_CLASS    *mFrontPageItemClass = NULL;
#define CURRENT_CLASS              mFrontPageItemClass

#if FixedPcdGet32(PcdH2OLmdeMultiLayout) == 0
CHAR16 *mFrontPageItemChilds = L""
  L"<VerticalLayout padding='15,20,15,20' name='ItemLayout'>"
    L"<HorizontalLayout name='ImageLayout'>"
      L"<Label name='ImageLayoutLeft'/>"
      L"<Texture name='FrontPageImage' background-image-style='stretch|light'/>"
    L"</HorizontalLayout>"
    L"<Label name='menu_prompt' text-align='center' height='15' padding='8,0,0,0' font-size='15' textcolor='0xFFFFFFFF'/>"
    L"<Label height='7'/>"
    L"<Texture background-color='0xFFF4F4F4' name='separator' height='1'/>"
    L"<Label height='8'/>"
    L"<Label text-align='center' textcolor='0xFFFFFFFF' font-size='13' name='menu_help' height='26'/>"
  L"</VerticalLayout>";
#endif

#define PROMPT_HEIGHT       23
#define SEPARATOR_HEIGHT    16
#define HELP_HEIGHT         26

VOID
UpdateFrontPageItem (
  UI_CONTROL                    *Control,
  CONST RECT                    *Rc
  )
{
  H2O_FRONT_PAGE_ITEM        *This;
  H2O_FORM_BROWSER_S        *Statement;
  CHAR16                    Str[20];
  UI_CONTROL                *Child;
  INT32                     Width;
  INT32                     Height;
  INT32                     ImageSize;

  Width  = Rc->right   -  Rc->left;
  Height = Rc->bottom -  Rc->top;
  ASSERT (Width > HELP_HEIGHT + SEPARATOR_HEIGHT + PROMPT_HEIGHT + 10 + 40);
  ASSERT (Height > HELP_HEIGHT + SEPARATOR_HEIGHT + PROMPT_HEIGHT + 10 + 30);
  Width -= 40;
  Height -= 30;

  ImageSize = Height - HELP_HEIGHT - SEPARATOR_HEIGHT - PROMPT_HEIGHT;
  ImageSize = MIN (Width / 2, ImageSize);

  This      = (H2O_FRONT_PAGE_ITEM *) Control;
  Statement = This->Statement;
  ASSERT (Statement != NULL);
  if (Statement == NULL) {
    return;
  }

  if (Statement->Image != NULL) {
    Child = UiFindChildByName (This, L"FrontPageImage");
    UnicodeSPrint (Str, sizeof (Str), L"0x%p", Statement->Image);
    UiSetAttribute (Child, L"background-image", Str);
    UnicodeSPrint (Str, sizeof (Str), L"%d", ImageSize);
    UiSetAttribute (Child, L"width", Str);
    Child = UiFindChildByName (This, L"ImageLayout");
    UiSetAttribute (Child, L"height", Str);
    Child = UiFindChildByName (This, L"ImageLayoutLeft");
    UnicodeSPrint (Str, sizeof (Str), L"%d", (Width - ImageSize) / 2);
    UiSetAttribute (Child, L"width", Str);
  }

  if (Statement->Prompt != NULL) {
    Child = UiFindChildByName (This, L"menu_prompt");
    UiSetAttribute (Child, L"text", Statement->Prompt);
  }
  if (Statement->Help != NULL) {
    Child = UiFindChildByName (This, L"menu_help");
    UiSetAttribute (Child, L"text", Statement->Help);
  }
}

STATIC
COLORREF
GetImageColor (
  IN  EFI_IMAGE_INPUT                *ImageIn
  )
{
  UINTN                           Index;
  UINTN                           BufferLen;
  EFI_GRAPHICS_OUTPUT_BLT_PIXEL   *Pixel;

  ASSERT (ImageIn != NULL && ImageIn->Bitmap != NULL);

  BufferLen = ImageIn->Width * ImageIn->Height;

  for (Index = 0; Index < BufferLen; Index++) {
    if (ImageIn->Bitmap[Index].Reserved != 0) {
      Pixel = &ImageIn->Bitmap[Index];
      return (0xFF000000 | (Pixel->Red << 16) | (Pixel->Green << 8) | Pixel->Blue);
    }
  }
  return 0;
}

STATIC
VOID
UpdateFrontPageItemColor (
  IN H2O_FRONT_PAGE_ITEM    *FrontPageItem,
  IN BOOLEAN               MenuColor
  )
{
  CHAR16                  TextColor[20];
  H2O_FORM_BROWSER_S      *Statement;
  COLORREF                ImageColor;
  UI_CONTROL              *Child;

  Child = UiFindChildByName (FrontPageItem, L"ItemLayout");
  ImageColor = 0;
  Statement  = FrontPageItem->Statement;
  if (Statement->Image != NULL) {
    ImageColor = GetImageColor (Statement->Image);
  }
  if (MenuColor) {
    UnicodeSPrint (TextColor, sizeof (TextColor), L"0x%x", ImageColor);
    UiSetAttribute (Child, L"background-color", TextColor);
  } else {
    UiSetAttribute (Child, L"background-color", L"0x0");
  }
}

VOID
H2OFrontPageItemOnSetState (
  UI_CONTROL                    *Control,
  UI_STATE                      SetState,
  UI_STATE                      ClearState
  )
{
  UI_CONTROL                    *Child;

  PARENT_CONTROL_CLASS(CURRENT_CLASS)->SetState (Control, SetState, ClearState);

  if ((SetState & UISTATE_HOVER) || (ClearState & UISTATE_HOVER)) {
    UpdateFrontPageItemColor ((H2O_FRONT_PAGE_ITEM *)Control, (SetState & UISTATE_HOVER) ? TRUE : FALSE);
  }

  Child = UiFindChildByName (Control, L"HotBkg");
  if (Child == NULL) {
    return;
  }

  if ((SetState & UISTATE_SELECTED) == UISTATE_SELECTED) {
    UiSetAttribute (Child, L"visibility", L"true");
  } else if ((ClearState & UISTATE_SELECTED) == UISTATE_SELECTED) {
    UiSetAttribute (Child, L"visibility", L"false");
  }
}

VOID
EFIAPI
H2OFrontPageItemSetPosition (
  UI_CONTROL                    *Control,
  CONST RECT                    *Pos
  )
{
  UpdateFrontPageItem (Control, Pos);
  PARENT_CONTROL_CLASS(CURRENT_CLASS)->SetPosition (Control,Pos);
}


BOOLEAN
EFIAPI
H2OFrontPageItemSetAttribute (
  UI_CONTROL *Control,
  CHAR16     *Name,
  CHAR16     *Value
  )
{
  H2O_FRONT_PAGE_ITEM  *This;
  EFI_STATUS          Status;

  This = (H2O_FRONT_PAGE_ITEM *) Control;

  if (StrCmp (Name, L"statement") == 0) {
    This->Statement = (H2O_FORM_BROWSER_S *)(UINTN) StrToUInt (Value, 16, &Status);
  } else {
    return PARENT_CLASS_SET_ATTRIBUTE (CURRENT_CLASS, Control, Name, Value);
  }

  CONTROL_CLASS_INVALIDATE (This);

  return TRUE;
}


LRESULT
EFIAPI
H2OFrontPageItemProc (
  HWND   Hwnd,
  UINT32 Msg,
  WPARAM WParam,
  LPARAM LParam
  )
{
  H2O_FRONT_PAGE_ITEM            *This;
  UI_CONTROL                     *Control;
  UI_CONTROL                     *Child;

  This = (H2O_FRONT_PAGE_ITEM *) GetWindowLongPtr (Hwnd, 0);
  if (This == NULL && Msg != WM_CREATE && Msg != WM_NCCALCSIZE) {
    ASSERT (FALSE);
    return 0;
  }
  Control = (UI_CONTROL *)This;

  switch (Msg) {

  case WM_CREATE:
    This = (H2O_FRONT_PAGE_ITEM *) AllocateZeroPool (sizeof (H2O_FRONT_PAGE_ITEM));
    if (This != NULL) {
      CONTROL_CLASS (This) = (UI_CONTROL_CLASS *) GetClassLongPtr (Hwnd, 0);
      SetWindowLongPtr (Hwnd, 0, (INTN)This);
      SendMessage (Hwnd, UI_NOTIFY_CREATE, WParam, LParam);
    }
    break;

  case UI_NOTIFY_CREATE:
    PARENT_CLASS_WNDPROC (CURRENT_CLASS, Hwnd, UI_NOTIFY_CREATE, WParam, LParam);
    XmlCreateControl (mFrontPageItemChilds, Control);
    break;

  case UI_NOTIFY_PAINT:
    if (WParam == PAINT_BKCOLOR || WParam == PAINT_STATUSIMAGE) {
      break;
    }
    PARENT_CLASS_WNDPROC (CURRENT_CLASS, Hwnd, Msg, WParam, LParam);

    Child = UiFindChildByName (Control, L"HotBkg");
    if (Child != NULL) {
      break;
    }

    if (WParam == PAINT_ALL &&
        (CONTROL_CLASS_GET_STATE (This) & UISTATE_FOCUSED)) {
      COLORREF                  Color;
      UI_MANAGER                *Manager;
      RECT                      Rc;
      HDC                       Hdc;

      Color = 0xFFFF0000;
      Manager = Control->Manager;
      Hdc     = Manager->PaintDC;

      Manager->GetControlRect (Manager, Control, &Rc);

      SelectObject (Hdc, GetStockObject (PS_NULL));
      SelectObject (Hdc, GetStockObject (DC_PEN));
      SetDCPenColor (Hdc, Color);
      Rectangle (Hdc, Rc.left, Rc.top, Rc.right, Rc.bottom);
    }
    break;

  case WM_NCHITTEST:
    return HTTRANSPARENT;
    break;

  case WM_DESTROY:
    KillTimer (Hwnd, 0);
    return PARENT_CLASS_WNDPROC (CURRENT_CLASS, Hwnd, Msg, WParam, LParam);

  default:
    return PARENT_CLASS_WNDPROC (CURRENT_CLASS, Hwnd, Msg, WParam, LParam);
  }

  return 0;
}

H2O_FRONT_PAGE_ITEM_CLASS *
EFIAPI
GetFrontPageItemClass (
  VOID
  )
{
  if (CURRENT_CLASS != NULL) {
    return CURRENT_CLASS;
  }

  InitUiClass ((UI_CONTROL_CLASS **)&CURRENT_CLASS, sizeof (*CURRENT_CLASS), L"FrontPageItem", (UI_CONTROL_CLASS *) GetControlClass());
  if (CURRENT_CLASS == NULL) {
    return NULL;
  }
  ((UI_CONTROL_CLASS *)CURRENT_CLASS)->WndProc      = H2OFrontPageItemProc;
  ((UI_CONTROL_CLASS *)CURRENT_CLASS)->SetAttribute = H2OFrontPageItemSetAttribute;
  ((UI_CONTROL_CLASS *)CURRENT_CLASS)->SetPosition  = H2OFrontPageItemSetPosition;
  ((UI_CONTROL_CLASS *)CURRENT_CLASS)->SetState     = H2OFrontPageItemOnSetState;

  return CURRENT_CLASS;
}