Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-05-15 01:31:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[FREELDR] Deduplicate common message-box creation code.

Browse source 
Also, make the (T)uiSave/RestoreScreen() helpers manage themselves the
temporary memory buffer used to store the temporary screen snapshot.
This commit is contained in:
Hermès Bélusca-Maïto 2024-03-11 21:44:45 +01:00
parent 64155019df
commit bbce6c3fdf
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
4 changed files with 148 additions and 127 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
boot/freeldr/freeldr/include/ui/gui.h
View file

@ -31,8 +31,17 @@ VOID GuiDrawText(ULONG X, ULONG Y, PUCHAR Text, UCHAR Attr); // Draws text
VOID GuiDrawText2(ULONG X, ULONG Y, ULONG MaxNumChars, PUCHAR Text, UCHAR Attr); // Draws text at coordinates specified
VOID GuiDrawStatusText(PCSTR StatusText); // Draws text at the very bottom line on the screen
VOID GuiUpdateDateTime(VOID); // Updates the date and time
VOID GuiSaveScreen(PUCHAR Buffer); // Saves the screen so that it can be restored later
VOID GuiRestoreScreen(PUCHAR Buffer); // Restores the screen from a previous save
/* Saves the screen so that it can be restored later */
_Ret_maybenull_
__drv_allocatesMem(Mem)
PUCHAR
GuiSaveScreen(VOID);
/* Restores the screen from a previous save */
VOID
GuiRestoreScreen(
_In_opt_ __drv_freesMem(Mem) PUCHAR Buffer);
/* Displays a message box on the screen with an ok button */
VOID

13
boot/freeldr/freeldr/include/ui/tui.h
View file

