/* $NetBSD: draw.c,v 1.10 2023/06/29 19:06:54 nia Exp $ */
/*-
 * Copyright (c) 2021 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Nia Alarie.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``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 FOUNDATION OR CONTRIBUTORS
 * 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.
 */
#include <sys/audioio.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <unistd.h>
#include <curses.h>
#include <err.h>
#include <stdlib.h>
#include "draw.h"

static int get_enum_color(const char *);
static void draw_enum(struct aiomixer_control *, int, bool, bool);
static void draw_set(struct aiomixer_control *, int, bool);
static void draw_levels(struct aiomixer_control *,
    const struct mixer_level *, bool, bool, bool);

void
draw_mixer_select(unsigned int num_mixers, unsigned int selected_mixer)
{
	struct audio_device dev;
	char mixer_path[16];
	unsigned int i;
	int fd;

	mvprintw(0, 0, "Select a mixer device:\n");

	for (i = 0; i < num_mixers; ++i) {
		(void)snprintf(mixer_path, sizeof(mixer_path),
		    "/dev/mixer%d", i);
		fd = open(mixer_path, O_RDWR);
		if (fd == -1)
			break;
		if (ioctl(fd, AUDIO_GETDEV, &dev) < 0) {
			close(fd);
			break;
		}
		close(fd);
		if (selected_mixer == i) {
			attron(A_STANDOUT);
			addstr("[*] ");
		} else {
			addstr("[ ] ");
		}
		printw("%s: %s %s %s\n", mixer_path,
		    dev.name, dev.version, dev.config);
		if (selected_mixer == i)
			attroff(A_STANDOUT);
	}
}

void
draw_control(struct aiomixer *aio,
    struct aiomixer_control *control, bool selected)
{
	struct mixer_ctrl value;

	value.dev = control->info.index;
	value.type = control->info.type;
	if (value.type == AUDIO_MIXER_VALUE)
		value.un.value.num_channels = control->info.un.v.num_channels;

	if (ioctl(aio->fd, AUDIO_MIXER_READ, &value) < 0)
		err(EXIT_FAILURE, "failed to read from mixer device");

	wclear(control->widgetpad);
	if (selected)
		wattron(control->widgetpad, A_STANDOUT);
	if (value.type == AUDIO_MIXER_VALUE)
		wprintw(control->widgetpad, "%s (%s)\n",
		    control->info.label.name, control->info.un.v.units.name);
	else
		wprintw(control->widgetpad, "%s\n", control->info.label.name);
	if (selected)
		wattroff(control->widgetpad, A_STANDOUT);

	switch (value.type) {
	case AUDIO_MIXER_ENUM:
		draw_enum(control, value.un.ord, selected, aio->use_colour);
		break;
	case AUDIO_MIXER_SET:
		draw_set(control, value.un.mask, aio->use_colour);
		break;
	case AUDIO_MIXER_VALUE:
		draw_levels(control, &value.un.value,
		    aio->channels_unlocked, selected, aio->use_colour);
		break;
	}

	wprintw(control->widgetpad, "\n");
}

void
draw_screen(struct aiomixer *aio)
{
	int i, max_y;

	/* Clear any leftovers if the screen changed. */
	if (aio->widgets_resized) {
		aio->widgets_resized = false;
		max_y = getmaxy(stdscr);
		for (i = 3; i < max_y; ++i) {
			mvprintw(i, 0, "\n");
		}
	}

	wnoutrefresh(stdscr);
	wnoutrefresh(aio->header);
	wnoutrefresh(aio->classbar);
	max_y = aio->classes[aio->curclass].height + 1;
	max_y -= aio->class_scroll_y;
	if (max_y > (getmaxy(stdscr) - 3))
		max_y = getmaxy(stdscr) - 3;
	pnoutrefresh(aio->classes[aio->curclass].widgetpad,
	    aio->class_scroll_y, 0,
	    3, 0,
	    max_y, getmaxx(stdscr));
	doupdate();
}

