#include <curses.h>
#include <menu.h>
#include <stdlib.h>
#include <string.h>
#include <locale.h>

#define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
#define CTRLD 4

char *choices[] =
{
    "Choice 1",
    "Choice 2",
    "Choice 3",
    "Choice 4",
    "Choice 5",
    "Choice 6",
    "Choice 7",
    "Choice 8",
    "Choice 9",
    "Choice 10",
    "Exit",
    (char *) NULL,
};

void print_in_middle(WINDOW *win, int starty, int startx, int width, char
		     *string, chtype color);

int
main(void)
{
    ITEM **my_items;
    int c;
    MENU *my_menu;
    WINDOW *my_menu_win;
    int n_choices, i;
    /* Added by address@hidden [2014-03-03 18:31:25] */
    int ret, cur_idx, count;

    setlocale(LC_ALL, "");

    /* Initialize curses */
    initscr();
    start_color();
    cbreak();
    noecho();
    keypad(stdscr, TRUE);
    init_pair(1, COLOR_RED, COLOR_BLACK);
    init_pair(2, COLOR_CYAN, COLOR_BLACK);

    /* Create items */
    n_choices = ARRAY_SIZE(choices);
    my_items = (ITEM **) calloc(n_choices, sizeof(ITEM *));
    for (i = 0; i < n_choices; ++i)
	my_items[i] = new_item(choices[i], choices[i]);

    /* Crate menu */
    my_menu = new_menu((ITEM **) my_items);

    /* Create the window to be associated with the menu */
    my_menu_win = newwin(10, 40, 4, 4);
    keypad(my_menu_win, TRUE);

    /* Set main window and sub window */
    set_menu_win(my_menu, my_menu_win);
    set_menu_sub(my_menu, derwin(my_menu_win, 6, 38, 3, 1));
    set_menu_format(my_menu, 5, 1);

    /* Set menu mark to the string " * " */
    set_menu_mark(my_menu, " * ");

    /* Print a border around the main window and print a title */
    box(my_menu_win, 0, 0);
    print_in_middle(my_menu_win, 1, 0, 40, "My Menu", COLOR_PAIR(1));
    mvwaddch(my_menu_win, 2, 0, ACS_LTEE);
    mvwhline(my_menu_win, 2, 1, ACS_HLINE, 38);
    mvwaddch(my_menu_win, 2, 39, ACS_RTEE);

    /* Added by address@hidden [2014-03-03 18:31:12] */
    ret = set_current_item(my_menu, my_items[8]);
    cur_idx = item_index(current_item(my_menu));
    count = item_count(my_menu);
    if (cur_idx >= count - my_menu->arows) {
	/* It's ugly that we use the menu's internal member [menu->toprow],
	 * because ether 'set_current_item' or 'set_top_row' will modify
	 * both of the member [menu->toprow] and [menu->curitem], and the
	 * logic between the two member is what we wanted */
	my_menu->toprow = count - my_menu->arows;
    }

    /* Post the menu */
    post_menu(my_menu);
    wrefresh(my_menu_win);
    attron(COLOR_PAIR(2));
    mvprintw(LINES - 2, 0,
	     "Use PageUp and PageDown to scoll down or up a page of items");
    mvprintw(LINES - 1, 0, "Arrow Keys to navigate (F1 to Exit)");
    attroff(COLOR_PAIR(2));

    /* Added by address@hidden [2014-03-03 18:31:17] */
    mvprintw(LINES - 4, 0, "ret: %d, top: %d, cur_idx: %d, cur->name: %s",
	     ret, top_row(my_menu), item_index(current_item(my_menu)),
	     item_name(current_item(my_menu)));

    refresh();
    while ((c = wgetch(my_menu_win)) != KEY_F(1)) {
	switch (c) {
	case KEY_DOWN:
	    menu_driver(my_menu, REQ_DOWN_ITEM);
	    break;
	case KEY_UP:
	    menu_driver(my_menu, REQ_UP_ITEM);
	    break;
	case KEY_NPAGE:
	    menu_driver(my_menu, REQ_SCR_DPAGE);
	    break;
	case KEY_PPAGE:
	    menu_driver(my_menu, REQ_SCR_UPAGE);
	    break;
	}
	wrefresh(my_menu_win);
    }

    /* Unpost and free all the memory taken up */
    unpost_menu(my_menu);
    free_menu(my_menu);
    for (i = 0; i < n_choices; ++i)
	free_item(my_items[i]);
    endwin();
    return 0;
}
void
print_in_middle(WINDOW *win, int starty, int startx, int width, char
		*string, chtype color)
{
    int length, x, y;
    float temp;

    if (win == NULL)
	win = stdscr;
    getyx(win, y, x);
    if (startx != 0)
	x = startx;
    if (starty != 0)
	y = starty;
    if (width == 0)
	width = 80;
    length = strlen(string);
    temp = (width - length) / 2;
    x = startx + (int) temp;
    wattron(win, color);
    mvwprintw(win, y, x, "%s", string);
    wattroff(win, color);
    refresh();
}