/* $NetBSD: bwai.c,v 1.4 2025/02/15 00:35:18 jmcneill Exp $ */

/*-
 * Copyright (c) 2024 Jared McNeill <jmcneill@invisible.ca>
 * All rights reserved.
 *
 * 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 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.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: bwai.c,v 1.4 2025/02/15 00:35:18 jmcneill Exp $");

#include <sys/param.h>
#include <sys/bus.h>
#include <sys/cpu.h>
#include <sys/device.h>
#include <sys/kmem.h>
#include <sys/mutex.h>
#include <sys/bitops.h>

#include <dev/audio/audio_dai.h>

#include <machine/wii.h>

#include "mainbus.h"
#include "avenc.h"
#include "bwai.h"

#define	AI_CONTROL		0x00
#define	 AI_CONTROL_RATE	__BIT(6)
#define	 AI_CONTROL_SCRESET	__BIT(5)
#define	 AI_CONTROL_AIINTVLD	__BIT(4)
#define	 AI_CONTROL_AIINT	__BIT(3)
#define	 AI_CONTROL_AIINTMSK	__BIT(2)
#define	 AI_CONTROL_AFR		__BIT(1)
#define	 AI_CONTROL_PSTAT	__BIT(0)
#define AI_AIIT			0x0c

struct bwai_softc {
	device_t		sc_dev;
	bus_space_tag_t		sc_bst;
	bus_space_handle_t	sc_bsh;
	int			sc_irq;

	struct audio_dai_device	sc_dai;

	void			(*sc_intr)(void *);
	void			*sc_intrarg;
	uint32_t		sc_intrnext;
	uint32_t		sc_intrstep;

	kmutex_t		*sc_intr_lock;
};

enum bwai_mixer_ctrl {
	BWAI_OUTPUT_CLASS,
	BWAI_INPUT_CLASS,

	BWAI_OUTPUT_MASTER_VOLUME,
	BWAI_INPUT_DAC_VOLUME,

	BWAI_MIXER_CTRL_LAST
};

#define	RD4(sc, reg)			\
	bus_space_read_4((sc)->sc_bst, (sc)->sc_bsh, (reg))
#define	WR4(sc, reg, val)		\
	bus_space_write_4((sc)->sc_bst, (sc)->sc_bsh, (reg), (val))

static int
bwai_intr(void *priv)
{
	struct bwai_softc * const sc = priv;
	uint32_t val;

	val = RD4(sc, AI_CONTROL);
	if ((val & AI_CONTROL_AIINT) != 0) {
		sc->sc_intrnext += sc->sc_intrstep;
		WR4(sc, AI_AIIT, sc->sc_intrnext);
		WR4(sc, AI_CONTROL, val);

		mutex_enter(sc->sc_intr_lock);
		if (sc->sc_intr) {
			sc->sc_intr(sc->sc_intrarg);
		}
		mutex_exit(sc->sc_intr_lock);
	}

	return 1;
}

audio_dai_tag_t
bwai_dsp_init(kmutex_t *intr_lock)
{
	struct bwai_softc *sc;
	device_t dev;

	dev = device_find_by_driver_unit("bwai", 0);
	if (dev == NULL) {
		return NULL;
	}
	sc = device_private(dev);

	sc->sc_intr_lock = intr_lock;

	intr_establish_xname(sc->sc_irq, IST_LEVEL, IPL_AUDIO, bwai_intr, sc,
	    device_xname(dev));

	return &sc->sc_dai;
}

static int
bwai_set_port(void *priv, mixer_ctrl_t *mc)
{
	if (mc->dev != BWAI_OUTPUT_MASTER_VOLUME &&
	    mc->dev != BWAI_INPUT_DAC_VOLUME) {
		return ENXIO;
	}

	avenc_set_volume(mc->un.value.level[AUDIO_MIXER_LEVEL_LEFT],
			 mc->un.value.level[AUDIO_MIXER_LEVEL_RIGHT]);

	return 0;
}

static int
bwai_get_port(void *priv, mixer_ctrl_t *mc)
{
	if (mc->dev != BWAI_OUTPUT_MASTER_VOLUME &&
	    mc->dev != BWAI_INPUT_DAC_VOLUME) {
		return ENXIO;
	}

	avenc_get_volume(&mc->un.value.level[AUDIO_MIXER_LEVEL_LEFT],
			 &mc->un.value.level[AUDIO_MIXER_LEVEL_RIGHT]);

	return 0;
}

static int
bwai_query_devinfo(void *priv, mixer_devinfo_t *di)
{
	switch (di->index) {
	case BWAI_OUTPUT_CLASS:
		di->mixer_class = di->index;
		strcpy(di->label.name, AudioCoutputs);
		di->type = AUDIO_MIXER_CLASS;
		di->next = di->prev = AUDIO_MIXER_LAST;
		return 0;

	case BWAI_INPUT_CLASS:
		di->mixer_class = di->index;
		strcpy(di->label.name, AudioCinputs);
		di->type = AUDIO_MIXER_CLASS;
		di->next = di->prev = AUDIO_MIXER_LAST;
		return 0;

	case BWAI_OUTPUT_MASTER_VOLUME:
		di->mixer_class = BWAI_OUTPUT_CLASS;
		strcpy(di->label.name, AudioNmaster);
		di->un.v.delta = 1;
		di->type = AUDIO_MIXER_VALUE;
		di->next = di->prev = AUDIO_MIXER_LAST;
		di->un.v.num_channels = 2;
		strcpy(di->un.v.units.name, AudioNvolume);
		return 0;

	case BWAI_INPUT_DAC_VOLUME:
		di->mixer_class = BWAI_INPUT_CLASS;
		strcpy(di->label.name, AudioNdac);
		di->un.v.delta = 1;
		di->type = AUDIO_MIXER_VALUE;
		di->next = di->prev = AUDIO_MIXER_LAST;
		di->un.v.num_channels = 2;
		strcpy(di->un.v.units.name, AudioNvolume);
		return 0;
	}

	return ENXIO;
}

static int
bwai_set_format(void *priv, int setmode,
    const audio_params_t *play, const audio_params_t *rec,
    audio_filter_reg_t *pfil, audio_filter_reg_t *rfil)
{
	return 0;
}

static int
bwai_trigger_output(void *priv, void *start, void *end, int blksize,
    void (*intr)(void *), void *intrarg, const audio_params_t *params)
{
	struct bwai_softc * const sc = priv;
	uint32_t val;

	sc->sc_intr = intr;
	sc->sc_intrarg = intrarg;

	val = RD4(sc, AI_CONTROL);
	if ((val & AI_CONTROL_PSTAT) != 0) {
		WR4(sc, AI_CONTROL, 0);
	}

	sc->sc_intrstep = blksize / 4;
	sc->sc_intrnext = sc->sc_intrstep;
	WR4(sc, AI_AIIT, sc->sc_intrnext);

	val = AI_CONTROL_SCRESET |
	      AI_CONTROL_AIINT |
	      AI_CONTROL_AIINTMSK |
	      AI_CONTROL_AFR;
	WR4(sc, AI_CONTROL, val);

	val |= AI_CONTROL_PSTAT;
	WR4(sc, AI_CONTROL, val);

	return 0;
}

static int
bwai_halt_output(void *priv)
{
	struct bwai_softc * const sc = priv;

	WR4(sc, AI_CONTROL, 0);

	sc->sc_intr = NULL;
	sc->sc_intrarg = NULL;

	return 0;
}

static const struct audio_hw_if bwai_hw_if = {
	.set_port = bwai_set_port,
	.get_port = bwai_get_port,
	.query_devinfo = bwai_query_devinfo,
	.set_format = bwai_set_format,
	.trigger_output = bwai_trigger_output,
	.halt_output = bwai_halt_output,
};

static int
bwai_match(device_t parent, cfdata_t cf, void *aux)
{
	struct mainbus_attach_args * const maa = aux;

	return strcmp(maa->maa_name, "bwai") == 0;
}

static void
bwai_attach(device_t parent, device_t self, void *aux)
{
	struct bwai_softc * const sc = device_private(self);
	struct mainbus_attach_args * const maa = aux;

	sc->sc_dev = self;
	sc->sc_bst = maa->maa_bst;
	if (bus_space_map(sc->sc_bst, maa->maa_addr, AI_SIZE, 0,
	    &sc->sc_bsh) != 0) {
		aprint_error(": couldn't map registers\n");
		return;
	}
	sc->sc_irq = maa->maa_irq;

	aprint_naive("\n");
	aprint_normal(": Audio Interface\n");

	sc->sc_dai.dai_hw_if = &bwai_hw_if;
	sc->sc_dai.dai_dev = self;
	sc->sc_dai.dai_priv = sc;
}

CFATTACH_DECL_NEW(bwai, sizeof(struct bwai_softc),
    bwai_match, bwai_attach, NULL, NULL);