source: mainline/uspace/lib/ui/src/menu.c@ e229148

/*
 * Copyright (c) 2024 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libui
 * @{
 */
/**
 * @file Menu
 */

#include <adt/list.h>
#include <ctype.h>
#include <errno.h>
#include <gfx/color.h>
#include <gfx/context.h>
#include <gfx/render.h>
#include <gfx/text.h>
#include <io/pos_event.h>
#include <stdlib.h>
#include <str.h>
#include <uchar.h>
#include <ui/ui.h>
#include <ui/accel.h>
#include <ui/control.h>
#include <ui/paint.h>
#include <ui/popup.h>
#include <ui/menu.h>
#include <ui/menuentry.h>
#include <ui/resource.h>
#include <ui/window.h>
#include "../private/menu.h"
#include "../private/resource.h"

enum {
        menu_frame_w = 4,
        menu_frame_h = 4,
        menu_frame_w_text = 2,
        menu_frame_h_text = 1,
        menu_frame_h_margin_text = 1
};

static void ui_menu_popup_close(ui_popup_t *, void *);
static void ui_menu_popup_kbd(ui_popup_t *, void *, kbd_event_t *);
static void ui_menu_popup_pos(ui_popup_t *, void *, pos_event_t *);
static void ui_menu_key_press_unmod(ui_menu_t *, kbd_event_t *);

static ui_popup_cb_t ui_menu_popup_cb = {
        .close = ui_menu_popup_close,
        .kbd = ui_menu_popup_kbd,
        .pos = ui_menu_popup_pos
};

/** Create new menu.
 *
 * @param parent Parent window
 * @param mbar Menu bar
 * @param rmenu Place to store pointer to new menu
 * @return EOK on success, ENOMEM if out of memory
 */
errno_t ui_menu_create(ui_window_t *parent, ui_menu_t **rmenu)
{
        ui_menu_t *menu;

        menu = calloc(1, sizeof(ui_menu_t));
        if (menu == NULL)
                return ENOMEM;

        menu->parent = parent;
        list_initialize(&menu->entries);

        *rmenu = menu;
        return EOK;
}

/** Destroy menu.
 *
 * @param menu Menu or @c NULL
 */
void ui_menu_destroy(ui_menu_t *menu)
{
        ui_menu_entry_t *mentry;

        if (menu == NULL)
                return;

        /* Destroy entries */
        mentry = ui_menu_entry_first(menu);
        while (mentry != NULL) {
                ui_menu_entry_destroy(mentry);
                mentry = ui_menu_entry_first(menu);
        }

        free(menu->caption);
        free(menu);
}

/** Set menu callbacks.
 *
 * @param menu Menu
 * @param cb Callbacks
 * @param arg Callback argument
 */
void ui_menu_set_cb(ui_menu_t *menu, ui_menu_cb_t *cb, void *arg)
{
        menu->cb = cb;
        menu->arg = arg;
}

/** Get menu geometry.
 *
 * @param menu Menu
 * @param spos Starting position
 * @param geom Structure to fill in with computed geometry
 */
void ui_menu_get_geom(ui_menu_t *menu, gfx_coord2_t *spos,
    ui_menu_geom_t *geom)
{
        gfx_coord2_t edim;
        gfx_coord_t frame_w;
        gfx_coord_t frame_h;
        ui_resource_t *res;

        res = ui_window_get_res(menu->parent);

        if (res->textmode) {
                frame_w = menu_frame_w_text;
                frame_h = menu_frame_h_text;
        } else {
                frame_w = menu_frame_w;
                frame_h = menu_frame_h;
        }

        edim.x = ui_menu_entry_calc_width(menu, menu->max_caption_w,
            menu->max_shortcut_w);
        edim.y = menu->total_h;

        geom->outer_rect.p0 = *spos;
        geom->outer_rect.p1.x = spos->x + edim.x + 2 * frame_w;
        geom->outer_rect.p1.y = spos->y + edim.y + 2 * frame_h;

        geom->entries_rect.p0.x = spos->x + frame_w;
        geom->entries_rect.p0.y = spos->y + frame_h;
        geom->entries_rect.p1.x = geom->entries_rect.p0.x + edim.x;
        geom->entries_rect.p1.y = geom->entries_rect.p0.x + edim.y;
}

/** Get UI resource from menu.
 *
 * @param menu Menu
 * @return UI resource
 */
ui_resource_t *ui_menu_get_res(ui_menu_t *menu)
{
        return ui_popup_get_res(menu->popup);
}

