/*
- * refit/screen.c
+ * refind/screen.c
* Screen handling functions
*
* Copyright (c) 2006 Christoph Pfisterer
#include "global.h"
#include "screen.h"
#include "config.h"
-#include "refit_call_wrapper.h"
+#include "libegint.h"
+#include "../include/refit_call_wrapper.h"
-#include "egemb_refind_banner.h"
+#include "../include/egemb_refind_banner.h"
// Console defines and variables
UINTN ConHeight;
CHAR16 *BlankLine;
-static VOID SwitchToText(IN BOOLEAN CursorEnabled);
-static VOID SwitchToGraphics(VOID);
static VOID DrawScreenHeader(IN CHAR16 *Title);
// UGA defines and variables
VOID InitScreen(VOID)
{
UINTN i;
-
+
// initialize libeg
egInitScreen();
-
+
if (egHasGraphicsMode()) {
egGetScreenSize(&UGAWidth, &UGAHeight);
AllowGraphicsMode = TRUE;
egSetGraphicsModeEnabled(FALSE); // just to be sure we are in text mode
}
GraphicsScreenDirty = TRUE;
-
+
// disable cursor
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, FALSE);
-
+
// get size of text console
if (refit_call4_wrapper(ST->ConOut->QueryMode, ST->ConOut, ST->ConOut->Mode->Mode, &ConWidth, &ConHeight) != EFI_SUCCESS) {
// use default values on error
ConWidth = 80;
ConHeight = 25;
}
-
+
// make a buffer for a whole text line
BlankLine = AllocatePool((ConWidth + 1) * sizeof(CHAR16));
for (i = 0; i < ConWidth; i++)
BlankLine[i] = ' ';
BlankLine[i] = 0;
-
+
// show the banner (even when in graphics mode)
DrawScreenHeader(L"Initializing...");
}
// switch to text mode if requested
AllowGraphicsMode = FALSE;
SwitchToText(FALSE);
-
+
} else if (AllowGraphicsMode) {
// clear screen and show banner
// (now we know we'll stay in graphics mode)
+ if ((GlobalConfig.RequestedScreenWidth > 0) && (GlobalConfig.RequestedScreenHeight > 0) &&
+ egSetScreenSize(GlobalConfig.RequestedScreenWidth, GlobalConfig.RequestedScreenHeight)) {
+ UGAWidth = GlobalConfig.RequestedScreenWidth;
+ UGAHeight = GlobalConfig.RequestedScreenHeight;
+ } // if user requested a particular screen resolution
SwitchToGraphics();
BltClearScreen(TRUE);
}
}
-static VOID SwitchToText(IN BOOLEAN CursorEnabled)
+VOID SwitchToText(IN BOOLEAN CursorEnabled)
{
egSetGraphicsModeEnabled(FALSE);
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, CursorEnabled);
}
-static VOID SwitchToGraphics(VOID)
+VOID SwitchToGraphics(VOID)
{
if (AllowGraphicsMode && !egIsGraphicsModeEnabled()) {
egSetGraphicsModeEnabled(TRUE);
{
DrawScreenHeader(Title);
SwitchToText(FALSE);
-
+
// reset error flag
haveError = FALSE;
}
SwitchToText(FALSE);
PauseForKey();
}
-
+
// reset error flag
haveError = FALSE;
}
VOID BeginExternalScreen(IN BOOLEAN UseGraphicsMode, IN CHAR16 *Title)
{
+ EG_PIXEL DarkBackgroundPixel = { 0x0, 0x0, 0x0, 0 };
+
if (!AllowGraphicsMode)
UseGraphicsMode = FALSE;
-
+
if (UseGraphicsMode) {
SwitchToGraphics();
BltClearScreen(FALSE);
- }
-
+ } else {
+ egClearScreen(&DarkBackgroundPixel);
+ DrawScreenHeader(Title);
+ } // if/else
+
// show the header
- DrawScreenHeader(Title);
-
+// DrawScreenHeader(Title);
+
if (!UseGraphicsMode)
SwitchToText(TRUE);
-
+
// reset error flag
haveError = FALSE;
}
{
// make sure we clean up later
GraphicsScreenDirty = TRUE;
-
+
if (haveError) {
SwitchToText(FALSE);
PauseForKey();
}
-
+
// reset error flag
haveError = FALSE;
}
// clear text screen
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
refit_call1_wrapper(ST->ConOut->ClearScreen, ST->ConOut);
-
+
// enable cursor
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, TRUE);
}
BOOLEAN GotKeyStrokes;
EFI_STATUS Status;
EFI_INPUT_KEY key;
-
+
GotKeyStrokes = FALSE;
for (;;) {
Status = refit_call2_wrapper(ST->ConIn->ReadKeyStroke, ST->ConIn, &key);
VOID PauseForKey(VOID)
{
UINTN index;
-
+
Print(L"\n* Hit any key to continue *");
-
+
if (ReadAllKeyStrokes()) { // remove buffered key strokes
refit_call1_wrapper(BS->Stall, 5000000); // 5 seconds delay
ReadAllKeyStrokes(); // empty the buffer again
}
-
+
refit_call3_wrapper(BS->WaitForEvent, 1, &ST->ConIn->WaitForKey, &index);
ReadAllKeyStrokes(); // empty the buffer to protect the menu
-
+
Print(L"\n");
}
// show console and wait for key
SwitchToText(FALSE);
PauseForKey();
-
+
// reset error flag
haveError = FALSE;
}
VOID EndlessIdleLoop(VOID)
{
UINTN index;
-
+
for (;;) {
ReadAllKeyStrokes();
refit_call3_wrapper(BS->WaitForEvent, 1, &ST->ConIn->WaitForKey, &index);
// Error handling
//
+#ifdef __MAKEWITH_GNUEFI
BOOLEAN CheckFatalError(IN EFI_STATUS Status, IN CHAR16 *where)
{
CHAR16 ErrorName[64];
-
+
if (!EFI_ERROR(Status))
return FALSE;
-
+
StatusToString(ErrorName, Status);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR);
Print(L"Fatal Error: %s %s\n", ErrorName, where);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
haveError = TRUE;
-
+
//BS->Exit(ImageHandle, ExitStatus, ExitDataSize, ExitData);
-
+
return TRUE;
}
BOOLEAN CheckError(IN EFI_STATUS Status, IN CHAR16 *where)
{
CHAR16 ErrorName[64];
-
+
if (!EFI_ERROR(Status))
return FALSE;
-
+
StatusToString(ErrorName, Status);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR);
Print(L"Error: %s %s\n", ErrorName, where);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
haveError = TRUE;
-
+
return TRUE;
}
+#else
+BOOLEAN CheckFatalError(IN EFI_STATUS Status, IN CHAR16 *where)
+{
+// CHAR16 ErrorName[64];
+
+ if (!EFI_ERROR(Status))
+ return FALSE;
+
+// StatusToString(ErrorName, Status);
+ gST->ConOut->SetAttribute (gST->ConOut, ATTR_ERROR);
+ Print(L"Fatal Error: %r %s\n", Status, where);
+ gST->ConOut->SetAttribute (gST->ConOut, ATTR_BASIC);
+ haveError = TRUE;
+
+ //gBS->Exit(ImageHandle, ExitStatus, ExitDataSize, ExitData);
+
+ return TRUE;
+}
+
+BOOLEAN CheckError(IN EFI_STATUS Status, IN CHAR16 *where)
+{
+// CHAR16 ErrorName[64];
+
+ if (!EFI_ERROR(Status))
+ return FALSE;
+
+// StatusToString(ErrorName, Status);
+ gST->ConOut->SetAttribute (gST->ConOut, ATTR_ERROR);
+ Print(L"Error: %r %s\n", Status, where);
+ gST->ConOut->SetAttribute (gST->ConOut, ATTR_BASIC);
+ haveError = TRUE;
+
+ return TRUE;
+}
+#endif
//
// Graphics functions
if (Banner != NULL)
MenuBackgroundPixel = Banner->PixelData[0];
}
-
+
// clear and draw banner
egClearScreen(&MenuBackgroundPixel);
if (Banner != NULL)
BltImage(Banner, (UGAWidth - Banner->Width) >> 1,
((UGAHeight - LAYOUT_TOTAL_HEIGHT) >> 1) + LAYOUT_BANNER_HEIGHT - Banner->Height);
-
+
} else {
// clear to standard background color
egClearScreen(&StdBackgroundPixel);
}
-
+
GraphicsScreenDirty = FALSE;
}
VOID BltImageAlpha(IN EG_IMAGE *Image, IN UINTN XPos, IN UINTN YPos, IN EG_PIXEL *BackgroundPixel)
{
EG_IMAGE *CompImage;
-
+
// compose on background
CompImage = egCreateFilledImage(Image->Width, Image->Height, FALSE, BackgroundPixel);
egComposeImage(CompImage, Image, 0, 0);
-
+
// blit to screen and clean up
egDrawImage(CompImage, XPos, YPos);
egFreeImage(CompImage);
// CompHeight = TotalHeight;
// OffsetY = (TotalHeight - CompHeight) >> 1;
// egComposeImage(CompImage, TopImage, OffsetX, OffsetY);
-//
+//
// // blit to screen and clean up
// egDrawImage(CompImage, XPos, YPos);
// egFreeImage(CompImage);
VOID BltImageCompositeBadge(IN EG_IMAGE *BaseImage, IN EG_IMAGE *TopImage, IN EG_IMAGE *BadgeImage, IN UINTN XPos, IN UINTN YPos)
{
- UINTN TotalWidth, TotalHeight, CompWidth = 0, CompHeight = 0, OffsetX = 0, OffsetY = 0;
+ UINTN TotalWidth = 0, TotalHeight = 0, CompWidth = 0, CompHeight = 0, OffsetX = 0, OffsetY = 0;
EG_IMAGE *CompImage = NULL;
// initialize buffer with base image
OffsetY = (TotalHeight - CompHeight) >> 1;
egComposeImage(CompImage, TopImage, OffsetX, OffsetY);
}
-
+
// place the badge image
if (BadgeImage != NULL && CompImage != NULL && (BadgeImage->Width + 8) < CompWidth && (BadgeImage->Height + 8) < CompHeight) {
OffsetX += CompWidth - 8 - BadgeImage->Width;
egFreeImage(CompImage);
GraphicsScreenDirty = TRUE;
}
+
+// Line-editing functions borrowed from gummiboot (cursor_left(), cursor_right(), & line_edit()).
+// gummiboot is copyright (c) 2012 by Kay Sievers <kay.sievers@vrfy.org> and Harald Hoyer
+// <harald@redhat.com> and is licensed under the LGPL 2.1.
+
+static void cursor_left(UINTN *cursor, UINTN *first)
+{
+ if ((*cursor) > 0)
+ (*cursor)--;
+ else if ((*first) > 0)
+ (*first)--;
+}
+
+static void cursor_right(UINTN *cursor, UINTN *first, UINTN x_max, UINTN len)
+{
+ if ((*cursor)+2 < x_max)
+ (*cursor)++;
+ else if ((*first) + (*cursor) < len)
+ (*first)++;
+}
+
+BOOLEAN line_edit(CHAR16 *line_in, CHAR16 **line_out, UINTN x_max, UINTN y_pos) {
+ CHAR16 *line;
+ UINTN size;
+ UINTN len;
+ UINTN first;
+ CHAR16 *print;
+ UINTN cursor;
+ BOOLEAN exit;
+ BOOLEAN enter;
+
+ if (!line_in)
+ line_in = L"";
+ size = StrLen(line_in) + 1024;
+ line = AllocatePool(size * sizeof(CHAR16));
+ StrCpy(line, line_in);
+ len = StrLen(line);
+ print = AllocatePool(x_max * sizeof(CHAR16));
+
+ refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, TRUE);
+
+ first = 0;
+ cursor = 0;
+ enter = FALSE;
+ exit = FALSE;
+ while (!exit) {
+ UINTN index;
+ EFI_STATUS err;
+ EFI_INPUT_KEY key;
+ UINTN i;
+
+ i = len - first;
+ if (i >= x_max-2)
+ i = x_max-2;
+ CopyMem(print, line + first, i * sizeof(CHAR16));
+ print[i++] = ' ';
+ print[i] = '\0';
+
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, 0, y_pos);
+ refit_call2_wrapper(ST->ConOut->OutputString, ST->ConOut, print);
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, cursor, y_pos);
+
+ refit_call3_wrapper(BS->WaitForEvent, 1, &ST->ConIn->WaitForKey, &index);
+ err = refit_call2_wrapper(ST->ConIn->ReadKeyStroke, ST->ConIn, &key);
+ if (EFI_ERROR(err))
+ continue;
+
+ switch (key.ScanCode) {
+ case SCAN_ESC:
+ exit = TRUE;
+ break;
+ case SCAN_HOME:
+ cursor = 0;
+ first = 0;
+ continue;
+ case SCAN_END:
+ cursor = len;
+ if (cursor >= x_max) {
+ cursor = x_max-2;
+ first = len - (x_max-2);
+ }
+ continue;
+ case SCAN_UP:
+ while((first + cursor) && line[first + cursor] == ' ')
+ cursor_left(&cursor, &first);
+ while((first + cursor) && line[first + cursor] != ' ')
+ cursor_left(&cursor, &first);
+ while((first + cursor) && line[first + cursor] == ' ')
+ cursor_left(&cursor, &first);
+ if (first + cursor != len && first + cursor)
+ cursor_right(&cursor, &first, x_max, len);
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, cursor, y_pos);
+ continue;
+ case SCAN_DOWN:
+ while(line[first + cursor] && line[first + cursor] == ' ')
+ cursor_right(&cursor, &first, x_max, len);
+ while(line[first + cursor] && line[first + cursor] != ' ')
+ cursor_right(&cursor, &first, x_max, len);
+ while(line[first + cursor] && line[first + cursor] == ' ')
+ cursor_right(&cursor, &first, x_max, len);
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, cursor, y_pos);
+ continue;
+ case SCAN_RIGHT:
+ if (first + cursor == len)
+ continue;
+ cursor_right(&cursor, &first, x_max, len);
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, cursor, y_pos);
+ continue;
+ case SCAN_LEFT:
+ cursor_left(&cursor, &first);
+ refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, cursor, y_pos);
+ continue;
+ case SCAN_DELETE:
+ if (len == 0)
+ continue;
+ if (first + cursor == len)
+ continue;
+ for (i = first + cursor; i < len; i++)
+ line[i] = line[i+1];
+ line[len-1] = ' ';
+ len--;
+ continue;
+ }
+
+ switch (key.UnicodeChar) {
+ case CHAR_LINEFEED:
+ case CHAR_CARRIAGE_RETURN:
+ if (StrCmp(line, line_in) != 0) {
+ *line_out = line;
+ line = NULL;
+ }
+ enter = TRUE;
+ exit = TRUE;
+ break;
+ case CHAR_BACKSPACE:
+ if (len == 0)
+ continue;
+ if (first == 0 && cursor == 0)
+ continue;
+ for (i = first + cursor-1; i < len; i++)
+ line[i] = line[i+1];
+ len--;
+ if (cursor > 0)
+ cursor--;
+ if (cursor > 0 || first == 0)
+ continue;
+ /* show full line if it fits */
+ if (len < x_max-2) {
+ cursor = first;
+ first = 0;
+ continue;
+ }
+ /* jump left to see what we delete */
+ if (first > 10) {
+ first -= 10;
+ cursor = 10;
+ } else {
+ cursor = first;
+ first = 0;
+ }
+ continue;
+ case '\t':
+ case ' ' ... '~':
+ case 0x80 ... 0xffff:
+ if (len+1 == size)
+ continue;
+ for (i = len; i > first + cursor; i--)
+ line[i] = line[i-1];
+ line[first + cursor] = key.UnicodeChar;
+ len++;
+ line[len] = '\0';
+ if (cursor+2 < x_max)
+ cursor++;
+ else if (first + cursor < len)
+ first++;
+ continue;
+ }
+ }
+
+ refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, FALSE);
+ FreePool(print);
+ FreePool(line);
+ return enter;
+} /* BOOLEAN line_edit() */