xref: /openbmc/linux/sound/soc/codecs/wm8731.c (revision a09d2831)
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");
751