diff --git a/util/console.c b/util/console.c index 8c9401c..53fc5d0 100644 --- a/util/console.c +++ b/util/console.c @@ -41,6 +41,28 @@ static int grub_console_attr = A_NORMAL; +grub_uint8_t grub_console_cur_color = 7; + +static grub_uint8_t grub_console_standard_color = 0x7; +static grub_uint8_t grub_console_normal_color = 0x7; +static grub_uint8_t grub_console_highlight_color = 0x70; + +#define NUM_COLORS 8 + +static grub_uint8_t color_map[NUM_COLORS] = +{ + COLOR_BLACK, + COLOR_BLUE, + COLOR_GREEN, + COLOR_CYAN, + COLOR_RED, + COLOR_MAGENTA, + COLOR_YELLOW, + COLOR_WHITE +}; + +static int use_color; + static void grub_ncurses_putchar (grub_uint32_t c) { @@ -100,24 +122,46 @@ grub_ncurses_setcolorstate (grub_term_color_state state) switch (state) { case GRUB_TERM_COLOR_STANDARD: + grub_console_cur_color = grub_console_standard_color; grub_console_attr = A_NORMAL; break; case GRUB_TERM_COLOR_NORMAL: + grub_console_cur_color = grub_console_normal_color; grub_console_attr = A_NORMAL; break; case GRUB_TERM_COLOR_HIGHLIGHT: + grub_console_cur_color = grub_console_highlight_color; grub_console_attr = A_STANDOUT; break; default: break; } + + if (use_color) + { + grub_uint8_t fg, bg; + + fg = (grub_console_cur_color & 7); + bg = (grub_console_cur_color >> 4) & 7; + + grub_console_attr = (grub_console_cur_color & 8) ? A_BOLD : A_NORMAL; + color_set ((bg << 3) + fg, 0); + } } /* XXX: This function is never called. */ static void grub_ncurses_setcolor (grub_uint8_t normal_color, grub_uint8_t highlight_color) { - color_set (normal_color << 8 | highlight_color, 0); + grub_console_normal_color = normal_color; + grub_console_highlight_color = highlight_color; +} + +static void +grub_ncurses_getcolor (grub_uint8_t *normal_color, grub_uint8_t *highlight_color) +{ + *normal_color = grub_console_normal_color; + *highlight_color = grub_console_highlight_color; } static int saved_char = ERR; @@ -272,7 +316,23 @@ grub_ncurses_init (void) nonl (); intrflush (stdscr, FALSE); keypad (stdscr, TRUE); - start_color (); + + if (has_colors ()) + { + start_color (); + + if ((COLORS >= NUM_COLORS) && (COLOR_PAIRS >= NUM_COLORS * NUM_COLORS)) + { + int i, j, n; + + n = 0; + for (i = 0; i < NUM_COLORS; i++) + for (j = 0; j < NUM_COLORS; j++) + init_pair(n++, color_map[j], color_map[i]); + + use_color = 1; + } + } return 0; } @@ -300,6 +360,7 @@ static struct grub_term grub_ncurses_term = .cls = grub_ncurses_cls, .setcolorstate = grub_ncurses_setcolorstate, .setcolor = grub_ncurses_setcolor, + .getcolor = grub_ncurses_getcolor, .setcursor = grub_ncurses_setcursor, .refresh = grub_ncurses_refresh, .flags = 0,