1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * wm8731.c  --  WM8731 ALSA SoC Audio driver
4 *
5 * Copyright 2005 Openedhand Ltd.
6 * Copyright 2006-12 Wolfson Microelectronics, plc
7 *
8 * Author: Richard Purdie <richard@openedhand.com>
9 *
10 * Based on wm8753.c by Liam Girdwood
11 */
12
13#include <linux/module.h>
14#include <linux/moduleparam.h>
15#include <linux/init.h>
16#include <linux/delay.h>
17#include <linux/pm.h>
18#include <linux/i2c.h>
19#include <linux/slab.h>
20#include <linux/regmap.h>
21#include <linux/regulator/consumer.h>
22#include <linux/spi/spi.h>
23#include <linux/of_device.h>
24#include <linux/mutex.h>
25#include <linux/clk.h>
26#include <sound/core.h>
27#include <sound/pcm.h>
28#include <sound/pcm_params.h>
29#include <sound/soc.h>
30#include <sound/initval.h>
31#include <sound/tlv.h>
32
33#include "wm8731.h"
34
35#define WM8731_NUM_SUPPLIES 4
36static const char *wm8731_supply_names[WM8731_NUM_SUPPLIES] = {
37	"AVDD",
38	"HPVDD",
39	"DCVDD",
40	"DBVDD",
41};
42
43/* codec private data */
44struct wm8731_priv {
45	struct regmap *regmap;
46	struct clk *mclk;
47	struct regulator_bulk_data supplies[WM8731_NUM_SUPPLIES];
48	const struct snd_pcm_hw_constraint_list *constraints;
49	unsigned int sysclk;
50	int sysclk_type;
51	int playback_fs;
52	bool deemph;
53
54	struct mutex lock;
55};
56
57
58/*
59 * wm8731 register cache
60 */
61static const struct reg_default wm8731_reg_defaults[] = {
62	{ 0, 0x0097 },
63	{ 1, 0x0097 },
64	{ 2, 0x0079 },
65	{ 3, 0x0079 },
66	{ 4, 0x000a },
67	{ 5, 0x0008 },
68	{ 6, 0x009f },
69	{ 7, 0x000a },
70	{ 8, 0x0000 },
71	{ 9, 0x0000 },
72};
73
74static bool wm8731_volatile(struct device *dev, unsigned int reg)
75{
76	return reg == WM8731_RESET;
77}
78
79#define wm8731_reset(m)	regmap_write(m, WM8731_RESET, 0)
80
81static const char *wm8731_input_select[] = {"Line In", "Mic"};
82
83static SOC_ENUM_SINGLE_DECL(wm8731_insel_enum,
84			    WM8731_APANA, 2, wm8731_input_select);
85
86static int wm8731_deemph[] = { 0, 32000, 44100, 48000 };
87
88static int wm8731_set_deemph(struct snd_soc_component *component)
89{
90	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
91	int val, i, best;
92
93	/* If we're using deemphasis select the nearest available sample
94	 * rate.
95	 */
96	if (wm8731->deemph) {
97		best = 1;
98		for (i = 2; i < ARRAY_SIZE(wm8731_deemph); i++) {
99			if (abs(wm8731_deemph[i] - wm8731->playback_fs) <
100			    abs(wm8731_deemph[best] - wm8731->playback_fs))
101				best = i;
102		}
103
104		val = best << 1;
105	} else {
106		best = 0;
107		val = 0;
108	}
109
110	dev_dbg(component->dev, "Set deemphasis %d (%dHz)\n",
111		best, wm8731_deemph[best]);
112
113	return snd_soc_component_update_bits(component, WM8731_APDIGI, 0x6, val);
114}
115
116static int wm8731_get_deemph(struct snd_kcontrol *kcontrol,
117			     struct snd_ctl_elem_value *ucontrol)
118{
119	struct snd_soc_component *component = snd_soc_kcontrol_component(kcontrol);
120	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
121
122	ucontrol->value.integer.value[0] = wm8731->deemph;
123
124	return 0;
125}
126
127static int wm8731_put_deemph(struct snd_kcontrol *kcontrol,
128			     struct snd_ctl_elem_value *ucontrol)
129{
130	struct snd_soc_component *component = snd_soc_kcontrol_component(kcontrol);
131	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
132	unsigned int deemph = ucontrol->value.integer.value[0];
133	int ret = 0;
134
135	if (deemph > 1)
136		return -EINVAL;
137
138	mutex_lock(&wm8731->lock);
139	if (wm8731->deemph != deemph) {
140		wm8731->deemph = deemph;
141
142		wm8731_set_deemph(component);
143
144		ret = 1;
145	}
146	mutex_unlock(&wm8731->lock);
147
148	return ret;
149}
150
151static const DECLARE_TLV_DB_SCALE(in_tlv, -3450, 150, 0);
152static const DECLARE_TLV_DB_SCALE(sidetone_tlv, -1500, 300, 0);
153static const DECLARE_TLV_DB_SCALE(out_tlv, -12100, 100, 1);
154static const DECLARE_TLV_DB_SCALE(mic_tlv, 0, 2000, 0);
155
156static const struct snd_kcontrol_new wm8731_snd_controls[] = {
157
158SOC_DOUBLE_R_TLV("Master Playback Volume", WM8731_LOUT1V, WM8731_ROUT1V,
159		 0, 127, 0, out_tlv),
160SOC_DOUBLE_R("Master Playback ZC Switch", WM8731_LOUT1V, WM8731_ROUT1V,
161	7, 1, 0),
162
163SOC_DOUBLE_R_TLV("Capture Volume", WM8731_LINVOL, WM8731_RINVOL, 0, 31, 0,
164		 in_tlv),
165SOC_DOUBLE_R("Line Capture Switch", WM8731_LINVOL, WM8731_RINVOL, 7, 1, 1),
166
167SOC_SINGLE_TLV("Mic Boost Volume", WM8731_APANA, 0, 1, 0, mic_tlv),
168SOC_SINGLE("Mic Capture Switch", WM8731_APANA, 1, 1, 1),
169
170SOC_SINGLE_TLV("Sidetone Playback Volume", WM8731_APANA, 6, 3, 1,
171	       sidetone_tlv),
172
173SOC_SINGLE("ADC High Pass Filter Switch", WM8731_APDIGI, 0, 1, 1),
174SOC_SINGLE("Store DC Offset Switch", WM8731_APDIGI, 4, 1, 0),
175
176SOC_SINGLE_BOOL_EXT("Playback Deemphasis Switch", 0,
177		    wm8731_get_deemph, wm8731_put_deemph),
178};
179
180/* Output Mixer */
181static const struct snd_kcontrol_new wm8731_output_mixer_controls[] = {
182SOC_DAPM_SINGLE("Line Bypass Switch", WM8731_APANA, 3, 1, 0),
183SOC_DAPM_SINGLE("Mic Sidetone Switch", WM8731_APANA, 5, 1, 0),
184SOC_DAPM_SINGLE("HiFi Playback Switch", WM8731_APANA, 4, 1, 0),
185};
186
187/* Input mux */
188static const struct snd_kcontrol_new wm8731_input_mux_controls =
189SOC_DAPM_ENUM("Input Select", wm8731_insel_enum);
190
191static const struct snd_soc_dapm_widget wm8731_dapm_widgets[] = {
192SND_SOC_DAPM_SUPPLY("ACTIVE",WM8731_ACTIVE, 0, 0, NULL, 0),
193SND_SOC_DAPM_SUPPLY("OSC", WM8731_PWR, 5, 1, NULL, 0),
194SND_SOC_DAPM_MIXER("Output Mixer", WM8731_PWR, 4, 1,
195	&wm8731_output_mixer_controls[0],
196	ARRAY_SIZE(wm8731_output_mixer_controls)),
197SND_SOC_DAPM_DAC("DAC", "HiFi Playback", WM8731_PWR, 3, 1),
198SND_SOC_DAPM_OUTPUT("LOUT"),
199SND_SOC_DAPM_OUTPUT("LHPOUT"),
200SND_SOC_DAPM_OUTPUT("ROUT"),
201SND_SOC_DAPM_OUTPUT("RHPOUT"),
202SND_SOC_DAPM_ADC("ADC", "HiFi Capture", WM8731_PWR, 2, 1),
203SND_SOC_DAPM_MUX("Input Mux", SND_SOC_NOPM, 0, 0, &wm8731_input_mux_controls),
204SND_SOC_DAPM_PGA("Line Input", WM8731_PWR, 0, 1, NULL, 0),
205SND_SOC_DAPM_MICBIAS("Mic Bias", WM8731_PWR, 1, 1),
206SND_SOC_DAPM_INPUT("MICIN"),
207SND_SOC_DAPM_INPUT("RLINEIN"),
208SND_SOC_DAPM_INPUT("LLINEIN"),
209};
210
211static int wm8731_check_osc(struct snd_soc_dapm_widget *source,
212			    struct snd_soc_dapm_widget *sink)
213{
214	struct snd_soc_component *component = snd_soc_dapm_to_component(source->dapm);
215	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
216
217	return wm8731->sysclk_type == WM8731_SYSCLK_XTAL;
218}
219
220static const struct snd_soc_dapm_route wm8731_intercon[] = {
221	{"DAC", NULL, "OSC", wm8731_check_osc},
222	{"ADC", NULL, "OSC", wm8731_check_osc},
223	{"DAC", NULL, "ACTIVE"},
224	{"ADC", NULL, "ACTIVE"},
225
226	/* output mixer */
227	{"Output Mixer", "Line Bypass Switch", "Line Input"},
228	{"Output Mixer", "HiFi Playback Switch", "DAC"},
229	{"Output Mixer", "Mic Sidetone Switch", "Mic Bias"},
230
231	/* outputs */
232	{"RHPOUT", NULL, "Output Mixer"},
233	{"ROUT", NULL, "Output Mixer"},
234	{"LHPOUT", NULL, "Output Mixer"},
235	{"LOUT", NULL, "Output Mixer"},
236
237	/* input mux */
238	{"Input Mux", "Line In", "Line Input"},
239	{"Input Mux", "Mic", "Mic Bias"},
240	{"ADC", NULL, "Input Mux"},
241
242	/* inputs */
243	{"Line Input", NULL, "LLINEIN"},
244	{"Line Input", NULL, "RLINEIN"},
245	{"Mic Bias", NULL, "MICIN"},
246};
247
248struct _coeff_div {
249	u32 mclk;
250	u32 rate;
251	u16 fs;
252	u8 sr:4;
253	u8 bosr:1;
254	u8 usb:1;
255};
256
257/* codec mclk clock divider coefficients */
258static const struct _coeff_div coeff_div[] = {
259	/* 48k */
260	{12288000, 48000, 256, 0x0, 0x0, 0x0},
261	{18432000, 48000, 384, 0x0, 0x1, 0x0},
262	{12000000, 48000, 250, 0x0, 0x0, 0x1},
263
264	/* 32k */
265	{12288000, 32000, 384, 0x6, 0x0, 0x0},
266	{18432000, 32000, 576, 0x6, 0x1, 0x0},
267	{12000000, 32000, 375, 0x6, 0x0, 0x1},
268
269	/* 8k */
270	{12288000, 8000, 1536, 0x3, 0x0, 0x0},
271	{18432000, 8000, 2304, 0x3, 0x1, 0x0},
272	{11289600, 8000, 1408, 0xb, 0x0, 0x0},
273	{16934400, 8000, 2112, 0xb, 0x1, 0x0},
274	{12000000, 8000, 1500, 0x3, 0x0, 0x1},
275
276	/* 96k */
277	{12288000, 96000, 128, 0x7, 0x0, 0x0},
278	{18432000, 96000, 192, 0x7, 0x1, 0x0},
279	{12000000, 96000, 125, 0x7, 0x0, 0x1},
280
281	/* 44.1k */
282	{11289600, 44100, 256, 0x8, 0x0, 0x0},
283	{16934400, 44100, 384, 0x8, 0x1, 0x0},
284	{12000000, 44100, 272, 0x8, 0x1, 0x1},
285
286	/* 88.2k */
287	{11289600, 88200, 128, 0xf, 0x0, 0x0},
288	{16934400, 88200, 192, 0xf, 0x1, 0x0},
289	{12000000, 88200, 136, 0xf, 0x1, 0x1},
290};
291
292/* rates constraints */
293static const unsigned int wm8731_rates_12000000[] = {
294	8000, 32000, 44100, 48000, 96000, 88200,
295};
296
297static const unsigned int wm8731_rates_12288000_18432000[] = {
298	8000, 32000, 48000, 96000,
299};
300
301static const unsigned int wm8731_rates_11289600_16934400[] = {
302	8000, 44100, 88200,
303};
304
305static const struct snd_pcm_hw_constraint_list wm8731_constraints_12000000 = {
306	.list = wm8731_rates_12000000,
307	.count = ARRAY_SIZE(wm8731_rates_12000000),
308};
309
310static const
311struct snd_pcm_hw_constraint_list wm8731_constraints_12288000_18432000 = {
312	.list = wm8731_rates_12288000_18432000,
313	.count = ARRAY_SIZE(wm8731_rates_12288000_18432000),
314};
315
316static const
317struct snd_pcm_hw_constraint_list wm8731_constraints_11289600_16934400 = {
318	.list = wm8731_rates_11289600_16934400,
319	.count = ARRAY_SIZE(wm8731_rates_11289600_16934400),
320};
321
322static inline int get_coeff(int mclk, int rate)
323{
324	int i;
325
326	for (i = 0; i < ARRAY_SIZE(coeff_div); i++) {
327		if (coeff_div[i].rate == rate && coeff_div[i].mclk == mclk)
328			return i;
329	}
330	return 0;
331}
332
333static int wm8731_hw_params(struct snd_pcm_substream *substream,
334			    struct snd_pcm_hw_params *params,
335			    struct snd_soc_dai *dai)
336{
337	struct snd_soc_component *component = dai->component;
338	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
339	u16 iface = snd_soc_component_read(component, WM8731_IFACE) & 0xfff3;
340	int i = get_coeff(wm8731->sysclk, params_rate(params));
341	u16 srate = (coeff_div[i].sr << 2) |
342		(coeff_div[i].bosr << 1) | coeff_div[i].usb;
343
344	wm8731->playback_fs = params_rate(params);
345
346	snd_soc_component_write(component, WM8731_SRATE, srate);
347
348	/* bit size */
349	switch (params_width(params)) {
350	case 16:
351		break;
352	case 20:
353		iface |= 0x0004;
354		break;
355	case 24:
356		iface |= 0x0008;
357		break;
358	case 32:
359		iface |= 0x000c;
360		break;
361	}
362
363	wm8731_set_deemph(component);
364
365	snd_soc_component_write(component, WM8731_IFACE, iface);
366	return 0;
367}
368
369static int wm8731_mute(struct snd_soc_dai *dai, int mute, int direction)
370{
371	struct snd_soc_component *component = dai->component;
372	u16 mute_reg = snd_soc_component_read(component, WM8731_APDIGI) & 0xfff7;
373
374	if (mute)
375		snd_soc_component_write(component, WM8731_APDIGI, mute_reg | 0x8);
376	else
377		snd_soc_component_write(component, WM8731_APDIGI, mute_reg);
378	return 0;
379}
380
381static int wm8731_set_dai_sysclk(struct snd_soc_dai *codec_dai,
382		int clk_id, unsigned int freq, int dir)
383{
384	struct snd_soc_component *component = codec_dai->component;
385	struct snd_soc_dapm_context *dapm = snd_soc_component_get_dapm(component);
386	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
387
388	switch (clk_id) {
389	case WM8731_SYSCLK_XTAL:
390	case WM8731_SYSCLK_MCLK:
391		if (wm8731->mclk && clk_set_rate(wm8731->mclk, freq))
392			return -EINVAL;
393		wm8731->sysclk_type = clk_id;
394		break;
395	default:
396		return -EINVAL;
397	}
398
399	switch (freq) {
400	case 0:
401		wm8731->constraints = NULL;
402		break;
403	case 12000000:
404		wm8731->constraints = &wm8731_constraints_12000000;
405		break;
406	case 12288000:
407	case 18432000:
408		wm8731->constraints = &wm8731_constraints_12288000_18432000;
409		break;
410	case 16934400:
411	case 11289600:
412		wm8731->constraints = &wm8731_constraints_11289600_16934400;
413		break;
414	default:
415		return -EINVAL;
416	}
417
418	wm8731->sysclk = freq;
419
420	snd_soc_dapm_sync(dapm);
421
422	return 0;
423}
424
425
426static int wm8731_set_dai_fmt(struct snd_soc_dai *codec_dai,
427		unsigned int fmt)
428{
429	struct snd_soc_component *component = codec_dai->component;
430	u16 iface = 0;
431
432	/* set master/slave audio interface */
433	switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
434	case SND_SOC_DAIFMT_CBM_CFM:
435		iface |= 0x0040;
436		break;
437	case SND_SOC_DAIFMT_CBS_CFS:
438		break;
439	default:
440		return -EINVAL;
441	}
442
443	/* interface format */
444	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
445	case SND_SOC_DAIFMT_I2S:
446		iface |= 0x0002;
447		break;
448	case SND_SOC_DAIFMT_RIGHT_J:
449		break;
450	case SND_SOC_DAIFMT_LEFT_J:
451		iface |= 0x0001;
452		break;
453	case SND_SOC_DAIFMT_DSP_A:
454		iface |= 0x0013;
455		break;
456	case SND_SOC_DAIFMT_DSP_B:
457		iface |= 0x0003;
458		break;
459	default:
460		return -EINVAL;
461	}
462
463	/* clock inversion */
464	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
465	case SND_SOC_DAIFMT_NB_NF:
466		break;
467	case SND_SOC_DAIFMT_IB_IF:
468		iface |= 0x0090;
469		break;
470	case SND_SOC_DAIFMT_IB_NF:
471		iface |= 0x0080;
472		break;
473	case SND_SOC_DAIFMT_NB_IF:
474		iface |= 0x0010;
475		break;
476	default:
477		return -EINVAL;
478	}
479
480	/* set iface */
481	snd_soc_component_write(component, WM8731_IFACE, iface);
482	return 0;
483}
484
485static int wm8731_set_bias_level(struct snd_soc_component *component,
486				 enum snd_soc_bias_level level)
487{
488	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(component);
489	int ret;
490	u16 reg;
491
492	switch (level) {
493	case SND_SOC_BIAS_ON:
494		if (wm8731->mclk) {
495			ret = clk_prepare_enable(wm8731->mclk);
496			if (ret)
497				return ret;
498		}
499		break;
500	case SND_SOC_BIAS_PREPARE:
501		break;
502	case SND_SOC_BIAS_STANDBY:
503		if (snd_soc_component_get_bias_level(component) == SND_SOC_BIAS_OFF) {
504			ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
505						    wm8731->supplies);
506			if (ret != 0)
507				return ret;
508
509			regcache_sync(wm8731->regmap);
510		}
511
512		/* Clear PWROFF, gate CLKOUT, everything else as-is */
513		reg = snd_soc_component_read(component, WM8731_PWR) & 0xff7f;
514		snd_soc_component_write(component, WM8731_PWR, reg | 0x0040);
515		break;
516	case SND_SOC_BIAS_OFF:
517		if (wm8731->mclk)
518			clk_disable_unprepare(wm8731->mclk);
519		snd_soc_component_write(component, WM8731_PWR, 0xffff);
520		regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies),
521				       wm8731->supplies);
522		regcache_mark_dirty(wm8731->regmap);
523		break;
524	}
525	return 0;
526}
527
528static int wm8731_startup(struct snd_pcm_substream *substream,
529	struct snd_soc_dai *dai)
530{
531	struct wm8731_priv *wm8731 = snd_soc_component_get_drvdata(dai->component);
532
533	if (wm8731->constraints)
534		snd_pcm_hw_constraint_list(substream->runtime, 0,
535					   SNDRV_PCM_HW_PARAM_RATE,
536					   wm8731->constraints);
537
538	return 0;
539}
540
541#define WM8731_RATES SNDRV_PCM_RATE_8000_96000
542
543#define WM8731_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
544	SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S32_LE)
545
546static const struct snd_soc_dai_ops wm8731_dai_ops = {
547	.startup	= wm8731_startup,
548	.hw_params	= wm8731_hw_params,
549	.mute_stream	= wm8731_mute,
550	.set_sysclk	= wm8731_set_dai_sysclk,
551	.set_fmt	= wm8731_set_dai_fmt,
552	.no_capture_mute = 1,
553};
554
555static struct snd_soc_dai_driver wm8731_dai = {
556	.name = "wm8731-hifi",
557	.playback = {
558		.stream_name = "Playback",
559		.channels_min = 1,
560		.channels_max = 2,
561		.rates = WM8731_RATES,
562		.formats = WM8731_FORMATS,},
563	.capture = {
564		.stream_name = "Capture",
565		.channels_min = 1,
566		.channels_max = 2,
567		.rates = WM8731_RATES,
568		.formats = WM8731_FORMATS,},
569	.ops = &wm8731_dai_ops,
570	.symmetric_rates = 1,
571};
572
573static int wm8731_request_supplies(struct device *dev,
574		struct wm8731_priv *wm8731)
575{
576	int ret = 0, i;
577
578	for (i = 0; i < ARRAY_SIZE(wm8731->supplies); i++)
579		wm8731->supplies[i].supply = wm8731_supply_names[i];
580
581	ret = devm_regulator_bulk_get(dev, ARRAY_SIZE(wm8731->supplies),
582				 wm8731->supplies);
583	if (ret != 0) {
584		dev_err(dev, "Failed to request supplies: %d\n", ret);
585		return ret;
586	}
587
588	ret = regulator_bulk_enable(ARRAY_SIZE(wm8731->supplies),
589				    wm8731->supplies);
590	if (ret != 0) {
591		dev_err(dev, "Failed to enable supplies: %d\n", ret);
592		return ret;
593	}
594
595	return 0;
596}
597
598static int wm8731_hw_init(struct device *dev, struct wm8731_priv *wm8731)
599{
600	int ret = 0;
601
602	ret = wm8731_reset(wm8731->regmap);
603	if (ret < 0) {
604		dev_err(dev, "Failed to issue reset: %d\n", ret);
605		goto err;
606	}
607
608	/* Clear POWEROFF, keep everything else disabled */
609	regmap_write(wm8731->regmap, WM8731_PWR, 0x7f);
610
611	/* Latch the update bits */
612	regmap_update_bits(wm8731->regmap, WM8731_LOUT1V, 0x100, 0);
613	regmap_update_bits(wm8731->regmap, WM8731_ROUT1V, 0x100, 0);
614	regmap_update_bits(wm8731->regmap, WM8731_LINVOL, 0x100, 0);
615	regmap_update_bits(wm8731->regmap, WM8731_RINVOL, 0x100, 0);
616
617	/* Disable bypass path by default */
618	regmap_update_bits(wm8731->regmap, WM8731_APANA, 0x8, 0);
619
620	regcache_mark_dirty(wm8731->regmap);
621
622err:
623	return ret;
624}
625
626static const struct snd_soc_component_driver soc_component_dev_wm8731 = {
627	.set_bias_level		= wm8731_set_bias_level,
628	.controls		= wm8731_snd_controls,
629	.num_controls		= ARRAY_SIZE(wm8731_snd_controls),
630	.dapm_widgets		= wm8731_dapm_widgets,
631	.num_dapm_widgets	= ARRAY_SIZE(wm8731_dapm_widgets),
632	.dapm_routes		= wm8731_intercon,
633	.num_dapm_routes	= ARRAY_SIZE(wm8731_intercon),
634	.suspend_bias_off	= 1,
635	.idle_bias_on		= 1,
636	.use_pmdown_time	= 1,
637	.endianness		= 1,
638	.non_legacy_dai_naming	= 1,
639};
640
641static const struct of_device_id wm8731_of_match[] = {
642	{ .compatible = "wlf,wm8731", },
643	{ }
644};
645
646MODULE_DEVICE_TABLE(of, wm8731_of_match);
647
648static const struct regmap_config wm8731_regmap = {
649	.reg_bits = 7,
650	.val_bits = 9,
651
652	.max_register = WM8731_RESET,
653	.volatile_reg = wm8731_volatile,
654
655	.cache_type = REGCACHE_RBTREE,
656	.reg_defaults = wm8731_reg_defaults,
657	.num_reg_defaults = ARRAY_SIZE(wm8731_reg_defaults),
658};
659
660#if defined(CONFIG_SPI_MASTER)
661static int wm8731_spi_probe(struct spi_device *spi)
662{
663	struct wm8731_priv *wm8731;
664	int ret;
665
666	wm8731 = devm_kzalloc(&spi->dev, sizeof(*wm8731), GFP_KERNEL);
667	if (wm8731 == NULL)
668		return -ENOMEM;
669
670	wm8731->mclk = devm_clk_get(&spi->dev, "mclk");
671	if (IS_ERR(wm8731->mclk)) {
672		ret = PTR_ERR(wm8731->mclk);
673		if (ret == -ENOENT) {
674			wm8731->mclk = NULL;
675			dev_warn(&spi->dev, "Assuming static MCLK\n");
676		} else {
677			dev_err(&spi->dev, "Failed to get MCLK: %d\n",
678				ret);
679			return ret;
680		}
681	}
682
683	mutex_init(&wm8731->lock);
684
685	spi_set_drvdata(spi, wm8731);
686
687	ret = wm8731_request_supplies(&spi->dev, wm8731);
688	if (ret != 0)
689		return ret;
690
691	wm8731->regmap = devm_regmap_init_spi(spi, &wm8731_regmap);
692	if (IS_ERR(wm8731->regmap)) {
693		ret = PTR_ERR(wm8731->regmap);
694		dev_err(&spi->dev, "Failed to allocate register map: %d\n",
695			ret);
696		return ret;
697	}
698
699	ret = wm8731_hw_init(&spi->dev, wm8731);
700	if (ret != 0)
701		return ret;
702
703	ret = devm_snd_soc_register_component(&spi->dev,
704			&soc_component_dev_wm8731, &wm8731_dai, 1);
705	if (ret != 0) {
706		dev_err(&spi->dev, "Failed to register CODEC: %d\n", ret);
707		return ret;
708	}
709
710	return 0;
711}
712
713static int wm8731_spi_remove(struct spi_device *spi)
714{
715	return 0;
716}
717
718static struct spi_driver wm8731_spi_driver = {
719	.driver = {
720		.name	= "wm8731",
721		.of_match_table = wm8731_of_match,
722	},
723	.probe		= wm8731_spi_probe,
724	.remove		= wm8731_spi_remove,
725};
726#endif /* CONFIG_SPI_MASTER */
727
728#if IS_ENABLED(CONFIG_I2C)
729static int wm8731_i2c_probe(struct i2c_client *i2c,
730			    const struct i2c_device_id *id)
731{
732	struct wm8731_priv *wm8731;
733	int ret;
734
735	wm8731 = devm_kzalloc(&i2c->dev, sizeof(struct wm8731_priv),
736			      GFP_KERNEL);
737	if (wm8731 == NULL)
738		return -ENOMEM;
739
740	wm8731->mclk = devm_clk_get(&i2c->dev, "mclk");
741	if (IS_ERR(wm8731->mclk)) {
742		ret = PTR_ERR(wm8731->mclk);
743		if (ret == -ENOENT) {
744			wm8731->mclk = NULL;
745			dev_warn(&i2c->dev, "Assuming static MCLK\n");
746		} else {
747			dev_err(&i2c->dev, "Failed to get MCLK: %d\n",
748				ret);
749			return ret;
750		}
751	}
752
753	mutex_init(&wm8731->lock);
754
755	i2c_set_clientdata(i2c, wm8731);
756
757	ret = wm8731_request_supplies(&i2c->dev, wm8731);
758	if (ret != 0)
759		return ret;
760
761	wm8731->regmap = devm_regmap_init_i2c(i2c, &wm8731_regmap);
762	if (IS_ERR(wm8731->regmap)) {
763		ret = PTR_ERR(wm8731->regmap);
764		dev_err(&i2c->dev, "Failed to allocate register map: %d\n",
765			ret);
766		goto err_regulator_enable;
767	}
768
769	ret = wm8731_hw_init(&i2c->dev, wm8731);
770	if (ret != 0)
771		goto err_regulator_enable;
772
773	ret = devm_snd_soc_register_component(&i2c->dev,
774			&soc_component_dev_wm8731, &wm8731_dai, 1);
775	if (ret != 0) {
776		dev_err(&i2c->dev, "Failed to register CODEC: %d\n", ret);
777		goto err_regulator_enable;
778	}
779
780	return 0;
781
782err_regulator_enable:
783	/* Regulators will be enabled by bias management */
784	regulator_bulk_disable(ARRAY_SIZE(wm8731->supplies), wm8731->supplies);
785
786	return ret;
787}
788
789static int wm8731_i2c_remove(struct i2c_client *client)
790{
791	return 0;
792}
793
794static const struct i2c_device_id wm8731_i2c_id[] = {
795	{ "wm8731", 0 },
796	{ }
797};
798MODULE_DEVICE_TABLE(i2c, wm8731_i2c_id);
799
800static struct i2c_driver wm8731_i2c_driver = {
801	.driver = {
802		.name = "wm8731",
803		.of_match_table = wm8731_of_match,
804	},
805	.probe =    wm8731_i2c_probe,
806	.remove =   wm8731_i2c_remove,
807	.id_table = wm8731_i2c_id,
808};
809#endif
810
811static int __init wm8731_modinit(void)
812{
813	int ret = 0;
814#if IS_ENABLED(CONFIG_I2C)
815	ret = i2c_add_driver(&wm8731_i2c_driver);
816	if (ret != 0) {
817		printk(KERN_ERR "Failed to register WM8731 I2C driver: %d\n",
818		       ret);
819	}
820#endif
821#if defined(CONFIG_SPI_MASTER)
822	ret = spi_register_driver(&wm8731_spi_driver);
823	if (ret != 0) {
824		printk(KERN_ERR "Failed to register WM8731 SPI driver: %d\n",
825		       ret);
826	}
827#endif
828	return ret;
829}
830module_init(wm8731_modinit);
831
832static void __exit wm8731_exit(void)
833{
834#if IS_ENABLED(CONFIG_I2C)
835	i2c_del_driver(&wm8731_i2c_driver);
836#endif
837#if defined(CONFIG_SPI_MASTER)
838	spi_unregister_driver(&wm8731_spi_driver);
839#endif
840}
841module_exit(wm8731_exit);
842
843MODULE_DESCRIPTION("ASoC WM8731 driver");
844MODULE_AUTHOR("Richard Purdie");
845MODULE_LICENSE("GPL");
846