lunaix-os/kernel/tty/tty.c

   1 #include <klibc/string.h>
   2 #include <lunaix/tty/tty.h>
   3 #include <lunaix/constants.h>
   4 #include <stdint.h>
   5
   6 #define TTY_WIDTH 80
   7 #define TTY_HEIGHT 25
   8
   9 vga_attribute* tty_vga_buffer = (vga_attribute*)VGA_BUFFER_PADDR;
  10
  11 vga_attribute tty_theme_color = VGA_COLOR_BLACK;
  12
  13 uint32_t tty_x = 0;
  14 uint16_t tty_y = 0;
  15
  16 void tty_init(void* vga_buf) {
  17     tty_vga_buffer = (vga_attribute*)vga_buf;
  18     tty_clear();
  19 }
  20
  21 void tty_set_buffer(void* vga_buf) {
  22     tty_vga_buffer = (vga_attribute*)vga_buf;
  23 }
  24
  25 void
  26 tty_set_theme(vga_attribute fg, vga_attribute bg)
  27 {
  28     tty_theme_color = (bg << 4 | fg) << 8;
  29 }
  30
  31 void
  32 tty_put_char(char chr)
  33 {
  34     switch (chr) {
  35         case '\t':
  36             tty_x += 4;
  37             break;
  38         case '\n':
  39             tty_y++;
  40             // fall through
  41         case '\r':
  42             tty_x = 0;
  43             break;
  44         default:
  45             *(tty_vga_buffer + tty_x + tty_y * TTY_WIDTH) = (tty_theme_color | chr);
  46             tty_x++;
  47             break;
  48     }
  49
  50     if (tty_x >= TTY_WIDTH) {
  51         tty_x = 0;
  52         tty_y++;
  53     }
  54     if (tty_y >= TTY_HEIGHT) {
  55         tty_scroll_up();
  56     }
  57 }
  58
  59 void
  60 tty_put_str(char* str)
  61 {
  62     while (*str != '\0') {
  63         tty_put_char(*str);
  64         str++;
  65     }
  66 }
  67
  68 void
  69 tty_scroll_up()
  70 {
  71     size_t last_line = TTY_WIDTH * (TTY_HEIGHT - 1);
  72     memcpy(tty_vga_buffer, tty_vga_buffer + TTY_WIDTH, last_line * 2);
  73     for (size_t i = 0; i < TTY_WIDTH; i++) {
  74         *(tty_vga_buffer + i + last_line) = tty_theme_color;
  75     }
  76     tty_y = tty_y == 0 ? 0 : TTY_HEIGHT - 1;
  77 }
  78
  79 void
  80 tty_clear()
  81 {
  82     for (uint32_t i = 0; i < TTY_WIDTH * TTY_HEIGHT; i++) {
  83         *(tty_vga_buffer + i) = tty_theme_color;
  84     }
  85     tty_x = 0;
  86     tty_y = 0;
  87 }
  88
  89 void
  90 tty_clear_line(unsigned int y) {
  91     for (size_t i = 0; i < TTY_WIDTH; i++)
  92     {
  93         *(tty_vga_buffer + i + y * TTY_WIDTH) = tty_theme_color;
  94     }
  95 }
  96
  97 void
  98 tty_set_cpos(unsigned int x, unsigned int y) {
  99     tty_x = x % TTY_WIDTH;
 100     tty_y = y % TTY_HEIGHT;
 101 }
 102
 103 void
 104 tty_get_cpos(unsigned int* x, unsigned int* y) {
 105     *x = tty_x;
 106     *y = tty_y;
 107 }
 108
 109 vga_attribute
 110 tty_get_theme() {
 111     return tty_theme_color;
 112 }