diff options
Diffstat (limited to 'Library/UefiShellUfmCommandLib/tbi/win.c')
| -rw-r--r-- | Library/UefiShellUfmCommandLib/tbi/win.c | 290 |
1 files changed, 290 insertions, 0 deletions
diff --git a/Library/UefiShellUfmCommandLib/tbi/win.c b/Library/UefiShellUfmCommandLib/tbi/win.c new file mode 100644 index 0000000..3d82e58 --- /dev/null +++ b/Library/UefiShellUfmCommandLib/tbi/win.c @@ -0,0 +1,290 @@ +#include <Uefi.h> +#include <Library/UefiLib.h> +#include <Library/DebugLib.h> +#include <Library/ShellLib.h> +#include <Library/PrintLib.h> // UnicodeVSPrint() +#include <Library/BaseMemoryLib.h> +#include <Library/MemoryAllocationLib.h> + +#include "screen.h" +#include "win.h" + +#define CHECK_POSITION(win, x, y) \ + if(x < 0 || y < 0 || x >= win->width || y >= win->height) \ + return FALSE + +#define SET_WINDOW_CHAR(win, col, row, character, attrib) \ + win->text[row][col] = character; \ + win->attr[row][col] = attrib + +#define SET_WINDOW_CHAR2(win, col, row, character) \ + SET_WINDOW_CHAR(win, col, row, character, win->cur_attr) + +struct window *newwin(struct screen *s, + INT32 ncols, INT32 nlines, INT32 begin_x, INT32 begin_y) +{ + INTN x, y; + struct window *win; + + ASSERT(s != NULL); + + if(ncols <= 0 || nlines <= 0 || begin_x < 0 || begin_y < 0) + return NULL; + + if((begin_x + ncols) > s->columns || (begin_y + nlines) > s->lines) + return NULL; + + win = AllocateZeroPool(sizeof(struct window)); + if(!win) + return NULL; + + win->text = AllocatePool(nlines * sizeof(CHAR16 *)); + if(!win->text) { + delwin(win); + return NULL; + } + + win->attr = AllocatePool(nlines * sizeof(INT32 *)); + if(!win->attr) { + delwin(win); + return NULL; + } + + win->scr = s; + win->curx = 0; + win->cury = 0; + win->begx = begin_x; + win->begy = begin_y; + win->width = ncols; + win->height = nlines; + win->cur_attr = s->attr; + + for(y = 0; y < nlines; y++) { + win->text[y] = AllocatePool((ncols + 1) * sizeof(CHAR16)); + win->attr[y] = AllocatePool((ncols + 1) * sizeof(INT32)); + if(!win->text[y] || !win->attr[y]) { + delwin(win); + return NULL; + } + + for(x = 0; x < ncols; x++) { + win->text[y][x] = L' '; + win->attr[y][x] = win->cur_attr; + } + win->text[y][x] = CHAR_NULL; + win->attr[y][x] = win->cur_attr; + } + + return win; +} + +VOID delwin(struct window *w) +{ + INTN i; + ASSERT(w != NULL); + + if(w->text) { + for(i = 0; i < w->height; i++) { + if(w->text[i]) + FreePool(w->text[i]); + } + FreePool(w->text); + } + + if(w->attr) { + for(i = 0; i < w->height; i++) { + if(w->attr[i]) + FreePool(w->attr[i]); + } + FreePool(w->attr); + } + + FreePool(w); +} + +VOID wattrset(struct window *w, INT32 attr) +{ + ASSERT(w != NULL); + + w->cur_attr = attr; +} + +VOID wattroff(struct window *w) +{ + ASSERT(w != NULL); + + w->cur_attr = w->scr->attr; +} + +BOOLEAN wmove(struct window *w, INT32 x, INT32 y) +{ + ASSERT(w != NULL); + CHECK_POSITION(w, x, y); + + w->curx = x; + w->cury = y; + return TRUE; +} + +BOOLEAN wborder(struct window *w, CHAR16 ls, CHAR16 rs, CHAR16 ts, + CHAR16 bs, CHAR16 tl, CHAR16 tr, CHAR16 bl, CHAR16 br) +{ + INT32 i, endx, endy; + ASSERT(w != NULL); + + endx = w->width - 1; + endy = w->height - 1; + + for(i = 0; i <= endx; i++) { + SET_WINDOW_CHAR2(w, i, 0, ts); + SET_WINDOW_CHAR2(w, i, endy, bs); + } + + for(i = 0; i <= endy; i++) { + SET_WINDOW_CHAR2(w, 0, i, ls); + SET_WINDOW_CHAR2(w, endx, i, rs); + } + + SET_WINDOW_CHAR2(w, 0, 0, tl); + SET_WINDOW_CHAR2(w, endx, 0, tr); + SET_WINDOW_CHAR2(w, 0, endy, bl); + SET_WINDOW_CHAR2(w, endx, endy, br); + return TRUE; +} + +BOOLEAN mvwhline(struct window *w, INT32 x, INT32 y, CHAR16 ch, INT32 n) +{ + UINTN i, length; + ASSERT(w != NULL); + CHECK_POSITION(w, x, y); + + length = w->width - x; + if(length > n) + length = n; + + length += x; + for(i = x; i < length; i++) { + w->text[y][i] = ch; + w->attr[y][i] = w->cur_attr; + } + return TRUE; +} + +BOOLEAN mvwvline(struct window *w, INT32 x, INT32 y, CHAR16 ch, INT32 n) +{ + UINTN i, length; + ASSERT(w != NULL); + CHECK_POSITION(w, x, y); + + length = w->height - y; + if(length > n) + length = n; + + length += y; + for(i = y; i < length; i++) { + w->text[i][x] = ch; + w->attr[i][x] = w->cur_attr; + } + return TRUE; +} + +UINTN EFIAPI wprintf(struct window *w, CONST CHAR16 *fmt, ...) +{ + VA_LIST arg; + UINTN retval; + + VA_START(arg, fmt); + retval = wvprintf(w, fmt, arg); + VA_END(arg); + return retval; +} + +UINTN EFIAPI mvwprintf(struct window *w, INT32 x, INT32 y, CONST CHAR16 *fmt, ...) +{ + VA_LIST arg; + UINTN retval; + BOOLEAN moved; + + moved = wmove(w, x, y); + if(moved == FALSE) + return 0; + + VA_START(arg, fmt); + retval = wvprintf(w, fmt, arg); + VA_END(arg); + return retval; +} + +UINTN EFIAPI wvprintf(struct window *w, CONST CHAR16 *fmt, VA_LIST args) +{ + INT32 x, y; + UINTN i, length, max_length, walker_size; + CHAR16 *fmt_walker; + + ASSERT(w != NULL); + ASSERT(fmt != NULL); + + x = w->curx; + y = w->cury; + + walker_size = w->width * sizeof(CHAR16); + fmt_walker = AllocateZeroPool(walker_size); + if(!fmt_walker) + return 0; + + UnicodeVSPrint(fmt_walker, walker_size, fmt, args); + length = StrLen(fmt_walker); + max_length = w->width - x; + if(length >= max_length) { + length = max_length; + + w->curx = 0; + if(++w->cury >= w->height) + w->cury = 0; + } + else + w->curx += length; + + CopyMem(w->text[y] + x, fmt_walker, length * 2); // multiply by 2 because CHAR16 + for(i = 0; i < length; i++) + w->attr[y][x + i] = w->cur_attr; + FreePool(fmt_walker); + return length; +} + +VOID wrefresh(struct window *w) +{ + UINTN x, y; + INT32 attributes; + CHAR16 tmp_ch, *print_ptr; + EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *stdout; + + ASSERT(w != NULL); + + attributes = w->attr[0][0]; + stdout = w->scr->stdout; + stdout->SetAttribute(stdout, attributes); + + for(y = 0; y < w->height; y++) { + stdout->SetCursorPosition(stdout, w->begx, w->begy + y); + print_ptr = w->text[y]; + + for(x = 0; x <= (w->width); x++) { + if(w->text[y][x] == CHAR_NULL || attributes != w->attr[y][x]) + { + tmp_ch = w->text[y][x]; + w->text[y][x] = CHAR_NULL; + stdout->OutputString(stdout, print_ptr); + w->text[y][x] = tmp_ch; + + print_ptr = &(w->text[y][x]); + if(w->text[y][x] != CHAR_NULL) { + attributes = w->attr[y][x]; + stdout->SetAttribute(stdout, attributes); + } + } + } + } + + stdout->SetAttribute(stdout, w->scr->attr); +} |