/*

 * WM8750 audio CODEC.

 *

 * Copyright (c) 2006 Openedhand Ltd.

 * Written by Andrzej Zaborowski <balrog@zabor.org>

 *

 * This file is licensed under GNU GPL.

 */

#include "hw.h"

#include "i2c.h"

#include "audio/audio.h"

#define IN_PORT_N	3

#define OUT_PORT_N	3

#define CODEC		"wm8750"

struct wm_rate_s;

struct wm8750_s {

    i2c_slave i2c;

    uint8_t i2c_data[2];

    int i2c_len;

    QEMUSoundCard card;

    SWVoiceIn *adc_voice[IN_PORT_N];

    SWVoiceOut *dac_voice[OUT_PORT_N];

    int enable;

    void (*data_req)(void *, int, int);

    void *opaque;

    uint8_t data_in[4096];

    uint8_t data_out[4096];

    int idx_in, req_in;

    int idx_out, req_out;

    SWVoiceOut **out[2];

    uint8_t outvol[7], outmute[2];

    SWVoiceIn **in[2];

    uint8_t invol[4], inmute[2];

    uint8_t diff[2], pol, ds, monomix[2], alc, mute;

    uint8_t path[4], mpath[2], power, format;

    const struct wm_rate_s *rate;

    int adc_hz, dac_hz, ext_adc_hz, ext_dac_hz, master;

};

/* pow(10.0, -i / 20.0) * 255, i = 0..42 */

static const uint8_t wm8750_vol_db_table[] = {

    255, 227, 203, 181, 161, 143, 128, 114, 102, 90, 81, 72, 64, 57, 51, 45,

    40, 36, 32, 29, 26, 23, 20, 18, 16, 14, 13, 11, 10, 9, 8, 7, 6, 6, 5, 5,

    4, 4, 3, 3, 3, 2, 2

};

#define WM8750_OUTVOL_TRANSFORM(x)	wm8750_vol_db_table[(0x7f - x) / 3]

#define WM8750_INVOL_TRANSFORM(x)	(x << 2)

static inline void wm8750_in_load(struct wm8750_s *s)

{

    int acquired;

    if (s->idx_in + s->req_in <= sizeof(s->data_in))

        return;

    s->idx_in = audio_MAX(0, (int) sizeof(s->data_in) - s->req_in);

    acquired = AUD_read(*s->in[0], s->data_in + s->idx_in,

                    sizeof(s->data_in) - s->idx_in);

}

static inline void wm8750_out_flush(struct wm8750_s *s)

{

    int sent = 0;

    while (sent < s->idx_out)

        sent += AUD_write(*s->out[0], s->data_out + sent, s->idx_out - sent)

                ?: s->idx_out;

    s->idx_out = 0;

}

static void wm8750_audio_in_cb(void *opaque, int avail_b)

{

    struct wm8750_s *s = (struct wm8750_s *) opaque;

    s->req_in = avail_b;

    s->data_req(s->opaque, s->req_out >> 2, avail_b >> 2);

}

static void wm8750_audio_out_cb(void *opaque, int free_b)

{

    struct wm8750_s *s = (struct wm8750_s *) opaque;

    if (s->idx_out >= free_b) {

        s->idx_out = free_b;

        s->req_out = 0;

        wm8750_out_flush(s);

    } else

        s->req_out = free_b - s->idx_out;

    s->data_req(s->opaque, s->req_out >> 2, s->req_in >> 2);

}

struct wm_rate_s {

    int adc;

    int adc_hz;

    int dac;

    int dac_hz;

};

