sound/soc/codecs/wm8994.c

   1 /*
   2  * wm8994.c  --  WM8994 ALSA SoC Audio driver
   3  *
   4  * Copyright 2009-12 Wolfson Microelectronics plc
   5  *
   6  * Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
   7  *
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as
  11  * published by the Free Software Foundation.
  12  */
  13
  14 #include <linux/module.h>
  15 #include <linux/moduleparam.h>
  16 #include <linux/init.h>
  17 #include <linux/delay.h>
  18 #include <linux/pm.h>
  19 #include <linux/i2c.h>
  20 #include <linux/platform_device.h>
  21 #include <linux/pm_runtime.h>
  22 #include <linux/regulator/consumer.h>
  23 #include <linux/slab.h>
  24 #include <sound/core.h>
  25 #include <sound/jack.h>
  26 #include <sound/pcm.h>
  27 #include <sound/pcm_params.h>
  28 #include <sound/soc.h>
  29 #include <sound/initval.h>
  30 #include <sound/tlv.h>
  31 #include <trace/events/asoc.h>
  32
  33 #include <linux/mfd/wm8994/core.h>
  34 #include <linux/mfd/wm8994/registers.h>
  35 #include <linux/mfd/wm8994/pdata.h>
  36 #include <linux/mfd/wm8994/gpio.h>
  37
  38 #include "wm8994.h"
  39 #include "wm_hubs.h"
  40
  41 #define WM1811_JACKDET_MODE_NONE  0x0000
  42 #define WM1811_JACKDET_MODE_JACK  0x0100
  43 #define WM1811_JACKDET_MODE_MIC   0x0080
  44 #define WM1811_JACKDET_MODE_AUDIO 0x0180
  45
  46 #define WM8994_NUM_DRC 3
  47 #define WM8994_NUM_EQ  3
  48
  49 static struct {
  50         unsigned int reg;
  51         unsigned int mask;
  52 } wm8994_vu_bits[] = {
  53         { WM8994_LEFT_LINE_INPUT_1_2_VOLUME, WM8994_IN1_VU },
  54         { WM8994_RIGHT_LINE_INPUT_1_2_VOLUME, WM8994_IN1_VU },
  55         { WM8994_LEFT_LINE_INPUT_3_4_VOLUME, WM8994_IN2_VU },
  56         { WM8994_RIGHT_LINE_INPUT_3_4_VOLUME, WM8994_IN2_VU },
  57         { WM8994_SPEAKER_VOLUME_LEFT, WM8994_SPKOUT_VU },
  58         { WM8994_SPEAKER_VOLUME_RIGHT, WM8994_SPKOUT_VU },
  59         { WM8994_LEFT_OUTPUT_VOLUME, WM8994_HPOUT1_VU },
  60         { WM8994_RIGHT_OUTPUT_VOLUME, WM8994_HPOUT1_VU },
  61         { WM8994_LEFT_OPGA_VOLUME, WM8994_MIXOUT_VU },
  62         { WM8994_RIGHT_OPGA_VOLUME, WM8994_MIXOUT_VU },
  63
  64         { WM8994_AIF1_DAC1_LEFT_VOLUME, WM8994_AIF1DAC1_VU },
  65         { WM8994_AIF1_DAC1_RIGHT_VOLUME, WM8994_AIF1DAC1_VU },
  66         { WM8994_AIF1_DAC2_LEFT_VOLUME, WM8994_AIF1DAC2_VU },
  67         { WM8994_AIF1_DAC2_RIGHT_VOLUME, WM8994_AIF1DAC2_VU },
  68         { WM8994_AIF2_DAC_LEFT_VOLUME, WM8994_AIF2DAC_VU },
  69         { WM8994_AIF2_DAC_RIGHT_VOLUME, WM8994_AIF2DAC_VU },
  70         { WM8994_AIF1_ADC1_LEFT_VOLUME, WM8994_AIF1ADC1_VU },
  71         { WM8994_AIF1_ADC1_RIGHT_VOLUME, WM8994_AIF1ADC1_VU },
  72         { WM8994_AIF1_ADC2_LEFT_VOLUME, WM8994_AIF1ADC2_VU },
  73         { WM8994_AIF1_ADC2_RIGHT_VOLUME, WM8994_AIF1ADC2_VU },
  74         { WM8994_AIF2_ADC_LEFT_VOLUME, WM8994_AIF2ADC_VU },
  75         { WM8994_AIF2_ADC_RIGHT_VOLUME, WM8994_AIF1ADC2_VU },
  76         { WM8994_DAC1_LEFT_VOLUME, WM8994_DAC1_VU },
  77         { WM8994_DAC1_RIGHT_VOLUME, WM8994_DAC1_VU },
  78         { WM8994_DAC2_LEFT_VOLUME, WM8994_DAC2_VU },
  79         { WM8994_DAC2_RIGHT_VOLUME, WM8994_DAC2_VU },
  80 };
  81
  82 static int wm8994_drc_base[] = {
  83         WM8994_AIF1_DRC1_1,
  84         WM8994_AIF1_DRC2_1,
  85         WM8994_AIF2_DRC_1,
  86 };
  87
  88 static int wm8994_retune_mobile_base[] = {
  89         WM8994_AIF1_DAC1_EQ_GAINS_1,
  90         WM8994_AIF1_DAC2_EQ_GAINS_1,
  91         WM8994_AIF2_EQ_GAINS_1,
  92 };
  93
  94 static const struct wm8958_micd_rate micdet_rates[] = {
  95         { 32768,       true,  1, 4 },
  96         { 32768,       false, 1, 1 },
  97         { 44100 * 256, true,  7, 10 },
  98         { 44100 * 256, false, 7, 10 },
  99 };
 100
 101 static const struct wm8958_micd_rate jackdet_rates[] = {
 102         { 32768,       true,  0, 1 },
 103         { 32768,       false, 0, 1 },
 104         { 44100 * 256, true,  10, 10 },
 105         { 44100 * 256, false, 7, 8 },
 106 };
 107
 108 static void wm8958_micd_set_rate(struct snd_soc_codec *codec)
 109 {
 110         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 111         struct wm8994 *control = wm8994->wm8994;
 112         int best, i, sysclk, val;
 113         bool idle;
 114         const struct wm8958_micd_rate *rates;
 115         int num_rates;
 116
 117         idle = !wm8994->jack_mic;
 118
 119         sysclk = snd_soc_read(codec, WM8994_CLOCKING_1);
 120         if (sysclk & WM8994_SYSCLK_SRC)
 121                 sysclk = wm8994->aifclk[1];
 122         else
 123                 sysclk = wm8994->aifclk[0];
 124
 125         if (control->pdata.micd_rates) {
 126                 rates = control->pdata.micd_rates;
 127                 num_rates = control->pdata.num_micd_rates;
 128         } else if (wm8994->jackdet) {
 129                 rates = jackdet_rates;
 130                 num_rates = ARRAY_SIZE(jackdet_rates);
 131         } else {
 132                 rates = micdet_rates;
 133                 num_rates = ARRAY_SIZE(micdet_rates);
 134         }
 135
 136         best = 0;
 137         for (i = 0; i < num_rates; i++) {
 138                 if (rates[i].idle != idle)
 139                         continue;
 140                 if (abs(rates[i].sysclk - sysclk) <
 141                     abs(rates[best].sysclk - sysclk))
 142                         best = i;
 143                 else if (rates[best].idle != idle)
 144                         best = i;
 145         }
 146
 147         val = rates[best].start << WM8958_MICD_BIAS_STARTTIME_SHIFT
 148                 | rates[best].rate << WM8958_MICD_RATE_SHIFT;
 149
 150         dev_dbg(codec->dev, "MICD rate %d,%d for %dHz %s\n",
 151                 rates[best].start, rates[best].rate, sysclk,
 152                 idle ? "idle" : "active");
 153
 154         snd_soc_update_bits(codec, WM8958_MIC_DETECT_1,
 155                             WM8958_MICD_BIAS_STARTTIME_MASK |
 156                             WM8958_MICD_RATE_MASK, val);
 157 }
 158
 159 static int configure_aif_clock(struct snd_soc_codec *codec, int aif)
 160 {
 161         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 162         int rate;
 163         int reg1 = 0;
 164         int offset;
 165
 166         if (aif)
 167                 offset = 4;
 168         else
 169                 offset = 0;
 170
 171         switch (wm8994->sysclk[aif]) {
 172         case WM8994_SYSCLK_MCLK1:
 173                 rate = wm8994->mclk[0];
 174                 break;
 175
 176         case WM8994_SYSCLK_MCLK2:
 177                 reg1 |= 0x8;
 178                 rate = wm8994->mclk[1];
 179                 break;
 180
 181         case WM8994_SYSCLK_FLL1:
 182                 reg1 |= 0x10;
 183                 rate = wm8994->fll[0].out;
 184                 break;
 185
 186         case WM8994_SYSCLK_FLL2:
 187                 reg1 |= 0x18;
 188                 rate = wm8994->fll[1].out;
 189                 break;
 190
 191         default:
 192                 return -EINVAL;
 193         }
 194
 195         if (rate >= 13500000) {
 196                 rate /= 2;
 197                 reg1 |= WM8994_AIF1CLK_DIV;
 198
 199                 dev_dbg(codec->dev, "Dividing AIF%d clock to %dHz\n",
 200                         aif + 1, rate);
 201         }
 202
 203         wm8994->aifclk[aif] = rate;
 204
 205         snd_soc_update_bits(codec, WM8994_AIF1_CLOCKING_1 + offset,
 206                             WM8994_AIF1CLK_SRC_MASK | WM8994_AIF1CLK_DIV,
 207                             reg1);
 208
 209         return 0;
 210 }
 211
 212 static int configure_clock(struct snd_soc_codec *codec)
 213 {
 214         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 215         int change, new;
 216
 217         /* Bring up the AIF clocks first */
 218         configure_aif_clock(codec, 0);
 219         configure_aif_clock(codec, 1);
 220
 221         /* Then switch CLK_SYS over to the higher of them; a change
 222          * can only happen as a result of a clocking change which can
 223          * only be made outside of DAPM so we can safely redo the
 224          * clocking.
 225          */
 226
 227         /* If they're equal it doesn't matter which is used */
 228         if (wm8994->aifclk[0] == wm8994->aifclk[1]) {
 229                 wm8958_micd_set_rate(codec);
 230                 return 0;
 231         }
 232
 233         if (wm8994->aifclk[0] < wm8994->aifclk[1])
 234                 new = WM8994_SYSCLK_SRC;
 235         else
 236                 new = 0;
 237
 238         change = snd_soc_update_bits(codec, WM8994_CLOCKING_1,
 239                                      WM8994_SYSCLK_SRC, new);
 240         if (change)
 241                 snd_soc_dapm_sync(&codec->dapm);
 242
 243         wm8958_micd_set_rate(codec);
 244
 245         return 0;
 246 }
 247
 248 static int check_clk_sys(struct snd_soc_dapm_widget *source,
 249                          struct snd_soc_dapm_widget *sink)
 250 {
 251         int reg = snd_soc_read(source->codec, WM8994_CLOCKING_1);
 252         const char *clk;
 253
 254         /* Check what we're currently using for CLK_SYS */
 255         if (reg & WM8994_SYSCLK_SRC)
 256                 clk = "AIF2CLK";
 257         else
 258                 clk = "AIF1CLK";
 259
 260         return strcmp(source->name, clk) == 0;
 261 }
 262
 263 static const char *sidetone_hpf_text[] = {
 264         "2.7kHz", "1.35kHz", "675Hz", "370Hz", "180Hz", "90Hz", "45Hz"
 265 };
 266
 267 static const struct soc_enum sidetone_hpf =
 268         SOC_ENUM_SINGLE(WM8994_SIDETONE, 7, 7, sidetone_hpf_text);
 269
 270 static const char *adc_hpf_text[] = {
 271         "HiFi", "Voice 1", "Voice 2", "Voice 3"
 272 };
 273
 274 static const struct soc_enum aif1adc1_hpf =
 275         SOC_ENUM_SINGLE(WM8994_AIF1_ADC1_FILTERS, 13, 4, adc_hpf_text);
 276
 277 static const struct soc_enum aif1adc2_hpf =
 278         SOC_ENUM_SINGLE(WM8994_AIF1_ADC2_FILTERS, 13, 4, adc_hpf_text);
 279
 280 static const struct soc_enum aif2adc_hpf =
 281         SOC_ENUM_SINGLE(WM8994_AIF2_ADC_FILTERS, 13, 4, adc_hpf_text);
 282
 283 static const DECLARE_TLV_DB_SCALE(aif_tlv, 0, 600, 0);
 284 static const DECLARE_TLV_DB_SCALE(digital_tlv, -7200, 75, 1);
 285 static const DECLARE_TLV_DB_SCALE(st_tlv, -3600, 300, 0);
 286 static const DECLARE_TLV_DB_SCALE(wm8994_3d_tlv, -1600, 183, 0);
 287 static const DECLARE_TLV_DB_SCALE(eq_tlv, -1200, 100, 0);
 288 static const DECLARE_TLV_DB_SCALE(ng_tlv, -10200, 600, 0);
 289 static const DECLARE_TLV_DB_SCALE(mixin_boost_tlv, 0, 900, 0);
 290
 291 #define WM8994_DRC_SWITCH(xname, reg, shift) \
 292 {       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, \
 293         .info = snd_soc_info_volsw, .get = snd_soc_get_volsw,\
 294         .put = wm8994_put_drc_sw, \
 295         .private_value =  SOC_SINGLE_VALUE(reg, shift, 1, 0) }
 296
 297 static int wm8994_put_drc_sw(struct snd_kcontrol *kcontrol,
 298                              struct snd_ctl_elem_value *ucontrol)
 299 {
 300         struct soc_mixer_control *mc =
 301                 (struct soc_mixer_control *)kcontrol->private_value;
 302         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 303         int mask, ret;
 304
 305         /* Can't enable both ADC and DAC paths simultaneously */
 306         if (mc->shift == WM8994_AIF1DAC1_DRC_ENA_SHIFT)
 307                 mask = WM8994_AIF1ADC1L_DRC_ENA_MASK |
 308                         WM8994_AIF1ADC1R_DRC_ENA_MASK;
 309         else
 310                 mask = WM8994_AIF1DAC1_DRC_ENA_MASK;
 311
 312         ret = snd_soc_read(codec, mc->reg);
 313         if (ret < 0)
 314                 return ret;
 315         if (ret & mask)
 316                 return -EINVAL;
 317
 318         return snd_soc_put_volsw(kcontrol, ucontrol);
 319 }
 320
 321 static void wm8994_set_drc(struct snd_soc_codec *codec, int drc)
 322 {
 323         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 324         struct wm8994 *control = wm8994->wm8994;
 325         struct wm8994_pdata *pdata = &control->pdata;
 326         int base = wm8994_drc_base[drc];
 327         int cfg = wm8994->drc_cfg[drc];
 328         int save, i;
 329
 330         /* Save any enables; the configuration should clear them. */
 331         save = snd_soc_read(codec, base);
 332         save &= WM8994_AIF1DAC1_DRC_ENA | WM8994_AIF1ADC1L_DRC_ENA |
 333                 WM8994_AIF1ADC1R_DRC_ENA;
 334
 335         for (i = 0; i < WM8994_DRC_REGS; i++)
 336                 snd_soc_update_bits(codec, base + i, 0xffff,
 337                                     pdata->drc_cfgs[cfg].regs[i]);
 338
 339         snd_soc_update_bits(codec, base, WM8994_AIF1DAC1_DRC_ENA |
 340                              WM8994_AIF1ADC1L_DRC_ENA |
 341                              WM8994_AIF1ADC1R_DRC_ENA, save);
 342 }
 343
 344 /* Icky as hell but saves code duplication */
 345 static int wm8994_get_drc(const char *name)
 346 {
 347         if (strcmp(name, "AIF1DRC1 Mode") == 0)
 348                 return 0;
 349         if (strcmp(name, "AIF1DRC2 Mode") == 0)
 350                 return 1;
 351         if (strcmp(name, "AIF2DRC Mode") == 0)
 352                 return 2;
 353         return -EINVAL;
 354 }
 355
 356 static int wm8994_put_drc_enum(struct snd_kcontrol *kcontrol,
 357                                struct snd_ctl_elem_value *ucontrol)
 358 {
 359         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 360         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 361         struct wm8994 *control = wm8994->wm8994;
 362         struct wm8994_pdata *pdata = &control->pdata;
 363         int drc = wm8994_get_drc(kcontrol->id.name);
 364         int value = ucontrol->value.integer.value[0];
 365
 366         if (drc < 0)
 367                 return drc;
 368
 369         if (value >= pdata->num_drc_cfgs)
 370                 return -EINVAL;
 371
 372         wm8994->drc_cfg[drc] = value;
 373
 374         wm8994_set_drc(codec, drc);
 375
 376         return 0;
 377 }
 378
 379 static int wm8994_get_drc_enum(struct snd_kcontrol *kcontrol,
 380                                struct snd_ctl_elem_value *ucontrol)
 381 {
 382         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 383         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 384         int drc = wm8994_get_drc(kcontrol->id.name);
 385
 386         ucontrol->value.enumerated.item[0] = wm8994->drc_cfg[drc];
 387
 388         return 0;
 389 }
 390
 391 static void wm8994_set_retune_mobile(struct snd_soc_codec *codec, int block)
 392 {
 393         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 394         struct wm8994 *control = wm8994->wm8994;
 395         struct wm8994_pdata *pdata = &control->pdata;
 396         int base = wm8994_retune_mobile_base[block];
 397         int iface, best, best_val, save, i, cfg;
 398
 399         if (!pdata || !wm8994->num_retune_mobile_texts)
 400                 return;
 401
 402         switch (block) {
 403         case 0:
 404         case 1:
 405                 iface = 0;
 406                 break;
 407         case 2:
 408                 iface = 1;
 409                 break;
 410         default:
 411                 return;
 412         }
 413
 414         /* Find the version of the currently selected configuration
 415          * with the nearest sample rate. */
 416         cfg = wm8994->retune_mobile_cfg[block];
 417         best = 0;
 418         best_val = INT_MAX;
 419         for (i = 0; i < pdata->num_retune_mobile_cfgs; i++) {
 420                 if (strcmp(pdata->retune_mobile_cfgs[i].name,
 421                            wm8994->retune_mobile_texts[cfg]) == 0 &&
 422                     abs(pdata->retune_mobile_cfgs[i].rate
 423                         - wm8994->dac_rates[iface]) < best_val) {
 424                         best = i;
 425                         best_val = abs(pdata->retune_mobile_cfgs[i].rate
 426                                        - wm8994->dac_rates[iface]);
 427                 }
 428         }
 429
 430         dev_dbg(codec->dev, "ReTune Mobile %d %s/%dHz for %dHz sample rate\n",
 431                 block,
 432                 pdata->retune_mobile_cfgs[best].name,
 433                 pdata->retune_mobile_cfgs[best].rate,
 434                 wm8994->dac_rates[iface]);
 435
 436         /* The EQ will be disabled while reconfiguring it, remember the
 437          * current configuration.
 438          */
 439         save = snd_soc_read(codec, base);
 440         save &= WM8994_AIF1DAC1_EQ_ENA;
 441
 442         for (i = 0; i < WM8994_EQ_REGS; i++)
 443                 snd_soc_update_bits(codec, base + i, 0xffff,
 444                                 pdata->retune_mobile_cfgs[best].regs[i]);
 445
 446         snd_soc_update_bits(codec, base, WM8994_AIF1DAC1_EQ_ENA, save);
 447 }
 448
 449 /* Icky as hell but saves code duplication */
 450 static int wm8994_get_retune_mobile_block(const char *name)
 451 {
 452         if (strcmp(name, "AIF1.1 EQ Mode") == 0)
 453                 return 0;
 454         if (strcmp(name, "AIF1.2 EQ Mode") == 0)
 455                 return 1;
 456         if (strcmp(name, "AIF2 EQ Mode") == 0)
 457                 return 2;
 458         return -EINVAL;
 459 }
 460
 461 static int wm8994_put_retune_mobile_enum(struct snd_kcontrol *kcontrol,
 462                                          struct snd_ctl_elem_value *ucontrol)
 463 {
 464         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 465         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 466         struct wm8994 *control = wm8994->wm8994;
 467         struct wm8994_pdata *pdata = &control->pdata;
 468         int block = wm8994_get_retune_mobile_block(kcontrol->id.name);
 469         int value = ucontrol->value.integer.value[0];
 470
 471         if (block < 0)
 472                 return block;
 473
 474         if (value >= pdata->num_retune_mobile_cfgs)
 475                 return -EINVAL;
 476
 477         wm8994->retune_mobile_cfg[block] = value;
 478
 479         wm8994_set_retune_mobile(codec, block);
 480
 481         return 0;
 482 }
 483
 484 static int wm8994_get_retune_mobile_enum(struct snd_kcontrol *kcontrol,
 485                                          struct snd_ctl_elem_value *ucontrol)
 486 {
 487         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 488         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 489         int block = wm8994_get_retune_mobile_block(kcontrol->id.name);
 490
 491         ucontrol->value.enumerated.item[0] = wm8994->retune_mobile_cfg[block];
 492
 493         return 0;
 494 }
 495
 496 static const char *aif_chan_src_text[] = {
 497         "Left", "Right"
 498 };
 499
 500 static const struct soc_enum aif1adcl_src =
 501         SOC_ENUM_SINGLE(WM8994_AIF1_CONTROL_1, 15, 2, aif_chan_src_text);
 502
 503 static const struct soc_enum aif1adcr_src =
 504         SOC_ENUM_SINGLE(WM8994_AIF1_CONTROL_1, 14, 2, aif_chan_src_text);
 505
 506 static const struct soc_enum aif2adcl_src =
 507         SOC_ENUM_SINGLE(WM8994_AIF2_CONTROL_1, 15, 2, aif_chan_src_text);
 508
 509 static const struct soc_enum aif2adcr_src =
 510         SOC_ENUM_SINGLE(WM8994_AIF2_CONTROL_1, 14, 2, aif_chan_src_text);
 511
 512 static const struct soc_enum aif1dacl_src =
 513         SOC_ENUM_SINGLE(WM8994_AIF1_CONTROL_2, 15, 2, aif_chan_src_text);
 514
 515 static const struct soc_enum aif1dacr_src =
 516         SOC_ENUM_SINGLE(WM8994_AIF1_CONTROL_2, 14, 2, aif_chan_src_text);
 517
 518 static const struct soc_enum aif2dacl_src =
 519         SOC_ENUM_SINGLE(WM8994_AIF2_CONTROL_2, 15, 2, aif_chan_src_text);
 520
 521 static const struct soc_enum aif2dacr_src =
 522         SOC_ENUM_SINGLE(WM8994_AIF2_CONTROL_2, 14, 2, aif_chan_src_text);
 523
 524 static const char *osr_text[] = {
 525         "Low Power", "High Performance",
 526 };
 527
 528 static const struct soc_enum dac_osr =
 529         SOC_ENUM_SINGLE(WM8994_OVERSAMPLING, 0, 2, osr_text);
 530
 531 static const struct soc_enum adc_osr =
 532         SOC_ENUM_SINGLE(WM8994_OVERSAMPLING, 1, 2, osr_text);
 533
 534 static const struct snd_kcontrol_new wm8994_snd_controls[] = {
 535 SOC_DOUBLE_R_TLV("AIF1ADC1 Volume", WM8994_AIF1_ADC1_LEFT_VOLUME,
 536                  WM8994_AIF1_ADC1_RIGHT_VOLUME,
 537                  1, 119, 0, digital_tlv),
 538 SOC_DOUBLE_R_TLV("AIF1ADC2 Volume", WM8994_AIF1_ADC2_LEFT_VOLUME,
 539                  WM8994_AIF1_ADC2_RIGHT_VOLUME,
 540                  1, 119, 0, digital_tlv),
 541 SOC_DOUBLE_R_TLV("AIF2ADC Volume", WM8994_AIF2_ADC_LEFT_VOLUME,
 542                  WM8994_AIF2_ADC_RIGHT_VOLUME,
 543                  1, 119, 0, digital_tlv),
 544
 545 SOC_ENUM("AIF1ADCL Source", aif1adcl_src),
 546 SOC_ENUM("AIF1ADCR Source", aif1adcr_src),
 547 SOC_ENUM("AIF2ADCL Source", aif2adcl_src),
 548 SOC_ENUM("AIF2ADCR Source", aif2adcr_src),
 549
 550 SOC_ENUM("AIF1DACL Source", aif1dacl_src),
 551 SOC_ENUM("AIF1DACR Source", aif1dacr_src),
 552 SOC_ENUM("AIF2DACL Source", aif2dacl_src),
 553 SOC_ENUM("AIF2DACR Source", aif2dacr_src),
 554
 555 SOC_DOUBLE_R_TLV("AIF1DAC1 Volume", WM8994_AIF1_DAC1_LEFT_VOLUME,
 556                  WM8994_AIF1_DAC1_RIGHT_VOLUME, 1, 96, 0, digital_tlv),
 557 SOC_DOUBLE_R_TLV("AIF1DAC2 Volume", WM8994_AIF1_DAC2_LEFT_VOLUME,
 558                  WM8994_AIF1_DAC2_RIGHT_VOLUME, 1, 96, 0, digital_tlv),
 559 SOC_DOUBLE_R_TLV("AIF2DAC Volume", WM8994_AIF2_DAC_LEFT_VOLUME,
 560                  WM8994_AIF2_DAC_RIGHT_VOLUME, 1, 96, 0, digital_tlv),
 561
 562 SOC_SINGLE_TLV("AIF1 Boost Volume", WM8994_AIF1_CONTROL_2, 10, 3, 0, aif_tlv),
 563 SOC_SINGLE_TLV("AIF2 Boost Volume", WM8994_AIF2_CONTROL_2, 10, 3, 0, aif_tlv),
 564
 565 SOC_SINGLE("AIF1DAC1 EQ Switch", WM8994_AIF1_DAC1_EQ_GAINS_1, 0, 1, 0),
 566 SOC_SINGLE("AIF1DAC2 EQ Switch", WM8994_AIF1_DAC2_EQ_GAINS_1, 0, 1, 0),
 567 SOC_SINGLE("AIF2 EQ Switch", WM8994_AIF2_EQ_GAINS_1, 0, 1, 0),
 568
 569 WM8994_DRC_SWITCH("AIF1DAC1 DRC Switch", WM8994_AIF1_DRC1_1, 2),
 570 WM8994_DRC_SWITCH("AIF1ADC1L DRC Switch", WM8994_AIF1_DRC1_1, 1),
 571 WM8994_DRC_SWITCH("AIF1ADC1R DRC Switch", WM8994_AIF1_DRC1_1, 0),
 572
 573 WM8994_DRC_SWITCH("AIF1DAC2 DRC Switch", WM8994_AIF1_DRC2_1, 2),
 574 WM8994_DRC_SWITCH("AIF1ADC2L DRC Switch", WM8994_AIF1_DRC2_1, 1),
 575 WM8994_DRC_SWITCH("AIF1ADC2R DRC Switch", WM8994_AIF1_DRC2_1, 0),
 576
 577 WM8994_DRC_SWITCH("AIF2DAC DRC Switch", WM8994_AIF2_DRC_1, 2),
 578 WM8994_DRC_SWITCH("AIF2ADCL DRC Switch", WM8994_AIF2_DRC_1, 1),
 579 WM8994_DRC_SWITCH("AIF2ADCR DRC Switch", WM8994_AIF2_DRC_1, 0),
 580
 581 SOC_SINGLE_TLV("DAC1 Right Sidetone Volume", WM8994_DAC1_MIXER_VOLUMES,
 582                5, 12, 0, st_tlv),
 583 SOC_SINGLE_TLV("DAC1 Left Sidetone Volume", WM8994_DAC1_MIXER_VOLUMES,
 584                0, 12, 0, st_tlv),
 585 SOC_SINGLE_TLV("DAC2 Right Sidetone Volume", WM8994_DAC2_MIXER_VOLUMES,
 586                5, 12, 0, st_tlv),
 587 SOC_SINGLE_TLV("DAC2 Left Sidetone Volume", WM8994_DAC2_MIXER_VOLUMES,
 588                0, 12, 0, st_tlv),
 589 SOC_ENUM("Sidetone HPF Mux", sidetone_hpf),
 590 SOC_SINGLE("Sidetone HPF Switch", WM8994_SIDETONE, 6, 1, 0),
 591
 592 SOC_ENUM("AIF1ADC1 HPF Mode", aif1adc1_hpf),
 593 SOC_DOUBLE("AIF1ADC1 HPF Switch", WM8994_AIF1_ADC1_FILTERS, 12, 11, 1, 0),
 594
 595 SOC_ENUM("AIF1ADC2 HPF Mode", aif1adc2_hpf),
 596 SOC_DOUBLE("AIF1ADC2 HPF Switch", WM8994_AIF1_ADC2_FILTERS, 12, 11, 1, 0),
 597
 598 SOC_ENUM("AIF2ADC HPF Mode", aif2adc_hpf),
 599 SOC_DOUBLE("AIF2ADC HPF Switch", WM8994_AIF2_ADC_FILTERS, 12, 11, 1, 0),
 600
 601 SOC_ENUM("ADC OSR", adc_osr),
 602 SOC_ENUM("DAC OSR", dac_osr),
 603
 604 SOC_DOUBLE_R_TLV("DAC1 Volume", WM8994_DAC1_LEFT_VOLUME,
 605                  WM8994_DAC1_RIGHT_VOLUME, 1, 96, 0, digital_tlv),
 606 SOC_DOUBLE_R("DAC1 Switch", WM8994_DAC1_LEFT_VOLUME,
 607              WM8994_DAC1_RIGHT_VOLUME, 9, 1, 1),
 608
 609 SOC_DOUBLE_R_TLV("DAC2 Volume", WM8994_DAC2_LEFT_VOLUME,
 610                  WM8994_DAC2_RIGHT_VOLUME, 1, 96, 0, digital_tlv),
 611 SOC_DOUBLE_R("DAC2 Switch", WM8994_DAC2_LEFT_VOLUME,
 612              WM8994_DAC2_RIGHT_VOLUME, 9, 1, 1),
 613
 614 SOC_SINGLE_TLV("SPKL DAC2 Volume", WM8994_SPKMIXL_ATTENUATION,
 615                6, 1, 1, wm_hubs_spkmix_tlv),
 616 SOC_SINGLE_TLV("SPKL DAC1 Volume", WM8994_SPKMIXL_ATTENUATION,
 617                2, 1, 1, wm_hubs_spkmix_tlv),
 618
 619 SOC_SINGLE_TLV("SPKR DAC2 Volume", WM8994_SPKMIXR_ATTENUATION,
 620                6, 1, 1, wm_hubs_spkmix_tlv),
 621 SOC_SINGLE_TLV("SPKR DAC1 Volume", WM8994_SPKMIXR_ATTENUATION,
 622                2, 1, 1, wm_hubs_spkmix_tlv),
 623
 624 SOC_SINGLE_TLV("AIF1DAC1 3D Stereo Volume", WM8994_AIF1_DAC1_FILTERS_2,
 625                10, 15, 0, wm8994_3d_tlv),
 626 SOC_SINGLE("AIF1DAC1 3D Stereo Switch", WM8994_AIF1_DAC1_FILTERS_2,
 627            8, 1, 0),
 628 SOC_SINGLE_TLV("AIF1DAC2 3D Stereo Volume", WM8994_AIF1_DAC2_FILTERS_2,
 629                10, 15, 0, wm8994_3d_tlv),
 630 SOC_SINGLE("AIF1DAC2 3D Stereo Switch", WM8994_AIF1_DAC2_FILTERS_2,
 631            8, 1, 0),
 632 SOC_SINGLE_TLV("AIF2DAC 3D Stereo Volume", WM8994_AIF2_DAC_FILTERS_2,
 633                10, 15, 0, wm8994_3d_tlv),
 634 SOC_SINGLE("AIF2DAC 3D Stereo Switch", WM8994_AIF2_DAC_FILTERS_2,
 635            8, 1, 0),
 636 };
 637
 638 static const struct snd_kcontrol_new wm8994_eq_controls[] = {
 639 SOC_SINGLE_TLV("AIF1DAC1 EQ1 Volume", WM8994_AIF1_DAC1_EQ_GAINS_1, 11, 31, 0,
 640                eq_tlv),
 641 SOC_SINGLE_TLV("AIF1DAC1 EQ2 Volume", WM8994_AIF1_DAC1_EQ_GAINS_1, 6, 31, 0,
 642                eq_tlv),
 643 SOC_SINGLE_TLV("AIF1DAC1 EQ3 Volume", WM8994_AIF1_DAC1_EQ_GAINS_1, 1, 31, 0,
 644                eq_tlv),
 645 SOC_SINGLE_TLV("AIF1DAC1 EQ4 Volume", WM8994_AIF1_DAC1_EQ_GAINS_2, 11, 31, 0,
 646                eq_tlv),
 647 SOC_SINGLE_TLV("AIF1DAC1 EQ5 Volume", WM8994_AIF1_DAC1_EQ_GAINS_2, 6, 31, 0,
 648                eq_tlv),
 649
 650 SOC_SINGLE_TLV("AIF1DAC2 EQ1 Volume", WM8994_AIF1_DAC2_EQ_GAINS_1, 11, 31, 0,
 651                eq_tlv),
 652 SOC_SINGLE_TLV("AIF1DAC2 EQ2 Volume", WM8994_AIF1_DAC2_EQ_GAINS_1, 6, 31, 0,
 653                eq_tlv),
 654 SOC_SINGLE_TLV("AIF1DAC2 EQ3 Volume", WM8994_AIF1_DAC2_EQ_GAINS_1, 1, 31, 0,
 655                eq_tlv),
 656 SOC_SINGLE_TLV("AIF1DAC2 EQ4 Volume", WM8994_AIF1_DAC2_EQ_GAINS_2, 11, 31, 0,
 657                eq_tlv),
 658 SOC_SINGLE_TLV("AIF1DAC2 EQ5 Volume", WM8994_AIF1_DAC2_EQ_GAINS_2, 6, 31, 0,
 659                eq_tlv),
 660
 661 SOC_SINGLE_TLV("AIF2 EQ1 Volume", WM8994_AIF2_EQ_GAINS_1, 11, 31, 0,
 662                eq_tlv),
 663 SOC_SINGLE_TLV("AIF2 EQ2 Volume", WM8994_AIF2_EQ_GAINS_1, 6, 31, 0,
 664                eq_tlv),
 665 SOC_SINGLE_TLV("AIF2 EQ3 Volume", WM8994_AIF2_EQ_GAINS_1, 1, 31, 0,
 666                eq_tlv),
 667 SOC_SINGLE_TLV("AIF2 EQ4 Volume", WM8994_AIF2_EQ_GAINS_2, 11, 31, 0,
 668                eq_tlv),
 669 SOC_SINGLE_TLV("AIF2 EQ5 Volume", WM8994_AIF2_EQ_GAINS_2, 6, 31, 0,
 670                eq_tlv),
 671 };
 672
 673 static const struct snd_kcontrol_new wm8994_drc_controls[] = {
 674 SND_SOC_BYTES_MASK("AIF1.1 DRC", WM8994_AIF1_DRC1_1, 5,
 675                    WM8994_AIF1DAC1_DRC_ENA | WM8994_AIF1ADC1L_DRC_ENA |
 676                    WM8994_AIF1ADC1R_DRC_ENA),
 677 SND_SOC_BYTES_MASK("AIF1.2 DRC", WM8994_AIF1_DRC2_1, 5,
 678                    WM8994_AIF1DAC2_DRC_ENA | WM8994_AIF1ADC2L_DRC_ENA |
 679                    WM8994_AIF1ADC2R_DRC_ENA),
 680 SND_SOC_BYTES_MASK("AIF2 DRC", WM8994_AIF2_DRC_1, 5,
 681                    WM8994_AIF2DAC_DRC_ENA | WM8994_AIF2ADCL_DRC_ENA |
 682                    WM8994_AIF2ADCR_DRC_ENA),
 683 };
 684
 685 static const char *wm8958_ng_text[] = {
 686         "30ms", "125ms", "250ms", "500ms",
 687 };
 688
 689 static const struct soc_enum wm8958_aif1dac1_ng_hold =
 690         SOC_ENUM_SINGLE(WM8958_AIF1_DAC1_NOISE_GATE,
 691                         WM8958_AIF1DAC1_NG_THR_SHIFT, 4, wm8958_ng_text);
 692
 693 static const struct soc_enum wm8958_aif1dac2_ng_hold =
 694         SOC_ENUM_SINGLE(WM8958_AIF1_DAC2_NOISE_GATE,
 695                         WM8958_AIF1DAC2_NG_THR_SHIFT, 4, wm8958_ng_text);
 696
 697 static const struct soc_enum wm8958_aif2dac_ng_hold =
 698         SOC_ENUM_SINGLE(WM8958_AIF2_DAC_NOISE_GATE,
 699                         WM8958_AIF2DAC_NG_THR_SHIFT, 4, wm8958_ng_text);
 700
 701 static const struct snd_kcontrol_new wm8958_snd_controls[] = {
 702 SOC_SINGLE_TLV("AIF3 Boost Volume", WM8958_AIF3_CONTROL_2, 10, 3, 0, aif_tlv),
 703
 704 SOC_SINGLE("AIF1DAC1 Noise Gate Switch", WM8958_AIF1_DAC1_NOISE_GATE,
 705            WM8958_AIF1DAC1_NG_ENA_SHIFT, 1, 0),
 706 SOC_ENUM("AIF1DAC1 Noise Gate Hold Time", wm8958_aif1dac1_ng_hold),
 707 SOC_SINGLE_TLV("AIF1DAC1 Noise Gate Threshold Volume",
 708                WM8958_AIF1_DAC1_NOISE_GATE, WM8958_AIF1DAC1_NG_THR_SHIFT,
 709                7, 1, ng_tlv),
 710
 711 SOC_SINGLE("AIF1DAC2 Noise Gate Switch", WM8958_AIF1_DAC2_NOISE_GATE,
 712            WM8958_AIF1DAC2_NG_ENA_SHIFT, 1, 0),
 713 SOC_ENUM("AIF1DAC2 Noise Gate Hold Time", wm8958_aif1dac2_ng_hold),
 714 SOC_SINGLE_TLV("AIF1DAC2 Noise Gate Threshold Volume",
 715                WM8958_AIF1_DAC2_NOISE_GATE, WM8958_AIF1DAC2_NG_THR_SHIFT,
 716                7, 1, ng_tlv),
 717
 718 SOC_SINGLE("AIF2DAC Noise Gate Switch", WM8958_AIF2_DAC_NOISE_GATE,
 719            WM8958_AIF2DAC_NG_ENA_SHIFT, 1, 0),
 720 SOC_ENUM("AIF2DAC Noise Gate Hold Time", wm8958_aif2dac_ng_hold),
 721 SOC_SINGLE_TLV("AIF2DAC Noise Gate Threshold Volume",
 722                WM8958_AIF2_DAC_NOISE_GATE, WM8958_AIF2DAC_NG_THR_SHIFT,
 723                7, 1, ng_tlv),
 724 };
 725
 726 static const struct snd_kcontrol_new wm1811_snd_controls[] = {
 727 SOC_SINGLE_TLV("MIXINL IN1LP Boost Volume", WM8994_INPUT_MIXER_1, 7, 1, 0,
 728                mixin_boost_tlv),
 729 SOC_SINGLE_TLV("MIXINL IN1RP Boost Volume", WM8994_INPUT_MIXER_1, 8, 1, 0,
 730                mixin_boost_tlv),
 731 };
 732
 733 /* We run all mode setting through a function to enforce audio mode */
 734 static void wm1811_jackdet_set_mode(struct snd_soc_codec *codec, u16 mode)
 735 {
 736         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 737
 738         if (!wm8994->jackdet || !wm8994->micdet[0].jack)
 739                 return;
 740
 741         if (wm8994->active_refcount)
 742                 mode = WM1811_JACKDET_MODE_AUDIO;
 743
 744         if (mode == wm8994->jackdet_mode)
 745                 return;
 746
 747         wm8994->jackdet_mode = mode;
 748
 749         /* Always use audio mode to detect while the system is active */
 750         if (mode != WM1811_JACKDET_MODE_NONE)
 751                 mode = WM1811_JACKDET_MODE_AUDIO;
 752
 753         snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 754                             WM1811_JACKDET_MODE_MASK, mode);
 755 }
 756
 757 static void active_reference(struct snd_soc_codec *codec)
 758 {
 759         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 760
 761         mutex_lock(&wm8994->accdet_lock);
 762
 763         wm8994->active_refcount++;
 764
 765         dev_dbg(codec->dev, "Active refcount incremented, now %d\n",
 766                 wm8994->active_refcount);
 767
 768         /* If we're using jack detection go into audio mode */
 769         wm1811_jackdet_set_mode(codec, WM1811_JACKDET_MODE_AUDIO);
 770
 771         mutex_unlock(&wm8994->accdet_lock);
 772 }
 773
 774 static void active_dereference(struct snd_soc_codec *codec)
 775 {
 776         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 777         u16 mode;
 778
 779         mutex_lock(&wm8994->accdet_lock);
 780
 781         wm8994->active_refcount--;
 782
 783         dev_dbg(codec->dev, "Active refcount decremented, now %d\n",
 784                 wm8994->active_refcount);
 785
 786         if (wm8994->active_refcount == 0) {
 787                 /* Go into appropriate detection only mode */
 788                 if (wm8994->jack_mic || wm8994->mic_detecting)
 789                         mode = WM1811_JACKDET_MODE_MIC;
 790                 else
 791                         mode = WM1811_JACKDET_MODE_JACK;
 792
 793                 wm1811_jackdet_set_mode(codec, mode);
 794         }
 795
 796         mutex_unlock(&wm8994->accdet_lock);
 797 }
 798
 799 static int clk_sys_event(struct snd_soc_dapm_widget *w,
 800                          struct snd_kcontrol *kcontrol, int event)
 801 {
 802         struct snd_soc_codec *codec = w->codec;
 803         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 804
 805         switch (event) {
 806         case SND_SOC_DAPM_PRE_PMU:
 807                 return configure_clock(codec);
 808
 809         case SND_SOC_DAPM_POST_PMU:
 810                 /*
 811                  * JACKDET won't run until we start the clock and it
 812                  * only reports deltas, make sure we notify the state
 813                  * up the stack on startup.  Use a *very* generous
 814                  * timeout for paranoia, there's no urgency and we
 815                  * don't want false reports.
 816                  */
 817                 if (wm8994->jackdet && !wm8994->clk_has_run) {
 818                         schedule_delayed_work(&wm8994->jackdet_bootstrap,
 819                                               msecs_to_jiffies(1000));
 820                         wm8994->clk_has_run = true;
 821                 }
 822                 break;
 823
 824         case SND_SOC_DAPM_POST_PMD:
 825                 configure_clock(codec);
 826                 break;
 827         }
 828
 829         return 0;
 830 }
 831
 832 static void vmid_reference(struct snd_soc_codec *codec)
 833 {
 834         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 835
 836         pm_runtime_get_sync(codec->dev);
 837
 838         wm8994->vmid_refcount++;
 839
 840         dev_dbg(codec->dev, "Referencing VMID, refcount is now %d\n",
 841                 wm8994->vmid_refcount);
 842
 843         if (wm8994->vmid_refcount == 1) {
 844                 snd_soc_update_bits(codec, WM8994_ANTIPOP_1,
 845                                     WM8994_LINEOUT1_DISCH |
 846                                     WM8994_LINEOUT2_DISCH, 0);
 847
 848                 wm_hubs_vmid_ena(codec);
 849
 850                 switch (wm8994->vmid_mode) {
 851                 default:
 852                         WARN_ON(NULL == "Invalid VMID mode");
 853                 case WM8994_VMID_NORMAL:
 854                         /* Startup bias, VMID ramp & buffer */
 855                         snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 856                                             WM8994_BIAS_SRC |
 857                                             WM8994_VMID_DISCH |
 858                                             WM8994_STARTUP_BIAS_ENA |
 859                                             WM8994_VMID_BUF_ENA |
 860                                             WM8994_VMID_RAMP_MASK,
 861                                             WM8994_BIAS_SRC |
 862                                             WM8994_STARTUP_BIAS_ENA |
 863                                             WM8994_VMID_BUF_ENA |
 864                                             (0x2 << WM8994_VMID_RAMP_SHIFT));
 865
 866                         /* Main bias enable, VMID=2x40k */
 867                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_1,
 868                                             WM8994_BIAS_ENA |
 869                                             WM8994_VMID_SEL_MASK,
 870                                             WM8994_BIAS_ENA | 0x2);
 871
 872                         msleep(300);
 873
 874                         snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 875                                             WM8994_VMID_RAMP_MASK |
 876                                             WM8994_BIAS_SRC,
 877                                             0);
 878                         break;
 879
 880                 case WM8994_VMID_FORCE:
 881                         /* Startup bias, slow VMID ramp & buffer */
 882                         snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 883                                             WM8994_BIAS_SRC |
 884                                             WM8994_VMID_DISCH |
 885                                             WM8994_STARTUP_BIAS_ENA |
 886                                             WM8994_VMID_BUF_ENA |
 887                                             WM8994_VMID_RAMP_MASK,
 888                                             WM8994_BIAS_SRC |
 889                                             WM8994_STARTUP_BIAS_ENA |
 890                                             WM8994_VMID_BUF_ENA |
 891                                             (0x2 << WM8994_VMID_RAMP_SHIFT));
 892
 893                         /* Main bias enable, VMID=2x40k */
 894                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_1,
 895                                             WM8994_BIAS_ENA |
 896                                             WM8994_VMID_SEL_MASK,
 897                                             WM8994_BIAS_ENA | 0x2);
 898
 899                         msleep(400);
 900
 901                         snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 902                                             WM8994_VMID_RAMP_MASK |
 903                                             WM8994_BIAS_SRC,
 904                                             0);
 905                         break;
 906                 }
 907         }
 908 }
 909
 910 static void vmid_dereference(struct snd_soc_codec *codec)
 911 {
 912         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
 913
 914         wm8994->vmid_refcount--;
 915
 916         dev_dbg(codec->dev, "Dereferencing VMID, refcount is now %d\n",
 917                 wm8994->vmid_refcount);
 918
 919         if (wm8994->vmid_refcount == 0) {
 920                 if (wm8994->hubs.lineout1_se)
 921                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_3,
 922                                             WM8994_LINEOUT1N_ENA |
 923                                             WM8994_LINEOUT1P_ENA,
 924                                             WM8994_LINEOUT1N_ENA |
 925                                             WM8994_LINEOUT1P_ENA);
 926
 927                 if (wm8994->hubs.lineout2_se)
 928                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_3,
 929                                             WM8994_LINEOUT2N_ENA |
 930                                             WM8994_LINEOUT2P_ENA,
 931                                             WM8994_LINEOUT2N_ENA |
 932                                             WM8994_LINEOUT2P_ENA);
 933
 934                 /* Start discharging VMID */
 935                 snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 936                                     WM8994_BIAS_SRC |
 937                                     WM8994_VMID_DISCH,
 938                                     WM8994_BIAS_SRC |
 939                                     WM8994_VMID_DISCH);
 940
 941                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_1,
 942                                     WM8994_VMID_SEL_MASK, 0);
 943
 944                 msleep(400);
 945
 946                 /* Active discharge */
 947                 snd_soc_update_bits(codec, WM8994_ANTIPOP_1,
 948                                     WM8994_LINEOUT1_DISCH |
 949                                     WM8994_LINEOUT2_DISCH,
 950                                     WM8994_LINEOUT1_DISCH |
 951                                     WM8994_LINEOUT2_DISCH);
 952
 953                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_3,
 954                                     WM8994_LINEOUT1N_ENA |
 955                                     WM8994_LINEOUT1P_ENA |
 956                                     WM8994_LINEOUT2N_ENA |
 957                                     WM8994_LINEOUT2P_ENA, 0);
 958
 959                 /* Switch off startup biases */
 960                 snd_soc_update_bits(codec, WM8994_ANTIPOP_2,
 961                                     WM8994_BIAS_SRC |
 962                                     WM8994_STARTUP_BIAS_ENA |
 963                                     WM8994_VMID_BUF_ENA |
 964                                     WM8994_VMID_RAMP_MASK, 0);
 965
 966                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_1,
 967                                     WM8994_VMID_SEL_MASK, 0);
 968         }
 969
 970         pm_runtime_put(codec->dev);
 971 }
 972
 973 static int vmid_event(struct snd_soc_dapm_widget *w,
 974                       struct snd_kcontrol *kcontrol, int event)
 975 {
 976         struct snd_soc_codec *codec = w->codec;
 977
 978         switch (event) {
 979         case SND_SOC_DAPM_PRE_PMU:
 980                 vmid_reference(codec);
 981                 break;
 982
 983         case SND_SOC_DAPM_POST_PMD:
 984                 vmid_dereference(codec);
 985                 break;
 986         }
 987
 988         return 0;
 989 }
 990
 991 static bool wm8994_check_class_w_digital(struct snd_soc_codec *codec)
 992 {
 993         int source = 0;  /* GCC flow analysis can't track enable */
 994         int reg, reg_r;
 995
 996         /* We also need the same AIF source for L/R and only one path */
 997         reg = snd_soc_read(codec, WM8994_DAC1_LEFT_MIXER_ROUTING);
 998         switch (reg) {
 999         case WM8994_AIF2DACL_TO_DAC1L:
1000                 dev_vdbg(codec->dev, "Class W source AIF2DAC\n");
1001                 source = 2 << WM8994_CP_DYN_SRC_SEL_SHIFT;
1002                 break;
1003         case WM8994_AIF1DAC2L_TO_DAC1L:
1004                 dev_vdbg(codec->dev, "Class W source AIF1DAC2\n");
1005                 source = 1 << WM8994_CP_DYN_SRC_SEL_SHIFT;
1006                 break;
1007         case WM8994_AIF1DAC1L_TO_DAC1L:
1008                 dev_vdbg(codec->dev, "Class W source AIF1DAC1\n");
1009                 source = 0 << WM8994_CP_DYN_SRC_SEL_SHIFT;
1010                 break;
1011         default:
1012                 dev_vdbg(codec->dev, "DAC mixer setting: %x\n", reg);
1013                 return false;
1014         }
1015
1016         reg_r = snd_soc_read(codec, WM8994_DAC1_RIGHT_MIXER_ROUTING);
1017         if (reg_r != reg) {
1018                 dev_vdbg(codec->dev, "Left and right DAC mixers different\n");
1019                 return false;
1020         }
1021
1022         /* Set the source up */
1023         snd_soc_update_bits(codec, WM8994_CLASS_W_1,
1024                             WM8994_CP_DYN_SRC_SEL_MASK, source);
1025
1026         return true;
1027 }
1028
1029 static int aif1clk_ev(struct snd_soc_dapm_widget *w,
1030                       struct snd_kcontrol *kcontrol, int event)
1031 {
1032         struct snd_soc_codec *codec = w->codec;
1033         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
1034         struct wm8994 *control = codec->control_data;
1035         int mask = WM8994_AIF1DAC1L_ENA | WM8994_AIF1DAC1R_ENA;
1036         int i;
1037         int dac;
1038         int adc;
1039         int val;
1040
1041         switch (control->type) {
1042         case WM8994:
1043         case WM8958:
1044                 mask |= WM8994_AIF1DAC2L_ENA | WM8994_AIF1DAC2R_ENA;
1045                 break;
1046         default:
1047                 break;
1048         }
1049
1050         switch (event) {
1051         case SND_SOC_DAPM_PRE_PMU:
1052                 /* Don't enable timeslot 2 if not in use */
1053                 if (wm8994->channels[0] <= 2)
1054                         mask &= ~(WM8994_AIF1DAC2L_ENA | WM8994_AIF1DAC2R_ENA);
1055
1056                 val = snd_soc_read(codec, WM8994_AIF1_CONTROL_1);
1057                 if ((val & WM8994_AIF1ADCL_SRC) &&
1058                     (val & WM8994_AIF1ADCR_SRC))
1059                         adc = WM8994_AIF1ADC1R_ENA | WM8994_AIF1ADC2R_ENA;
1060                 else if (!(val & WM8994_AIF1ADCL_SRC) &&
1061                          !(val & WM8994_AIF1ADCR_SRC))
1062                         adc = WM8994_AIF1ADC1L_ENA | WM8994_AIF1ADC2L_ENA;
1063                 else
1064                         adc = WM8994_AIF1ADC1R_ENA | WM8994_AIF1ADC2R_ENA |
1065                                 WM8994_AIF1ADC1L_ENA | WM8994_AIF1ADC2L_ENA;
1066
1067                 val = snd_soc_read(codec, WM8994_AIF1_CONTROL_2);
1068                 if ((val & WM8994_AIF1DACL_SRC) &&
1069                     (val & WM8994_AIF1DACR_SRC))
1070                         dac = WM8994_AIF1DAC1R_ENA | WM8994_AIF1DAC2R_ENA;
1071                 else if (!(val & WM8994_AIF1DACL_SRC) &&
1072                          !(val & WM8994_AIF1DACR_SRC))
1073                         dac = WM8994_AIF1DAC1L_ENA | WM8994_AIF1DAC2L_ENA;
1074                 else
1075                         dac = WM8994_AIF1DAC1R_ENA | WM8994_AIF1DAC2R_ENA |
1076                                 WM8994_AIF1DAC1L_ENA | WM8994_AIF1DAC2L_ENA;
1077
1078                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4,
1079                                     mask, adc);
1080                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1081                                     mask, dac);
1082                 snd_soc_update_bits(codec, WM8994_CLOCKING_1,
1083                                     WM8994_AIF1DSPCLK_ENA |
1084                                     WM8994_SYSDSPCLK_ENA,
1085                                     WM8994_AIF1DSPCLK_ENA |
1086                                     WM8994_SYSDSPCLK_ENA);
1087                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4, mask,
1088                                     WM8994_AIF1ADC1R_ENA |
1089                                     WM8994_AIF1ADC1L_ENA |
1090                                     WM8994_AIF1ADC2R_ENA |
1091                                     WM8994_AIF1ADC2L_ENA);
1092                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5, mask,
1093                                     WM8994_AIF1DAC1R_ENA |
1094                                     WM8994_AIF1DAC1L_ENA |
1095                                     WM8994_AIF1DAC2R_ENA |
1096                                     WM8994_AIF1DAC2L_ENA);
1097                 break;
1098
1099         case SND_SOC_DAPM_POST_PMU:
1100                 for (i = 0; i < ARRAY_SIZE(wm8994_vu_bits); i++)
1101                         snd_soc_write(codec, wm8994_vu_bits[i].reg,
1102                                       snd_soc_read(codec,
1103                                                    wm8994_vu_bits[i].reg));
1104                 break;
1105
1106         case SND_SOC_DAPM_PRE_PMD:
1107         case SND_SOC_DAPM_POST_PMD:
1108                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1109                                     mask, 0);
1110                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4,
1111                                     mask, 0);
1112
1113                 val = snd_soc_read(codec, WM8994_CLOCKING_1);
1114                 if (val & WM8994_AIF2DSPCLK_ENA)
1115                         val = WM8994_SYSDSPCLK_ENA;
1116                 else
1117                         val = 0;
1118                 snd_soc_update_bits(codec, WM8994_CLOCKING_1,
1119                                     WM8994_SYSDSPCLK_ENA |
1120                                     WM8994_AIF1DSPCLK_ENA, val);
1121                 break;
1122         }
1123
1124         return 0;
1125 }
1126
1127 static int aif2clk_ev(struct snd_soc_dapm_widget *w,
1128                       struct snd_kcontrol *kcontrol, int event)
1129 {
1130         struct snd_soc_codec *codec = w->codec;
1131         int i;
1132         int dac;
1133         int adc;
1134         int val;
1135
1136         switch (event) {
1137         case SND_SOC_DAPM_PRE_PMU:
1138                 val = snd_soc_read(codec, WM8994_AIF2_CONTROL_1);
1139                 if ((val & WM8994_AIF2ADCL_SRC) &&
1140                     (val & WM8994_AIF2ADCR_SRC))
1141                         adc = WM8994_AIF2ADCR_ENA;
1142                 else if (!(val & WM8994_AIF2ADCL_SRC) &&
1143                          !(val & WM8994_AIF2ADCR_SRC))
1144                         adc = WM8994_AIF2ADCL_ENA;
1145                 else
1146                         adc = WM8994_AIF2ADCL_ENA | WM8994_AIF2ADCR_ENA;
1147
1148
1149                 val = snd_soc_read(codec, WM8994_AIF2_CONTROL_2);
1150                 if ((val & WM8994_AIF2DACL_SRC) &&
1151                     (val & WM8994_AIF2DACR_SRC))
1152                         dac = WM8994_AIF2DACR_ENA;
1153                 else if (!(val & WM8994_AIF2DACL_SRC) &&
1154                          !(val & WM8994_AIF2DACR_SRC))
1155                         dac = WM8994_AIF2DACL_ENA;
1156                 else
1157                         dac = WM8994_AIF2DACL_ENA | WM8994_AIF2DACR_ENA;
1158
1159                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4,
1160                                     WM8994_AIF2ADCL_ENA |
1161                                     WM8994_AIF2ADCR_ENA, adc);
1162                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1163                                     WM8994_AIF2DACL_ENA |
1164                                     WM8994_AIF2DACR_ENA, dac);
1165                 snd_soc_update_bits(codec, WM8994_CLOCKING_1,
1166                                     WM8994_AIF2DSPCLK_ENA |
1167                                     WM8994_SYSDSPCLK_ENA,
1168                                     WM8994_AIF2DSPCLK_ENA |
1169                                     WM8994_SYSDSPCLK_ENA);
1170                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4,
1171                                     WM8994_AIF2ADCL_ENA |
1172                                     WM8994_AIF2ADCR_ENA,
1173                                     WM8994_AIF2ADCL_ENA |
1174                                     WM8994_AIF2ADCR_ENA);
1175                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1176                                     WM8994_AIF2DACL_ENA |
1177                                     WM8994_AIF2DACR_ENA,
1178                                     WM8994_AIF2DACL_ENA |
1179                                     WM8994_AIF2DACR_ENA);
1180                 break;
1181
1182         case SND_SOC_DAPM_POST_PMU:
1183                 for (i = 0; i < ARRAY_SIZE(wm8994_vu_bits); i++)
1184                         snd_soc_write(codec, wm8994_vu_bits[i].reg,
1185                                       snd_soc_read(codec,
1186                                                    wm8994_vu_bits[i].reg));
1187                 break;
1188
1189         case SND_SOC_DAPM_PRE_PMD:
1190         case SND_SOC_DAPM_POST_PMD:
1191                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1192                                     WM8994_AIF2DACL_ENA |
1193                                     WM8994_AIF2DACR_ENA, 0);
1194                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_4,
1195                                     WM8994_AIF2ADCL_ENA |
1196                                     WM8994_AIF2ADCR_ENA, 0);
1197
1198                 val = snd_soc_read(codec, WM8994_CLOCKING_1);
1199                 if (val & WM8994_AIF1DSPCLK_ENA)
1200                         val = WM8994_SYSDSPCLK_ENA;
1201                 else
1202                         val = 0;
1203                 snd_soc_update_bits(codec, WM8994_CLOCKING_1,
1204                                     WM8994_SYSDSPCLK_ENA |
1205                                     WM8994_AIF2DSPCLK_ENA, val);
1206                 break;
1207         }
1208
1209         return 0;
1210 }
1211
1212 static int aif1clk_late_ev(struct snd_soc_dapm_widget *w,
1213                            struct snd_kcontrol *kcontrol, int event)
1214 {
1215         struct snd_soc_codec *codec = w->codec;
1216         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
1217
1218         switch (event) {
1219         case SND_SOC_DAPM_PRE_PMU:
1220                 wm8994->aif1clk_enable = 1;
1221                 break;
1222         case SND_SOC_DAPM_POST_PMD:
1223                 wm8994->aif1clk_disable = 1;
1224                 break;
1225         }
1226
1227         return 0;
1228 }
1229
1230 static int aif2clk_late_ev(struct snd_soc_dapm_widget *w,
1231                            struct snd_kcontrol *kcontrol, int event)
1232 {
1233         struct snd_soc_codec *codec = w->codec;
1234         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
1235
1236         switch (event) {
1237         case SND_SOC_DAPM_PRE_PMU:
1238                 wm8994->aif2clk_enable = 1;
1239                 break;
1240         case SND_SOC_DAPM_POST_PMD:
1241                 wm8994->aif2clk_disable = 1;
1242                 break;
1243         }
1244
1245         return 0;
1246 }
1247
1248 static int late_enable_ev(struct snd_soc_dapm_widget *w,
1249                           struct snd_kcontrol *kcontrol, int event)
1250 {
1251         struct snd_soc_codec *codec = w->codec;
1252         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
1253
1254         switch (event) {
1255         case SND_SOC_DAPM_PRE_PMU:
1256                 if (wm8994->aif1clk_enable) {
1257                         aif1clk_ev(w, kcontrol, SND_SOC_DAPM_PRE_PMU);
1258                         snd_soc_update_bits(codec, WM8994_AIF1_CLOCKING_1,
1259                                             WM8994_AIF1CLK_ENA_MASK,
1260                                             WM8994_AIF1CLK_ENA);
1261                         aif1clk_ev(w, kcontrol, SND_SOC_DAPM_POST_PMU);
1262                         wm8994->aif1clk_enable = 0;
1263                 }
1264                 if (wm8994->aif2clk_enable) {
1265                         aif2clk_ev(w, kcontrol, SND_SOC_DAPM_PRE_PMU);
1266                         snd_soc_update_bits(codec, WM8994_AIF2_CLOCKING_1,
1267                                             WM8994_AIF2CLK_ENA_MASK,
1268                                             WM8994_AIF2CLK_ENA);
1269                         aif2clk_ev(w, kcontrol, SND_SOC_DAPM_POST_PMU);
1270                         wm8994->aif2clk_enable = 0;
1271                 }
1272                 break;
1273         }
1274
1275         /* We may also have postponed startup of DSP, handle that. */
1276         wm8958_aif_ev(w, kcontrol, event);
1277
1278         return 0;
1279 }
1280
1281 static int late_disable_ev(struct snd_soc_dapm_widget *w,
1282                            struct snd_kcontrol *kcontrol, int event)
1283 {
1284         struct snd_soc_codec *codec = w->codec;
1285         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
1286
1287         switch (event) {
1288         case SND_SOC_DAPM_POST_PMD:
1289                 if (wm8994->aif1clk_disable) {
1290                         aif1clk_ev(w, kcontrol, SND_SOC_DAPM_PRE_PMD);
1291                         snd_soc_update_bits(codec, WM8994_AIF1_CLOCKING_1,
1292                                             WM8994_AIF1CLK_ENA_MASK, 0);
1293                         aif1clk_ev(w, kcontrol, SND_SOC_DAPM_POST_PMD);
1294                         wm8994->aif1clk_disable = 0;
1295                 }
1296                 if (wm8994->aif2clk_disable) {
1297                         aif2clk_ev(w, kcontrol, SND_SOC_DAPM_PRE_PMD);
1298                         snd_soc_update_bits(codec, WM8994_AIF2_CLOCKING_1,
1299                                             WM8994_AIF2CLK_ENA_MASK, 0);
1300                         aif2clk_ev(w, kcontrol, SND_SOC_DAPM_POST_PMD);
1301                         wm8994->aif2clk_disable = 0;
1302                 }
1303                 break;
1304         }
1305
1306         return 0;
1307 }
1308
1309 static int adc_mux_ev(struct snd_soc_dapm_widget *w,
1310                       struct snd_kcontrol *kcontrol, int event)
1311 {
1312         late_enable_ev(w, kcontrol, event);
1313         return 0;
1314 }
1315
1316 static int micbias_ev(struct snd_soc_dapm_widget *w,
1317                       struct snd_kcontrol *kcontrol, int event)
1318 {
1319         late_enable_ev(w, kcontrol, event);
1320         return 0;
1321 }
1322
1323 static int dac_ev(struct snd_soc_dapm_widget *w,
1324                   struct snd_kcontrol *kcontrol, int event)
1325 {
1326         struct snd_soc_codec *codec = w->codec;
1327         unsigned int mask = 1 << w->shift;
1328
1329         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
1330                             mask, mask);
1331         return 0;
1332 }
1333
1334 static const char *adc_mux_text[] = {
1335         "ADC",
1336         "DMIC",
1337 };
1338
1339 static const struct soc_enum adc_enum =
1340         SOC_ENUM_SINGLE(0, 0, 2, adc_mux_text);
1341
1342 static const struct snd_kcontrol_new adcl_mux =
1343         SOC_DAPM_ENUM_VIRT("ADCL Mux", adc_enum);
1344
1345 static const struct snd_kcontrol_new adcr_mux =
1346         SOC_DAPM_ENUM_VIRT("ADCR Mux", adc_enum);
1347
1348 static const struct snd_kcontrol_new left_speaker_mixer[] = {
1349 SOC_DAPM_SINGLE("DAC2 Switch", WM8994_SPEAKER_MIXER, 9, 1, 0),
1350 SOC_DAPM_SINGLE("Input Switch", WM8994_SPEAKER_MIXER, 7, 1, 0),
1351 SOC_DAPM_SINGLE("IN1LP Switch", WM8994_SPEAKER_MIXER, 5, 1, 0),
1352 SOC_DAPM_SINGLE("Output Switch", WM8994_SPEAKER_MIXER, 3, 1, 0),
1353 SOC_DAPM_SINGLE("DAC1 Switch", WM8994_SPEAKER_MIXER, 1, 1, 0),
1354 };
1355
1356 static const struct snd_kcontrol_new right_speaker_mixer[] = {
1357 SOC_DAPM_SINGLE("DAC2 Switch", WM8994_SPEAKER_MIXER, 8, 1, 0),
1358 SOC_DAPM_SINGLE("Input Switch", WM8994_SPEAKER_MIXER, 6, 1, 0),
1359 SOC_DAPM_SINGLE("IN1RP Switch", WM8994_SPEAKER_MIXER, 4, 1, 0),
1360 SOC_DAPM_SINGLE("Output Switch", WM8994_SPEAKER_MIXER, 2, 1, 0),
1361 SOC_DAPM_SINGLE("DAC1 Switch", WM8994_SPEAKER_MIXER, 0, 1, 0),
1362 };
1363
1364 /* Debugging; dump chip status after DAPM transitions */
1365 static int post_ev(struct snd_soc_dapm_widget *w,
1366             struct snd_kcontrol *kcontrol, int event)
1367 {
1368         struct snd_soc_codec *codec = w->codec;
1369         dev_dbg(codec->dev, "SRC status: %x\n",
1370                 snd_soc_read(codec,
1371                              WM8994_RATE_STATUS));
1372         return 0;
1373 }
1374
1375 static const struct snd_kcontrol_new aif1adc1l_mix[] = {
1376 SOC_DAPM_SINGLE("ADC/DMIC Switch", WM8994_AIF1_ADC1_LEFT_MIXER_ROUTING,
1377                 1, 1, 0),
1378 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_AIF1_ADC1_LEFT_MIXER_ROUTING,
1379                 0, 1, 0),
1380 };
1381
1382 static const struct snd_kcontrol_new aif1adc1r_mix[] = {
1383 SOC_DAPM_SINGLE("ADC/DMIC Switch", WM8994_AIF1_ADC1_RIGHT_MIXER_ROUTING,
1384                 1, 1, 0),
1385 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_AIF1_ADC1_RIGHT_MIXER_ROUTING,
1386                 0, 1, 0),
1387 };
1388
1389 static const struct snd_kcontrol_new aif1adc2l_mix[] = {
1390 SOC_DAPM_SINGLE("DMIC Switch", WM8994_AIF1_ADC2_LEFT_MIXER_ROUTING,
1391                 1, 1, 0),
1392 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_AIF1_ADC2_LEFT_MIXER_ROUTING,
1393                 0, 1, 0),
1394 };
1395
1396 static const struct snd_kcontrol_new aif1adc2r_mix[] = {
1397 SOC_DAPM_SINGLE("DMIC Switch", WM8994_AIF1_ADC2_RIGHT_MIXER_ROUTING,
1398                 1, 1, 0),
1399 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_AIF1_ADC2_RIGHT_MIXER_ROUTING,
1400                 0, 1, 0),
1401 };
1402
1403 static const struct snd_kcontrol_new aif2dac2l_mix[] = {
1404 SOC_DAPM_SINGLE("Right Sidetone Switch", WM8994_DAC2_LEFT_MIXER_ROUTING,
1405                 5, 1, 0),
1406 SOC_DAPM_SINGLE("Left Sidetone Switch", WM8994_DAC2_LEFT_MIXER_ROUTING,
1407                 4, 1, 0),
1408 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_DAC2_LEFT_MIXER_ROUTING,
1409                 2, 1, 0),
1410 SOC_DAPM_SINGLE("AIF1.2 Switch", WM8994_DAC2_LEFT_MIXER_ROUTING,
1411                 1, 1, 0),
1412 SOC_DAPM_SINGLE("AIF1.1 Switch", WM8994_DAC2_LEFT_MIXER_ROUTING,
1413                 0, 1, 0),
1414 };
1415
1416 static const struct snd_kcontrol_new aif2dac2r_mix[] = {
1417 SOC_DAPM_SINGLE("Right Sidetone Switch", WM8994_DAC2_RIGHT_MIXER_ROUTING,
1418                 5, 1, 0),
1419 SOC_DAPM_SINGLE("Left Sidetone Switch", WM8994_DAC2_RIGHT_MIXER_ROUTING,
1420                 4, 1, 0),
1421 SOC_DAPM_SINGLE("AIF2 Switch", WM8994_DAC2_RIGHT_MIXER_ROUTING,
1422                 2, 1, 0),
1423 SOC_DAPM_SINGLE("AIF1.2 Switch", WM8994_DAC2_RIGHT_MIXER_ROUTING,
1424                 1, 1, 0),
1425 SOC_DAPM_SINGLE("AIF1.1 Switch", WM8994_DAC2_RIGHT_MIXER_ROUTING,
1426                 0, 1, 0),
1427 };
1428
1429 #define WM8994_CLASS_W_SWITCH(xname, reg, shift, max, invert) \
1430 {       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, \
1431         .info = snd_soc_info_volsw, \
1432         .get = snd_soc_dapm_get_volsw, .put = wm8994_put_class_w, \
1433         .private_value =  SOC_SINGLE_VALUE(reg, shift, max, invert) }
1434
1435 static int wm8994_put_class_w(struct snd_kcontrol *kcontrol,
1436                               struct snd_ctl_elem_value *ucontrol)
1437 {
1438         struct snd_soc_dapm_widget_list *wlist = snd_kcontrol_chip(kcontrol);
1439         struct snd_soc_dapm_widget *w = wlist->widgets[0];
1440         struct snd_soc_codec *codec = w->codec;
1441         int ret;
1442
1443         ret = snd_soc_dapm_put_volsw(kcontrol, ucontrol);
1444
1445         wm_hubs_update_class_w(codec);
1446
1447         return ret;
1448 }
1449
1450 static const struct snd_kcontrol_new dac1l_mix[] = {
1451 WM8994_CLASS_W_SWITCH("Right Sidetone Switch", WM8994_DAC1_LEFT_MIXER_ROUTING,
1452                       5, 1, 0),
1453 WM8994_CLASS_W_SWITCH("Left Sidetone Switch", WM8994_DAC1_LEFT_MIXER_ROUTING,
1454                       4, 1, 0),
1455 WM8994_CLASS_W_SWITCH("AIF2 Switch", WM8994_DAC1_LEFT_MIXER_ROUTING,
1456                       2, 1, 0),
1457 WM8994_CLASS_W_SWITCH("AIF1.2 Switch", WM8994_DAC1_LEFT_MIXER_ROUTING,
1458                       1, 1, 0),
1459 WM8994_CLASS_W_SWITCH("AIF1.1 Switch", WM8994_DAC1_LEFT_MIXER_ROUTING,
1460                       0, 1, 0),
1461 };
1462
1463 static const struct snd_kcontrol_new dac1r_mix[] = {
1464 WM8994_CLASS_W_SWITCH("Right Sidetone Switch", WM8994_DAC1_RIGHT_MIXER_ROUTING,
1465                       5, 1, 0),
1466 WM8994_CLASS_W_SWITCH("Left Sidetone Switch", WM8994_DAC1_RIGHT_MIXER_ROUTING,
1467                       4, 1, 0),
1468 WM8994_CLASS_W_SWITCH("AIF2 Switch", WM8994_DAC1_RIGHT_MIXER_ROUTING,
1469                       2, 1, 0),
1470 WM8994_CLASS_W_SWITCH("AIF1.2 Switch", WM8994_DAC1_RIGHT_MIXER_ROUTING,
1471                       1, 1, 0),
1472 WM8994_CLASS_W_SWITCH("AIF1.1 Switch", WM8994_DAC1_RIGHT_MIXER_ROUTING,
1473                       0, 1, 0),
1474 };
1475
1476 static const char *sidetone_text[] = {
1477         "ADC/DMIC1", "DMIC2",
1478 };
1479
1480 static const struct soc_enum sidetone1_enum =
1481         SOC_ENUM_SINGLE(WM8994_SIDETONE, 0, 2, sidetone_text);
1482
1483 static const struct snd_kcontrol_new sidetone1_mux =
1484         SOC_DAPM_ENUM("Left Sidetone Mux", sidetone1_enum);
1485
1486 static const struct soc_enum sidetone2_enum =
1487         SOC_ENUM_SINGLE(WM8994_SIDETONE, 1, 2, sidetone_text);
1488
1489 static const struct snd_kcontrol_new sidetone2_mux =
1490         SOC_DAPM_ENUM("Right Sidetone Mux", sidetone2_enum);
1491
1492 static const char *aif1dac_text[] = {
1493         "AIF1DACDAT", "AIF3DACDAT",
1494 };
1495
1496 static const struct soc_enum aif1dac_enum =
1497         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 0, 2, aif1dac_text);
1498
1499 static const struct snd_kcontrol_new aif1dac_mux =
1500         SOC_DAPM_ENUM("AIF1DAC Mux", aif1dac_enum);
1501
1502 static const char *aif2dac_text[] = {
1503         "AIF2DACDAT", "AIF3DACDAT",
1504 };
1505
1506 static const struct soc_enum aif2dac_enum =
1507         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 1, 2, aif2dac_text);
1508
1509 static const struct snd_kcontrol_new aif2dac_mux =
1510         SOC_DAPM_ENUM("AIF2DAC Mux", aif2dac_enum);
1511
1512 static const char *aif2adc_text[] = {
1513         "AIF2ADCDAT", "AIF3DACDAT",
1514 };
1515
1516 static const struct soc_enum aif2adc_enum =
1517         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 2, 2, aif2adc_text);
1518
1519 static const struct snd_kcontrol_new aif2adc_mux =
1520         SOC_DAPM_ENUM("AIF2ADC Mux", aif2adc_enum);
1521
1522 static const char *aif3adc_text[] = {
1523         "AIF1ADCDAT", "AIF2ADCDAT", "AIF2DACDAT", "Mono PCM",
1524 };
1525
1526 static const struct soc_enum wm8994_aif3adc_enum =
1527         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 3, 3, aif3adc_text);
1528
1529 static const struct snd_kcontrol_new wm8994_aif3adc_mux =
1530         SOC_DAPM_ENUM("AIF3ADC Mux", wm8994_aif3adc_enum);
1531
1532 static const struct soc_enum wm8958_aif3adc_enum =
1533         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 3, 4, aif3adc_text);
1534
1535 static const struct snd_kcontrol_new wm8958_aif3adc_mux =
1536         SOC_DAPM_ENUM("AIF3ADC Mux", wm8958_aif3adc_enum);
1537
1538 static const char *mono_pcm_out_text[] = {
1539         "None", "AIF2ADCL", "AIF2ADCR",
1540 };
1541
1542 static const struct soc_enum mono_pcm_out_enum =
1543         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 9, 3, mono_pcm_out_text);
1544
1545 static const struct snd_kcontrol_new mono_pcm_out_mux =
1546         SOC_DAPM_ENUM("Mono PCM Out Mux", mono_pcm_out_enum);
1547
1548 static const char *aif2dac_src_text[] = {
1549         "AIF2", "AIF3",
1550 };
1551
1552 /* Note that these two control shouldn't be simultaneously switched to AIF3 */
1553 static const struct soc_enum aif2dacl_src_enum =
1554         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 7, 2, aif2dac_src_text);
1555
1556 static const struct snd_kcontrol_new aif2dacl_src_mux =
1557         SOC_DAPM_ENUM("AIF2DACL Mux", aif2dacl_src_enum);
1558
1559 static const struct soc_enum aif2dacr_src_enum =
1560         SOC_ENUM_SINGLE(WM8994_POWER_MANAGEMENT_6, 8, 2, aif2dac_src_text);
1561
1562 static const struct snd_kcontrol_new aif2dacr_src_mux =
1563         SOC_DAPM_ENUM("AIF2DACR Mux", aif2dacr_src_enum);
1564
1565 static const struct snd_soc_dapm_widget wm8994_lateclk_revd_widgets[] = {
1566 SND_SOC_DAPM_SUPPLY("AIF1CLK", SND_SOC_NOPM, 0, 0, aif1clk_late_ev,
1567         SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
1568 SND_SOC_DAPM_SUPPLY("AIF2CLK", SND_SOC_NOPM, 0, 0, aif2clk_late_ev,
1569         SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
1570
1571 SND_SOC_DAPM_PGA_E("Late DAC1L Enable PGA", SND_SOC_NOPM, 0, 0, NULL, 0,
1572         late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1573 SND_SOC_DAPM_PGA_E("Late DAC1R Enable PGA", SND_SOC_NOPM, 0, 0, NULL, 0,
1574         late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1575 SND_SOC_DAPM_PGA_E("Late DAC2L Enable PGA", SND_SOC_NOPM, 0, 0, NULL, 0,
1576         late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1577 SND_SOC_DAPM_PGA_E("Late DAC2R Enable PGA", SND_SOC_NOPM, 0, 0, NULL, 0,
1578         late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1579 SND_SOC_DAPM_PGA_E("Direct Voice", SND_SOC_NOPM, 0, 0, NULL, 0,
1580         late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1581
1582 SND_SOC_DAPM_MIXER_E("SPKL", WM8994_POWER_MANAGEMENT_3, 8, 0,
1583                      left_speaker_mixer, ARRAY_SIZE(left_speaker_mixer),
1584                      late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1585 SND_SOC_DAPM_MIXER_E("SPKR", WM8994_POWER_MANAGEMENT_3, 9, 0,
1586                      right_speaker_mixer, ARRAY_SIZE(right_speaker_mixer),
1587                      late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1588 SND_SOC_DAPM_MUX_E("Left Headphone Mux", SND_SOC_NOPM, 0, 0, &wm_hubs_hpl_mux,
1589                    late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1590 SND_SOC_DAPM_MUX_E("Right Headphone Mux", SND_SOC_NOPM, 0, 0, &wm_hubs_hpr_mux,
1591                    late_enable_ev, SND_SOC_DAPM_PRE_PMU),
1592
1593 SND_SOC_DAPM_POST("Late Disable PGA", late_disable_ev)
1594 };
1595
1596 static const struct snd_soc_dapm_widget wm8994_lateclk_widgets[] = {
1597 SND_SOC_DAPM_SUPPLY("AIF1CLK", WM8994_AIF1_CLOCKING_1, 0, 0, aif1clk_ev,
1598                     SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMU |
1599                     SND_SOC_DAPM_PRE_PMD),
1600 SND_SOC_DAPM_SUPPLY("AIF2CLK", WM8994_AIF2_CLOCKING_1, 0, 0, aif2clk_ev,
1601                     SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMU |
1602                     SND_SOC_DAPM_PRE_PMD),
1603 SND_SOC_DAPM_PGA("Direct Voice", SND_SOC_NOPM, 0, 0, NULL, 0),
1604 SND_SOC_DAPM_MIXER("SPKL", WM8994_POWER_MANAGEMENT_3, 8, 0,
1605                    left_speaker_mixer, ARRAY_SIZE(left_speaker_mixer)),
1606 SND_SOC_DAPM_MIXER("SPKR", WM8994_POWER_MANAGEMENT_3, 9, 0,
1607                    right_speaker_mixer, ARRAY_SIZE(right_speaker_mixer)),
1608 SND_SOC_DAPM_MUX("Left Headphone Mux", SND_SOC_NOPM, 0, 0, &wm_hubs_hpl_mux),
1609 SND_SOC_DAPM_MUX("Right Headphone Mux", SND_SOC_NOPM, 0, 0, &wm_hubs_hpr_mux),
1610 };
1611
1612 static const struct snd_soc_dapm_widget wm8994_dac_revd_widgets[] = {
1613 SND_SOC_DAPM_DAC_E("DAC2L", NULL, SND_SOC_NOPM, 3, 0,
1614         dac_ev, SND_SOC_DAPM_PRE_PMU),
1615 SND_SOC_DAPM_DAC_E("DAC2R", NULL, SND_SOC_NOPM, 2, 0,
1616         dac_ev, SND_SOC_DAPM_PRE_PMU),
1617 SND_SOC_DAPM_DAC_E("DAC1L", NULL, SND_SOC_NOPM, 1, 0,
1618         dac_ev, SND_SOC_DAPM_PRE_PMU),
1619 SND_SOC_DAPM_DAC_E("DAC1R", NULL, SND_SOC_NOPM, 0, 0,
1620         dac_ev, SND_SOC_DAPM_PRE_PMU),
1621 };
1622
1623 static const struct snd_soc_dapm_widget wm8994_dac_widgets[] = {
1624 SND_SOC_DAPM_DAC("DAC2L", NULL, WM8994_POWER_MANAGEMENT_5, 3, 0),
1625 SND_SOC_DAPM_DAC("DAC2R", NULL, WM8994_POWER_MANAGEMENT_5, 2, 0),
1626 SND_SOC_DAPM_DAC("DAC1L", NULL, WM8994_POWER_MANAGEMENT_5, 1, 0),
1627 SND_SOC_DAPM_DAC("DAC1R", NULL, WM8994_POWER_MANAGEMENT_5, 0, 0),
1628 };
1629
1630 static const struct snd_soc_dapm_widget wm8994_adc_revd_widgets[] = {
1631 SND_SOC_DAPM_VIRT_MUX_E("ADCL Mux", WM8994_POWER_MANAGEMENT_4, 1, 0, &adcl_mux,
1632                         adc_mux_ev, SND_SOC_DAPM_PRE_PMU),
1633 SND_SOC_DAPM_VIRT_MUX_E("ADCR Mux", WM8994_POWER_MANAGEMENT_4, 0, 0, &adcr_mux,
1634                         adc_mux_ev, SND_SOC_DAPM_PRE_PMU),
1635 };
1636
1637 static const struct snd_soc_dapm_widget wm8994_adc_widgets[] = {
1638 SND_SOC_DAPM_VIRT_MUX("ADCL Mux", WM8994_POWER_MANAGEMENT_4, 1, 0, &adcl_mux),
1639 SND_SOC_DAPM_VIRT_MUX("ADCR Mux", WM8994_POWER_MANAGEMENT_4, 0, 0, &adcr_mux),
1640 };
1641
1642 static const struct snd_soc_dapm_widget wm8994_dapm_widgets[] = {
1643 SND_SOC_DAPM_INPUT("DMIC1DAT"),
1644 SND_SOC_DAPM_INPUT("DMIC2DAT"),
1645 SND_SOC_DAPM_INPUT("Clock"),
1646
1647 SND_SOC_DAPM_SUPPLY_S("MICBIAS Supply", 1, SND_SOC_NOPM, 0, 0, micbias_ev,
1648                       SND_SOC_DAPM_PRE_PMU),
1649 SND_SOC_DAPM_SUPPLY("VMID", SND_SOC_NOPM, 0, 0, vmid_event,
1650                     SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
1651
1652 SND_SOC_DAPM_SUPPLY("CLK_SYS", SND_SOC_NOPM, 0, 0, clk_sys_event,
1653                     SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMU |
1654                     SND_SOC_DAPM_PRE_PMD),
1655
1656 SND_SOC_DAPM_SUPPLY("DSP1CLK", SND_SOC_NOPM, 3, 0, NULL, 0),
1657 SND_SOC_DAPM_SUPPLY("DSP2CLK", SND_SOC_NOPM, 2, 0, NULL, 0),
1658 SND_SOC_DAPM_SUPPLY("DSPINTCLK", SND_SOC_NOPM, 1, 0, NULL, 0),
1659
1660 SND_SOC_DAPM_AIF_OUT("AIF1ADC1L", NULL,
1661                      0, SND_SOC_NOPM, 9, 0),
1662 SND_SOC_DAPM_AIF_OUT("AIF1ADC1R", NULL,
1663                      0, SND_SOC_NOPM, 8, 0),
1664 SND_SOC_DAPM_AIF_IN_E("AIF1DAC1L", NULL, 0,
1665                       SND_SOC_NOPM, 9, 0, wm8958_aif_ev,
1666                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
1667 SND_SOC_DAPM_AIF_IN_E("AIF1DAC1R", NULL, 0,
1668                       SND_SOC_NOPM, 8, 0, wm8958_aif_ev,
1669                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
1670
1671 SND_SOC_DAPM_AIF_OUT("AIF1ADC2L", NULL,
1672                      0, SND_SOC_NOPM, 11, 0),
1673 SND_SOC_DAPM_AIF_OUT("AIF1ADC2R", NULL,
1674                      0, SND_SOC_NOPM, 10, 0),
1675 SND_SOC_DAPM_AIF_IN_E("AIF1DAC2L", NULL, 0,
1676                       SND_SOC_NOPM, 11, 0, wm8958_aif_ev,
1677                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
1678 SND_SOC_DAPM_AIF_IN_E("AIF1DAC2R", NULL, 0,
1679                       SND_SOC_NOPM, 10, 0, wm8958_aif_ev,
1680                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
1681
1682 SND_SOC_DAPM_MIXER("AIF1ADC1L Mixer", SND_SOC_NOPM, 0, 0,
1683                    aif1adc1l_mix, ARRAY_SIZE(aif1adc1l_mix)),
1684 SND_SOC_DAPM_MIXER("AIF1ADC1R Mixer", SND_SOC_NOPM, 0, 0,
1685                    aif1adc1r_mix, ARRAY_SIZE(aif1adc1r_mix)),
1686
1687 SND_SOC_DAPM_MIXER("AIF1ADC2L Mixer", SND_SOC_NOPM, 0, 0,
1688                    aif1adc2l_mix, ARRAY_SIZE(aif1adc2l_mix)),
1689 SND_SOC_DAPM_MIXER("AIF1ADC2R Mixer", SND_SOC_NOPM, 0, 0,
1690                    aif1adc2r_mix, ARRAY_SIZE(aif1adc2r_mix)),
1691
1692 SND_SOC_DAPM_MIXER("AIF2DAC2L Mixer", SND_SOC_NOPM, 0, 0,
1693                    aif2dac2l_mix, ARRAY_SIZE(aif2dac2l_mix)),
1694 SND_SOC_DAPM_MIXER("AIF2DAC2R Mixer", SND_SOC_NOPM, 0, 0,
1695                    aif2dac2r_mix, ARRAY_SIZE(aif2dac2r_mix)),
1696
1697 SND_SOC_DAPM_MUX("Left Sidetone", SND_SOC_NOPM, 0, 0, &sidetone1_mux),
1698 SND_SOC_DAPM_MUX("Right Sidetone", SND_SOC_NOPM, 0, 0, &sidetone2_mux),
1699
1700 SND_SOC_DAPM_MIXER("DAC1L Mixer", SND_SOC_NOPM, 0, 0,
1701                    dac1l_mix, ARRAY_SIZE(dac1l_mix)),
1702 SND_SOC_DAPM_MIXER("DAC1R Mixer", SND_SOC_NOPM, 0, 0,
1703                    dac1r_mix, ARRAY_SIZE(dac1r_mix)),
1704
1705 SND_SOC_DAPM_AIF_OUT("AIF2ADCL", NULL, 0,
1706                      SND_SOC_NOPM, 13, 0),
1707 SND_SOC_DAPM_AIF_OUT("AIF2ADCR", NULL, 0,
1708                      SND_SOC_NOPM, 12, 0),
1709 SND_SOC_DAPM_AIF_IN_E("AIF2DACL", NULL, 0,
1710                       SND_SOC_NOPM, 13, 0, wm8958_aif_ev,
1711                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_PRE_PMD),
1712 SND_SOC_DAPM_AIF_IN_E("AIF2DACR", NULL, 0,
1713                       SND_SOC_NOPM, 12, 0, wm8958_aif_ev,
1714                       SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_PRE_PMD),
1715
1716 SND_SOC_DAPM_AIF_IN("AIF1DACDAT", NULL, 0, SND_SOC_NOPM, 0, 0),
1717 SND_SOC_DAPM_AIF_IN("AIF2DACDAT", NULL, 0, SND_SOC_NOPM, 0, 0),
1718 SND_SOC_DAPM_AIF_OUT("AIF1ADCDAT", NULL, 0, SND_SOC_NOPM, 0, 0),
1719 SND_SOC_DAPM_AIF_OUT("AIF2ADCDAT",  NULL, 0, SND_SOC_NOPM, 0, 0),
1720
1721 SND_SOC_DAPM_MUX("AIF1DAC Mux", SND_SOC_NOPM, 0, 0, &aif1dac_mux),
1722 SND_SOC_DAPM_MUX("AIF2DAC Mux", SND_SOC_NOPM, 0, 0, &aif2dac_mux),
1723 SND_SOC_DAPM_MUX("AIF2ADC Mux", SND_SOC_NOPM, 0, 0, &aif2adc_mux),
1724
1725 SND_SOC_DAPM_AIF_IN("AIF3DACDAT", NULL, 0, SND_SOC_NOPM, 0, 0),
1726 SND_SOC_DAPM_AIF_OUT("AIF3ADCDAT", NULL, 0, SND_SOC_NOPM, 0, 0),
1727
1728 SND_SOC_DAPM_SUPPLY("TOCLK", WM8994_CLOCKING_1, 4, 0, NULL, 0),
1729
1730 SND_SOC_DAPM_ADC("DMIC2L", NULL, WM8994_POWER_MANAGEMENT_4, 5, 0),
1731 SND_SOC_DAPM_ADC("DMIC2R", NULL, WM8994_POWER_MANAGEMENT_4, 4, 0),
1732 SND_SOC_DAPM_ADC("DMIC1L", NULL, WM8994_POWER_MANAGEMENT_4, 3, 0),
1733 SND_SOC_DAPM_ADC("DMIC1R", NULL, WM8994_POWER_MANAGEMENT_4, 2, 0),
1734
1735 /* Power is done with the muxes since the ADC power also controls the
1736  * downsampling chain, the chip will automatically manage the analogue
1737  * specific portions.
1738  */
1739 SND_SOC_DAPM_ADC("ADCL", NULL, SND_SOC_NOPM, 1, 0),
1740 SND_SOC_DAPM_ADC("ADCR", NULL, SND_SOC_NOPM, 0, 0),
1741
1742 SND_SOC_DAPM_POST("Debug log", post_ev),
1743 };
1744
1745 static const struct snd_soc_dapm_widget wm8994_specific_dapm_widgets[] = {
1746 SND_SOC_DAPM_MUX("AIF3ADC Mux", SND_SOC_NOPM, 0, 0, &wm8994_aif3adc_mux),
1747 };
1748
1749 static const struct snd_soc_dapm_widget wm8958_dapm_widgets[] = {
1750 SND_SOC_DAPM_SUPPLY("AIF3", WM8994_POWER_MANAGEMENT_6, 5, 1, NULL, 0),
1751 SND_SOC_DAPM_MUX("Mono PCM Out Mux", SND_SOC_NOPM, 0, 0, &mono_pcm_out_mux),
1752 SND_SOC_DAPM_MUX("AIF2DACL Mux", SND_SOC_NOPM, 0, 0, &aif2dacl_src_mux),
1753 SND_SOC_DAPM_MUX("AIF2DACR Mux", SND_SOC_NOPM, 0, 0, &aif2dacr_src_mux),
1754 SND_SOC_DAPM_MUX("AIF3ADC Mux", SND_SOC_NOPM, 0, 0, &wm8958_aif3adc_mux),
1755 };
1756
1757 static const struct snd_soc_dapm_route intercon[] = {
1758         { "CLK_SYS", NULL, "AIF1CLK", check_clk_sys },
1759         { "CLK_SYS", NULL, "AIF2CLK", check_clk_sys },
1760
1761         { "DSP1CLK", NULL, "CLK_SYS" },
1762         { "DSP2CLK", NULL, "CLK_SYS" },
1763         { "DSPINTCLK", NULL, "CLK_SYS" },
1764
1765         { "AIF1ADC1L", NULL, "AIF1CLK" },
1766         { "AIF1ADC1L", NULL, "DSP1CLK" },
1767         { "AIF1ADC1R", NULL, "AIF1CLK" },
1768         { "AIF1ADC1R", NULL, "DSP1CLK" },
1769         { "AIF1ADC1R", NULL, "DSPINTCLK" },
1770
1771         { "AIF1DAC1L", NULL, "AIF1CLK" },
1772         { "AIF1DAC1L", NULL, "DSP1CLK" },
1773         { "AIF1DAC1R", NULL, "AIF1CLK" },
1774         { "AIF1DAC1R", NULL, "DSP1CLK" },
1775         { "AIF1DAC1R", NULL, "DSPINTCLK" },
1776
1777         { "AIF1ADC2L", NULL, "AIF1CLK" },
1778         { "AIF1ADC2L", NULL, "DSP1CLK" },
1779         { "AIF1ADC2R", NULL, "AIF1CLK" },
1780         { "AIF1ADC2R", NULL, "DSP1CLK" },
1781         { "AIF1ADC2R", NULL, "DSPINTCLK" },
1782
1783         { "AIF1DAC2L", NULL, "AIF1CLK" },
1784         { "AIF1DAC2L", NULL, "DSP1CLK" },
1785         { "AIF1DAC2R", NULL, "AIF1CLK" },
1786         { "AIF1DAC2R", NULL, "DSP1CLK" },
1787         { "AIF1DAC2R", NULL, "DSPINTCLK" },
1788
1789         { "AIF2ADCL", NULL, "AIF2CLK" },
1790         { "AIF2ADCL", NULL, "DSP2CLK" },
1791         { "AIF2ADCR", NULL, "AIF2CLK" },
1792         { "AIF2ADCR", NULL, "DSP2CLK" },
1793         { "AIF2ADCR", NULL, "DSPINTCLK" },
1794
1795         { "AIF2DACL", NULL, "AIF2CLK" },
1796         { "AIF2DACL", NULL, "DSP2CLK" },
1797         { "AIF2DACR", NULL, "AIF2CLK" },
1798         { "AIF2DACR", NULL, "DSP2CLK" },
1799         { "AIF2DACR", NULL, "DSPINTCLK" },
1800
1801         { "DMIC1L", NULL, "DMIC1DAT" },
1802         { "DMIC1L", NULL, "CLK_SYS" },
1803         { "DMIC1R", NULL, "DMIC1DAT" },
1804         { "DMIC1R", NULL, "CLK_SYS" },
1805         { "DMIC2L", NULL, "DMIC2DAT" },
1806         { "DMIC2L", NULL, "CLK_SYS" },
1807         { "DMIC2R", NULL, "DMIC2DAT" },
1808         { "DMIC2R", NULL, "CLK_SYS" },
1809
1810         { "ADCL", NULL, "AIF1CLK" },
1811         { "ADCL", NULL, "DSP1CLK" },
1812         { "ADCL", NULL, "DSPINTCLK" },
1813
1814         { "ADCR", NULL, "AIF1CLK" },
1815         { "ADCR", NULL, "DSP1CLK" },
1816         { "ADCR", NULL, "DSPINTCLK" },
1817
1818         { "ADCL Mux", "ADC", "ADCL" },
1819         { "ADCL Mux", "DMIC", "DMIC1L" },
1820         { "ADCR Mux", "ADC", "ADCR" },
1821         { "ADCR Mux", "DMIC", "DMIC1R" },
1822
1823         { "DAC1L", NULL, "AIF1CLK" },
1824         { "DAC1L", NULL, "DSP1CLK" },
1825         { "DAC1L", NULL, "DSPINTCLK" },
1826
1827         { "DAC1R", NULL, "AIF1CLK" },
1828         { "DAC1R", NULL, "DSP1CLK" },
1829         { "DAC1R", NULL, "DSPINTCLK" },
1830
1831         { "DAC2L", NULL, "AIF2CLK" },
1832         { "DAC2L", NULL, "DSP2CLK" },
1833         { "DAC2L", NULL, "DSPINTCLK" },
1834
1835         { "DAC2R", NULL, "AIF2DACR" },
1836         { "DAC2R", NULL, "AIF2CLK" },
1837         { "DAC2R", NULL, "DSP2CLK" },
1838         { "DAC2R", NULL, "DSPINTCLK" },
1839
1840         { "TOCLK", NULL, "CLK_SYS" },
1841
1842         { "AIF1DACDAT", NULL, "AIF1 Playback" },
1843         { "AIF2DACDAT", NULL, "AIF2 Playback" },
1844         { "AIF3DACDAT", NULL, "AIF3 Playback" },
1845
1846         { "AIF1 Capture", NULL, "AIF1ADCDAT" },
1847         { "AIF2 Capture", NULL, "AIF2ADCDAT" },
1848         { "AIF3 Capture", NULL, "AIF3ADCDAT" },
1849
1850         /* AIF1 outputs */
1851         { "AIF1ADC1L", NULL, "AIF1ADC1L Mixer" },
1852         { "AIF1ADC1L Mixer", "ADC/DMIC Switch", "ADCL Mux" },
1853         { "AIF1ADC1L Mixer", "AIF2 Switch", "AIF2DACL" },
1854
1855         { "AIF1ADC1R", NULL, "AIF1ADC1R Mixer" },
1856         { "AIF1ADC1R Mixer", "ADC/DMIC Switch", "ADCR Mux" },
1857         { "AIF1ADC1R Mixer", "AIF2 Switch", "AIF2DACR" },
1858
1859         { "AIF1ADC2L", NULL, "AIF1ADC2L Mixer" },
1860         { "AIF1ADC2L Mixer", "DMIC Switch", "DMIC2L" },
1861         { "AIF1ADC2L Mixer", "AIF2 Switch", "AIF2DACL" },
1862
1863         { "AIF1ADC2R", NULL, "AIF1ADC2R Mixer" },
1864         { "AIF1ADC2R Mixer", "DMIC Switch", "DMIC2R" },
1865         { "AIF1ADC2R Mixer", "AIF2 Switch", "AIF2DACR" },
1866
1867         /* Pin level routing for AIF3 */
1868         { "AIF1DAC1L", NULL, "AIF1DAC Mux" },
1869         { "AIF1DAC1R", NULL, "AIF1DAC Mux" },
1870         { "AIF1DAC2L", NULL, "AIF1DAC Mux" },
1871         { "AIF1DAC2R", NULL, "AIF1DAC Mux" },
1872
1873         { "AIF1DAC Mux", "AIF1DACDAT", "AIF1DACDAT" },
1874         { "AIF1DAC Mux", "AIF3DACDAT", "AIF3DACDAT" },
1875         { "AIF2DAC Mux", "AIF2DACDAT", "AIF2DACDAT" },
1876         { "AIF2DAC Mux", "AIF3DACDAT", "AIF3DACDAT" },
1877         { "AIF2ADC Mux", "AIF2ADCDAT", "AIF2ADCL" },
1878         { "AIF2ADC Mux", "AIF2ADCDAT", "AIF2ADCR" },
1879         { "AIF2ADC Mux", "AIF3DACDAT", "AIF3ADCDAT" },
1880
1881         /* DAC1 inputs */
1882         { "DAC1L Mixer", "AIF2 Switch", "AIF2DACL" },
1883         { "DAC1L Mixer", "AIF1.2 Switch", "AIF1DAC2L" },
1884         { "DAC1L Mixer", "AIF1.1 Switch", "AIF1DAC1L" },
1885         { "DAC1L Mixer", "Left Sidetone Switch", "Left Sidetone" },
1886         { "DAC1L Mixer", "Right Sidetone Switch", "Right Sidetone" },
1887
1888         { "DAC1R Mixer", "AIF2 Switch", "AIF2DACR" },
1889         { "DAC1R Mixer", "AIF1.2 Switch", "AIF1DAC2R" },
1890         { "DAC1R Mixer", "AIF1.1 Switch", "AIF1DAC1R" },
1891         { "DAC1R Mixer", "Left Sidetone Switch", "Left Sidetone" },
1892         { "DAC1R Mixer", "Right Sidetone Switch", "Right Sidetone" },
1893
1894         /* DAC2/AIF2 outputs  */
1895         { "AIF2ADCL", NULL, "AIF2DAC2L Mixer" },
1896         { "AIF2DAC2L Mixer", "AIF2 Switch", "AIF2DACL" },
1897         { "AIF2DAC2L Mixer", "AIF1.2 Switch", "AIF1DAC2L" },
1898         { "AIF2DAC2L Mixer", "AIF1.1 Switch", "AIF1DAC1L" },
1899         { "AIF2DAC2L Mixer", "Left Sidetone Switch", "Left Sidetone" },
1900         { "AIF2DAC2L Mixer", "Right Sidetone Switch", "Right Sidetone" },
1901
1902         { "AIF2ADCR", NULL, "AIF2DAC2R Mixer" },
1903         { "AIF2DAC2R Mixer", "AIF2 Switch", "AIF2DACR" },
1904         { "AIF2DAC2R Mixer", "AIF1.2 Switch", "AIF1DAC2R" },
1905         { "AIF2DAC2R Mixer", "AIF1.1 Switch", "AIF1DAC1R" },
1906         { "AIF2DAC2R Mixer", "Left Sidetone Switch", "Left Sidetone" },
1907         { "AIF2DAC2R Mixer", "Right Sidetone Switch", "Right Sidetone" },
1908
1909         { "AIF1ADCDAT", NULL, "AIF1ADC1L" },
1910         { "AIF1ADCDAT", NULL, "AIF1ADC1R" },
1911         { "AIF1ADCDAT", NULL, "AIF1ADC2L" },
1912         { "AIF1ADCDAT", NULL, "AIF1ADC2R" },
1913
1914         { "AIF2ADCDAT", NULL, "AIF2ADC Mux" },
1915
1916         /* AIF3 output */
1917         { "AIF3ADCDAT", "AIF1ADCDAT", "AIF1ADC1L" },
1918         { "AIF3ADCDAT", "AIF1ADCDAT", "AIF1ADC1R" },
1919         { "AIF3ADCDAT", "AIF1ADCDAT", "AIF1ADC2L" },
1920         { "AIF3ADCDAT", "AIF1ADCDAT", "AIF1ADC2R" },
1921         { "AIF3ADCDAT", "AIF2ADCDAT", "AIF2ADCL" },
1922         { "AIF3ADCDAT", "AIF2ADCDAT", "AIF2ADCR" },
1923         { "AIF3ADCDAT", "AIF2DACDAT", "AIF2DACL" },
1924         { "AIF3ADCDAT", "AIF2DACDAT", "AIF2DACR" },
1925
1926         /* Sidetone */
1927         { "Left Sidetone", "ADC/DMIC1", "ADCL Mux" },
1928         { "Left Sidetone", "DMIC2", "DMIC2L" },
1929         { "Right Sidetone", "ADC/DMIC1", "ADCR Mux" },
1930         { "Right Sidetone", "DMIC2", "DMIC2R" },
1931
1932         /* Output stages */
1933         { "Left Output Mixer", "DAC Switch", "DAC1L" },
1934         { "Right Output Mixer", "DAC Switch", "DAC1R" },
1935
1936         { "SPKL", "DAC1 Switch", "DAC1L" },
1937         { "SPKL", "DAC2 Switch", "DAC2L" },
1938
1939         { "SPKR", "DAC1 Switch", "DAC1R" },
1940         { "SPKR", "DAC2 Switch", "DAC2R" },
1941
1942         { "Left Headphone Mux", "DAC", "DAC1L" },
1943         { "Right Headphone Mux", "DAC", "DAC1R" },
1944 };
1945
1946 static const struct snd_soc_dapm_route wm8994_lateclk_revd_intercon[] = {
1947         { "DAC1L", NULL, "Late DAC1L Enable PGA" },
1948         { "Late DAC1L Enable PGA", NULL, "DAC1L Mixer" },
1949         { "DAC1R", NULL, "Late DAC1R Enable PGA" },
1950         { "Late DAC1R Enable PGA", NULL, "DAC1R Mixer" },
1951         { "DAC2L", NULL, "Late DAC2L Enable PGA" },
1952         { "Late DAC2L Enable PGA", NULL, "AIF2DAC2L Mixer" },
1953         { "DAC2R", NULL, "Late DAC2R Enable PGA" },
1954         { "Late DAC2R Enable PGA", NULL, "AIF2DAC2R Mixer" }
1955 };
1956
1957 static const struct snd_soc_dapm_route wm8994_lateclk_intercon[] = {
1958         { "DAC1L", NULL, "DAC1L Mixer" },
1959         { "DAC1R", NULL, "DAC1R Mixer" },
1960         { "DAC2L", NULL, "AIF2DAC2L Mixer" },
1961         { "DAC2R", NULL, "AIF2DAC2R Mixer" },
1962 };
1963
1964 static const struct snd_soc_dapm_route wm8994_revd_intercon[] = {
1965         { "AIF1DACDAT", NULL, "AIF2DACDAT" },
1966         { "AIF2DACDAT", NULL, "AIF1DACDAT" },
1967         { "AIF1ADCDAT", NULL, "AIF2ADCDAT" },
1968         { "AIF2ADCDAT", NULL, "AIF1ADCDAT" },
1969         { "MICBIAS1", NULL, "CLK_SYS" },
1970         { "MICBIAS1", NULL, "MICBIAS Supply" },
1971         { "MICBIAS2", NULL, "CLK_SYS" },
1972         { "MICBIAS2", NULL, "MICBIAS Supply" },
1973 };
1974
1975 static const struct snd_soc_dapm_route wm8994_intercon[] = {
1976         { "AIF2DACL", NULL, "AIF2DAC Mux" },
1977         { "AIF2DACR", NULL, "AIF2DAC Mux" },
1978         { "MICBIAS1", NULL, "VMID" },
1979         { "MICBIAS2", NULL, "VMID" },
1980 };
1981
1982 static const struct snd_soc_dapm_route wm8958_intercon[] = {
1983         { "AIF2DACL", NULL, "AIF2DACL Mux" },
1984         { "AIF2DACR", NULL, "AIF2DACR Mux" },
1985
1986         { "AIF2DACL Mux", "AIF2", "AIF2DAC Mux" },
1987         { "AIF2DACL Mux", "AIF3", "AIF3DACDAT" },
1988         { "AIF2DACR Mux", "AIF2", "AIF2DAC Mux" },
1989         { "AIF2DACR Mux", "AIF3", "AIF3DACDAT" },
1990
1991         { "AIF3DACDAT", NULL, "AIF3" },
1992         { "AIF3ADCDAT", NULL, "AIF3" },
1993
1994         { "Mono PCM Out Mux", "AIF2ADCL", "AIF2ADCL" },
1995         { "Mono PCM Out Mux", "AIF2ADCR", "AIF2ADCR" },
1996
1997         { "AIF3ADC Mux", "Mono PCM", "Mono PCM Out Mux" },
1998 };
1999
2000 /* The size in bits of the FLL divide multiplied by 10
2001  * to allow rounding later */
2002 #define FIXED_FLL_SIZE ((1 << 16) * 10)
2003
2004 struct fll_div {
2005         u16 outdiv;
2006         u16 n;
2007         u16 k;
2008         u16 clk_ref_div;
2009         u16 fll_fratio;
2010 };
2011
2012 static int wm8994_get_fll_config(struct fll_div *fll,
2013                                  int freq_in, int freq_out)
2014 {
2015         u64 Kpart;
2016         unsigned int K, Ndiv, Nmod;
2017
2018         pr_debug("FLL input=%dHz, output=%dHz\n", freq_in, freq_out);
2019
2020         /* Scale the input frequency down to <= 13.5MHz */
2021         fll->clk_ref_div = 0;
2022         while (freq_in > 13500000) {
2023                 fll->clk_ref_div++;
2024                 freq_in /= 2;
2025
2026                 if (fll->clk_ref_div > 3)
2027                         return -EINVAL;
2028         }
2029         pr_debug("CLK_REF_DIV=%d, Fref=%dHz\n", fll->clk_ref_div, freq_in);
2030
2031         /* Scale the output to give 90MHz<=Fvco<=100MHz */
2032         fll->outdiv = 3;
2033         while (freq_out * (fll->outdiv + 1) < 90000000) {
2034                 fll->outdiv++;
2035                 if (fll->outdiv > 63)
2036                         return -EINVAL;
2037         }
2038         freq_out *= fll->outdiv + 1;
2039         pr_debug("OUTDIV=%d, Fvco=%dHz\n", fll->outdiv, freq_out);
2040
2041         if (freq_in > 1000000) {
2042                 fll->fll_fratio = 0;
2043         } else if (freq_in > 256000) {
2044                 fll->fll_fratio = 1;
2045                 freq_in *= 2;
2046         } else if (freq_in > 128000) {
2047                 fll->fll_fratio = 2;
2048                 freq_in *= 4;
2049         } else if (freq_in > 64000) {
2050                 fll->fll_fratio = 3;
2051                 freq_in *= 8;
2052         } else {
2053                 fll->fll_fratio = 4;
2054                 freq_in *= 16;
2055         }
2056         pr_debug("FLL_FRATIO=%d, Fref=%dHz\n", fll->fll_fratio, freq_in);
2057
2058         /* Now, calculate N.K */
2059         Ndiv = freq_out / freq_in;
2060
2061         fll->n = Ndiv;
2062         Nmod = freq_out % freq_in;
2063         pr_debug("Nmod=%d\n", Nmod);
2064
2065         /* Calculate fractional part - scale up so we can round. */
2066         Kpart = FIXED_FLL_SIZE * (long long)Nmod;
2067
2068         do_div(Kpart, freq_in);
2069
2070         K = Kpart & 0xFFFFFFFF;
2071
2072         if ((K % 10) >= 5)
2073                 K += 5;
2074
2075         /* Move down to proper range now rounding is done */
2076         fll->k = K / 10;
2077
2078         pr_debug("N=%x K=%x\n", fll->n, fll->k);
2079
2080         return 0;
2081 }
2082
2083 static int _wm8994_set_fll(struct snd_soc_codec *codec, int id, int src,
2084                           unsigned int freq_in, unsigned int freq_out)
2085 {
2086         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2087         struct wm8994 *control = wm8994->wm8994;
2088         int reg_offset, ret;
2089         struct fll_div fll;
2090         u16 reg, clk1, aif_reg, aif_src;
2091         unsigned long timeout;
2092         bool was_enabled;
2093
2094         switch (id) {
2095         case WM8994_FLL1:
2096                 reg_offset = 0;
2097                 id = 0;
2098                 aif_src = 0x10;
2099                 break;
2100         case WM8994_FLL2:
2101                 reg_offset = 0x20;
2102                 id = 1;
2103                 aif_src = 0x18;
2104                 break;
2105         default:
2106                 return -EINVAL;
2107         }
2108
2109         reg = snd_soc_read(codec, WM8994_FLL1_CONTROL_1 + reg_offset);
2110         was_enabled = reg & WM8994_FLL1_ENA;
2111
2112         switch (src) {
2113         case 0:
2114                 /* Allow no source specification when stopping */
2115                 if (freq_out)
2116                         return -EINVAL;
2117                 src = wm8994->fll[id].src;
2118                 break;
2119         case WM8994_FLL_SRC_MCLK1:
2120         case WM8994_FLL_SRC_MCLK2:
2121         case WM8994_FLL_SRC_LRCLK:
2122         case WM8994_FLL_SRC_BCLK:
2123                 break;
2124         case WM8994_FLL_SRC_INTERNAL:
2125                 freq_in = 12000000;
2126                 freq_out = 12000000;
2127                 break;
2128         default:
2129                 return -EINVAL;
2130         }
2131
2132         /* Are we changing anything? */
2133         if (wm8994->fll[id].src == src &&
2134             wm8994->fll[id].in == freq_in && wm8994->fll[id].out == freq_out)
2135                 return 0;
2136
2137         /* If we're stopping the FLL redo the old config - no
2138          * registers will actually be written but we avoid GCC flow
2139          * analysis bugs spewing warnings.
2140          */
2141         if (freq_out)
2142                 ret = wm8994_get_fll_config(&fll, freq_in, freq_out);
2143         else
2144                 ret = wm8994_get_fll_config(&fll, wm8994->fll[id].in,
2145                                             wm8994->fll[id].out);
2146         if (ret < 0)
2147                 return ret;
2148
2149         /* Make sure that we're not providing SYSCLK right now */
2150         clk1 = snd_soc_read(codec, WM8994_CLOCKING_1);
2151         if (clk1 & WM8994_SYSCLK_SRC)
2152                 aif_reg = WM8994_AIF2_CLOCKING_1;
2153         else
2154                 aif_reg = WM8994_AIF1_CLOCKING_1;
2155         reg = snd_soc_read(codec, aif_reg);
2156
2157         if ((reg & WM8994_AIF1CLK_ENA) &&
2158             (reg & WM8994_AIF1CLK_SRC_MASK) == aif_src) {
2159                 dev_err(codec->dev, "FLL%d is currently providing SYSCLK\n",
2160                         id + 1);
2161                 return -EBUSY;
2162         }
2163
2164         /* We always need to disable the FLL while reconfiguring */
2165         snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_1 + reg_offset,
2166                             WM8994_FLL1_ENA, 0);
2167
2168         if (wm8994->fll_byp && src == WM8994_FLL_SRC_BCLK &&
2169             freq_in == freq_out && freq_out) {
2170                 dev_dbg(codec->dev, "Bypassing FLL%d\n", id + 1);
2171                 snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_5 + reg_offset,
2172                                     WM8958_FLL1_BYP, WM8958_FLL1_BYP);
2173                 goto out;
2174         }
2175
2176         reg = (fll.outdiv << WM8994_FLL1_OUTDIV_SHIFT) |
2177                 (fll.fll_fratio << WM8994_FLL1_FRATIO_SHIFT);
2178         snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_2 + reg_offset,
2179                             WM8994_FLL1_OUTDIV_MASK |
2180                             WM8994_FLL1_FRATIO_MASK, reg);
2181
2182         snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_3 + reg_offset,
2183                             WM8994_FLL1_K_MASK, fll.k);
2184
2185         snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_4 + reg_offset,
2186                             WM8994_FLL1_N_MASK,
2187                             fll.n << WM8994_FLL1_N_SHIFT);
2188
2189         snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_5 + reg_offset,
2190                             WM8994_FLL1_FRC_NCO | WM8958_FLL1_BYP |
2191                             WM8994_FLL1_REFCLK_DIV_MASK |
2192                             WM8994_FLL1_REFCLK_SRC_MASK,
2193                             ((src == WM8994_FLL_SRC_INTERNAL)
2194                              << WM8994_FLL1_FRC_NCO_SHIFT) |
2195                             (fll.clk_ref_div << WM8994_FLL1_REFCLK_DIV_SHIFT) |
2196                             (src - 1));
2197
2198         /* Clear any pending completion from a previous failure */
2199         try_wait_for_completion(&wm8994->fll_locked[id]);
2200
2201         /* Enable (with fractional mode if required) */
2202         if (freq_out) {
2203                 /* Enable VMID if we need it */
2204                 if (!was_enabled) {
2205                         active_reference(codec);
2206
2207                         switch (control->type) {
2208                         case WM8994:
2209                                 vmid_reference(codec);
2210                                 break;
2211                         case WM8958:
2212                                 if (wm8994->revision < 1)
2213                                         vmid_reference(codec);
2214                                 break;
2215                         default:
2216                                 break;
2217                         }
2218                 }
2219
2220                 reg = WM8994_FLL1_ENA;
2221
2222                 if (fll.k)
2223                         reg |= WM8994_FLL1_FRAC;
2224                 if (src == WM8994_FLL_SRC_INTERNAL)
2225                         reg |= WM8994_FLL1_OSC_ENA;
2226
2227                 snd_soc_update_bits(codec, WM8994_FLL1_CONTROL_1 + reg_offset,
2228                                     WM8994_FLL1_ENA | WM8994_FLL1_OSC_ENA |
2229                                     WM8994_FLL1_FRAC, reg);
2230
2231                 if (wm8994->fll_locked_irq) {
2232                         timeout = wait_for_completion_timeout(&wm8994->fll_locked[id],
2233                                                               msecs_to_jiffies(10));
2234                         if (timeout == 0)
2235                                 dev_warn(codec->dev,
2236                                          "Timed out waiting for FLL lock\n");
2237                 } else {
2238                         msleep(5);
2239                 }
2240         } else {
2241                 if (was_enabled) {
2242                         switch (control->type) {
2243                         case WM8994:
2244                                 vmid_dereference(codec);
2245                                 break;
2246                         case WM8958:
2247                                 if (wm8994->revision < 1)
2248                                         vmid_dereference(codec);
2249                                 break;
2250                         default:
2251                                 break;
2252                         }
2253
2254                         active_dereference(codec);
2255                 }
2256         }
2257
2258 out:
2259         wm8994->fll[id].in = freq_in;
2260         wm8994->fll[id].out = freq_out;
2261         wm8994->fll[id].src = src;
2262
2263         configure_clock(codec);
2264
2265         /*
2266          * If SYSCLK will be less than 50kHz adjust AIFnCLK dividers
2267          * for detection.
2268          */
2269         if (max(wm8994->aifclk[0], wm8994->aifclk[1]) < 50000) {
2270                 dev_dbg(codec->dev, "Configuring AIFs for 128fs\n");
2271                 snd_soc_update_bits(codec, WM8994_AIF1_RATE,
2272                                     WM8994_AIF1CLK_RATE_MASK, 0x1);
2273                 snd_soc_update_bits(codec, WM8994_AIF2_RATE,
2274                                     WM8994_AIF2CLK_RATE_MASK, 0x1);
2275         }
2276
2277         return 0;
2278 }
2279
2280 static irqreturn_t wm8994_fll_locked_irq(int irq, void *data)
2281 {
2282         struct completion *completion = data;
2283
2284         complete(completion);
2285
2286         return IRQ_HANDLED;
2287 }
2288
2289 static int opclk_divs[] = { 10, 20, 30, 40, 55, 60, 80, 120, 160 };
2290
2291 static int wm8994_set_fll(struct snd_soc_dai *dai, int id, int src,
2292                           unsigned int freq_in, unsigned int freq_out)
2293 {
2294         return _wm8994_set_fll(dai->codec, id, src, freq_in, freq_out);
2295 }
2296
2297 static int wm8994_set_dai_sysclk(struct snd_soc_dai *dai,
2298                 int clk_id, unsigned int freq, int dir)
2299 {
2300         struct snd_soc_codec *codec = dai->codec;
2301         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2302         int i;
2303
2304         switch (dai->id) {
2305         case 1:
2306         case 2:
2307                 break;
2308
2309         default:
2310                 /* AIF3 shares clocking with AIF1/2 */
2311                 return -EINVAL;
2312         }
2313
2314         switch (clk_id) {
2315         case WM8994_SYSCLK_MCLK1:
2316                 wm8994->sysclk[dai->id - 1] = WM8994_SYSCLK_MCLK1;
2317                 wm8994->mclk[0] = freq;
2318                 dev_dbg(dai->dev, "AIF%d using MCLK1 at %uHz\n",
2319                         dai->id, freq);
2320                 break;
2321
2322         case WM8994_SYSCLK_MCLK2:
2323                 /* TODO: Set GPIO AF */
2324                 wm8994->sysclk[dai->id - 1] = WM8994_SYSCLK_MCLK2;
2325                 wm8994->mclk[1] = freq;
2326                 dev_dbg(dai->dev, "AIF%d using MCLK2 at %uHz\n",
2327                         dai->id, freq);
2328                 break;
2329
2330         case WM8994_SYSCLK_FLL1:
2331                 wm8994->sysclk[dai->id - 1] = WM8994_SYSCLK_FLL1;
2332                 dev_dbg(dai->dev, "AIF%d using FLL1\n", dai->id);
2333                 break;
2334
2335         case WM8994_SYSCLK_FLL2:
2336                 wm8994->sysclk[dai->id - 1] = WM8994_SYSCLK_FLL2;
2337                 dev_dbg(dai->dev, "AIF%d using FLL2\n", dai->id);
2338                 break;
2339
2340         case WM8994_SYSCLK_OPCLK:
2341                 /* Special case - a division (times 10) is given and
2342                  * no effect on main clocking.
2343                  */
2344                 if (freq) {
2345                         for (i = 0; i < ARRAY_SIZE(opclk_divs); i++)
2346                                 if (opclk_divs[i] == freq)
2347                                         break;
2348                         if (i == ARRAY_SIZE(opclk_divs))
2349                                 return -EINVAL;
2350                         snd_soc_update_bits(codec, WM8994_CLOCKING_2,
2351                                             WM8994_OPCLK_DIV_MASK, i);
2352                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_2,
2353                                             WM8994_OPCLK_ENA, WM8994_OPCLK_ENA);
2354                 } else {
2355                         snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_2,
2356                                             WM8994_OPCLK_ENA, 0);
2357                 }
2358
2359         default:
2360                 return -EINVAL;
2361         }
2362
2363         configure_clock(codec);
2364
2365         /*
2366          * If SYSCLK will be less than 50kHz adjust AIFnCLK dividers
2367          * for detection.
2368          */
2369         if (max(wm8994->aifclk[0], wm8994->aifclk[1]) < 50000) {
2370                 dev_dbg(codec->dev, "Configuring AIFs for 128fs\n");
2371                 snd_soc_update_bits(codec, WM8994_AIF1_RATE,
2372                                     WM8994_AIF1CLK_RATE_MASK, 0x1);
2373                 snd_soc_update_bits(codec, WM8994_AIF2_RATE,
2374                                     WM8994_AIF2CLK_RATE_MASK, 0x1);
2375         }
2376
2377         return 0;
2378 }
2379
2380 static int wm8994_set_bias_level(struct snd_soc_codec *codec,
2381                                  enum snd_soc_bias_level level)
2382 {
2383         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2384         struct wm8994 *control = wm8994->wm8994;
2385
2386         wm_hubs_set_bias_level(codec, level);
2387
2388         switch (level) {
2389         case SND_SOC_BIAS_ON:
2390                 break;
2391
2392         case SND_SOC_BIAS_PREPARE:
2393                 /* MICBIAS into regulating mode */
2394                 switch (control->type) {
2395                 case WM8958:
2396                 case WM1811:
2397                         snd_soc_update_bits(codec, WM8958_MICBIAS1,
2398                                             WM8958_MICB1_MODE, 0);
2399                         snd_soc_update_bits(codec, WM8958_MICBIAS2,
2400                                             WM8958_MICB2_MODE, 0);
2401                         break;
2402                 default:
2403                         break;
2404                 }
2405
2406                 if (codec->dapm.bias_level == SND_SOC_BIAS_STANDBY)
2407                         active_reference(codec);
2408                 break;
2409
2410         case SND_SOC_BIAS_STANDBY:
2411                 if (codec->dapm.bias_level == SND_SOC_BIAS_OFF) {
2412                         switch (control->type) {
2413                         case WM8958:
2414                                 if (wm8994->revision == 0) {
2415                                         /* Optimise performance for rev A */
2416                                         snd_soc_update_bits(codec,
2417                                                             WM8958_CHARGE_PUMP_2,
2418                                                             WM8958_CP_DISCH,
2419                                                             WM8958_CP_DISCH);
2420                                 }
2421                                 break;
2422
2423                         default:
2424                                 break;
2425                         }
2426
2427                         /* Discharge LINEOUT1 & 2 */
2428                         snd_soc_update_bits(codec, WM8994_ANTIPOP_1,
2429                                             WM8994_LINEOUT1_DISCH |
2430                                             WM8994_LINEOUT2_DISCH,
2431                                             WM8994_LINEOUT1_DISCH |
2432                                             WM8994_LINEOUT2_DISCH);
2433                 }
2434
2435                 if (codec->dapm.bias_level == SND_SOC_BIAS_PREPARE)
2436                         active_dereference(codec);
2437
2438                 /* MICBIAS into bypass mode on newer devices */
2439                 switch (control->type) {
2440                 case WM8958:
2441                 case WM1811:
2442                         snd_soc_update_bits(codec, WM8958_MICBIAS1,
2443                                             WM8958_MICB1_MODE,
2444                                             WM8958_MICB1_MODE);
2445                         snd_soc_update_bits(codec, WM8958_MICBIAS2,
2446                                             WM8958_MICB2_MODE,
2447                                             WM8958_MICB2_MODE);
2448                         break;
2449                 default:
2450                         break;
2451                 }
2452                 break;
2453
2454         case SND_SOC_BIAS_OFF:
2455                 if (codec->dapm.bias_level == SND_SOC_BIAS_STANDBY)
2456                         wm8994->cur_fw = NULL;
2457                 break;
2458         }
2459
2460         codec->dapm.bias_level = level;
2461
2462         return 0;
2463 }
2464
2465 int wm8994_vmid_mode(struct snd_soc_codec *codec, enum wm8994_vmid_mode mode)
2466 {
2467         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2468
2469         switch (mode) {
2470         case WM8994_VMID_NORMAL:
2471                 if (wm8994->hubs.lineout1_se) {
2472                         snd_soc_dapm_disable_pin(&codec->dapm,
2473                                                  "LINEOUT1N Driver");
2474                         snd_soc_dapm_disable_pin(&codec->dapm,
2475                                                  "LINEOUT1P Driver");
2476                 }
2477                 if (wm8994->hubs.lineout2_se) {
2478                         snd_soc_dapm_disable_pin(&codec->dapm,
2479                                                  "LINEOUT2N Driver");
2480                         snd_soc_dapm_disable_pin(&codec->dapm,
2481                                                  "LINEOUT2P Driver");
2482                 }
2483
2484                 /* Do the sync with the old mode to allow it to clean up */
2485                 snd_soc_dapm_sync(&codec->dapm);
2486                 wm8994->vmid_mode = mode;
2487                 break;
2488
2489         case WM8994_VMID_FORCE:
2490                 if (wm8994->hubs.lineout1_se) {
2491                         snd_soc_dapm_force_enable_pin(&codec->dapm,
2492                                                       "LINEOUT1N Driver");
2493                         snd_soc_dapm_force_enable_pin(&codec->dapm,
2494                                                       "LINEOUT1P Driver");
2495                 }
2496                 if (wm8994->hubs.lineout2_se) {
2497                         snd_soc_dapm_force_enable_pin(&codec->dapm,
2498                                                       "LINEOUT2N Driver");
2499                         snd_soc_dapm_force_enable_pin(&codec->dapm,
2500                                                       "LINEOUT2P Driver");
2501                 }
2502
2503                 wm8994->vmid_mode = mode;
2504                 snd_soc_dapm_sync(&codec->dapm);
2505                 break;
2506
2507         default:
2508                 return -EINVAL;
2509         }
2510
2511         return 0;
2512 }
2513
2514 static int wm8994_set_dai_fmt(struct snd_soc_dai *dai, unsigned int fmt)
2515 {
2516         struct snd_soc_codec *codec = dai->codec;
2517         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2518         struct wm8994 *control = wm8994->wm8994;
2519         int ms_reg;
2520         int aif1_reg;
2521         int ms = 0;
2522         int aif1 = 0;
2523
2524         switch (dai->id) {
2525         case 1:
2526                 ms_reg = WM8994_AIF1_MASTER_SLAVE;
2527                 aif1_reg = WM8994_AIF1_CONTROL_1;
2528                 break;
2529         case 2:
2530                 ms_reg = WM8994_AIF2_MASTER_SLAVE;
2531                 aif1_reg = WM8994_AIF2_CONTROL_1;
2532                 break;
2533         default:
2534                 return -EINVAL;
2535         }
2536
2537         switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
2538         case SND_SOC_DAIFMT_CBS_CFS:
2539                 break;
2540         case SND_SOC_DAIFMT_CBM_CFM:
2541                 ms = WM8994_AIF1_MSTR;
2542                 break;
2543         default:
2544                 return -EINVAL;
2545         }
2546
2547         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
2548         case SND_SOC_DAIFMT_DSP_B:
2549                 aif1 |= WM8994_AIF1_LRCLK_INV;
2550         case SND_SOC_DAIFMT_DSP_A:
2551                 aif1 |= 0x18;
2552                 break;
2553         case SND_SOC_DAIFMT_I2S:
2554                 aif1 |= 0x10;
2555                 break;
2556         case SND_SOC_DAIFMT_RIGHT_J:
2557                 break;
2558         case SND_SOC_DAIFMT_LEFT_J:
2559                 aif1 |= 0x8;
2560                 break;
2561         default:
2562                 return -EINVAL;
2563         }
2564
2565         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
2566         case SND_SOC_DAIFMT_DSP_A:
2567         case SND_SOC_DAIFMT_DSP_B:
2568                 /* frame inversion not valid for DSP modes */
2569                 switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
2570                 case SND_SOC_DAIFMT_NB_NF:
2571                         break;
2572                 case SND_SOC_DAIFMT_IB_NF:
2573                         aif1 |= WM8994_AIF1_BCLK_INV;
2574                         break;
2575                 default:
2576                         return -EINVAL;
2577                 }
2578                 break;
2579
2580         case SND_SOC_DAIFMT_I2S:
2581         case SND_SOC_DAIFMT_RIGHT_J:
2582         case SND_SOC_DAIFMT_LEFT_J:
2583                 switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
2584                 case SND_SOC_DAIFMT_NB_NF:
2585                         break;
2586                 case SND_SOC_DAIFMT_IB_IF:
2587                         aif1 |= WM8994_AIF1_BCLK_INV | WM8994_AIF1_LRCLK_INV;
2588                         break;
2589                 case SND_SOC_DAIFMT_IB_NF:
2590                         aif1 |= WM8994_AIF1_BCLK_INV;
2591                         break;
2592                 case SND_SOC_DAIFMT_NB_IF:
2593                         aif1 |= WM8994_AIF1_LRCLK_INV;
2594                         break;
2595                 default:
2596                         return -EINVAL;
2597                 }
2598                 break;
2599         default:
2600                 return -EINVAL;
2601         }
2602
2603         /* The AIF2 format configuration needs to be mirrored to AIF3
2604          * on WM8958 if it's in use so just do it all the time. */
2605         switch (control->type) {
2606         case WM1811:
2607         case WM8958:
2608                 if (dai->id == 2)
2609                         snd_soc_update_bits(codec, WM8958_AIF3_CONTROL_1,
2610                                             WM8994_AIF1_LRCLK_INV |
2611                                             WM8958_AIF3_FMT_MASK, aif1);
2612                 break;
2613
2614         default:
2615                 break;
2616         }
2617
2618         snd_soc_update_bits(codec, aif1_reg,
2619                             WM8994_AIF1_BCLK_INV | WM8994_AIF1_LRCLK_INV |
2620                             WM8994_AIF1_FMT_MASK,
2621                             aif1);
2622         snd_soc_update_bits(codec, ms_reg, WM8994_AIF1_MSTR,
2623                             ms);
2624
2625         return 0;
2626 }
2627
2628 static struct {
2629         int val, rate;
2630 } srs[] = {
2631         { 0,   8000 },
2632         { 1,  11025 },
2633         { 2,  12000 },
2634         { 3,  16000 },
2635         { 4,  22050 },
2636         { 5,  24000 },
2637         { 6,  32000 },
2638         { 7,  44100 },
2639         { 8,  48000 },
2640         { 9,  88200 },
2641         { 10, 96000 },
2642 };
2643
2644 static int fs_ratios[] = {
2645         64, 128, 192, 256, 348, 512, 768, 1024, 1408, 1536
2646 };
2647
2648 static int bclk_divs[] = {
2649         10, 15, 20, 30, 40, 50, 60, 80, 110, 120, 160, 220, 240, 320, 440, 480,
2650         640, 880, 960, 1280, 1760, 1920
2651 };
2652
2653 static int wm8994_hw_params(struct snd_pcm_substream *substream,
2654                             struct snd_pcm_hw_params *params,
2655                             struct snd_soc_dai *dai)
2656 {
2657         struct snd_soc_codec *codec = dai->codec;
2658         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2659         int aif1_reg;
2660         int aif2_reg;
2661         int bclk_reg;
2662         int lrclk_reg;
2663         int rate_reg;
2664         int aif1 = 0;
2665         int aif2 = 0;
2666         int bclk = 0;
2667         int lrclk = 0;
2668         int rate_val = 0;
2669         int id = dai->id - 1;
2670
2671         int i, cur_val, best_val, bclk_rate, best;
2672
2673         switch (dai->id) {
2674         case 1:
2675                 aif1_reg = WM8994_AIF1_CONTROL_1;
2676                 aif2_reg = WM8994_AIF1_CONTROL_2;
2677                 bclk_reg = WM8994_AIF1_BCLK;
2678                 rate_reg = WM8994_AIF1_RATE;
2679                 if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK ||
2680                     wm8994->lrclk_shared[0]) {
2681                         lrclk_reg = WM8994_AIF1DAC_LRCLK;
2682                 } else {
2683                         lrclk_reg = WM8994_AIF1ADC_LRCLK;
2684                         dev_dbg(codec->dev, "AIF1 using split LRCLK\n");
2685                 }
2686                 break;
2687         case 2:
2688                 aif1_reg = WM8994_AIF2_CONTROL_1;
2689                 aif2_reg = WM8994_AIF2_CONTROL_2;
2690                 bclk_reg = WM8994_AIF2_BCLK;
2691                 rate_reg = WM8994_AIF2_RATE;
2692                 if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK ||
2693                     wm8994->lrclk_shared[1]) {
2694                         lrclk_reg = WM8994_AIF2DAC_LRCLK;
2695                 } else {
2696                         lrclk_reg = WM8994_AIF2ADC_LRCLK;
2697                         dev_dbg(codec->dev, "AIF2 using split LRCLK\n");
2698                 }
2699                 break;
2700         default:
2701                 return -EINVAL;
2702         }
2703
2704         bclk_rate = params_rate(params);
2705         switch (params_format(params)) {
2706         case SNDRV_PCM_FORMAT_S16_LE:
2707                 bclk_rate *= 16;
2708                 break;
2709         case SNDRV_PCM_FORMAT_S20_3LE:
2710                 bclk_rate *= 20;
2711                 aif1 |= 0x20;
2712                 break;
2713         case SNDRV_PCM_FORMAT_S24_LE:
2714                 bclk_rate *= 24;
2715                 aif1 |= 0x40;
2716                 break;
2717         case SNDRV_PCM_FORMAT_S32_LE:
2718                 bclk_rate *= 32;
2719                 aif1 |= 0x60;
2720                 break;
2721         default:
2722                 return -EINVAL;
2723         }
2724
2725         wm8994->channels[id] = params_channels(params);
2726         switch (params_channels(params)) {
2727         case 1:
2728         case 2:
2729                 bclk_rate *= 2;
2730                 break;
2731         default:
2732                 bclk_rate *= 4;
2733                 break;
2734         }
2735
2736         /* Try to find an appropriate sample rate; look for an exact match. */
2737         for (i = 0; i < ARRAY_SIZE(srs); i++)
2738                 if (srs[i].rate == params_rate(params))
2739                         break;
2740         if (i == ARRAY_SIZE(srs))
2741                 return -EINVAL;
2742         rate_val |= srs[i].val << WM8994_AIF1_SR_SHIFT;
2743
2744         dev_dbg(dai->dev, "Sample rate is %dHz\n", srs[i].rate);
2745         dev_dbg(dai->dev, "AIF%dCLK is %dHz, target BCLK %dHz\n",
2746                 dai->id, wm8994->aifclk[id], bclk_rate);
2747
2748         if (params_channels(params) == 1 &&
2749             (snd_soc_read(codec, aif1_reg) & 0x18) == 0x18)
2750                 aif2 |= WM8994_AIF1_MONO;
2751
2752         if (wm8994->aifclk[id] == 0) {
2753                 dev_err(dai->dev, "AIF%dCLK not configured\n", dai->id);
2754                 return -EINVAL;
2755         }
2756
2757         /* AIFCLK/fs ratio; look for a close match in either direction */
2758         best = 0;
2759         best_val = abs((fs_ratios[0] * params_rate(params))
2760                        - wm8994->aifclk[id]);
2761         for (i = 1; i < ARRAY_SIZE(fs_ratios); i++) {
2762                 cur_val = abs((fs_ratios[i] * params_rate(params))
2763                               - wm8994->aifclk[id]);
2764                 if (cur_val >= best_val)
2765                         continue;
2766                 best = i;
2767                 best_val = cur_val;
2768         }
2769         dev_dbg(dai->dev, "Selected AIF%dCLK/fs = %d\n",
2770                 dai->id, fs_ratios[best]);
2771         rate_val |= best;
2772
2773         /* We may not get quite the right frequency if using
2774          * approximate clocks so look for the closest match that is
2775          * higher than the target (we need to ensure that there enough
2776          * BCLKs to clock out the samples).
2777          */
2778         best = 0;
2779         for (i = 0; i < ARRAY_SIZE(bclk_divs); i++) {
2780                 cur_val = (wm8994->aifclk[id] * 10 / bclk_divs[i]) - bclk_rate;
2781                 if (cur_val < 0) /* BCLK table is sorted */
2782                         break;
2783                 best = i;
2784         }
2785         bclk_rate = wm8994->aifclk[id] * 10 / bclk_divs[best];
2786         dev_dbg(dai->dev, "Using BCLK_DIV %d for actual BCLK %dHz\n",
2787                 bclk_divs[best], bclk_rate);
2788         bclk |= best << WM8994_AIF1_BCLK_DIV_SHIFT;
2789
2790         lrclk = bclk_rate / params_rate(params);
2791         if (!lrclk) {
2792                 dev_err(dai->dev, "Unable to generate LRCLK from %dHz BCLK\n",
2793                         bclk_rate);
2794                 return -EINVAL;
2795         }
2796         dev_dbg(dai->dev, "Using LRCLK rate %d for actual LRCLK %dHz\n",
2797                 lrclk, bclk_rate / lrclk);
2798
2799         snd_soc_update_bits(codec, aif1_reg, WM8994_AIF1_WL_MASK, aif1);
2800         snd_soc_update_bits(codec, aif2_reg, WM8994_AIF1_MONO, aif2);
2801         snd_soc_update_bits(codec, bclk_reg, WM8994_AIF1_BCLK_DIV_MASK, bclk);
2802         snd_soc_update_bits(codec, lrclk_reg, WM8994_AIF1DAC_RATE_MASK,
2803                             lrclk);
2804         snd_soc_update_bits(codec, rate_reg, WM8994_AIF1_SR_MASK |
2805                             WM8994_AIF1CLK_RATE_MASK, rate_val);
2806
2807         if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
2808                 switch (dai->id) {
2809                 case 1:
2810                         wm8994->dac_rates[0] = params_rate(params);
2811                         wm8994_set_retune_mobile(codec, 0);
2812                         wm8994_set_retune_mobile(codec, 1);
2813                         break;
2814                 case 2:
2815                         wm8994->dac_rates[1] = params_rate(params);
2816                         wm8994_set_retune_mobile(codec, 2);
2817                         break;
2818                 }
2819         }
2820
2821         return 0;
2822 }
2823
2824 static int wm8994_aif3_hw_params(struct snd_pcm_substream *substream,
2825                                  struct snd_pcm_hw_params *params,
2826                                  struct snd_soc_dai *dai)
2827 {
2828         struct snd_soc_codec *codec = dai->codec;
2829         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
2830         struct wm8994 *control = wm8994->wm8994;
2831         int aif1_reg;
2832         int aif1 = 0;
2833
2834         switch (dai->id) {
2835         case 3:
2836                 switch (control->type) {
2837                 case WM1811:
2838                 case WM8958:
2839                         aif1_reg = WM8958_AIF3_CONTROL_1;
2840                         break;
2841                 default:
2842                         return 0;
2843                 }
2844                 break;
2845         default:
2846                 return 0;
2847         }
2848
2849         switch (params_format(params)) {
2850         case SNDRV_PCM_FORMAT_S16_LE:
2851                 break;
2852         case SNDRV_PCM_FORMAT_S20_3LE:
2853                 aif1 |= 0x20;
2854                 break;
2855         case SNDRV_PCM_FORMAT_S24_LE:
2856                 aif1 |= 0x40;
2857                 break;
2858         case SNDRV_PCM_FORMAT_S32_LE:
2859                 aif1 |= 0x60;
2860                 break;
2861         default:
2862                 return -EINVAL;
2863         }
2864
2865         return snd_soc_update_bits(codec, aif1_reg, WM8994_AIF1_WL_MASK, aif1);
2866 }
2867
2868 static int wm8994_aif_mute(struct snd_soc_dai *codec_dai, int mute)
2869 {
2870         struct snd_soc_codec *codec = codec_dai->codec;
2871         int mute_reg;
2872         int reg;
2873
2874         switch (codec_dai->id) {
2875         case 1:
2876                 mute_reg = WM8994_AIF1_DAC1_FILTERS_1;
2877                 break;
2878         case 2:
2879                 mute_reg = WM8994_AIF2_DAC_FILTERS_1;
2880                 break;
2881         default:
2882                 return -EINVAL;
2883         }
2884
2885         if (mute)
2886                 reg = WM8994_AIF1DAC1_MUTE;
2887         else
2888                 reg = 0;
2889
2890         snd_soc_update_bits(codec, mute_reg, WM8994_AIF1DAC1_MUTE, reg);
2891
2892         return 0;
2893 }
2894
2895 static int wm8994_set_tristate(struct snd_soc_dai *codec_dai, int tristate)
2896 {
2897         struct snd_soc_codec *codec = codec_dai->codec;
2898         int reg, val, mask;
2899
2900         switch (codec_dai->id) {
2901         case 1:
2902                 reg = WM8994_AIF1_MASTER_SLAVE;
2903                 mask = WM8994_AIF1_TRI;
2904                 break;
2905         case 2:
2906                 reg = WM8994_AIF2_MASTER_SLAVE;
2907                 mask = WM8994_AIF2_TRI;
2908                 break;
2909         default:
2910                 return -EINVAL;
2911         }
2912
2913         if (tristate)
2914                 val = mask;
2915         else
2916                 val = 0;
2917
2918         return snd_soc_update_bits(codec, reg, mask, val);
2919 }
2920
2921 static int wm8994_aif2_probe(struct snd_soc_dai *dai)
2922 {
2923         struct snd_soc_codec *codec = dai->codec;
2924
2925         /* Disable the pulls on the AIF if we're using it to save power. */
2926         snd_soc_update_bits(codec, WM8994_GPIO_3,
2927                             WM8994_GPN_PU | WM8994_GPN_PD, 0);
2928         snd_soc_update_bits(codec, WM8994_GPIO_4,
2929                             WM8994_GPN_PU | WM8994_GPN_PD, 0);
2930         snd_soc_update_bits(codec, WM8994_GPIO_5,
2931                             WM8994_GPN_PU | WM8994_GPN_PD, 0);
2932
2933         return 0;
2934 }
2935
2936 #define WM8994_RATES SNDRV_PCM_RATE_8000_96000
2937
2938 #define WM8994_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
2939                         SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S32_LE)
2940
2941 static const struct snd_soc_dai_ops wm8994_aif1_dai_ops = {
2942         .set_sysclk     = wm8994_set_dai_sysclk,
2943         .set_fmt        = wm8994_set_dai_fmt,
2944         .hw_params      = wm8994_hw_params,
2945         .digital_mute   = wm8994_aif_mute,
2946         .set_pll        = wm8994_set_fll,
2947         .set_tristate   = wm8994_set_tristate,
2948 };
2949
2950 static const struct snd_soc_dai_ops wm8994_aif2_dai_ops = {
2951         .set_sysclk     = wm8994_set_dai_sysclk,
2952         .set_fmt        = wm8994_set_dai_fmt,
2953         .hw_params      = wm8994_hw_params,
2954         .digital_mute   = wm8994_aif_mute,
2955         .set_pll        = wm8994_set_fll,
2956         .set_tristate   = wm8994_set_tristate,
2957 };
2958
2959 static const struct snd_soc_dai_ops wm8994_aif3_dai_ops = {
2960         .hw_params      = wm8994_aif3_hw_params,
2961 };
2962
2963 static struct snd_soc_dai_driver wm8994_dai[] = {
2964         {
2965                 .name = "wm8994-aif1",
2966                 .id = 1,
2967                 .playback = {
2968                         .stream_name = "AIF1 Playback",
2969                         .channels_min = 1,
2970                         .channels_max = 2,
2971                         .rates = WM8994_RATES,
2972                         .formats = WM8994_FORMATS,
2973                         .sig_bits = 24,
2974                 },
2975                 .capture = {
2976                         .stream_name = "AIF1 Capture",
2977                         .channels_min = 1,
2978                         .channels_max = 2,
2979                         .rates = WM8994_RATES,
2980                         .formats = WM8994_FORMATS,
2981                         .sig_bits = 24,
2982                  },
2983                 .ops = &wm8994_aif1_dai_ops,
2984         },
2985         {
2986                 .name = "wm8994-aif2",
2987                 .id = 2,
2988                 .playback = {
2989                         .stream_name = "AIF2 Playback",
2990                         .channels_min = 1,
2991                         .channels_max = 2,
2992                         .rates = WM8994_RATES,
2993                         .formats = WM8994_FORMATS,
2994                         .sig_bits = 24,
2995                 },
2996                 .capture = {
2997                         .stream_name = "AIF2 Capture",
2998                         .channels_min = 1,
2999                         .channels_max = 2,
3000                         .rates = WM8994_RATES,
3001                         .formats = WM8994_FORMATS,
3002                         .sig_bits = 24,
3003                 },
3004                 .probe = wm8994_aif2_probe,
3005                 .ops = &wm8994_aif2_dai_ops,
3006         },
3007         {
3008                 .name = "wm8994-aif3",
3009                 .id = 3,
3010                 .playback = {
3011                         .stream_name = "AIF3 Playback",
3012                         .channels_min = 1,
3013                         .channels_max = 2,
3014                         .rates = WM8994_RATES,
3015                         .formats = WM8994_FORMATS,
3016                         .sig_bits = 24,
3017                 },
3018                 .capture = {
3019                         .stream_name = "AIF3 Capture",
3020                         .channels_min = 1,
3021                         .channels_max = 2,
3022                         .rates = WM8994_RATES,
3023                         .formats = WM8994_FORMATS,
3024                         .sig_bits = 24,
3025                  },
3026                 .ops = &wm8994_aif3_dai_ops,
3027         }
3028 };
3029
3030 #ifdef CONFIG_PM
3031 static int wm8994_codec_suspend(struct snd_soc_codec *codec)
3032 {
3033         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3034         int i, ret;
3035
3036         for (i = 0; i < ARRAY_SIZE(wm8994->fll); i++) {
3037                 memcpy(&wm8994->fll_suspend[i], &wm8994->fll[i],
3038                        sizeof(struct wm8994_fll_config));
3039                 ret = _wm8994_set_fll(codec, i + 1, 0, 0, 0);
3040                 if (ret < 0)
3041                         dev_warn(codec->dev, "Failed to stop FLL%d: %d\n",
3042                                  i + 1, ret);
3043         }
3044
3045         wm8994_set_bias_level(codec, SND_SOC_BIAS_OFF);
3046
3047         return 0;
3048 }
3049
3050 static int wm8994_codec_resume(struct snd_soc_codec *codec)
3051 {
3052         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3053         struct wm8994 *control = wm8994->wm8994;
3054         int i, ret;
3055         unsigned int val, mask;
3056
3057         if (wm8994->revision < 4) {
3058                 /* force a HW read */
3059                 ret = regmap_read(control->regmap,
3060                                   WM8994_POWER_MANAGEMENT_5, &val);
3061
3062                 /* modify the cache only */
3063                 codec->cache_only = 1;
3064                 mask =  WM8994_DAC1R_ENA | WM8994_DAC1L_ENA |
3065                         WM8994_DAC2R_ENA | WM8994_DAC2L_ENA;
3066                 val &= mask;
3067                 snd_soc_update_bits(codec, WM8994_POWER_MANAGEMENT_5,
3068                                     mask, val);
3069                 codec->cache_only = 0;
3070         }
3071
3072         for (i = 0; i < ARRAY_SIZE(wm8994->fll); i++) {
3073                 if (!wm8994->fll_suspend[i].out)
3074                         continue;
3075
3076                 ret = _wm8994_set_fll(codec, i + 1,
3077                                      wm8994->fll_suspend[i].src,
3078                                      wm8994->fll_suspend[i].in,
3079                                      wm8994->fll_suspend[i].out);
3080                 if (ret < 0)
3081                         dev_warn(codec->dev, "Failed to restore FLL%d: %d\n",
3082                                  i + 1, ret);
3083         }
3084
3085         return 0;
3086 }
3087 #else
3088 #define wm8994_codec_suspend NULL
3089 #define wm8994_codec_resume NULL
3090 #endif
3091
3092 static void wm8994_handle_retune_mobile_pdata(struct wm8994_priv *wm8994)
3093 {
3094         struct snd_soc_codec *codec = wm8994->hubs.codec;
3095         struct wm8994 *control = wm8994->wm8994;
3096         struct wm8994_pdata *pdata = &control->pdata;
3097         struct snd_kcontrol_new controls[] = {
3098                 SOC_ENUM_EXT("AIF1.1 EQ Mode",
3099                              wm8994->retune_mobile_enum,
3100                              wm8994_get_retune_mobile_enum,
3101                              wm8994_put_retune_mobile_enum),
3102                 SOC_ENUM_EXT("AIF1.2 EQ Mode",
3103                              wm8994->retune_mobile_enum,
3104                              wm8994_get_retune_mobile_enum,
3105                              wm8994_put_retune_mobile_enum),
3106                 SOC_ENUM_EXT("AIF2 EQ Mode",
3107                              wm8994->retune_mobile_enum,
3108                              wm8994_get_retune_mobile_enum,
3109                              wm8994_put_retune_mobile_enum),
3110         };
3111         int ret, i, j;
3112         const char **t;
3113
3114         /* We need an array of texts for the enum API but the number
3115          * of texts is likely to be less than the number of
3116          * configurations due to the sample rate dependency of the
3117          * configurations. */
3118         wm8994->num_retune_mobile_texts = 0;
3119         wm8994->retune_mobile_texts = NULL;
3120         for (i = 0; i < pdata->num_retune_mobile_cfgs; i++) {
3121                 for (j = 0; j < wm8994->num_retune_mobile_texts; j++) {
3122                         if (strcmp(pdata->retune_mobile_cfgs[i].name,
3123                                    wm8994->retune_mobile_texts[j]) == 0)
3124                                 break;
3125                 }
3126
3127                 if (j != wm8994->num_retune_mobile_texts)
3128                         continue;
3129
3130                 /* Expand the array... */
3131                 t = krealloc(wm8994->retune_mobile_texts,
3132                              sizeof(char *) *
3133                              (wm8994->num_retune_mobile_texts + 1),
3134                              GFP_KERNEL);
3135                 if (t == NULL)
3136                         continue;
3137
3138                 /* ...store the new entry... */
3139                 t[wm8994->num_retune_mobile_texts] =
3140                         pdata->retune_mobile_cfgs[i].name;
3141
3142                 /* ...and remember the new version. */
3143                 wm8994->num_retune_mobile_texts++;
3144                 wm8994->retune_mobile_texts = t;
3145         }
3146
3147         dev_dbg(codec->dev, "Allocated %d unique ReTune Mobile names\n",
3148                 wm8994->num_retune_mobile_texts);
3149
3150         wm8994->retune_mobile_enum.max = wm8994->num_retune_mobile_texts;
3151         wm8994->retune_mobile_enum.texts = wm8994->retune_mobile_texts;
3152
3153         ret = snd_soc_add_codec_controls(wm8994->hubs.codec, controls,
3154                                    ARRAY_SIZE(controls));
3155         if (ret != 0)
3156                 dev_err(wm8994->hubs.codec->dev,
3157                         "Failed to add ReTune Mobile controls: %d\n", ret);
3158 }
3159
3160 static void wm8994_handle_pdata(struct wm8994_priv *wm8994)
3161 {
3162         struct snd_soc_codec *codec = wm8994->hubs.codec;
3163         struct wm8994 *control = wm8994->wm8994;
3164         struct wm8994_pdata *pdata = &control->pdata;
3165         int ret, i;
3166
3167         if (!pdata)
3168                 return;
3169
3170         wm_hubs_handle_analogue_pdata(codec, pdata->lineout1_diff,
3171                                       pdata->lineout2_diff,
3172                                       pdata->lineout1fb,
3173                                       pdata->lineout2fb,
3174                                       pdata->jd_scthr,
3175                                       pdata->jd_thr,
3176                                       pdata->micb1_delay,
3177                                       pdata->micb2_delay,
3178                                       pdata->micbias1_lvl,
3179                                       pdata->micbias2_lvl);
3180
3181         dev_dbg(codec->dev, "%d DRC configurations\n", pdata->num_drc_cfgs);
3182
3183         if (pdata->num_drc_cfgs) {
3184                 struct snd_kcontrol_new controls[] = {
3185                         SOC_ENUM_EXT("AIF1DRC1 Mode", wm8994->drc_enum,
3186                                      wm8994_get_drc_enum, wm8994_put_drc_enum),
3187                         SOC_ENUM_EXT("AIF1DRC2 Mode", wm8994->drc_enum,
3188                                      wm8994_get_drc_enum, wm8994_put_drc_enum),
3189                         SOC_ENUM_EXT("AIF2DRC Mode", wm8994->drc_enum,
3190                                      wm8994_get_drc_enum, wm8994_put_drc_enum),
3191                 };
3192
3193                 /* We need an array of texts for the enum API */
3194                 wm8994->drc_texts = devm_kzalloc(wm8994->hubs.codec->dev,
3195                             sizeof(char *) * pdata->num_drc_cfgs, GFP_KERNEL);
3196                 if (!wm8994->drc_texts) {
3197                         dev_err(wm8994->hubs.codec->dev,
3198                                 "Failed to allocate %d DRC config texts\n",
3199                                 pdata->num_drc_cfgs);
3200                         return;
3201                 }
3202
3203                 for (i = 0; i < pdata->num_drc_cfgs; i++)
3204                         wm8994->drc_texts[i] = pdata->drc_cfgs[i].name;
3205
3206                 wm8994->drc_enum.max = pdata->num_drc_cfgs;
3207                 wm8994->drc_enum.texts = wm8994->drc_texts;
3208
3209                 ret = snd_soc_add_codec_controls(wm8994->hubs.codec, controls,
3210                                            ARRAY_SIZE(controls));
3211                 for (i = 0; i < WM8994_NUM_DRC; i++)
3212                         wm8994_set_drc(codec, i);
3213         } else {
3214                 ret = snd_soc_add_codec_controls(wm8994->hubs.codec,
3215                                                  wm8994_drc_controls,
3216                                                  ARRAY_SIZE(wm8994_drc_controls));
3217         }
3218
3219         if (ret != 0)
3220                 dev_err(wm8994->hubs.codec->dev,
3221                         "Failed to add DRC mode controls: %d\n", ret);
3222
3223
3224         dev_dbg(codec->dev, "%d ReTune Mobile configurations\n",
3225                 pdata->num_retune_mobile_cfgs);
3226
3227         if (pdata->num_retune_mobile_cfgs)
3228                 wm8994_handle_retune_mobile_pdata(wm8994);
3229         else
3230                 snd_soc_add_codec_controls(wm8994->hubs.codec, wm8994_eq_controls,
3231                                      ARRAY_SIZE(wm8994_eq_controls));
3232
3233         for (i = 0; i < ARRAY_SIZE(pdata->micbias); i++) {
3234                 if (pdata->micbias[i]) {
3235                         snd_soc_write(codec, WM8958_MICBIAS1 + i,
3236                                 pdata->micbias[i] & 0xffff);
3237                 }
3238         }
3239 }
3240
3241 /**
3242  * wm8994_mic_detect - Enable microphone detection via the WM8994 IRQ
3243  *
3244  * @codec:   WM8994 codec
3245  * @jack:    jack to report detection events on
3246  * @micbias: microphone bias to detect on
3247  *
3248  * Enable microphone detection via IRQ on the WM8994.  If GPIOs are
3249  * being used to bring out signals to the processor then only platform
3250  * data configuration is needed for WM8994 and processor GPIOs should
3251  * be configured using snd_soc_jack_add_gpios() instead.
3252  *
3253  * Configuration of detection levels is available via the micbias1_lvl
3254  * and micbias2_lvl platform data members.
3255  */
3256 int wm8994_mic_detect(struct snd_soc_codec *codec, struct snd_soc_jack *jack,
3257                       int micbias)
3258 {
3259         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3260         struct wm8994_micdet *micdet;
3261         struct wm8994 *control = wm8994->wm8994;
3262         int reg, ret;
3263
3264         if (control->type != WM8994) {
3265                 dev_warn(codec->dev, "Not a WM8994\n");
3266                 return -EINVAL;
3267         }
3268
3269         switch (micbias) {
3270         case 1:
3271                 micdet = &wm8994->micdet[0];
3272                 if (jack)
3273                         ret = snd_soc_dapm_force_enable_pin(&codec->dapm,
3274                                                             "MICBIAS1");
3275                 else
3276                         ret = snd_soc_dapm_disable_pin(&codec->dapm,
3277                                                        "MICBIAS1");
3278                 break;
3279         case 2:
3280                 micdet = &wm8994->micdet[1];
3281                 if (jack)
3282                         ret = snd_soc_dapm_force_enable_pin(&codec->dapm,
3283                                                             "MICBIAS1");
3284                 else
3285                         ret = snd_soc_dapm_disable_pin(&codec->dapm,
3286                                                        "MICBIAS1");
3287                 break;
3288         default:
3289                 dev_warn(codec->dev, "Invalid MICBIAS %d\n", micbias);
3290                 return -EINVAL;
3291         }
3292
3293         if (ret != 0)
3294                 dev_warn(codec->dev, "Failed to configure MICBIAS%d: %d\n",
3295                          micbias, ret);
3296
3297         dev_dbg(codec->dev, "Configuring microphone detection on %d %p\n",
3298                 micbias, jack);
3299
3300         /* Store the configuration */
3301         micdet->jack = jack;
3302         micdet->detecting = true;
3303
3304         /* If either of the jacks is set up then enable detection */
3305         if (wm8994->micdet[0].jack || wm8994->micdet[1].jack)
3306                 reg = WM8994_MICD_ENA;
3307         else
3308                 reg = 0;
3309
3310         snd_soc_update_bits(codec, WM8994_MICBIAS, WM8994_MICD_ENA, reg);
3311
3312         /* enable MICDET and MICSHRT deboune */
3313         snd_soc_update_bits(codec, WM8994_IRQ_DEBOUNCE,
3314                             WM8994_MIC1_DET_DB_MASK | WM8994_MIC1_SHRT_DB_MASK |
3315                             WM8994_MIC2_DET_DB_MASK | WM8994_MIC2_SHRT_DB_MASK,
3316                             WM8994_MIC1_DET_DB | WM8994_MIC1_SHRT_DB);
3317
3318         snd_soc_dapm_sync(&codec->dapm);
3319
3320         return 0;
3321 }
3322 EXPORT_SYMBOL_GPL(wm8994_mic_detect);
3323
3324 static void wm8994_mic_work(struct work_struct *work)
3325 {
3326         struct wm8994_priv *priv = container_of(work,
3327                                                 struct wm8994_priv,
3328                                                 mic_work.work);
3329         struct regmap *regmap = priv->wm8994->regmap;
3330         struct device *dev = priv->wm8994->dev;
3331         unsigned int reg;
3332         int ret;
3333         int report;
3334
3335         pm_runtime_get_sync(dev);
3336
3337         ret = regmap_read(regmap, WM8994_INTERRUPT_RAW_STATUS_2, &reg);
3338         if (ret < 0) {
3339                 dev_err(dev, "Failed to read microphone status: %d\n",
3340                         ret);
3341                 pm_runtime_put(dev);
3342                 return;
3343         }
3344
3345         dev_dbg(dev, "Microphone status: %x\n", reg);
3346
3347         report = 0;
3348         if (reg & WM8994_MIC1_DET_STS) {
3349                 if (priv->micdet[0].detecting)
3350                         report = SND_JACK_HEADSET;
3351         }
3352         if (reg & WM8994_MIC1_SHRT_STS) {
3353                 if (priv->micdet[0].detecting)
3354                         report = SND_JACK_HEADPHONE;
3355                 else
3356                         report |= SND_JACK_BTN_0;
3357         }
3358         if (report)
3359                 priv->micdet[0].detecting = false;
3360         else
3361                 priv->micdet[0].detecting = true;
3362
3363         snd_soc_jack_report(priv->micdet[0].jack, report,
3364                             SND_JACK_HEADSET | SND_JACK_BTN_0);
3365
3366         report = 0;
3367         if (reg & WM8994_MIC2_DET_STS) {
3368                 if (priv->micdet[1].detecting)
3369                         report = SND_JACK_HEADSET;
3370         }
3371         if (reg & WM8994_MIC2_SHRT_STS) {
3372                 if (priv->micdet[1].detecting)
3373                         report = SND_JACK_HEADPHONE;
3374                 else
3375                         report |= SND_JACK_BTN_0;
3376         }
3377         if (report)
3378                 priv->micdet[1].detecting = false;
3379         else
3380                 priv->micdet[1].detecting = true;
3381
3382         snd_soc_jack_report(priv->micdet[1].jack, report,
3383                             SND_JACK_HEADSET | SND_JACK_BTN_0);
3384
3385         pm_runtime_put(dev);
3386 }
3387
3388 static irqreturn_t wm8994_mic_irq(int irq, void *data)
3389 {
3390         struct wm8994_priv *priv = data;
3391         struct snd_soc_codec *codec = priv->hubs.codec;
3392
3393 #ifndef CONFIG_SND_SOC_WM8994_MODULE
3394         trace_snd_soc_jack_irq(dev_name(codec->dev));
3395 #endif
3396
3397         pm_wakeup_event(codec->dev, 300);
3398
3399         schedule_delayed_work(&priv->mic_work, msecs_to_jiffies(250));
3400
3401         return IRQ_HANDLED;
3402 }
3403
3404 static void wm1811_micd_stop(struct snd_soc_codec *codec)
3405 {
3406         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3407
3408         if (!wm8994->jackdet)
3409                 return;
3410
3411         mutex_lock(&wm8994->accdet_lock);
3412
3413         snd_soc_update_bits(codec, WM8958_MIC_DETECT_1, WM8958_MICD_ENA, 0);
3414
3415         wm1811_jackdet_set_mode(codec, WM1811_JACKDET_MODE_JACK);
3416
3417         mutex_unlock(&wm8994->accdet_lock);
3418
3419         if (wm8994->wm8994->pdata.jd_ext_cap)
3420                 snd_soc_dapm_disable_pin(&codec->dapm,
3421                                          "MICBIAS2");
3422 }
3423
3424 static void wm8958_button_det(struct snd_soc_codec *codec, u16 status)
3425 {
3426         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3427         int report;
3428
3429         report = 0;
3430         if (status & 0x4)
3431                 report |= SND_JACK_BTN_0;
3432
3433         if (status & 0x8)
3434                 report |= SND_JACK_BTN_1;
3435
3436         if (status & 0x10)
3437                 report |= SND_JACK_BTN_2;
3438
3439         if (status & 0x20)
3440                 report |= SND_JACK_BTN_3;
3441
3442         if (status & 0x40)
3443                 report |= SND_JACK_BTN_4;
3444
3445         if (status & 0x80)
3446                 report |= SND_JACK_BTN_5;
3447
3448         snd_soc_jack_report(wm8994->micdet[0].jack, report,
3449                             wm8994->btn_mask);
3450 }
3451
3452 static void wm8958_mic_id(void *data, u16 status)
3453 {
3454         struct snd_soc_codec *codec = data;
3455         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3456
3457         /* Either nothing present or just starting detection */
3458         if (!(status & WM8958_MICD_STS)) {
3459                 /* If nothing present then clear our statuses */
3460                 dev_dbg(codec->dev, "Detected open circuit\n");
3461                 wm8994->jack_mic = false;
3462                 wm8994->mic_detecting = true;
3463
3464                 wm1811_micd_stop(codec);
3465
3466                 wm8958_micd_set_rate(codec);
3467
3468                 snd_soc_jack_report(wm8994->micdet[0].jack, 0,
3469                                     wm8994->btn_mask |
3470                                     SND_JACK_HEADSET);
3471                 return;
3472         }
3473
3474         /* If the measurement is showing a high impedence we've got a
3475          * microphone.
3476          */
3477         if (status & 0x600) {
3478                 dev_dbg(codec->dev, "Detected microphone\n");
3479
3480                 wm8994->mic_detecting = false;
3481                 wm8994->jack_mic = true;
3482
3483                 wm8958_micd_set_rate(codec);
3484
3485                 snd_soc_jack_report(wm8994->micdet[0].jack, SND_JACK_HEADSET,
3486                                     SND_JACK_HEADSET);
3487         }
3488
3489
3490         if (status & 0xfc) {
3491                 dev_dbg(codec->dev, "Detected headphone\n");
3492                 wm8994->mic_detecting = false;
3493
3494                 wm8958_micd_set_rate(codec);
3495
3496                 /* If we have jackdet that will detect removal */
3497                 wm1811_micd_stop(codec);
3498
3499                 snd_soc_jack_report(wm8994->micdet[0].jack, SND_JACK_HEADPHONE,
3500                                     SND_JACK_HEADSET);
3501         }
3502 }
3503
3504 /* Deferred mic detection to allow for extra settling time */
3505 static void wm1811_mic_work(struct work_struct *work)
3506 {
3507         struct wm8994_priv *wm8994 = container_of(work, struct wm8994_priv,
3508                                                   mic_work.work);
3509         struct wm8994 *control = wm8994->wm8994;
3510         struct snd_soc_codec *codec = wm8994->hubs.codec;
3511
3512         pm_runtime_get_sync(codec->dev);
3513
3514         /* If required for an external cap force MICBIAS on */
3515         if (control->pdata.jd_ext_cap) {
3516                 snd_soc_dapm_force_enable_pin(&codec->dapm,
3517                                               "MICBIAS2");
3518                 snd_soc_dapm_sync(&codec->dapm);
3519         }
3520
3521         mutex_lock(&wm8994->accdet_lock);
3522
3523         dev_dbg(codec->dev, "Starting mic detection\n");
3524
3525         /* Use a user-supplied callback if we have one */
3526         if (wm8994->micd_cb) {
3527                 wm8994->micd_cb(wm8994->micd_cb_data);
3528         } else {
3529                 /*
3530                  * Start off measument of microphone impedence to find out
3531                  * what's actually there.
3532                  */
3533                 wm8994->mic_detecting = true;
3534                 wm1811_jackdet_set_mode(codec, WM1811_JACKDET_MODE_MIC);
3535
3536                 snd_soc_update_bits(codec, WM8958_MIC_DETECT_1,
3537                                     WM8958_MICD_ENA, WM8958_MICD_ENA);
3538         }
3539
3540         mutex_unlock(&wm8994->accdet_lock);
3541
3542         pm_runtime_put(codec->dev);
3543 }
3544
3545 static irqreturn_t wm1811_jackdet_irq(int irq, void *data)
3546 {
3547         struct wm8994_priv *wm8994 = data;
3548         struct wm8994 *control = wm8994->wm8994;
3549         struct snd_soc_codec *codec = wm8994->hubs.codec;
3550         int reg, delay;
3551         bool present;
3552
3553         pm_runtime_get_sync(codec->dev);
3554
3555         mutex_lock(&wm8994->accdet_lock);
3556
3557         reg = snd_soc_read(codec, WM1811_JACKDET_CTRL);
3558         if (reg < 0) {
3559                 dev_err(codec->dev, "Failed to read jack status: %d\n", reg);
3560                 mutex_unlock(&wm8994->accdet_lock);
3561                 pm_runtime_put(codec->dev);
3562                 return IRQ_NONE;
3563         }
3564
3565         dev_dbg(codec->dev, "JACKDET %x\n", reg);
3566
3567         present = reg & WM1811_JACKDET_LVL;
3568
3569         if (present) {
3570                 dev_dbg(codec->dev, "Jack detected\n");
3571
3572                 wm8958_micd_set_rate(codec);
3573
3574                 snd_soc_update_bits(codec, WM8958_MICBIAS2,
3575                                     WM8958_MICB2_DISCH, 0);
3576
3577                 /* Disable debounce while inserted */
3578                 snd_soc_update_bits(codec, WM1811_JACKDET_CTRL,
3579                                     WM1811_JACKDET_DB, 0);
3580
3581                 delay = control->pdata.micdet_delay;
3582                 schedule_delayed_work(&wm8994->mic_work,
3583                                       msecs_to_jiffies(delay));
3584         } else {
3585                 dev_dbg(codec->dev, "Jack not detected\n");
3586
3587                 cancel_delayed_work_sync(&wm8994->mic_work);
3588
3589                 snd_soc_update_bits(codec, WM8958_MICBIAS2,
3590                                     WM8958_MICB2_DISCH, WM8958_MICB2_DISCH);
3591
3592                 /* Enable debounce while removed */
3593                 snd_soc_update_bits(codec, WM1811_JACKDET_CTRL,
3594                                     WM1811_JACKDET_DB, WM1811_JACKDET_DB);
3595
3596                 wm8994->mic_detecting = false;
3597                 wm8994->jack_mic = false;
3598                 snd_soc_update_bits(codec, WM8958_MIC_DETECT_1,
3599                                     WM8958_MICD_ENA, 0);
3600                 wm1811_jackdet_set_mode(codec, WM1811_JACKDET_MODE_JACK);
3601         }
3602
3603         mutex_unlock(&wm8994->accdet_lock);
3604
3605         /* Turn off MICBIAS if it was on for an external cap */
3606         if (control->pdata.jd_ext_cap && !present)
3607                 snd_soc_dapm_disable_pin(&codec->dapm, "MICBIAS2");
3608
3609         if (present)
3610                 snd_soc_jack_report(wm8994->micdet[0].jack,
3611                                     SND_JACK_MECHANICAL, SND_JACK_MECHANICAL);
3612         else
3613                 snd_soc_jack_report(wm8994->micdet[0].jack, 0,
3614                                     SND_JACK_MECHANICAL | SND_JACK_HEADSET |
3615                                     wm8994->btn_mask);
3616
3617         /* Since we only report deltas force an update, ensures we
3618          * avoid bootstrapping issues with the core. */
3619         snd_soc_jack_report(wm8994->micdet[0].jack, 0, 0);
3620
3621         pm_runtime_put(codec->dev);
3622         return IRQ_HANDLED;
3623 }
3624
3625 static void wm1811_jackdet_bootstrap(struct work_struct *work)
3626 {
3627         struct wm8994_priv *wm8994 = container_of(work,
3628                                                 struct wm8994_priv,
3629                                                 jackdet_bootstrap.work);
3630         wm1811_jackdet_irq(0, wm8994);
3631 }
3632
3633 /**
3634  * wm8958_mic_detect - Enable microphone detection via the WM8958 IRQ
3635  *
3636  * @codec:   WM8958 codec
3637  * @jack:    jack to report detection events on
3638  *
3639  * Enable microphone detection functionality for the WM8958.  By
3640  * default simple detection which supports the detection of up to 6
3641  * buttons plus video and microphone functionality is supported.
3642  *
3643  * The WM8958 has an advanced jack detection facility which is able to
3644  * support complex accessory detection, especially when used in
3645  * conjunction with external circuitry.  In order to provide maximum
3646  * flexiblity a callback is provided which allows a completely custom
3647  * detection algorithm.
3648  */
3649 int wm8958_mic_detect(struct snd_soc_codec *codec, struct snd_soc_jack *jack,
3650                       wm1811_micdet_cb det_cb, void *det_cb_data,
3651                       wm1811_mic_id_cb id_cb, void *id_cb_data)
3652 {
3653         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3654         struct wm8994 *control = wm8994->wm8994;
3655         u16 micd_lvl_sel;
3656
3657         switch (control->type) {
3658         case WM1811:
3659         case WM8958:
3660                 break;
3661         default:
3662                 return -EINVAL;
3663         }
3664
3665         if (jack) {
3666                 snd_soc_dapm_force_enable_pin(&codec->dapm, "CLK_SYS");
3667                 snd_soc_dapm_sync(&codec->dapm);
3668
3669                 wm8994->micdet[0].jack = jack;
3670
3671                 if (det_cb) {
3672                         wm8994->micd_cb = det_cb;
3673                         wm8994->micd_cb_data = det_cb_data;
3674                 } else {
3675                         wm8994->mic_detecting = true;
3676                         wm8994->jack_mic = false;
3677                 }
3678
3679                 if (id_cb) {
3680                         wm8994->mic_id_cb = id_cb;
3681                         wm8994->mic_id_cb_data = id_cb_data;
3682                 } else {
3683                         wm8994->mic_id_cb = wm8958_mic_id;
3684                         wm8994->mic_id_cb_data = codec;
3685                 }
3686
3687                 wm8958_micd_set_rate(codec);
3688
3689                 /* Detect microphones and short circuits by default */
3690                 if (control->pdata.micd_lvl_sel)
3691                         micd_lvl_sel = control->pdata.micd_lvl_sel;
3692                 else
3693                         micd_lvl_sel = 0x41;
3694
3695                 wm8994->btn_mask = SND_JACK_BTN_0 | SND_JACK_BTN_1 |
3696                         SND_JACK_BTN_2 | SND_JACK_BTN_3 |
3697                         SND_JACK_BTN_4 | SND_JACK_BTN_5;
3698
3699                 snd_soc_update_bits(codec, WM8958_MIC_DETECT_2,
3700                                     WM8958_MICD_LVL_SEL_MASK, micd_lvl_sel);
3701
3702                 WARN_ON(codec->dapm.bias_level > SND_SOC_BIAS_STANDBY);
3703
3704                 /*
3705                  * If we can use jack detection start off with that,
3706                  * otherwise jump straight to microphone detection.
3707                  */
3708                 if (wm8994->jackdet) {
3709                         /* Disable debounce for the initial detect */
3710                         snd_soc_update_bits(codec, WM1811_JACKDET_CTRL,
3711                                             WM1811_JACKDET_DB, 0);
3712
3713                         snd_soc_update_bits(codec, WM8958_MICBIAS2,
3714                                             WM8958_MICB2_DISCH,
3715                                             WM8958_MICB2_DISCH);
3716                         snd_soc_update_bits(codec, WM8994_LDO_1,
3717                                             WM8994_LDO1_DISCH, 0);
3718                         wm1811_jackdet_set_mode(codec,
3719                                                 WM1811_JACKDET_MODE_JACK);
3720                 } else {
3721                         snd_soc_update_bits(codec, WM8958_MIC_DETECT_1,
3722                                             WM8958_MICD_ENA, WM8958_MICD_ENA);
3723                 }
3724
3725         } else {
3726                 snd_soc_update_bits(codec, WM8958_MIC_DETECT_1,
3727                                     WM8958_MICD_ENA, 0);
3728                 wm1811_jackdet_set_mode(codec, WM1811_JACKDET_MODE_NONE);
3729                 snd_soc_dapm_disable_pin(&codec->dapm, "CLK_SYS");
3730                 snd_soc_dapm_sync(&codec->dapm);
3731         }
3732
3733         return 0;
3734 }
3735 EXPORT_SYMBOL_GPL(wm8958_mic_detect);
3736
3737 static irqreturn_t wm8958_mic_irq(int irq, void *data)
3738 {
3739         struct wm8994_priv *wm8994 = data;
3740         struct snd_soc_codec *codec = wm8994->hubs.codec;
3741         int reg, count, ret;
3742
3743         /*
3744          * Jack detection may have detected a removal simulataneously
3745          * with an update of the MICDET status; if so it will have
3746          * stopped detection and we can ignore this interrupt.
3747          */
3748         if (!(snd_soc_read(codec, WM8958_MIC_DETECT_1) & WM8958_MICD_ENA))
3749                 return IRQ_HANDLED;
3750
3751         pm_runtime_get_sync(codec->dev);
3752
3753         /* We may occasionally read a detection without an impedence
3754          * range being provided - if that happens loop again.
3755          */
3756         count = 10;
3757         do {
3758                 reg = snd_soc_read(codec, WM8958_MIC_DETECT_3);
3759                 if (reg < 0) {
3760                         dev_err(codec->dev,
3761                                 "Failed to read mic detect status: %d\n",
3762                                 reg);
3763                         pm_runtime_put(codec->dev);
3764                         return IRQ_NONE;
3765                 }
3766
3767                 if (!(reg & WM8958_MICD_VALID)) {
3768                         dev_dbg(codec->dev, "Mic detect data not valid\n");
3769                         goto out;
3770                 }
3771
3772                 if (!(reg & WM8958_MICD_STS) || (reg & WM8958_MICD_LVL_MASK))
3773                         break;
3774
3775                 msleep(1);
3776         } while (count--);
3777
3778         if (count == 0)
3779                 dev_warn(codec->dev, "No impedance range reported for jack\n");
3780
3781 #ifndef CONFIG_SND_SOC_WM8994_MODULE
3782         trace_snd_soc_jack_irq(dev_name(codec->dev));
3783 #endif
3784
3785         /* Avoid a transient report when the accessory is being removed */
3786         if (wm8994->jackdet) {
3787                 ret = snd_soc_read(codec, WM1811_JACKDET_CTRL);
3788                 if (ret < 0) {
3789                         dev_err(codec->dev, "Failed to read jack status: %d\n",
3790                                 ret);
3791                 } else if (!(ret & WM1811_JACKDET_LVL)) {
3792                         dev_dbg(codec->dev, "Ignoring removed jack\n");
3793                         return IRQ_HANDLED;
3794                 }
3795         }
3796
3797         if (wm8994->mic_detecting)
3798                 wm8994->mic_id_cb(wm8994->mic_id_cb_data, reg);
3799         else
3800                 wm8958_button_det(codec, reg);
3801
3802 out:
3803         pm_runtime_put(codec->dev);
3804         return IRQ_HANDLED;
3805 }
3806
3807 static irqreturn_t wm8994_fifo_error(int irq, void *data)
3808 {
3809         struct snd_soc_codec *codec = data;
3810
3811         dev_err(codec->dev, "FIFO error\n");
3812
3813         return IRQ_HANDLED;
3814 }
3815
3816 static irqreturn_t wm8994_temp_warn(int irq, void *data)
3817 {
3818         struct snd_soc_codec *codec = data;
3819
3820         dev_err(codec->dev, "Thermal warning\n");
3821
3822         return IRQ_HANDLED;
3823 }
3824
3825 static irqreturn_t wm8994_temp_shut(int irq, void *data)
3826 {
3827         struct snd_soc_codec *codec = data;
3828
3829         dev_crit(codec->dev, "Thermal shutdown\n");
3830
3831         return IRQ_HANDLED;
3832 }
3833
3834 static int wm8994_codec_probe(struct snd_soc_codec *codec)
3835 {
3836         struct wm8994 *control = dev_get_drvdata(codec->dev->parent);
3837         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
3838         struct snd_soc_dapm_context *dapm = &codec->dapm;
3839         unsigned int reg;
3840         int ret, i;
3841
3842         wm8994->hubs.codec = codec;
3843         codec->control_data = control->regmap;
3844
3845         snd_soc_codec_set_cache_io(codec, 16, 16, SND_SOC_REGMAP);
3846
3847         mutex_init(&wm8994->accdet_lock);
3848         INIT_DELAYED_WORK(&wm8994->jackdet_bootstrap,
3849                           wm1811_jackdet_bootstrap);
3850
3851         switch (control->type) {
3852         case WM8994:
3853                 INIT_DELAYED_WORK(&wm8994->mic_work, wm8994_mic_work);
3854                 break;
3855         case WM1811:
3856                 INIT_DELAYED_WORK(&wm8994->mic_work, wm1811_mic_work);
3857                 break;
3858         default:
3859                 break;
3860         }
3861
3862         for (i = 0; i < ARRAY_SIZE(wm8994->fll_locked); i++)
3863                 init_completion(&wm8994->fll_locked[i]);
3864
3865         wm8994->micdet_irq = control->pdata.micdet_irq;
3866
3867         pm_runtime_enable(codec->dev);
3868         pm_runtime_idle(codec->dev);
3869
3870         /* By default use idle_bias_off, will override for WM8994 */
3871         codec->dapm.idle_bias_off = 1;
3872
3873         /* Set revision-specific configuration */
3874         wm8994->revision = snd_soc_read(codec, WM8994_CHIP_REVISION);
3875         switch (control->type) {
3876         case WM8994:
3877                 /* Single ended line outputs should have VMID on. */
3878                 if (!control->pdata.lineout1_diff ||
3879                     !control->pdata.lineout2_diff)
3880                         codec->dapm.idle_bias_off = 0;
3881
3882                 switch (wm8994->revision) {
3883                 case 2:
3884                 case 3:
3885                         wm8994->hubs.dcs_codes_l = -5;
3886                         wm8994->hubs.dcs_codes_r = -5;
3887                         wm8994->hubs.hp_startup_mode = 1;
3888                         wm8994->hubs.dcs_readback_mode = 1;
3889                         wm8994->hubs.series_startup = 1;
3890                         break;
3891                 default:
3892                         wm8994->hubs.dcs_readback_mode = 2;
3893                         break;
3894                 }
3895                 break;
3896
3897         case WM8958:
3898                 wm8994->hubs.dcs_readback_mode = 1;
3899                 wm8994->hubs.hp_startup_mode = 1;
3900
3901                 switch (wm8994->revision) {
3902                 case 0:
3903                         break;
3904                 default:
3905                         wm8994->fll_byp = true;
3906                         break;
3907                 }
3908                 break;
3909
3910         case WM1811:
3911                 wm8994->hubs.dcs_readback_mode = 2;
3912                 wm8994->hubs.no_series_update = 1;
3913                 wm8994->hubs.hp_startup_mode = 1;
3914                 wm8994->hubs.no_cache_dac_hp_direct = true;
3915                 wm8994->fll_byp = true;
3916
3917                 wm8994->hubs.dcs_codes_l = -9;
3918                 wm8994->hubs.dcs_codes_r = -7;
3919
3920                 snd_soc_update_bits(codec, WM8994_ANALOGUE_HP_1,
3921                                     WM1811_HPOUT1_ATTN, WM1811_HPOUT1_ATTN);
3922                 break;
3923
3924         default:
3925                 break;
3926         }
3927
3928         wm8994_request_irq(wm8994->wm8994, WM8994_IRQ_FIFOS_ERR,
3929                            wm8994_fifo_error, "FIFO error", codec);
3930         wm8994_request_irq(wm8994->wm8994, WM8994_IRQ_TEMP_WARN,
3931                            wm8994_temp_warn, "Thermal warning", codec);
3932         wm8994_request_irq(wm8994->wm8994, WM8994_IRQ_TEMP_SHUT,
3933                            wm8994_temp_shut, "Thermal shutdown", codec);
3934
3935         ret = wm8994_request_irq(wm8994->wm8994, WM8994_IRQ_DCS_DONE,
3936                                  wm_hubs_dcs_done, "DC servo done",
3937                                  &wm8994->hubs);
3938         if (ret == 0)
3939                 wm8994->hubs.dcs_done_irq = true;
3940
3941         switch (control->type) {
3942         case WM8994:
3943                 if (wm8994->micdet_irq) {
3944                         ret = request_threaded_irq(wm8994->micdet_irq, NULL,
3945                                                    wm8994_mic_irq,
3946                                                    IRQF_TRIGGER_RISING,
3947                                                    "Mic1 detect",
3948                                                    wm8994);
3949                         if (ret != 0)
3950                                 dev_warn(codec->dev,
3951                                          "Failed to request Mic1 detect IRQ: %d\n",
3952                                          ret);
3953                 }
3954
3955                 ret = wm8994_request_irq(wm8994->wm8994,
3956                                          WM8994_IRQ_MIC1_SHRT,
3957                                          wm8994_mic_irq, "Mic 1 short",
3958                                          wm8994);
3959                 if (ret != 0)
3960                         dev_warn(codec->dev,
3961                                  "Failed to request Mic1 short IRQ: %d\n",
3962                                  ret);
3963
3964                 ret = wm8994_request_irq(wm8994->wm8994,
3965                                          WM8994_IRQ_MIC2_DET,
3966                                          wm8994_mic_irq, "Mic 2 detect",
3967                                          wm8994);
3968                 if (ret != 0)
3969                         dev_warn(codec->dev,
3970                                  "Failed to request Mic2 detect IRQ: %d\n",
3971                                  ret);
3972
3973                 ret = wm8994_request_irq(wm8994->wm8994,
3974                                          WM8994_IRQ_MIC2_SHRT,
3975                                          wm8994_mic_irq, "Mic 2 short",
3976                                          wm8994);
3977                 if (ret != 0)
3978                         dev_warn(codec->dev,
3979                                  "Failed to request Mic2 short IRQ: %d\n",
3980                                  ret);
3981                 break;
3982
3983         case WM8958:
3984         case WM1811:
3985                 if (wm8994->micdet_irq) {
3986                         ret = request_threaded_irq(wm8994->micdet_irq, NULL,
3987                                                    wm8958_mic_irq,
3988                                                    IRQF_TRIGGER_RISING,
3989                                                    "Mic detect",
3990                                                    wm8994);
3991                         if (ret != 0)
3992                                 dev_warn(codec->dev,
3993                                          "Failed to request Mic detect IRQ: %d\n",
3994                                          ret);
3995                 } else {
3996                         wm8994_request_irq(wm8994->wm8994, WM8994_IRQ_MIC1_DET,
3997                                            wm8958_mic_irq, "Mic detect",
3998                                            wm8994);
3999                 }
4000         }
4001
4002         switch (control->type) {
4003         case WM1811:
4004                 if (control->cust_id > 1 || wm8994->revision > 1) {
4005                         ret = wm8994_request_irq(wm8994->wm8994,
4006                                                  WM8994_IRQ_GPIO(6),
4007                                                  wm1811_jackdet_irq, "JACKDET",
4008                                                  wm8994);
4009                         if (ret == 0)
4010                                 wm8994->jackdet = true;
4011                 }
4012                 break;
4013         default:
4014                 break;
4015         }
4016
4017         wm8994->fll_locked_irq = true;
4018         for (i = 0; i < ARRAY_SIZE(wm8994->fll_locked); i++) {
4019                 ret = wm8994_request_irq(wm8994->wm8994,
4020                                          WM8994_IRQ_FLL1_LOCK + i,
4021                                          wm8994_fll_locked_irq, "FLL lock",
4022                                          &wm8994->fll_locked[i]);
4023                 if (ret != 0)
4024                         wm8994->fll_locked_irq = false;
4025         }
4026
4027         /* Make sure we can read from the GPIOs if they're inputs */
4028         pm_runtime_get_sync(codec->dev);
4029
4030         /* Remember if AIFnLRCLK is configured as a GPIO.  This should be
4031          * configured on init - if a system wants to do this dynamically
4032          * at runtime we can deal with that then.
4033          */
4034         ret = regmap_read(control->regmap, WM8994_GPIO_1, &reg);
4035         if (ret < 0) {
4036                 dev_err(codec->dev, "Failed to read GPIO1 state: %d\n", ret);
4037                 goto err_irq;
4038         }
4039         if ((reg & WM8994_GPN_FN_MASK) != WM8994_GP_FN_PIN_SPECIFIC) {
4040                 wm8994->lrclk_shared[0] = 1;
4041                 wm8994_dai[0].symmetric_rates = 1;
4042         } else {
4043                 wm8994->lrclk_shared[0] = 0;
4044         }
4045
4046         ret = regmap_read(control->regmap, WM8994_GPIO_6, &reg);
4047         if (ret < 0) {
4048                 dev_err(codec->dev, "Failed to read GPIO6 state: %d\n", ret);
4049                 goto err_irq;
4050         }
4051         if ((reg & WM8994_GPN_FN_MASK) != WM8994_GP_FN_PIN_SPECIFIC) {
4052                 wm8994->lrclk_shared[1] = 1;
4053                 wm8994_dai[1].symmetric_rates = 1;
4054         } else {
4055                 wm8994->lrclk_shared[1] = 0;
4056         }
4057
4058         pm_runtime_put(codec->dev);
4059
4060         /* Latch volume update bits */
4061         for (i = 0; i < ARRAY_SIZE(wm8994_vu_bits); i++)
4062                 snd_soc_update_bits(codec, wm8994_vu_bits[i].reg,
4063                                     wm8994_vu_bits[i].mask,
4064                                     wm8994_vu_bits[i].mask);
4065
4066         /* Set the low bit of the 3D stereo depth so TLV matches */
4067         snd_soc_update_bits(codec, WM8994_AIF1_DAC1_FILTERS_2,
4068                             1 << WM8994_AIF1DAC1_3D_GAIN_SHIFT,
4069                             1 << WM8994_AIF1DAC1_3D_GAIN_SHIFT);
4070         snd_soc_update_bits(codec, WM8994_AIF1_DAC2_FILTERS_2,
4071                             1 << WM8994_AIF1DAC2_3D_GAIN_SHIFT,
4072                             1 << WM8994_AIF1DAC2_3D_GAIN_SHIFT);
4073         snd_soc_update_bits(codec, WM8994_AIF2_DAC_FILTERS_2,
4074                             1 << WM8994_AIF2DAC_3D_GAIN_SHIFT,
4075                             1 << WM8994_AIF2DAC_3D_GAIN_SHIFT);
4076
4077         /* Unconditionally enable AIF1 ADC TDM mode on chips which can
4078          * use this; it only affects behaviour on idle TDM clock
4079          * cycles. */
4080         switch (control->type) {
4081         case WM8994:
4082         case WM8958:
4083                 snd_soc_update_bits(codec, WM8994_AIF1_CONTROL_1,
4084                                     WM8994_AIF1ADC_TDM, WM8994_AIF1ADC_TDM);
4085                 break;
4086         default:
4087                 break;
4088         }
4089
4090         /* Put MICBIAS into bypass mode by default on newer devices */
4091         switch (control->type) {
4092         case WM8958:
4093         case WM1811:
4094                 snd_soc_update_bits(codec, WM8958_MICBIAS1,
4095                                     WM8958_MICB1_MODE, WM8958_MICB1_MODE);
4096                 snd_soc_update_bits(codec, WM8958_MICBIAS2,
4097                                     WM8958_MICB2_MODE, WM8958_MICB2_MODE);
4098                 break;
4099         default:
4100                 break;
4101         }
4102
4103         wm8994->hubs.check_class_w_digital = wm8994_check_class_w_digital;
4104         wm_hubs_update_class_w(codec);
4105
4106         wm8994_handle_pdata(wm8994);
4107
4108         wm_hubs_add_analogue_controls(codec);
4109         snd_soc_add_codec_controls(codec, wm8994_snd_controls,
4110                              ARRAY_SIZE(wm8994_snd_controls));
4111         snd_soc_dapm_new_controls(dapm, wm8994_dapm_widgets,
4112                                   ARRAY_SIZE(wm8994_dapm_widgets));
4113
4114         switch (control->type) {
4115         case WM8994:
4116                 snd_soc_dapm_new_controls(dapm, wm8994_specific_dapm_widgets,
4117                                           ARRAY_SIZE(wm8994_specific_dapm_widgets));
4118                 if (wm8994->revision < 4) {
4119                         snd_soc_dapm_new_controls(dapm, wm8994_lateclk_revd_widgets,
4120                                                   ARRAY_SIZE(wm8994_lateclk_revd_widgets));
4121                         snd_soc_dapm_new_controls(dapm, wm8994_adc_revd_widgets,
4122                                                   ARRAY_SIZE(wm8994_adc_revd_widgets));
4123                         snd_soc_dapm_new_controls(dapm, wm8994_dac_revd_widgets,
4124                                                   ARRAY_SIZE(wm8994_dac_revd_widgets));
4125                 } else {
4126                         snd_soc_dapm_new_controls(dapm, wm8994_lateclk_widgets,
4127                                                   ARRAY_SIZE(wm8994_lateclk_widgets));
4128                         snd_soc_dapm_new_controls(dapm, wm8994_adc_widgets,
4129                                                   ARRAY_SIZE(wm8994_adc_widgets));
4130                         snd_soc_dapm_new_controls(dapm, wm8994_dac_widgets,
4131                                                   ARRAY_SIZE(wm8994_dac_widgets));
4132                 }
4133                 break;
4134         case WM8958:
4135                 snd_soc_add_codec_controls(codec, wm8958_snd_controls,
4136                                      ARRAY_SIZE(wm8958_snd_controls));
4137                 snd_soc_dapm_new_controls(dapm, wm8958_dapm_widgets,
4138                                           ARRAY_SIZE(wm8958_dapm_widgets));
4139                 if (wm8994->revision < 1) {
4140                         snd_soc_dapm_new_controls(dapm, wm8994_lateclk_revd_widgets,
4141                                                   ARRAY_SIZE(wm8994_lateclk_revd_widgets));
4142                         snd_soc_dapm_new_controls(dapm, wm8994_adc_revd_widgets,
4143                                                   ARRAY_SIZE(wm8994_adc_revd_widgets));
4144                         snd_soc_dapm_new_controls(dapm, wm8994_dac_revd_widgets,
4145                                                   ARRAY_SIZE(wm8994_dac_revd_widgets));
4146                 } else {
4147                         snd_soc_dapm_new_controls(dapm, wm8994_lateclk_widgets,
4148                                                   ARRAY_SIZE(wm8994_lateclk_widgets));
4149                         snd_soc_dapm_new_controls(dapm, wm8994_adc_widgets,
4150                                                   ARRAY_SIZE(wm8994_adc_widgets));
4151                         snd_soc_dapm_new_controls(dapm, wm8994_dac_widgets,
4152                                                   ARRAY_SIZE(wm8994_dac_widgets));
4153                 }
4154                 break;
4155
4156         case WM1811:
4157                 snd_soc_add_codec_controls(codec, wm8958_snd_controls,
4158                                      ARRAY_SIZE(wm8958_snd_controls));
4159                 snd_soc_dapm_new_controls(dapm, wm8958_dapm_widgets,
4160                                           ARRAY_SIZE(wm8958_dapm_widgets));
4161                 snd_soc_dapm_new_controls(dapm, wm8994_lateclk_widgets,
4162                                           ARRAY_SIZE(wm8994_lateclk_widgets));
4163                 snd_soc_dapm_new_controls(dapm, wm8994_adc_widgets,
4164                                           ARRAY_SIZE(wm8994_adc_widgets));
4165                 snd_soc_dapm_new_controls(dapm, wm8994_dac_widgets,
4166                                           ARRAY_SIZE(wm8994_dac_widgets));
4167                 break;
4168         }
4169
4170         wm_hubs_add_analogue_routes(codec, 0, 0);
4171         snd_soc_dapm_add_routes(dapm, intercon, ARRAY_SIZE(intercon));
4172
4173         switch (control->type) {
4174         case WM8994:
4175                 snd_soc_dapm_add_routes(dapm, wm8994_intercon,
4176                                         ARRAY_SIZE(wm8994_intercon));
4177
4178                 if (wm8994->revision < 4) {
4179                         snd_soc_dapm_add_routes(dapm, wm8994_revd_intercon,
4180                                                 ARRAY_SIZE(wm8994_revd_intercon));
4181                         snd_soc_dapm_add_routes(dapm, wm8994_lateclk_revd_intercon,
4182                                                 ARRAY_SIZE(wm8994_lateclk_revd_intercon));
4183                 } else {
4184                         snd_soc_dapm_add_routes(dapm, wm8994_lateclk_intercon,
4185                                                 ARRAY_SIZE(wm8994_lateclk_intercon));
4186                 }
4187                 break;
4188         case WM8958:
4189                 if (wm8994->revision < 1) {
4190                         snd_soc_dapm_add_routes(dapm, wm8994_intercon,
4191                                                 ARRAY_SIZE(wm8994_intercon));
4192                         snd_soc_dapm_add_routes(dapm, wm8994_revd_intercon,
4193                                                 ARRAY_SIZE(wm8994_revd_intercon));
4194                         snd_soc_dapm_add_routes(dapm, wm8994_lateclk_revd_intercon,
4195                                                 ARRAY_SIZE(wm8994_lateclk_revd_intercon));
4196                 } else {
4197                         snd_soc_dapm_add_routes(dapm, wm8994_lateclk_intercon,
4198                                                 ARRAY_SIZE(wm8994_lateclk_intercon));
4199                         snd_soc_dapm_add_routes(dapm, wm8958_intercon,
4200                                                 ARRAY_SIZE(wm8958_intercon));
4201                 }
4202
4203                 wm8958_dsp2_init(codec);
4204                 break;
4205         case WM1811:
4206                 snd_soc_dapm_add_routes(dapm, wm8994_lateclk_intercon,
4207                                         ARRAY_SIZE(wm8994_lateclk_intercon));
4208                 snd_soc_dapm_add_routes(dapm, wm8958_intercon,
4209                                         ARRAY_SIZE(wm8958_intercon));
4210                 break;
4211         }
4212
4213         return 0;
4214
4215 err_irq:
4216         if (wm8994->jackdet)
4217                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_GPIO(6), wm8994);
4218         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC2_SHRT, wm8994);
4219         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC2_DET, wm8994);
4220         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC1_SHRT, wm8994);
4221         if (wm8994->micdet_irq)
4222                 free_irq(wm8994->micdet_irq, wm8994);
4223         for (i = 0; i < ARRAY_SIZE(wm8994->fll_locked); i++)
4224                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_FLL1_LOCK + i,
4225                                 &wm8994->fll_locked[i]);
4226         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_DCS_DONE,
4227                         &wm8994->hubs);
4228         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_FIFOS_ERR, codec);
4229         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_TEMP_SHUT, codec);
4230         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_TEMP_WARN, codec);
4231
4232         return ret;
4233 }
4234
4235 static int wm8994_codec_remove(struct snd_soc_codec *codec)
4236 {
4237         struct wm8994_priv *wm8994 = snd_soc_codec_get_drvdata(codec);
4238         struct wm8994 *control = wm8994->wm8994;
4239         int i;
4240
4241         wm8994_set_bias_level(codec, SND_SOC_BIAS_OFF);
4242
4243         pm_runtime_disable(codec->dev);
4244
4245         for (i = 0; i < ARRAY_SIZE(wm8994->fll_locked); i++)
4246                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_FLL1_LOCK + i,
4247                                 &wm8994->fll_locked[i]);
4248
4249         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_DCS_DONE,
4250                         &wm8994->hubs);
4251         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_FIFOS_ERR, codec);
4252         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_TEMP_SHUT, codec);
4253         wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_TEMP_WARN, codec);
4254
4255         if (wm8994->jackdet)
4256                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_GPIO(6), wm8994);
4257
4258         switch (control->type) {
4259         case WM8994:
4260                 if (wm8994->micdet_irq)
4261                         free_irq(wm8994->micdet_irq, wm8994);
4262                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC2_DET,
4263                                 wm8994);
4264                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC1_SHRT,
4265                                 wm8994);
4266                 wm8994_free_irq(wm8994->wm8994, WM8994_IRQ_MIC1_DET,
4267                                 wm8994);
4268                 break;
4269
4270         case WM1811:
4271         case WM8958:
4272                 if (wm8994->micdet_irq)
4273                         free_irq(wm8994->micdet_irq, wm8994);
4274                 break;
4275         }
4276         release_firmware(wm8994->mbc);
4277         release_firmware(wm8994->mbc_vss);
4278         release_firmware(wm8994->enh_eq);
4279         kfree(wm8994->retune_mobile_texts);
4280         return 0;
4281 }
4282
4283 static struct snd_soc_codec_driver soc_codec_dev_wm8994 = {
4284         .probe =        wm8994_codec_probe,
4285         .remove =       wm8994_codec_remove,
4286         .suspend =      wm8994_codec_suspend,
4287         .resume =       wm8994_codec_resume,
4288         .set_bias_level = wm8994_set_bias_level,
4289 };
4290
4291 static int wm8994_probe(struct platform_device *pdev)
4292 {
4293         struct wm8994_priv *wm8994;
4294
4295         wm8994 = devm_kzalloc(&pdev->dev, sizeof(struct wm8994_priv),
4296                               GFP_KERNEL);
4297         if (wm8994 == NULL)
4298                 return -ENOMEM;
4299         platform_set_drvdata(pdev, wm8994);
4300
4301         wm8994->wm8994 = dev_get_drvdata(pdev->dev.parent);
4302
4303         return snd_soc_register_codec(&pdev->dev, &soc_codec_dev_wm8994,
4304                         wm8994_dai, ARRAY_SIZE(wm8994_dai));
4305 }
4306
4307 static int wm8994_remove(struct platform_device *pdev)
4308 {
4309         snd_soc_unregister_codec(&pdev->dev);
4310         return 0;
4311 }
4312
4313 #ifdef CONFIG_PM_SLEEP
4314 static int wm8994_suspend(struct device *dev)
4315 {
4316         struct wm8994_priv *wm8994 = dev_get_drvdata(dev);
4317
4318         /* Drop down to power saving mode when system is suspended */
4319         if (wm8994->jackdet && !wm8994->active_refcount)
4320                 regmap_update_bits(wm8994->wm8994->regmap, WM8994_ANTIPOP_2,
4321                                    WM1811_JACKDET_MODE_MASK,
4322                                    wm8994->jackdet_mode);
4323
4324         return 0;
4325 }
4326
4327 static int wm8994_resume(struct device *dev)
4328 {
4329         struct wm8994_priv *wm8994 = dev_get_drvdata(dev);
4330
4331         if (wm8994->jackdet && wm8994->jackdet_mode)
4332                 regmap_update_bits(wm8994->wm8994->regmap, WM8994_ANTIPOP_2,
4333                                    WM1811_JACKDET_MODE_MASK,
4334                                    WM1811_JACKDET_MODE_AUDIO);
4335
4336         return 0;
4337 }
4338 #endif
4339
4340 static const struct dev_pm_ops wm8994_pm_ops = {
4341         SET_SYSTEM_SLEEP_PM_OPS(wm8994_suspend, wm8994_resume)
4342 };
4343
4344 static struct platform_driver wm8994_codec_driver = {
4345         .driver = {
4346                 .name = "wm8994-codec",
4347                 .owner = THIS_MODULE,
4348                 .pm = &wm8994_pm_ops,
4349         },
4350         .probe = wm8994_probe,
4351         .remove = wm8994_remove,
4352 };
4353
4354 module_platform_driver(wm8994_codec_driver);
4355
4356 MODULE_DESCRIPTION("ASoC WM8994 driver");
4357 MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
4358 MODULE_LICENSE("GPL");
4359 MODULE_ALIAS("platform:wm8994-codec");