u8 interrupt_clk;
        bool jd_inverted;
        unsigned int sysclk;
+
+       bool calibrated;
+       int version;
 };
 
 static const SNDRV_CTL_TLVD_DECLARE_DB_SCALE(dac_vol_tlv, -9550, 50, 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("Analog Power", ES8326_ANA_PDN, 7, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("IBias Power", ES8326_ANA_PDN, 6, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("ADC Vref", ES8326_ANA_PDN, 5, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("DAC Vref", ES8326_ANA_PDN, 4, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("Vref Power", ES8326_ANA_PDN, 3, 1, NULL, 0),
        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),
 
-       /* Headphone Charge Pump and Output */
-       SND_SOC_DAPM_SUPPLY("HPOR Cal", ES8326_HP_CAL, 7, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("HPOL Cal", ES8326_HP_CAL, 3, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("Headphone Charge Pump", ES8326_HP_DRIVER,
-                           3, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("Headphone Driver Bias", ES8326_HP_DRIVER,
-                           2, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("Headphone LDO", ES8326_HP_DRIVER,
-                           1, 1, NULL, 0),
-       SND_SOC_DAPM_SUPPLY("Headphone Reference", ES8326_HP_DRIVER,
-                           0, 1, NULL, 0),
-       SND_SOC_DAPM_REG(snd_soc_dapm_supply, "HPOR Supply", ES8326_HP_CAL,
-                        ES8326_HPOR_SHIFT, 7, 7, 0),
-       SND_SOC_DAPM_REG(snd_soc_dapm_supply, "HPOL Supply", ES8326_HP_CAL,
-                        0, 7, 7, 0),
-
        SND_SOC_DAPM_OUTPUT("HPOL"),
        SND_SOC_DAPM_OUTPUT("HPOR"),
 };
        {"I2S OUT", NULL, "ADC L"},
        {"I2S OUT", NULL, "ADC R"},
 
-       {"I2S OUT", NULL, "Analog Power"},
-       {"I2S OUT", NULL, "ADC Vref"},
-       {"I2S OUT", NULL, "Vref Power"},
-       {"I2S OUT", NULL, "IBias Power"},
-       {"I2S IN", NULL, "Analog Power"},
-       {"I2S IN", NULL, "DAC Vref"},
-       {"I2S IN", NULL, "Vref Power"},
-       {"I2S IN", NULL, "IBias Power"},
-
        {"Right DAC", NULL, "I2S IN"},
        {"Left DAC", NULL, "I2S IN"},
 
        {"LHPMIX", NULL, "Left DAC"},
        {"RHPMIX", NULL, "Right DAC"},
 
-       {"HPOR", NULL, "HPOR Cal"},
-       {"HPOL", NULL, "HPOL Cal"},
-       {"HPOR", NULL, "HPOR Supply"},
-       {"HPOL", NULL, "HPOL Supply"},
-       {"HPOL", NULL, "Headphone Charge Pump"},
-       {"HPOR", NULL, "Headphone Charge Pump"},
-       {"HPOL", NULL, "Headphone Driver Bias"},
-       {"HPOR", NULL, "Headphone Driver Bias"},
-       {"HPOL", NULL, "Headphone LDO"},
-       {"HPOR", NULL, "Headphone LDO"},
-       {"HPOL", NULL, "Headphone Reference"},
-       {"HPOR", NULL, "Headphone Reference"},
-
        {"HPOL", NULL, "LHPMIX"},
        {"HPOR", NULL, "RHPMIX"},
 };
        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);
+
+       if (mute) {
+               regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_OFF);
+               regmap_update_bits(es8326->regmap, ES8326_DAC_MUTE,
+                               ES8326_MUTE_MASK, ES8326_MUTE);
+               regmap_write(es8326->regmap, ES8326_HP_DRIVER, 0xf0);
+       } else {
+               if (!es8326->calibrated) {
+                       regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_FORCE_CAL);
+                       msleep(30);
+                       es8326->calibrated = true;
+               }
+               regmap_write(es8326->regmap, ES8326_HP_DRIVER, 0xa0);
+               regmap_write(es8326->regmap, ES8326_HP_VOL, 0x00);
+               regmap_write(es8326->regmap, ES8326_HP_CAL, ES8326_HP_ON);
+               regmap_update_bits(es8326->regmap, ES8326_DAC_MUTE,
+                               ES8326_MUTE_MASK, ~(ES8326_MUTE));
+       }
+       return 0;
+}
+
 static int es8326_set_bias_level(struct snd_soc_component *codec,
                                 enum snd_soc_bias_level level)
 {
        .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 = 1,
 };
 
 static struct snd_soc_dai_driver es8326_dai = {
 
        cancel_delayed_work_sync(&es8326->jack_detect_work);
        es8326_disable_micbias(component);
-
+       es8326->calibrated = false;
        regmap_write(es8326->regmap, ES8326_CLK_CTL, ES8326_CLK_OFF);
        regcache_cache_only(es8326->regmap, true);
        regcache_mark_dirty(es8326->regmap);