static const struct wm_rate_s wm_rate_table[] = {

    {  256, 48000,  256, 48000 },	/* SR: 00000 */

    {  384, 48000,  384, 48000 },	/* SR: 00001 */

    {  256, 48000, 1536,  8000 },	/* SR: 00010 */

    {  384, 48000, 2304,  8000 },	/* SR: 00011 */

    { 1536,  8000,  256, 48000 },	/* SR: 00100 */

    { 2304,  8000,  384, 48000 },	/* SR: 00101 */

    { 1536,  8000, 1536,  8000 },	/* SR: 00110 */

    { 2304,  8000, 2304,  8000 },	/* SR: 00111 */

    { 1024, 12000, 1024, 12000 },	/* SR: 01000 */

    { 1526, 12000, 1536, 12000 },	/* SR: 01001 */

    {  768, 16000,  768, 16000 },	/* SR: 01010 */

    { 1152, 16000, 1152, 16000 },	/* SR: 01011 */

    {  384, 32000,  384, 32000 },	/* SR: 01100 */

    {  576, 32000,  576, 32000 },	/* SR: 01101 */

    {  128, 96000,  128, 96000 },	/* SR: 01110 */

    {  192, 96000,  192, 96000 },	/* SR: 01111 */

    {  256, 44100,  256, 44100 },	/* SR: 10000 */

    {  384, 44100,  384, 44100 },	/* SR: 10001 */

    {  256, 44100, 1408,  8018 },	/* SR: 10010 */

    {  384, 44100, 2112,  8018 },	/* SR: 10011 */

    { 1408,  8018,  256, 44100 },	/* SR: 10100 */

    { 2112,  8018,  384, 44100 },	/* SR: 10101 */

    { 1408,  8018, 1408,  8018 },	/* SR: 10110 */

    { 2112,  8018, 2112,  8018 },	/* SR: 10111 */

    { 1024, 11025, 1024, 11025 },	/* SR: 11000 */

    { 1536, 11025, 1536, 11025 },	/* SR: 11001 */

    {  512, 22050,  512, 22050 },	/* SR: 11010 */

    {  768, 22050,  768, 22050 },	/* SR: 11011 */

    {  512, 24000,  512, 24000 },	/* SR: 11100 */

    {  768, 24000,  768, 24000 },	/* SR: 11101 */

    {  128, 88200,  128, 88200 },	/* SR: 11110 */

    {  192, 88200,  192, 88200 },	/* SR: 11111 */

};

static void wm8750_vol_update(struct wm8750_s *s)

{

    /* FIXME: multiply all volumes by s->invol[2], s->invol[3] */

    AUD_set_volume_in(s->adc_voice[0], s->mute,

                    s->inmute[0] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[0]),

                    s->inmute[1] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[1]));

    AUD_set_volume_in(s->adc_voice[1], s->mute,

                    s->inmute[0] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[0]),

                    s->inmute[1] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[1]));

    AUD_set_volume_in(s->adc_voice[2], s->mute,

                    s->inmute[0] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[0]),

                    s->inmute[1] ? 0 : WM8750_INVOL_TRANSFORM(s->invol[1]));

    /* FIXME: multiply all volumes by s->outvol[0], s->outvol[1] */

    /* Speaker: LOUT2VOL ROUT2VOL */

    AUD_set_volume_out(s->dac_voice[0], s->mute,

                    s->outmute[0] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[4]),

                    s->outmute[1] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[5]));

    /* Headphone: LOUT1VOL ROUT1VOL */

    AUD_set_volume_out(s->dac_voice[1], s->mute,

                    s->outmute[0] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[2]),

                    s->outmute[1] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[3]));

    /* MONOOUT: MONOVOL MONOVOL */

    AUD_set_volume_out(s->dac_voice[2], s->mute,

                    s->outmute[0] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[6]),

                    s->outmute[1] ? 0 : WM8750_OUTVOL_TRANSFORM(s->outvol[6]));

}

static void wm8750_set_format(struct wm8750_s *s)

