CloverBootloader/rEFIt_UEFI/libeg/XPointer.cpp

/*
 * a class for mouse support
 */
#ifdef __cplusplus
extern "C" {
#endif
#include "Library/BaseLib.h"
#ifdef __cplusplus
}
#endif
#include "XPointer.h"
#include "libegint.h"   //this includes platform.h 
#include "../refit/screen.h"

#ifndef DEBUG_ALL
#define DEBUG_MOUSE 1
#else
#define DEBUG_MOUSE DEBUG_ALL
#endif

#if DEBUG_MOUSE == 0
#define DBG(...)
#else
#define DBG(...) DebugLog(DEBUG_MOUSE, __VA_ARGS__)
#endif

//extern EFI_AUDIO_IO_PROTOCOL *AudioIo;

// make them theme dependent? No, 32 is good value for all.
#define POINTER_WIDTH  32
#define POINTER_HEIGHT 32

XPointer::XPointer()
{
  PointerImage = new XImage(BuiltinIcon(BUILTIN_ICON_POINTER));

  oldImage = new XImage(POINTER_WIDTH, POINTER_HEIGHT);
  newImage = new XImage(POINTER_WIDTH, POINTER_HEIGHT);

}

XPointer::~XPointer()
{
  delete PointerImage;
  delete newImage;
  delete oldImage;
}

void XPointer::Hide()
{
  if (SimplePointerProtocol) {
    oldImage->Draw(oldPlace.XPos, oldPlace.YPos, 1.f);
  }
}

EFI_STATUS XPointer::MouseBirth()
{
  EFI_STATUS Status = EFI_UNSUPPORTED;

  if (!gSettings.PointerEnabled) {
    return EFI_SUCCESS;
  }

  if (SimplePointerProtocol) { //do not double
//    DBG("DrawPointer\n");
    DrawPointer();
    return EFI_SUCCESS;
  }

  // Try first to use mouse from System Table
  Status = gBS->HandleProtocol(gST->ConsoleInHandle, &gEfiSimplePointerProtocolGuid, (VOID**)&SimplePointerProtocol);
  if (EFI_ERROR(Status)) {
    // not found, so use the first found device
    DBG("MouseBirth: No mouse at ConIn, checking if any other device exists\n");
    Status = gBS->LocateProtocol(&gEfiSimplePointerProtocolGuid, NULL, (VOID**)&SimplePointerProtocol);
  }

  if (EFI_ERROR(Status)) {
    PointerImage = NULL;
    MouseEvent = NoEvents;
    SimplePointerProtocol = NULL;
    MsgLog("No mouse!\n");
    gSettings.PointerEnabled = FALSE;
    return Status;
  }

  PointerImage = new XImage(BuiltinIcon(BUILTIN_ICON_POINTER));
  if (!PointerImage) {
    //this is impossible after BuiltinIcon
    DBG("No pointer image!\n");
    SimplePointerProtocol = NULL;
    return EFI_NOT_FOUND;
  }
  LastClickTime = 0; //AsmReadTsc();
  oldPlace.XPos = (INTN)(UGAWidth >> 2);
  oldPlace.YPos = (INTN)(UGAHeight >> 2);
  oldPlace.Width = POINTER_WIDTH;
  oldPlace.Height = POINTER_HEIGHT;
  CopyMem(&newPlace, &oldPlace, sizeof(EG_RECT));

  //newImage->Fill(&MenuBackgroundPixel),
  //  egTakeImage(oldImage, oldPlace.XPos, oldPlace.YPos,
  //              POINTER_WIDTH, POINTER_HEIGHT); // DrawPointer repeats it
  DrawPointer();
  MouseEvent = NoEvents;
  return Status;
}

