sound/soc/codecs/wm8731.c

   1 /*
   2  * wm8731.c  --  WM8731 ALSA SoC Audio driver
   3  *
   4  * Copyright 2005 Openedhand Ltd.
   5  *
   6  * Author: Richard Purdie <richard@openedhand.com>
   7  *
   8  * Based on wm8753.c by Liam Girdwood
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License version 2 as
  12  * published by the Free Software Foundation.
  13  */
  14
  15 #include <linux/module.h>
  16 #include <linux/moduleparam.h>
  17 #include <linux/init.h>
  18 #include <linux/delay.h>
  19 #include <linux/pm.h>
  20 #include <linux/i2c.h>
  21 #include <linux/slab.h>
  22 #include <linux/regulator/consumer.h>
  23 #include <linux/spi/spi.h>
  24 #include <linux/of_device.h>
  25 #include <sound/core.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
  32 #include "wm8731.h"
  33
  34 #define WM8731_NUM_SUPPLIES 4
  35 static const char *wm8731_supply_names[WM8731_NUM_SUPPLIES] = {
  36         "AVDD",
  37         "HPVDD",
  38         "DCVDD",
  39         "DBVDD",
  40 };
  41
  42 /* codec private data */
  43 struct wm8731_priv {
  44         enum snd_soc_control_type control_type;
  45         struct regulator_bulk_data supplies[WM8731_NUM_SUPPLIES];
  46         unsigned int sysclk;
  47         int sysclk_type;
  48         int playback_fs;
  49         bool deemph;
  50 };
  51
  52
  53 /*
  54  * wm8731 register cache
  55  * We can't read the WM8731 register space when we are
  56  * using 2 wire for device control, so we cache them instead.
  57  * There is no point in caching the reset register
  58  */
  59 static const u16 wm8731_reg[WM8731_CACHEREGNUM] = {
  60         0x0097, 0x0097, 0x0079, 0x0079,
  61         0x000a, 0x0008, 0x009f, 0x000a,
  62         0x0000, 0x0000
  63 };
  64
  65 #define wm8731_reset(c) snd_soc_write(c, WM8731_RESET, 0)
  66
  67 static const char *wm8731_input_select[] = {"Line In", "Mic"};
  68
  69 static const struct soc_enum wm8731_insel_enum =
  70         SOC_ENUM_SINGLE(WM8731_APANA, 2, 2, wm8731_input_select);
  71
  72 static int wm8731_deemph[] = { 0, 32000, 44100, 48000 };
  73
  74 static int wm8731_set_deemph(struct snd_soc_codec *codec)
  75 {
  76         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
  77         int val, i, best;
  78
  79         /* If we're using deemphasis select the nearest available sample
  80          * rate.
  81          */
  82         if (wm8731->deemph) {
  83                 best = 1;
  84                 for (i = 2; i < ARRAY_SIZE(wm8731_deemph); i++) {
  85                         if (abs(wm8731_deemph[i] - wm8731->playback_fs) <
  86                             abs(wm8731_deemph[best] - wm8731->playback_fs))
  87                                 best = i;
  88                 }
  89
  90                 val = best << 1;
  91         } else {
  92                 best = 0;
  93                 val = 0;
  94         }
  95
  96         dev_dbg(codec->dev, "Set deemphasis %d (%dHz)\n",
  97                 best, wm8731_deemph[best]);
  98
  99         return snd_soc_update_bits(codec, WM8731_APDIGI, 0x6, val);
 100 }
 101
 102 static int wm8731_get_deemph(struct snd_kcontrol *kcontrol,
 103                              struct snd_ctl_elem_value *ucontrol)
 104 {
 105         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 106         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 107
 108         ucontrol->value.enumerated.item[0] = wm8731->deemph;
 109
 110         return 0;
 111 }
 112
 113 static int wm8731_put_deemph(struct snd_kcontrol *kcontrol,
 114                              struct snd_ctl_elem_value *ucontrol)
 115 {
 116         struct snd_soc_codec *codec = snd_kcontrol_chip(kcontrol);
 117         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 118         int deemph = ucontrol->value.enumerated.item[0];
 119         int ret = 0;
 120
 121         if (deemph > 1)
 122                 return -EINVAL;
 123
 124         mutex_lock(&codec->mutex);
 125         if (wm8731->deemph != deemph) {
 126                 wm8731->deemph = deemph;
 127
 128                 wm8731_set_deemph(codec);
 129
 130                 ret = 1;
 131         }
 132         mutex_unlock(&codec->mutex);
 133
 134         return ret;
 135 }
 136
 137 static const DECLARE_TLV_DB_SCALE(in_tlv, -3450, 150, 0);
 138 static const DECLARE_TLV_DB_SCALE(sidetone_tlv, -1500, 300, 0);
 139 static const DECLARE_TLV_DB_SCALE(out_tlv, -12100, 100, 1);
 140 static const DECLARE_TLV_DB_SCALE(mic_tlv, 0, 2000, 0);
 141
 142 static const struct snd_kcontrol_new wm8731_snd_controls[] = {
 143
 144 SOC_DOUBLE_R_TLV("Master Playback Volume", WM8731_LOUT1V, WM8731_ROUT1V,
 145                  0, 127, 0, out_tlv),
 146 SOC_DOUBLE_R("Master Playback ZC Switch", WM8731_LOUT1V, WM8731_ROUT1V,
 147         7, 1, 0),
 148
 149 SOC_DOUBLE_R_TLV("Capture Volume", WM8731_LINVOL, WM8731_RINVOL, 0, 31, 0,
 150                  in_tlv),
 151 SOC_DOUBLE_R("Line Capture Switch", WM8731_LINVOL, WM8731_RINVOL, 7, 1, 1),
 152
 153 SOC_SINGLE_TLV("Mic Boost Volume", WM8731_APANA, 0, 1, 0, mic_tlv),
 154 SOC_SINGLE("Mic Capture Switch", WM8731_APANA, 1, 1, 1),
 155
 156 SOC_SINGLE_TLV("Sidetone Playback Volume", WM8731_APANA, 6, 3, 1,
 157                sidetone_tlv),
 158
 159 SOC_SINGLE("ADC High Pass Filter Switch", WM8731_APDIGI, 0, 1, 1),
 160 SOC_SINGLE("Store DC Offset Switch", WM8731_APDIGI, 4, 1, 0),
 161
 162 SOC_SINGLE_BOOL_EXT("Playback Deemphasis Switch", 0,
 163                     wm8731_get_deemph, wm8731_put_deemph),
 164 };
 165
 166 /* Output Mixer */
 167 static const struct snd_kcontrol_new wm8731_output_mixer_controls[] = {
 168 SOC_DAPM_SINGLE("Line Bypass Switch", WM8731_APANA, 3, 1, 0),
 169 SOC_DAPM_SINGLE("Mic Sidetone Switch", WM8731_APANA, 5, 1, 0),
 170 SOC_DAPM_SINGLE("HiFi Playback Switch", WM8731_APANA, 4, 1, 0),
 171 };
 172
 173 /* Input mux */
 174 static const struct snd_kcontrol_new wm8731_input_mux_controls =
 175 SOC_DAPM_ENUM("Input Select", wm8731_insel_enum);
 176
 177 static const struct snd_soc_dapm_widget wm8731_dapm_widgets[] = {
 178 SND_SOC_DAPM_SUPPLY("ACTIVE",WM8731_ACTIVE, 0, 0, NULL, 0),
 179 SND_SOC_DAPM_SUPPLY("OSC", WM8731_PWR, 5, 1, NULL, 0),
 180 SND_SOC_DAPM_MIXER("Output Mixer", WM8731_PWR, 4, 1,
 181         &wm8731_output_mixer_controls[0],
 182         ARRAY_SIZE(wm8731_output_mixer_controls)),
 183 SND_SOC_DAPM_DAC("DAC", "HiFi Playback", WM8731_PWR, 3, 1),
 184 SND_SOC_DAPM_OUTPUT("LOUT"),
 185 SND_SOC_DAPM_OUTPUT("LHPOUT"),
 186 SND_SOC_DAPM_OUTPUT("ROUT"),
 187 SND_SOC_DAPM_OUTPUT("RHPOUT"),
 188 SND_SOC_DAPM_ADC("ADC", "HiFi Capture", WM8731_PWR, 2, 1),
 189 SND_SOC_DAPM_MUX("Input Mux", SND_SOC_NOPM, 0, 0, &wm8731_input_mux_controls),
 190 SND_SOC_DAPM_PGA("Line Input", WM8731_PWR, 0, 1, NULL, 0),
 191 SND_SOC_DAPM_MICBIAS("Mic Bias", WM8731_PWR, 1, 1),
 192 SND_SOC_DAPM_INPUT("MICIN"),
 193 SND_SOC_DAPM_INPUT("RLINEIN"),
 194 SND_SOC_DAPM_INPUT("LLINEIN"),
 195 };
 196
 197 static int wm8731_check_osc(struct snd_soc_dapm_widget *source,
 198                             struct snd_soc_dapm_widget *sink)
 199 {
 200         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(source->codec);
 201
 202         return wm8731->sysclk_type == WM8731_SYSCLK_XTAL;
 203 }
 204
 205 static const struct snd_soc_dapm_route wm8731_intercon[] = {
 206         {"DAC", NULL, "OSC", wm8731_check_osc},
 207         {"ADC", NULL, "OSC", wm8731_check_osc},
 208         {"DAC", NULL, "ACTIVE"},
 209         {"ADC", NULL, "ACTIVE"},
 210
 211         /* output mixer */
 212         {"Output Mixer", "Line Bypass Switch", "Line Input"},
 213         {"Output Mixer", "HiFi Playback Switch", "DAC"},
 214         {"Output Mixer", "Mic Sidetone Switch", "Mic Bias"},
 215
 216         /* outputs */
 217         {"RHPOUT", NULL, "Output Mixer"},
 218         {"ROUT", NULL, "Output Mixer"},
 219         {"LHPOUT", NULL, "Output Mixer"},
 220         {"LOUT", NULL, "Output Mixer"},
 221
 222         /* input mux */
 223         {"Input Mux", "Line In", "Line Input"},
 224         {"Input Mux", "Mic", "Mic Bias"},
 225         {"ADC", NULL, "Input Mux"},
 226
 227         /* inputs */
 228         {"Line Input", NULL, "LLINEIN"},
 229         {"Line Input", NULL, "RLINEIN"},
 230         {"Mic Bias", NULL, "MICIN"},
 231 };
 232
 233 struct _coeff_div {
 234         u32 mclk;
 235         u32 rate;
 236         u16 fs;
 237         u8 sr:4;
 238         u8 bosr:1;
 239         u8 usb:1;
 240 };
 241
 242 /* codec mclk clock divider coefficients */
 243 static const struct _coeff_div coeff_div[] = {
 244         /* 48k */
 245         {12288000, 48000, 256, 0x0, 0x0, 0x0},
 246         {18432000, 48000, 384, 0x0, 0x1, 0x0},
 247         {12000000, 48000, 250, 0x0, 0x0, 0x1},
 248
 249         /* 32k */
 250         {12288000, 32000, 384, 0x6, 0x0, 0x0},
 251         {18432000, 32000, 576, 0x6, 0x1, 0x0},
 252         {12000000, 32000, 375, 0x6, 0x0, 0x1},
 253
 254         /* 8k */
 255         {12288000, 8000, 1536, 0x3, 0x0, 0x0},
 256         {18432000, 8000, 2304, 0x3, 0x1, 0x0},
 257         {11289600, 8000, 1408, 0xb, 0x0, 0x0},
 258         {16934400, 8000, 2112, 0xb, 0x1, 0x0},
 259         {12000000, 8000, 1500, 0x3, 0x0, 0x1},
 260
 261         /* 96k */
 262         {12288000, 96000, 128, 0x7, 0x0, 0x0},
 263         {18432000, 96000, 192, 0x7, 0x1, 0x0},
 264         {12000000, 96000, 125, 0x7, 0x0, 0x1},
 265
 266         /* 44.1k */
 267         {11289600, 44100, 256, 0x8, 0x0, 0x0},
 268         {16934400, 44100, 384, 0x8, 0x1, 0x0},
 269         {12000000, 44100, 272, 0x8, 0x1, 0x1},
 270
 271         /* 88.2k */
 272         {11289600, 88200, 128, 0xf, 0x0, 0x0},
 273         {16934400, 88200, 192, 0xf, 0x1, 0x0},
 274         {12000000, 88200, 136, 0xf, 0x1, 0x1},
 275 };
 276
 277 static inline int get_coeff(int mclk, int rate)
 278 {
 279         int i;
 280
 281         for (i = 0; i < ARRAY_SIZE(coeff_div); i++) {
 282                 if (coeff_div[i].rate == rate && coeff_div[i].mclk == mclk)
 283                         return i;
 284         }
 285         return 0;
 286 }
 287
 288 static int wm8731_hw_params(struct snd_pcm_substream *substream,
 289                             struct snd_pcm_hw_params *params,
 290                             struct snd_soc_dai *dai)
 291 {
 292         struct snd_soc_codec *codec = dai->codec;
 293         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 294         u16 iface = snd_soc_read(codec, WM8731_IFACE) & 0xfff3;
 295         int i = get_coeff(wm8731->sysclk, params_rate(params));
 296         u16 srate = (coeff_div[i].sr << 2) |
 297                 (coeff_div[i].bosr << 1) | coeff_div[i].usb;
 298
 299         wm8731->playback_fs = params_rate(params);
 300
 301         snd_soc_write(codec, WM8731_SRATE, srate);
 302
 303         /* bit size */
 304         switch (params_format(params)) {
 305         case SNDRV_PCM_FORMAT_S16_LE:
 306                 break;
 307         case SNDRV_PCM_FORMAT_S20_3LE:
 308                 iface |= 0x0004;
 309                 break;
 310         case SNDRV_PCM_FORMAT_S24_LE:
 311                 iface |= 0x0008;
 312                 break;
 313         }
 314
 315         wm8731_set_deemph(codec);
 316
 317         snd_soc_write(codec, WM8731_IFACE, iface);
 318         return 0;
 319 }
 320
 321 static int wm8731_mute(struct snd_soc_dai *dai, int mute)
 322 {
 323         struct snd_soc_codec *codec = dai->codec;
 324         u16 mute_reg = snd_soc_read(codec, WM8731_APDIGI) & 0xfff7;
 325
 326         if (mute)
 327                 snd_soc_write(codec, WM8731_APDIGI, mute_reg | 0x8);
 328         else
 329                 snd_soc_write(codec, WM8731_APDIGI, mute_reg);
 330         return 0;
 331 }
 332
 333 static int wm8731_set_dai_sysclk(struct snd_soc_dai *codec_dai,
 334                 int clk_id, unsigned int freq, int dir)
 335 {
 336         struct snd_soc_codec *codec = codec_dai->codec;
 337         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 338
 339         switch (clk_id) {
 340         case WM8731_SYSCLK_XTAL:
 341         case WM8731_SYSCLK_MCLK:
 342                 wm8731->sysclk_type = clk_id;
 343                 break;
 344         default:
 345                 return -EINVAL;
 346         }
 347
 348         switch (freq) {
 349         case 11289600:
 350         case 12000000:
 351         case 12288000:
 352         case 16934400:
 353         case 18432000:
 354                 wm8731->sysclk = freq;
 355                 break;
 356         default:
 357                 return -EINVAL;
 358         }
 359
 360         snd_soc_dapm_sync(&codec->dapm);
 361
 362         return 0;
 363 }
 364
 365
 366 static int wm8731_set_dai_fmt(struct snd_soc_dai *codec_dai,
 367                 unsigned int fmt)
 368 {
 369         struct snd_soc_codec *codec = codec_dai->codec;
 370         u16 iface = 0;
 371
 372         /* set master/slave audio interface */
 373         switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
 374         case SND_SOC_DAIFMT_CBM_CFM:
 375                 iface |= 0x0040;
 376                 break;
 377         case SND_SOC_DAIFMT_CBS_CFS:
 378                 break;
 379         default:
 380                 return -EINVAL;
 381         }
 382
 383         /* interface format */
 384         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
 385         case SND_SOC_DAIFMT_I2S:
 386                 iface |= 0x0002;
 387                 break;
 388         case SND_SOC_DAIFMT_RIGHT_J:
 389                 break;
 390         case SND_SOC_DAIFMT_LEFT_J:
 391                 iface |= 0x0001;
 392                 break;
 393         case SND_SOC_DAIFMT_DSP_A:
 394                 iface |= 0x0003;
 395                 break;
 396         case SND_SOC_DAIFMT_DSP_B:
 397                 iface |= 0x0013;
 398                 break;
 399         default:
 400                 return -EINVAL;
 401         }
 402
 403         /* clock inversion */
 404         switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
 405         case SND_SOC_DAIFMT_NB_NF:
 406                 break;
 407         case SND_SOC_DAIFMT_IB_IF:
 408                 iface |= 0x0090;
 409                 break;
 410         case SND_SOC_DAIFMT_IB_NF:
 411                 iface |= 0x0080;
 412                 break;
 413         case SND_SOC_DAIFMT_NB_IF:
 414                 iface |= 0x0010;
 415                 break;
 416         default:
 417                 return -EINVAL;
 418         }
 419
 420         /* set iface */
 421         snd_soc_write(codec, WM8731_IFACE, iface);
 422         return 0;
 423 }
 424
 425 static int wm8731_set_bias_level(struct snd_soc_codec *codec,
 426                                  enum snd_soc_bias_level level)
 427 {
 428         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 429         int ret;
 430         u16 reg;
 431
 432         switch (level) {
 433         case SND_SOC_BIAS_ON:
 434                 break;
 435         case SND_SOC_BIAS_PREPARE:
 436                 break;
 437         case SND_SOC_BIAS_STANDBY:
 438                 if (codec->dapm.bias_level == SND_SOC_BIAS_OFF) {
 439                         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
 440                                                     wm8731->supplies);
 441                         if (ret != 0)
 442                                 return ret;
 443
 444                         snd_soc_cache_sync(codec);
 445                 }
 446
 447                 /* Clear PWROFF, gate CLKOUT, everything else as-is */
 448                 reg = snd_soc_read(codec, WM8731_PWR) & 0xff7f;
 449                 snd_soc_write(codec, WM8731_PWR, reg | 0x0040);
 450                 break;
 451         case SND_SOC_BIAS_OFF:
 452                 snd_soc_write(codec, WM8731_PWR, 0xffff);
 453                 regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies),
 454                                        wm8731->supplies);
 455                 codec->cache_sync = 1;
 456                 break;
 457         }
 458         codec->dapm.bias_level = level;
 459         return 0;
 460 }
 461
 462 #define WM8731_RATES SNDRV_PCM_RATE_8000_96000
 463
 464 #define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
 465         SNDRV_PCM_FMTBIT_S24_LE)
 466
 467 static const struct snd_soc_dai_ops wm8731_dai_ops = {
 468         .hw_params      = wm8731_hw_params,
 469         .digital_mute   = wm8731_mute,
 470         .set_sysclk     = wm8731_set_dai_sysclk,
 471         .set_fmt        = wm8731_set_dai_fmt,
 472 };
 473
 474 static struct snd_soc_dai_driver wm8731_dai = {
 475         .name = "wm8731-hifi",
 476         .playback = {
 477                 .stream_name = "Playback",
 478                 .channels_min = 1,
 479                 .channels_max = 2,
 480                 .rates = WM8731_RATES,
 481                 .formats = WM8731_FORMATS,},
 482         .capture = {
 483                 .stream_name = "Capture",
 484                 .channels_min = 1,
 485                 .channels_max = 2,
 486                 .rates = WM8731_RATES,
 487                 .formats = WM8731_FORMATS,},
 488         .ops = &wm8731_dai_ops,
 489         .symmetric_rates = 1,
 490 };
 491
 492 #ifdef CONFIG_PM
 493 static int wm8731_suspend(struct snd_soc_codec *codec)
 494 {
 495         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 496
 497         return 0;
 498 }
 499
 500 static int wm8731_resume(struct snd_soc_codec *codec)
 501 {
 502         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 503
 504         return 0;
 505 }
 506 #else
 507 #define wm8731_suspend NULL
 508 #define wm8731_resume NULL
 509 #endif
 510
 511 static int wm8731_probe(struct snd_soc_codec *codec)
 512 {
 513         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 514         int ret = 0, i;
 515
 516         ret = snd_soc_codec_set_cache_io(codec, 7, 9, wm8731->control_type);
 517         if (ret < 0) {
 518                 dev_err(codec->dev, "Failed to set cache I/O: %d\n", ret);
 519                 return ret;
 520         }
 521
 522         for (i = 0; i < ARRAY_SIZE(wm8731->supplies); i++)
 523                 wm8731->supplies[i].supply = wm8731_supply_names[i];
 524
 525         ret = regulator_bulk_get(codec->dev, ARRAY_SIZE(wm8731->supplies),
 526                                  wm8731->supplies);
 527         if (ret != 0) {
 528                 dev_err(codec->dev, "Failed to request supplies: %d\n", ret);
 529                 return ret;
 530         }
 531
 532         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
 533                                     wm8731->supplies);
 534         if (ret != 0) {
 535                 dev_err(codec->dev, "Failed to enable supplies: %d\n", ret);
 536                 goto err_regulator_get;
 537         }
 538
 539         ret = wm8731_reset(codec);
 540         if (ret < 0) {
 541                 dev_err(codec->dev, "Failed to issue reset: %d\n", ret);
 542                 goto err_regulator_enable;
 543         }
 544
 545         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 546
 547         /* Latch the update bits */
 548         snd_soc_update_bits(codec, WM8731_LOUT1V, 0x100, 0);
 549         snd_soc_update_bits(codec, WM8731_ROUT1V, 0x100, 0);
 550         snd_soc_update_bits(codec, WM8731_LINVOL, 0x100, 0);
 551         snd_soc_update_bits(codec, WM8731_RINVOL, 0x100, 0);
 552
 553         /* Disable bypass path by default */
 554         snd_soc_update_bits(codec, WM8731_APANA, 0x8, 0);
 555
 556         /* Regulators will have been enabled by bias management */
 557         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
 558
 559         return 0;
 560
 561 err_regulator_enable:
 562         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
 563 err_regulator_get:
 564         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
 565
 566         return ret;
 567 }
 568
 569 /* power down chip */
 570 static int wm8731_remove(struct snd_soc_codec *codec)
 571 {
 572         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
 573
 574         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 575
 576         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
 577         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
 578
 579         return 0;
 580 }
 581
 582 static struct snd_soc_codec_driver soc_codec_dev_wm8731 = {
 583         .probe =        wm8731_probe,
 584         .remove =       wm8731_remove,
 585         .suspend =      wm8731_suspend,
 586         .resume =       wm8731_resume,
 587         .set_bias_level = wm8731_set_bias_level,
 588         .reg_cache_size = ARRAY_SIZE(wm8731_reg),
 589         .reg_word_size = sizeof(u16),
 590         .reg_cache_default = wm8731_reg,
 591         .dapm_widgets = wm8731_dapm_widgets,
 592         .num_dapm_widgets = ARRAY_SIZE(wm8731_dapm_widgets),
 593         .dapm_routes = wm8731_intercon,
 594         .num_dapm_routes = ARRAY_SIZE(wm8731_intercon),
 595         .controls =     wm8731_snd_controls,
 596         .num_controls = ARRAY_SIZE(wm8731_snd_controls),
 597 };
 598
 599 static const struct of_device_id wm8731_of_match[] = {
 600         { .compatible = "wlf,wm8731", },
 601         { }
 602 };
 603
 604 MODULE_DEVICE_TABLE(of, wm8731_of_match);
 605
 606 #if defined(CONFIG_SPI_MASTER)
 607 static int __devinit wm8731_spi_probe(struct spi_device *spi)
 608 {
 609         struct wm8731_priv *wm8731;
 610         int ret;
 611
 612         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
 613         if (wm8731 == NULL)
 614                 return -ENOMEM;
 615
 616         wm8731->control_type = SND_SOC_SPI;
 617         spi_set_drvdata(spi, wm8731);
 618
 619         ret = snd_soc_register_codec(&spi->dev,
 620                         &soc_codec_dev_wm8731, &wm8731_dai, 1);
 621         if (ret < 0)
 622                 kfree(wm8731);
 623         return ret;
 624 }
 625
 626 static int __devexit wm8731_spi_remove(struct spi_device *spi)
 627 {
 628         snd_soc_unregister_codec(&spi->dev);
 629         kfree(spi_get_drvdata(spi));
 630         return 0;
 631 }
 632
 633 static struct spi_driver wm8731_spi_driver = {
 634         .driver = {
 635                 .name   = "wm8731",
 636                 .owner  = THIS_MODULE,
 637                 .of_match_table = wm8731_of_match,
 638         },
 639         .probe          = wm8731_spi_probe,
 640         .remove         = __devexit_p(wm8731_spi_remove),
 641 };
 642 #endif /* CONFIG_SPI_MASTER */
 643
 644 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 645 static __devinit int wm8731_i2c_probe(struct i2c_client *i2c,
 646                                       const struct i2c_device_id *id)
 647 {
 648         struct wm8731_priv *wm8731;
 649         int ret;
 650
 651         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
 652         if (wm8731 == NULL)
 653                 return -ENOMEM;
 654
 655         i2c_set_clientdata(i2c, wm8731);
 656         wm8731->control_type = SND_SOC_I2C;
 657
 658         ret =  snd_soc_register_codec(&i2c->dev,
 659                         &soc_codec_dev_wm8731, &wm8731_dai, 1);
 660         if (ret < 0)
 661                 kfree(wm8731);
 662         return ret;
 663 }
 664
 665 static __devexit int wm8731_i2c_remove(struct i2c_client *client)
 666 {
 667         snd_soc_unregister_codec(&client->dev);
 668         kfree(i2c_get_clientdata(client));
 669         return 0;
 670 }
 671
 672 static const struct i2c_device_id wm8731_i2c_id[] = {
 673         { "wm8731", 0 },
 674         { }
 675 };
 676 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
 677
 678 static struct i2c_driver wm8731_i2c_driver = {
 679         .driver = {
 680                 .name = "wm8731",
 681                 .owner = THIS_MODULE,
 682                 .of_match_table = wm8731_of_match,
 683         },
 684         .probe =    wm8731_i2c_probe,
 685         .remove =   __devexit_p(wm8731_i2c_remove),
 686         .id_table = wm8731_i2c_id,
 687 };
 688 #endif
 689
 690 static int __init wm8731_modinit(void)
 691 {
 692         int ret = 0;
 693 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 694         ret = i2c_add_driver(&wm8731_i2c_driver);
 695         if (ret != 0) {
 696                 printk(KERN_ERR "Failed to register WM8731 I2C driver: %d\n",
 697                        ret);
 698         }
 699 #endif
 700 #if defined(CONFIG_SPI_MASTER)
 701         ret = spi_register_driver(&wm8731_spi_driver);
 702         if (ret != 0) {
 703                 printk(KERN_ERR "Failed to register WM8731 SPI driver: %d\n",
 704                        ret);
 705         }
 706 #endif
 707         return ret;
 708 }
 709 module_init(wm8731_modinit);
 710
 711 static void __exit wm8731_exit(void)
 712 {
 713 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 714         i2c_del_driver(&wm8731_i2c_driver);
 715 #endif
 716 #if defined(CONFIG_SPI_MASTER)
 717         spi_unregister_driver(&wm8731_spi_driver);
 718 #endif
 719 }
 720 module_exit(wm8731_exit);
 721
 722 MODULE_DESCRIPTION("ASoC WM8731 driver");
 723 MODULE_AUTHOR("Richard Purdie");
 724 MODULE_LICENSE("GPL");