/* $NetBSD: bwdsp.c,v 1.2 2024/01/23 21:49:51 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: bwdsp.c,v 1.2 2024/01/23 21:49:51 jmcneill Exp $");

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

#include <sys/audioio.h>
#include <dev/audio/audio_if.h>
#include <dev/audio/audio_dai.h>

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

#define	BWDSP_MAP_FLAGS		BUS_DMA_NOCACHE

#define DSP_DMA_START_ADDR_H	0x30
#define	DSP_DMA_START_ADDR_L	0x32
#define DSP_DMA_CONTROL_LENGTH	0x36
#define  DSP_DMA_CONTROL_LENGTH_CTRL	__BIT(15)
#define	 DSP_DMA_CONTROL_LENGTH_NUM_CLS	__BITS(14,0)

#define	DSP_DMA_ALIGN		32
#define	DSP_DMA_MAX_BUFSIZE	(DSP_DMA_CONTROL_LENGTH_NUM_CLS * 32)

struct bwdsp_dma {
	LIST_ENTRY(bwdsp_dma)	dma_list;
	bus_dmamap_t		dma_map;
	void			*dma_addr;
	size_t			dma_size;
	bus_dma_segment_t	dma_segs[1];
	int			dma_nsegs;
};

struct bwdsp_softc {
	device_t		sc_dev;
	bus_space_tag_t		sc_bst;
	bus_space_handle_t	sc_bsh;
	bus_dma_tag_t		sc_dmat;

	LIST_HEAD(, bwdsp_dma) sc_dmalist;

	kmutex_t		sc_lock;
	kmutex_t		sc_intr_lock;

	struct audio_format	sc_format;

	audio_dai_tag_t		sc_dai;
};

#define	RD2(sc, reg)			\
	bus_space_read_2((sc)->sc_bst, (sc)->sc_bsh, (reg))
#define	WR2(sc, reg, val)		\
	bus_space_write_2((sc)->sc_bst, (sc)->sc_bsh, (reg), (val))

static int
bwdsp_allocdma(struct bwdsp_softc *sc, size_t size,
    size_t align, struct bwdsp_dma *dma)
{
	int error;

	dma->dma_size = size;
	error = bus_dmamem_alloc(sc->sc_dmat, dma->dma_size, align, 0,
	    dma->dma_segs, 1, &dma->dma_nsegs, BUS_DMA_WAITOK);
	if (error)
		return error;

	error = bus_dmamem_map(sc->sc_dmat, dma->dma_segs, dma->dma_nsegs,
	    dma->dma_size, &dma->dma_addr, BUS_DMA_WAITOK | BWDSP_MAP_FLAGS);
	if (error)
		goto free;

	error = bus_dmamap_create(sc->sc_dmat, dma->dma_size, dma->dma_nsegs,
	    dma->dma_size, 0, BUS_DMA_WAITOK, &dma->dma_map);
	if (error)
		goto unmap;

	error = bus_dmamap_load(sc->sc_dmat, dma->dma_map, dma->dma_addr,
	    dma->dma_size, NULL, BUS_DMA_WAITOK);
	if (error)
		goto destroy;

	return 0;

destroy:
	bus_dmamap_destroy(sc->sc_dmat, dma->dma_map);
unmap:
	bus_dmamem_unmap(sc->sc_dmat, dma->dma_addr, dma->dma_size);
free:
	bus_dmamem_free(sc->sc_dmat, dma->dma_segs, dma->dma_nsegs);

	return error;
}

static void
bwdsp_freedma(struct bwdsp_softc *sc, struct bwdsp_dma *dma)
{
	bus_dmamap_unload(sc->sc_dmat, dma->dma_map);
	bus_dmamap_destroy(sc->sc_dmat, dma->dma_map);
	bus_dmamem_unmap(sc->sc_dmat, dma->dma_addr, dma->dma_size);
	bus_dmamem_free(sc->sc_dmat, dma->dma_segs, dma->dma_nsegs);
}

static int
bwdsp_query_format(void *priv, audio_format_query_t *afp)
{
	struct bwdsp_softc * const sc = priv;

	return audio_query_format(&sc->sc_format, 1, afp);
}

static int
bwdsp_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)
{
	struct bwdsp_softc * const sc = priv;

	return audio_dai_mi_set_format(sc->sc_dai, setmode, play, rec,
	    pfil, rfil);
}

static int
bwdsp_set_port(void *priv, mixer_ctrl_t *mc)
{
	struct bwdsp_softc * const sc = priv;

	return audio_dai_set_port(sc->sc_dai, mc);
}

static int
bwdsp_get_port(void *priv, mixer_ctrl_t *mc)
{
	struct bwdsp_softc * const sc = priv;

	return audio_dai_get_port(sc->sc_dai, mc);
}

static int
bwdsp_query_devinfo(void *priv, mixer_devinfo_t *di)
{
	struct bwdsp_softc * const sc = priv;

	return audio_dai_query_devinfo(sc->sc_dai, di);
}

static void *
bwdsp_allocm(void *priv, int dir, size_t size)
{
	struct bwdsp_softc * const sc = priv;
	struct bwdsp_dma *dma;
	int error;

	dma = kmem_alloc(sizeof(*dma), KM_SLEEP);

	error = bwdsp_allocdma(sc, size, DSP_DMA_ALIGN, dma);
	if (error) {
		kmem_free(dma, sizeof(*dma));
		device_printf(sc->sc_dev, "couldn't allocate DMA memory (%d)\n",
		    error);
		return NULL;
	}

	LIST_INSERT_HEAD(&sc->sc_dmalist, dma, dma_list);

	return dma->dma_addr;
}

static void
bwdsp_freem(void *priv, void *addr, size_t size)
{
	struct bwdsp_softc * const sc = priv;
	struct bwdsp_dma *dma;

	LIST_FOREACH(dma, &sc->sc_dmalist, dma_list)
		if (dma->dma_addr == addr) {
			bwdsp_freedma(sc, dma);
			LIST_REMOVE(dma, dma_list);
			kmem_free(dma, sizeof(*dma));
			break;
		}
}

static int
bwdsp_getdev(void *priv, struct audio_device *adev)
{
	snprintf(adev->name, sizeof(adev->name), "Broadway DSP");
	snprintf(adev->version, sizeof(adev->version), "");
	snprintf(adev->config, sizeof(adev->config), "bwdsp");

	return 0;
}

static int
bwdsp_get_props(void *priv)
{
	return AUDIO_PROP_PLAYBACK;
}

static int
bwdsp_round_blocksize(void *priv, int bs, int mode,
    const audio_params_t *params)
{
	bs = roundup(bs, DSP_DMA_ALIGN);
	if (bs > DSP_DMA_MAX_BUFSIZE) {
		bs = DSP_DMA_MAX_BUFSIZE;
	}
	return bs;
}

static size_t
bwdsp_round_buffersize(void *priv, int dir, size_t bufsize)
{
	if (bufsize > DSP_DMA_MAX_BUFSIZE) {
		bufsize = DSP_DMA_MAX_BUFSIZE;
	}
	return bufsize;
}

static void
bwdsp_transfer(struct bwdsp_softc *sc, uint32_t phys_addr, size_t bufsize)
{
	if (bufsize != 0) {
		WR2(sc, DSP_DMA_START_ADDR_H, phys_addr >> 16);
		WR2(sc, DSP_DMA_START_ADDR_L, phys_addr & 0xffff);
		WR2(sc, DSP_DMA_CONTROL_LENGTH,
		    DSP_DMA_CONTROL_LENGTH_CTRL | (bufsize / 32));
	} else {
		WR2(sc, DSP_DMA_CONTROL_LENGTH, 0);
	}
}

static int
bwdsp_trigger_output(void *priv, void *start, void *end, int blksize,
    void (*intr)(void *), void *intrarg, const audio_params_t *params)
{
	struct bwdsp_softc * const sc = priv;
	struct bwdsp_dma *dma;
	bus_addr_t pstart;
	bus_size_t psize;
	int error;

	pstart = 0;
	psize = (uintptr_t)end - (uintptr_t)start;

	LIST_FOREACH(dma, &sc->sc_dmalist, dma_list)
		if (dma->dma_addr == start) {
			pstart = dma->dma_map->dm_segs[0].ds_addr;
			break;
		}
	if (pstart == 0) {
		device_printf(sc->sc_dev, "bad addr %p\n", start);
		return EINVAL;
	}

	error = audio_dai_trigger(sc->sc_dai, start, end, blksize,
	    intr, intrarg, params, AUMODE_PLAY);
	if (error != 0) {
		return error;
	}

	/* Start DMA transfer */
	bwdsp_transfer(sc, pstart, psize);

	return 0;
}