static int
get_enum_color(const char *name)
{
	if (strcmp(name, AudioNon) == 0) {
		return COLOR_ENUM_ON;
	}
	if (strcmp(name, AudioNoff) == 0) {
		return COLOR_ENUM_OFF;
	}

	return COLOR_ENUM_MISC;
}

static void
draw_enum(struct aiomixer_control *control, int ord,
    bool selected, bool colour)
{
	struct audio_mixer_enum *e;
	int color = COLOR_ENUM_MISC;
	int i;

	for (i = 0; i < control->info.un.e.num_mem; ++i) {
		e = &control->info.un.e;
		if (ord == e->member[i].ord && selected) {
			if (termattrs() & A_BOLD)
				wattron(control->widgetpad, A_BOLD);
			else
				wattron(control->widgetpad, A_STANDOUT);
		}
		waddch(control->widgetpad, '[');
		if (ord == e->member[i].ord) {
			if (colour) {
				color = get_enum_color(e->member[i].label.name);
				wattron(control->widgetpad,
					COLOR_PAIR(color));
			} else {
				waddch(control->widgetpad, '*');
			}
		}
		wprintw(control->widgetpad, "%s", e->member[i].label.name);
		if (ord == control->info.un.e.member[i].ord) {
			if (colour) {
				wattroff(control->widgetpad,
					COLOR_PAIR(color));
			}
		}
		waddch(control->widgetpad, ']');
		if (ord == e->member[i].ord && selected) {
			if (termattrs() & A_BOLD)
				wattroff(control->widgetpad, A_BOLD);
			else
				wattroff(control->widgetpad, A_STANDOUT);
		}
		if (i != (e->num_mem - 1))
			waddstr(control->widgetpad, ", ");
	}
	waddch(control->widgetpad, '\n');
}

static void
draw_set(struct aiomixer_control *control, int mask, bool colour)
{
	int i;

	for (i = 0; i < control->info.un.s.num_mem; ++i) {
		waddch(control->widgetpad, '[');
		if (mask & control->info.un.s.member[i].mask) {
			if (colour) {
				wattron(control->widgetpad,
					COLOR_PAIR(COLOR_SET_SELECTED));
			}
			waddch(control->widgetpad, '*');
			if (colour) {
				wattroff(control->widgetpad,
					COLOR_PAIR(COLOR_SET_SELECTED));
			}
		} else {
			waddch(control->widgetpad, ' ');
		}
		waddstr(control->widgetpad, "] ");
		if (control->setindex == i) {
			if (termattrs() & A_BOLD)
				wattron(control->widgetpad, A_BOLD);
			else
				wattron(control->widgetpad, A_STANDOUT);
		}
		wprintw(control->widgetpad, "%s",
		    control->info.un.s.member[i].label.name);
		if (control->setindex == i) {
			if (termattrs() & A_BOLD)
				wattroff(control->widgetpad, A_BOLD);
			else
				wattroff(control->widgetpad, A_STANDOUT);
		}
		if (i != (control->info.un.s.num_mem - 1))
			waddstr(control->widgetpad, ", ");
	}
}

static void
draw_levels(struct aiomixer_control *control,
    const struct mixer_level *levels, bool channels_unlocked,
    bool selected, bool colour)
{
	int i;
	int j, nchars;

	for (i = 0; i < control->info.un.v.num_channels; ++i) {
		if ((selected && !channels_unlocked) ||
		    (control->setindex == i && channels_unlocked)) {
			if (termattrs() & A_BOLD)
				wattron(control->widgetpad, A_BOLD);
			else
				wattron(control->widgetpad, A_STANDOUT);
		}
		wprintw(control->widgetpad, "[%3u/%3u ",
		    levels->level[i], AUDIO_MAX_GAIN);
		if (colour) {
			wattron(control->widgetpad,
				COLOR_PAIR(COLOR_LEVELS));
		}
		nchars = (levels->level[i] *
		    (getmaxx(control->widgetpad) - 11)) / AUDIO_MAX_GAIN;
		for (j = 0; j < nchars; ++j)
			waddch(control->widgetpad, '*');
		if (colour) {
			wattroff(control->widgetpad,
				COLOR_PAIR(COLOR_LEVELS));
		}
		nchars = getmaxx(control->widgetpad) - 11 - nchars;
		for (j = 0; j < nchars; ++j)
			waddch(control->widgetpad, ' ');
		wprintw(control->widgetpad, "]\n");
		if ((selected && !channels_unlocked) ||
		    (control->setindex == i && channels_unlocked)) {
			if (termattrs() & A_BOLD)
				wattroff(control->widgetpad, A_BOLD);
			else
				wattroff(control->widgetpad, A_STANDOUT);
		}
	}
}

