/*      $NetBSD: slurm.c,v 1.4 2019/01/22 06:47:20 skrll Exp $ */

/*
 * Copyright (c) 2012 Jonathan A. Kollasch
 * 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 COPYRIGHT HOLDERS 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 COPYRIGHT HOLDER 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/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: slurm.c,v 1.4 2019/01/22 06:47:20 skrll Exp $");

#include <sys/param.h>
#include <sys/proc.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/device.h>
#include <sys/conf.h>

#include <dev/usb/usb.h>
#include <dev/usb/usbdi.h>
#include <dev/usb/usbdivar.h>
#include <dev/usb/usbdi_util.h>
#include <dev/usb/usbdevs.h>
#include <dev/ic/si470x_reg.h>

#include <sys/radioio.h>
#include <dev/radio_if.h>

#ifdef SLURM_DEBUG
int     slurmdebug = 0;
#define DPRINTFN(n, x)  do { if (slurmdebug > (n)) printf x; } while (0)
#else
#define DPRINTFN(n, x)
#endif

#define DPRINTF(x) DPRINTFN(0, x)

#define SI470X_VOLFACT (255 / __SHIFTOUT_MASK(SI470X_VOLUME))

struct slurm_softc {
        device_t                sc_dev;
        struct usbd_device *    sc_udev;
        struct usbd_interface * sc_uif;
        uint32_t                sc_band;
        uint32_t                sc_space;
};

static const struct usb_devno slurm_devs[] = {
        { USB_VENDOR_ADS, USB_PRODUCT_ADS_RDX155 },
};

static int slurm_match(device_t, cfdata_t, void *);
static void slurm_attach(device_t, device_t, void *);
static int slurm_detach(device_t, int);

static int slurm_get_info(void *, struct radio_info *);
static int slurm_set_info(void *, struct radio_info *);
static int slurm_search(void *, int);

static usbd_status slurm_setreg(struct slurm_softc *, int, uint16_t);
static usbd_status slurm_getreg(struct slurm_softc *, int, uint16_t *);

static uint32_t slurm_si470x_get_freq(struct slurm_softc *, uint16_t);
static void slurm_si470x_get_bandspace(struct slurm_softc *, uint16_t);
static int slurm_si470x_get_info(uint16_t);
static int slurm_si470x_get_mute(uint16_t);
static int slurm_si470x_get_stereo(uint16_t);
static int slurm_si470x_get_volume(uint16_t);

static int slurm_si470x_search(struct slurm_softc *, int);

static void slurm_si470x_set_freq(struct slurm_softc *, uint32_t);
static void slurm_si470x_set_powercfg(struct slurm_softc *, int, int);
static void slurm_si470x_set_volume(struct slurm_softc *, int);

static const struct radio_hw_if slurm_radio = {
        .get_info = slurm_get_info,
        .set_info = slurm_set_info,
        .search = slurm_search,
};

CFATTACH_DECL_NEW(slurm, sizeof(struct slurm_softc),
    slurm_match, slurm_attach, slurm_detach, NULL);

static int
slurm_match(device_t parent, cfdata_t match, void *aux)
{
        const struct usbif_attach_arg * const uiaa = aux;

        if (uiaa->uiaa_ifaceno != 2)
                return UMATCH_NONE;

        if (usb_lookup(slurm_devs, uiaa->uiaa_vendor, uiaa->uiaa_product) != NULL) {
                return UMATCH_VENDOR_PRODUCT;
        }

        return UMATCH_NONE;
}

static void
slurm_attach(device_t parent, device_t self, void *aux)
{
        struct slurm_softc * const sc = device_private(self);
        const struct usbif_attach_arg * const uiaa = aux;

        sc->sc_dev = self;
        sc->sc_udev = uiaa->uiaa_device;
        sc->sc_uif = uiaa->uiaa_iface;

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

        usbd_add_drv_event(USB_EVENT_DRIVER_ATTACH, sc->sc_udev, sc->sc_dev);

#ifdef SLURM_DEBUG
        {
                uint16_t val;
                for (int i = 0; i < 16; i++) {
                        slurm_getreg(sc, i, &val);
                        device_printf(self, "%02x -> %04x\n", i, val);
                }
        }
#endif

        radio_attach_mi(&slurm_radio, sc, self);
}

static int
slurm_detach(device_t self, int flags)
{
        struct slurm_softc * const sc = device_private(self);
        int rv = 0;

        if ((rv = config_detach_children(self, flags)) != 0)
                return rv;

        usbd_add_drv_event(USB_EVENT_DRIVER_DETACH, sc->sc_udev, sc->sc_dev);

        return rv;
}

static int
slurm_get_info(void *v, struct radio_info *ri)
{
        struct slurm_softc * const sc = v;
        uint16_t powercfg, sysconfig2, readchannel, statusrssi;

        slurm_getreg(sc, SI470X_POWERCFG, &powercfg);
        slurm_getreg(sc, SI470X_SYSCONFIG2, &sysconfig2);
        slurm_getreg(sc, SI470X_STATUSRSSI, &statusrssi);
        slurm_getreg(sc, SI470X_READCHANNEL, &readchannel);

        ri->mute = slurm_si470x_get_mute(powercfg);
        ri->volume = slurm_si470x_get_volume(sysconfig2);
        ri->stereo = slurm_si470x_get_stereo(powercfg);
        ri->rfreq = 0;
        ri->lock = 0;
        slurm_si470x_get_bandspace(sc, sysconfig2);
        ri->freq = slurm_si470x_get_freq(sc, readchannel);
        ri->caps = RADIO_CAPS_DETECT_STEREO | RADIO_CAPS_DETECT_SIGNAL |
                   RADIO_CAPS_SET_MONO | RADIO_CAPS_HW_SEARCH |
                   RADIO_CAPS_HW_AFC | RADIO_CAPS_LOCK_SENSITIVITY;
        ri->info = slurm_si470x_get_info(statusrssi);

        return 0;
}

static int
slurm_set_info(void *v, struct radio_info *ri)
{
        struct slurm_softc * const sc = v;

        slurm_si470x_set_freq(sc, ri->freq);
        slurm_si470x_set_powercfg(sc, ri->mute, ri->stereo);
        slurm_si470x_set_volume(sc, ri->volume);

        return 0;
}

static int
slurm_search(void *v, int f)
{
        struct slurm_softc * const sc = v;

        return slurm_si470x_search(sc, f);
}

static usbd_status
slurm_getreg(struct slurm_softc *sc, int reg, uint16_t *val)
{
        usbd_status status;
        uint8_t s[3];

        ++reg;

        s[0] = reg;
        s[1] = s[2] = 0;

        status = usbd_get_report(sc->sc_uif, UHID_FEATURE_REPORT,
                reg, &s, sizeof(s));

        *val = (s[1] << 8) | s[2];

        return status;
}

static usbd_status
slurm_setreg(struct slurm_softc *sc, int reg, uint16_t val)
{
        usbd_status status;
        uint8_t s[3];

        ++reg;

        s[0] = reg;
        s[1] = (val >> 8) & 0xff;
        s[2] = (val >> 0) & 0xff;

        status = usbd_set_report(sc->sc_uif, UHID_FEATURE_REPORT,
                reg, &s, sizeof(s));

        return status;
}

static int
slurm_si470x_await_stc(struct slurm_softc *sc)
{
        int i;
        uint16_t statusrssi;

        for (i = 50; i > 0; i--) {
                usbd_delay_ms(sc->sc_udev, 2);
                slurm_getreg(sc, SI470X_STATUSRSSI, &statusrssi);
                if ((statusrssi & (SI470X_STC|SI470X_SF_BL)) != 0)
                        break;
        }

        if (i == 0)
                return -1;
        else
                return 0;
}

static void
slurm_si470x_get_bandspace(struct slurm_softc *sc, uint16_t sysconfig2)
{
        switch (__SHIFTOUT(sysconfig2, SI470X_SPACE)) {
        default:
        case 0:
                sc->sc_space = 200;
                break;
        case 1:
                sc->sc_space = 100;
                break;
        case 2:
                sc->sc_space = 50;
                break;
        }

        switch (__SHIFTOUT(sysconfig2, SI470X_BAND)) {
        default:
        case 0:
                sc->sc_band = 87500;
                break;
        case 1:
        case 2:
                sc->sc_band = 76000;
                break;
        }
}

static uint32_t
slurm_si470x_get_freq(struct slurm_softc *sc, uint16_t readchannel)
{
        readchannel = __SHIFTOUT(readchannel, SI470X_READCHAN);
        return sc->sc_band + readchannel * sc->sc_space;
}

static int
slurm_si470x_get_info(uint16_t statusrssi)
{
        return (__SHIFTOUT(statusrssi, SI470X_ST) ? RADIO_INFO_STEREO : 0)
            | (__SHIFTOUT(statusrssi, SI470X_AFCRL) ? 0 : RADIO_INFO_SIGNAL);
}

static int
slurm_si470x_get_mute(uint16_t powercfg)
{
        return __SHIFTOUT(powercfg, SI470X_DMUTE) ? 0 : 1;
}

static int
slurm_si470x_get_stereo(uint16_t powercfg)
{
        return __SHIFTOUT(powercfg, SI470X_MONO) ? 0 : 1;
}

static int
slurm_si470x_get_volume(uint16_t sysconfig2)
{
        return __SHIFTOUT(sysconfig2, SI470X_VOLUME) * SI470X_VOLFACT;
}

static int
slurm_si470x_search(struct slurm_softc *sc, int up)
{
        uint16_t powercfg;

        slurm_getreg(sc, SI470X_POWERCFG, &powercfg);
        powercfg &= ~(SI470X_SKMODE|SI470X_SEEKUP|SI470X_SEEK);
        powercfg |= up ? SI470X_SEEKUP : 0;
        slurm_setreg(sc, SI470X_POWERCFG, SI470X_SEEK|powercfg);
        slurm_si470x_await_stc(sc);
        slurm_setreg(sc, SI470X_POWERCFG, powercfg);

        return 0;
}

static void
slurm_si470x_set_freq(struct slurm_softc *sc, uint32_t freq)
{
        uint16_t channel;

        channel = (freq - sc->sc_band) / sc->sc_space;

        slurm_setreg(sc, SI470X_CHANNEL, SI470X_TUNE|channel);
        slurm_si470x_await_stc(sc);
        slurm_setreg(sc, SI470X_CHANNEL, channel);

#ifdef SLURM_DEBUG
        device_printf(sc->sc_dev, "%s 0a -> %04x after %d\n", __func__, val, i);
#endif
}

static void
slurm_si470x_set_powercfg(struct slurm_softc *sc, int mute, int stereo)
{
        uint16_t powercfg;

        slurm_getreg(sc, SI470X_POWERCFG, &powercfg);
        powercfg &= ~(SI470X_DMUTE|SI470X_MONO);
        powercfg |= SI470X_DSMUTE;
        powercfg |= mute ? 0 : SI470X_DMUTE;
        powercfg |= stereo ? 0 : SI470X_MONO;
        slurm_setreg(sc, SI470X_POWERCFG, powercfg);
}

static void
slurm_si470x_set_volume(struct slurm_softc *sc, int volume)
{
        uint16_t sysconfig2;

        slurm_getreg(sc, SI470X_SYSCONFIG2, &sysconfig2);
        sysconfig2 &= ~SI470X_VOLUME;
        sysconfig2 |= __SHIFTIN(volume / SI470X_VOLFACT, SI470X_VOLUME);
        slurm_setreg(sc, SI470X_SYSCONFIG2, sysconfig2);
}