static int
bwdsp_halt_output(void *priv)
{
	struct bwdsp_softc * const sc = priv;

	/* Stop DMA transfer */
	bwdsp_transfer(sc, 0, 0);

	return audio_dai_halt(sc->sc_dai, AUMODE_PLAY);
}

static void
bwdsp_get_locks(void *priv, kmutex_t **intr, kmutex_t **thread)
{
	struct bwdsp_softc * const sc = priv;

	*intr = &sc->sc_intr_lock;
	*thread = &sc->sc_lock;
}

static const struct audio_hw_if bwdsp_hw_if = {
	.query_format = bwdsp_query_format,
	.set_format = bwdsp_set_format,
	.allocm = bwdsp_allocm,
	.freem = bwdsp_freem,
	.getdev = bwdsp_getdev,
	.set_port = bwdsp_set_port,
	.get_port = bwdsp_get_port,
	.query_devinfo = bwdsp_query_devinfo,
	.get_props = bwdsp_get_props,
	.round_blocksize = bwdsp_round_blocksize,
	.round_buffersize = bwdsp_round_buffersize,
	.trigger_output = bwdsp_trigger_output,
	.halt_output = bwdsp_halt_output,
	.get_locks = bwdsp_get_locks,
};

static void
bwdsp_late_attach(device_t dev)
{
	struct bwdsp_softc * const sc = device_private(dev);

	sc->sc_dai = bwai_dsp_init(&sc->sc_intr_lock);
	if (sc->sc_dai == NULL) {
		aprint_error_dev(dev, "can't find bwai device\n");
		return;
	}

	audio_attach_mi(&bwdsp_hw_if, sc, dev);
}

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

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

