root/sound/soc/codecs/es8326.c
// SPDX-License-Identifier: GPL-2.0-only
//
// es8326.c -- es8326 ALSA SoC audio driver
// Copyright Everest Semiconductor Co., Ltd
//
// Authors: David Yang <yangxiaohua@everest-semi.com>
//

#include <linux/clk.h>
#include <linux/i2c.h>
#include <linux/interrupt.h>
#include <linux/irq.h>
#include <linux/module.h>
#include <sound/jack.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/soc-dapm.h>
#include <sound/tlv.h>
#include "es8326.h"

struct es8326_priv {
        struct clk *mclk;
        struct i2c_client *i2c;
        struct regmap *regmap;
        struct snd_soc_component *component;
        struct delayed_work jack_detect_work;
        struct delayed_work button_press_work;
        struct snd_soc_jack *jack;
        int irq;
        /* The lock protects the situation that an irq is generated
         * while enabling or disabling or during an irq.
         */
        struct mutex lock;
        u8 jack_pol;
        u8 interrupt_src;
        u8 interrupt_clk;
        u8 hpl_vol;
        u8 hpr_vol;
        bool jd_inverted;
        unsigned int sysclk;

        bool calibrated;
        int version;
        int hp;
        int jack_remove_retry;
};

static int es8326_crosstalk1_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int crosstalk_h, crosstalk_l;
        unsigned int crosstalk;

        regmap_read(es8326->regmap, ES8326_DAC_RAMPRATE, &crosstalk_h);
        regmap_read(es8326->regmap, ES8326_DAC_CROSSTALK, &crosstalk_l);
        crosstalk_h &= 0x20;
        crosstalk_l &= 0xf0;
        crosstalk = crosstalk_h >> 1 | crosstalk_l >> 4;
        ucontrol->value.integer.value[0] = crosstalk;

        return 0;
}

static int es8326_crosstalk1_set(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int crosstalk_h, crosstalk_l;
        unsigned int crosstalk;

        crosstalk = ucontrol->value.integer.value[0];
        regmap_read(es8326->regmap, ES8326_DAC_CROSSTALK, &crosstalk_l);
        crosstalk_h = (crosstalk & 0x10) << 1;
        crosstalk_l &= 0x0f;
        crosstalk_l |= (crosstalk & 0x0f) << 4;
        regmap_update_bits(es8326->regmap, ES8326_DAC_RAMPRATE,
                        0x20, crosstalk_h);
        regmap_write(es8326->regmap, ES8326_DAC_CROSSTALK, crosstalk_l);

        return 0;
}

static int es8326_crosstalk2_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int crosstalk_h, crosstalk_l;
        unsigned int crosstalk;

        regmap_read(es8326->regmap, ES8326_DAC_RAMPRATE, &crosstalk_h);
        regmap_read(es8326->regmap, ES8326_DAC_CROSSTALK, &crosstalk_l);
        crosstalk_h &= 0x10;
        crosstalk_l &= 0x0f;
        crosstalk = crosstalk_h  | crosstalk_l;
        ucontrol->value.integer.value[0] = crosstalk;

        return 0;
}

static int es8326_crosstalk2_set(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int crosstalk_h, crosstalk_l;
        unsigned int crosstalk;

        crosstalk = ucontrol->value.integer.value[0];
        regmap_read(es8326->regmap, ES8326_DAC_CROSSTALK, &crosstalk_l);
        crosstalk_h = crosstalk & 0x10;
        crosstalk_l &= 0xf0;
        crosstalk_l |= crosstalk & 0x0f;
        regmap_update_bits(es8326->regmap, ES8326_DAC_RAMPRATE,
                        0x10, crosstalk_h);
        regmap_write(es8326->regmap, ES8326_DAC_CROSSTALK, crosstalk_l);

        return 0;
}

static int es8326_hplvol_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = es8326->hpl_vol;

        return 0;
}

static int es8326_hplvol_set(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int hp_vol;

        hp_vol = ucontrol->value.integer.value[0];
        if (hp_vol > 5)
                return -EINVAL;
        if (es8326->hpl_vol != hp_vol) {
                es8326->hpl_vol = hp_vol;
                if (hp_vol >= 3)
                        hp_vol++;
                regmap_update_bits(es8326->regmap, ES8326_HP_VOL,
                                0x70, (hp_vol << 4));
                return 1;
        }

        return 0;
}

static int es8326_hprvol_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = es8326->hpr_vol;

        return 0;
}

static int es8326_hprvol_set(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int hp_vol;

        hp_vol = ucontrol->value.integer.value[0];
        if (hp_vol > 5)
                return -EINVAL;
        if (es8326->hpr_vol != hp_vol) {
                es8326->hpr_vol = hp_vol;
                if (hp_vol >= 3)
                        hp_vol++;
                regmap_update_bits(es8326->regmap, ES8326_HP_VOL,
                                0x07, hp_vol);
                return 1;
        }

        return 0;
}