/** Open menu.
 *
 * @param menu Menu
 * @param prect Parent rectangle around which the menu should be placed
 * @param idev_id Input device associated with the menu's seat
 */
errno_t ui_menu_open(ui_menu_t *menu, gfx_rect_t *prect, sysarg_t idev_id)
{
        ui_popup_t *popup = NULL;
        ui_popup_params_t params;
        ui_menu_geom_t geom;
        gfx_coord2_t mpos;
        errno_t rc;

        /* Select first entry */
        menu->selected = ui_menu_entry_first(menu);

        /* Determine menu dimensions */

        mpos.x = 0;
        mpos.y = 0;
        ui_menu_get_geom(menu, &mpos, &geom);

        ui_popup_params_init(&params);
        params.rect = geom.outer_rect;
        params.place = *prect;
        params.idev_id = idev_id;

        rc = ui_popup_create(ui_window_get_ui(menu->parent), menu->parent,
            &params, &popup);
        if (rc != EOK)
                return rc;

        menu->popup = popup;
        ui_popup_set_cb(popup, &ui_menu_popup_cb, menu);

        return ui_menu_paint(menu, &mpos);
}

/** Close menu.
 *
 * @param menu Menu
 */
void ui_menu_close(ui_menu_t *menu)
{
        ui_popup_destroy(menu->popup);
        menu->popup = NULL;
}

/** Determine if menu is open.
 *
 * @param menu Menu
 * @return @c true iff menu is open
 */
bool ui_menu_is_open(ui_menu_t *menu)
{
        return menu->popup != NULL;
}

/** Paint menu.
 *
 * @param menu Menu
 * @param spos Starting position (top-left corner)
 * @return EOK on success or an error code
 */
errno_t ui_menu_paint_bg_gfx(ui_menu_t *menu, gfx_coord2_t *spos)
{
        ui_resource_t *res;
        ui_menu_geom_t geom;
        gfx_rect_t bg_rect;
        errno_t rc;

        res = ui_menu_get_res(menu);
        ui_menu_get_geom(menu, spos, &geom);

        /* Paint menu frame */

        rc = gfx_set_color(res->gc, res->wnd_face_color);
        if (rc != EOK)
                goto error;

        rc = ui_paint_outset_frame(res, &geom.outer_rect, &bg_rect);
        if (rc != EOK)
                goto error;

        /* Paint menu background */

        rc = gfx_set_color(res->gc, res->wnd_face_color);
        if (rc != EOK)
                goto error;

        rc = gfx_fill_rect(res->gc, &bg_rect);
        if (rc != EOK)
                goto error;

        return EOK;
error:
        return rc;
}

/** Paint menu.
 *
 * @param menu Menu
 * @param spos Starting position (top-left corner)
 * @return EOK on success or an error code
 */
errno_t ui_menu_paint_bg_text(ui_menu_t *menu, gfx_coord2_t *spos)
{
        ui_resource_t *res;
        ui_menu_geom_t geom;
        gfx_rect_t rect;
        errno_t rc;

        res = ui_menu_get_res(menu);
        ui_menu_get_geom(menu, spos, &geom);

        /* Paint menu background */

        rc = gfx_set_color(res->gc, res->wnd_face_color);
        if (rc != EOK)
                goto error;

        rc = gfx_fill_rect(res->gc, &geom.outer_rect);
        if (rc != EOK)
                goto error;

        /* Paint menu box */

        rect = geom.outer_rect;
        rect.p0.x += menu_frame_h_margin_text;
        rect.p1.x -= menu_frame_h_margin_text;

        rc = ui_paint_text_box(res, &rect, ui_box_single, res->wnd_face_color);
        if (rc != EOK)
                goto error;

        return EOK;
error:
        return rc;
}

/** Paint menu.
 *
 * @param menu Menu
 * @param spos Starting position (top-left corner)
 * @return EOK on success or an error code
 */
