1/*
2 * wm8940.c  --  WM8940 ALSA Soc Audio driver
3 *
4 * Author: Jonathan Cameron <jic23@cam.ac.uk>
5 *
6 * Based on wm8510.c
7 *    Copyright  2006 Wolfson Microelectronics PLC.
8 *    Author:  Liam Girdwood <lrg@slimlogic.co.uk>
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 * Not currently handled:
15 * Notch filter control
16 * AUXMode (inverting vs mixer)
17 * No means to obtain current gain if alc enabled.
18 * No use made of gpio
19 * Fast VMID discharge for power down
20 * Soft Start
21 * DLR and ALR Swaps not enabled
22 * Digital Sidetone not supported
23 */
24#include <linux/module.h>
25#include <linux/moduleparam.h>
26#include <linux/kernel.h>
27#include <linux/init.h>
28#include <linux/delay.h>
29#include <linux/pm.h>
30#include <linux/i2c.h>
31#include <linux/platform_device.h>
32#include <linux/spi/spi.h>
33#include <linux/slab.h>
34#include <sound/core.h>
35#include <sound/pcm.h>
36#include <sound/pcm_params.h>
37#include <sound/soc.h>
38#include <sound/initval.h>
39#include <sound/tlv.h>
40
41#include "wm8940.h"
42
43struct wm8940_priv {
44	unsigned int sysclk;
45	enum snd_soc_control_type control_type;
46	void *control_data;
47};
48
49static u16 wm8940_reg_defaults[] = {
50	0x8940, /* Soft Reset */
51	0x0000, /* Power 1 */
52	0x0000, /* Power 2 */
53	0x0000, /* Power 3 */
54	0x0010, /* Interface Control */
55	0x0000, /* Companding Control */
56	0x0140, /* Clock Control */
57	0x0000, /* Additional Controls */
58	0x0000, /* GPIO Control */
59	0x0002, /* Auto Increment Control */
60	0x0000, /* DAC Control */
61	0x00FF, /* DAC Volume */
62	0,
63	0,
64	0x0100, /* ADC Control */
65	0x00FF, /* ADC Volume */
66	0x0000, /* Notch Filter 1 Control 1 */
67	0x0000, /* Notch Filter 1 Control 2 */
68	0x0000, /* Notch Filter 2 Control 1 */
69	0x0000, /* Notch Filter 2 Control 2 */
70	0x0000, /* Notch Filter 3 Control 1 */
71	0x0000, /* Notch Filter 3 Control 2 */
72	0x0000, /* Notch Filter 4 Control 1 */
73	0x0000, /* Notch Filter 4 Control 2 */
74	0x0032, /* DAC Limit Control 1 */
75	0x0000, /* DAC Limit Control 2 */
76	0,
77	0,
78	0,
79	0,
80	0,
81	0,
82	0x0038, /* ALC Control 1 */
83	0x000B, /* ALC Control 2 */
84	0x0032, /* ALC Control 3 */
85	0x0000, /* Noise Gate */
86	0x0041, /* PLLN */
87	0x000C, /* PLLK1 */
88	0x0093, /* PLLK2 */
89	0x00E9, /* PLLK3 */
90	0,
91	0,
92	0x0030, /* ALC Control 4 */
93	0,
94	0x0002, /* Input Control */
95	0x0050, /* PGA Gain */
96	0,
97	0x0002, /* ADC Boost Control */
98	0,
99	0x0002, /* Output Control */
100	0x0000, /* Speaker Mixer Control */
101	0,
102	0,
103	0,
104	0x0079, /* Speaker Volume */
105	0,
106	0x0000, /* Mono Mixer Control */
107};
108
109static const char *wm8940_companding[] = { "Off", "NC", "u-law", "A-law" };
110static const struct soc_enum wm8940_adc_companding_enum
111= SOC_ENUM_SINGLE(WM8940_COMPANDINGCTL, 1, 4, wm8940_companding);
112static const struct soc_enum wm8940_dac_companding_enum
113= SOC_ENUM_SINGLE(WM8940_COMPANDINGCTL, 3, 4, wm8940_companding);
114
115static const char *wm8940_alc_mode_text[] = {"ALC", "Limiter"};
116static const struct soc_enum wm8940_alc_mode_enum
117= SOC_ENUM_SINGLE(WM8940_ALC3, 8, 2, wm8940_alc_mode_text);
118
119static const char *wm8940_mic_bias_level_text[] = {"0.9", "0.65"};
120static const struct soc_enum wm8940_mic_bias_level_enum
121= SOC_ENUM_SINGLE(WM8940_INPUTCTL, 8, 2, wm8940_mic_bias_level_text);
122
123static const char *wm8940_filter_mode_text[] = {"Audio", "Application"};
124static const struct soc_enum wm8940_filter_mode_enum
125= SOC_ENUM_SINGLE(WM8940_ADC, 7, 2, wm8940_filter_mode_text);
126
127static DECLARE_TLV_DB_SCALE(wm8940_spk_vol_tlv, -5700, 100, 1);
128static DECLARE_TLV_DB_SCALE(wm8940_att_tlv, -1000, 1000, 0);
129static DECLARE_TLV_DB_SCALE(wm8940_pga_vol_tlv, -1200, 75, 0);
130static DECLARE_TLV_DB_SCALE(wm8940_alc_min_tlv, -1200, 600, 0);
131static DECLARE_TLV_DB_SCALE(wm8940_alc_max_tlv, 675, 600, 0);
132static DECLARE_TLV_DB_SCALE(wm8940_alc_tar_tlv, -2250, 50, 0);
133static DECLARE_TLV_DB_SCALE(wm8940_lim_boost_tlv, 0, 100, 0);
134static DECLARE_TLV_DB_SCALE(wm8940_lim_thresh_tlv, -600, 100, 0);
135static DECLARE_TLV_DB_SCALE(wm8940_adc_tlv, -12750, 50, 1);
136static DECLARE_TLV_DB_SCALE(wm8940_capture_boost_vol_tlv, 0, 2000, 0);
137
138static const struct snd_kcontrol_new wm8940_snd_controls[] = {
139	SOC_SINGLE("Digital Loopback Switch", WM8940_COMPANDINGCTL,
140		   6, 1, 0),
141	SOC_ENUM("DAC Companding", wm8940_dac_companding_enum),
142	SOC_ENUM("ADC Companding", wm8940_adc_companding_enum),
143
144	SOC_ENUM("ALC Mode", wm8940_alc_mode_enum),
145	SOC_SINGLE("ALC Switch", WM8940_ALC1, 8, 1, 0),
146	SOC_SINGLE_TLV("ALC Capture Max Gain", WM8940_ALC1,
147		       3, 7, 1, wm8940_alc_max_tlv),
148	SOC_SINGLE_TLV("ALC Capture Min Gain", WM8940_ALC1,
149		       0, 7, 0, wm8940_alc_min_tlv),
150	SOC_SINGLE_TLV("ALC Capture Target", WM8940_ALC2,
151		       0, 14, 0, wm8940_alc_tar_tlv),
152	SOC_SINGLE("ALC Capture Hold", WM8940_ALC2, 4, 10, 0),
153	SOC_SINGLE("ALC Capture Decay", WM8940_ALC3, 4, 10, 0),
154	SOC_SINGLE("ALC Capture Attach", WM8940_ALC3, 0, 10, 0),
155	SOC_SINGLE("ALC ZC Switch", WM8940_ALC4, 1, 1, 0),
156	SOC_SINGLE("ALC Capture Noise Gate Switch", WM8940_NOISEGATE,
157		   3, 1, 0),
158	SOC_SINGLE("ALC Capture Noise Gate Threshold", WM8940_NOISEGATE,
159		   0, 7, 0),
160
161	SOC_SINGLE("DAC Playback Limiter Switch", WM8940_DACLIM1, 8, 1, 0),
162	SOC_SINGLE("DAC Playback Limiter Attack", WM8940_DACLIM1, 0, 9, 0),
163	SOC_SINGLE("DAC Playback Limiter Decay", WM8940_DACLIM1, 4, 11, 0),
164	SOC_SINGLE_TLV("DAC Playback Limiter Threshold", WM8940_DACLIM2,
165		       4, 9, 1, wm8940_lim_thresh_tlv),
166	SOC_SINGLE_TLV("DAC Playback Limiter Boost", WM8940_DACLIM2,
167		       0, 12, 0, wm8940_lim_boost_tlv),
168
169	SOC_SINGLE("Capture PGA ZC Switch", WM8940_PGAGAIN, 7, 1, 0),
170	SOC_SINGLE_TLV("Capture PGA Volume", WM8940_PGAGAIN,
171		       0, 63, 0, wm8940_pga_vol_tlv),
172	SOC_SINGLE_TLV("Digital Playback Volume", WM8940_DACVOL,
173		       0, 255, 0, wm8940_adc_tlv),
174	SOC_SINGLE_TLV("Digital Capture Volume", WM8940_ADCVOL,
175		       0, 255, 0, wm8940_adc_tlv),
176	SOC_ENUM("Mic Bias Level", wm8940_mic_bias_level_enum),
177	SOC_SINGLE_TLV("Capture Boost Volue", WM8940_ADCBOOST,
178		       8, 1, 0, wm8940_capture_boost_vol_tlv),
179	SOC_SINGLE_TLV("Speaker Playback Volume", WM8940_SPKVOL,
180		       0, 63, 0, wm8940_spk_vol_tlv),
181	SOC_SINGLE("Speaker Playback Switch", WM8940_SPKVOL,  6, 1, 1),
182
183	SOC_SINGLE_TLV("Speaker Mixer Line Bypass Volume", WM8940_SPKVOL,
184		       8, 1, 1, wm8940_att_tlv),
185	SOC_SINGLE("Speaker Playback ZC Switch", WM8940_SPKVOL, 7, 1, 0),
186
187	SOC_SINGLE("Mono Out Switch", WM8940_MONOMIX, 6, 1, 1),
188	SOC_SINGLE_TLV("Mono Mixer Line Bypass Volume", WM8940_MONOMIX,
189		       7, 1, 1, wm8940_att_tlv),
190
191	SOC_SINGLE("High Pass Filter Switch", WM8940_ADC, 8, 1, 0),
192	SOC_ENUM("High Pass Filter Mode", wm8940_filter_mode_enum),
193	SOC_SINGLE("High Pass Filter Cut Off", WM8940_ADC, 4, 7, 0),
194	SOC_SINGLE("ADC Inversion Switch", WM8940_ADC, 0, 1, 0),
195	SOC_SINGLE("DAC Inversion Switch", WM8940_DAC, 0, 1, 0),
196	SOC_SINGLE("DAC Auto Mute Switch", WM8940_DAC, 2, 1, 0),
197	SOC_SINGLE("ZC Timeout Clock Switch", WM8940_ADDCNTRL, 0, 1, 0),
198};
199
200static const struct snd_kcontrol_new wm8940_speaker_mixer_controls[] = {
201	SOC_DAPM_SINGLE("Line Bypass Switch", WM8940_SPKMIX, 1, 1, 0),
202	SOC_DAPM_SINGLE("Aux Playback Switch", WM8940_SPKMIX, 5, 1, 0),
203	SOC_DAPM_SINGLE("PCM Playback Switch", WM8940_SPKMIX, 0, 1, 0),
204};
205
206static const struct snd_kcontrol_new wm8940_mono_mixer_controls[] = {
207	SOC_DAPM_SINGLE("Line Bypass Switch", WM8940_MONOMIX, 1, 1, 0),
208	SOC_DAPM_SINGLE("Aux Playback Switch", WM8940_MONOMIX, 2, 1, 0),
209	SOC_DAPM_SINGLE("PCM Playback Switch", WM8940_MONOMIX, 0, 1, 0),
210};
211
212static DECLARE_TLV_DB_SCALE(wm8940_boost_vol_tlv, -1500, 300, 1);
213static const struct snd_kcontrol_new wm8940_input_boost_controls[] = {
214	SOC_DAPM_SINGLE("Mic PGA Switch", WM8940_PGAGAIN, 6, 1, 1),
215	SOC_DAPM_SINGLE_TLV("Aux Volume", WM8940_ADCBOOST,
216			    0, 7, 0, wm8940_boost_vol_tlv),
217	SOC_DAPM_SINGLE_TLV("Mic Volume", WM8940_ADCBOOST,
218			    4, 7, 0, wm8940_boost_vol_tlv),
219};
220
221static const struct snd_kcontrol_new wm8940_micpga_controls[] = {
222	SOC_DAPM_SINGLE("AUX Switch", WM8940_INPUTCTL, 2, 1, 0),
223	SOC_DAPM_SINGLE("MICP Switch", WM8940_INPUTCTL, 0, 1, 0),
224	SOC_DAPM_SINGLE("MICN Switch", WM8940_INPUTCTL, 1, 1, 0),
225};
226
227static const struct snd_soc_dapm_widget wm8940_dapm_widgets[] = {
228	SND_SOC_DAPM_MIXER("Speaker Mixer", WM8940_POWER3, 2, 0,
229			   &wm8940_speaker_mixer_controls[0],
230			   ARRAY_SIZE(wm8940_speaker_mixer_controls)),
231	SND_SOC_DAPM_MIXER("Mono Mixer", WM8940_POWER3, 3, 0,
232			   &wm8940_mono_mixer_controls[0],
233			   ARRAY_SIZE(wm8940_mono_mixer_controls)),
234	SND_SOC_DAPM_DAC("DAC", "HiFi Playback", WM8940_POWER3, 0, 0),
235
236	SND_SOC_DAPM_PGA("SpkN Out", WM8940_POWER3, 5, 0, NULL, 0),
237	SND_SOC_DAPM_PGA("SpkP Out", WM8940_POWER3, 6, 0, NULL, 0),
238	SND_SOC_DAPM_PGA("Mono Out", WM8940_POWER3, 7, 0, NULL, 0),
239	SND_SOC_DAPM_OUTPUT("MONOOUT"),
240	SND_SOC_DAPM_OUTPUT("SPKOUTP"),
241	SND_SOC_DAPM_OUTPUT("SPKOUTN"),
242
243	SND_SOC_DAPM_PGA("Aux Input", WM8940_POWER1, 6, 0, NULL, 0),
244	SND_SOC_DAPM_ADC("ADC", "HiFi Capture", WM8940_POWER2, 0, 0),
245	SND_SOC_DAPM_MIXER("Mic PGA", WM8940_POWER2, 2, 0,
246			   &wm8940_micpga_controls[0],
247			   ARRAY_SIZE(wm8940_micpga_controls)),
248	SND_SOC_DAPM_MIXER("Boost Mixer", WM8940_POWER2, 4, 0,
249			   &wm8940_input_boost_controls[0],
250			   ARRAY_SIZE(wm8940_input_boost_controls)),
251	SND_SOC_DAPM_MICBIAS("Mic Bias", WM8940_POWER1, 4, 0),
252
253	SND_SOC_DAPM_INPUT("MICN"),
254	SND_SOC_DAPM_INPUT("MICP"),
255	SND_SOC_DAPM_INPUT("AUX"),
256};
257
258static const struct snd_soc_dapm_route audio_map[] = {
259	/* Mono output mixer */
260	{"Mono Mixer", "PCM Playback Switch", "DAC"},
261	{"Mono Mixer", "Aux Playback Switch", "Aux Input"},
262	{"Mono Mixer", "Line Bypass Switch", "Boost Mixer"},
263
264	/* Speaker output mixer */
265	{"Speaker Mixer", "PCM Playback Switch", "DAC"},
266	{"Speaker Mixer", "Aux Playback Switch", "Aux Input"},
267	{"Speaker Mixer", "Line Bypass Switch", "Boost Mixer"},
268
269	/* Outputs */
270	{"Mono Out", NULL, "Mono Mixer"},
271	{"MONOOUT", NULL, "Mono Out"},
272	{"SpkN Out", NULL, "Speaker Mixer"},
273	{"SpkP Out", NULL, "Speaker Mixer"},
274	{"SPKOUTN", NULL, "SpkN Out"},
275	{"SPKOUTP", NULL, "SpkP Out"},
276
277	/*  Microphone PGA */
278	{"Mic PGA", "MICN Switch", "MICN"},
279	{"Mic PGA", "MICP Switch", "MICP"},
280	{"Mic PGA", "AUX Switch", "AUX"},
281
282	/* Boost Mixer */
283	{"Boost Mixer", "Mic PGA Switch", "Mic PGA"},
284	{"Boost Mixer", "Mic Volume",  "MICP"},
285	{"Boost Mixer", "Aux Volume", "Aux Input"},
286
287	{"ADC", NULL, "Boost Mixer"},
288};
289
290static int wm8940_add_widgets(struct snd_soc_codec *codec)
291{
292	struct snd_soc_dapm_context *dapm = &codec->dapm;
293	int ret;
294
295	ret = snd_soc_dapm_new_controls(dapm, wm8940_dapm_widgets,
296					ARRAY_SIZE(wm8940_dapm_widgets));
297	if (ret)
298		goto error_ret;
299	ret = snd_soc_dapm_add_routes(dapm, audio_map, ARRAY_SIZE(audio_map));
300	if (ret)
301		goto error_ret;
302
303error_ret:
304	return ret;
305}
306
307#define wm8940_reset(c) snd_soc_write(c, WM8940_SOFTRESET, 0);
308
309static int wm8940_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 = snd_soc_read(codec, WM8940_IFACE) & 0xFE67;
314	u16 clk = snd_soc_read(codec, WM8940_CLOCK) & 0x1fe;
315
316	switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
317	case SND_SOC_DAIFMT_CBM_CFM:
318		clk |= 1;
319		break;
320	case SND_SOC_DAIFMT_CBS_CFS:
321		break;
322	default:
323		return -EINVAL;
324	}
325	snd_soc_write(codec, WM8940_CLOCK, clk);
326
327	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
328	case SND_SOC_DAIFMT_I2S:
329		iface |= (2 << 3);
330		break;
331	case SND_SOC_DAIFMT_LEFT_J:
332		iface |= (1 << 3);
333		break;
334	case SND_SOC_DAIFMT_RIGHT_J:
335		break;
336	case SND_SOC_DAIFMT_DSP_A:
337		iface |= (3 << 3);
338		break;
339	case SND_SOC_DAIFMT_DSP_B:
340		iface |= (3 << 3) | (1 << 7);
341		break;
342	}
343
344	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
345	case SND_SOC_DAIFMT_NB_NF:
346		break;
347	case SND_SOC_DAIFMT_NB_IF:
348		iface |= (1 << 7);
349		break;
350	case SND_SOC_DAIFMT_IB_NF:
351		iface |= (1 << 8);
352		break;
353	case SND_SOC_DAIFMT_IB_IF:
354		iface |= (1 << 8) | (1 << 7);
355		break;
356	}
357
358	snd_soc_write(codec, WM8940_IFACE, iface);
359
360	return 0;
361}
362
363static int wm8940_i2s_hw_params(struct snd_pcm_substream *substream,
364				struct snd_pcm_hw_params *params,
365				struct snd_soc_dai *dai)
366{
367	struct snd_soc_pcm_runtime *rtd = substream->private_data;
368	struct snd_soc_codec *codec = rtd->codec;
369	u16 iface = snd_soc_read(codec, WM8940_IFACE) & 0xFD9F;
370	u16 addcntrl = snd_soc_read(codec, WM8940_ADDCNTRL) & 0xFFF1;
371	u16 companding =  snd_soc_read(codec,
372						WM8940_COMPANDINGCTL) & 0xFFDF;
373	int ret;
374
375	/* LoutR control */
376	if (substream->stream == SNDRV_PCM_STREAM_CAPTURE
377	    && params_channels(params) == 2)
378		iface |= (1 << 9);
379
380	switch (params_rate(params)) {
381	case 8000:
382		addcntrl |= (0x5 << 1);
383		break;
384	case 11025:
385		addcntrl |= (0x4 << 1);
386		break;
387	case 16000:
388		addcntrl |= (0x3 << 1);
389		break;
390	case 22050:
391		addcntrl |= (0x2 << 1);
392		break;
393	case 32000:
394		addcntrl |= (0x1 << 1);
395		break;
396	case 44100:
397	case 48000:
398		break;
399	}
400	ret = snd_soc_write(codec, WM8940_ADDCNTRL, addcntrl);
401	if (ret)
402		goto error_ret;
403
404	switch (params_format(params)) {
405	case SNDRV_PCM_FORMAT_S8:
406		companding = companding | (1 << 5);
407		break;
408	case SNDRV_PCM_FORMAT_S16_LE:
409		break;
410	case SNDRV_PCM_FORMAT_S20_3LE:
411		iface |= (1 << 5);
412		break;
413	case SNDRV_PCM_FORMAT_S24_LE:
414		iface |= (2 << 5);
415		break;
416	case SNDRV_PCM_FORMAT_S32_LE:
417		iface |= (3 << 5);
418		break;
419	}
420	ret = snd_soc_write(codec, WM8940_COMPANDINGCTL, companding);
421	if (ret)
422		goto error_ret;
423	ret = snd_soc_write(codec, WM8940_IFACE, iface);
424
425error_ret:
426	return ret;
427}
428
429static int wm8940_mute(struct snd_soc_dai *dai, int mute)
430{
431	struct snd_soc_codec *codec = dai->codec;
432	u16 mute_reg = snd_soc_read(codec, WM8940_DAC) & 0xffbf;
433
434	if (mute)
435		mute_reg |= 0x40;
436
437	return snd_soc_write(codec, WM8940_DAC, mute_reg);
438}
439
440static int wm8940_set_bias_level(struct snd_soc_codec *codec,
441				 enum snd_soc_bias_level level)
442{
443	u16 val;
444	u16 pwr_reg = snd_soc_read(codec, WM8940_POWER1) & 0x1F0;
445	int ret = 0;
446
447	switch (level) {
448	case SND_SOC_BIAS_ON:
449		/* ensure bufioen and biasen */
450		pwr_reg |= (1 << 2) | (1 << 3);
451		/* Enable thermal shutdown */
452		val = snd_soc_read(codec, WM8940_OUTPUTCTL);
453		ret = snd_soc_write(codec, WM8940_OUTPUTCTL, val | 0x2);
454		if (ret)
455			break;
456		/* set vmid to 75k */
457		ret = snd_soc_write(codec, WM8940_POWER1, pwr_reg | 0x1);
458		break;
459	case SND_SOC_BIAS_PREPARE:
460		/* ensure bufioen and biasen */
461		pwr_reg |= (1 << 2) | (1 << 3);
462		ret = snd_soc_write(codec, WM8940_POWER1, pwr_reg | 0x1);
463		break;
464	case SND_SOC_BIAS_STANDBY:
465		/* ensure bufioen and biasen */
466		pwr_reg |= (1 << 2) | (1 << 3);
467		/* set vmid to 300k for standby */
468		ret = snd_soc_write(codec, WM8940_POWER1, pwr_reg | 0x2);
469		break;
470	case SND_SOC_BIAS_OFF:
471		ret = snd_soc_write(codec, WM8940_POWER1, pwr_reg);
472		break;
473	}
474
475	codec->dapm.bias_level = level;
476
477	return ret;
478}
479
480struct pll_ {
481	unsigned int pre_scale:2;
482	unsigned int n:4;
483	unsigned int k;
484};
485
486static struct pll_ pll_div;
487
488/* The size in bits of the pll divide multiplied by 10
489 * to allow rounding later */
490#define FIXED_PLL_SIZE ((1 << 24) * 10)
491static void pll_factors(unsigned int target, unsigned int source)
492{
493	unsigned long long Kpart;
494	unsigned int K, Ndiv, Nmod;
495	/* The left shift ist to avoid accuracy loss when right shifting */
496	Ndiv = target / source;
497
498	if (Ndiv > 12) {
499		source <<= 1;
500		/* Multiply by 2 */
501		pll_div.pre_scale = 0;
502		Ndiv = target / source;
503	} else if (Ndiv < 3) {
504		source >>= 2;
505		/* Divide by 4 */
506		pll_div.pre_scale = 3;
507		Ndiv = target / source;
508	} else if (Ndiv < 6) {
509		source >>= 1;
510		/* divide by 2 */
511		pll_div.pre_scale = 2;
512		Ndiv = target / source;
513	} else
514		pll_div.pre_scale = 1;
515
516	if ((Ndiv < 6) || (Ndiv > 12))
517		printk(KERN_WARNING
518			"WM8940 N value %d outwith recommended range!d\n",
519			Ndiv);
520
521	pll_div.n = Ndiv;
522	Nmod = target % source;
523	Kpart = FIXED_PLL_SIZE * (long long)Nmod;
524
525	do_div(Kpart, source);
526
527	K = Kpart & 0xFFFFFFFF;
528
529	/* Check if we need to round */
530	if ((K % 10) >= 5)
531		K += 5;
532
533	/* Move down to proper range now rounding is done */
534	K /= 10;
535
536	pll_div.k = K;
537}
538
539/* Untested at the moment */
540static int wm8940_set_dai_pll(struct snd_soc_dai *codec_dai, int pll_id,
541		int source, unsigned int freq_in, unsigned int freq_out)
542{
543	struct snd_soc_codec *codec = codec_dai->codec;
544	u16 reg;
545
546	/* Turn off PLL */
547	reg = snd_soc_read(codec, WM8940_POWER1);
548	snd_soc_write(codec, WM8940_POWER1, reg & 0x1df);
549
550	if (freq_in == 0 || freq_out == 0) {
551		/* Clock CODEC directly from MCLK */
552		reg = snd_soc_read(codec, WM8940_CLOCK);
553		snd_soc_write(codec, WM8940_CLOCK, reg & 0x0ff);
554		/* Pll power down */
555		snd_soc_write(codec, WM8940_PLLN, (1 << 7));
556		return 0;
557	}
558
559	/* Pll is followed by a frequency divide by 4 */
560	pll_factors(freq_out*4, freq_in);
561	if (pll_div.k)
562		snd_soc_write(codec, WM8940_PLLN,
563			     (pll_div.pre_scale << 4) | pll_div.n | (1 << 6));
564	else /* No factional component */
565		snd_soc_write(codec, WM8940_PLLN,
566			     (pll_div.pre_scale << 4) | pll_div.n);
567	snd_soc_write(codec, WM8940_PLLK1, pll_div.k >> 18);
568	snd_soc_write(codec, WM8940_PLLK2, (pll_div.k >> 9) & 0x1ff);
569	snd_soc_write(codec, WM8940_PLLK3, pll_div.k & 0x1ff);
570	/* Enable the PLL */
571	reg = snd_soc_read(codec, WM8940_POWER1);
572	snd_soc_write(codec, WM8940_POWER1, reg | 0x020);
573
574	/* Run CODEC from PLL instead of MCLK */
575	reg = snd_soc_read(codec, WM8940_CLOCK);
576	snd_soc_write(codec, WM8940_CLOCK, reg | 0x100);
577
578	return 0;
579}
580
581static int wm8940_set_dai_sysclk(struct snd_soc_dai *codec_dai,
582				 int clk_id, unsigned int freq, int dir)
583{
584	struct snd_soc_codec *codec = codec_dai->codec;
585	struct wm8940_priv *wm8940 = snd_soc_codec_get_drvdata(codec);
586
587	switch (freq) {
588	case 11289600:
589	case 12000000:
590	case 12288000:
591	case 16934400:
592	case 18432000:
593		wm8940->sysclk = freq;
594		return 0;
595	}
596	return -EINVAL;
597}
598
599static int wm8940_set_dai_clkdiv(struct snd_soc_dai *codec_dai,
600				 int div_id, int div)
601{
602	struct snd_soc_codec *codec = codec_dai->codec;
603	u16 reg;
604	int ret = 0;
605
606	switch (div_id) {
607	case WM8940_BCLKDIV:
608		reg = snd_soc_read(codec, WM8940_CLOCK) & 0xFFEF3;
609		ret = snd_soc_write(codec, WM8940_CLOCK, reg | (div << 2));
610		break;
611	case WM8940_MCLKDIV:
612		reg = snd_soc_read(codec, WM8940_CLOCK) & 0xFF1F;
613		ret = snd_soc_write(codec, WM8940_CLOCK, reg | (div << 5));
614		break;
615	case WM8940_OPCLKDIV:
616		reg = snd_soc_read(codec, WM8940_ADDCNTRL) & 0xFFCF;
617		ret = snd_soc_write(codec, WM8940_ADDCNTRL, reg | (div << 4));
618		break;
619	}
620	return ret;
621}
622
623#define WM8940_RATES SNDRV_PCM_RATE_8000_48000
624
625#define WM8940_FORMATS (SNDRV_PCM_FMTBIT_S8 |				\
626			SNDRV_PCM_FMTBIT_S16_LE |			\
627			SNDRV_PCM_FMTBIT_S20_3LE |			\
628			SNDRV_PCM_FMTBIT_S24_LE |			\
629			SNDRV_PCM_FMTBIT_S32_LE)
630
631static struct snd_soc_dai_ops wm8940_dai_ops = {
632	.hw_params = wm8940_i2s_hw_params,
633	.set_sysclk = wm8940_set_dai_sysclk,
634	.digital_mute = wm8940_mute,
635	.set_fmt = wm8940_set_dai_fmt,
636	.set_clkdiv = wm8940_set_dai_clkdiv,
637	.set_pll = wm8940_set_dai_pll,
638};
639
640static struct snd_soc_dai_driver wm8940_dai = {
641	.name = "wm8940-hifi",
642	.playback = {
643		.stream_name = "Playback",
644		.channels_min = 1,
645		.channels_max = 2,
646		.rates = WM8940_RATES,
647		.formats = WM8940_FORMATS,
648	},
649	.capture = {
650		.stream_name = "Capture",
651		.channels_min = 1,
652		.channels_max = 2,
653		.rates = WM8940_RATES,
654		.formats = WM8940_FORMATS,
655	},
656	.ops = &wm8940_dai_ops,
657	.symmetric_rates = 1,
658};
659
660static int wm8940_suspend(struct snd_soc_codec *codec, pm_message_t state)
661{
662	return wm8940_set_bias_level(codec, SND_SOC_BIAS_OFF);
663}
664
665static int wm8940_resume(struct snd_soc_codec *codec)
666{
667	int i;
668	int ret;
669	u8 data[3];
670	u16 *cache = codec->reg_cache;
671
672	/* Sync reg_cache with the hardware
673	 * Could use auto incremented writes to speed this up
674	 */
675	for (i = 0; i < ARRAY_SIZE(wm8940_reg_defaults); i++) {
676		data[0] = i;
677		data[1] = (cache[i] & 0xFF00) >> 8;
678		data[2] = cache[i] & 0x00FF;
679		ret = codec->hw_write(codec->control_data, data, 3);
680		if (ret < 0)
681			goto error_ret;
682		else if (ret != 3) {
683			ret = -EIO;
684			goto error_ret;
685		}
686	}
687	ret = wm8940_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
688	if (ret)
689		goto error_ret;
690
691error_ret:
692	return ret;
693}
694
695static int wm8940_probe(struct snd_soc_codec *codec)
696{
697	struct wm8940_priv *wm8940 = snd_soc_codec_get_drvdata(codec);
698	struct wm8940_setup_data *pdata = codec->dev->platform_data;
699	int ret;
700	u16 reg;
701
702	codec->control_data = wm8940->control_data;
703	ret = snd_soc_codec_set_cache_io(codec, 8, 16, wm8940->control_type);
704	if (ret < 0) {
705		dev_err(codec->dev, "Failed to set cache I/O: %d\n", ret);
706		return ret;
707	}
708
709	ret = wm8940_reset(codec);
710	if (ret < 0) {
711		dev_err(codec->dev, "Failed to issue reset\n");
712		return ret;
713	}
714
715	wm8940_set_bias_level(codec, SND_SOC_BIAS_STANDBY);
716
717	ret = snd_soc_write(codec, WM8940_POWER1, 0x180);
718	if (ret < 0)
719		return ret;
720
721	if (!pdata)
722		dev_warn(codec->dev, "No platform data supplied\n");
723	else {
724		reg = snd_soc_read(codec, WM8940_OUTPUTCTL);
725		ret = snd_soc_write(codec, WM8940_OUTPUTCTL, reg | pdata->vroi);
726		if (ret < 0)
727			return ret;
728	}
729
730	ret = snd_soc_add_controls(codec, wm8940_snd_controls,
731			     ARRAY_SIZE(wm8940_snd_controls));
732	if (ret)
733		return ret;
734	ret = wm8940_add_widgets(codec);
735	if (ret)
736		return ret;
737
738	return ret;
739}
740
741static int wm8940_remove(struct snd_soc_codec *codec)
742{
743	wm8940_set_bias_level(codec, SND_SOC_BIAS_OFF);
744	return 0;
745}
746
747static struct snd_soc_codec_driver soc_codec_dev_wm8940 = {
748	.probe =	wm8940_probe,
749	.remove =	wm8940_remove,
750	.suspend =	wm8940_suspend,
751	.resume =	wm8940_resume,
752	.set_bias_level = wm8940_set_bias_level,
753	.reg_cache_size = ARRAY_SIZE(wm8940_reg_defaults),
754	.reg_word_size = sizeof(u16),
755	.reg_cache_default = wm8940_reg_defaults,
756};
757
758#if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
759static __devinit int wm8940_i2c_probe(struct i2c_client *i2c,
760				      const struct i2c_device_id *id)
761{
762	struct wm8940_priv *wm8940;
763	int ret;
764
765	wm8940 = kzalloc(sizeof(struct wm8940_priv), GFP_KERNEL);
766	if (wm8940 == NULL)
767		return -ENOMEM;
768
769	i2c_set_clientdata(i2c, wm8940);
770	wm8940->control_data = i2c;
771	wm8940->control_type = SND_SOC_I2C;
772
773	ret = snd_soc_register_codec(&i2c->dev,
774			&soc_codec_dev_wm8940, &wm8940_dai, 1);
775	if (ret < 0)
776		kfree(wm8940);
777	return ret;
778}
779
780static __devexit int wm8940_i2c_remove(struct i2c_client *client)
781{
782	snd_soc_unregister_codec(&client->dev);
783	kfree(i2c_get_clientdata(client));
784	return 0;
785}
786
787static const struct i2c_device_id wm8940_i2c_id[] = {
788	{ "wm8940", 0 },
789	{ }
790};
791MODULE_DEVICE_TABLE(i2c, wm8940_i2c_id);
792
793static struct i2c_driver wm8940_i2c_driver = {
794	.driver = {
795		.name = "wm8940-codec",
796		.owner = THIS_MODULE,
797	},
798	.probe =    wm8940_i2c_probe,
799	.remove =   __devexit_p(wm8940_i2c_remove),
800	.id_table = wm8940_i2c_id,
801};
802#endif
803
804static int __init wm8940_modinit(void)
805{
806	int ret = 0;
807#if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
808	ret = i2c_add_driver(&wm8940_i2c_driver);
809	if (ret != 0) {
810		printk(KERN_ERR "Failed to register wm8940 I2C driver: %d\n",
811		       ret);
812	}
813#endif
814	return ret;
815}
816module_init(wm8940_modinit);
817
818static void __exit wm8940_exit(void)
819{
820#if defined(CONFIG_I2C) || defined(CONFIG_I2C_MODULE)
821	i2c_del_driver(&wm8940_i2c_driver);
822#endif
823}
824module_exit(wm8940_exit);
825
826MODULE_DESCRIPTION("ASoC WM8940 driver");
827MODULE_AUTHOR("Jonathan Cameron");
828MODULE_LICENSE("GPL");
829