menu.c

   1
   2 #include <stdlib.h>
   3 #include <stddef.h>
   4 #include <string.h>
   5 #include <stdbool.h>
   6 #include <ncurses.h>
   7
   8 #include "headers/menu.h"
   9 #include "headers/render.h"
  10 #include "headers/utils.h"
  11
  12 #define MENU_PAD_TOP 2
  13 #define MENU_PAD_BOTTOM 1
  14 #define MENU_PAD_LEFT 2
  15 #define MENU_PAD_RIGHT 1
  16
  17 #define MAX_CONTENTS_LENGTH 256
  18
  19 typedef struct MenuItem {
  20     char* title;
  21     char* description;
  22     void* userdata;
  23 } MenuItem;
  24
  25 typedef struct Menu {
  26     char* menu_name;
  27     MenuItem** menu_items;
  28     int menu_length;
  29     int selected_item;
  30     int scroll_offset;
  31     bool focused;
  32     WINDOW* menu_win;
  33     WINDOW* sub_win;
  34     int max_height;
  35     int max_width;
  36     void* userdata;
  37 } Menu;
  38
  39 int swap_item(Menu* menu, int src_index, int dest_index);
  40
  41 /* insert mode */
  42 int menu_insert_mode(Menu* menu, int insert_index);
  43
  44 /* prob temp for now */
  45 MenuItem* create_blank_menuitem(void);
  46
  47
  48 MenuItem*
  49 create_menuitem(char* title)
  50 {
  51     MenuItem* new_menuitem;
  52
  53     new_menuitem = malloc(sizeof(MenuItem));
  54     new_menuitem->title = title;
  55     new_menuitem->description = strdup(""); //TEMP FOR NOW
  56
  57     return new_menuitem;
  58 }
  59
  60 MenuItem*
  61 create_blank_menuitem(void)
  62 {
  63     return create_menuitem(strdup(""));
  64 }
  65
  66 Menu*
  67 create_menu(char* menu_name, MenuItem** item_list)
  68 {
  69     Menu* new_menu;
  70
  71     new_menu = malloc(sizeof(Menu));
  72     new_menu->menu_name = menu_name;
  73     new_menu->menu_items = item_list;
  74     new_menu->menu_length = array_length(MenuItem*, item_list);
  75     new_menu->selected_item = 0;
  76     new_menu->scroll_offset = 0;
  77     new_menu->focused = false;
  78     set_menu_win(new_menu, stdscr);
  79
  80     return new_menu;
  81 }
  82
  83 /* getters */
  84 WINDOW*
  85 get_menu_win(Menu* menu)
  86 {
  87     return menu->menu_win;
  88 }
  89
  90 WINDOW*
  91 get_menu_subwin(Menu* menu)
  92 {
  93     return menu->sub_win;
  94 }
  95
  96 MenuItem*
  97 get_menu_item(Menu* menu, int index)
  98 {
  99     if (index < 0 || index >= menu->menu_length) return NULL;
 100
 101     return menu->menu_items[index];
 102 }
 103
 104 int
 105 get_selected_item(Menu* menu)
 106 {
 107     return menu->selected_item;
 108 }
 109
 110 int
 111 get_menu_length(Menu* menu)
 112 {
 113     return menu->menu_length;
 114 }
 115
 116 char*
 117 get_menu_name(Menu* menu)
 118 {
 119     return menu->menu_name;
 120 }
 121
 122 void*
 123 get_menu_userdata(Menu* menu)
 124 {
 125     return menu->userdata;
 126 }
 127
 128 char*
 129 get_menuitem_title(MenuItem* menuitem)
 130 {
 131     return menuitem->title;
 132 }
 133
 134 char*
 135 get_menuitem_descrip(MenuItem* menuitem)
 136 {
 137     return menuitem->description;
 138 }
 139
 140 void*
 141 get_menuitem_userdata(MenuItem* menuitem)
 142 {
 143     return menuitem->userdata;
 144 }
 145
 146
 147 /* setters */
 148 int
 149 set_menu_win(Menu* menu, WINDOW* win)
 150 {
 151     int height, width;
 152
 153     menu->menu_win = win;
 154     getmaxyx(menu->menu_win, height, width);
 155
 156     /* create a subwin (also prob free old subwin?) */
 157     menu->max_height = height-MENU_PAD_TOP-MENU_PAD_BOTTOM;
 158     menu->max_width = width-MENU_PAD_LEFT-MENU_PAD_RIGHT;
 159     menu->sub_win = derwin(
 160             menu->menu_win,
 161             menu->max_height,
 162             menu->max_width,
 163             MENU_PAD_TOP,
 164             MENU_PAD_LEFT
 165     );
 166
 167     return 0;
 168 }
 169
 170 int
 171 set_selected_item(Menu* menu, int selected_item)
 172 {
 173     menu->selected_item = selected_item;
 174     return 0;
 175 }
 176
 177 int
 178 set_menu_focus(Menu* menu, bool focus)
 179 {
 180     menu->focused = focus;
 181     return 0;
 182 }
 183
 184 int
 185 set_menu_userdata(Menu* menu, void* userdata)
 186 {
 187     menu->userdata = userdata;
 188     return 0;
 189 }
 190
 191 int
 192 set_menuitem_descrip(MenuItem* menuitem, char* descrip)
 193 {
 194     menuitem->description = descrip;
 195     return 0;
 196 }
 197
 198 int
 199 set_menuitem_userdata(MenuItem* menuitem, void* userdata)
 200 {
 201     menuitem->userdata = userdata;
 202     return 0;
 203 }
 204
 205
 206 int
 207 swap_item(Menu* menu, int src_index, int dest_index)
 208 {
 209     ar_swap_item((void**)menu->menu_items, src_index, dest_index);
 210
 211     return 0;
 212 }
 213
 214 int
 215 delete_item(Menu* menu, int index)
 216 {
 217     if (index < 0 || index > menu->menu_length-1) return -1;
 218
 219     for (int i = index; i <= menu->menu_length-1; i++) {
 220         menu->menu_items[i] = menu->menu_items[i+1];
 221     }
 222
 223     menu->menu_items = realloc(menu->menu_items, menu->menu_length*sizeof(MenuItem*));
 224     menu->menu_items[menu->menu_length-1] = 0; // preserve null at end
 225
 226     menu->menu_length -= 1;
 227
 228     /* also move the current selected position if it's last */
 229     if (menu->selected_item > menu->menu_length-1) {
 230         menu->selected_item = menu->menu_length-1;
 231     }
 232
 233     return 0;
 234 }
 235
 236 int
 237 insert_item(Menu* menu, MenuItem* menuitem, int index)
 238 { // note, this func does not validate index
 239
 240     /* resize array and insert */
 241     menu->menu_items = realloc(menu->menu_items, (menu->menu_length+2)*sizeof(MenuItem*));
 242
 243     for (int i = menu->menu_length; i > index; i--) {
 244         menu->menu_items[i] = menu->menu_items[i-1];
 245     }
 246
 247     menu->menu_items[index] = menuitem;
 248     menu->menu_items[menu->menu_length+1] = 0; // remember null at end
 249     menu->menu_length += 1;
 250
 251     /* move cursor pos */
 252     menu->selected_item = index;
 253
 254     return 0;
 255 }
 256
 257 int
 258 menu_insert_mode(Menu* menu, int insert_index)
 259 {
 260     char temp[MAX_CONTENTS_LENGTH+1]; // remember null
 261     char* new_contents;
 262
 263     curs_on();
 264
 265     /* move cursor to right spot */
 266     ungetstr(menu->menu_items[insert_index]->title);
 267     mvwgetnstr(menu->sub_win,
 268         insert_index, // account for wrap later too
 269         0,
 270         temp,
 271         MAX_CONTENTS_LENGTH
 272     );
 273     curs_off();
 274
 275     /* copy out */
 276     new_contents = strdup(temp);
 277     menu->menu_items[insert_index]->title = new_contents;
 278
 279     /* delete if empty - maybe move this to a cleanup stage */
 280     if (strlen(new_contents) == 0) {
 281         delete_item(menu, insert_index);
 282     }
 283
 284     return 0;
 285 }
 286
 287 int
 288 menu_driver(Menu* menu, MenuAction action)
 289 {
 290
 291     switch (action) {
 292         case MENU_UP:
 293             menu->selected_item = menu->selected_item-1 >= 0 ? menu->selected_item-1 : 0;
 294             break;
 295
 296         case MENU_DOWN:
 297             menu->selected_item = menu->selected_item+1 <= menu->menu_length-1 ? menu->selected_item+1 : menu->menu_length-1;
 298             break;
 299
 300         case MENU_TOP:
 301             menu->selected_item = 0;
 302             break;
 303
 304         case MENU_BOTTOM:
 305             menu->selected_item = menu->menu_length-1;
 306             break;
 307
 308         case MENU_MOVE_UP:
 309             if (menu->selected_item <= 0) break;
 310             swap_item(menu, menu->selected_item, menu->selected_item-1);
 311             menu->selected_item -= 1;
 312             break;
 313
 314         case MENU_MOVE_DOWN:
 315             if (menu->selected_item >= menu->menu_length-1) break;
 316             swap_item(menu, menu->selected_item, menu->selected_item+1);
 317             menu->selected_item += 1;
 318             break;
 319
 320         case MENU_DELETE:
 321             delete_item(menu, menu->selected_item);
 322             break;
 323
 324         case MENU_APPEND:
 325             insert_item(menu, create_blank_menuitem(), menu->menu_length);
 326             menu_insert_mode(menu, menu->selected_item);
 327             break;
 328
 329         case MENU_INSERT_ABOVE:
 330             insert_item(menu, create_blank_menuitem(), menu->selected_item);
 331             menu_insert_mode(menu, menu->selected_item);
 332             break;
 333
 334         case MENU_INSERT_BELOW:
 335             insert_item(menu, create_blank_menuitem(), menu->selected_item+1);
 336             menu_insert_mode(menu, menu->selected_item); // inserted item is cur now
 337             break;
 338
 339         case MENU_EDIT:
 340             menu_insert_mode(menu, menu->selected_item);
 341             break;
 342
 343         default: // This is here for debug, disable later
 344             fprintf(stderr, "Invalid menu action");
 345     }
 346
 347     return 0;
 348 }
 349
 350 int
 351 render_menu(Menu* menu)
 352 {
 353     /* draw outer menu (prob dont need this every render) */
 354     /* wclear(menu->menu_win); */
 355     wattron(menu->menu_win, COLOR_PAIR(
 356         (menu->focused == true) ?
 357         TS_MENU_SELECTED: TS_MENU_NONSELECTED
 358     ));
 359     mvwprintw(menu->menu_win, 0, MENU_PAD_LEFT, menu->menu_name);
 360     wattroff(menu->menu_win, COLOR_PAIR(0));
 361
 362     /* draw inner menu */
 363     wclear(menu->sub_win);
 364
 365     int cur_line = 0;
 366     for (int i = 0; i < menu->menu_length-menu->scroll_offset; i++) {
 367
 368         MenuItem* curitem;
 369         int wrapped_lines;
 370         char* wrapped_text;
 371
 372         curitem = menu->menu_items[i];
 373
 374         /* wrap text by inserting newlines (maxwidth-1 for newline char)*/
 375         wrapped_text = wrap_text(curitem->title, menu->max_width-1, &wrapped_lines);
 376
 377         /* color selected item */
 378         wattron(menu->sub_win, COLOR_PAIR(
 379            (i == menu->selected_item && menu->focused == true) ?
 380            TS_SELECTED : TS_NONSELECTED
 381         ));
 382         mvwprintw(menu->sub_win, cur_line, 0, wrapped_text);
 383         wattroff(menu->sub_win, COLOR_PAIR(0));
 384
 385         cur_line += wrapped_lines;
 386
 387         /* display number of items */
 388         if (strlen(curitem->description) > 0) {
 389             mvwprintw(menu->sub_win, cur_line, 0, curitem->description);
 390             cur_line += 1;
 391         }
 392
 393         free(wrapped_text);
 394
 395     }
 396
 397     wrefresh(menu->sub_win);
 398     wrefresh(menu->menu_win);
 399
 400     return 0;
 401 }
 402
 403 int
 404 free_menu(Menu* menu)
 405 {
 406     return 0;
 407 }
 408