/*
 *   AD1843 low level driver
 *
 *   Copyright 2003 Vivien Chappelier <vivien.chappelier@linux-mips.org>
 *   Copyright 2008 Thomas Bogendoerfer <tsbogend@alpha.franken.de>
 *
 *   inspired from vwsnd.c (SGI VW audio driver)
 *     Copyright 1999 Silicon Graphics, Inc.  All rights reserved.
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 *
 */

#include <linux/init.h>
#include <linux/sched.h>
#include <linux/errno.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/ad1843.h>

/*
 * AD1843 bitfield definitions.  All are named as in the AD1843 data
 * sheet, with ad1843_ prepended and individual bit numbers removed.
 *
 * E.g., bits LSS0 through LSS2 become ad1843_LSS.
 *
 * Only the bitfields we need are defined.
 */

struct ad1843_bitfield {
        char reg;
        char lo_bit;
        char nbits;
};

static const struct ad1843_bitfield
        ad1843_PDNO   = {  0, 14,  1 }, /* Converter Power-Down Flag */
        ad1843_INIT   = {  0, 15,  1 }, /* Clock Initialization Flag */
        ad1843_RIG    = {  2,  0,  4 }, /* Right ADC Input Gain */
        ad1843_RMGE   = {  2,  4,  1 }, /* Right ADC Mic Gain Enable */
        ad1843_RSS    = {  2,  5,  3 }, /* Right ADC Source Select */
        ad1843_LIG    = {  2,  8,  4 }, /* Left ADC Input Gain */
        ad1843_LMGE   = {  2, 12,  1 }, /* Left ADC Mic Gain Enable */
        ad1843_LSS    = {  2, 13,  3 }, /* Left ADC Source Select */
        ad1843_RD2M   = {  3,  0,  5 }, /* Right DAC 2 Mix Gain/Atten */
        ad1843_RD2MM  = {  3,  7,  1 }, /* Right DAC 2 Mix Mute */
        ad1843_LD2M   = {  3,  8,  5 }, /* Left DAC 2 Mix Gain/Atten */
        ad1843_LD2MM  = {  3, 15,  1 }, /* Left DAC 2 Mix Mute */
        ad1843_RX1M   = {  4,  0,  5 }, /* Right Aux 1 Mix Gain/Atten */
        ad1843_RX1MM  = {  4,  7,  1 }, /* Right Aux 1 Mix Mute */
        ad1843_LX1M   = {  4,  8,  5 }, /* Left Aux 1 Mix Gain/Atten */
        ad1843_LX1MM  = {  4, 15,  1 }, /* Left Aux 1 Mix Mute */
        ad1843_RX2M   = {  5,  0,  5 }, /* Right Aux 2 Mix Gain/Atten */
        ad1843_RX2MM  = {  5,  7,  1 }, /* Right Aux 2 Mix Mute */
        ad1843_LX2M   = {  5,  8,  5 }, /* Left Aux 2 Mix Gain/Atten */
        ad1843_LX2MM  = {  5, 15,  1 }, /* Left Aux 2 Mix Mute */
        ad1843_RMCM   = {  7,  0,  5 }, /* Right Mic Mix Gain/Atten */
        ad1843_RMCMM  = {  7,  7,  1 }, /* Right Mic Mix Mute */
        ad1843_LMCM   = {  7,  8,  5 }, /* Left Mic Mix Gain/Atten */
        ad1843_LMCMM  = {  7, 15,  1 }, /* Left Mic Mix Mute */
        ad1843_HPOS   = {  8,  4,  1 }, /* Headphone Output Voltage Swing */
        ad1843_HPOM   = {  8,  5,  1 }, /* Headphone Output Mute */
        ad1843_MPOM   = {  8,  6,  1 }, /* Mono Output Mute */
        ad1843_RDA1G  = {  9,  0,  6 }, /* Right DAC1 Analog/Digital Gain */
        ad1843_RDA1GM = {  9,  7,  1 }, /* Right DAC1 Analog Mute */
        ad1843_LDA1G  = {  9,  8,  6 }, /* Left DAC1 Analog/Digital Gain */
        ad1843_LDA1GM = {  9, 15,  1 }, /* Left DAC1 Analog Mute */
        ad1843_RDA2G  = { 10,  0,  6 }, /* Right DAC2 Analog/Digital Gain */
        ad1843_RDA2GM = { 10,  7,  1 }, /* Right DAC2 Analog Mute */
        ad1843_LDA2G  = { 10,  8,  6 }, /* Left DAC2 Analog/Digital Gain */
        ad1843_LDA2GM = { 10, 15,  1 }, /* Left DAC2 Analog Mute */
        ad1843_RDA1AM = { 11,  7,  1 }, /* Right DAC1 Digital Mute */
        ad1843_LDA1AM = { 11, 15,  1 }, /* Left DAC1 Digital Mute */
        ad1843_RDA2AM = { 12,  7,  1 }, /* Right DAC2 Digital Mute */
        ad1843_LDA2AM = { 12, 15,  1 }, /* Left DAC2 Digital Mute */
        ad1843_ADLC   = { 15,  0,  2 }, /* ADC Left Sample Rate Source */
        ad1843_ADRC   = { 15,  2,  2 }, /* ADC Right Sample Rate Source */
        ad1843_DA1C   = { 15,  8,  2 }, /* DAC1 Sample Rate Source */
        ad1843_DA2C   = { 15, 10,  2 }, /* DAC2 Sample Rate Source */
        ad1843_C1C    = { 17,  0, 16 }, /* Clock 1 Sample Rate Select */
        ad1843_C2C    = { 20,  0, 16 }, /* Clock 2 Sample Rate Select */
        ad1843_C3C    = { 23,  0, 16 }, /* Clock 3 Sample Rate Select */
        ad1843_DAADL  = { 25,  4,  2 }, /* Digital ADC Left Source Select */
        ad1843_DAADR  = { 25,  6,  2 }, /* Digital ADC Right Source Select */
        ad1843_DAMIX  = { 25, 14,  1 }, /* DAC Digital Mix Enable */
        ad1843_DRSFLT = { 25, 15,  1 }, /* Digital Reampler Filter Mode */
        ad1843_ADLF   = { 26,  0,  2 }, /* ADC Left Channel Data Format */
        ad1843_ADRF   = { 26,  2,  2 }, /* ADC Right Channel Data Format */
        ad1843_ADTLK  = { 26,  4,  1 }, /* ADC Transmit Lock Mode Select */
        ad1843_SCF    = { 26,  7,  1 }, /* SCLK Frequency Select */
        ad1843_DA1F   = { 26,  8,  2 }, /* DAC1 Data Format Select */
        ad1843_DA2F   = { 26, 10,  2 }, /* DAC2 Data Format Select */
        ad1843_DA1SM  = { 26, 14,  1 }, /* DAC1 Stereo/Mono Mode Select */
        ad1843_DA2SM  = { 26, 15,  1 }, /* DAC2 Stereo/Mono Mode Select */
        ad1843_ADLEN  = { 27,  0,  1 }, /* ADC Left Channel Enable */
        ad1843_ADREN  = { 27,  1,  1 }, /* ADC Right Channel Enable */
        ad1843_AAMEN  = { 27,  4,  1 }, /* Analog to Analog Mix Enable */
        ad1843_ANAEN  = { 27,  7,  1 }, /* Analog Channel Enable */
        ad1843_DA1EN  = { 27,  8,  1 }, /* DAC1 Enable */
        ad1843_DA2EN  = { 27,  9,  1 }, /* DAC2 Enable */
        ad1843_DDMEN  = { 27, 12,  1 }, /* DAC2 to DAC1 Mix  Enable */
        ad1843_C1EN   = { 28, 11,  1 }, /* Clock Generator 1 Enable */
        ad1843_C2EN   = { 28, 12,  1 }, /* Clock Generator 2 Enable */
        ad1843_C3EN   = { 28, 13,  1 }, /* Clock Generator 3 Enable */
        ad1843_PDNI   = { 28, 15,  1 }; /* Converter Power Down */