VOID XPointer::DrawPointer()
{

// take background image
  oldImage->GetArea(newPlace);
  CopyMem(&oldPlace, &newPlace, sizeof(EG_RECT));  //can we use oldPlace = newPlace; ?

//  CopyMem(newImage->PixelData, oldImage->PixelData, (UINTN)(POINTER_WIDTH * POINTER_HEIGHT * sizeof(EG_PIXEL)));
  newImage->CopyScaled(*oldImage, 1.f);

  newImage->Compose(0, 0, *PointerImage, true);
  newImage->Draw(newPlace.XPos, newPlace.YPos, 1.f);
}

VOID XPointer::KillMouse()
{
  //  EG_PIXEL pi;
  if (!SimplePointerProtocol) {
    return;
  }
 
  delete newImage;
  delete oldImage;

  delete PointerImage;

  MouseEvent = NoEvents;
  SimplePointerProtocol = NULL;
}

// input - tsc
// output - milliseconds
// the caller is responsible for t1 > t0
UINT64 XPointer::TimeDiff(UINT64 t0, UINT64 t1)
{
  return DivU64x64Remainder((t1 - t0), DivU64x32(gCPUStructure.TSCFrequency, 1000), 0);
}

VOID XPointer::UpdatePointer()
{
  UINT64                    Now;
  EFI_STATUS                Status;
  EFI_SIMPLE_POINTER_STATE	tmpState;
  EFI_SIMPLE_POINTER_MODE   *CurrentMode;
  INTN                      ScreenRelX;
  INTN                      ScreenRelY;

  //  Now = gRT->GetTime(&Now, NULL);
  Now = AsmReadTsc();
  Status = SimplePointerProtocol->GetState(SimplePointerProtocol, &tmpState);
  if (!EFI_ERROR(Status)) {
    if (!State.LeftButton && tmpState.LeftButton) // press left
      MouseEvent = LeftMouseDown;
    else if (!State.RightButton && tmpState.RightButton) // press right
      MouseEvent = RightMouseDown;
    else if (State.LeftButton && !tmpState.LeftButton) { //release left
      // time for double click 500ms into menu
      if (TimeDiff(LastClickTime, Now) < gSettings.DoubleClickTime)
        MouseEvent = DoubleClick;
      else
        MouseEvent = LeftClick;
      LastClickTime = Now;
    }
    else if (State.RightButton && !tmpState.RightButton) //release right
      MouseEvent = RightClick;
    else if (State.RelativeMovementZ > 0)
      MouseEvent = ScrollDown;
    else if (State.RelativeMovementZ < 0)
      MouseEvent = ScrollUp;
    else if (State.RelativeMovementX || State.RelativeMovementY)
      MouseEvent = MouseMove;
    else
      MouseEvent = NoEvents;

    CopyMem(&State, &tmpState, sizeof(EFI_SIMPLE_POINTER_STATE));
    CurrentMode = SimplePointerProtocol->Mode;

    ScreenRelX = ((UGAWidth * State.RelativeMovementX / (INTN)CurrentMode->ResolutionX) * gSettings.PointerSpeed) >> 10;
    if (gSettings.PointerMirror) {
      newPlace.XPos -= ScreenRelX;
    }
    else {
      newPlace.XPos += ScreenRelX;
    }
    if (newPlace.XPos < 0) newPlace.XPos = 0;
    if (newPlace.XPos > UGAWidth - 1) newPlace.XPos = UGAWidth - 1;

    //    YPosPrev = newPlace.YPos;
    ScreenRelY = ((UGAHeight * State.RelativeMovementY / (INTN)CurrentMode->ResolutionY) * gSettings.PointerSpeed) >> 10;
    newPlace.YPos += ScreenRelY;
    if (newPlace.YPos < 0) newPlace.YPos = 0;
    if (newPlace.YPos > UGAHeight - 1) newPlace.YPos = UGAHeight - 1;

    HidePointer();
    DrawPointer();
  }
}

bool XPointer::MouseInRect(EG_RECT *Place)
{
  return  ((newPlace.XPos >= Place->XPos) &&
    (newPlace.XPos < (Place->XPos + (INTN)Place->Width)) &&
    (newPlace.YPos >= Place->YPos) &&
    (newPlace.YPos < (Place->YPos + (INTN)Place->Height)));
}

