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 {
 269         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 270         struct snd_soc_device *socdev = rtd->socdev;
 271         struct snd_soc_codec *codec = socdev->codec;
 272         struct wm8731_priv *wm8731 = codec->private_data;
 273         u16 iface = wm8731_read_reg_cache(codec, WM8731_IFACE) & 0xfff3;
 274         int i = get_coeff(wm8731->sysclk, params_rate(params));
 275         u16 srate = (coeff_div[i].sr << 2) |
 276                 (coeff_div[i].bosr << 1) | coeff_div[i].usb;
 277
 278         wm8731_write(codec, WM8731_SRATE, srate);
 279
 280         /* bit size */
 281         switch (params_format(params)) {
 282         case SNDRV_PCM_FORMAT_S16_LE:
 283                 break;
 284         case SNDRV_PCM_FORMAT_S20_3LE:
 285                 iface |= 0x0004;
 286                 break;
 287         case SNDRV_PCM_FORMAT_S24_LE:
 288                 iface |= 0x0008;
 289                 break;
 290         }
 291
 292         wm8731_write(codec, WM8731_IFACE, iface);
 293         return 0;
 294 }
 295
 296 static int wm8731_pcm_prepare(struct snd_pcm_substream *substream)
 297 {
 298         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 299         struct snd_soc_device *socdev = rtd->socdev;
 300         struct snd_soc_codec *codec = socdev->codec;
 301
 302         /* set active */
 303         wm8731_write(codec, WM8731_ACTIVE, 0x0001);
 304
 305         return 0;
 306 }
 307
 308 static void wm8731_shutdown(struct snd_pcm_substream *substream)
 309 {
 310         struct snd_soc_pcm_runtime *rtd = substream->private_data;
 311         struct snd_soc_device *socdev = rtd->socdev;
 312         struct snd_soc_codec *codec = socdev->codec;
 313
 314         /* deactivate */
 315         if (!codec->active) {
 316                 udelay(50);
 317                 wm8731_write(codec, WM8731_ACTIVE, 0x0);
 318         }
 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 = wm8731_read_reg_cache(codec, WM8731_APDIGI) & 0xfff7;
 325
 326         if (mute)
 327                 wm8731_write(codec, WM8731_APDIGI, mute_reg | 0x8);
 328         else
 329                 wm8731_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 = codec->private_data;
 338
 339         switch (freq) {
 340         case 11289600:
 341         case 12000000:
 342         case 12288000:
 343         case 16934400:
 344         case 18432000:
 345                 wm8731->sysclk = freq;
 346                 return 0;
 347         }
 348         return -EINVAL;
 349 }
 350
 351
 352 static int wm8731_set_dai_fmt(struct snd_soc_dai *codec_dai,
 353                 unsigned int fmt)
 354 {
 355         struct snd_soc_codec *codec = codec_dai->codec;
 356         u16 iface = 0;
 357
 358         /* set master/slave audio interface */
 359         switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
 360         case SND_SOC_DAIFMT_CBM_CFM:
 361                 iface |= 0x0040;
 362                 break;
 363         case SND_SOC_DAIFMT_CBS_CFS:
 364                 break;
 365         default:
 366                 return -EINVAL;
 367         }
 368
 369         /* interface format */
 370         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
 371         case SND_SOC_DAIFMT_I2S:
 372                 iface |= 0x0002;
 373                 break;
 374         case SND_SOC_DAIFMT_RIGHT_J:
 375                 break;
 376         case SND_SOC_DAIFMT_LEFT_J:
 377                 iface |= 0x0001;
 378                 break;
 379         case SND_SOC_DAIFMT_DSP_A:
 380                 iface |= 0x0003;
 381                 break;
 382         case SND_SOC_DAIFMT_DSP_B:
 383                 iface |= 0x0013;
 384                 break;
 385         default:
 386                 return -EINVAL;
 387         }
 388
 389         /* clock inversion */
 390         switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
 391         case SND_SOC_DAIFMT_NB_NF:
 392                 break;
 393         case SND_SOC_DAIFMT_IB_IF:
 394                 iface |= 0x0090;
 395                 break;
 396         case SND_SOC_DAIFMT_IB_NF:
 397                 iface |= 0x0080;
 398                 break;
 399         case SND_SOC_DAIFMT_NB_IF:
 400                 iface |= 0x0010;
 401                 break;
 402         default:
 403                 return -EINVAL;
 404         }
 405
 406         /* set iface */
 407         wm8731_write(codec, WM8731_IFACE, iface);
 408         return 0;
 409 }
 410
 411 static int wm8731_set_bias_level(struct snd_soc_codec *codec,
 412                                  enum snd_soc_bias_level level)
 413 {
 414         u16 reg = wm8731_read_reg_cache(codec, WM8731_PWR) & 0xff7f;
 415
 416         switch (level) {
 417         case SND_SOC_BIAS_ON:
 418                 /* vref/mid, osc on, dac unmute */
 419                 wm8731_write(codec, WM8731_PWR, reg);
 420                 break;
 421         case SND_SOC_BIAS_PREPARE:
 422                 break;
 423         case SND_SOC_BIAS_STANDBY:
 424                 /* everything off except vref/vmid, */
 425                 wm8731_write(codec, WM8731_PWR, reg | 0x0040);
 426                 break;
 427         case SND_SOC_BIAS_OFF:
 428                 /* everything off, dac mute, inactive */
 429                 wm8731_write(codec, WM8731_ACTIVE, 0x0);
 430                 wm8731_write(codec, WM8731_PWR, 0xffff);
 431                 break;
 432         }
 433         codec->bias_level = level;
 434         return 0;
 435 }
 436
 437 #define WM8731_RATES (SNDRV_PCM_RATE_8000 | SNDRV_PCM_RATE_11025 |\
 438                 SNDRV_PCM_RATE_16000 | SNDRV_PCM_RATE_22050 |\
 439                 SNDRV_PCM_RATE_32000 | SNDRV_PCM_RATE_44100 |\
 440                 SNDRV_PCM_RATE_48000 | SNDRV_PCM_RATE_88200 |\
 441                 SNDRV_PCM_RATE_96000)
 442
 443 #define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
 444         SNDRV_PCM_FMTBIT_S24_LE)
 445
 446 struct snd_soc_dai wm8731_dai = {
 447         .name = "WM8731",
 448         .playback = {
 449                 .stream_name = "Playback",
 450                 .channels_min = 1,
 451                 .channels_max = 2,
 452                 .rates = WM8731_RATES,
 453                 .formats = WM8731_FORMATS,},
 454         .capture = {
 455                 .stream_name = "Capture",
 456                 .channels_min = 1,
 457                 .channels_max = 2,
 458                 .rates = WM8731_RATES,
 459                 .formats = WM8731_FORMATS,},
 460         .ops = {
 461                 .prepare = wm8731_pcm_prepare,
 462                 .hw_params = wm8731_hw_params,
 463                 .shutdown = wm8731_shutdown,
 464         },
 465         .dai_ops = {
 466                 .digital_mute = wm8731_mute,
 467                 .set_sysclk = wm8731_set_dai_sysclk,
 468                 .set_fmt = wm8731_set_dai_fmt,
 469         }
 470 };
 471 EXPORT_SYMBOL_GPL(wm8731_dai);
 472
 473 static int wm8731_suspend(struct platform_device *pdev, pm_message_t state)
 474 {
 475         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 476         struct snd_soc_codec *codec = socdev->codec;
 477
 478         wm8731_write(codec, WM8731_ACTIVE, 0x0);
 479         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 480         return 0;
 481 }
 482
 483 static int wm8731_resume(struct platform_device *pdev)
 484 {
 485         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 486         struct snd_soc_codec *codec = socdev->codec;
 487         int i;
 488         u8 data[2];
 489         u16 *cache = codec->reg_cache;
 490
 491         /* Sync reg_cache with the hardware */
 492         for (i = 0; i < ARRAY_SIZE(wm8731_reg); i++) {
 493                 data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
 494                 data[1] = cache[i] & 0x00ff;
 495                 codec->hw_write(codec->control_data, data, 2);
 496         }
 497         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 498         wm8731_set_bias_level(codec, codec->suspend_bias_level);
 499         return 0;
 500 }
 501
 502 /*
 503  * initialise the WM8731 driver
 504  * register the mixer and dsp interfaces with the kernel
 505  */
 506 static int wm8731_init(struct snd_soc_device *socdev)
 507 {
 508         struct snd_soc_codec *codec = socdev->codec;
 509         int reg, ret = 0;
 510
 511         codec->name = "WM8731";
 512         codec->owner = THIS_MODULE;
 513         codec->read = wm8731_read_reg_cache;
 514         codec->write = wm8731_write;
 515         codec->set_bias_level = wm8731_set_bias_level;
 516         codec->dai = &wm8731_dai;
 517         codec->num_dai = 1;
 518         codec->reg_cache_size = ARRAY_SIZE(wm8731_reg);
 519         codec->reg_cache = kmemdup(wm8731_reg, sizeof(wm8731_reg), GFP_KERNEL);
 520         if (codec->reg_cache == NULL)
 521                 return -ENOMEM;
 522
 523         wm8731_reset(codec);
 524
 525         /* register pcms */
 526         ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
 527         if (ret < 0) {
 528                 printk(KERN_ERR "wm8731: failed to create pcms\n");
 529                 goto pcm_err;
 530         }
 531
 532         /* power on device */
 533         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
 534
 535         /* set the update bits */
 536         reg = wm8731_read_reg_cache(codec, WM8731_LOUT1V);
 537         wm8731_write(codec, WM8731_LOUT1V, reg & ~0x0100);
 538         reg = wm8731_read_reg_cache(codec, WM8731_ROUT1V);
 539         wm8731_write(codec, WM8731_ROUT1V, reg & ~0x0100);
 540         reg = wm8731_read_reg_cache(codec, WM8731_LINVOL);
 541         wm8731_write(codec, WM8731_LINVOL, reg & ~0x0100);
 542         reg = wm8731_read_reg_cache(codec, WM8731_RINVOL);
 543         wm8731_write(codec, WM8731_RINVOL, reg & ~0x0100);
 544
 545         wm8731_add_controls(codec);
 546         wm8731_add_widgets(codec);
 547         ret = snd_soc_register_card(socdev);
 548         if (ret < 0) {
 549                 printk(KERN_ERR "wm8731: failed to register card\n");
 550                 goto card_err;
 551         }
 552
 553         return ret;
 554
 555 card_err:
 556         snd_soc_free_pcms(socdev);
 557         snd_soc_dapm_free(socdev);
 558 pcm_err:
 559         kfree(codec->reg_cache);
 560         return ret;
 561 }
 562
 563 static struct snd_soc_device *wm8731_socdev;
 564
 565 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 566
 567 /*
 568  * WM8731 2 wire address is determined by GPIO5
 569  * state during powerup.
 570  *    low  = 0x1a
 571  *    high = 0x1b
 572  */
 573
 574 static int wm8731_i2c_probe(struct i2c_client *i2c,
 575                             const struct i2c_device_id *id)
 576 {
 577         struct snd_soc_device *socdev = wm8731_socdev;
 578         struct snd_soc_codec *codec = socdev->codec;
 579         int ret;
 580
 581         i2c_set_clientdata(i2c, codec);
 582         codec->control_data = i2c;
 583
 584         ret = wm8731_init(socdev);
 585         if (ret < 0)
 586                 pr_err("failed to initialise WM8731\n");
 587
 588         return ret;
 589 }
 590
 591 static int wm8731_i2c_remove(struct i2c_client *client)
 592 {
 593         struct snd_soc_codec *codec = i2c_get_clientdata(client);
 594         kfree(codec->reg_cache);
 595         return 0;
 596 }
 597
 598 static const struct i2c_device_id wm8731_i2c_id[] = {
 599         { "wm8731", 0 },
 600         { }
 601 };
 602 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
 603
 604 static struct i2c_driver wm8731_i2c_driver = {
 605         .driver = {
 606                 .name = "WM8731 I2C Codec",
 607                 .owner = THIS_MODULE,
 608         },
 609         .probe =    wm8731_i2c_probe,
 610         .remove =   wm8731_i2c_remove,
 611         .id_table = wm8731_i2c_id,
 612 };
 613
 614 static int wm8731_add_i2c_device(struct platform_device *pdev,
 615                                  const struct wm8731_setup_data *setup)
 616 {
 617         struct i2c_board_info info;
 618         struct i2c_adapter *adapter;
 619         struct i2c_client *client;
 620         int ret;
 621
 622         ret = i2c_add_driver(&wm8731_i2c_driver);
 623         if (ret != 0) {
 624                 dev_err(&pdev->dev, "can't add i2c driver\n");
 625                 return ret;
 626         }
 627
 628         memset(&info, 0, sizeof(struct i2c_board_info));
 629         info.addr = setup->i2c_address;
 630         strlcpy(info.type, "wm8731", I2C_NAME_SIZE);
 631
 632         adapter = i2c_get_adapter(setup->i2c_bus);
 633         if (!adapter) {
 634                 dev_err(&pdev->dev, "can't get i2c adapter %d\n",
 635                         setup->i2c_bus);
 636                 goto err_driver;
 637         }
 638
 639         client = i2c_new_device(adapter, &info);
 640         i2c_put_adapter(adapter);
 641         if (!client) {
 642                 dev_err(&pdev->dev, "can't add i2c device at 0x%x\n",
 643                         (unsigned int)info.addr);
 644                 goto err_driver;
 645         }
 646
 647         return 0;
 648
 649 err_driver:
 650         i2c_del_driver(&wm8731_i2c_driver);
 651         return -ENODEV;
 652 }
 653 #endif
 654
 655 #if defined(CONFIG_SPI_MASTER)
 656 static int __devinit wm8731_spi_probe(struct spi_device *spi)
 657 {
 658         struct snd_soc_device *socdev = wm8731_socdev;
 659         struct snd_soc_codec *codec = socdev->codec;
 660         int ret;
 661
 662         codec->control_data = spi;
 663
 664         ret = wm8731_init(socdev);
 665         if (ret < 0)
 666                 dev_err(&spi->dev, "failed to initialise WM8731\n");
 667
 668         return ret;
 669 }
 670
 671 static int __devexit wm8731_spi_remove(struct spi_device *spi)
 672 {
 673         return 0;
 674 }
 675
 676 static struct spi_driver wm8731_spi_driver = {
 677         .driver = {
 678                 .name   = "wm8731",
 679                 .bus    = &spi_bus_type,
 680                 .owner  = THIS_MODULE,
 681         },
 682         .probe          = wm8731_spi_probe,
 683         .remove         = __devexit_p(wm8731_spi_remove),
 684 };
 685
 686 static int wm8731_spi_write(struct spi_device *spi, const char *data, int len)
 687 {
 688         struct spi_transfer t;
 689         struct spi_message m;
 690         u8 msg[2];
 691
 692         if (len <= 0)
 693                 return 0;
 694
 695         msg[0] = data[0];
 696         msg[1] = data[1];
 697
 698         spi_message_init(&m);
 699         memset(&t, 0, (sizeof t));
 700
 701         t.tx_buf = &msg[0];
 702         t.len = len;
 703
 704         spi_message_add_tail(&t, &m);
 705         spi_sync(spi, &m);
 706
 707         return len;
 708 }
 709 #endif /* CONFIG_SPI_MASTER */
 710
 711 static int wm8731_probe(struct platform_device *pdev)
 712 {
 713         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 714         struct wm8731_setup_data *setup;
 715         struct snd_soc_codec *codec;
 716         struct wm8731_priv *wm8731;
 717         int ret = 0;
 718
 719         pr_info("WM8731 Audio Codec %s", WM8731_VERSION);
 720
 721         setup = socdev->codec_data;
 722         codec = kzalloc(sizeof(struct snd_soc_codec), GFP_KERNEL);
 723         if (codec == NULL)
 724                 return -ENOMEM;
 725
 726         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
 727         if (wm8731 == NULL) {
 728                 kfree(codec);
 729                 return -ENOMEM;
 730         }
 731
 732         codec->private_data = wm8731;
 733         socdev->codec = codec;
 734         mutex_init(&codec->mutex);
 735         INIT_LIST_HEAD(&codec->dapm_widgets);
 736         INIT_LIST_HEAD(&codec->dapm_paths);
 737
 738         wm8731_socdev = socdev;
 739         ret = -ENODEV;
 740
 741 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 742         if (setup->i2c_address) {
 743                 codec->hw_write = (hw_write_t)i2c_master_send;
 744                 ret = wm8731_add_i2c_device(pdev, setup);
 745         }
 746 #endif
 747 #if defined(CONFIG_SPI_MASTER)
 748         if (setup->spi) {
 749                 codec->hw_write = (hw_write_t)wm8731_spi_write;
 750                 ret = spi_register_driver(&wm8731_spi_driver);
 751                 if (ret != 0)
 752                         printk(KERN_ERR "can't add spi driver");
 753         }
 754 #endif
 755
 756         if (ret != 0) {
 757                 kfree(codec->private_data);
 758                 kfree(codec);
 759         }
 760         return ret;
 761 }
 762
 763 /* power down chip */
 764 static int wm8731_remove(struct platform_device *pdev)
 765 {
 766         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
 767         struct snd_soc_codec *codec = socdev->codec;
 768
 769         if (codec->control_data)
 770                 wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
 771
 772         snd_soc_free_pcms(socdev);
 773         snd_soc_dapm_free(socdev);
 774 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
 775         i2c_unregister_device(codec->control_data);
 776         i2c_del_driver(&wm8731_i2c_driver);
 777 #endif
 778 #if defined(CONFIG_SPI_MASTER)
 779         spi_unregister_driver(&wm8731_spi_driver);
 780 #endif
 781         kfree(codec->private_data);
 782         kfree(codec);
 783
 784         return 0;
 785 }
 786
 787 struct snd_soc_codec_device soc_codec_dev_wm8731 = {
 788         .probe =        wm8731_probe,
 789         .remove =       wm8731_remove,
 790         .suspend =      wm8731_suspend,
 791         .resume =       wm8731_resume,
 792 };
 793 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8731);
 794
 795 MODULE_DESCRIPTION("ASoC WM8731 driver");
 796 MODULE_AUTHOR("Richard Purdie");
 797 MODULE_LICENSE("GPL");