static void
bwdsp_attach(device_t parent, device_t self, void *aux)
{
	struct bwdsp_softc * const sc = device_private(self);
	struct mainbus_attach_args * const maa = aux;
	bus_addr_t addr = maa->maa_addr;
	bus_size_t size = 0x200;

	sc->sc_dev = self;
	sc->sc_bst = maa->maa_bst;
	if (bus_space_map(sc->sc_bst, addr, size, 0, &sc->sc_bsh) != 0) {
		aprint_error(": couldn't map registers\n");
		return;
	}
	sc->sc_dmat = maa->maa_dmat;
	LIST_INIT(&sc->sc_dmalist);
	mutex_init(&sc->sc_lock, MUTEX_DEFAULT, IPL_NONE);
	mutex_init(&sc->sc_intr_lock, MUTEX_DEFAULT, IPL_SCHED);

	aprint_naive("\n");
	aprint_normal(": DSP\n");

	sc->sc_format.mode = AUMODE_PLAY;
	sc->sc_format.encoding = AUDIO_ENCODING_SLINEAR_BE;
	sc->sc_format.validbits = 16;
	sc->sc_format.precision = 16;
	sc->sc_format.channels = 2;
	sc->sc_format.channel_mask = AUFMT_STEREO;
	sc->sc_format.frequency_type = 1;
	sc->sc_format.frequency[0] = 48000;

	config_defer(self, bwdsp_late_attach);
}

CFATTACH_DECL_NEW(bwdsp, sizeof(struct bwdsp_softc),
    bwdsp_match, bwdsp_attach, NULL, NULL);