@ -101,8 +101,17 @@ TuiDrawCenteredText(
VOID TuiDrawStatusText(PCSTR StatusText); // Draws text at the very bottom line on the screen
VOID TuiUpdateDateTime(VOID); // Updates the date and time
VOID TuiSaveScreen(PUCHAR Buffer); // Saves the screen so that it can be restored later
VOID TuiRestoreScreen(PUCHAR Buffer); // Restores the screen from a previous save
/* Saves the screen so that it can be restored later */
_Ret_maybenull_
__drv_allocatesMem(Mem)
PUCHAR
TuiSaveScreen(VOID);
/* Restores the screen from a previous save */
VOID
TuiRestoreScreen(
_In_opt_ __drv_freesMem(Mem) PUCHAR Buffer);
/* Displays a message box on the screen with an ok button */
VOID

10
boot/freeldr/freeldr/ui/gui.c
View file

@ -51,11 +51,17 @@ VOID GuiUpdateDateTime(VOID)
{
}
VOID GuiSaveScreen(PUCHAR Buffer)
_Ret_maybenull_
__drv_allocatesMem(Mem)
PUCHAR
GuiSaveScreen(VOID)
{
return NULL;
}
VOID GuiRestoreScreen(PUCHAR Buffer)
VOID
GuiRestoreScreen(
_In_opt_ __drv_freesMem(Mem) PUCHAR Buffer)
{
}

239
boot/freeldr/freeldr/ui/tui.c
View file

@ -19,6 +19,14 @@
#include <freeldr.h>
typedef struct _SMALL_RECT
{
SHORT Left;
SHORT Top;
SHORT Right;
SHORT Bottom;
} SMALL_RECT, *PSMALL_RECT;
PVOID TextVideoBuffer = NULL;
/* GENERIC TUI UTILS *********************************************************/
@ -606,72 +614,76 @@ VOID TuiUpdateDateTime(VOID)
Buffer, ATTR(UiTitleBoxFgColor, UiTitleBoxBgColor));
}
VOID TuiSaveScreen(PUCHAR Buffer)
_Ret_maybenull_
__drv_allocatesMem(Mem)
PUCHAR
TuiSaveScreen(VOID)
{
PUCHAR ScreenMemory = (PUCHAR)TextVideoBuffer;
ULONG i;
PUCHAR Buffer;
PUCHAR ScreenMemory = (PUCHAR)TextVideoBuffer;
ULONG i;
/* Allocate the buffer */
Buffer = FrLdrTempAlloc(UiScreenWidth * UiScreenHeight * 2,
TAG_TUI_SCREENBUFFER);
if (!Buffer)
return NULL;
/* Loop through each cell and copy it */
for (i=0; i < (UiScreenWidth * UiScreenHeight * 2); i++)
{
Buffer[i] = ScreenMemory[i];
}
return Buffer;
}
VOID TuiRestoreScreen(PUCHAR Buffer)
VOID
TuiRestoreScreen(
_In_opt_ __drv_freesMem(Mem) PUCHAR Buffer)
{
PUCHAR ScreenMemory = (PUCHAR)TextVideoBuffer;
ULONG i;
PUCHAR ScreenMemory = (PUCHAR)TextVideoBuffer;
ULONG i;
if (!Buffer)
return;
/* Loop through each cell and copy it */
for (i=0; i < (UiScreenWidth * UiScreenHeight * 2); i++)
{
ScreenMemory[i] = Buffer[i];
}
/* Free the buffer */
FrLdrTempFree(Buffer, TAG_TUI_SCREENBUFFER);
VideoCopyOffScreenBufferToVRAM();
}
VOID
TuiMessageBox(
_In_ PCSTR MessageText)
static VOID
TuiDrawMsgBoxCommon(
_In_ PCSTR MessageText,
_Out_ PSMALL_RECT MsgBoxRect)
{
PVOID ScreenBuffer;
INT width = 8;
ULONG height = 1;
INT curline = 0;
INT k;
size_t i, j;
INT x1, x2, y1, y2;
CHAR temp[260];
// Save the screen contents
ScreenBuffer = FrLdrTempAlloc(UiScreenWidth * UiScreenHeight * 2,
TAG_TUI_SCREENBUFFER);
TuiSaveScreen(ScreenBuffer);
// Display the message box
TuiMessageBoxCritical(MessageText);
// Restore the screen contents
TuiRestoreScreen(ScreenBuffer);
FrLdrTempFree(ScreenBuffer, TAG_TUI_SCREENBUFFER);
}
VOID
TuiMessageBoxCritical(
_In_ PCSTR MessageText)
{
int width = 8;
unsigned int height = 1;
int curline = 0;
int k;
size_t i , j;
int x1, x2, y1, y2;
char temp[260];
char key;
// Find the height
for (i=0; i<strlen(MessageText); i++)
/* Find the height */
for (i = 0; i < strlen(MessageText); i++)
{
if (MessageText[i] == '\n')
height++;
}
// Find the width
for (i=0,j=0,k=0; i<height; i++)
/* Find the width */
for (i = j = k = 0; i < height; i++)
{
while ((MessageText[j] != '\n') && (MessageText[j] != 0))
while ((MessageText[j] != '\n') && (MessageText[j] != ANSI_NULL))
{
j++;
k++;
@ -684,34 +696,71 @@ TuiMessageBoxCritical(
j++;
}
// Calculate box area
/* Calculate box area */
x1 = (UiScreenWidth - (width+2))/2;
x2 = x1 + width + 3;
y1 = ((UiScreenHeight - height - 2)/2) + 1;
y2 = y1 + height + 4;
// Draw the box
TuiDrawBox(x1, y1, x2, y2, D_VERT, D_HORZ, TRUE, TRUE, ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
MsgBoxRect->Left = x1; MsgBoxRect->Right = x2;
MsgBoxRect->Top = y1; MsgBoxRect->Bottom = y2;
// Draw the text
for (i=0,j=0; i<strlen(MessageText)+1; i++)
/* Draw the box */
TuiDrawBox(x1, y1, x2, y2, D_VERT, D_HORZ, TRUE, TRUE,
ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
/* Draw the text */
for (i = j = 0; i < strlen(MessageText) + 1; i++)
{
if ((MessageText[i] == '\n') || (MessageText[i] == 0))
if ((MessageText[i] == '\n') || (MessageText[i] == ANSI_NULL))
{
temp[j] = 0;
j = 0;
UiDrawText(x1+2, y1+1+curline, temp, ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
UiDrawText(x1 + 2, y1 + 1 + curline, temp,
ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
curline++;
}
else
{
temp[j++] = MessageText[i];
}
}
}
// Draw OK button
strcpy(temp, " OK ");
UiDrawText(x1+((x2-x1)/2)-3, y2-2, temp, ATTR(COLOR_BLACK, COLOR_GRAY));
VOID
TuiMessageBox(
_In_ PCSTR MessageText)
{
PVOID ScreenBuffer;
// Draw status text
/* Save the screen contents */
ScreenBuffer = TuiSaveScreen();
/* Display the message box */
TuiMessageBoxCritical(MessageText);
/* Restore the screen contents */
TuiRestoreScreen(ScreenBuffer);
}
VOID
TuiMessageBoxCritical(
_In_ PCSTR MessageText)
{
SMALL_RECT BoxRect;
CHAR key;
/* Draw the common parts of the message box */
TuiDrawMsgBoxCommon(MessageText, &BoxRect);
/* Draw centered OK button */
UiDrawText((BoxRect.Left + BoxRect.Right) / 2 - 3,
BoxRect.Bottom - 2,
" OK ",
ATTR(COLOR_BLACK, COLOR_GRAY));
/* Draw status text */
UiDrawStatusText("Press ENTER to continue");
VideoCopyOffScreenBufferToVRAM();
@ -975,80 +1024,29 @@ VOID TuiFadeOut(VOID)
BOOLEAN TuiEditBox(PCSTR MessageText, PCHAR EditTextBuffer, ULONG Length)
{
INT width = 8;
ULONG height = 1;
INT curline = 0;
INT k;
size_t i , j;
INT x1, x2, y1, y2;
CHAR temp[260];
CHAR key;
BOOLEAN Extended;
INT EditBoxLine;
ULONG EditBoxStartX, EditBoxEndX;
INT EditBoxCursorX;
ULONG EditBoxTextLength, EditBoxTextPosition;
INT EditBoxTextDisplayIndex;
BOOLEAN ReturnCode;
PVOID ScreenBuffer;
BOOLEAN Extended;
INT EditBoxLine;
ULONG EditBoxStartX, EditBoxEndX;
INT EditBoxCursorX;
ULONG EditBoxTextLength, EditBoxTextPosition;
INT EditBoxTextDisplayIndex;
BOOLEAN ReturnCode;
SMALL_RECT BoxRect;
PVOID ScreenBuffer;
// Save the screen contents
ScreenBuffer = FrLdrTempAlloc(UiScreenWidth * UiScreenHeight * 2,
TAG_TUI_SCREENBUFFER);
TuiSaveScreen(ScreenBuffer);
/* Save the screen contents */
ScreenBuffer = TuiSaveScreen();
// Find the height
for (i=0; i<strlen(MessageText); i++)
{
if (MessageText[i] == '\n')
height++;
}
// Find the width
for (i=0,j=0,k=0; i<height; i++)
{
while ((MessageText[j] != '\n') && (MessageText[j] != 0))
{
j++;
k++;
}
if (k > width)
width = k;
k = 0;
j++;
}
// Calculate box area
x1 = (UiScreenWidth - (width+2))/2;
x2 = x1 + width + 3;
y1 = ((UiScreenHeight - height - 2)/2) + 1;
y2 = y1 + height + 4;
// Draw the box
TuiDrawBox(x1, y1, x2, y2, D_VERT, D_HORZ, TRUE, TRUE, ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
// Draw the text
for (i=0,j=0; i<strlen(MessageText)+1; i++)
{
if ((MessageText[i] == '\n') || (MessageText[i] == 0))
{
temp[j] = 0;
j = 0;
UiDrawText(x1+2, y1+1+curline, temp, ATTR(UiMessageBoxFgColor, UiMessageBoxBgColor));
curline++;
}
else
temp[j++] = MessageText[i];
}
/* Draw the common parts of the message box */
TuiDrawMsgBoxCommon(MessageText, &BoxRect);
EditBoxTextLength = (ULONG)strlen(EditTextBuffer);
EditBoxTextLength = min(EditBoxTextLength, Length - 1);
EditBoxTextPosition = 0;
EditBoxLine = y2 - 2;
EditBoxStartX = x1 + 3;
EditBoxEndX = x2 - 3;
EditBoxLine = BoxRect.Bottom - 2;
EditBoxStartX = BoxRect.Left + 3;
EditBoxEndX = BoxRect.Right - 3;
// Draw the edit box background and the text
UiFillArea(EditBoxStartX, EditBoxLine, EditBoxEndX, EditBoxLine, ' ', ATTR(UiEditBoxTextColor, UiEditBoxBgColor));
@ -1199,9 +1197,8 @@ BOOLEAN TuiEditBox(PCSTR MessageText, PCHAR EditTextBuffer, ULONG Length)
// Hide the cursor again
MachVideoHideShowTextCursor(FALSE);
// Restore the screen contents
/* Restore the screen contents */
TuiRestoreScreen(ScreenBuffer);
FrLdrTempFree(ScreenBuffer, TAG_TUI_SCREENBUFFER);
return ReturnCode;
}