void
draw_classbar(struct aiomixer *aio)
{
	unsigned int i;

	wmove(aio->classbar, 0, 0);

	for (i = 0; i < aio->numclasses; ++i) {
		if (aio->curclass == i)
			wattron(aio->classbar, A_STANDOUT);
		wprintw(aio->classbar, "[%u:%s]",
		    i + 1, aio->classes[i].name);
		if (aio->curclass == i)
			wattroff(aio->classbar, A_STANDOUT);
		waddch(aio->classbar, ' ');
	}

	wprintw(aio->classbar, "\n\n");
}

void
draw_header(struct aiomixer *aio)
{
	wprintw(aio->header, "\n");
	mvwaddstr(aio->header, 0,
	    getmaxx(aio->header) - (int)sizeof("NetBSD audio mixer"),
	    "NetBSD audio mixer");

	if (aio->mixerdev.version[0] != '\0') {
		mvwprintw(aio->header, 0, 0, "%s %s",
		    aio->mixerdev.name, aio->mixerdev.version);
	} else {
		mvwprintw(aio->header, 0, 0, "%s", aio->mixerdev.name);
	}
}

void
create_widgets(struct aiomixer *aio)
{
	size_t i, j;
	struct aiomixer_class *class;
	struct aiomixer_control *control;

	aio->header = newwin(1, getmaxx(stdscr), 0, 0);
	if (aio->header == NULL)
		errx(EXIT_FAILURE, "failed to create window");

	aio->classbar = newwin(2, getmaxx(stdscr), 1, 0);
	if (aio->classbar == NULL)
		errx(EXIT_FAILURE, "failed to create window");

	for (i = 0; i < aio->numclasses; ++i) {
		class = &aio->classes[i];
		class->height = 0;
		class->widgetpad = newpad(4 * __arraycount(class->controls),
		    getmaxx(stdscr));
		if (class->widgetpad == NULL)
			errx(EXIT_FAILURE, "failed to create curses pad");
		for (j = 0; j < class->numcontrols; ++j) {
			control = &class->controls[j];
			switch (control->info.type) {
			case AUDIO_MIXER_VALUE:
				control->height = 2 +
				    control->info.un.v.num_channels;
				break;
			case AUDIO_MIXER_ENUM:
			case AUDIO_MIXER_SET:
				control->height = 3;
				break;
			}
			control->widgetpad = subpad(class->widgetpad,
			    control->height, getmaxx(stdscr),
			    class->height, 0);
			if (control->widgetpad == NULL)
				errx(EXIT_FAILURE, "failed to create curses pad");
			control->widget_y = class->height;
			class->height += control->height;
		}
		wresize(class->widgetpad, class->height, getmaxx(stdscr));
	}

	aio->last_max_x = getmaxx(stdscr);
}

void
resize_widgets(struct aiomixer *aio)
{
	size_t i, j;
	struct aiomixer_class *class;
	struct aiomixer_control *control;
	int max_x;

	max_x = getmaxx(stdscr);

	if (aio->last_max_x != max_x) {
		aio->last_max_x = max_x;
		wresize(aio->header, 1, max_x);
		wresize(aio->classbar, 2, max_x);

		for (i = 0; i < aio->numclasses; ++i) {
			class = &aio->classes[i];
			wresize(class->widgetpad, class->height, max_x);
			for (j = 0; j < class->numcontrols; ++j) {
				control = &class->controls[j];
				wresize(control->widgetpad,
				    control->height, max_x);
			}
		}
	}

	aio->widgets_resized = true;
}