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