/*
 * The various registers of the AD1843 use three different formats for
 * specifying gain.  The ad1843_gain structure parameterizes the
 * formats.
 */

struct ad1843_gain {
        int     negative;               /* nonzero if gain is negative. */
        const struct ad1843_bitfield *lfield;
        const struct ad1843_bitfield *rfield;
        const struct ad1843_bitfield *lmute;
        const struct ad1843_bitfield *rmute;
};

static const struct ad1843_gain ad1843_gain_RECLEV = {
        .negative = 0,
        .lfield   = &ad1843_LIG,
        .rfield   = &ad1843_RIG
};
static const struct ad1843_gain ad1843_gain_LINE = {
        .negative = 1,
        .lfield   = &ad1843_LX1M,
        .rfield   = &ad1843_RX1M,
        .lmute    = &ad1843_LX1MM,
        .rmute    = &ad1843_RX1MM
};
static const struct ad1843_gain ad1843_gain_LINE_2 = {
        .negative = 1,
        .lfield   = &ad1843_LDA2G,
        .rfield   = &ad1843_RDA2G,
        .lmute    = &ad1843_LDA2GM,
        .rmute    = &ad1843_RDA2GM
};
static const struct ad1843_gain ad1843_gain_MIC = {
        .negative = 1,
        .lfield   = &ad1843_LMCM,
        .rfield   = &ad1843_RMCM,
        .lmute    = &ad1843_LMCMM,
        .rmute    = &ad1843_RMCMM
};
static const struct ad1843_gain ad1843_gain_PCM_0 = {
        .negative = 1,
        .lfield   = &ad1843_LDA1G,
        .rfield   = &ad1843_RDA1G,
        .lmute    = &ad1843_LDA1GM,
        .rmute    = &ad1843_RDA1GM
};
static const struct ad1843_gain ad1843_gain_PCM_1 = {
        .negative = 1,
        .lfield   = &ad1843_LD2M,
        .rfield   = &ad1843_RD2M,
        .lmute    = &ad1843_LD2MM,
        .rmute    = &ad1843_RD2MM
};