errno_t ui_menu_paint(ui_menu_t *menu, gfx_coord2_t *spos)
{
        ui_resource_t *res;
        gfx_coord2_t pos;
        ui_menu_entry_t *mentry;
        ui_menu_geom_t geom;
        errno_t rc;

        res = ui_menu_get_res(menu);
        ui_menu_get_geom(menu, spos, &geom);

        /* Paint menu frame and background */
        if (res->textmode)
                rc = ui_menu_paint_bg_text(menu, spos);
        else
                rc = ui_menu_paint_bg_gfx(menu, spos);
        if (rc != EOK)
                goto error;

        /* Paint entries */
        pos = geom.entries_rect.p0;

        mentry = ui_menu_entry_first(menu);
        while (mentry != NULL) {
                rc = ui_menu_entry_paint(mentry, &pos);
                if (rc != EOK)
                        goto error;

                pos.y += ui_menu_entry_height(mentry);
                mentry = ui_menu_entry_next(mentry);
        }

        rc = gfx_update(res->gc);
        if (rc != EOK)
                goto error;

        return EOK;
error:
        return rc;
}

/** Handle position event in menu.
 *
 * @param menu Menu
 * @param spos Starting position (top-left corner)
 * @param event Position event
 * @return ui_claimed iff the event was claimed
 */
ui_evclaim_t ui_menu_pos_event(ui_menu_t *menu, gfx_coord2_t *spos,
    pos_event_t *event)
{
        ui_menu_geom_t geom;
        ui_menu_entry_t *mentry;
        gfx_coord2_t pos;
        gfx_coord2_t epos;
        ui_evclaim_t claimed;

        ui_menu_get_geom(menu, spos, &geom);
        epos.x = event->hpos;
        epos.y = event->vpos;

        pos = geom.entries_rect.p0;

        mentry = ui_menu_entry_first(menu);
        while (mentry != NULL) {
                claimed = ui_menu_entry_pos_event(mentry, &pos, event);
                if (claimed == ui_claimed)
                        return ui_claimed;

                pos.y += ui_menu_entry_height(mentry);
                mentry = ui_menu_entry_next(mentry);
        }

        /* Event inside menu? */
        if (gfx_pix_inside_rect(&epos, &geom.outer_rect)) {
                /* Claim event */
                return ui_claimed;
        } else {
                /* Press outside menu - close it */
                if (event->type == POS_PRESS)
                        ui_menu_close_req(menu);
        }

        return ui_unclaimed;
}

/** Handle keyboard event in menu.
 *
 * @param menu Menu
 * @param event Keyboard event
 * @return ui_claimed iff the event was claimed
 */
ui_evclaim_t ui_menu_kbd_event(ui_menu_t *menu, kbd_event_t *event)
{
        if (event->type == KEY_PRESS && (event->mods &
            (KM_CTRL | KM_ALT | KM_SHIFT)) == 0) {
                ui_menu_key_press_unmod(menu, event);
        }

        if (event->type == KEY_PRESS && (event->mods & KM_ALT) != 0 &&
            (event->mods & (KM_CTRL | KM_SHIFT)) == 0 && event->c != '\0')
                ui_menu_press_accel(menu, event->c, event->kbd_id);

        return ui_claimed;
}

/** Move one entry up.
 *
 * Non-selectable entries are skipped. If we are already at the top,
 * we wrap around.
 *
 * @param menu Menu
 */
void ui_menu_up(ui_menu_t *menu)
{
        gfx_coord2_t mpos;
        ui_menu_entry_t *nentry;

        if (menu->selected == NULL)
                return;

        nentry = ui_menu_entry_prev(menu->selected);
        if (nentry == NULL)
                nentry = ui_menu_entry_last(menu);

        /* Need to find a selectable entry */
        while (!ui_menu_entry_selectable(nentry)) {
                nentry = ui_menu_entry_prev(nentry);
                if (nentry == NULL)
                        nentry = ui_menu_entry_last(menu);

                /* Went completely around and found nothing? */
                if (nentry == menu->selected)
                        return;
        }

        menu->selected = nentry;

        mpos.x = 0;
        mpos.y = 0;
        (void) ui_menu_paint(menu, &mpos);
}

/** Move one entry down.
 *
 * Non-selectable entries are skipped. If we are already at the bottom,
 * we wrap around.
 *
 * @param menu Menu
 */
void ui_menu_down(ui_menu_t *menu)
{
        gfx_coord2_t mpos;
        ui_menu_entry_t *nentry;

        if (menu->selected == NULL)
                return;

        nentry = ui_menu_entry_next(menu->selected);
        if (nentry == NULL)
                nentry = ui_menu_entry_first(menu);

        /* Need to find a selectable entry */
        while (!ui_menu_entry_selectable(nentry)) {
                nentry = ui_menu_entry_next(nentry);
                if (nentry == NULL)
                        nentry = ui_menu_entry_first(menu);

                /* Went completely around and found nothing? */
                if (nentry == menu->selected)
                        return;
        }

        menu->selected = nentry;

        mpos.x = 0;
        mpos.y = 0;
        (void) ui_menu_paint(menu, &mpos);
}

