ui.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * Subject to the GPL, version 2.
   4  */
   5
   6 #include <curses.h>
   7
   8 #include "ui.h"
   9 #include "xmalloc.h"
  10
  11 void ui_table_init(struct ui_table *tbl)
  12 {
  13         memset(tbl, 0, sizeof(*tbl));
  14
  15         getsyx(tbl->y, tbl->x);
  16
  17         tbl->rows_y  = tbl->y;
  18         tbl->width   = COLS;
  19         tbl->col_pad = 1;
  20
  21         INIT_LIST_HEAD(&tbl->cols);
  22 }
  23
  24 void ui_table_uninit(struct ui_table *tbl)
  25 {
  26         struct ui_col *col, *tmp;
  27
  28         list_for_each_entry_safe(col, tmp, &tbl->cols, entry)
  29                 xfree(col);
  30 }
  31
  32 void ui_table_pos_set(struct ui_table *tbl, int y, int x)
  33 {
  34         tbl->y      = y;
  35         tbl->x      = x;
  36         tbl->rows_y = y;
  37 }
  38
  39 static struct ui_col *ui_table_col_get(struct ui_table *tbl, uint32_t id)
  40 {
  41         struct ui_col *col;
  42
  43         list_for_each_entry(col, &tbl->cols, entry) {
  44                 if (col->id == id)
  45                         return col;
  46         }
  47
  48         bug();
  49 }
  50
  51 static void __ui_table_pos_update(struct ui_table *tbl)
  52 {
  53         struct ui_col *col;
  54         uint32_t pos = tbl->x;
  55
  56         list_for_each_entry(col, &tbl->cols, entry) {
  57                 col->pos  = pos;
  58                 pos      += col->len + tbl->col_pad;
  59         }
  60 }
  61
  62 void ui_table_col_add(struct ui_table *tbl, uint32_t id, char *name, uint32_t len)
  63 {
  64         struct ui_col *col = xzmalloc(sizeof(*col));
  65
  66         col->id    = id;
  67         col->name  = name;
  68         col->len   = len;
  69         col->align = UI_ALIGN_LEFT;
  70
  71         list_add_tail(&col->entry, &tbl->cols);
  72
  73         __ui_table_pos_update(tbl);
  74 }
  75
  76 void ui_table_col_color_set(struct ui_table *tbl, int col_id, int color)
  77 {
  78         struct ui_col *col = ui_table_col_get(tbl, col_id);
  79
  80         col->color = color;
  81 }
  82
  83 void ui_table_col_align_set(struct ui_table *tbl, int col_id, enum ui_align align)
  84 {
  85         struct ui_col *col = ui_table_col_get(tbl, col_id);
  86
  87         col->align = align;
  88 }
  89
  90 void ui_table_row_add(struct ui_table *tbl)
  91 {
  92         tbl->rows_y++;
  93 }
  94
  95 void ui_table_clear(struct ui_table *tbl)
  96 {
  97         tbl->rows_y = tbl->y;
  98 }
  99
 100 #define UI_ALIGN_COL(col) (((col)->align == UI_ALIGN_LEFT) ? "%-*.*s" : "%*.*s")
 101
 102 static void __ui_table_row_print(struct ui_table *tbl, struct ui_col *col,
 103                                  const char *str)
 104 {
 105         mvprintw(tbl->rows_y, col->pos, UI_ALIGN_COL(col), col->len, col->len, str);
 106         mvprintw(tbl->rows_y, col->pos + col->len, "%*s", tbl->col_pad, " ");
 107 }
 108
 109 void ui_table_row_print(struct ui_table *tbl, uint32_t col_id, const char *str)
 110 {
 111         struct ui_col *col = ui_table_col_get(tbl, col_id);
 112
 113         attron(col->color);
 114         __ui_table_row_print(tbl, col, str);
 115         attroff(col->color);
 116 }
 117
 118 void ui_table_header_color_set(struct ui_table *tbl, int color)
 119 {
 120         tbl->hdr_color = color;
 121 }
 122
 123 void ui_table_header_print(struct ui_table *tbl)
 124 {
 125         struct ui_col *col;
 126         int max_width = tbl->width;
 127         int width = 0;
 128
 129         attron(tbl->hdr_color);
 130
 131         mvprintw(tbl->y, tbl->x, "%-*.*s", max_width - tbl->x, max_width - tbl->x, "");
 132         mvprintw(tbl->y, tbl->x, "");
 133
 134         list_for_each_entry(col, &tbl->cols, entry) {
 135                 __ui_table_row_print(tbl, col, col->name);
 136                 width += col->len + tbl->col_pad;
 137         }
 138
 139         mvprintw(tbl->y, width, "%*s", max_width - width, " ");
 140         attroff(tbl->hdr_color);
 141 }