{

    int i;

    struct audsettings in_fmt;

    struct audsettings out_fmt;

    struct audsettings monoout_fmt;

    wm8750_out_flush(s);

    if (s->in[0] && *s->in[0])

        AUD_set_active_in(*s->in[0], 0);

    if (s->out[0] && *s->out[0])

        AUD_set_active_out(*s->out[0], 0);

    for (i = 0; i < IN_PORT_N; i ++)

        if (s->adc_voice[i]) {

            AUD_close_in(&s->card, s->adc_voice[i]);

            s->adc_voice[i] = 0;

        }

    for (i = 0; i < OUT_PORT_N; i ++)

        if (s->dac_voice[i]) {

            AUD_close_out(&s->card, s->dac_voice[i]);

            s->dac_voice[i] = 0;

        }

    if (!s->enable)

        return;

    /* Setup input */

    in_fmt.endianness = 0;

    in_fmt.nchannels = 2;

    in_fmt.freq = s->adc_hz;

    in_fmt.fmt = AUD_FMT_S16;

    s->adc_voice[0] = AUD_open_in(&s->card, s->adc_voice[0],

                    CODEC ".input1", s, wm8750_audio_in_cb, &in_fmt);

    s->adc_voice[1] = AUD_open_in(&s->card, s->adc_voice[1],

                    CODEC ".input2", s, wm8750_audio_in_cb, &in_fmt);

    s->adc_voice[2] = AUD_open_in(&s->card, s->adc_voice[2],

                    CODEC ".input3", s, wm8750_audio_in_cb, &in_fmt);

    /* Setup output */

    out_fmt.endianness = 0;

    out_fmt.nchannels = 2;

    out_fmt.freq = s->dac_hz;

    out_fmt.fmt = AUD_FMT_S16;

    monoout_fmt.endianness = 0;

    monoout_fmt.nchannels = 1;

    monoout_fmt.freq = s->rate->dac_hz;

    monoout_fmt.fmt = AUD_FMT_S16;

    s->dac_voice[0] = AUD_open_out(&s->card, s->dac_voice[0],

                    CODEC ".speaker", s, wm8750_audio_out_cb, &out_fmt);

    s->dac_voice[1] = AUD_open_out(&s->card, s->dac_voice[1],

                    CODEC ".headphone", s, wm8750_audio_out_cb, &out_fmt);

    /* MONOMIX is also in stereo for simplicity */

    s->dac_voice[2] = AUD_open_out(&s->card, s->dac_voice[2],

                    CODEC ".monomix", s, wm8750_audio_out_cb, &out_fmt);

    /* no sense emulating OUT3 which is a mix of other outputs */

    wm8750_vol_update(s);

    /* We should connect the left and right channels to their

     * respective inputs/outputs but we have completely no need

     * for mixing or combining paths to different ports, so we

     * connect both channels to where the left channel is routed.  */

    if (s->in[0] && *s->in[0])

        AUD_set_active_in(*s->in[0], 1);

    if (s->out[0] && *s->out[0])

        AUD_set_active_out(*s->out[0], 1);

}

static void wm8750_clk_update(struct wm8750_s *s, int ext)

{

    if (s->master || !s->ext_dac_hz)

        s->dac_hz = s->rate->dac_hz;

    else

        s->dac_hz = s->ext_dac_hz;

    if (s->master || !s->ext_adc_hz)

        s->adc_hz = s->rate->adc_hz;

    else

        s->adc_hz = s->ext_adc_hz;

    if (s->master || (!s->ext_dac_hz && !s->ext_adc_hz)) {

        if (!ext)

            wm8750_set_format(s);

    } else {

        if (ext)

            wm8750_set_format(s);

    }

}

void wm8750_reset(i2c_slave *i2c)

{

    struct wm8750_s *s = (struct wm8750_s *) i2c;

    s->rate = &wm_rate_table[0];

    s->enable = 0;

    wm8750_clk_update(s, 1);

    s->diff[0] = 0;

    s->diff[1] = 0;

    s->ds = 0;

    s->alc = 0;

    s->in[0] = &s->adc_voice[0];

    s->invol[0] = 0x17;

    s->invol[1] = 0x17;

    s->invol[2] = 0xc3;

    s->invol[3] = 0xc3;

    s->out[0] = &s->dac_voice[0];

    s->outvol[0] = 0xff;

    s->outvol[1] = 0xff;

    s->outvol[2] = 0x79;

    s->outvol[3] = 0x79;

    s->outvol[4] = 0x79;

    s->outvol[5] = 0x79;

    s->outvol[6] = 0x79;

    s->inmute[0] = 0;

    s->inmute[1] = 0;

    s->outmute[0] = 0;

    s->outmute[1] = 0;

    s->mute = 1;

    s->path[0] = 0;

    s->path[1] = 0;

    s->path[2] = 0;

    s->path[3] = 0;

    s->mpath[0] = 0;

    s->mpath[1] = 0;

    s->format = 0x0a;

    s->idx_in = sizeof(s->data_in);

    s->req_in = 0;

    s->idx_out = 0;

    s->req_out = 0;

    wm8750_vol_update(s);

    s->i2c_len = 0;

}

