xref: /openbmc/linux/sound/soc/codecs/tas2552.c (revision 6a87e0f0)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * tas2552.c - ALSA SoC Texas Instruments TAS2552 Mono Audio Amplifier
4  *
5  * Copyright (C) 2014 Texas Instruments Incorporated -  https://www.ti.com
6  *
7  * Author: Dan Murphy <dmurphy@ti.com>
8  */
9 
10 #include <linux/module.h>
11 #include <linux/errno.h>
12 #include <linux/device.h>
13 #include <linux/i2c.h>
14 #include <linux/gpio.h>
15 #include <linux/of_gpio.h>
16 #include <linux/pm_runtime.h>
17 #include <linux/regmap.h>
18 #include <linux/slab.h>
19 
20 #include <linux/gpio/consumer.h>
21 #include <linux/regulator/consumer.h>
22 
23 #include <sound/pcm.h>
24 #include <sound/pcm_params.h>
25 #include <sound/soc.h>
26 #include <sound/soc-dapm.h>
27 #include <sound/tlv.h>
28 #include <sound/tas2552-plat.h>
29 #include <dt-bindings/sound/tas2552.h>
30 
31 #include "tas2552.h"
32 
33 static const struct reg_default tas2552_reg_defs[] = {
34 	{TAS2552_CFG_1, 0x22},
35 	{TAS2552_CFG_3, 0x80},
36 	{TAS2552_DOUT, 0x00},
37 	{TAS2552_OUTPUT_DATA, 0xc0},
38 	{TAS2552_PDM_CFG, 0x01},
39 	{TAS2552_PGA_GAIN, 0x00},
40 	{TAS2552_BOOST_APT_CTRL, 0x0f},
41 	{TAS2552_RESERVED_0D, 0xbe},
42 	{TAS2552_LIMIT_RATE_HYS, 0x08},
43 	{TAS2552_CFG_2, 0xef},
44 	{TAS2552_SER_CTRL_1, 0x00},
45 	{TAS2552_SER_CTRL_2, 0x00},
46 	{TAS2552_PLL_CTRL_1, 0x10},
47 	{TAS2552_PLL_CTRL_2, 0x00},
48 	{TAS2552_PLL_CTRL_3, 0x00},
49 	{TAS2552_BTIP, 0x8f},
50 	{TAS2552_BTS_CTRL, 0x80},
51 	{TAS2552_LIMIT_RELEASE, 0x04},
52 	{TAS2552_LIMIT_INT_COUNT, 0x00},
53 	{TAS2552_EDGE_RATE_CTRL, 0x40},
54 	{TAS2552_VBAT_DATA, 0x00},
55 };
56 
57 #define TAS2552_NUM_SUPPLIES	3
58 static const char *tas2552_supply_names[TAS2552_NUM_SUPPLIES] = {
59 	"vbat",		/* vbat voltage */
60 	"iovdd",	/* I/O Voltage */
61 	"avdd",		/* Analog DAC Voltage */
62 };
63 
64 struct tas2552_data {
65 	struct snd_soc_component *component;
66 	struct regmap *regmap;
67 	struct i2c_client *tas2552_client;
68 	struct regulator_bulk_data supplies[TAS2552_NUM_SUPPLIES];
69 	struct gpio_desc *enable_gpio;
70 	unsigned char regs[TAS2552_VBAT_DATA];
71 	unsigned int pll_clkin;
72 	int pll_clk_id;
73 	unsigned int pdm_clk;
74 	int pdm_clk_id;
75 
76 	unsigned int dai_fmt;
77 	unsigned int tdm_delay;
78 };
79 
80 static int tas2552_post_event(struct snd_soc_dapm_widget *w,
81 			      struct snd_kcontrol *kcontrol, int event)
82 {
83 	struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
84 
85 	switch (event) {
86 	case SND_SOC_DAPM_POST_PMU:
87 		snd_soc_component_write(component, TAS2552_RESERVED_0D, 0xc0);
88 		snd_soc_component_update_bits(component, TAS2552_LIMIT_RATE_HYS, (1 << 5),
89 				    (1 << 5));
90 		snd_soc_component_update_bits(component, TAS2552_CFG_2, 1, 0);
91 		snd_soc_component_update_bits(component, TAS2552_CFG_1, TAS2552_SWS, 0);
92 		break;
93 	case SND_SOC_DAPM_POST_PMD:
94 		snd_soc_component_update_bits(component, TAS2552_CFG_1, TAS2552_SWS,
95 				    TAS2552_SWS);
96 		snd_soc_component_update_bits(component, TAS2552_CFG_2, 1, 1);
97 		snd_soc_component_update_bits(component, TAS2552_LIMIT_RATE_HYS, (1 << 5), 0);
98 		snd_soc_component_write(component, TAS2552_RESERVED_0D, 0xbe);
99 		break;
100 	}
101 	return 0;
102 }
103 
104 /* Input mux controls */
105 static const char * const tas2552_input_texts[] = {
106 	"Digital", "Analog" };
107 static SOC_ENUM_SINGLE_DECL(tas2552_input_mux_enum, TAS2552_CFG_3, 7,
108 			    tas2552_input_texts);
109 
110 static const struct snd_kcontrol_new tas2552_input_mux_control =
111 	SOC_DAPM_ENUM("Route", tas2552_input_mux_enum);
112 
113 static const struct snd_soc_dapm_widget tas2552_dapm_widgets[] =
114 {
115 	SND_SOC_DAPM_INPUT("IN"),
116 
117 	/* MUX Controls */
118 	SND_SOC_DAPM_MUX("Input selection", SND_SOC_NOPM, 0, 0,
119 			 &tas2552_input_mux_control),
120 
121 	SND_SOC_DAPM_AIF_IN("DAC IN", "DAC Playback", 0, SND_SOC_NOPM, 0, 0),
122 	SND_SOC_DAPM_DAC("DAC", NULL, SND_SOC_NOPM, 0, 0),
123 	SND_SOC_DAPM_OUT_DRV("ClassD", TAS2552_CFG_2, 7, 0, NULL, 0),
124 	SND_SOC_DAPM_SUPPLY("PLL", TAS2552_CFG_2, 3, 0, NULL, 0),
125 	SND_SOC_DAPM_POST("Post Event", tas2552_post_event),
126 
127 	SND_SOC_DAPM_OUTPUT("OUT")
128 };
129 
130 static const struct snd_soc_dapm_route tas2552_audio_map[] = {
131 	{"DAC", NULL, "DAC IN"},
132 	{"Input selection", "Digital", "DAC"},
133 	{"Input selection", "Analog", "IN"},
134 	{"ClassD", NULL, "Input selection"},
135 	{"OUT", NULL, "ClassD"},
136 	{"ClassD", NULL, "PLL"},
137 };
138 
139 #ifdef CONFIG_PM
140 static void tas2552_sw_shutdown(struct tas2552_data *tas2552, int sw_shutdown)
141 {
142 	u8 cfg1_reg = 0;
143 
144 	if (!tas2552->component)
145 		return;
146 
147 	if (sw_shutdown)
148 		cfg1_reg = TAS2552_SWS;
149 
150 	snd_soc_component_update_bits(tas2552->component, TAS2552_CFG_1, TAS2552_SWS,
151 			    cfg1_reg);
152 }
153 #endif
154 
155 static int tas2552_setup_pll(struct snd_soc_component *component,
156 			     struct snd_pcm_hw_params *params)
157 {
158 	struct tas2552_data *tas2552 = dev_get_drvdata(component->dev);
159 	bool bypass_pll = false;
160 	unsigned int pll_clk = params_rate(params) * 512;
161 	unsigned int pll_clkin = tas2552->pll_clkin;
162 	u8 pll_enable;
163 
164 	if (!pll_clkin) {
165 		if (tas2552->pll_clk_id != TAS2552_PLL_CLKIN_BCLK)
166 			return -EINVAL;
167 
168 		pll_clkin = snd_soc_params_to_bclk(params);
169 		pll_clkin += tas2552->tdm_delay;
170 	}
171 
172 	pll_enable = snd_soc_component_read(component, TAS2552_CFG_2) & TAS2552_PLL_ENABLE;
173 	snd_soc_component_update_bits(component, TAS2552_CFG_2, TAS2552_PLL_ENABLE, 0);
174 
175 	if (pll_clkin == pll_clk)
176 		bypass_pll = true;
177 
178 	if (bypass_pll) {
179 		/* By pass the PLL configuration */
180 		snd_soc_component_update_bits(component, TAS2552_PLL_CTRL_2,
181 				    TAS2552_PLL_BYPASS, TAS2552_PLL_BYPASS);
182 	} else {
183 		/* Fill in the PLL control registers for J & D
184 		 * pll_clk = (.5 * pll_clkin * J.D) / 2^p
185 		 * Need to fill in J and D here based on incoming freq
186 		 */
187 		unsigned int d, q, t;
188 		u8 j;
189 		u8 pll_sel = (tas2552->pll_clk_id << 3) & TAS2552_PLL_SRC_MASK;
190 		u8 p = snd_soc_component_read(component, TAS2552_PLL_CTRL_1);
191 
192 		p = (p >> 7);
193 
194 recalc:
195 		t = (pll_clk * 2) << p;
196 		j = t / pll_clkin;
197 		d = t % pll_clkin;
198 		t = pll_clkin / 10000;
199 		q = d / (t + 1);
200 		d = q + ((9999 - pll_clkin % 10000) * (d / t - q)) / 10000;
201 
202 		if (d && (pll_clkin < 512000 || pll_clkin > 9200000)) {
203 			if (tas2552->pll_clk_id == TAS2552_PLL_CLKIN_BCLK) {
204 				pll_clkin = 1800000;
205 				pll_sel = (TAS2552_PLL_CLKIN_1_8_FIXED << 3) &
206 							TAS2552_PLL_SRC_MASK;
207 			} else {
208 				pll_clkin = snd_soc_params_to_bclk(params);
209 				pll_clkin += tas2552->tdm_delay;
210 				pll_sel = (TAS2552_PLL_CLKIN_BCLK << 3) &
211 							TAS2552_PLL_SRC_MASK;
212 			}
213 			goto recalc;
214 		}
215 
216 		snd_soc_component_update_bits(component, TAS2552_CFG_1, TAS2552_PLL_SRC_MASK,
217 				    pll_sel);
218 
219 		snd_soc_component_update_bits(component, TAS2552_PLL_CTRL_1,
220 				    TAS2552_PLL_J_MASK, j);
221 		/* Will clear the PLL_BYPASS bit */
222 		snd_soc_component_write(component, TAS2552_PLL_CTRL_2,
223 			      TAS2552_PLL_D_UPPER(d));
224 		snd_soc_component_write(component, TAS2552_PLL_CTRL_3,
225 			      TAS2552_PLL_D_LOWER(d));
226 	}
227 
228 	/* Restore PLL status */
229 	snd_soc_component_update_bits(component, TAS2552_CFG_2, TAS2552_PLL_ENABLE,
230 			    pll_enable);
231 
232 	return 0;
233 }
234 
235 static int tas2552_hw_params(struct snd_pcm_substream *substream,
236 			     struct snd_pcm_hw_params *params,
237 			     struct snd_soc_dai *dai)
238 {
239 	struct snd_soc_component *component = dai->component;
240 	struct tas2552_data *tas2552 = dev_get_drvdata(component->dev);
241 	int cpf;
242 	u8 ser_ctrl1_reg, wclk_rate;
243 
244 	switch (params_width(params)) {
245 	case 16:
246 		ser_ctrl1_reg = TAS2552_WORDLENGTH_16BIT;
247 		cpf = 32 + tas2552->tdm_delay;
248 		break;
249 	case 20:
250 		ser_ctrl1_reg = TAS2552_WORDLENGTH_20BIT;
251 		cpf = 64 + tas2552->tdm_delay;
252 		break;
253 	case 24:
254 		ser_ctrl1_reg = TAS2552_WORDLENGTH_24BIT;
255 		cpf = 64 + tas2552->tdm_delay;
256 		break;
257 	case 32:
258 		ser_ctrl1_reg = TAS2552_WORDLENGTH_32BIT;
259 		cpf = 64 + tas2552->tdm_delay;
260 		break;
261 	default:
262 		dev_err(component->dev, "Not supported sample size: %d\n",
263 			params_width(params));
264 		return -EINVAL;
265 	}
266 
267 	if (cpf <= 32)
268 		ser_ctrl1_reg |= TAS2552_CLKSPERFRAME_32;
269 	else if (cpf <= 64)
270 		ser_ctrl1_reg |= TAS2552_CLKSPERFRAME_64;
271 	else if (cpf <= 128)
272 		ser_ctrl1_reg |= TAS2552_CLKSPERFRAME_128;
273 	else
274 		ser_ctrl1_reg |= TAS2552_CLKSPERFRAME_256;
275 
276 	snd_soc_component_update_bits(component, TAS2552_SER_CTRL_1,
277 			    TAS2552_WORDLENGTH_MASK | TAS2552_CLKSPERFRAME_MASK,
278 			    ser_ctrl1_reg);
279 
280 	switch (params_rate(params)) {
281 	case 8000:
282 		wclk_rate = TAS2552_WCLK_FREQ_8KHZ;
283 		break;
284 	case 11025:
285 	case 12000:
286 		wclk_rate = TAS2552_WCLK_FREQ_11_12KHZ;
287 		break;
288 	case 16000:
289 		wclk_rate = TAS2552_WCLK_FREQ_16KHZ;
290 		break;
291 	case 22050:
292 	case 24000:
293 		wclk_rate = TAS2552_WCLK_FREQ_22_24KHZ;
294 		break;
295 	case 32000:
296 		wclk_rate = TAS2552_WCLK_FREQ_32KHZ;
297 		break;
298 	case 44100:
299 	case 48000:
300 		wclk_rate = TAS2552_WCLK_FREQ_44_48KHZ;
301 		break;
302 	case 88200:
303 	case 96000:
304 		wclk_rate = TAS2552_WCLK_FREQ_88_96KHZ;
305 		break;
306 	case 176400:
307 	case 192000:
308 		wclk_rate = TAS2552_WCLK_FREQ_176_192KHZ;
309 		break;
310 	default:
311 		dev_err(component->dev, "Not supported sample rate: %d\n",
312 			params_rate(params));
313 		return -EINVAL;
314 	}
315 
316 	snd_soc_component_update_bits(component, TAS2552_CFG_3, TAS2552_WCLK_FREQ_MASK,
317 			    wclk_rate);
318 
319 	return tas2552_setup_pll(component, params);
320 }
321 
322 #define TAS2552_DAI_FMT_MASK	(TAS2552_BCLKDIR | \
323 				 TAS2552_WCLKDIR | \
324 				 TAS2552_DATAFORMAT_MASK)
325 static int tas2552_prepare(struct snd_pcm_substream *substream,
326 			   struct snd_soc_dai *dai)
327 {
328 	struct snd_soc_component *component = dai->component;
329 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
330 	int delay = 0;
331 
332 	/* TDM slot selection only valid in DSP_A/_B mode */
333 	if (tas2552->dai_fmt == SND_SOC_DAIFMT_DSP_A)
334 		delay += (tas2552->tdm_delay + 1);
335 	else if (tas2552->dai_fmt == SND_SOC_DAIFMT_DSP_B)
336 		delay += tas2552->tdm_delay;
337 
338 	/* Configure data delay */
339 	snd_soc_component_write(component, TAS2552_SER_CTRL_2, delay);
340 
341 	return 0;
342 }
343 
344 static int tas2552_set_dai_fmt(struct snd_soc_dai *dai, unsigned int fmt)
345 {
346 	struct snd_soc_component *component = dai->component;
347 	struct tas2552_data *tas2552 = dev_get_drvdata(component->dev);
348 	u8 serial_format;
349 
350 	switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
351 	case SND_SOC_DAIFMT_CBC_CFC:
352 		serial_format = 0x00;
353 		break;
354 	case SND_SOC_DAIFMT_CBC_CFP:
355 		serial_format = TAS2552_WCLKDIR;
356 		break;
357 	case SND_SOC_DAIFMT_CBP_CFC:
358 		serial_format = TAS2552_BCLKDIR;
359 		break;
360 	case SND_SOC_DAIFMT_CBP_CFP:
361 		serial_format = (TAS2552_BCLKDIR | TAS2552_WCLKDIR);
362 		break;
363 	default:
364 		dev_vdbg(component->dev, "DAI Format master is not found\n");
365 		return -EINVAL;
366 	}
367 
368 	switch (fmt & (SND_SOC_DAIFMT_FORMAT_MASK |
369 		       SND_SOC_DAIFMT_INV_MASK)) {
370 	case (SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF):
371 		break;
372 	case (SND_SOC_DAIFMT_DSP_A | SND_SOC_DAIFMT_IB_NF):
373 	case (SND_SOC_DAIFMT_DSP_B | SND_SOC_DAIFMT_IB_NF):
374 		serial_format |= TAS2552_DATAFORMAT_DSP;
375 		break;
376 	case (SND_SOC_DAIFMT_RIGHT_J | SND_SOC_DAIFMT_NB_NF):
377 		serial_format |= TAS2552_DATAFORMAT_RIGHT_J;
378 		break;
379 	case (SND_SOC_DAIFMT_LEFT_J | SND_SOC_DAIFMT_NB_NF):
380 		serial_format |= TAS2552_DATAFORMAT_LEFT_J;
381 		break;
382 	default:
383 		dev_vdbg(component->dev, "DAI Format is not found\n");
384 		return -EINVAL;
385 	}
386 	tas2552->dai_fmt = fmt & SND_SOC_DAIFMT_FORMAT_MASK;
387 
388 	snd_soc_component_update_bits(component, TAS2552_SER_CTRL_1, TAS2552_DAI_FMT_MASK,
389 			    serial_format);
390 	return 0;
391 }
392 
393 static int tas2552_set_dai_sysclk(struct snd_soc_dai *dai, int clk_id,
394 				  unsigned int freq, int dir)
395 {
396 	struct snd_soc_component *component = dai->component;
397 	struct tas2552_data *tas2552 = dev_get_drvdata(component->dev);
398 	u8 reg, mask, val;
399 
400 	switch (clk_id) {
401 	case TAS2552_PLL_CLKIN_MCLK:
402 	case TAS2552_PLL_CLKIN_IVCLKIN:
403 		if (freq < 512000 || freq > 24576000) {
404 			/* out of range PLL_CLKIN, fall back to use BCLK */
405 			dev_warn(component->dev, "Out of range PLL_CLKIN: %u\n",
406 				 freq);
407 			clk_id = TAS2552_PLL_CLKIN_BCLK;
408 			freq = 0;
409 		}
410 		fallthrough;
411 	case TAS2552_PLL_CLKIN_BCLK:
412 	case TAS2552_PLL_CLKIN_1_8_FIXED:
413 		mask = TAS2552_PLL_SRC_MASK;
414 		val = (clk_id << 3) & mask; /* bit 4:5 in the register */
415 		reg = TAS2552_CFG_1;
416 		tas2552->pll_clk_id = clk_id;
417 		tas2552->pll_clkin = freq;
418 		break;
419 	case TAS2552_PDM_CLK_PLL:
420 	case TAS2552_PDM_CLK_IVCLKIN:
421 	case TAS2552_PDM_CLK_BCLK:
422 	case TAS2552_PDM_CLK_MCLK:
423 		mask = TAS2552_PDM_CLK_SEL_MASK;
424 		val = (clk_id >> 1) & mask; /* bit 0:1 in the register */
425 		reg = TAS2552_PDM_CFG;
426 		tas2552->pdm_clk_id = clk_id;
427 		tas2552->pdm_clk = freq;
428 		break;
429 	default:
430 		dev_err(component->dev, "Invalid clk id: %d\n", clk_id);
431 		return -EINVAL;
432 	}
433 
434 	snd_soc_component_update_bits(component, reg, mask, val);
435 
436 	return 0;
437 }
438 
439 static int tas2552_set_dai_tdm_slot(struct snd_soc_dai *dai,
440 				    unsigned int tx_mask, unsigned int rx_mask,
441 				    int slots, int slot_width)
442 {
443 	struct snd_soc_component *component = dai->component;
444 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
445 	unsigned int lsb;
446 
447 	if (unlikely(!tx_mask)) {
448 		dev_err(component->dev, "tx masks need to be non 0\n");
449 		return -EINVAL;
450 	}
451 
452 	/* TDM based on DSP mode requires slots to be adjacent */
453 	lsb = __ffs(tx_mask);
454 	if ((lsb + 1) != __fls(tx_mask)) {
455 		dev_err(component->dev, "Invalid mask, slots must be adjacent\n");
456 		return -EINVAL;
457 	}
458 
459 	tas2552->tdm_delay = lsb * slot_width;
460 
461 	/* DOUT in high-impedance on inactive bit clocks */
462 	snd_soc_component_update_bits(component, TAS2552_DOUT,
463 			    TAS2552_SDOUT_TRISTATE, TAS2552_SDOUT_TRISTATE);
464 
465 	return 0;
466 }
467 
468 static int tas2552_mute(struct snd_soc_dai *dai, int mute, int direction)
469 {
470 	u8 cfg1_reg = 0;
471 	struct snd_soc_component *component = dai->component;
472 
473 	if (mute)
474 		cfg1_reg |= TAS2552_MUTE;
475 
476 	snd_soc_component_update_bits(component, TAS2552_CFG_1, TAS2552_MUTE, cfg1_reg);
477 
478 	return 0;
479 }
480 
481 #ifdef CONFIG_PM
482 static int tas2552_runtime_suspend(struct device *dev)
483 {
484 	struct tas2552_data *tas2552 = dev_get_drvdata(dev);
485 
486 	tas2552_sw_shutdown(tas2552, 1);
487 
488 	regcache_cache_only(tas2552->regmap, true);
489 	regcache_mark_dirty(tas2552->regmap);
490 
491 	gpiod_set_value(tas2552->enable_gpio, 0);
492 
493 	return 0;
494 }
495 
496 static int tas2552_runtime_resume(struct device *dev)
497 {
498 	struct tas2552_data *tas2552 = dev_get_drvdata(dev);
499 
500 	gpiod_set_value(tas2552->enable_gpio, 1);
501 
502 	tas2552_sw_shutdown(tas2552, 0);
503 
504 	regcache_cache_only(tas2552->regmap, false);
505 	regcache_sync(tas2552->regmap);
506 
507 	return 0;
508 }
509 #endif
510 
511 static const struct dev_pm_ops tas2552_pm = {
512 	SET_RUNTIME_PM_OPS(tas2552_runtime_suspend, tas2552_runtime_resume,
513 			   NULL)
514 };
515 
516 static const struct snd_soc_dai_ops tas2552_speaker_dai_ops = {
517 	.hw_params	= tas2552_hw_params,
518 	.prepare	= tas2552_prepare,
519 	.set_sysclk	= tas2552_set_dai_sysclk,
520 	.set_fmt	= tas2552_set_dai_fmt,
521 	.set_tdm_slot	= tas2552_set_dai_tdm_slot,
522 	.mute_stream	= tas2552_mute,
523 	.no_capture_mute = 1,
524 };
525 
526 /* Formats supported by TAS2552 driver. */
527 #define TAS2552_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
528 			 SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S32_LE)
529 
530 /* TAS2552 dai structure. */
531 static struct snd_soc_dai_driver tas2552_dai[] = {
532 	{
533 		.name = "tas2552-amplifier",
534 		.playback = {
535 			.stream_name = "Playback",
536 			.channels_min = 2,
537 			.channels_max = 2,
538 			.rates = SNDRV_PCM_RATE_8000_192000,
539 			.formats = TAS2552_FORMATS,
540 		},
541 		.ops = &tas2552_speaker_dai_ops,
542 	},
543 };
544 
545 /*
546  * DAC digital volumes. From -7 to 24 dB in 1 dB steps
547  */
548 static DECLARE_TLV_DB_SCALE(dac_tlv, -700, 100, 0);
549 
550 static const char * const tas2552_din_source_select[] = {
551 	"Muted",
552 	"Left",
553 	"Right",
554 	"Left + Right average",
555 };
556 static SOC_ENUM_SINGLE_DECL(tas2552_din_source_enum,
557 			    TAS2552_CFG_3, 3,
558 			    tas2552_din_source_select);
559 
560 static const struct snd_kcontrol_new tas2552_snd_controls[] = {
561 	SOC_SINGLE_TLV("Speaker Driver Playback Volume",
562 			 TAS2552_PGA_GAIN, 0, 0x1f, 0, dac_tlv),
563 	SOC_ENUM("DIN source", tas2552_din_source_enum),
564 };
565 
566 static int tas2552_component_probe(struct snd_soc_component *component)
567 {
568 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
569 	int ret;
570 
571 	tas2552->component = component;
572 
573 	ret = regulator_bulk_enable(ARRAY_SIZE(tas2552->supplies),
574 				    tas2552->supplies);
575 
576 	if (ret != 0) {
577 		dev_err(component->dev, "Failed to enable supplies: %d\n",
578 			ret);
579 		return ret;
580 	}
581 
582 	gpiod_set_value(tas2552->enable_gpio, 1);
583 
584 	ret = pm_runtime_resume_and_get(component->dev);
585 	if (ret < 0) {
586 		dev_err(component->dev, "Enabling device failed: %d\n",
587 			ret);
588 		goto probe_fail;
589 	}
590 
591 	snd_soc_component_update_bits(component, TAS2552_CFG_1, TAS2552_MUTE, TAS2552_MUTE);
592 	snd_soc_component_write(component, TAS2552_CFG_3, TAS2552_I2S_OUT_SEL |
593 					    TAS2552_DIN_SRC_SEL_AVG_L_R);
594 	snd_soc_component_write(component, TAS2552_OUTPUT_DATA,
595 		      TAS2552_PDM_DATA_SEL_V_I |
596 		      TAS2552_R_DATA_OUT(TAS2552_DATA_OUT_V_DATA));
597 	snd_soc_component_write(component, TAS2552_BOOST_APT_CTRL, TAS2552_APT_DELAY_200 |
598 						     TAS2552_APT_THRESH_20_17);
599 
600 	snd_soc_component_write(component, TAS2552_CFG_2, TAS2552_BOOST_EN | TAS2552_APT_EN |
601 					    TAS2552_LIM_EN);
602 
603 	return 0;
604 
605 probe_fail:
606 	pm_runtime_put_noidle(component->dev);
607 	gpiod_set_value(tas2552->enable_gpio, 0);
608 
609 	regulator_bulk_disable(ARRAY_SIZE(tas2552->supplies),
610 					tas2552->supplies);
611 	return ret;
612 }
613 
614 static void tas2552_component_remove(struct snd_soc_component *component)
615 {
616 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
617 
618 	pm_runtime_put(component->dev);
619 
620 	gpiod_set_value(tas2552->enable_gpio, 0);
621 };
622 
623 #ifdef CONFIG_PM
624 static int tas2552_suspend(struct snd_soc_component *component)
625 {
626 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
627 	int ret;
628 
629 	ret = regulator_bulk_disable(ARRAY_SIZE(tas2552->supplies),
630 					tas2552->supplies);
631 
632 	if (ret != 0)
633 		dev_err(component->dev, "Failed to disable supplies: %d\n",
634 			ret);
635 	return ret;
636 }
637 
638 static int tas2552_resume(struct snd_soc_component *component)
639 {
640 	struct tas2552_data *tas2552 = snd_soc_component_get_drvdata(component);
641 	int ret;
642 
643 	ret = regulator_bulk_enable(ARRAY_SIZE(tas2552->supplies),
644 				    tas2552->supplies);
645 
646 	if (ret != 0) {
647 		dev_err(component->dev, "Failed to enable supplies: %d\n",
648 			ret);
649 	}
650 
651 	return ret;
652 }
653 #else
654 #define tas2552_suspend NULL
655 #define tas2552_resume NULL
656 #endif
657 
658 static const struct snd_soc_component_driver soc_component_dev_tas2552 = {
659 	.probe			= tas2552_component_probe,
660 	.remove			= tas2552_component_remove,
661 	.suspend		= tas2552_suspend,
662 	.resume			= tas2552_resume,
663 	.controls		= tas2552_snd_controls,
664 	.num_controls		= ARRAY_SIZE(tas2552_snd_controls),
665 	.dapm_widgets		= tas2552_dapm_widgets,
666 	.num_dapm_widgets	= ARRAY_SIZE(tas2552_dapm_widgets),
667 	.dapm_routes		= tas2552_audio_map,
668 	.num_dapm_routes	= ARRAY_SIZE(tas2552_audio_map),
669 	.idle_bias_on		= 1,
670 	.endianness		= 1,
671 };
672 
673 static const struct regmap_config tas2552_regmap_config = {
674 	.reg_bits = 8,
675 	.val_bits = 8,
676 
677 	.max_register = TAS2552_MAX_REG,
678 	.reg_defaults = tas2552_reg_defs,
679 	.num_reg_defaults = ARRAY_SIZE(tas2552_reg_defs),
680 	.cache_type = REGCACHE_RBTREE,
681 };
682 
683 static int tas2552_probe(struct i2c_client *client)
684 {
685 	struct device *dev;
686 	struct tas2552_data *data;
687 	int ret;
688 	int i;
689 
690 	dev = &client->dev;
691 	data = devm_kzalloc(&client->dev, sizeof(*data), GFP_KERNEL);
692 	if (data == NULL)
693 		return -ENOMEM;
694 
695 	data->enable_gpio = devm_gpiod_get_optional(dev, "enable",
696 						    GPIOD_OUT_LOW);
697 	if (IS_ERR(data->enable_gpio))
698 		return PTR_ERR(data->enable_gpio);
699 
700 	data->tas2552_client = client;
701 	data->regmap = devm_regmap_init_i2c(client, &tas2552_regmap_config);
702 	if (IS_ERR(data->regmap)) {
703 		ret = PTR_ERR(data->regmap);
704 		dev_err(&client->dev, "Failed to allocate register map: %d\n",
705 			ret);
706 		return ret;
707 	}
708 
709 	for (i = 0; i < ARRAY_SIZE(data->supplies); i++)
710 		data->supplies[i].supply = tas2552_supply_names[i];
711 
712 	ret = devm_regulator_bulk_get(dev, ARRAY_SIZE(data->supplies),
713 				      data->supplies);
714 	if (ret != 0) {
715 		dev_err(dev, "Failed to request supplies: %d\n", ret);
716 		return ret;
717 	}
718 
719 	pm_runtime_set_active(&client->dev);
720 	pm_runtime_set_autosuspend_delay(&client->dev, 1000);
721 	pm_runtime_use_autosuspend(&client->dev);
722 	pm_runtime_enable(&client->dev);
723 	pm_runtime_mark_last_busy(&client->dev);
724 	pm_runtime_put_sync_autosuspend(&client->dev);
725 
726 	dev_set_drvdata(&client->dev, data);
727 
728 	ret = devm_snd_soc_register_component(&client->dev,
729 				      &soc_component_dev_tas2552,
730 				      tas2552_dai, ARRAY_SIZE(tas2552_dai));
731 	if (ret < 0) {
732 		dev_err(&client->dev, "Failed to register component: %d\n", ret);
733 		pm_runtime_get_noresume(&client->dev);
734 	}
735 
736 	return ret;
737 }
738 
739 static void tas2552_i2c_remove(struct i2c_client *client)
740 {
741 	pm_runtime_disable(&client->dev);
742 }
743 
744 static const struct i2c_device_id tas2552_id[] = {
745 	{ "tas2552", 0 },
746 	{ }
747 };
748 MODULE_DEVICE_TABLE(i2c, tas2552_id);
749 
750 #if IS_ENABLED(CONFIG_OF)
751 static const struct of_device_id tas2552_of_match[] = {
752 	{ .compatible = "ti,tas2552", },
753 	{},
754 };
755 MODULE_DEVICE_TABLE(of, tas2552_of_match);
756 #endif
757 
758 static struct i2c_driver tas2552_i2c_driver = {
759 	.driver = {
760 		.name = "tas2552",
761 		.of_match_table = of_match_ptr(tas2552_of_match),
762 		.pm = &tas2552_pm,
763 	},
764 	.probe = tas2552_probe,
765 	.remove = tas2552_i2c_remove,
766 	.id_table = tas2552_id,
767 };
768 
769 module_i2c_driver(tas2552_i2c_driver);
770 
771 MODULE_AUTHOR("Dan Muprhy <dmurphy@ti.com>");
772 MODULE_DESCRIPTION("TAS2552 Audio amplifier driver");
773 MODULE_LICENSE("GPL");
774