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