xref: /openbmc/linux/sound/soc/codecs/wm8776.c (revision 35b1207b)
1 /*
2  * wm8776.c  --  WM8776 ALSA SoC Audio driver
3  *
4  * Copyright 2009 Wolfson Microelectronics plc
5  *
6  * Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 as
10  * published by the Free Software Foundation.
11  *
12  * TODO: Input ALC/limiter support
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 "wm8776.h"
32 
33 static struct snd_soc_codec *wm8776_codec;
34 struct snd_soc_codec_device soc_codec_dev_wm8776;
35 
36 /* codec private data */
37 struct wm8776_priv {
38 	struct snd_soc_codec codec;
39 	u16 reg_cache[WM8776_CACHEREGNUM];
40 	int sysclk[2];
41 };
42 
43 #ifdef CONFIG_SPI_MASTER
44 static int wm8776_spi_write(struct spi_device *spi, const char *data, int len);
45 #endif
46 
47 static const u16 wm8776_reg[WM8776_CACHEREGNUM] = {
48 	0x79, 0x79, 0x79, 0xff, 0xff,  /* 4 */
49 	0xff, 0x00, 0x90, 0x00, 0x00,  /* 9 */
50 	0x22, 0x22, 0x22, 0x08, 0xcf,  /* 14 */
51 	0xcf, 0x7b, 0x00, 0x32, 0x00,  /* 19 */
52 	0xa6, 0x01, 0x01
53 };
54 
55 static int wm8776_reset(struct snd_soc_codec *codec)
56 {
57 	return snd_soc_write(codec, WM8776_RESET, 0);
58 }
59 
60 static const DECLARE_TLV_DB_SCALE(hp_tlv, -12100, 100, 1);
61 static const DECLARE_TLV_DB_SCALE(dac_tlv, -12750, 50, 1);
62 static const DECLARE_TLV_DB_SCALE(adc_tlv, -10350, 50, 1);
63 
64 static const struct snd_kcontrol_new wm8776_snd_controls[] = {
65 SOC_DOUBLE_R_TLV("Headphone Playback Volume", WM8776_HPLVOL, WM8776_HPRVOL,
66 		 0, 127, 0, hp_tlv),
67 SOC_DOUBLE_R_TLV("Digital Playback Volume", WM8776_DACLVOL, WM8776_DACRVOL,
68 		 0, 255, 0, dac_tlv),
69 SOC_SINGLE("Digital Playback ZC Switch", WM8776_DACCTRL1, 0, 1, 0),
70 
71 SOC_SINGLE("Deemphasis Switch", WM8776_DACCTRL2, 0, 1, 0),
72 
73 SOC_DOUBLE_R_TLV("Capture Volume", WM8776_ADCLVOL, WM8776_ADCRVOL,
74 		 0, 255, 0, adc_tlv),
75 SOC_DOUBLE("Capture Switch", WM8776_ADCMUX, 7, 6, 1, 1),
76 SOC_DOUBLE_R("Capture ZC Switch", WM8776_ADCLVOL, WM8776_ADCRVOL, 8, 1, 0),
77 SOC_SINGLE("Capture HPF Switch", WM8776_ADCIFCTRL, 8, 1, 1),
78 };
79 
80 static const struct snd_kcontrol_new inmix_controls[] = {
81 SOC_DAPM_SINGLE("AIN1 Switch", WM8776_ADCMUX, 0, 1, 0),
82 SOC_DAPM_SINGLE("AIN2 Switch", WM8776_ADCMUX, 1, 1, 0),
83 SOC_DAPM_SINGLE("AIN3 Switch", WM8776_ADCMUX, 2, 1, 0),
84 SOC_DAPM_SINGLE("AIN4 Switch", WM8776_ADCMUX, 3, 1, 0),
85 SOC_DAPM_SINGLE("AIN5 Switch", WM8776_ADCMUX, 4, 1, 0),
86 };
87 
88 static const struct snd_kcontrol_new outmix_controls[] = {
89 SOC_DAPM_SINGLE("DAC Switch", WM8776_OUTMUX, 0, 1, 0),
90 SOC_DAPM_SINGLE("AUX Switch", WM8776_OUTMUX, 1, 1, 0),
91 SOC_DAPM_SINGLE("Bypass Switch", WM8776_OUTMUX, 2, 1, 0),
92 };
93 
94 static const struct snd_soc_dapm_widget wm8776_dapm_widgets[] = {
95 SND_SOC_DAPM_INPUT("AUX"),
96 SND_SOC_DAPM_INPUT("AUX"),
97 
98 SND_SOC_DAPM_INPUT("AIN1"),
99 SND_SOC_DAPM_INPUT("AIN2"),
100 SND_SOC_DAPM_INPUT("AIN3"),
101 SND_SOC_DAPM_INPUT("AIN4"),
102 SND_SOC_DAPM_INPUT("AIN5"),
103 
104 SND_SOC_DAPM_MIXER("Input Mixer", WM8776_PWRDOWN, 6, 1,
105 		   inmix_controls, ARRAY_SIZE(inmix_controls)),
106 
107 SND_SOC_DAPM_ADC("ADC", "Capture", WM8776_PWRDOWN, 1, 1),
108 SND_SOC_DAPM_DAC("DAC", "Playback", WM8776_PWRDOWN, 2, 1),
109 
110 SND_SOC_DAPM_MIXER("Output Mixer", SND_SOC_NOPM, 0, 0,
111 		   outmix_controls, ARRAY_SIZE(outmix_controls)),
112 
113 SND_SOC_DAPM_PGA("Headphone PGA", WM8776_PWRDOWN, 3, 1, NULL, 0),
114 
115 SND_SOC_DAPM_OUTPUT("VOUT"),
116 
117 SND_SOC_DAPM_OUTPUT("HPOUTL"),
118 SND_SOC_DAPM_OUTPUT("HPOUTR"),
119 };
120 
121 static const struct snd_soc_dapm_route routes[] = {
122 	{ "Input Mixer", "AIN1 Switch", "AIN1" },
123 	{ "Input Mixer", "AIN2 Switch", "AIN2" },
124 	{ "Input Mixer", "AIN3 Switch", "AIN3" },
125 	{ "Input Mixer", "AIN4 Switch", "AIN4" },
126 	{ "Input Mixer", "AIN5 Switch", "AIN5" },
127 
128 	{ "ADC", NULL, "Input Mixer" },
129 
130 	{ "Output Mixer", "DAC Switch", "DAC" },
131 	{ "Output Mixer", "AUX Switch", "AUX" },
132 	{ "Output Mixer", "Bypass Switch", "Input Mixer" },
133 
134 	{ "VOUT", NULL, "Output Mixer" },
135 
136 	{ "Headphone PGA", NULL, "Output Mixer" },
137 
138 	{ "HPOUTL", NULL, "Headphone PGA" },
139 	{ "HPOUTR", NULL, "Headphone PGA" },
140 };
141 
142 static int wm8776_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
143 {
144 	struct snd_soc_codec *codec = dai->codec;
145 	int reg, iface, master;
146 
147 	switch (dai->id) {
148 	case WM8776_DAI_DAC:
149 		reg = WM8776_DACIFCTRL;
150 		master = 0x80;
151 		break;
152 	case WM8776_DAI_ADC:
153 		reg = WM8776_ADCIFCTRL;
154 		master = 0x100;
155 		break;
156 	default:
157 		return -EINVAL;
158 	}
159 
160 	iface = 0;
161 
162 	switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
163 	case SND_SOC_DAIFMT_CBM_CFM:
164 		break;
165 	case SND_SOC_DAIFMT_CBS_CFS:
166 		master = 0;
167 		break;
168 	default:
169 		return -EINVAL;
170 	}
171 
172 	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
173 	case SND_SOC_DAIFMT_I2S:
174 		iface |= 0x0002;
175 		break;
176 	case SND_SOC_DAIFMT_RIGHT_J:
177 		break;
178 	case SND_SOC_DAIFMT_LEFT_J:
179 		iface |= 0x0001;
180 		break;
181 		/* FIXME: CHECK A/B */
182 	case SND_SOC_DAIFMT_DSP_A:
183 		iface |= 0x0003;
184 		break;
185 	case SND_SOC_DAIFMT_DSP_B:
186 		iface |= 0x0007;
187 		break;
188 	default:
189 		return -EINVAL;
190 	}
191 
192 	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
193 	case SND_SOC_DAIFMT_NB_NF:
194 		break;
195 	case SND_SOC_DAIFMT_IB_IF:
196 		iface |= 0x00c;
197 		break;
198 	case SND_SOC_DAIFMT_IB_NF:
199 		iface |= 0x008;
200 		break;
201 	case SND_SOC_DAIFMT_NB_IF:
202 		iface |= 0x004;
203 		break;
204 	default:
205 		return -EINVAL;
206 	}
207 
208 	/* Finally, write out the values */
209 	snd_soc_update_bits(codec, reg, 0xf, iface);
210 	snd_soc_update_bits(codec, WM8776_MSTRCTRL, 0x180, master);
211 
212 	return 0;
213 }
214 
215 static int mclk_ratios[] = {
216 	128,
217 	192,
218 	256,
219 	384,
220 	512,
221 	768,
222 };
223 
224 static int wm8776_hw_params(struct snd_pcm_substream *substream,
225 			    struct snd_pcm_hw_params *params,
226 			    struct snd_soc_dai *dai)
227 {
228 	struct snd_soc_codec *codec = dai->codec;
229 	struct wm8776_priv *wm8776 = codec->private_data;
230 	int iface_reg, iface;
231 	int ratio_shift, master;
232 	int i;
233 
234 	iface = 0;
235 
236 	switch (dai->id) {
237 	case WM8776_DAI_DAC:
238 		iface_reg = WM8776_DACIFCTRL;
239 		master = 0x80;
240 		ratio_shift = 4;
241 		break;
242 	case WM8776_DAI_ADC:
243 		iface_reg = WM8776_ADCIFCTRL;
244 		master = 0x100;
245 		ratio_shift = 0;
246 		break;
247 	default:
248 		return -EINVAL;
249 	}
250 
251 
252 	/* Set word length */
253 	switch (params_format(params)) {
254 	case SNDRV_PCM_FORMAT_S16_LE:
255 		break;
256 	case SNDRV_PCM_FORMAT_S20_3LE:
257 		iface |= 0x10;
258 		break;
259 	case SNDRV_PCM_FORMAT_S24_LE:
260 		iface |= 0x20;
261 		break;
262 	case SNDRV_PCM_FORMAT_S32_LE:
263 		iface |= 0x30;
264 		break;
265 	}
266 
267 	/* Only need to set MCLK/LRCLK ratio if we're master */
268 	if (snd_soc_read(codec, WM8776_MSTRCTRL) & master) {
269 		for (i = 0; i < ARRAY_SIZE(mclk_ratios); i++) {
270 			if (wm8776->sysclk[dai->id] / params_rate(params)
271 			    == mclk_ratios[i])
272 				break;
273 		}
274 
275 		if (i == ARRAY_SIZE(mclk_ratios)) {
276 			dev_err(codec->dev,
277 				"Unable to configure MCLK ratio %d/%d\n",
278 				wm8776->sysclk[dai->id], params_rate(params));
279 			return -EINVAL;
280 		}
281 
282 		dev_dbg(codec->dev, "MCLK is %dfs\n", mclk_ratios[i]);
283 
284 		snd_soc_update_bits(codec, WM8776_MSTRCTRL,
285 				    0x7 << ratio_shift, i << ratio_shift);
286 	} else {
287 		dev_dbg(codec->dev, "DAI in slave mode\n");
288 	}
289 
290 	snd_soc_update_bits(codec, iface_reg, 0x30, iface);
291 
292 	return 0;
293 }
294 
295 static int wm8776_mute(struct snd_soc_dai *dai, int mute)
296 {
297 	struct snd_soc_codec *codec = dai->codec;
298 
299 	return snd_soc_write(codec, WM8776_DACMUTE, !!mute);
300 }
301 
302 static int wm8776_set_sysclk(struct snd_soc_dai *dai,
303 			     int clk_id, unsigned int freq, int dir)
304 {
305 	struct snd_soc_codec *codec = dai->codec;
306 	struct wm8776_priv *wm8776 = codec->private_data;
307 
308 	BUG_ON(dai->id >= ARRAY_SIZE(wm8776->sysclk));
309 
310 	wm8776->sysclk[dai->id] = freq;
311 
312 	return 0;
313 }
314 
315 static int wm8776_set_bias_level(struct snd_soc_codec *codec,
316 				 enum snd_soc_bias_level level)
317 {
318 	switch (level) {
319 	case SND_SOC_BIAS_ON:
320 		break;
321 	case SND_SOC_BIAS_PREPARE:
322 		break;
323 	case SND_SOC_BIAS_STANDBY:
324 		if (codec->bias_level == SND_SOC_BIAS_OFF) {
325 			/* Disable the global powerdown; DAPM does the rest */
326 			snd_soc_update_bits(codec, WM8776_PWRDOWN, 1, 0);
327 		}
328 
329 		break;
330 	case SND_SOC_BIAS_OFF:
331 		snd_soc_update_bits(codec, WM8776_PWRDOWN, 1, 1);
332 		break;
333 	}
334 
335 	codec->bias_level = level;
336 	return 0;
337 }
338 
339 #define WM8776_RATES (SNDRV_PCM_RATE_32000 | SNDRV_PCM_RATE_44100 |\
340 		      SNDRV_PCM_RATE_48000 | SNDRV_PCM_RATE_88200 |\
341 		      SNDRV_PCM_RATE_96000)
342 
343 
344 #define WM8776_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
345 			SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S32_LE)
346 
347 static struct snd_soc_dai_ops wm8776_dac_ops = {
348 	.digital_mute	= wm8776_mute,
349 	.hw_params      = wm8776_hw_params,
350 	.set_fmt        = wm8776_set_fmt,
351 	.set_sysclk     = wm8776_set_sysclk,
352 };
353 
354 static struct snd_soc_dai_ops wm8776_adc_ops = {
355 	.hw_params      = wm8776_hw_params,
356 	.set_fmt        = wm8776_set_fmt,
357 	.set_sysclk     = wm8776_set_sysclk,
358 };
359 
360 struct snd_soc_dai wm8776_dai[] = {
361 	{
362 		.name = "WM8776 Playback",
363 		.id = WM8776_DAI_DAC,
364 		.playback = {
365 			.stream_name = "Playback",
366 			.channels_min = 2,
367 			.channels_max = 2,
368 			.rates = WM8776_RATES,
369 			.formats = WM8776_FORMATS,
370 		},
371 		.ops = &wm8776_dac_ops,
372 	},
373 	{
374 		.name = "WM8776 Capture",
375 		.id = WM8776_DAI_ADC,
376 		.capture = {
377 			.stream_name = "Capture",
378 			.channels_min = 2,
379 			.channels_max = 2,
380 			.rates = WM8776_RATES,
381 			.formats = WM8776_FORMATS,
382 		},
383 		.ops = &wm8776_adc_ops,
384 	},
385 };
386 EXPORT_SYMBOL_GPL(wm8776_dai);
387 
388 #ifdef CONFIG_PM
389 static int wm8776_suspend(struct platform_device *pdev, pm_message_t state)
390 {
391 	struct snd_soc_device *socdev = platform_get_drvdata(pdev);
392 	struct snd_soc_codec *codec = socdev->card->codec;
393 
394 	wm8776_set_bias_level(codec, SND_SOC_BIAS_OFF);
395 
396 	return 0;
397 }
398 
399 static int wm8776_resume(struct platform_device *pdev)
400 {
401 	struct snd_soc_device *socdev = platform_get_drvdata(pdev);
402 	struct snd_soc_codec *codec = socdev->card->codec;
403 	int i;
404 	u8 data[2];
405 	u16 *cache = codec->reg_cache;
406 
407 	/* Sync reg_cache with the hardware */
408 	for (i = 0; i < ARRAY_SIZE(wm8776_reg); i++) {
409 		data[0] = (i << 1) | ((cache[i] >> 8) & 0x0001);
410 		data[1] = cache[i] & 0x00ff;
411 		codec->hw_write(codec->control_data, data, 2);
412 	}
413 
414 	wm8776_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
415 
416 	return 0;
417 }
418 #else
419 #define wm8776_suspend NULL
420 #define wm8776_resume NULL
421 #endif
422 
423 static int wm8776_probe(struct platform_device *pdev)
424 {
425 	struct snd_soc_device *socdev = platform_get_drvdata(pdev);
426 	struct snd_soc_codec *codec;
427 	int ret = 0;
428 
429 	if (wm8776_codec == NULL) {
430 		dev_err(&pdev->dev, "Codec device not registered\n");
431 		return -ENODEV;
432 	}
433 
434 	socdev->card->codec = wm8776_codec;
435 	codec = wm8776_codec;
436 
437 	/* register pcms */
438 	ret = snd_soc_new_pcms(socdev, SNDRV_DEFAULT_IDX1, SNDRV_DEFAULT_STR1);
439 	if (ret < 0) {
440 		dev_err(codec->dev, "failed to create pcms: %d\n", ret);
441 		goto pcm_err;
442 	}
443 
444 	snd_soc_add_controls(codec, wm8776_snd_controls,
445 			     ARRAY_SIZE(wm8776_snd_controls));
446 	snd_soc_dapm_new_controls(codec, wm8776_dapm_widgets,
447 				  ARRAY_SIZE(wm8776_dapm_widgets));
448 	snd_soc_dapm_add_routes(codec, routes, ARRAY_SIZE(routes));
449 
450 	ret = snd_soc_init_card(socdev);
451 	if (ret < 0) {
452 		dev_err(codec->dev, "failed to register card: %d\n", ret);
453 		goto card_err;
454 	}
455 
456 	return ret;
457 
458 card_err:
459 	snd_soc_free_pcms(socdev);
460 	snd_soc_dapm_free(socdev);
461 pcm_err:
462 	return ret;
463 }
464 
465 /* power down chip */
466 static int wm8776_remove(struct platform_device *pdev)
467 {
468 	struct snd_soc_device *socdev = platform_get_drvdata(pdev);
469 
470 	snd_soc_free_pcms(socdev);
471 	snd_soc_dapm_free(socdev);
472 
473 	return 0;
474 }
475 
476 struct snd_soc_codec_device soc_codec_dev_wm8776 = {
477 	.probe = 	wm8776_probe,
478 	.remove = 	wm8776_remove,
479 	.suspend = 	wm8776_suspend,
480 	.resume =	wm8776_resume,
481 };
482 EXPORT_SYMBOL_GPL(soc_codec_dev_wm8776);
483 
484 static int wm8776_register(struct wm8776_priv *wm8776,
485 			   enum snd_soc_control_type control)
486 {
487 	int ret, i;
488 	struct snd_soc_codec *codec = &wm8776->codec;
489 
490 	if (wm8776_codec) {
491 		dev_err(codec->dev, "Another WM8776 is registered\n");
492 		ret = -EINVAL;
493 		goto err;
494 	}
495 
496 	mutex_init(&codec->mutex);
497 	INIT_LIST_HEAD(&codec->dapm_widgets);
498 	INIT_LIST_HEAD(&codec->dapm_paths);
499 
500 	codec->private_data = wm8776;
501 	codec->name = "WM8776";
502 	codec->owner = THIS_MODULE;
503 	codec->bias_level = SND_SOC_BIAS_OFF;
504 	codec->set_bias_level = wm8776_set_bias_level;
505 	codec->dai = wm8776_dai;
506 	codec->num_dai = ARRAY_SIZE(wm8776_dai);
507 	codec->reg_cache_size = WM8776_CACHEREGNUM;
508 	codec->reg_cache = &wm8776->reg_cache;
509 
510 	memcpy(codec->reg_cache, wm8776_reg, sizeof(wm8776_reg));
511 
512 	ret = snd_soc_codec_set_cache_io(codec, 7, 9, control);
513 	if (ret < 0) {
514 		dev_err(codec->dev, "Failed to set cache I/O: %d\n", ret);
515 		goto err;
516 	}
517 
518 	for (i = 0; i < ARRAY_SIZE(wm8776_dai); i++)
519 		wm8776_dai[i].dev = codec->dev;
520 
521 	ret = wm8776_reset(codec);
522 	if (ret < 0) {
523 		dev_err(codec->dev, "Failed to issue reset: %d\n", ret);
524 		goto err;
525 	}
526 
527 	wm8776_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
528 
529 	/* Latch the update bits; right channel only since we always
530 	 * update both. */
531 	snd_soc_update_bits(codec, WM8776_HPRVOL, 0x100, 0x100);
532 	snd_soc_update_bits(codec, WM8776_DACRVOL, 0x100, 0x100);
533 
534 	wm8776_codec = codec;
535 
536 	ret = snd_soc_register_codec(codec);
537 	if (ret != 0) {
538 		dev_err(codec->dev, "Failed to register codec: %d\n", ret);
539 		goto err;
540 	}
541 
542 	ret = snd_soc_register_dais(wm8776_dai, ARRAY_SIZE(wm8776_dai));
543 	if (ret != 0) {
544 		dev_err(codec->dev, "Failed to register DAIs: %d\n", ret);
545 		goto err_codec;
546 	}
547 
548 	return 0;
549 
550 err_codec:
551 	snd_soc_unregister_codec(codec);
552 err:
553 	kfree(wm8776);
554 	return ret;
555 }
556 
557 static void wm8776_unregister(struct wm8776_priv *wm8776)
558 {
559 	wm8776_set_bias_level(&wm8776->codec, SND_SOC_BIAS_OFF);
560 	snd_soc_unregister_dais(wm8776_dai, ARRAY_SIZE(wm8776_dai));
561 	snd_soc_unregister_codec(&wm8776->codec);
562 	kfree(wm8776);
563 	wm8776_codec = NULL;
564 }
565 
566 #if defined(CONFIG_SPI_MASTER)
567 static int wm8776_spi_write(struct spi_device *spi, const char *data, int len)
568 {
569 	struct spi_transfer t;
570 	struct spi_message m;
571 	u8 msg[2];
572 
573 	if (len <= 0)
574 		return 0;
575 
576 	msg[0] = data[0];
577 	msg[1] = data[1];
578 
579 	spi_message_init(&m);
580 	memset(&t, 0, (sizeof t));
581 
582 	t.tx_buf = &msg[0];
583 	t.len = len;
584 
585 	spi_message_add_tail(&t, &m);
586 	spi_sync(spi, &m);
587 
588 	return len;
589 }
590 
591 static int __devinit wm8776_spi_probe(struct spi_device *spi)
592 {
593 	struct snd_soc_codec *codec;
594 	struct wm8776_priv *wm8776;
595 
596 	wm8776 = kzalloc(sizeof(struct wm8776_priv), GFP_KERNEL);
597 	if (wm8776 == NULL)
598 		return -ENOMEM;
599 
600 	codec = &wm8776->codec;
601 	codec->control_data = spi;
602 	codec->hw_write = (hw_write_t)wm8776_spi_write;
603 	codec->dev = &spi->dev;
604 
605 	dev_set_drvdata(&spi->dev, wm8776);
606 
607 	return wm8776_register(wm8776, SND_SOC_SPI);
608 }
609 
610 static int __devexit wm8776_spi_remove(struct spi_device *spi)
611 {
612 	struct wm8776_priv *wm8776 = dev_get_drvdata(&spi->dev);
613 
614 	wm8776_unregister(wm8776);
615 
616 	return 0;
617 }
618 
619 #ifdef CONFIG_PM
620 static int wm8776_spi_suspend(struct spi_device *spi, pm_message_t msg)
621 {
622 	return snd_soc_suspend_device(&spi->dev);
623 }
624 
625 static int wm8776_spi_resume(struct spi_device *spi)
626 {
627 	return snd_soc_resume_device(&spi->dev);
628 }
629 #else
630 #define wm8776_spi_suspend NULL
631 #define wm8776_spi_resume NULL
632 #endif
633 
634 static struct spi_driver wm8776_spi_driver = {
635 	.driver = {
636 		.name	= "wm8776",
637 		.bus	= &spi_bus_type,
638 		.owner	= THIS_MODULE,
639 	},
640 	.probe		= wm8776_spi_probe,
641 	.suspend	= wm8776_spi_suspend,
642 	.resume		= wm8776_spi_resume,
643 	.remove		= __devexit_p(wm8776_spi_remove),
644 };
645 #endif /* CONFIG_SPI_MASTER */
646 
647 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
648 static __devinit int wm8776_i2c_probe(struct i2c_client *i2c,
649 				      const struct i2c_device_id *id)
650 {
651 	struct wm8776_priv *wm8776;
652 	struct snd_soc_codec *codec;
653 
654 	wm8776 = kzalloc(sizeof(struct wm8776_priv), GFP_KERNEL);
655 	if (wm8776 == NULL)
656 		return -ENOMEM;
657 
658 	codec = &wm8776->codec;
659 	codec->hw_write = (hw_write_t)i2c_master_send;
660 
661 	i2c_set_clientdata(i2c, wm8776);
662 	codec->control_data = i2c;
663 
664 	codec->dev = &i2c->dev;
665 
666 	return wm8776_register(wm8776, SND_SOC_I2C);
667 }
668 
669 static __devexit int wm8776_i2c_remove(struct i2c_client *client)
670 {
671 	struct wm8776_priv *wm8776 = i2c_get_clientdata(client);
672 	wm8776_unregister(wm8776);
673 	return 0;
674 }
675 
676 #ifdef CONFIG_PM
677 static int wm8776_i2c_suspend(struct i2c_client *i2c, pm_message_t msg)
678 {
679 	return snd_soc_suspend_device(&i2c->dev);
680 }
681 
682 static int wm8776_i2c_resume(struct i2c_client *i2c)
683 {
684 	return snd_soc_resume_device(&i2c->dev);
685 }
686 #else
687 #define wm8776_i2c_suspend NULL
688 #define wm8776_i2c_resume NULL
689 #endif
690 
691 static const struct i2c_device_id wm8776_i2c_id[] = {
692 	{ "wm8776", 0 },
693 	{ }
694 };
695 MODULE_DEVICE_TABLE(i2c, wm8776_i2c_id);
696 
697 static struct i2c_driver wm8776_i2c_driver = {
698 	.driver = {
699 		.name = "wm8776",
700 		.owner = THIS_MODULE,
701 	},
702 	.probe =    wm8776_i2c_probe,
703 	.remove =   __devexit_p(wm8776_i2c_remove),
704 	.suspend =  wm8776_i2c_suspend,
705 	.resume =   wm8776_i2c_resume,
706 	.id_table = wm8776_i2c_id,
707 };
708 #endif
709 
710 static int __init wm8776_modinit(void)
711 {
712 	int ret;
713 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
714 	ret = i2c_add_driver(&wm8776_i2c_driver);
715 	if (ret != 0) {
716 		printk(KERN_ERR "Failed to register WM8776 I2C driver: %d\n",
717 		       ret);
718 	}
719 #endif
720 #if defined(CONFIG_SPI_MASTER)
721 	ret = spi_register_driver(&wm8776_spi_driver);
722 	if (ret != 0) {
723 		printk(KERN_ERR "Failed to register WM8776 SPI driver: %d\n",
724 		       ret);
725 	}
726 #endif
727 	return 0;
728 }
729 module_init(wm8776_modinit);
730 
731 static void __exit wm8776_exit(void)
732 {
733 #if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
734 	i2c_del_driver(&wm8776_i2c_driver);
735 #endif
736 #if defined(CONFIG_SPI_MASTER)
737 	spi_unregister_driver(&wm8776_spi_driver);
738 #endif
739 }
740 module_exit(wm8776_exit);
741 
742 MODULE_DESCRIPTION("ASoC WM8776 driver");
743 MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
744 MODULE_LICENSE("GPL");
745