static const struct ad1843_gain *ad1843_gain[AD1843_GAIN_SIZE] =
{
        &ad1843_gain_RECLEV,
        &ad1843_gain_LINE,
        &ad1843_gain_LINE_2,
        &ad1843_gain_MIC,
        &ad1843_gain_PCM_0,
        &ad1843_gain_PCM_1,
};

/* read the current value of an AD1843 bitfield. */

static int ad1843_read_bits(struct snd_ad1843 *ad1843,
                            const struct ad1843_bitfield *field)
{
        int w;

        w = ad1843->read(ad1843->chip, field->reg);
        return w >> field->lo_bit & ((1 << field->nbits) - 1);
}

/*
 * write a new value to an AD1843 bitfield and return the old value.
 */

static int ad1843_write_bits(struct snd_ad1843 *ad1843,
                             const struct ad1843_bitfield *field,
                             int newval)
{
        int w, mask, oldval, newbits;

        w = ad1843->read(ad1843->chip, field->reg);
        mask = ((1 << field->nbits) - 1) << field->lo_bit;
        oldval = (w & mask) >> field->lo_bit;
        newbits = (newval << field->lo_bit) & mask;
        w = (w & ~mask) | newbits;
        ad1843->write(ad1843->chip, field->reg, w);

        return oldval;
}

