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/platform_device.h>
  22 #include <linux/spi/spi.h>
  23 #include <sound/core.h>
  24 #include <sound/pcm.h>
  25 #include <sound/pcm_params.h>
  26 #include <sound/soc.h>
  27 #include <sound/soc-dapm.h>
  28 #include <sound/initval.h>
  29
  30 #include "wm8731.h"
  31
  32 #define WM8731_VERSION "0.13"
  33
  34 struct snd_soc_codec_device soc_codec_dev_wm8731;
  35
  36 /* codec private data */
  37 struct wm8731_priv {
  38         unsigned int sysclk;
  39 };
  40
  41 /*
  42  * wm8731 register cache
  43  * We can't read the WM8731 register space when we are
  44  * using 2 wire for device control, so we cache them instead.
  45  * There is no point in caching the reset register
  46  */
  47 static const u16 wm8731_reg[WM8731_CACHEREGNUM] = {
  48     0x0097, 0x0097, 0x0079, 0x0079,
  49     0x000a, 0x0008, 0x009f, 0x000a,
  50     0x0000, 0x0000
  51 };
  52
  53 /*
  54  * read wm8731 register cache
  55  */
  56 static inline unsigned int wm8731_read_reg_cache(struct snd_soc_codec *codec,
  57         unsigned int reg)
  58 {
  59         u16 *cache = codec->reg_cache;
  60         if (reg == WM8731_RESET)
  61                 return 0;
  62         if (reg >= WM8731_CACHEREGNUM)
  63                 return -1;
  64         return cache[reg];
  65 }
  66
  67 /*
  68  * write wm8731 register cache
  69  */
  70 static inline void wm8731_write_reg_cache(struct snd_soc_codec *codec,
  71         u16 reg, unsigned int value)
  72 {
  73         u16 *cache = codec->reg_cache;
  74         if (reg >= WM8731_CACHEREGNUM)
  75                 return;
  76         cache[reg] = value;
  77 }
  78
  79 /*
  80  * write to the WM8731 register space
  81  */
  82 static int wm8731_write(struct snd_soc_codec *codec, unsigned int reg,
  83         unsigned int value)
  84 {
  85         u8 data[2];
  86
  87         /* data is
  88          *   D15..D9 WM8731 register offset
  89          *   D8...D0 register data
  90          */
  91         data[0] = (reg << 1) | ((value >> 8) & 0x0001);
  92         data[1] = value & 0x00ff;
  93
  94         wm8731_write_reg_cache(codec, reg, value);
  95         if (codec->hw_write(codec->control_data, data, 2) == 2)
  96                 return 0;
  97         else
  98                 return -EIO;
  99 }
 100
 101 #define wm8731_reset(c) wm8731_write(c, WM8731_RESET, 0)
 102
 103 static const char *wm8731_input_select[] = {"Line In", "Mic"};
 104 static const char *wm8731_deemph[] = {"None", "32Khz", "44.1Khz", "48Khz"};
 105
 106 static const struct soc_enum wm8731_enum[] = {
 107         SOC_ENUM_SINGLE(WM8731_APANA, 2, 2, wm8731_input_select),
 108         SOC_ENUM_SINGLE(WM8731_APDIGI, 1, 4, wm8731_deemph),
 109 };
 110
 111 static const struct snd_kcontrol_new wm8731_snd_controls[] = {
 112
 113 SOC_DOUBLE_R("Master Playback Volume", WM8731_LOUT1V, WM8731_ROUT1V,
 114         0, 127, 0),
 115 SOC_DOUBLE_R("Master Playback ZC Switch", WM8731_LOUT1V, WM8731_ROUT1V,
 116         7, 1, 0),
 117
 118 SOC_DOUBLE_R("Capture Volume", WM8731_LINVOL, WM8731_RINVOL, 0, 31, 0),
 119 SOC_DOUBLE_R("Line Capture Switch", WM8731_LINVOL, WM8731_RINVOL, 7, 1, 1),
 120
 121 SOC_SINGLE("Mic Boost (+20dB)", WM8731_APANA, 0, 1, 0),
 122 SOC_SINGLE("Capture Mic Switch", WM8731_APANA, 1, 1, 1),
 123
 124 SOC_SINGLE("Sidetone Playback Volume", WM8731_APANA, 6, 3, 1),
 125
 126 SOC_SINGLE("ADC High Pass Filter Switch", WM8731_APDIGI, 0, 1, 1),
 127 SOC_SINGLE("Store DC Offset Switch", WM8731_APDIGI, 4, 1, 0),
 128
 129 SOC_ENUM("Playback De-emphasis", wm8731_enum[1]),
 130 };
 131
 132 /* add non dapm controls */
 133 static int wm8731_add_controls(struct snd_soc_codec *codec)
 134 {
 135         int err, i;
 136
 137         for (i = 0; i < ARRAY_SIZE(wm8731_snd_controls); i++) {
 138                 err = snd_ctl_add(codec->card,
 139                                   snd_soc_cnew(&wm8731_snd_controls[i],
 140                                                 codec, NULL));
 141                 if (err < 0)
 142                         return err;
 143         }
 144
 145         return 0;
 146 }
 147
 148 /* Output Mixer */
 149 static const struct snd_kcontrol_new wm8731_output_mixer_controls[] = {
 150 SOC_DAPM_SINGLE("Line Bypass Switch", WM8731_APANA, 3, 1, 0),
 151 SOC_DAPM_SINGLE("Mic Sidetone Switch", WM8731_APANA, 5, 1, 0),
 152 SOC_DAPM_SINGLE("HiFi Playback Switch", WM8731_APANA, 4, 1, 0),
 153 };
 154
 155 /* Input mux */
 156 static const struct snd_kcontrol_new wm8731_input_mux_controls =
 157 SOC_DAPM_ENUM("Input Select", wm8731_enum[0]);
 158
 159 static const struct snd_soc_dapm_widget wm8731_dapm_widgets[] = {
 160 SND_SOC_DAPM_MIXER("Output Mixer", WM8731_PWR, 4, 1,
 161         &wm8731_output_mixer_controls[0],
 162         ARRAY_SIZE(wm8731_output_mixer_controls)),
 163 SND_SOC_DAPM_DAC("DAC", "HiFi Playback", WM8731_PWR, 3, 1),
 164 SND_SOC_DAPM_OUTPUT("LOUT"),
 165 SND_SOC_DAPM_OUTPUT("LHPOUT"),
 166 SND_SOC_DAPM_OUTPUT("ROUT"),
 167 SND_SOC_DAPM_OUTPUT("RHPOUT"),
 168 SND_SOC_DAPM_ADC("ADC", "HiFi Capture", WM8731_PWR, 2, 1),
 169 SND_SOC_DAPM_MUX("Input Mux", SND_SOC_NOPM, 0, 0, &wm8731_input_mux_controls),
 170 SND_SOC_DAPM_PGA("Line Input", WM8731_PWR, 0, 1, NULL, 0),
 171 SND_SOC_DAPM_MICBIAS("Mic Bias", WM8731_PWR, 1, 1),
 172 SND_SOC_DAPM_INPUT("MICIN"),
 173 SND_SOC_DAPM_INPUT("RLINEIN"),
 174 SND_SOC_DAPM_INPUT("LLINEIN"),
 175 };
 176
 177 static const struct snd_soc_dapm_route intercon[] = {
 178         /* output mixer */
 179         {"Output Mixer", "Line Bypass Switch", "Line Input"},
 180         {"Output Mixer", "HiFi Playback Switch", "DAC"},
 181         {"Output Mixer", "Mic Sidetone Switch", "Mic Bias"},
 182
 183         /* outputs */
 184         {"RHPOUT", NULL, "Output Mixer"},
 185         {"ROUT", NULL, "Output Mixer"},
 186         {"LHPOUT", NULL, "Output Mixer"},
 187         {"LOUT", NULL, "Output Mixer"},
 188
 189         /* input mux */
 190         {"Input Mux", "Line In", "Line Input"},
 191         {"Input Mux", "Mic", "Mic Bias"},
 192         {"ADC", NULL, "Input Mux"},
 193
 194         /* inputs */
 195         {"Line Input", NULL, "LLINEIN"},
 196         {"Line Input", NULL, "RLINEIN"},
 197         {"Mic Bias", NULL, "MICIN"},
 198 };
 199
 200 static int wm8731_add_widgets(struct snd_soc_codec *codec)
 201 {
 202         snd_soc_dapm_new_controls(codec, wm8731_dapm_widgets,
 203                                   ARRAY_SIZE(wm8731_dapm_widgets));
 204
 205         snd_soc_dapm_add_routes(codec, intercon, ARRAY_SIZE(intercon));
 206
 207         snd_soc_dapm_new_widgets(codec);
 208         return 0;
 209 }
 210
 211 struct _coeff_div {
 212         u32 mclk;
 213         u32 rate;
 214         u16 fs;
 215         u8 sr:4;
 216         u8 bosr:1;
 217         u8 usb:1;
 218 };
 219
 220 /* codec mclk clock divider coefficients */
 221 static const struct _coeff_div coeff_div[] = {
 222         /* 48k */
 223         {12288000, 48000, 256, 0x0, 0x0, 0x0},
 224         {18432000, 48000, 384, 0x0, 0x1, 0x0},
 225         {12000000, 48000, 250, 0x0, 0x0, 0x1},
 226
 227         /* 32k */
 228         {12288000, 32000, 384, 0x6, 0x0, 0x0},
 229         {18432000, 32000, 576, 0x6, 0x1, 0x0},
 230         {12000000, 32000, 375, 0x6, 0x0, 0x1},
 231
 232         /* 8k */
 233         {12288000, 8000, 1536, 0x3, 0x0, 0x0},
 234         {18432000, 8000, 2304, 0x3, 0x1, 0x0},
 235         {11289600, 8000, 1408, 0xb, 0x0, 0x0},
 236         {16934400, 8000, 2112, 0xb, 0x1, 0x0},
 237         {12000000, 8000, 1500, 0x3, 0x0, 0x1},
 238
 239         /* 96k */
 240         {12288000, 96000, 128, 0x7, 0x0, 0x0},
 241         {18432000, 96000, 192, 0x7, 0x1, 0x0},
 242         {12000000, 96000, 125, 0x7, 0x0, 0x1},
 243
 244         /* 44.1k */
 245         {11289600, 44100, 256, 0x8, 0x0, 0x0},
 246         {16934400, 44100, 384, 0x8, 0x1, 0x0},
 247         {12000000, 44100, 272, 0x8, 0x1, 0x1},
 248
 249         /* 88.2k */
 250         {11289600, 88200, 128, 0xf, 0x0, 0x0},
 251         {16934400, 88200, 192, 0xf, 0x1, 0x0},
 252         {12000000, 88200, 136, 0xf, 0x1, 0x1},
 253 };
 254
 255 static inline int get_coeff(int mclk, int rate)
 256 {
 257         int i;
 258
 259         for (i = 0; i < ARRAY_SIZE(coeff_div); i++) {
 260                 if (coeff_div[i].rate == rate && coeff_div[i].mclk == mclk)
 261                         return i;
 262         }
 263         return 0;
 264 }
 265
 266 static int wm8731_hw_params(struct snd_pcm_substream *substream,
 267                             struct snd_pcm_hw_params *params,
 268                             struct snd_soc_dai *dai)
 269 {
 270         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 271         struct snd_soc_device *socdev = rtd->socdev;
 272         struct snd_soc_codec *codec = socdev->codec;
 273         struct wm8731_priv *wm8731 = codec->private_data;
 274         u16 iface = wm8731_read_reg_cache(codec, WM8731_IFACE) & 0xfff3;
 275         int i = get_coeff(wm8731->sysclk, params_rate(params));
 276         u16 srate = (coeff_div[i].sr << 2) |
 277                 (coeff_div[i].bosr << 1) | coeff_div[i].usb;
 278
 279         wm8731_write(codec, WM8731_SRATE, srate);
 280
 281         /* bit size */
 282         switch (params_format(params)) {
 283         case SNDRV_PCM_FORMAT_S16_LE:
 284                 break;
 285         case SNDRV_PCM_FORMAT_S20_3LE:
 286                 iface |= 0x0004;
 287                 break;
 288         case SNDRV_PCM_FORMAT_S24_LE:
 289                 iface |= 0x0008;
 290                 break;
 291         }
 292
 293         wm8731_write(codec, WM8731_IFACE, iface);
 294         return 0;
 295 }
 296
 297 static int wm8731_pcm_prepare(struct snd_pcm_substream *substream,
 298                               struct snd_soc_dai *dai)
 299 {
 300         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 301         struct snd_soc_device *socdev = rtd->socdev;
 302         struct snd_soc_codec *codec = socdev->codec;
 303
 304         /* set active */
 305         wm8731_write(codec, WM8731_ACTIVE, 0x0001);
 306
 307         return 0;
 308 }
 309
 310 static void wm8731_shutdown(struct snd_pcm_substream *substream,
 311                             struct snd_soc_dai *dai)
 312 {
 313         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 314         struct snd_soc_device *socdev = rtd->socdev;
 315         struct snd_soc_codec *codec = socdev->codec;
 316
 317         /* deactivate */
 318         if (!codec->active) {
 319                 udelay(50);
 320                 wm8731_write(codec, WM8731_ACTIVE, 0x0);
 321         }
 322 }
 323
 324 static int wm8731_mute(struct snd_soc_dai *dai, int mute)
 325 {
 326         struct snd_soc_codec *codec = dai->codec;
 327         u16 mute_reg = wm8731_read_reg_cache(codec, WM8731_APDIGI) & 0xfff7;
 328
 329         if (mute)
 330                 wm8731_write(codec, WM8731_APDIGI, mute_reg | 0x8);
 331         else
 332                 wm8731_write(codec, WM8731_APDIGI, mute_reg);
 333         return 0;
 334 }
 335
 336 static int wm8731_set_dai_sysclk(struct snd_soc_dai *codec_dai,
 337                 int clk_id, unsigned int freq, int dir)
 338 {
 339         struct snd_soc_codec *codec = codec_dai->codec;
 340         struct wm8731_priv *wm8731 = codec->private_data;
 341
 342         switch (freq) {
 343         case 11289600:
 344         case 12000000:
 345         case 12288000:
 346         case 16934400:
 347         case 18432000:
 348                 wm8731->sysclk = freq;
 349                 return 0;
 350         }
 351         return -EINVAL;
 352 }
 353
 354
 355 static int wm8731_set_dai_fmt(struct snd_soc_dai *codec_dai,
 356                 unsigned int fmt)
 357 {
 358         struct snd_soc_codec *codec = codec_dai->codec;
 359         u16 iface = 0;
 360
 361         /* set master/slave audio interface */
 362         switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
 363         case SND_SOC_DAIFMT_CBM_CFM:
 364                 iface |= 0x0040;
 365                 break;
 366         case SND_SOC_DAIFMT_CBS_CFS:
 367                 break;
 368         default:
 369                 return -EINVAL;
 370         }
 371
 372         /* interface format */
 373         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
 374         case SND_SOC_DAIFMT_I2S:
 375                 iface |= 0x0002;
 376                 break;
 377         case SND_SOC_DAIFMT_RIGHT_J:
 378                 break;
 379         case SND_SOC_DAIFMT_LEFT_J:
 380                 iface |= 0x0001;
 381                 break;
 382         case SND_SOC_DAIFMT_DSP_A:
 383                 iface |= 0x0003;
 384                 break;
 385         case SND_SOC_DAIFMT_DSP_B:
 386                 iface |= 0x0013;
 387                 break;
 388         default:
 389                 return -EINVAL;
 390         }
 391
 392         /* clock inversion */
 393         switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
 394         case SND_SOC_DAIFMT_NB_NF:
 395                 break;
 396         case SND_SOC_DAIFMT_IB_IF:
 397                 iface |= 0x0090;
 398                 break;
 399         case SND_SOC_DAIFMT_IB_NF:
 400                 iface |= 0x0080;
 401                 break;
 402         case SND_SOC_DAIFMT_NB_IF:
 403                 iface |= 0x0010;
 404                 break;
 405         default:
 406                 return -EINVAL;
 407         }
 408
 409         /* set iface */
 410         wm8731_write(codec, WM8731_IFACE, iface);
 411         return 0;
 412 }
 413
 414 static int wm8731_set_bias_level(struct snd_soc_codec *codec,
 415                                  enum snd_soc_bias_level level)
 416 {
 417         u16 reg = wm8731_read_reg_cache(codec, WM8731_PWR) & 0xff7f;
 418
 419         switch (level) {
 420         case SND_SOC_BIAS_ON:
 421                 /* vref/mid, osc on, dac unmute */
 422                 wm8731_write(codec, WM8731_PWR, reg);
 423                 break;
 424         case SND_SOC_BIAS_PREPARE:
 425                 break;
 426         case SND_SOC_BIAS_STANDBY:
 427                 /* everything off except vref/vmid, */
 428                 wm8731_write(codec, WM8731_PWR, reg | 0x0040);
 429                 break;
 430         case SND_SOC_BIAS_OFF:
 431                 /* everything off, dac mute, inactive */
 432                 wm8731_write(codec, WM8731_ACTIVE, 0x0);
 433                 wm8731_write(codec, WM8731_PWR, 0xffff);
 434                 break;
 435         }
 436         codec->bias_level = level;
 437         return 0;
 438 }
 439
 440 #define WM8731_RATES (SNDRV_PCM_RATE_8000 | SNDRV_PCM_RATE_11025 |\
 441                 SNDRV_PCM_RATE_16000 | SNDRV_PCM_RATE_22050 |\
 442                 SNDRV_PCM_RATE_32000 | SNDRV_PCM_RATE_44100 |\
 443                 SNDRV_PCM_RATE_48000 | SNDRV_PCM_RATE_88200 |\
 444                 SNDRV_PCM_RATE_96000)
 445
 446 #define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
 447         SNDRV_PCM_FMTBIT_S24_LE)
 448
 449 struct snd_soc_dai wm8731_dai = {
 450         .name = "WM8731",
 451         .playback = {
 452                 .stream_name = "Playback",
 453                 .channels_min = 1,
 454                 .channels_max = 2,
 455                 .rates = WM8731_RATES,
 456                 .formats = WM8731_FORMATS,},
 457         .capture = {
 458                 .stream_name = "Capture",
 459                 .channels_min = 1,
 460                 .channels_max = 2,
 461                 .rates = WM8731_RATES,
 462                 .formats = WM8731_FORMATS,},
 463         .ops = {
 464                 .prepare = wm8731_pcm_prepare,
 465                 .hw_params = wm8731_hw_params,
 466                 .shutdown = wm8731_shutdown,
 467                 .digital_mute = wm8731_mute,
 468                 .set_sysclk = wm8731_set_dai_sysclk,
 469                 .set_fmt = wm8731_set_dai_fmt,
 470         }
 471 };
 472 EXPORT_SYMBOL_GPL(wm8731_dai);
 473
 474 static int wm8731_suspend(struct platform_device *pdev, pm_message_t state)
 475 {
 476         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 477         struct snd_soc_codec *codec = socdev->codec;
 478
 479         wm8731_write(codec, WM8731_ACTIVE, 0x0);
 480         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 481         return 0;
 482 }
 483
 484 static int wm8731_resume(struct platform_device *pdev)
 485 {
 486         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 487         struct snd_soc_codec *codec = socdev->codec;
 488         int i;
 489         u8 data[2];
 490         u16 *cache = codec->reg_cache;
 491
 492         /* Sync reg_cache with the hardware */
 493         for (i = 0; i < ARRAY_SIZE(wm8731_reg); i++) {
 494                 data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
 495                 data[1] = cache[i] & 0x00ff;
 496                 codec->hw_write(codec->control_data, data, 2);
 497         }
 498         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 499         wm8731_set_bias_level(codec, codec->suspend_bias_level);
 500         return 0;
 501 }
 502
 503 /*
 504  * initialise the WM8731 driver
 505  * register the mixer and dsp interfaces with the kernel
 506  */
 507 static int wm8731_init(struct snd_soc_device *socdev)
 508 {
 509         struct snd_soc_codec *codec = socdev->codec;
 510         int reg, ret = 0;
 511
 512         codec->name = "WM8731";
 513         codec->owner = THIS_MODULE;
 514         codec->read = wm8731_read_reg_cache;
 515         codec->write = wm8731_write;
 516         codec->set_bias_level = wm8731_set_bias_level;
 517         codec->dai = &wm8731_dai;
 518         codec->num_dai = 1;
 519         codec->reg_cache_size = ARRAY_SIZE(wm8731_reg);
 520         codec->reg_cache = kmemdup(wm8731_reg, sizeof(wm8731_reg), GFP_KERNEL);
 521         if (codec->reg_cache == NULL)
 522                 return -ENOMEM;
 523
 524         wm8731_reset(codec);
 525
 526         /* register pcms */
 527         ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
 528         if (ret < 0) {
 529                 printk(KERN_ERR "wm8731: failed to create pcms\n");
 530                 goto pcm_err;
 531         }
 532
 533         /* power on device */
 534         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 535
 536         /* set the update bits */
 537         reg = wm8731_read_reg_cache(codec, WM8731_LOUT1V);
 538         wm8731_write(codec, WM8731_LOUT1V, reg & ~0x0100);
 539         reg = wm8731_read_reg_cache(codec, WM8731_ROUT1V);
 540         wm8731_write(codec, WM8731_ROUT1V, reg & ~0x0100);
 541         reg = wm8731_read_reg_cache(codec, WM8731_LINVOL);
 542         wm8731_write(codec, WM8731_LINVOL, reg & ~0x0100);
 543         reg = wm8731_read_reg_cache(codec, WM8731_RINVOL);
 544         wm8731_write(codec, WM8731_RINVOL, reg & ~0x0100);
 545
 546         wm8731_add_controls(codec);
 547         wm8731_add_widgets(codec);
 548         ret = snd_soc_register_card(socdev);
 549         if (ret < 0) {
 550                 printk(KERN_ERR "wm8731: failed to register card\n");
 551                 goto card_err;
 552         }
 553
 554         return ret;
 555
 556 card_err:
 557         snd_soc_free_pcms(socdev);
 558         snd_soc_dapm_free(socdev);
 559 pcm_err:
 560         kfree(codec->reg_cache);
 561         return ret;
 562 }
 563
 564 static struct snd_soc_device *wm8731_socdev;
 565
 566 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 567
 568 /*
 569  * WM8731 2 wire address is determined by GPIO5
 570  * state during powerup.
 571  *    low  = 0x1a
 572  *    high = 0x1b
 573  */
 574
 575 static int wm8731_i2c_probe(struct i2c_client *i2c,
 576                             const struct i2c_device_id *id)
 577 {
 578         struct snd_soc_device *socdev = wm8731_socdev;
 579         struct snd_soc_codec *codec = socdev->codec;
 580         int ret;
 581
 582         i2c_set_clientdata(i2c, codec);
 583         codec->control_data = i2c;
 584
 585         ret = wm8731_init(socdev);
 586         if (ret < 0)
 587                 pr_err("failed to initialise WM8731\n");
 588
 589         return ret;
 590 }
 591
 592 static int wm8731_i2c_remove(struct i2c_client *client)
 593 {
 594         struct snd_soc_codec *codec = i2c_get_clientdata(client);
 595         kfree(codec->reg_cache);
 596         return 0;
 597 }
 598
 599 static const struct i2c_device_id wm8731_i2c_id[] = {
 600         { "wm8731", 0 },
 601         { }
 602 };
 603 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
 604
 605 static struct i2c_driver wm8731_i2c_driver = {
 606         .driver = {
 607                 .name = "WM8731 I2C Codec",
 608                 .owner = THIS_MODULE,
 609         },
 610         .probe =    wm8731_i2c_probe,
 611         .remove =   wm8731_i2c_remove,
 612         .id_table = wm8731_i2c_id,
 613 };
 614
 615 static int wm8731_add_i2c_device(struct platform_device *pdev,
 616                                  const struct wm8731_setup_data *setup)
 617 {
 618         struct i2c_board_info info;
 619         struct i2c_adapter *adapter;
 620         struct i2c_client *client;
 621         int ret;
 622
 623         ret = i2c_add_driver(&wm8731_i2c_driver);
 624         if (ret != 0) {
 625                 dev_err(&pdev->dev, "can't add i2c driver\n");
 626                 return ret;
 627         }
 628
 629         memset(&info, 0, sizeof(struct i2c_board_info));
 630         info.addr = setup->i2c_address;
 631         strlcpy(info.type, "wm8731", I2C_NAME_SIZE);
 632
 633         adapter = i2c_get_adapter(setup->i2c_bus);
 634         if (!adapter) {
 635                 dev_err(&pdev->dev, "can't get i2c adapter %d\n",
 636                         setup->i2c_bus);
 637                 goto err_driver;
 638         }
 639
 640         client = i2c_new_device(adapter, &info);
 641         i2c_put_adapter(adapter);
 642         if (!client) {
 643                 dev_err(&pdev->dev, "can't add i2c device at 0x%x\n",
 644                         (unsigned int)info.addr);
 645                 goto err_driver;
 646         }
 647
 648         return 0;
 649
 650 err_driver:
 651         i2c_del_driver(&wm8731_i2c_driver);
 652         return -ENODEV;
 653 }
 654 #endif
 655
 656 #if defined(CONFIG_SPI_MASTER)
 657 static int __devinit wm8731_spi_probe(struct spi_device *spi)
 658 {
 659         struct snd_soc_device *socdev = wm8731_socdev;
 660         struct snd_soc_codec *codec = socdev->codec;
 661         int ret;
 662
 663         codec->control_data = spi;
 664
 665         ret = wm8731_init(socdev);
 666         if (ret < 0)
 667                 dev_err(&spi->dev, "failed to initialise WM8731\n");
 668
 669         return ret;
 670 }
 671
 672 static int __devexit wm8731_spi_remove(struct spi_device *spi)
 673 {
 674         return 0;
 675 }
 676
 677 static struct spi_driver wm8731_spi_driver = {
 678         .driver = {
 679                 .name   = "wm8731",
 680                 .bus    = &spi_bus_type,
 681                 .owner  = THIS_MODULE,
 682         },
 683         .probe          = wm8731_spi_probe,
 684         .remove         = __devexit_p(wm8731_spi_remove),
 685 };
 686
 687 static int wm8731_spi_write(struct spi_device *spi, const char *data, int len)
 688 {
 689         struct spi_transfer t;
 690         struct spi_message m;
 691         u8 msg[2];
 692
 693         if (len <= 0)
 694                 return 0;
 695
 696         msg[0] = data[0];
 697         msg[1] = data[1];
 698
 699         spi_message_init(&m);
 700         memset(&t, 0, (sizeof t));
 701
 702         t.tx_buf = &msg[0];
 703         t.len = len;
 704
 705         spi_message_add_tail(&t, &m);
 706         spi_sync(spi, &m);
 707
 708         return len;
 709 }
 710 #endif /* CONFIG_SPI_MASTER */
 711
 712 static int wm8731_probe(struct platform_device *pdev)
 713 {
 714         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 715         struct wm8731_setup_data *setup;
 716         struct snd_soc_codec *codec;
 717         struct wm8731_priv *wm8731;
 718         int ret = 0;
 719
 720         pr_info("WM8731 Audio Codec %s", WM8731_VERSION);
 721
 722         setup = socdev->codec_data;
 723         codec = kzalloc(sizeof(struct snd_soc_codec), GFP_KERNEL);
 724         if (codec == NULL)
 725                 return -ENOMEM;
 726
 727         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
 728         if (wm8731 == NULL) {
 729                 kfree(codec);
 730                 return -ENOMEM;
 731         }
 732
 733         codec->private_data = wm8731;
 734         socdev->codec = codec;
 735         mutex_init(&codec->mutex);
 736         INIT_LIST_HEAD(&codec->dapm_widgets);
 737         INIT_LIST_HEAD(&codec->dapm_paths);
 738
 739         wm8731_socdev = socdev;
 740         ret = -ENODEV;
 741
 742 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 743         if (setup->i2c_address) {
 744                 codec->hw_write = (hw_write_t)i2c_master_send;
 745                 ret = wm8731_add_i2c_device(pdev, setup);
 746         }
 747 #endif
 748 #if defined(CONFIG_SPI_MASTER)
 749         if (setup->spi) {
 750                 codec->hw_write = (hw_write_t)wm8731_spi_write;
 751                 ret = spi_register_driver(&wm8731_spi_driver);
 752                 if (ret != 0)
 753                         printk(KERN_ERR "can't add spi driver");
 754         }
 755 #endif
 756
 757         if (ret != 0) {
 758                 kfree(codec->private_data);
 759                 kfree(codec);
 760         }
 761         return ret;
 762 }
 763
 764 /* power down chip */
 765 static int wm8731_remove(struct platform_device *pdev)
 766 {
 767         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 768         struct snd_soc_codec *codec = socdev->codec;
 769
 770         if (codec->control_data)
 771                 wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 772
 773         snd_soc_free_pcms(socdev);
 774         snd_soc_dapm_free(socdev);
 775 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 776         i2c_unregister_device(codec->control_data);
 777         i2c_del_driver(&wm8731_i2c_driver);
 778 #endif
 779 #if defined(CONFIG_SPI_MASTER)
 780         spi_unregister_driver(&wm8731_spi_driver);
 781 #endif
 782         kfree(codec->private_data);
 783         kfree(codec);
 784
 785         return 0;
 786 }
 787
 788 struct snd_soc_codec_device soc_codec_dev_wm8731 = {
 789         .probe =        wm8731_probe,
 790         .remove =       wm8731_remove,
 791         .suspend =      wm8731_suspend,
 792         .resume =       wm8731_resume,
 793 };
 794 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8731);
 795
 796 MODULE_DESCRIPTION("ASoC WM8731 driver");
 797 MODULE_AUTHOR("Richard Purdie");
 798 MODULE_LICENSE("GPL");