EFI_STATUS XPointer::CheckMouseEvent(REFIT_MENU_SCREEN *Screen)
{
  EFI_STATUS Status = EFI_TIMEOUT;
  //  INTN EntryId;

  Screen->mAction = ActionNone;

  if (!Screen) {
    return EFI_TIMEOUT;
  }

  if (!IsDragging && MouseEvent == MouseMove)
    MouseEvent = NoEvents;

  // if (MouseEvent != NoEvents){
  if (ScrollEnabled && MouseInRect(&UpButton) && MouseEvent == LeftClick)
    Screen->mAction = ActionScrollUp;
  else if (ScrollEnabled && MouseInRect(&DownButton) && MouseEvent == LeftClick)
    Screen->mAction = ActionScrollDown;
  else if (ScrollEnabled && MouseInRect(&Scrollbar) && MouseEvent == LeftMouseDown) {
    IsDragging = TRUE;
    ScrollbarYMovement = 0;
    ScrollbarOldPointerPlace.XPos = ScrollbarNewPointerPlace.XPos = newPlace.XPos;
    ScrollbarOldPointerPlace.YPos = ScrollbarNewPointerPlace.YPos = newPlace.YPos;
  }
  else if (ScrollEnabled && IsDragging && MouseEvent == LeftClick) {
    IsDragging = FALSE;
  }
  else if (ScrollEnabled && IsDragging && MouseEvent == MouseMove) {
    Screen->mAction = ActionMoveScrollbar;
    ScrollbarNewPointerPlace.XPos = newPlace.XPos;
    ScrollbarNewPointerPlace.YPos = newPlace.YPos;
  }
  else if (ScrollEnabled && MouseInRect(&ScrollbarBackground) &&
    MouseEvent == LeftClick) {
    if (newPlace.YPos < Scrollbar.YPos) // up
      Screen->mAction = ActionPageUp;
    else // down
      Screen->mAction = ActionPageDown;
    // page up/down, like in OS X
  }
  else if (ScrollEnabled &&
    MouseEvent == ScrollDown) {
    Screen->mAction = ActionScrollDown;
  }
  else if (ScrollEnabled &&
    MouseEvent == ScrollUp) {
    Screen->mAction = ActionScrollUp;
  }
  else {
    for (UINTN EntryId = 0; EntryId < Screen->Entries.size(); EntryId++) {
      if (MouseInRect(&(Screen->Entries[EntryId].Place))) {
        switch (MouseEvent) {
        case LeftClick:
          Screen->mAction = Screen->Entries[EntryId].AtClick;
          //          DBG("Click\n");
          break;
        case RightClick:
          Screen->mAction = Screen->Entries[EntryId].AtRightClick;
          break;
        case DoubleClick:
          Screen->mAction = Screen->Entries[EntryId].AtDoubleClick;
          break;
        case ScrollDown:
          Screen->mAction = ActionScrollDown;
          break;
        case ScrollUp:
          Screen->mAction = ActionScrollUp;
          break;
        case MouseMove:
          Screen->mAction = Screen->Entries[EntryId].AtMouseOver;
          //how to do the action once?
          break;
        default:
          Screen->mAction = ActionNone;
          break;
        }
        Screen->mItemID = EntryId;
        break;
      }
      else { //click in milk
        switch (MouseEvent) {
        case LeftClick:
          Screen->mAction = ActionDeselect;
          break;
        case RightClick:
          Screen->mAction = ActionFinish;
          break;
        case ScrollDown:
          Screen->mAction = ActionScrollDown;
          break;
        case ScrollUp:
          Screen->mAction = ActionScrollUp;
          break;
        default:
          Screen->mAction = ActionNone;
          break;
        }
        Screen->mItemID = 0xFFFF;
      }
    }
  }
  // }
  if (Screen->mAction != ActionNone) {
    Status = EFI_SUCCESS;
    MouseEvent = NoEvents; //clear event as set action
  }
  return Status;
}