ASoC: Remove redundant snd_soc_dapm_new_widgets() calls
[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         return 0;
163 }
164
165 struct _coeff_div {
166         u32 mclk;
167         u32 rate;
168         u16 fs;
169         u8 sr:4;
170         u8 bosr:1;
171         u8 usb:1;
172 };
173
174 /* codec mclk clock divider coefficients */
175 static const struct _coeff_div coeff_div[] = {
176         /* 48k */
177         {12288000, 48000, 256, 0x0, 0x0, 0x0},
178         {18432000, 48000, 384, 0x0, 0x1, 0x0},
179         {12000000, 48000, 250, 0x0, 0x0, 0x1},
180
181         /* 32k */
182         {12288000, 32000, 384, 0x6, 0x0, 0x0},
183         {18432000, 32000, 576, 0x6, 0x1, 0x0},
184         {12000000, 32000, 375, 0x6, 0x0, 0x1},
185
186         /* 8k */
187         {12288000, 8000, 1536, 0x3, 0x0, 0x0},
188         {18432000, 8000, 2304, 0x3, 0x1, 0x0},
189         {11289600, 8000, 1408, 0xb, 0x0, 0x0},
190         {16934400, 8000, 2112, 0xb, 0x1, 0x0},
191         {12000000, 8000, 1500, 0x3, 0x0, 0x1},
192
193         /* 96k */
194         {12288000, 96000, 128, 0x7, 0x0, 0x0},
195         {18432000, 96000, 192, 0x7, 0x1, 0x0},
196         {12000000, 96000, 125, 0x7, 0x0, 0x1},
197
198         /* 44.1k */
199         {11289600, 44100, 256, 0x8, 0x0, 0x0},
200         {16934400, 44100, 384, 0x8, 0x1, 0x0},
201         {12000000, 44100, 272, 0x8, 0x1, 0x1},
202
203         /* 88.2k */
204         {11289600, 88200, 128, 0xf, 0x0, 0x0},
205         {16934400, 88200, 192, 0xf, 0x1, 0x0},
206         {12000000, 88200, 136, 0xf, 0x1, 0x1},
207 };
208
209 static inline int get_coeff(int mclk, int rate)
210 {
211         int i;
212
213         for (i = 0; i < ARRAY_SIZE(coeff_div); i++) {
214                 if (coeff_div[i].rate == rate && coeff_div[i].mclk == mclk)
215                         return i;
216         }
217         return 0;
218 }
219
220 static int wm8731_hw_params(struct snd_pcm_substream *substream,
221                             struct snd_pcm_hw_params *params,
222                             struct snd_soc_dai *dai)
223 {
224         struct snd_soc_pcm_runtime *rtd = substream->private_data;
225         struct snd_soc_device *socdev = rtd->socdev;
226         struct snd_soc_codec *codec = socdev->card->codec;
227         struct wm8731_priv *wm8731 = codec->private_data;
228         u16 iface = snd_soc_read(codec, WM8731_IFACE) & 0xfff3;
229         int i = get_coeff(wm8731->sysclk, params_rate(params));
230         u16 srate = (coeff_div[i].sr << 2) |
231                 (coeff_div[i].bosr << 1) | coeff_div[i].usb;
232
233         snd_soc_write(codec, WM8731_SRATE, srate);
234
235         /* bit size */
236         switch (params_format(params)) {
237         case SNDRV_PCM_FORMAT_S16_LE:
238                 break;
239         case SNDRV_PCM_FORMAT_S20_3LE:
240                 iface |= 0x0004;
241                 break;
242         case SNDRV_PCM_FORMAT_S24_LE:
243                 iface |= 0x0008;
244                 break;
245         }
246
247         snd_soc_write(codec, WM8731_IFACE, iface);
248         return 0;
249 }
250
251 static int wm8731_pcm_prepare(struct snd_pcm_substream *substream,
252                               struct snd_soc_dai *dai)
253 {
254         struct snd_soc_pcm_runtime *rtd = substream->private_data;
255         struct snd_soc_device *socdev = rtd->socdev;
256         struct snd_soc_codec *codec = socdev->card->codec;
257
258         /* set active */
259         snd_soc_write(codec, WM8731_ACTIVE, 0x0001);
260
261         return 0;
262 }
263
264 static void wm8731_shutdown(struct snd_pcm_substream *substream,
265                             struct snd_soc_dai *dai)
266 {
267         struct snd_soc_pcm_runtime *rtd = substream->private_data;
268         struct snd_soc_device *socdev = rtd->socdev;
269         struct snd_soc_codec *codec = socdev->card->codec;
270
271         /* deactivate */
272         if (!codec->active) {
273                 udelay(50);
274                 snd_soc_write(codec, WM8731_ACTIVE, 0x0);
275         }
276 }
277
278 static int wm8731_mute(struct snd_soc_dai *dai, int mute)
279 {
280         struct snd_soc_codec *codec = dai->codec;
281         u16 mute_reg = snd_soc_read(codec, WM8731_APDIGI) & 0xfff7;
282
283         if (mute)
284                 snd_soc_write(codec, WM8731_APDIGI, mute_reg | 0x8);
285         else
286                 snd_soc_write(codec, WM8731_APDIGI, mute_reg);
287         return 0;
288 }
289
290 static int wm8731_set_dai_sysclk(struct snd_soc_dai *codec_dai,
291                 int clk_id, unsigned int freq, int dir)
292 {
293         struct snd_soc_codec *codec = codec_dai->codec;
294         struct wm8731_priv *wm8731 = codec->private_data;
295
296         switch (freq) {
297         case 11289600:
298         case 12000000:
299         case 12288000:
300         case 16934400:
301         case 18432000:
302                 wm8731->sysclk = freq;
303                 return 0;
304         }
305         return -EINVAL;
306 }
307
308
309 static int wm8731_set_dai_fmt(struct snd_soc_dai *codec_dai,
310                 unsigned int fmt)
311 {
312         struct snd_soc_codec *codec = codec_dai->codec;
313         u16 iface = 0;
314
315         /* set master/slave audio interface */
316         switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
317         case SND_SOC_DAIFMT_CBM_CFM:
318                 iface |= 0x0040;
319                 break;
320         case SND_SOC_DAIFMT_CBS_CFS:
321                 break;
322         default:
323                 return -EINVAL;
324         }
325
326         /* interface format */
327         switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
328         case SND_SOC_DAIFMT_I2S:
329                 iface |= 0x0002;
330                 break;
331         case SND_SOC_DAIFMT_RIGHT_J:
332                 break;
333         case SND_SOC_DAIFMT_LEFT_J:
334                 iface |= 0x0001;
335                 break;
336         case SND_SOC_DAIFMT_DSP_A:
337                 iface |= 0x0003;
338                 break;
339         case SND_SOC_DAIFMT_DSP_B:
340                 iface |= 0x0013;
341                 break;
342         default:
343                 return -EINVAL;
344         }
345
346         /* clock inversion */
347         switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
348         case SND_SOC_DAIFMT_NB_NF:
349                 break;
350         case SND_SOC_DAIFMT_IB_IF:
351                 iface |= 0x0090;
352                 break;
353         case SND_SOC_DAIFMT_IB_NF:
354                 iface |= 0x0080;
355                 break;
356         case SND_SOC_DAIFMT_NB_IF:
357                 iface |= 0x0010;
358                 break;
359         default:
360                 return -EINVAL;
361         }
362
363         /* set iface */
364         snd_soc_write(codec, WM8731_IFACE, iface);
365         return 0;
366 }
367
368 static int wm8731_set_bias_level(struct snd_soc_codec *codec,
369                                  enum snd_soc_bias_level level)
370 {
371         u16 reg;
372
373         switch (level) {
374         case SND_SOC_BIAS_ON:
375                 break;
376         case SND_SOC_BIAS_PREPARE:
377                 break;
378         case SND_SOC_BIAS_STANDBY:
379                 /* Clear PWROFF, gate CLKOUT, everything else as-is */
380                 reg = snd_soc_read(codec, WM8731_PWR) & 0xff7f;
381                 snd_soc_write(codec, WM8731_PWR, reg | 0x0040);
382                 break;
383         case SND_SOC_BIAS_OFF:
384                 snd_soc_write(codec, WM8731_ACTIVE, 0x0);
385                 snd_soc_write(codec, WM8731_PWR, 0xffff);
386                 break;
387         }
388         codec->bias_level = level;
389         return 0;
390 }
391
392 #define WM8731_RATES (SNDRV_PCM_RATE_8000 | SNDRV_PCM_RATE_11025 |\
393                 SNDRV_PCM_RATE_16000 | SNDRV_PCM_RATE_22050 |\
394                 SNDRV_PCM_RATE_32000 | SNDRV_PCM_RATE_44100 |\
395                 SNDRV_PCM_RATE_48000 | SNDRV_PCM_RATE_88200 |\
396                 SNDRV_PCM_RATE_96000)
397
398 #define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
399         SNDRV_PCM_FMTBIT_S24_LE)
400
401 static struct snd_soc_dai_ops wm8731_dai_ops = {
402         .prepare        = wm8731_pcm_prepare,
403         .hw_params      = wm8731_hw_params,
404         .shutdown       = wm8731_shutdown,
405         .digital_mute   = wm8731_mute,
406         .set_sysclk     = wm8731_set_dai_sysclk,
407         .set_fmt        = wm8731_set_dai_fmt,
408 };
409
410 struct snd_soc_dai wm8731_dai = {
411         .name = "WM8731",
412         .playback = {
413                 .stream_name = "Playback",
414                 .channels_min = 1,
415                 .channels_max = 2,
416                 .rates = WM8731_RATES,
417                 .formats = WM8731_FORMATS,},
418         .capture = {
419                 .stream_name = "Capture",
420                 .channels_min = 1,
421                 .channels_max = 2,
422                 .rates = WM8731_RATES,
423                 .formats = WM8731_FORMATS,},
424         .ops = &wm8731_dai_ops,
425         .symmetric_rates = 1,
426 };
427 EXPORT_SYMBOL_GPL(wm8731_dai);
428
429 #ifdef CONFIG_PM
430 static int wm8731_suspend(struct platform_device *pdev, pm_message_t state)
431 {
432         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
433         struct snd_soc_codec *codec = socdev->card->codec;
434         struct wm8731_priv *wm8731 = codec->private_data;
435
436         snd_soc_write(codec, WM8731_ACTIVE, 0x0);
437         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
438         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies),
439                                wm8731->supplies);
440         return 0;
441 }
442
443 static int wm8731_resume(struct platform_device *pdev)
444 {
445         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
446         struct snd_soc_codec *codec = socdev->card->codec;
447         struct wm8731_priv *wm8731 = codec->private_data;
448         int i, ret;
449         u8 data[2];
450         u16 *cache = codec->reg_cache;
451
452         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
453                                     wm8731->supplies);
454         if (ret != 0)
455                 return ret;
456
457         /* Sync reg_cache with the hardware */
458         for (i = 0; i < ARRAY_SIZE(wm8731_reg); i++) {
459                 data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
460                 data[1] = cache[i] & 0x00ff;
461                 codec->hw_write(codec->control_data, data, 2);
462         }
463         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
464         wm8731_set_bias_level(codec, codec->suspend_bias_level);
465
466         return 0;
467 }
468 #else
469 #define wm8731_suspend NULL
470 #define wm8731_resume NULL
471 #endif
472
473 static int wm8731_probe(struct platform_device *pdev)
474 {
475         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
476         struct snd_soc_codec *codec;
477         int ret = 0;
478
479         if (wm8731_codec == NULL) {
480                 dev_err(&pdev->dev, "Codec device not registered\n");
481                 return -ENODEV;
482         }
483
484         socdev->card->codec = wm8731_codec;
485         codec = wm8731_codec;
486
487         /* register pcms */
488         ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
489         if (ret < 0) {
490                 dev_err(codec->dev, "failed to create pcms: %d\n", ret);
491                 goto pcm_err;
492         }
493
494         snd_soc_add_controls(codec, wm8731_snd_controls,
495                              ARRAY_SIZE(wm8731_snd_controls));
496         wm8731_add_widgets(codec);
497
498         return ret;
499
500 pcm_err:
501         return ret;
502 }
503
504 /* power down chip */
505 static int wm8731_remove(struct platform_device *pdev)
506 {
507         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
508
509         snd_soc_free_pcms(socdev);
510         snd_soc_dapm_free(socdev);
511
512         return 0;
513 }
514
515 struct snd_soc_codec_device soc_codec_dev_wm8731 = {
516         .probe =        wm8731_probe,
517         .remove =       wm8731_remove,
518         .suspend =      wm8731_suspend,
519         .resume =       wm8731_resume,
520 };
521 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8731);
522
523 static int wm8731_register(struct wm8731_priv *wm8731,
524                            enum snd_soc_control_type control)
525 {
526         int ret, i;
527         struct snd_soc_codec *codec = &wm8731->codec;
528
529         if (wm8731_codec) {
530                 dev_err(codec->dev, "Another WM8731 is registered\n");
531                 ret = -EINVAL;
532                 goto err;
533         }
534
535         mutex_init(&codec->mutex);
536         INIT_LIST_HEAD(&codec->dapm_widgets);
537         INIT_LIST_HEAD(&codec->dapm_paths);
538
539         codec->private_data = wm8731;
540         codec->name = "WM8731";
541         codec->owner = THIS_MODULE;
542         codec->bias_level = SND_SOC_BIAS_OFF;
543         codec->set_bias_level = wm8731_set_bias_level;
544         codec->dai = &wm8731_dai;
545         codec->num_dai = 1;
546         codec->reg_cache_size = WM8731_CACHEREGNUM;
547         codec->reg_cache = &wm8731->reg_cache;
548
549         memcpy(codec->reg_cache, wm8731_reg, sizeof(wm8731_reg));
550
551         ret = snd_soc_codec_set_cache_io(codec, 7, 9, control);
552         if (ret < 0) {
553                 dev_err(codec->dev, "Failed to set cache I/O: %d\n", ret);
554                 goto err;
555         }
556
557         for (i = 0; i < ARRAY_SIZE(wm8731->supplies); i++)
558                 wm8731->supplies[i].supply = wm8731_supply_names[i];
559
560         ret = regulator_bulk_get(codec->dev, ARRAY_SIZE(wm8731->supplies),
561                                  wm8731->supplies);
562         if (ret != 0) {
563                 dev_err(codec->dev, "Failed to request supplies: %d\n", ret);
564                 goto err;
565         }
566
567         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
568                                     wm8731->supplies);
569         if (ret != 0) {
570                 dev_err(codec->dev, "Failed to enable supplies: %d\n", ret);
571                 goto err_regulator_get;
572         }
573
574         ret = wm8731_reset(codec);
575         if (ret < 0) {
576                 dev_err(codec->dev, "Failed to issue reset: %d\n", ret);
577                 goto err_regulator_enable;
578         }
579
580         wm8731_dai.dev = codec->dev;
581
582         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
583
584         /* Latch the update bits */
585         snd_soc_update_bits(codec, WM8731_LOUT1V, 0x100, 0);
586         snd_soc_update_bits(codec, WM8731_ROUT1V, 0x100, 0);
587         snd_soc_update_bits(codec, WM8731_LINVOL, 0x100, 0);
588         snd_soc_update_bits(codec, WM8731_RINVOL, 0x100, 0);
589
590         /* Disable bypass path by default */
591         snd_soc_update_bits(codec, WM8731_APANA, 0x4, 0);
592
593         wm8731_codec = codec;
594
595         ret = snd_soc_register_codec(codec);
596         if (ret != 0) {
597                 dev_err(codec->dev, "Failed to register codec: %d\n", ret);
598                 goto err_regulator_enable;
599         }
600
601         ret = snd_soc_register_dai(&wm8731_dai);
602         if (ret != 0) {
603                 dev_err(codec->dev, "Failed to register DAI: %d\n", ret);
604                 snd_soc_unregister_codec(codec);
605                 goto err_codec;
606         }
607
608         return 0;
609
610 err_codec:
611         snd_soc_unregister_codec(codec);
612 err_regulator_enable:
613         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
614 err_regulator_get:
615         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
616 err:
617         kfree(wm8731);
618         return ret;
619 }
620
621 static void wm8731_unregister(struct wm8731_priv *wm8731)
622 {
623         wm8731_set_bias_level(&wm8731->codec, SND_SOC_BIAS_OFF);
624         snd_soc_unregister_dai(&wm8731_dai);
625         snd_soc_unregister_codec(&wm8731->codec);
626         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
627         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
628         kfree(wm8731);
629         wm8731_codec = NULL;
630 }
631
632 #if defined(CONFIG_SPI_MASTER)
633 static int __devinit wm8731_spi_probe(struct spi_device *spi)
634 {
635         struct snd_soc_codec *codec;
636         struct wm8731_priv *wm8731;
637
638         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
639         if (wm8731 == NULL)
640                 return -ENOMEM;
641
642         codec = &wm8731->codec;
643         codec->control_data = spi;
644         codec->dev = &spi->dev;
645
646         dev_set_drvdata(&spi->dev, wm8731);
647
648         return wm8731_register(wm8731, SND_SOC_SPI);
649 }
650
651 static int __devexit wm8731_spi_remove(struct spi_device *spi)
652 {
653         struct wm8731_priv *wm8731 = dev_get_drvdata(&spi->dev);
654
655         wm8731_unregister(wm8731);
656
657         return 0;
658 }
659
660 static struct spi_driver wm8731_spi_driver = {
661         .driver = {
662                 .name   = "wm8731",
663                 .bus    = &spi_bus_type,
664                 .owner  = THIS_MODULE,
665         },
666         .probe          = wm8731_spi_probe,
667         .remove         = __devexit_p(wm8731_spi_remove),
668 };
669 #endif /* CONFIG_SPI_MASTER */
670
671 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
672 static __devinit int wm8731_i2c_probe(struct i2c_client *i2c,
673                                       const struct i2c_device_id *id)
674 {
675         struct wm8731_priv *wm8731;
676         struct snd_soc_codec *codec;
677
678         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
679         if (wm8731 == NULL)
680                 return -ENOMEM;
681
682         codec = &wm8731->codec;
683
684         i2c_set_clientdata(i2c, wm8731);
685         codec->control_data = i2c;
686
687         codec->dev = &i2c->dev;
688
689         return wm8731_register(wm8731, SND_SOC_I2C);
690 }
691
692 static __devexit int wm8731_i2c_remove(struct i2c_client *client)
693 {
694         struct wm8731_priv *wm8731 = i2c_get_clientdata(client);
695         wm8731_unregister(wm8731);
696         return 0;
697 }
698
699 static const struct i2c_device_id wm8731_i2c_id[] = {
700         { "wm8731", 0 },
701         { }
702 };
703 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
704
705 static struct i2c_driver wm8731_i2c_driver = {
706         .driver = {
707                 .name = "WM8731 I2C Codec",
708                 .owner = THIS_MODULE,
709         },
710         .probe =    wm8731_i2c_probe,
711         .remove =   __devexit_p(wm8731_i2c_remove),
712         .id_table = wm8731_i2c_id,
713 };
714 #endif
715
716 static int __init wm8731_modinit(void)
717 {
718         int ret;
719 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
720         ret = i2c_add_driver(&wm8731_i2c_driver);
721         if (ret != 0) {
722                 printk(KERN_ERR "Failed to register WM8731 I2C driver: %d\n",
723                        ret);
724         }
725 #endif
726 #if defined(CONFIG_SPI_MASTER)
727         ret = spi_register_driver(&wm8731_spi_driver);
728         if (ret != 0) {
729                 printk(KERN_ERR "Failed to register WM8731 SPI driver: %d\n",
730                        ret);
731         }
732 #endif
733         return 0;
734 }
735 module_init(wm8731_modinit);
736
737 static void __exit wm8731_exit(void)
738 {
739 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
740         i2c_del_driver(&wm8731_i2c_driver);
741 #endif
742 #if defined(CONFIG_SPI_MASTER)
743         spi_unregister_driver(&wm8731_spi_driver);
744 #endif
745 }
746 module_exit(wm8731_exit);
747
748 MODULE_DESCRIPTION("ASoC WM8731 driver");
749 MODULE_AUTHOR("Richard Purdie");
750 MODULE_LICENSE("GPL");