/*
 * ad1843_read_multi reads multiple bitfields from the same AD1843
 * register.  It uses a single read cycle to do it.  (Reading the
 * ad1843 requires 256 bit times at 12.288 MHz, or nearly 20
 * microseconds.)
 *
 * Called like this.
 *
 *  ad1843_read_multi(ad1843, nfields,
 *                    &ad1843_FIELD1, &val1,
 *                    &ad1843_FIELD2, &val2, ...);
 */

static void ad1843_read_multi(struct snd_ad1843 *ad1843, int argcount, ...)
{
        va_list ap;
        const struct ad1843_bitfield *fp;
        int w = 0, mask, *value, reg = -1;

        va_start(ap, argcount);
        while (--argcount >= 0) {
                fp = va_arg(ap, const struct ad1843_bitfield *);
                value = va_arg(ap, int *);
                if (reg == -1) {
                        reg = fp->reg;
                        w = ad1843->read(ad1843->chip, reg);
                }

                mask = (1 << fp->nbits) - 1;
                *value = w >> fp->lo_bit & mask;
        }
        va_end(ap);
}

/*
 * ad1843_write_multi stores multiple bitfields into the same AD1843
 * register.  It uses one read and one write cycle to do it.
 *
 * Called like this.
 *
 *  ad1843_write_multi(ad1843, nfields,
 *                     &ad1843_FIELD1, val1,
 *                     &ad1843_FIELF2, val2, ...);
 */

static void ad1843_write_multi(struct snd_ad1843 *ad1843, int argcount, ...)
{
        va_list ap;
        int reg;
        const struct ad1843_bitfield *fp;
        int value;
        int w, m, mask, bits;

        mask = 0;
        bits = 0;
        reg = -1;

        va_start(ap, argcount);
        while (--argcount >= 0) {
                fp = va_arg(ap, const struct ad1843_bitfield *);
                value = va_arg(ap, int);
                if (reg == -1)
                        reg = fp->reg;
                else
                        WARN_ON(reg != fp->reg);
                m = ((1 << fp->nbits) - 1) << fp->lo_bit;
                mask |= m;
                bits |= (value << fp->lo_bit) & m;
        }
        va_end(ap);

        if (~mask & 0xFFFF)
                w = ad1843->read(ad1843->chip, reg);
        else
                w = 0;
        w = (w & ~mask) | bits;
        ad1843->write(ad1843->chip, reg, w);
}

int ad1843_get_gain_max(struct snd_ad1843 *ad1843, int id)
{
        const struct ad1843_gain *gp = ad1843_gain[id];
        int ret;

        ret = (1 << gp->lfield->nbits);
        if (!gp->lmute)
                ret -= 1;
        return ret;
}

/*
 * ad1843_get_gain reads the specified register and extracts the gain value
 * using the supplied gain type.
 */

int ad1843_get_gain(struct snd_ad1843 *ad1843, int id)
{
        int lg, rg, lm, rm;
        const struct ad1843_gain *gp = ad1843_gain[id];
        unsigned short mask = (1 << gp->lfield->nbits) - 1;

        ad1843_read_multi(ad1843, 2, gp->lfield, &lg, gp->rfield, &rg);
        if (gp->negative) {
                lg = mask - lg;
                rg = mask - rg;
        }
        if (gp->lmute) {
                ad1843_read_multi(ad1843, 2, gp->lmute, &lm, gp->rmute, &rm);
                if (lm)
                        lg = 0;
                if (rm)
                        rg = 0;
        }
        return lg << 0 | rg << 8;
}

/*
 * Set an audio channel's gain.
 *
 * Returns the new gain, which may be lower than the old gain.
 */

int ad1843_set_gain(struct snd_ad1843 *ad1843, int id, int newval)
{
        const struct ad1843_gain *gp = ad1843_gain[id];
        unsigned short mask = (1 << gp->lfield->nbits) - 1;

        int lg = (newval >> 0) & mask;
        int rg = (newval >> 8) & mask;
        int lm = (lg == 0) ? 1 : 0;
        int rm = (rg == 0) ? 1 : 0;

        if (gp->negative) {
                lg = mask - lg;
                rg = mask - rg;
        }
        if (gp->lmute)
                ad1843_write_multi(ad1843, 2, gp->lmute, lm, gp->rmute, rm);
        ad1843_write_multi(ad1843, 2, gp->lfield, lg, gp->rfield, rg);
        return ad1843_get_gain(ad1843, id);
}

