ASoC: Add TLV information for WM8731
[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("Capture Mic 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 };
468 EXPORT_SYMBOL_GPL(wm8731_dai);
469
470 #ifdef CONFIG_PM
471 static int wm8731_suspend(struct platform_device *pdev, pm_message_t state)
472 {
473         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
474         struct snd_soc_codec *codec = socdev->card->codec;
475
476         wm8731_write(codec, WM8731_ACTIVE, 0x0);
477         wm8731_set_bias_level(codec, SND_SOC_BIAS_OFF);
478         return 0;
479 }
480
481 static int wm8731_resume(struct platform_device *pdev)
482 {
483         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
484         struct snd_soc_codec *codec = socdev->card->codec;
485         int i;
486         u8 data[2];
487         u16 *cache = codec->reg_cache;
488
489         /* Sync reg_cache with the hardware */
490         for (i = 0; i < ARRAY_SIZE(wm8731_reg); i++) {
491                 data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
492                 data[1] = cache[i] & 0x00ff;
493                 codec->hw_write(codec->control_data, data, 2);
494         }
495         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
496         wm8731_set_bias_level(codec, codec->suspend_bias_level);
497         return 0;
498 }
499 #else
500 #define wm8731_suspend NULL
501 #define wm8731_resume NULL
502 #endif
503
504 static int wm8731_probe(struct platform_device *pdev)
505 {
506         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
507         struct snd_soc_codec *codec;
508         int ret = 0;
509
510         if (wm8731_codec == NULL) {
511                 dev_err(&pdev->dev, "Codec device not registered\n");
512                 return -ENODEV;
513         }
514
515         socdev->card->codec = wm8731_codec;
516         codec = wm8731_codec;
517
518         /* register pcms */
519         ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
520         if (ret < 0) {
521                 dev_err(codec->dev, "failed to create pcms: %d\n", ret);
522                 goto pcm_err;
523         }
524
525         snd_soc_add_controls(codec, wm8731_snd_controls,
526                              ARRAY_SIZE(wm8731_snd_controls));
527         wm8731_add_widgets(codec);
528         ret = snd_soc_init_card(socdev);
529         if (ret < 0) {
530                 dev_err(codec->dev, "failed to register card: %d\n", ret);
531                 goto card_err;
532         }
533
534         return ret;
535
536 card_err:
537         snd_soc_free_pcms(socdev);
538         snd_soc_dapm_free(socdev);
539 pcm_err:
540         return ret;
541 }
542
543 /* power down chip */
544 static int wm8731_remove(struct platform_device *pdev)
545 {
546         struct snd_soc_device *socdev = platform_get_drvdata(pdev);
547
548         snd_soc_free_pcms(socdev);
549         snd_soc_dapm_free(socdev);
550
551         return 0;
552 }
553
554 struct snd_soc_codec_device soc_codec_dev_wm8731 = {
555         .probe =        wm8731_probe,
556         .remove =       wm8731_remove,
557         .suspend =      wm8731_suspend,
558         .resume =       wm8731_resume,
559 };
560 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8731);
561
562 static int wm8731_register(struct wm8731_priv *wm8731)
563 {
564         int ret;
565         struct snd_soc_codec *codec = &wm8731->codec;
566         u16 reg;
567
568         if (wm8731_codec) {
569                 dev_err(codec->dev, "Another WM8731 is registered\n");
570                 ret = -EINVAL;
571                 goto err;
572         }
573
574         mutex_init(&codec->mutex);
575         INIT_LIST_HEAD(&codec->dapm_widgets);
576         INIT_LIST_HEAD(&codec->dapm_paths);
577
578         codec->private_data = wm8731;
579         codec->name = "WM8731";
580         codec->owner = THIS_MODULE;
581         codec->read = wm8731_read_reg_cache;
582         codec->write = wm8731_write;
583         codec->bias_level = SND_SOC_BIAS_OFF;
584         codec->set_bias_level = wm8731_set_bias_level;
585         codec->dai = &wm8731_dai;
586         codec->num_dai = 1;
587         codec->reg_cache_size = WM8731_CACHEREGNUM;
588         codec->reg_cache = &wm8731->reg_cache;
589
590         memcpy(codec->reg_cache, wm8731_reg, sizeof(wm8731_reg));
591
592         ret = wm8731_reset(codec);
593         if (ret < 0) {
594                 dev_err(codec->dev, "Failed to issue reset: %d\n", ret);
595                 goto err;
596         }
597
598         wm8731_dai.dev = codec->dev;
599
600         wm8731_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
601
602         /* Latch the update bits */
603         reg = wm8731_read_reg_cache(codec, WM8731_LOUT1V);
604         wm8731_write(codec, WM8731_LOUT1V, reg & ~0x0100);
605         reg = wm8731_read_reg_cache(codec, WM8731_ROUT1V);
606         wm8731_write(codec, WM8731_ROUT1V, reg & ~0x0100);
607         reg = wm8731_read_reg_cache(codec, WM8731_LINVOL);
608         wm8731_write(codec, WM8731_LINVOL, reg & ~0x0100);
609         reg = wm8731_read_reg_cache(codec, WM8731_RINVOL);
610         wm8731_write(codec, WM8731_RINVOL, reg & ~0x0100);
611
612         /* Disable bypass path by default */
613         reg = wm8731_read_reg_cache(codec, WM8731_APANA);
614         wm8731_write(codec, WM8731_APANA, reg & ~0x4);
615
616         wm8731_codec = codec;
617
618         ret = snd_soc_register_codec(codec);
619         if (ret != 0) {
620                 dev_err(codec->dev, "Failed to register codec: %d\n", ret);
621                 goto err;
622         }
623
624         ret = snd_soc_register_dai(&wm8731_dai);
625         if (ret != 0) {
626                 dev_err(codec->dev, "Failed to register DAI: %d\n", ret);
627                 snd_soc_unregister_codec(codec);
628                 goto err_codec;
629         }
630
631         return 0;
632
633 err_codec:
634         snd_soc_unregister_codec(codec);
635 err:
636         kfree(wm8731);
637         return ret;
638 }
639
640 static void wm8731_unregister(struct wm8731_priv *wm8731)
641 {
642         wm8731_set_bias_level(&wm8731->codec, SND_SOC_BIAS_OFF);
643         snd_soc_unregister_dai(&wm8731_dai);
644         snd_soc_unregister_codec(&wm8731->codec);
645         kfree(wm8731);
646         wm8731_codec = NULL;
647 }
648
649 #if defined(CONFIG_SPI_MASTER)
650 static int wm8731_spi_write(struct spi_device *spi, const char *data, int len)
651 {
652         struct spi_transfer t;
653         struct spi_message m;
654         u8 msg[2];
655
656         if (len <= 0)
657                 return 0;
658
659         msg[0] = data[0];
660         msg[1] = data[1];
661
662         spi_message_init(&m);
663         memset(&t, 0, (sizeof t));
664
665         t.tx_buf = &msg[0];
666         t.len = len;
667
668         spi_message_add_tail(&t, &m);
669         spi_sync(spi, &m);
670
671         return len;
672 }
673
674 static int __devinit wm8731_spi_probe(struct spi_device *spi)
675 {
676         struct snd_soc_codec *codec;
677         struct wm8731_priv *wm8731;
678
679         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
680         if (wm8731 == NULL)
681                 return -ENOMEM;
682
683         codec = &wm8731->codec;
684         codec->control_data = spi;
685         codec->hw_write = (hw_write_t)wm8731_spi_write;
686         codec->dev = &spi->dev;
687
688         dev_set_drvdata(&spi->dev, wm8731);
689
690         return wm8731_register(wm8731);
691 }
692
693 static int __devexit wm8731_spi_remove(struct spi_device *spi)
694 {
695         struct wm8731_priv *wm8731 = dev_get_drvdata(&spi->dev);
696
697         wm8731_unregister(wm8731);
698
699         return 0;
700 }
701
702 #ifdef CONFIG_PM
703 static int wm8731_spi_suspend(struct spi_device *spi, pm_message_t msg)
704 {
705         return snd_soc_suspend_device(&spi->dev);
706 }
707
708 static int wm8731_spi_resume(struct spi_device *spi)
709 {
710         return snd_soc_resume_device(&spi->dev);
711 }
712 #else
713 #define wm8731_spi_suspend NULL
714 #define wm8731_spi_resume NULL
715 #endif
716
717 static struct spi_driver wm8731_spi_driver = {
718         .driver = {
719                 .name   = "wm8731",
720                 .bus    = &spi_bus_type,
721                 .owner  = THIS_MODULE,
722         },
723         .probe          = wm8731_spi_probe,
724         .suspend        = wm8731_spi_suspend,
725         .resume         = wm8731_spi_resume,
726         .remove         = __devexit_p(wm8731_spi_remove),
727 };
728 #endif /* CONFIG_SPI_MASTER */
729
730 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
731 static __devinit int wm8731_i2c_probe(struct i2c_client *i2c,
732                                       const struct i2c_device_id *id)
733 {
734         struct wm8731_priv *wm8731;
735         struct snd_soc_codec *codec;
736
737         wm8731 = kzalloc(sizeof(struct wm8731_priv), GFP_KERNEL);
738         if (wm8731 == NULL)
739                 return -ENOMEM;
740
741         codec = &wm8731->codec;
742         codec->hw_write = (hw_write_t)i2c_master_send;
743
744         i2c_set_clientdata(i2c, wm8731);
745         codec->control_data = i2c;
746
747         codec->dev = &i2c->dev;
748
749         return wm8731_register(wm8731);
750 }
751
752 static __devexit int wm8731_i2c_remove(struct i2c_client *client)
753 {
754         struct wm8731_priv *wm8731 = i2c_get_clientdata(client);
755         wm8731_unregister(wm8731);
756         return 0;
757 }
758
759 #ifdef CONFIG_PM
760 static int wm8731_i2c_suspend(struct i2c_client *i2c, pm_message_t msg)
761 {
762         return snd_soc_suspend_device(&i2c->dev);
763 }
764
765 static int wm8731_i2c_resume(struct i2c_client *i2c)
766 {
767         return snd_soc_resume_device(&i2c->dev);
768 }
769 #else
770 #define wm8731_i2c_suspend NULL
771 #define wm8731_i2c_resume NULL
772 #endif
773
774 static const struct i2c_device_id wm8731_i2c_id[] = {
775         { "wm8731", 0 },
776         { }
777 };
778 MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
779
780 static struct i2c_driver wm8731_i2c_driver = {
781         .driver = {
782                 .name = "WM8731 I2C Codec",
783                 .owner = THIS_MODULE,
784         },
785         .probe =    wm8731_i2c_probe,
786         .remove =   __devexit_p(wm8731_i2c_remove),
787         .suspend =  wm8731_i2c_suspend,
788         .resume =   wm8731_i2c_resume,
789         .id_table = wm8731_i2c_id,
790 };
791 #endif
792
793 static int __init wm8731_modinit(void)
794 {
795         int ret;
796 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
797         ret = i2c_add_driver(&wm8731_i2c_driver);
798         if (ret != 0) {
799                 printk(KERN_ERR "Failed to register WM8731 I2C driver: %d\n",
800                        ret);
801         }
802 #endif
803 #if defined(CONFIG_SPI_MASTER)
804         ret = spi_register_driver(&wm8731_spi_driver);
805         if (ret != 0) {
806                 printk(KERN_ERR "Failed to register WM8731 SPI driver: %d\n",
807                        ret);
808         }
809 #endif
810         return 0;
811 }
812 module_init(wm8731_modinit);
813
814 static void __exit wm8731_exit(void)
815 {
816 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
817         i2c_del_driver(&wm8731_i2c_driver);
818 #endif
819 #if defined(CONFIG_SPI_MASTER)
820         spi_unregister_driver(&wm8731_spi_driver);
821 #endif
822 }
823 module_exit(wm8731_exit);
824
825 MODULE_DESCRIPTION("ASoC WM8731 driver");
826 MODULE_AUTHOR("Richard Purdie");
827 MODULE_LICENSE("GPL");