ASoC: Remove unneeded suspend bias managment from CODEC drivers
[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 = snd_soc_codec_get_drvdata(codec);
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 = snd_soc_codec_get_drvdata(codec);
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_96000
393
394 #define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
395         SNDRV_PCM_FMTBIT_S24_LE)
396
397 static struct snd_soc_dai_ops wm8731_dai_ops = {
398         .prepare        = wm8731_pcm_prepare,
399         .hw_params      = wm8731_hw_params,
400         .shutdown       = wm8731_shutdown,
401         .digital_mute   = wm8731_mute,
402         .set_sysclk     = wm8731_set_dai_sysclk,
403         .set_fmt        = wm8731_set_dai_fmt,
404 };
405
406 struct snd_soc_dai wm8731_dai = {
407         .name = "WM8731",
408         .playback = {
409                 .stream_name = "Playback",
410                 .channels_min = 1,
411                 .channels_max = 2,
412                 .rates = WM8731_RATES,
413                 .formats = WM8731_FORMATS,},
414         .capture = {
415                 .stream_name = "Capture",
416                 .channels_min = 1,
417                 .channels_max = 2,
418                 .rates = WM8731_RATES,
419                 .formats = WM8731_FORMATS,},
420         .ops = &wm8731_dai_ops,
421         .symmetric_rates = 1,
422 };
423 EXPORT_SYMBOL_GPL(wm8731_dai);
424
425 #ifdef CONFIG_PM
426 static int wm8731_suspend(struct platform_device *pdev, pm_message_t state)
427 {
428         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
429         struct snd_soc_codec *codec = socdev->card->codec;
430         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
431
432         snd_soc_write(codec, WM8731_ACTIVE, 0x0);
433         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
434         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies),
435                                wm8731->supplies);
436         return 0;
437 }
438
439 static int wm8731_resume(struct platform_device *pdev)
440 {
441         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
442         struct snd_soc_codec *codec = socdev->card->codec;
443         struct wm8731_priv *wm8731 = snd_soc_codec_get_drvdata(codec);
444         int i, ret;
445         u8 data[2];
446         u16 *cache = codec->reg_cache;
447
448         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
449                                     wm8731->supplies);
450         if (ret != 0)
451                 return ret;
452
453         /* Sync reg_cache with the hardware */
454         for (i = 0; i < ARRAY_SIZE(wm8731_reg); i++) {
455                 if (cache[i] == wm8731_reg[i])
456                         continue;
457
458                 data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
459                 data[1] = cache[i] & 0x00ff;
460                 codec->hw_write(codec->control_data, data, 2);
461         }
462         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
463
464         return 0;
465 }
466 #else
467 #define wm8731_suspend NULL
468 #define wm8731_resume NULL
469 #endif
470
471 static int wm8731_probe(struct platform_device *pdev)
472 {
473         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
474         struct snd_soc_codec *codec;
475         int ret = 0;
476
477         if (wm8731_codec == NULL) {
478                 dev_err(&pdev->dev, "Codec device not registered\n");
479                 return -ENODEV;
480         }
481
482         socdev->card->codec = wm8731_codec;
483         codec = wm8731_codec;
484
485         /* register pcms */
486         ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
487         if (ret < 0) {
488                 dev_err(codec->dev, "failed to create pcms: %d\n", ret);
489                 goto pcm_err;
490         }
491
492         snd_soc_add_controls(codec, wm8731_snd_controls,
493                              ARRAY_SIZE(wm8731_snd_controls));
494         wm8731_add_widgets(codec);
495
496         return ret;
497
498 pcm_err:
499         return ret;
500 }
501
502 /* power down chip */
503 static int wm8731_remove(struct platform_device *pdev)
504 {
505         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
506
507         snd_soc_free_pcms(socdev);
508         snd_soc_dapm_free(socdev);
509
510         return 0;
511 }
512
513 struct snd_soc_codec_device soc_codec_dev_wm8731 = {
514         .probe =        wm8731_probe,
515         .remove =       wm8731_remove,
516         .suspend =      wm8731_suspend,
517         .resume =       wm8731_resume,
518 };
519 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8731);
520
521 static int wm8731_register(struct wm8731_priv *wm8731,
522                            enum snd_soc_control_type control)
523 {
524         int ret, i;
525         struct snd_soc_codec *codec = &wm8731->codec;
526
527         if (wm8731_codec) {
528                 dev_err(codec->dev, "Another WM8731 is registered\n");
529                 ret = -EINVAL;
530                 goto err;
531         }
532
533         mutex_init(&codec->mutex);
534         INIT_LIST_HEAD(&codec->dapm_widgets);
535         INIT_LIST_HEAD(&codec->dapm_paths);
536
537         snd_soc_codec_set_drvdata(codec, wm8731);
538         codec->name = "WM8731";
539         codec->owner = THIS_MODULE;
540         codec->bias_level = SND_SOC_BIAS_OFF;
541         codec->set_bias_level = wm8731_set_bias_level;
542         codec->dai = &wm8731_dai;
543         codec->num_dai = 1;
544         codec->reg_cache_size = WM8731_CACHEREGNUM;
545         codec->reg_cache = &wm8731->reg_cache;
546
547         memcpy(codec->reg_cache, wm8731_reg, sizeof(wm8731_reg));
548
549         ret = snd_soc_codec_set_cache_io(codec, 7, 9, control);
550         if (ret < 0) {
551                 dev_err(codec->dev, "Failed to set cache I/O: %d\n", ret);
552                 goto err;
553         }
554
555         for (i = 0; i < ARRAY_SIZE(wm8731->supplies); i++)
556                 wm8731->supplies[i].supply = wm8731_supply_names[i];
557
558         ret = regulator_bulk_get(codec->dev, ARRAY_SIZE(wm8731->supplies),
559                                  wm8731->supplies);
560         if (ret != 0) {
561                 dev_err(codec->dev, "Failed to request supplies: %d\n", ret);
562                 goto err;
563         }
564
565         ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
566                                     wm8731->supplies);
567         if (ret != 0) {
568                 dev_err(codec->dev, "Failed to enable supplies: %d\n", ret);
569                 goto err_regulator_get;
570         }
571
572         ret = wm8731_reset(codec);
573         if (ret < 0) {
574                 dev_err(codec->dev, "Failed to issue reset: %d\n", ret);
575                 goto err_regulator_enable;
576         }
577
578         wm8731_dai.dev = codec->dev;
579
580         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
581
582         /* Latch the update bits */
583         snd_soc_update_bits(codec, WM8731_LOUT1V, 0x100, 0);
584         snd_soc_update_bits(codec, WM8731_ROUT1V, 0x100, 0);
585         snd_soc_update_bits(codec, WM8731_LINVOL, 0x100, 0);
586         snd_soc_update_bits(codec, WM8731_RINVOL, 0x100, 0);
587
588         /* Disable bypass path by default */
589         snd_soc_update_bits(codec, WM8731_APANA, 0x4, 0);
590
591         wm8731_codec = codec;
592
593         ret = snd_soc_register_codec(codec);
594         if (ret != 0) {
595                 dev_err(codec->dev, "Failed to register codec: %d\n", ret);
596                 goto err_regulator_enable;
597         }
598
599         ret = snd_soc_register_dai(&wm8731_dai);
600         if (ret != 0) {
601                 dev_err(codec->dev, "Failed to register DAI: %d\n", ret);
602                 snd_soc_unregister_codec(codec);
603                 goto err_codec;
604         }
605
606         return 0;
607
608 err_codec:
609         snd_soc_unregister_codec(codec);
610 err_regulator_enable:
611         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
612 err_regulator_get:
613         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
614 err:
615         kfree(wm8731);
616         return ret;
617 }
618
619 static void wm8731_unregister(struct wm8731_priv *wm8731)
620 {
621         wm8731_set_bias_level(&wm8731->codec, SND_SOC_BIAS_OFF);
622         snd_soc_unregister_dai(&wm8731_dai);
623         snd_soc_unregister_codec(&wm8731->codec);
624         regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
625         regulator_bulk_free(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
626         kfree(wm8731);
627         wm8731_codec = NULL;
628 }
629
630 #if defined(CONFIG_SPI_MASTER)
631 static int __devinit wm8731_spi_probe(struct spi_device *spi)
632 {
633         struct snd_soc_codec *codec;
634         struct wm8731_priv *wm8731;
635
636         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
637         if (wm8731 == NULL)
638                 return -ENOMEM;
639
640         codec = &wm8731->codec;
641         codec->control_data = spi;
642         codec->dev = &spi->dev;
643
644         dev_set_drvdata(&spi->dev, wm8731);
645
646         return wm8731_register(wm8731, SND_SOC_SPI);
647 }
648
649 static int __devexit wm8731_spi_remove(struct spi_device *spi)
650 {
651         struct wm8731_priv *wm8731 = dev_get_drvdata(&spi->dev);
652
653         wm8731_unregister(wm8731);
654
655         return 0;
656 }
657
658 static struct spi_driver wm8731_spi_driver = {
659         .driver = {
660                 .name   = "wm8731",
661                 .bus    = &spi_bus_type,
662                 .owner  = THIS_MODULE,
663         },
664         .probe          = wm8731_spi_probe,
665         .remove         = __devexit_p(wm8731_spi_remove),
666 };
667 #endif /* CONFIG_SPI_MASTER */
668
669 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
670 static __devinit int wm8731_i2c_probe(struct i2c_client *i2c,
671                                       const struct i2c_device_id *id)
672 {
673         struct wm8731_priv *wm8731;
674         struct snd_soc_codec *codec;
675
676         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
677         if (wm8731 == NULL)
678                 return -ENOMEM;
679
680         codec = &wm8731->codec;
681
682         i2c_set_clientdata(i2c, wm8731);
683         codec->control_data = i2c;
684
685         codec->dev = &i2c->dev;
686
687         return wm8731_register(wm8731, SND_SOC_I2C);
688 }
689
690 static __devexit int wm8731_i2c_remove(struct i2c_client *client)
691 {
692         struct wm8731_priv *wm8731 = i2c_get_clientdata(client);
693         wm8731_unregister(wm8731);
694         return 0;
695 }
696
697 static const struct i2c_device_id wm8731_i2c_id[] = {
698         { "wm8731", 0 },
699         { }
700 };
701 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
702
703 static struct i2c_driver wm8731_i2c_driver = {
704         .driver = {
705                 .name = "WM8731 I2C Codec",
706                 .owner = THIS_MODULE,
707         },
708         .probe =    wm8731_i2c_probe,
709         .remove =   __devexit_p(wm8731_i2c_remove),
710         .id_table = wm8731_i2c_id,
711 };
712 #endif
713
714 static int __init wm8731_modinit(void)
715 {
716         int ret;
717 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
718         ret = i2c_add_driver(&wm8731_i2c_driver);
719         if (ret != 0) {
720                 printk(KERN_ERR "Failed to register WM8731 I2C driver: %d\n",
721                        ret);
722         }
723 #endif
724 #if defined(CONFIG_SPI_MASTER)
725         ret = spi_register_driver(&wm8731_spi_driver);
726         if (ret != 0) {
727                 printk(KERN_ERR "Failed to register WM8731 SPI driver: %d\n",
728                        ret);
729         }
730 #endif
731         return 0;
732 }
733 module_init(wm8731_modinit);
734
735 static void __exit wm8731_exit(void)
736 {
737 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
738         i2c_del_driver(&wm8731_i2c_driver);
739 #endif
740 #if defined(CONFIG_SPI_MASTER)
741         spi_unregister_driver(&wm8731_spi_driver);
742 #endif
743 }
744 module_exit(wm8731_exit);
745
746 MODULE_DESCRIPTION("ASoC WM8731 driver");
747 MODULE_AUTHOR("Richard Purdie");
748 MODULE_LICENSE("GPL");