static void wm8750_event(i2c_slave *i2c, enum i2c_event event)

{

    struct wm8750_s *s = (struct wm8750_s *) i2c;

    switch (event) {

    case I2C_START_SEND:

        s->i2c_len = 0;

        break;

    case I2C_FINISH:

#ifdef VERBOSE

        if (s->i2c_len < 2)

            printf("%s: message too short (%i bytes)\n",

                            __FUNCTION__, s->i2c_len);

#endif

        break;

    default:

        break;

    }

}

#define WM8750_LINVOL	0x00

#define WM8750_RINVOL	0x01

#define WM8750_LOUT1V	0x02

#define WM8750_ROUT1V	0x03

#define WM8750_ADCDAC	0x05

#define WM8750_IFACE	0x07

#define WM8750_SRATE	0x08

#define WM8750_LDAC	0x0a

#define WM8750_RDAC	0x0b

#define WM8750_BASS	0x0c

#define WM8750_TREBLE	0x0d

#define WM8750_RESET	0x0f

#define WM8750_3D	0x10

#define WM8750_ALC1	0x11

#define WM8750_ALC2	0x12

#define WM8750_ALC3	0x13

#define WM8750_NGATE	0x14

#define WM8750_LADC	0x15

#define WM8750_RADC	0x16

#define WM8750_ADCTL1	0x17

#define WM8750_ADCTL2	0x18

#define WM8750_PWR1	0x19

#define WM8750_PWR2	0x1a

#define WM8750_ADCTL3	0x1b

#define WM8750_ADCIN	0x1f

#define WM8750_LADCIN	0x20

#define WM8750_RADCIN	0x21

#define WM8750_LOUTM1	0x22

#define WM8750_LOUTM2	0x23

#define WM8750_ROUTM1	0x24

#define WM8750_ROUTM2	0x25

#define WM8750_MOUTM1	0x26

#define WM8750_MOUTM2	0x27

#define WM8750_LOUT2V	0x28

#define WM8750_ROUT2V	0x29

#define WM8750_MOUTV	0x2a

static int wm8750_tx(i2c_slave *i2c, uint8_t data)