/* Returns the current recording source */

int ad1843_get_recsrc(struct snd_ad1843 *ad1843)
{
        int val = ad1843_read_bits(ad1843, &ad1843_LSS);

        if (val < 0 || val > 2) {
                val = 2;
                ad1843_write_multi(ad1843, 2,
                                   &ad1843_LSS, val, &ad1843_RSS, val);
        }
        return val;
}

/*
 * Set recording source.
 *
 * Returns newsrc on success, -errno on failure.
 */

int ad1843_set_recsrc(struct snd_ad1843 *ad1843, int newsrc)
{
        if (newsrc < 0 || newsrc > 2)
                return -EINVAL;

        ad1843_write_multi(ad1843, 2, &ad1843_LSS, newsrc, &ad1843_RSS, newsrc);
        return newsrc;
}

/* Setup ad1843 for D/A conversion. */

void ad1843_setup_dac(struct snd_ad1843 *ad1843,
                      unsigned int id,
                      unsigned int framerate,
                      snd_pcm_format_t fmt,
                      unsigned int channels)
{
        int ad_fmt = 0, ad_mode = 0;

        switch (fmt) {
        case SNDRV_PCM_FORMAT_S8:
                ad_fmt = 0;
                break;
        case SNDRV_PCM_FORMAT_U8:
                ad_fmt = 0;
                break;
        case SNDRV_PCM_FORMAT_S16_LE:
                ad_fmt = 1;
                break;
        case SNDRV_PCM_FORMAT_MU_LAW:
                ad_fmt = 2;
                break;
        case SNDRV_PCM_FORMAT_A_LAW:
                ad_fmt = 3;
                break;
        default:
                break;
        }

        switch (channels) {
        case 2:
                ad_mode = 0;
                break;
        case 1:
                ad_mode = 1;
                break;
        default:
                break;
        }

        if (id) {
                ad1843_write_bits(ad1843, &ad1843_C2C, framerate);
                ad1843_write_multi(ad1843, 2,
                                   &ad1843_DA2SM, ad_mode,
                                   &ad1843_DA2F, ad_fmt);
        } else {
                ad1843_write_bits(ad1843, &ad1843_C1C, framerate);
                ad1843_write_multi(ad1843, 2,
                                   &ad1843_DA1SM, ad_mode,
                                   &ad1843_DA1F, ad_fmt);
        }
}

void ad1843_shutdown_dac(struct snd_ad1843 *ad1843, unsigned int id)
{
        if (id)
                ad1843_write_bits(ad1843, &ad1843_DA2F, 1);
        else
                ad1843_write_bits(ad1843, &ad1843_DA1F, 1);
}

void ad1843_setup_adc(struct snd_ad1843 *ad1843,
                      unsigned int framerate,
                      snd_pcm_format_t fmt,
                      unsigned int channels)
{
        int da_fmt = 0;

        switch (fmt) {
        case SNDRV_PCM_FORMAT_S8:       da_fmt = 0; break;
        case SNDRV_PCM_FORMAT_U8:       da_fmt = 0; break;
        case SNDRV_PCM_FORMAT_S16_LE:   da_fmt = 1; break;
        case SNDRV_PCM_FORMAT_MU_LAW:   da_fmt = 2; break;
        case SNDRV_PCM_FORMAT_A_LAW:    da_fmt = 3; break;
        default:                break;
        }

        ad1843_write_bits(ad1843, &ad1843_C3C, framerate);
        ad1843_write_multi(ad1843, 2,
                           &ad1843_ADLF, da_fmt, &ad1843_ADRF, da_fmt);
}