static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(dac_vol_tlv, -9550, 50, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(adc_vol_tlv, -9550, 50, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(adc_analog_pga_tlv, 0, 300, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(adc_pga_tlv, 0, 600, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(softramp_rate, 0, 100, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(drc_target_tlv, -3200, 200, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(drc_recovery_tlv, -125, 250, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(dre_gain_tlv, -9550, 400, 0);
static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(dre_gate_tlv, -9600, 600, 0);

static const char *const winsize[] = {
        "0.25db/2  LRCK",
        "0.25db/4  LRCK",
        "0.25db/8  LRCK",
        "0.25db/16  LRCK",
        "0.25db/32  LRCK",
        "0.25db/64  LRCK",
        "0.25db/128  LRCK",
        "0.25db/256  LRCK",
        "0.25db/512  LRCK",
        "0.25db/1024  LRCK",
        "0.25db/2048  LRCK",
        "0.25db/4096  LRCK",
        "0.25db/8192  LRCK",
        "0.25db/16384  LRCK",
        "0.25db/32768  LRCK",
        "0.25db/65536  LRCK",
};

static const char *const dacpol_txt[] = {
        "Normal", "R Invert", "L Invert", "L + R Invert" };

static const char *const hp_spkvol_switch[] = {
        "HPVOL: HPL+HPL, SPKVOL: HPL+HPL",
        "HPVOL: HPL+HPR, SPKVOL: HPL+HPR",
        "HPVOL: HPL+HPL, SPKVOL: SPKL+SPKR",
        "HPVOL: HPL+HPR, SPKVOL: SPKL+SPKR",
};

static const struct soc_enum dacpol =
        SOC_ENUM_SINGLE(ES8326_DAC_DSM, 4, 4, dacpol_txt);
static const struct soc_enum dre_winsize =
        SOC_ENUM_SINGLE(ES8326_ADC_DRE, 0, 16, winsize);
static const struct soc_enum alc_winsize =
        SOC_ENUM_SINGLE(ES8326_ADC_RAMPRATE, 4, 16, winsize);
static const struct soc_enum drc_winsize =
        SOC_ENUM_SINGLE(ES8326_DRC_WINSIZE, 4, 16, winsize);
static const struct soc_enum hpvol_spkvol_switch =
        SOC_ENUM_SINGLE(ES8326_HP_MISC, 6, 4, hp_spkvol_switch);

static const struct snd_kcontrol_new es8326_snd_controls[] = {
        SOC_SINGLE_TLV("DAC Playback Volume", ES8326_DACL_VOL, 0, 0xbf, 0, dac_vol_tlv),
        SOC_ENUM("Playback Polarity", dacpol),
        SOC_SINGLE_TLV("DAC Ramp Rate", ES8326_DAC_RAMPRATE, 0, 0x0f, 0, softramp_rate),
        SOC_SINGLE_TLV("DRC Recovery Level", ES8326_DRC_RECOVERY, 0, 4, 0, drc_recovery_tlv),
        SOC_ENUM("DRC Winsize", drc_winsize),
        SOC_SINGLE_TLV("DRC Target Level", ES8326_DRC_WINSIZE, 0, 0x0f, 0, drc_target_tlv),

        SOC_DOUBLE_R_TLV("ADC Capture Volume", ES8326_ADC1_VOL, ES8326_ADC2_VOL, 0, 0xff, 0,
                         adc_vol_tlv),
        SOC_DOUBLE_TLV("ADC PGA Volume", ES8326_ADC_SCALE, 4, 0, 5, 0, adc_pga_tlv),
        SOC_SINGLE_TLV("ADC PGA Gain Volume", ES8326_PGAGAIN, 0, 10, 0, adc_analog_pga_tlv),
        SOC_SINGLE("ADC PGA SE Switch", ES8326_PGAGAIN, 7, 1, 0),
        SOC_SINGLE_TLV("ADC Ramp Rate", ES8326_ADC_RAMPRATE, 0, 0x0f, 0, softramp_rate),
        SOC_SINGLE("ADC4 DRE Switch", ES8326_ADC_DRE, 4, 1, 0),
        SOC_SINGLE("ADC3 DRE Switch", ES8326_ADC_DRE, 5, 1, 0),
        SOC_SINGLE("ADC2 DRE Switch", ES8326_ADC_DRE, 6, 1, 0),
        SOC_SINGLE("ADC1 DRE Switch", ES8326_ADC_DRE, 7, 1, 0),
        SOC_ENUM("DRE Winsize", dre_winsize),
        SOC_SINGLE("DRE Gain Switch", ES8326_ADC_DRE_GAIN, 5, 1, 0),
        SOC_SINGLE_TLV("DRE Gain Volume", ES8326_ADC_DRE_GAIN,
                        0, 0x1F, 0, dre_gain_tlv),
        SOC_SINGLE_TLV("DRE Gate Volume", ES8326_ADC_DRE_GATE,
                        4, 0x07, 0, dre_gate_tlv),
        SOC_SINGLE("ALC Capture Switch", ES8326_ALC_RECOVERY, 3, 1, 0),
        SOC_SINGLE_TLV("ALC Capture Recovery Level", ES8326_ALC_LEVEL,
                        0, 4, 0, drc_recovery_tlv),
        SOC_ENUM("ALC Capture Winsize", alc_winsize),
        SOC_SINGLE_TLV("ALC Capture Target Level", ES8326_ALC_LEVEL,
                        0, 0x0f, 0, drc_target_tlv),

        SOC_SINGLE_EXT("CROSSTALK1", SND_SOC_NOPM, 0, 31, 0,
                        es8326_crosstalk1_get, es8326_crosstalk1_set),
        SOC_SINGLE_EXT("CROSSTALK2", SND_SOC_NOPM, 0, 31, 0,
                        es8326_crosstalk2_get, es8326_crosstalk2_set),
        SOC_SINGLE_EXT("HPL Volume", SND_SOC_NOPM, 0, 5, 0,
                        es8326_hplvol_get, es8326_hplvol_set),
        SOC_SINGLE_EXT("HPR Volume", SND_SOC_NOPM, 0, 5, 0,
                        es8326_hprvol_get, es8326_hprvol_set),

        SOC_SINGLE_TLV("HPL Playback Volume", ES8326_DACL_VOL, 0, 0xbf, 0, dac_vol_tlv),
        SOC_SINGLE_TLV("HPR Playback Volume", ES8326_DACR_VOL, 0, 0xbf, 0, dac_vol_tlv),
        SOC_SINGLE_TLV("SPKL Playback Volume", ES8326_SPKL_VOL, 0, 0xbf, 0, dac_vol_tlv),
        SOC_SINGLE_TLV("SPKR Playback Volume", ES8326_SPKR_VOL, 0, 0xbf, 0, dac_vol_tlv),

        SOC_ENUM("HPVol SPKVol Switch", hpvol_spkvol_switch),
};

static const struct snd_soc_dapm_widget es8326_dapm_widgets[] = {
        SND_SOC_DAPM_INPUT("MIC1"),
        SND_SOC_DAPM_INPUT("MIC2"),
        SND_SOC_DAPM_INPUT("MIC3"),
        SND_SOC_DAPM_INPUT("MIC4"),

        SND_SOC_DAPM_ADC("ADC L", NULL, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_ADC("ADC R", NULL, SND_SOC_NOPM, 0, 0),

        /* Digital Interface */
        SND_SOC_DAPM_AIF_OUT("I2S OUT", "I2S1 Capture", 0, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_AIF_IN("I2S IN", "I2S1 Playback", 0, SND_SOC_NOPM, 0, 0),

        /* Analog Power Supply*/
        SND_SOC_DAPM_DAC("Right DAC", NULL, ES8326_ANA_PDN, 0, 1),
        SND_SOC_DAPM_DAC("Left DAC", NULL, ES8326_ANA_PDN, 1, 1),
        SND_SOC_DAPM_SUPPLY("MICBIAS1", ES8326_ANA_MICBIAS, 2, 0, NULL, 0),
        SND_SOC_DAPM_SUPPLY("MICBIAS2", ES8326_ANA_MICBIAS, 3, 0, NULL, 0),

        SND_SOC_DAPM_PGA("LHPMIX", ES8326_DAC2HPMIX, 7, 0, NULL, 0),
        SND_SOC_DAPM_PGA("RHPMIX", ES8326_DAC2HPMIX, 3, 0, NULL, 0),

        SND_SOC_DAPM_OUTPUT("HPOL"),
        SND_SOC_DAPM_OUTPUT("HPOR"),
};

static const struct snd_soc_dapm_route es8326_dapm_routes[] = {
        {"ADC L", NULL, "MIC1"},
        {"ADC R", NULL, "MIC2"},
        {"ADC L", NULL, "MIC3"},
        {"ADC R", NULL, "MIC4"},

        {"I2S OUT", NULL, "ADC L"},
        {"I2S OUT", NULL, "ADC R"},

        {"Right DAC", NULL, "I2S IN"},
        {"Left DAC", NULL, "I2S IN"},

        {"LHPMIX", NULL, "Left DAC"},
        {"RHPMIX", NULL, "Right DAC"},

        {"HPOL", NULL, "LHPMIX"},
        {"HPOR", NULL, "RHPMIX"},
};

static bool es8326_volatile_register(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case ES8326_HPL_OFFSET_INI:
        case ES8326_HPR_OFFSET_INI:
        case ES8326_HPDET_STA:
        case ES8326_CTIA_OMTP_STA:
        case ES8326_CSM_MUTE_STA:
                return true;
        default:
                return false;
        }
}

static bool es8326_writeable_register(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case ES8326_BIAS_SW1:
        case ES8326_BIAS_SW2:
        case ES8326_BIAS_SW3:
        case ES8326_BIAS_SW4:
        case ES8326_ADC_HPFS1:
        case ES8326_ADC_HPFS2:
                return false;
        default:
                return true;
        }
}

static const struct regmap_config es8326_regmap_config = {
        .reg_bits = 8,
        .val_bits = 8,
        .max_register = 0xff,
        .use_single_read = true,
        .use_single_write = true,
        .volatile_reg = es8326_volatile_register,
        .writeable_reg = es8326_writeable_register,
        .cache_type = REGCACHE_RBTREE,
};

struct _coeff_div {
        u16 fs;
        u32 rate;
        u32 mclk;
        u8 reg4;
        u8 reg5;
        u8 reg6;
        u8 reg7;
        u8 reg8;
        u8 reg9;
        u8 rega;
        u8 regb;
};

/* codec hifi mclk clock divider coefficients */
/* {ratio, LRCK, MCLK, REG04, REG05, REG06, REG07, REG08, REG09, REG10, REG11} */
static const struct _coeff_div coeff_div_v0[] = {
        {64, 8000, 512000, 0x60, 0x01, 0x0F, 0x75, 0x0A, 0x1B, 0x1F, 0x7F},
        {64, 16000, 1024000, 0x20, 0x00, 0x33, 0x35, 0x0A, 0x1B, 0x1F, 0x3F},
        {64, 44100, 2822400, 0xE0, 0x00, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {64, 48000, 3072000, 0xE0, 0x00, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {128, 8000, 1024000, 0x60, 0x00, 0x33, 0x35, 0x0A, 0x1B, 0x1F, 0x7F},
        {128, 16000, 2048000, 0x20, 0x00, 0x03, 0x35, 0x0A, 0x1B, 0x1F, 0x3F},
        {128, 44100, 5644800, 0xE0, 0x01, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {128, 48000, 6144000, 0xE0, 0x01, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},

        {192, 32000, 6144000, 0xE0, 0x02, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {256, 8000, 2048000, 0x60, 0x00, 0x03, 0x35, 0x0A, 0x1B, 0x1F, 0x7F},
        {256, 16000, 4096000, 0x20, 0x01, 0x03, 0x35, 0x0A, 0x1B, 0x1F, 0x3F},
        {256, 44100, 11289600, 0xE0, 0x00, 0x30, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {256, 48000, 12288000, 0xE0, 0x00, 0x30, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {384, 32000, 12288000, 0xE0, 0x05, 0x03, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {400, 48000, 19200000, 0xE9, 0x04, 0x0F, 0x6d, 0x4A, 0x0A, 0x1F, 0x1F},

        {500, 48000, 24000000, 0xF8, 0x04, 0x3F, 0x6D, 0x4A, 0x0A, 0x1F, 0x1F},
        {512, 8000, 4096000, 0x60, 0x01, 0x03, 0x35, 0x0A, 0x1B, 0x1F, 0x7F},
        {512, 16000, 8192000, 0x20, 0x00, 0x30, 0x35, 0x0A, 0x1B, 0x1F, 0x3F},
        {512, 44100, 22579200, 0xE0, 0x00, 0x00, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {512, 48000, 24576000, 0xE0, 0x00, 0x00, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {768, 32000, 24576000, 0xE0, 0x02, 0x30, 0x2D, 0x4A, 0x0A, 0x1F, 0x1F},
        {1024, 8000, 8192000, 0x60, 0x00, 0x30, 0x35, 0x0A, 0x1B, 0x1F, 0x7F},
        {1024, 16000, 16384000, 0x20, 0x00, 0x00, 0x35, 0x0A, 0x1B, 0x1F, 0x3F},
};

static const struct _coeff_div coeff_div_v3[] = {
        {32, 8000, 256000, 0x60, 0x00, 0x0F, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {32, 16000, 512000, 0x20, 0x00, 0x0D, 0x75, 0x8A, 0x1B, 0x1F, 0x3F},
        {32, 44100, 1411200, 0x00, 0x00, 0x13, 0x2D, 0x8A, 0x0A, 0x1F, 0x1F},
        {32, 48000, 1536000, 0x00, 0x00, 0x13, 0x2D, 0x8A, 0x0A, 0x1F, 0x1F},
        {36, 8000, 288000, 0x20, 0x00, 0x0D, 0x75, 0x8A, 0x1B, 0x23, 0x47},
        {36, 16000, 576000, 0x20, 0x00, 0x0D, 0x75, 0x8A, 0x1B, 0x23, 0x47},
        {48, 8000, 384000, 0x60, 0x02, 0x1F, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {48, 16000, 768000, 0x20, 0x02, 0x0F, 0x75, 0x8A, 0x1B, 0x1F, 0x3F},
        {48, 48000, 2304000, 0x00, 0x02, 0x0D, 0x2D, 0x8A, 0x0A, 0x1F, 0x1F},

        {64, 8000, 512000, 0x60, 0x00, 0x35, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {64, 16000, 1024000, 0x20, 0x00, 0x05, 0x75, 0x8A, 0x1B, 0x1F, 0x3F},
        {64, 44100, 2822400, 0xE0, 0x00, 0x31, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {64, 48000, 3072000, 0xE0, 0x00, 0x31, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {72, 8000, 576000, 0x20, 0x00, 0x13, 0x35, 0x8A, 0x1B, 0x23, 0x47},
        {72, 16000, 1152000, 0x20, 0x00, 0x05, 0x75, 0x8A, 0x1B, 0x23, 0x47},
        {96, 8000, 768000, 0x60, 0x02, 0x1D, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {96, 16000, 1536000, 0x20, 0x02, 0x0D, 0x75, 0x8A, 0x1B, 0x1F, 0x3F},
        {100, 48000, 4800000, 0x04, 0x04, 0x3F, 0x6D, 0xB8, 0x08, 0x4f, 0x1f},
        {125, 48000, 6000000, 0x04, 0x04, 0x1F, 0x2D, 0x8A, 0x0A, 0x27, 0x27},

        {128, 8000, 1024000, 0x60, 0x00, 0x05, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {128, 16000, 2048000, 0x20, 0x00, 0x31, 0x35, 0x08, 0x19, 0x1F, 0x3F},
        {128, 44100, 5644800, 0xE0, 0x00, 0x01, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {128, 48000, 6144000, 0xE0, 0x00, 0x01, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {144, 8000, 1152000, 0x20, 0x00, 0x03, 0x35, 0x8A, 0x1B, 0x23, 0x47},
        {144, 16000, 2304000, 0x20, 0x00, 0x11, 0x35, 0x8A, 0x1B, 0x23, 0x47},
        {192, 8000, 1536000, 0x60, 0x02, 0x0D, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {192, 32000, 6144000, 0xE0, 0x02, 0x31, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {192, 16000, 3072000, 0x20, 0x02, 0x05, 0x75, 0xCA, 0x1B, 0x1F, 0x3F},

        {200, 48000, 9600000, 0x04, 0x04, 0x0F, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {250, 48000, 12000000, 0x04, 0x04, 0x0F, 0x2D, 0xCA, 0x0A, 0x27, 0x27},
        {256, 8000, 2048000, 0x60, 0x00, 0x31, 0x35, 0x08, 0x19, 0x1F, 0x7F},
        {256, 16000, 4096000, 0x20, 0x00, 0x01, 0x35, 0x08, 0x19, 0x1F, 0x3F},
        {256, 44100, 11289600, 0xE0, 0x01, 0x01, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {256, 48000, 12288000, 0xE0, 0x01, 0x01, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {288, 8000, 2304000, 0x20, 0x00, 0x01, 0x35, 0x8A, 0x1B, 0x23, 0x47},
        {384, 8000, 3072000, 0x60, 0x02, 0x05, 0x75, 0x8A, 0x1B, 0x1F, 0x7F},
        {384, 16000, 6144000, 0x20, 0x02, 0x03, 0x35, 0x8A, 0x1B, 0x1F, 0x3F},
        {384, 32000, 12288000, 0xE0, 0x02, 0x01, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {384, 48000, 18432000, 0x00, 0x02, 0x01, 0x2D, 0x8A, 0x0A, 0x1F, 0x1F},

        {400, 48000, 19200000, 0xE4, 0x04, 0x35, 0x6d, 0xCA, 0x0A, 0x1F, 0x1F},
        {500, 48000, 24000000, 0xF8, 0x04, 0x3F, 0x6D, 0xCA, 0x0A, 0x1F, 0x1F},
        {512, 8000, 4096000, 0x60, 0x00, 0x01, 0x08, 0x19, 0x1B, 0x1F, 0x7F},
        {512, 16000, 8192000, 0x20, 0x00, 0x30, 0x35, 0x08, 0x19, 0x1F, 0x3F},
        {512, 44100, 22579200, 0xE0, 0x00, 0x00, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {512, 48000, 24576000, 0xE0, 0x00, 0x00, 0x2D, 0x48, 0x08, 0x1F, 0x1F},
        {768, 8000, 6144000, 0x60, 0x02, 0x11, 0x35, 0x8A, 0x1B, 0x1F, 0x7F},
        {768, 16000, 12288000, 0x20, 0x02, 0x01, 0x35, 0x8A, 0x1B, 0x1F, 0x3F},
        {768, 32000, 24576000, 0xE0, 0x02, 0x30, 0x2D, 0xCA, 0x0A, 0x1F, 0x1F},
        {800, 48000, 38400000, 0x00, 0x18, 0x13, 0x2D, 0x8A, 0x0A, 0x1F, 0x1F},

        {1024, 8000, 8192000, 0x60, 0x00, 0x30, 0x35, 0x8A, 0x1B, 0x1F, 0x7F},
        {1024, 16000, 16384000, 0x20, 0x00, 0x00, 0x35, 0x8A, 0x1B, 0x1F, 0x3F},
        {1152, 16000, 18432000, 0x20, 0x08, 0x11, 0x35, 0x8A, 0x1B, 0x1F, 0x3F},
        {1536, 8000, 12288000, 0x60, 0x02, 0x01, 0x35, 0x8A, 0x1B, 0x1F, 0x7F},
        {1536, 16000, 24576000, 0x20, 0x02, 0x10, 0x35, 0x8A, 0x1B, 0x1F, 0x3F},
        {1625, 8000, 13000000, 0x0C, 0x18, 0x1F, 0x2D, 0x8A, 0x0A, 0x27, 0x27},
        {1625, 16000, 26000000, 0x0C, 0x18, 0x1F, 0x2D, 0x8A, 0x0A, 0x27, 0x27},
        {2048, 8000, 16384000, 0x60, 0x00, 0x00, 0x35, 0x8A, 0x1B, 0x1F, 0x7F},
        {2304, 8000, 18432000, 0x40, 0x02, 0x10, 0x35, 0x8A, 0x1B, 0x1F, 0x5F},
        {3072, 8000, 24576000, 0x60, 0x02, 0x10, 0x35, 0x8A, 0x1B, 0x1F, 0x7F},
        {3250, 8000, 26000000, 0x0C, 0x18, 0x0F, 0x2D, 0x8A, 0x0A, 0x27, 0x27},
};

static inline int get_coeff(int mclk, int rate, int array,
                                const struct _coeff_div *coeff_div)
{
        int i;

        for (i = 0; i < array; i++) {
                if (coeff_div[i].rate == rate && coeff_div[i].mclk == mclk)
                        return i;
        }

        return -EINVAL;
}

static int es8326_set_dai_sysclk(struct snd_soc_dai *codec_dai,
                                 int clk_id, unsigned int freq, int dir)
{
        struct snd_soc_component *codec = codec_dai->component;
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(codec);

        es8326->sysclk = freq;

        return 0;
}

static int es8326_set_dai_fmt(struct snd_soc_dai *codec_dai, unsigned int fmt)
{
        struct snd_soc_component *component = codec_dai->component;
        u8 iface = 0;

        switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        case SND_SOC_DAIFMT_CBC_CFP:
                snd_soc_component_update_bits(component, ES8326_RESET,
                                              ES8326_MASTER_MODE_EN, ES8326_MASTER_MODE_EN);
                break;
        case SND_SOC_DAIFMT_CBC_CFC:
                break;
        default:
                return -EINVAL;
        }

        /* interface format */
        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:
                break;
        case SND_SOC_DAIFMT_RIGHT_J:
                dev_err(component->dev, "Codec driver does not support right justified\n");
                return -EINVAL;
        case SND_SOC_DAIFMT_LEFT_J:
                iface |= ES8326_DAIFMT_LEFT_J;
                break;
        case SND_SOC_DAIFMT_DSP_A:
                iface |= ES8326_DAIFMT_DSP_A;
                break;
        case SND_SOC_DAIFMT_DSP_B:
                iface |= ES8326_DAIFMT_DSP_B;
                break;
        default:
                return -EINVAL;
        }

        snd_soc_component_update_bits(component, ES8326_FMT, ES8326_DAIFMT_MASK, iface);

        return 0;
}

static int es8326_pcm_hw_params(struct snd_pcm_substream *substream,
                                struct snd_pcm_hw_params *params,
                                struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        const struct _coeff_div *coeff_div;
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        u8 srate = 0;
        int coeff, array;

        if (es8326->version == 0) {
                coeff_div =  coeff_div_v0;
                array = ARRAY_SIZE(coeff_div_v0);
        } else {
                coeff_div =  coeff_div_v3;
                array = ARRAY_SIZE(coeff_div_v3);
        }
        coeff = get_coeff(es8326->sysclk, params_rate(params), array, coeff_div);
        /* bit size */
        switch (params_format(params)) {
        case SNDRV_PCM_FORMAT_S16_LE:
                srate |= ES8326_S16_LE;
                break;
        case SNDRV_PCM_FORMAT_S20_3LE:
                srate |= ES8326_S20_3_LE;
                break;
        case SNDRV_PCM_FORMAT_S18_3LE:
                srate |= ES8326_S18_LE;
                break;
        case SNDRV_PCM_FORMAT_S24_LE:
                srate |= ES8326_S24_LE;
                break;
        case SNDRV_PCM_FORMAT_S32_LE:
                srate |= ES8326_S32_LE;
                break;
        default:
                return -EINVAL;
        }

        /* set iface & srate */
        snd_soc_component_update_bits(component, ES8326_FMT, ES8326_DATA_LEN_MASK, srate);

        if (coeff >= 0) {
                regmap_write(es8326->regmap,  ES8326_CLK_DIV1,
                             coeff_div[coeff].reg4);
                regmap_write(es8326->regmap,  ES8326_CLK_DIV2,
                             coeff_div[coeff].reg5);
                regmap_write(es8326->regmap,  ES8326_CLK_DLL,
                             coeff_div[coeff].reg6);
                regmap_write(es8326->regmap,  ES8326_CLK_MUX,
                             coeff_div[coeff].reg7);
                regmap_write(es8326->regmap,  ES8326_CLK_ADC_SEL,
                             coeff_div[coeff].reg8);
                regmap_write(es8326->regmap,  ES8326_CLK_DAC_SEL,
                             coeff_div[coeff].reg9);
                regmap_write(es8326->regmap,  ES8326_CLK_ADC_OSR,
                             coeff_div[coeff].rega);
                regmap_write(es8326->regmap,  ES8326_CLK_DAC_OSR,
                             coeff_div[coeff].regb);
        } else {
                dev_warn(component->dev, "Clock coefficients do not match");
        }

        return 0;
}

static int es8326_mute(struct snd_soc_dai *dai, int mute, int direction)
{
        struct snd_soc_component *component = dai->component;
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int offset_l, offset_r;

        if (mute) {
                if (direction == SNDRV_PCM_STREAM_PLAYBACK) {
                        regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_OFF);
                        regmap_update_bits(es8326->regmap, ES8326_DAC_MUTE,
                                        ES8326_MUTE_MASK, ES8326_MUTE);
                        regmap_update_bits(es8326->regmap, ES8326_HP_DRIVER_REF,
                                        0x30, 0x00);
                } else {
                        regmap_update_bits(es8326->regmap,  ES8326_ADC_MUTE,
                                        0x0F, 0x0F);
                        if (es8326->version > ES8326_VERSION_B) {
                                regmap_update_bits(es8326->regmap, ES8326_VMIDSEL, 0x40, 0x40);
                                regmap_update_bits(es8326->regmap, ES8326_ANA_MICBIAS, 0x70, 0x30);
                        }
                }
        } else {
                if (!es8326->calibrated) {
                        regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_FORCE_CAL);
                        msleep(30);
                        regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_OFF);
                        regmap_read(es8326->regmap, ES8326_HPL_OFFSET_INI, &offset_l);
                        regmap_read(es8326->regmap, ES8326_HPR_OFFSET_INI, &offset_r);
                        regmap_write(es8326->regmap, ES8326_HP_OFFSET_CAL, 0x8c);
                        regmap_write(es8326->regmap, ES8326_HPL_OFFSET_INI, offset_l);
                        regmap_write(es8326->regmap, ES8326_HPR_OFFSET_INI, offset_r);
                        es8326->calibrated = true;
                }
                regmap_update_bits(es8326->regmap, ES8326_CLK_INV, 0xc0, 0x00);
                regmap_update_bits(es8326->regmap, ES8326_CLK_MUX, 0x80, 0x00);
                if (direction == SNDRV_PCM_STREAM_PLAYBACK) {
                        regmap_update_bits(es8326->regmap, ES8326_DAC_DSM, 0x01, 0x01);
                        usleep_range(1000, 5000);
                        regmap_update_bits(es8326->regmap, ES8326_DAC_DSM, 0x01, 0x00);
                        usleep_range(1000, 5000);
                        regmap_update_bits(es8326->regmap, ES8326_HP_DRIVER_REF, 0x30, 0x20);
                        regmap_update_bits(es8326->regmap, ES8326_HP_DRIVER_REF, 0x30, 0x30);
                        regmap_write(es8326->regmap, ES8326_HP_DRIVER, 0xa1);
                        regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_ON);
                        regmap_update_bits(es8326->regmap, ES8326_DAC_MUTE,
                                        ES8326_MUTE_MASK, ~(ES8326_MUTE));
                } else {
                        msleep(300);
                        if (es8326->version > ES8326_VERSION_B) {
                                regmap_update_bits(es8326->regmap, ES8326_ANA_MICBIAS, 0x70, 0x70);
                                regmap_update_bits(es8326->regmap, ES8326_VMIDSEL, 0x40, 0x00);
                        }
                        regmap_update_bits(es8326->regmap,  ES8326_ADC_MUTE,
                                        0x0F, 0x00);
                }
        }
        return 0;
}

static int es8326_set_bias_level(struct snd_soc_component *codec,
                                 enum snd_soc_bias_level level)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(codec);
        int ret;

        switch (level) {
        case SND_SOC_BIAS_ON:
                ret = clk_prepare_enable(es8326->mclk);
                if (ret)
                        return ret;

                regmap_update_bits(es8326->regmap, ES8326_RESET, 0x02, 0x02);
                usleep_range(5000, 10000);
                regmap_write(es8326->regmap, ES8326_INTOUT_IO, es8326->interrupt_clk);
                regmap_write(es8326->regmap, ES8326_SDINOUT1_IO,
                            (ES8326_IO_DMIC_CLK << ES8326_SDINOUT1_SHIFT));
                regmap_write(es8326->regmap, ES8326_PGA_PDN, 0x40);
                regmap_write(es8326->regmap, ES8326_ANA_PDN, 0x00);
                regmap_update_bits(es8326->regmap,  ES8326_CLK_CTL, 0x20, 0x20);
                regmap_update_bits(es8326->regmap, ES8326_RESET, 0x02, 0x00);
                if (es8326->version > ES8326_VERSION_B) {
                        regmap_update_bits(es8326->regmap, ES8326_VMIDSEL, 0x40, 0x40);
                        regmap_update_bits(es8326->regmap, ES8326_ANA_MICBIAS, 0x70, 0x30);
                }
                break;
        case SND_SOC_BIAS_PREPARE:
                break;
        case SND_SOC_BIAS_STANDBY:
                regmap_write(es8326->regmap, ES8326_ANA_PDN, 0x3b);
                regmap_update_bits(es8326->regmap, ES8326_CLK_CTL, 0x20, 0x00);
                regmap_write(es8326->regmap, ES8326_SDINOUT1_IO, ES8326_IO_INPUT);
                if (es8326->version > ES8326_VERSION_B) {
                        regmap_update_bits(es8326->regmap, ES8326_VMIDSEL, 0x40, 0x40);
                        regmap_update_bits(es8326->regmap, ES8326_ANA_MICBIAS, 0x70, 0x10);
                }
                regmap_update_bits(es8326->regmap, ES8326_CLK_INV, 0xc0, 0xc0);
                regmap_update_bits(es8326->regmap, ES8326_CLK_MUX, 0x80, 0x80);
                break;
        case SND_SOC_BIAS_OFF:
                clk_disable_unprepare(es8326->mclk);
                break;
        }

        return 0;
}

#define es8326_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
        SNDRV_PCM_FMTBIT_S24_LE)

static const struct snd_soc_dai_ops es8326_ops = {
        .hw_params = es8326_pcm_hw_params,
        .set_fmt = es8326_set_dai_fmt,
        .set_sysclk = es8326_set_dai_sysclk,
        .mute_stream = es8326_mute,
        .no_capture_mute = 0,
};

static struct snd_soc_dai_driver es8326_dai = {
        .name = "ES8326 HiFi",
        .playback = {
                .stream_name = "Playback",
                .channels_min = 1,
                .channels_max = 2,
                .rates = SNDRV_PCM_RATE_8000_48000,
                .formats = es8326_FORMATS,
                },
        .capture = {
                .stream_name = "Capture",
                .channels_min = 1,
                .channels_max = 2,
                .rates = SNDRV_PCM_RATE_8000_48000,
                .formats = es8326_FORMATS,
                },
        .ops = &es8326_ops,
        .symmetric_rate = 1,
};

static void es8326_enable_micbias(struct snd_soc_component *component)
{
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);

        snd_soc_dapm_mutex_lock(dapm);
        snd_soc_dapm_force_enable_pin_unlocked(dapm, "MICBIAS1");
        snd_soc_dapm_force_enable_pin_unlocked(dapm, "MICBIAS2");
        snd_soc_dapm_sync_unlocked(dapm);
        snd_soc_dapm_mutex_unlock(dapm);
}

static void es8326_disable_micbias(struct snd_soc_component *component)
{
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);

        snd_soc_dapm_mutex_lock(dapm);
        snd_soc_dapm_disable_pin_unlocked(dapm, "MICBIAS1");
        snd_soc_dapm_disable_pin_unlocked(dapm, "MICBIAS2");
        snd_soc_dapm_sync_unlocked(dapm);
        snd_soc_dapm_mutex_unlock(dapm);
}

/*
 *      For button detection, set the following in soundcard
 *      snd_jack_set_key(jack->jack, SND_JACK_BTN_0, KEY_PLAYPAUSE);
 *      snd_jack_set_key(jack->jack, SND_JACK_BTN_1, KEY_VOLUMEUP);
 *      snd_jack_set_key(jack->jack, SND_JACK_BTN_2, KEY_VOLUMEDOWN);
 */
static void es8326_jack_button_handler(struct work_struct *work)
{
        struct es8326_priv *es8326 =
                container_of(work, struct es8326_priv, button_press_work.work);
        struct snd_soc_component *comp = es8326->component;
        unsigned int iface;
        static int button_to_report, press_count;
        static int prev_button, cur_button;

        if (!(es8326->jack->status & SND_JACK_HEADSET)) /* Jack unplugged */
                return;

        mutex_lock(&es8326->lock);
        iface = snd_soc_component_read(comp, ES8326_HPDET_STA);
        switch (iface) {
        case 0x93:
                /* pause button detected */
                cur_button = SND_JACK_BTN_0;
                break;
        case 0x6f:
        case 0x4b:
                /* button volume up */
                if ((iface == 0x6f) && (es8326->version > ES8326_VERSION_B))
                        cur_button = SND_JACK_BTN_0;
                else
                        cur_button = SND_JACK_BTN_1;
                break;
        case 0x27:
                /* button volume down */
                cur_button = SND_JACK_BTN_2;
                break;
        case 0x1e:
        case 0xe2:
                /* button released or not pressed */
                cur_button = 0;
                break;
        default:
                break;
        }

        if ((prev_button == cur_button) && (cur_button != 0)) {
                press_count++;
                if (press_count > 3) {
                        /* report a press every 120ms */
                        snd_soc_jack_report(es8326->jack, cur_button,
                                        SND_JACK_BTN_0 | SND_JACK_BTN_1 | SND_JACK_BTN_2);
                        press_count = 0;
                }
                button_to_report = cur_button;
                queue_delayed_work(system_dfl_wq, &es8326->button_press_work,
                                   msecs_to_jiffies(35));
        } else if (prev_button != cur_button) {
                /* mismatch, detect again */
                prev_button = cur_button;
                queue_delayed_work(system_dfl_wq, &es8326->button_press_work,
                                   msecs_to_jiffies(35));
        } else {
                /* released or no pressed */
                if (button_to_report != 0) {
                        snd_soc_jack_report(es8326->jack, button_to_report,
                                    SND_JACK_BTN_0 | SND_JACK_BTN_1 | SND_JACK_BTN_2);
                        snd_soc_jack_report(es8326->jack, 0,
                                    SND_JACK_BTN_0 | SND_JACK_BTN_1 | SND_JACK_BTN_2);
                        button_to_report = 0;
                }
                es8326_disable_micbias(es8326->component);
        }
        mutex_unlock(&es8326->lock);
}

static void es8326_jack_detect_handler(struct work_struct *work)
{
        struct es8326_priv *es8326 =
                container_of(work, struct es8326_priv, jack_detect_work.work);
        struct snd_soc_component *comp = es8326->component;
        unsigned int iface;

        mutex_lock(&es8326->lock);
        iface = snd_soc_component_read(comp, ES8326_HPDET_STA);
        dev_dbg(comp->dev, "gpio flag %#04x", iface);

        if ((es8326->jack_remove_retry == 1) && (es8326->version < ES8326_VERSION_B)) {
                if (iface & ES8326_HPINSERT_FLAG)
                        es8326->jack_remove_retry = 2;
                else
                        es8326->jack_remove_retry = 0;

                dev_dbg(comp->dev, "remove event check, set HPJACK_POL normal, cnt = %d\n",
                                es8326->jack_remove_retry);
                /*
                 * Inverted HPJACK_POL bit to trigger one IRQ to double check HP Removal event
                 */
                regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE,
                                        ES8326_HP_DET_JACK_POL, (es8326->jd_inverted ?
                                        ~es8326->jack_pol : es8326->jack_pol));
                goto exit;
        }

        if ((iface & ES8326_HPINSERT_FLAG) == 0) {
                /* Jack unplugged or spurious IRQ */
                dev_dbg(comp->dev, "No headset detected\n");
                es8326_disable_micbias(es8326->component);
                if (es8326->jack->status & SND_JACK_HEADPHONE) {
                        dev_dbg(comp->dev, "Report hp remove event\n");
                        snd_soc_jack_report(es8326->jack, 0,
                                    SND_JACK_BTN_0 | SND_JACK_BTN_1 | SND_JACK_BTN_2);
                        snd_soc_jack_report(es8326->jack, 0, SND_JACK_HEADSET);
                        /* mute adc when mic path switch */
                        regmap_write(es8326->regmap, ES8326_ADC1_SRC, 0x44);
                        regmap_write(es8326->regmap, ES8326_ADC2_SRC, 0x66);
                }
                es8326->hp = 0;
                regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x03, 0x01);
                regmap_write(es8326->regmap, ES8326_SYS_BIAS, 0x0a);
                regmap_update_bits(es8326->regmap, ES8326_HP_DRIVER_REF, 0x0f, 0x03);
                regmap_write(es8326->regmap, ES8326_INT_SOURCE, ES8326_INT_SRC_PIN9);
                /*
                 * Inverted HPJACK_POL bit to trigger one IRQ to double check HP Removal event
                 */
                if ((es8326->jack_remove_retry == 0) && (es8326->version < ES8326_VERSION_B)) {
                        es8326->jack_remove_retry = 1;
                        dev_dbg(comp->dev, "remove event check, invert HPJACK_POL, cnt = %d\n",
                                        es8326->jack_remove_retry);
                        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE,
                                        ES8326_HP_DET_JACK_POL, (es8326->jd_inverted ?
                                        es8326->jack_pol : ~es8326->jack_pol));

                } else {
                        es8326->jack_remove_retry = 0;
                }
        } else if ((iface & ES8326_HPINSERT_FLAG) == ES8326_HPINSERT_FLAG) {
                es8326->jack_remove_retry = 0;
                if (es8326->hp == 0) {
                        dev_dbg(comp->dev, "First insert, start OMTP/CTIA type check\n");
                        /*
                         * set auto-check mode, then restart jack_detect_work after 400ms.
                         * Don't report jack status.
                         */
                        regmap_write(es8326->regmap, ES8326_INT_SOURCE, 0x00);
                        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x03, 0x01);
                        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x10, 0x00);
                        usleep_range(50000, 70000);
                        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x03, 0x00);
                        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x10, 0x10);
                        usleep_range(50000, 70000);
                        regmap_write(es8326->regmap, ES8326_INT_SOURCE,
                                        (ES8326_INT_SRC_PIN9 | ES8326_INT_SRC_BUTTON));
                        regmap_write(es8326->regmap, ES8326_SYS_BIAS, 0x1f);
                        regmap_update_bits(es8326->regmap, ES8326_HP_DRIVER_REF, 0x0f, 0x0d);
                        queue_delayed_work(system_dfl_wq, &es8326->jack_detect_work,
                                        msecs_to_jiffies(400));
                        es8326->hp = 1;
                        goto exit;
                }
                if (es8326->jack->status & SND_JACK_HEADSET) {
                        /* detect button */
                        dev_dbg(comp->dev, "button pressed\n");
                        regmap_write(es8326->regmap, ES8326_INT_SOURCE,
                                        (ES8326_INT_SRC_PIN9 | ES8326_INT_SRC_BUTTON));
                        es8326_enable_micbias(es8326->component);
                        queue_delayed_work(system_dfl_wq, &es8326->button_press_work, 10);
                        goto exit;
                }
                if ((iface & ES8326_HPBUTTON_FLAG) == 0x01) {
                        dev_dbg(comp->dev, "Headphone detected\n");
                        snd_soc_jack_report(es8326->jack,
                                        SND_JACK_HEADPHONE, SND_JACK_HEADSET);
                } else {
                        dev_dbg(comp->dev, "Headset detected\n");
                        snd_soc_jack_report(es8326->jack,
                                        SND_JACK_HEADSET, SND_JACK_HEADSET);
                        regmap_update_bits(es8326->regmap, ES8326_PGA_PDN,
                                        0x08, 0x08);
                        regmap_write(es8326->regmap, ES8326_ADC1_SRC, 0x00);
                        regmap_write(es8326->regmap, ES8326_ADC2_SRC, 0x00);
                        regmap_update_bits(es8326->regmap, ES8326_PGA_PDN,
                                        0x08, 0x00);
                        usleep_range(10000, 15000);
                }
        }
exit:
        mutex_unlock(&es8326->lock);
}

static irqreturn_t es8326_irq(int irq, void *dev_id)
{
        struct es8326_priv *es8326 = dev_id;

        if (!es8326->jack)
                goto out;

        if (es8326->jack->status & SND_JACK_HEADSET)
                queue_delayed_work(system_dfl_wq, &es8326->jack_detect_work,
                                   msecs_to_jiffies(10));
        else
                queue_delayed_work(system_dfl_wq, &es8326->jack_detect_work,
                                   msecs_to_jiffies(300));

out:
        return IRQ_HANDLED;
}

static int es8326_calibrate(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int reg;
        unsigned int offset_l, offset_r;

        regmap_read(es8326->regmap, ES8326_CHIP_VERSION, &reg);
        es8326->version = reg;

        if ((es8326->version >= ES8326_VERSION_B) && (es8326->calibrated == false)) {
                dev_dbg(component->dev, "ES8326_VERSION_B, calibrating\n");
                regmap_write(es8326->regmap, ES8326_CLK_INV, 0xc0);
                regmap_write(es8326->regmap, ES8326_CLK_DIV1, 0x03);
                regmap_write(es8326->regmap, ES8326_CLK_DLL, 0x30);
                regmap_write(es8326->regmap, ES8326_CLK_MUX, 0xed);
                regmap_write(es8326->regmap, ES8326_CLK_DAC_SEL, 0x08);
                regmap_write(es8326->regmap, ES8326_CLK_TRI, 0xc1);
                regmap_write(es8326->regmap, ES8326_DAC_MUTE, 0x03);
                regmap_write(es8326->regmap, ES8326_ANA_VSEL, 0x7f);
                regmap_write(es8326->regmap, ES8326_VMIDLOW, 0x23);
                regmap_write(es8326->regmap, ES8326_DAC2HPMIX, 0x88);
                usleep_range(15000, 20000);
                regmap_write(es8326->regmap, ES8326_HP_OFFSET_CAL, 0x8c);
                usleep_range(15000, 20000);
                regmap_write(es8326->regmap, ES8326_RESET, 0xc0);
                usleep_range(15000, 20000);

                regmap_write(es8326->regmap, ES8326_HP_OFFSET_CAL, ES8326_HP_OFF);
                regmap_read(es8326->regmap, ES8326_CSM_MUTE_STA, &reg);
                if ((reg & 0xf0) != 0x40)
                        msleep(50);

                regmap_write(es8326->regmap, ES8326_HP_CAL, 0xd4);
                msleep(200);
                regmap_write(es8326->regmap, ES8326_HP_CAL, 0x4d);
                msleep(200);
                regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_OFF);
                regmap_read(es8326->regmap, ES8326_HPL_OFFSET_INI, &offset_l);
                regmap_read(es8326->regmap, ES8326_HPR_OFFSET_INI, &offset_r);
                regmap_write(es8326->regmap, ES8326_HP_OFFSET_CAL, 0x8c);
                regmap_write(es8326->regmap, ES8326_HPL_OFFSET_INI, offset_l);
                regmap_write(es8326->regmap, ES8326_HPR_OFFSET_INI, offset_r);
                regmap_write(es8326->regmap, ES8326_CLK_INV, 0x00);

                es8326->calibrated = true;
        }

        return 0;
}

static void es8326_init(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        regmap_write(es8326->regmap, ES8326_RESET, 0x1f);
        regmap_write(es8326->regmap, ES8326_VMIDSEL, 0x3E);
        regmap_write(es8326->regmap, ES8326_ANA_LP, 0xf0);
        usleep_range(10000, 15000);
        regmap_write(es8326->regmap, ES8326_HPJACK_TIMER, 0xd9);
        regmap_write(es8326->regmap, ES8326_ANA_MICBIAS, 0xd8);
        /* set headphone default type and detect pin */
        regmap_write(es8326->regmap, ES8326_HPDET_TYPE, 0x83);
        regmap_write(es8326->regmap, ES8326_CLK_RESAMPLE, 0x05);

        /* set internal oscillator as clock source of headpone cp */
        regmap_write(es8326->regmap, ES8326_CLK_DIV_CPC, 0x89);
        regmap_write(es8326->regmap, ES8326_CLK_CTL, ES8326_CLK_ON);
        /* clock manager reset release */
        regmap_write(es8326->regmap, ES8326_RESET, 0x17);
        /* set headphone detection as half scan mode */
        regmap_write(es8326->regmap, ES8326_HP_MISC, 0x3d);
        regmap_write(es8326->regmap, ES8326_PULLUP_CTL, 0x00);

        /* enable headphone driver */
        regmap_write(es8326->regmap, ES8326_HP_VOL, 0xc4);
        regmap_write(es8326->regmap, ES8326_HP_DRIVER, 0xa7);
        usleep_range(2000, 5000);
        regmap_write(es8326->regmap, ES8326_HP_DRIVER_REF, 0x23);
        regmap_write(es8326->regmap, ES8326_HP_DRIVER_REF, 0x33);
        regmap_write(es8326->regmap, ES8326_HP_DRIVER, 0xa1);

        regmap_write(es8326->regmap, ES8326_CLK_INV, 0x00);
        regmap_write(es8326->regmap, ES8326_CLK_VMIDS1, 0xc4);
        regmap_write(es8326->regmap, ES8326_CLK_VMIDS2, 0x81);
        regmap_write(es8326->regmap, ES8326_CLK_CAL_TIME, 0x00);
        /* calibrate for B version */
        es8326_calibrate(component);
        regmap_write(es8326->regmap, ES8326_DAC_CROSSTALK, 0xaa);
        regmap_write(es8326->regmap, ES8326_DAC_RAMPRATE, 0x00);
        /* turn off headphone out */
        regmap_write(es8326->regmap, ES8326_HP_CAL, 0x00);
        /* set ADC and DAC in low power mode */
        regmap_write(es8326->regmap, ES8326_ANA_LP, 0xf0);

        regmap_write(es8326->regmap, ES8326_ANA_VSEL, 0x7F);
        /* select vdda as micbias source */
        regmap_write(es8326->regmap, ES8326_VMIDLOW, 0x03);
        /* set dac dsmclip = 1 */
        regmap_write(es8326->regmap, ES8326_DAC_DSM, 0x08);
        regmap_write(es8326->regmap, ES8326_DAC_VPPSCALE, 0x15);

        regmap_write(es8326->regmap, ES8326_HPDET_TYPE, 0x80 |
                        ((es8326->version >= ES8326_VERSION_B) ?
                        (ES8326_HP_DET_SRC_PIN9 | es8326->jack_pol) :
                        (ES8326_HP_DET_SRC_PIN9 | es8326->jack_pol | 0x04)));
        usleep_range(5000, 10000);
        es8326_enable_micbias(es8326->component);
        usleep_range(50000, 70000);
        regmap_update_bits(es8326->regmap, ES8326_HPDET_TYPE, 0x03, 0x00);
        regmap_write(es8326->regmap, ES8326_INTOUT_IO,
                     es8326->interrupt_clk);
        regmap_write(es8326->regmap, ES8326_SDINOUT1_IO, ES8326_IO_INPUT);
        regmap_write(es8326->regmap, ES8326_SDINOUT23_IO, ES8326_IO_INPUT);

        regmap_write(es8326->regmap, ES8326_ANA_PDN, 0x00);
        regmap_write(es8326->regmap, ES8326_RESET, ES8326_CSM_ON);
        regmap_update_bits(es8326->regmap, ES8326_PGAGAIN, ES8326_MIC_SEL_MASK,
                           ES8326_MIC1_SEL);

        regmap_update_bits(es8326->regmap, ES8326_DAC_MUTE, ES8326_MUTE_MASK,
                           ES8326_MUTE);

        regmap_write(es8326->regmap, ES8326_ADC_MUTE, 0x0f);
        regmap_write(es8326->regmap, ES8326_CLK_DIV_LRCK, 0xff);
        regmap_write(es8326->regmap, ES8326_ADC1_SRC, 0x44);
        regmap_write(es8326->regmap, ES8326_ADC2_SRC, 0x66);
        es8326_disable_micbias(es8326->component);
        if (es8326->version > ES8326_VERSION_B) {
                regmap_update_bits(es8326->regmap, ES8326_ANA_MICBIAS, 0x73, 0x10);
                regmap_update_bits(es8326->regmap, ES8326_VMIDSEL, 0x40, 0x40);
        }

        msleep(200);
        regmap_write(es8326->regmap, ES8326_INT_SOURCE, ES8326_INT_SRC_PIN9);
}

static int es8326_resume(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        unsigned int reg;

        regcache_cache_only(es8326->regmap, false);
        regcache_cache_bypass(es8326->regmap, true);
        regmap_read(es8326->regmap, ES8326_CLK_RESAMPLE, &reg);
        regcache_cache_bypass(es8326->regmap, false);
        /* reset internal clock state */
        if (reg == 0x05)
                regmap_write(es8326->regmap, ES8326_CLK_CTL, ES8326_CLK_ON);
        else
                es8326_init(component);

        regcache_sync(es8326->regmap);

        es8326_irq(es8326->irq, es8326);
        return 0;
}

static int es8326_suspend(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        cancel_delayed_work_sync(&es8326->jack_detect_work);
        es8326_disable_micbias(component);
        es8326->calibrated = false;
        regmap_write(es8326->regmap, ES8326_CLK_MUX, 0x2d);
        regmap_write(es8326->regmap, ES8326_DAC2HPMIX, 0x00);
        regmap_write(es8326->regmap, ES8326_ANA_PDN, 0x3b);
        regmap_write(es8326->regmap, ES8326_CLK_CTL, ES8326_CLK_OFF);
        regcache_cache_only(es8326->regmap, true);

        /* reset register value to default */
        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x01);
        usleep_range(1000, 3000);
        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x00);

        regcache_mark_dirty(es8326->regmap);
        return 0;
}

static int es8326_probe(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);
        int ret;

        es8326->component = component;
        es8326->jd_inverted = device_property_read_bool(component->dev,
                                                        "everest,jack-detect-inverted");

        ret = device_property_read_u8(component->dev, "everest,jack-pol", &es8326->jack_pol);
        if (ret != 0) {
                dev_dbg(component->dev, "jack-pol return %d", ret);
                es8326->jack_pol = ES8326_HP_TYPE_AUTO;
        }
        dev_dbg(component->dev, "jack-pol %x", es8326->jack_pol);

        ret = device_property_read_u8(component->dev, "everest,interrupt-src",
                                      &es8326->interrupt_src);
        if (ret != 0) {
                dev_dbg(component->dev, "interrupt-src return %d", ret);
                es8326->interrupt_src = ES8326_HP_DET_SRC_PIN9;
        }
        dev_dbg(component->dev, "interrupt-src %x", es8326->interrupt_src);

        ret = device_property_read_u8(component->dev, "everest,interrupt-clk",
                                      &es8326->interrupt_clk);
        if (ret != 0) {
                dev_dbg(component->dev, "interrupt-clk return %d", ret);
                es8326->interrupt_clk = 0x00;
        }
        dev_dbg(component->dev, "interrupt-clk %x", es8326->interrupt_clk);

        es8326_init(component);
        return 0;
}

static void es8326_enable_jack_detect(struct snd_soc_component *component,
                                struct snd_soc_jack *jack)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        mutex_lock(&es8326->lock);
        if (es8326->jd_inverted)
                snd_soc_component_update_bits(component, ES8326_HPDET_TYPE,
                                              ES8326_HP_DET_JACK_POL, ~es8326->jack_pol);
        es8326->jack = jack;

        mutex_unlock(&es8326->lock);
        es8326_irq(es8326->irq, es8326);
}

static void es8326_disable_jack_detect(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        dev_dbg(component->dev, "Enter into %s\n", __func__);
        if (!es8326->jack)
                return; /* Already disabled (or never enabled) */
        cancel_delayed_work_sync(&es8326->jack_detect_work);

        mutex_lock(&es8326->lock);
        if (es8326->jack->status & SND_JACK_MICROPHONE) {
                es8326_disable_micbias(component);
                snd_soc_jack_report(es8326->jack, 0, SND_JACK_HEADSET);
        }
        es8326->jack = NULL;
        mutex_unlock(&es8326->lock);
}

static int es8326_set_jack(struct snd_soc_component *component,
                        struct snd_soc_jack *jack, void *data)
{
        if (jack)
                es8326_enable_jack_detect(component, jack);
        else
                es8326_disable_jack_detect(component);

        return 0;
}

static void es8326_remove(struct snd_soc_component *component)
{
        struct es8326_priv *es8326 = snd_soc_component_get_drvdata(component);

        es8326_disable_jack_detect(component);
        es8326_set_bias_level(component, SND_SOC_BIAS_OFF);
        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x01);
        usleep_range(1000, 3000);
        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x00);
}

static const struct snd_soc_component_driver soc_component_dev_es8326 = {
        .probe          = es8326_probe,
        .remove         = es8326_remove,
        .resume         = es8326_resume,
        .suspend        = es8326_suspend,
        .set_bias_level = es8326_set_bias_level,
        .set_jack       = es8326_set_jack,
        .dapm_widgets   = es8326_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(es8326_dapm_widgets),
        .dapm_routes            = es8326_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(es8326_dapm_routes),
        .controls               = es8326_snd_controls,
        .num_controls           = ARRAY_SIZE(es8326_snd_controls),
        .use_pmdown_time        = 1,
        .endianness             = 1,
};

static int es8326_i2c_probe(struct i2c_client *i2c)
{
        struct es8326_priv *es8326;
        int ret;

        es8326 = devm_kzalloc(&i2c->dev, sizeof(struct es8326_priv), GFP_KERNEL);
        if (!es8326)
                return -ENOMEM;

        i2c_set_clientdata(i2c, es8326);
        es8326->i2c = i2c;
        mutex_init(&es8326->lock);
        es8326->regmap = devm_regmap_init_i2c(i2c, &es8326_regmap_config);
        if (IS_ERR(es8326->regmap)) {
                ret = PTR_ERR(es8326->regmap);
                dev_err(&i2c->dev, "Failed to init regmap: %d\n", ret);
                return ret;
        }

        es8326->irq = i2c->irq;
        es8326->jack_remove_retry = 0;
        es8326->hp = 0;
        es8326->hpl_vol = 0x03;
        es8326->hpr_vol = 0x03;
        INIT_DELAYED_WORK(&es8326->jack_detect_work,
                          es8326_jack_detect_handler);
        INIT_DELAYED_WORK(&es8326->button_press_work,
                          es8326_jack_button_handler);
        /* ES8316 is level-based while ES8326 is edge-based */
        ret = devm_request_threaded_irq(&i2c->dev, es8326->irq, NULL, es8326_irq,
                                        IRQF_TRIGGER_RISING | IRQF_ONESHOT,
                                        "es8326", es8326);
        if (ret) {
                dev_warn(&i2c->dev, "Failed to request IRQ: %d: %d\n",
                es8326->irq, ret);
                es8326->irq = -ENXIO;
        }

        es8326->mclk = devm_clk_get_optional(&i2c->dev, "mclk");
        if (IS_ERR(es8326->mclk)) {
                dev_err(&i2c->dev, "unable to get mclk\n");
                return PTR_ERR(es8326->mclk);
        }
        if (!es8326->mclk)
                dev_warn(&i2c->dev, "assuming static mclk\n");

        ret = clk_prepare_enable(es8326->mclk);
        if (ret) {
                dev_err(&i2c->dev, "unable to enable mclk\n");
                return ret;
        }
        return devm_snd_soc_register_component(&i2c->dev,
                                        &soc_component_dev_es8326,
                                        &es8326_dai, 1);
}


static void es8326_i2c_shutdown(struct i2c_client *i2c)
{
        struct snd_soc_component *component;
        struct es8326_priv *es8326;

        es8326 = i2c_get_clientdata(i2c);
        component = es8326->component;
        dev_dbg(component->dev, "Enter into %s\n", __func__);
        cancel_delayed_work_sync(&es8326->jack_detect_work);
        cancel_delayed_work_sync(&es8326->button_press_work);

        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x01);
        usleep_range(1000, 3000);
        regmap_write(es8326->regmap, ES8326_CSM_I2C_STA, 0x00);

}

static void es8326_i2c_remove(struct i2c_client *i2c)
{
        es8326_i2c_shutdown(i2c);
}

static const struct i2c_device_id es8326_i2c_id[] = {
        {"es8326" },
        {}
};
MODULE_DEVICE_TABLE(i2c, es8326_i2c_id);

#ifdef CONFIG_OF
static const struct of_device_id es8326_of_match[] = {
        { .compatible = "everest,es8326", },
        {}
};
MODULE_DEVICE_TABLE(of, es8326_of_match);
#endif

#ifdef CONFIG_ACPI
static const struct acpi_device_id es8326_acpi_match[] = {
        {"ESSX8326", 0},
        {},
};
MODULE_DEVICE_TABLE(acpi, es8326_acpi_match);
#endif

static struct i2c_driver es8326_i2c_driver = {
        .driver = {
                .name = "es8326",
                .acpi_match_table = ACPI_PTR(es8326_acpi_match),
                .of_match_table = of_match_ptr(es8326_of_match),
        },
        .probe = es8326_i2c_probe,
        .shutdown = es8326_i2c_shutdown,
        .remove = es8326_i2c_remove,
        .id_table = es8326_i2c_id,
};
module_i2c_driver(es8326_i2c_driver);

MODULE_DESCRIPTION("ASoC es8326 driver");
MODULE_AUTHOR("David Yang <yangxiaohua@everest-semi.com>");
MODULE_LICENSE("GPL");