{

    struct wm8750_s *s = (struct wm8750_s *) i2c;

    uint8_t cmd;

    uint16_t value;

    if (s->i2c_len >= 2) {

        printf("%s: long message (%i bytes)\n", __FUNCTION__, s->i2c_len);

#ifdef VERBOSE

        return 1;

#endif

    }

    s->i2c_data[s->i2c_len ++] = data;

    if (s->i2c_len != 2)

        return 0;

    cmd = s->i2c_data[0] >> 1;

    value = ((s->i2c_data[0] << 8) | s->i2c_data[1]) & 0x1ff;

    switch (cmd) {

    case WM8750_LADCIN:	/* ADC Signal Path Control (Left) */

        s->diff[0] = (((value >> 6) & 3) == 3);	/* LINSEL */

        if (s->diff[0])

            s->in[0] = &s->adc_voice[0 + s->ds * 1];

        else

            s->in[0] = &s->adc_voice[((value >> 6) & 3) * 1 + 0];

        break;

    case WM8750_RADCIN:	/* ADC Signal Path Control (Right) */

        s->diff[1] = (((value >> 6) & 3) == 3);	/* RINSEL */

        if (s->diff[1])

            s->in[1] = &s->adc_voice[0 + s->ds * 1];

        else

            s->in[1] = &s->adc_voice[((value >> 6) & 3) * 1 + 0];

        break;

    case WM8750_ADCIN:	/* ADC Input Mode */

        s->ds = (value >> 8) & 1;	/* DS */

        if (s->diff[0])

            s->in[0] = &s->adc_voice[0 + s->ds * 1];

        if (s->diff[1])

            s->in[1] = &s->adc_voice[0 + s->ds * 1];

        s->monomix[0] = (value >> 6) & 3;	/* MONOMIX */

        break;

    case WM8750_ADCTL1:	/* Additional Control (1) */

        s->monomix[1] = (value >> 1) & 1;	/* DMONOMIX */

        break;

    case WM8750_PWR1:	/* Power Management (1) */

        s->enable = ((value >> 6) & 7) == 3;	/* VMIDSEL, VREF */

        wm8750_set_format(s);

        break;

    case WM8750_LINVOL:	/* Left Channel PGA */

        s->invol[0] = value & 0x3f;		/* LINVOL */

        s->inmute[0] = (value >> 7) & 1;	/* LINMUTE */

        wm8750_vol_update(s);

        break;

    case WM8750_RINVOL:	/* Right Channel PGA */

        s->invol[1] = value & 0x3f;		/* RINVOL */

        s->inmute[1] = (value >> 7) & 1;	/* RINMUTE */

        wm8750_vol_update(s);

        break;

    case WM8750_ADCDAC:	/* ADC and DAC Control */

        s->pol = (value >> 5) & 3;		/* ADCPOL */

        s->mute = (value >> 3) & 1;		/* DACMU */

        wm8750_vol_update(s);

        break;

    case WM8750_ADCTL3:	/* Additional Control (3) */

        break;

    case WM8750_LADC:	/* Left ADC Digital Volume */

        s->invol[2] = value & 0xff;		/* LADCVOL */

        wm8750_vol_update(s);

        break;

    case WM8750_RADC:	/* Right ADC Digital Volume */

        s->invol[3] = value & 0xff;		/* RADCVOL */

        wm8750_vol_update(s);

        break;

    case WM8750_ALC1:	/* ALC Control (1) */

        s->alc = (value >> 7) & 3;		/* ALCSEL */

        break;

    case WM8750_NGATE:	/* Noise Gate Control */

    case WM8750_3D:	/* 3D enhance */

        break;

    case WM8750_LDAC:	/* Left Channel Digital Volume */

        s->outvol[0] = value & 0xff;		/* LDACVOL */

        wm8750_vol_update(s);

        break;

    case WM8750_RDAC:	/* Right Channel Digital Volume */

        s->outvol[1] = value & 0xff;		/* RDACVOL */

        wm8750_vol_update(s);

        break;

    case WM8750_BASS:	/* Bass Control */

        break;

    case WM8750_LOUTM1:	/* Left Mixer Control (1) */

        s->path[0] = (value >> 8) & 1;		/* LD2LO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_LOUTM2:	/* Left Mixer Control (2) */

        s->path[1] = (value >> 8) & 1;		/* RD2LO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_ROUTM1:	/* Right Mixer Control (1) */

        s->path[2] = (value >> 8) & 1;		/* LD2RO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_ROUTM2:	/* Right Mixer Control (2) */

        s->path[3] = (value >> 8) & 1;		/* RD2RO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_MOUTM1:	/* Mono Mixer Control (1) */

        s->mpath[0] = (value >> 8) & 1;		/* LD2MO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_MOUTM2:	/* Mono Mixer Control (2) */

        s->mpath[1] = (value >> 8) & 1;		/* RD2MO */

        /* TODO: mute/unmute respective paths */

        wm8750_vol_update(s);

        break;

    case WM8750_LOUT1V:	/* LOUT1 Volume */

        s->outvol[2] = value & 0x7f;		/* LOUT1VOL */

        wm8750_vol_update(s);

        break;

    case WM8750_LOUT2V:	/* LOUT2 Volume */

        s->outvol[4] = value & 0x7f;		/* LOUT2VOL */

        wm8750_vol_update(s);

        break;

    case WM8750_ROUT1V:	/* ROUT1 Volume */

        s->outvol[3] = value & 0x7f;		/* ROUT1VOL */

        wm8750_vol_update(s);

        break;

    case WM8750_ROUT2V:	/* ROUT2 Volume */

        s->outvol[5] = value & 0x7f;		/* ROUT2VOL */

        wm8750_vol_update(s);

        break;

    case WM8750_MOUTV:	/* MONOOUT Volume */

        s->outvol[6] = value & 0x7f;		/* MONOOUTVOL */

        wm8750_vol_update(s);

        break;