void ad1843_shutdown_adc(struct snd_ad1843 *ad1843)
{
        /* nothing to do */
}

/*
 * Fully initialize the ad1843.  As described in the AD1843 data
 * sheet, section "START-UP SEQUENCE".  The numbered comments are
 * subsection headings from the data sheet.  See the data sheet, pages
 * 52-54, for more info.
 *
 * return 0 on success, -errno on failure.  */

int ad1843_init(struct snd_ad1843 *ad1843)
{
        unsigned long later;

        if (ad1843_read_bits(ad1843, &ad1843_INIT) != 0) {
                printk(KERN_ERR "ad1843: AD1843 won't initialize\n");
                return -EIO;
        }

        ad1843_write_bits(ad1843, &ad1843_SCF, 1);

        /* 4. Put the conversion resources into standby. */
        ad1843_write_bits(ad1843, &ad1843_PDNI, 0);
        later = jiffies + msecs_to_jiffies(500);

        while (ad1843_read_bits(ad1843, &ad1843_PDNO)) {
                if (time_after(jiffies, later)) {
                        printk(KERN_ERR
                               "ad1843: AD1843 won't power up\n");
                        return -EIO;
                }
                schedule_timeout_interruptible(5);
        }

        /* 5. Power up the clock generators and enable clock output pins. */
        ad1843_write_multi(ad1843, 3,
                           &ad1843_C1EN, 1,
                           &ad1843_C2EN, 1,
                           &ad1843_C3EN, 1);

        /* 6. Configure conversion resources while they are in standby. */

        /* DAC1/2 use clock 1/2 as source, ADC uses clock 3.  Always. */
        ad1843_write_multi(ad1843, 4,
                           &ad1843_DA1C, 1,
                           &ad1843_DA2C, 2,
                           &ad1843_ADLC, 3,
                           &ad1843_ADRC, 3);

        /* 7. Enable conversion resources. */
        ad1843_write_bits(ad1843, &ad1843_ADTLK, 1);
        ad1843_write_multi(ad1843, 7,
                           &ad1843_ANAEN, 1,
                           &ad1843_AAMEN, 1,
                           &ad1843_DA1EN, 1,
                           &ad1843_DA2EN, 1,
                           &ad1843_DDMEN, 1,
                           &ad1843_ADLEN, 1,
                           &ad1843_ADREN, 1);

        /* 8. Configure conversion resources while they are enabled. */

        /* set gain to 0 for all channels */
        ad1843_set_gain(ad1843, AD1843_GAIN_RECLEV, 0);
        ad1843_set_gain(ad1843, AD1843_GAIN_LINE, 0);
        ad1843_set_gain(ad1843, AD1843_GAIN_LINE_2, 0);
        ad1843_set_gain(ad1843, AD1843_GAIN_MIC, 0);
        ad1843_set_gain(ad1843, AD1843_GAIN_PCM_0, 0);
        ad1843_set_gain(ad1843, AD1843_GAIN_PCM_1, 0);

        /* Unmute all channels. */
        /* DAC1 */
        ad1843_write_multi(ad1843, 2, &ad1843_LDA1GM, 0, &ad1843_RDA1GM, 0);
        /* DAC2 */
        ad1843_write_multi(ad1843, 2, &ad1843_LDA2GM, 0, &ad1843_RDA2GM, 0);

        /* Set default recording source to Line In and set
         * mic gain to +20 dB.
         */
        ad1843_set_recsrc(ad1843, 2);
        ad1843_write_multi(ad1843, 2, &ad1843_LMGE, 1, &ad1843_RMGE, 1);

        /* Set Speaker Out level to +/- 4V and unmute it. */
        ad1843_write_multi(ad1843, 3,
                           &ad1843_HPOS, 1,
                           &ad1843_HPOM, 0,
                           &ad1843_MPOM, 0);

        return 0;
}