/** Handle key press without modifiers in menu popup window.
 *
 * @param menu Menu
 * @param event Keyboard event
 */
static void ui_menu_key_press_unmod(ui_menu_t *menu, kbd_event_t *event)
{
        ui_menu_entry_t *mentry;
        char32_t c;

        switch (event->key) {
        case KC_ESCAPE:
                ui_menu_close_req(menu);
                break;
        case KC_LEFT:
                ui_menu_left(menu, event->kbd_id);
                break;
        case KC_RIGHT:
                ui_menu_right(menu, event->kbd_id);
                break;
        case KC_UP:
                ui_menu_up(menu);
                break;
        case KC_DOWN:
                ui_menu_down(menu);
                break;
        case KC_ENTER:
                if (menu->selected != NULL &&
                    !ui_menu_entry_is_disabled(menu->selected))
                        ui_menu_entry_activate(menu->selected);
                break;
        default:
                if (event->c != '\0') {
                        mentry = ui_menu_entry_first(menu);
                        while (mentry != NULL) {
                                c = ui_menu_entry_get_accel(mentry);
                                if (c == (char32_t)tolower(event->c) &&
                                    !ui_menu_entry_is_disabled(mentry)) {
                                        ui_menu_entry_activate(mentry);
                                        break;
                                }
                                mentry = ui_menu_entry_next(mentry);
                        }
                }
                break;
        }
}

/** Handle close event in menu popup window.
 *
 * @param popup Menu popup window
 * @param arg Argument (ui_menu_t *)
 */
static void ui_menu_popup_close(ui_popup_t *popup, void *arg)
{
        ui_menu_t *menu = (ui_menu_t *)arg;

        /* Forward close request to caller */
        ui_menu_close_req(menu);
}

/** Handle keyboard event in menu popup window.
 *
 * @param popup Menu popup window
 * @param arg Argument (ui_menu_t *)
 * @param event Keyboard event
 */
static void ui_menu_popup_kbd(ui_popup_t *popup, void *arg, kbd_event_t *event)
{
        ui_menu_t *menu = (ui_menu_t *)arg;

        menu->idev_id = ui_popup_get_idev_id(menu->popup);
        ui_menu_kbd_event(menu, event);
}

/** Handle position event in menu popup window.
 *
 * @param popup Menu popup window
 * @param arg Argument (ui_menu_t *)
 * @param event Position event
 */
static void ui_menu_popup_pos(ui_popup_t *popup, void *arg, pos_event_t *event)
{
        ui_menu_t *menu = (ui_menu_t *)arg;
        gfx_coord2_t spos;

        menu->idev_id = ui_popup_get_idev_id(menu->popup);

        spos.x = 0;
        spos.y = 0;
        ui_menu_pos_event(menu, &spos, event);
}

/** Send menu left event.
 *
 * @param menu Menu
 * @param idev_id Input device ID
 */
void ui_menu_left(ui_menu_t *menu, sysarg_t idev_id)
{
        if (menu->cb != NULL && menu->cb->left != NULL)
                menu->cb->left(menu, menu->arg, idev_id);
}

/** Send menu right event.
 *
 * @param menu Menu
 * @param idev_id Input device ID
 */
void ui_menu_right(ui_menu_t *menu, sysarg_t idev_id)
{
        if (menu->cb != NULL && menu->cb->right != NULL)
                menu->cb->right(menu, menu->arg, idev_id);
}

/** Send menu close request event.
 *
 * @param menu Menu
 */
void ui_menu_close_req(ui_menu_t *menu)
{
        if (menu->cb != NULL && menu->cb->close_req != NULL)
                menu->cb->close_req(menu, menu->arg);
}

/** Send menu accelerator key press event.
 *
 * @param menu Menu
 * @param c Character
 * @param kbd_id Keyboard ID
 */
void ui_menu_press_accel(ui_menu_t *menu, char32_t c, sysarg_t kbd_id)
{
        if (menu->cb != NULL && menu->cb->press_accel != NULL)
                menu->cb->press_accel(menu, menu->arg, c, kbd_id);
}

/** Get ID of last device that input event.
 *
 * @param menu Menu
 * @return Input device ID
 */
sysarg_t ui_menu_get_idev_id(ui_menu_t *menu)
{
        return menu->idev_id;
}

/** @}
 */