Merge branch 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/ebiederm...
[linux-2.6-microblaze.git] / sound / soc / codecs / wsa881x.c
1 // SPDX-License-Identifier: GPL-2.0
2 // Copyright (c) 2015-2017, The Linux Foundation.
3 // Copyright (c) 2019, Linaro Limited
4
5 #include <linux/bitops.h>
6 #include <linux/gpio.h>
7 #include <linux/gpio/consumer.h>
8 #include <linux/interrupt.h>
9 #include <linux/module.h>
10 #include <linux/of.h>
11 #include <linux/of_gpio.h>
12 #include <linux/regmap.h>
13 #include <linux/slab.h>
14 #include <linux/soundwire/sdw.h>
15 #include <linux/soundwire/sdw_registers.h>
16 #include <linux/soundwire/sdw_type.h>
17 #include <sound/soc.h>
18 #include <sound/tlv.h>
19
20 #define WSA881X_DIGITAL_BASE            0x3000
21 #define WSA881X_ANALOG_BASE             0x3100
22
23 /* Digital register address space */
24 #define WSA881X_CHIP_ID0                        (WSA881X_DIGITAL_BASE + 0x0000)
25 #define WSA881X_CHIP_ID1                        (WSA881X_DIGITAL_BASE + 0x0001)
26 #define WSA881X_CHIP_ID2                        (WSA881X_DIGITAL_BASE + 0x0002)
27 #define WSA881X_CHIP_ID3                        (WSA881X_DIGITAL_BASE + 0x0003)
28 #define WSA881X_BUS_ID                          (WSA881X_DIGITAL_BASE + 0x0004)
29 #define WSA881X_CDC_RST_CTL                     (WSA881X_DIGITAL_BASE + 0x0005)
30 #define WSA881X_CDC_TOP_CLK_CTL                 (WSA881X_DIGITAL_BASE + 0x0006)
31 #define WSA881X_CDC_ANA_CLK_CTL                 (WSA881X_DIGITAL_BASE + 0x0007)
32 #define WSA881X_CDC_DIG_CLK_CTL                 (WSA881X_DIGITAL_BASE + 0x0008)
33 #define WSA881X_CLOCK_CONFIG                    (WSA881X_DIGITAL_BASE + 0x0009)
34 #define WSA881X_ANA_CTL                         (WSA881X_DIGITAL_BASE + 0x000A)
35 #define WSA881X_SWR_RESET_EN                    (WSA881X_DIGITAL_BASE + 0x000B)
36 #define WSA881X_RESET_CTL                       (WSA881X_DIGITAL_BASE + 0x000C)
37 #define WSA881X_TADC_VALUE_CTL                  (WSA881X_DIGITAL_BASE + 0x000F)
38 #define WSA881X_TEMP_DETECT_CTL                 (WSA881X_DIGITAL_BASE + 0x0010)
39 #define WSA881X_TEMP_MSB                        (WSA881X_DIGITAL_BASE + 0x0011)
40 #define WSA881X_TEMP_LSB                        (WSA881X_DIGITAL_BASE + 0x0012)
41 #define WSA881X_TEMP_CONFIG0                    (WSA881X_DIGITAL_BASE + 0x0013)
42 #define WSA881X_TEMP_CONFIG1                    (WSA881X_DIGITAL_BASE + 0x0014)
43 #define WSA881X_CDC_CLIP_CTL                    (WSA881X_DIGITAL_BASE + 0x0015)
44 #define WSA881X_SDM_PDM9_LSB                    (WSA881X_DIGITAL_BASE + 0x0016)
45 #define WSA881X_SDM_PDM9_MSB                    (WSA881X_DIGITAL_BASE + 0x0017)
46 #define WSA881X_CDC_RX_CTL                      (WSA881X_DIGITAL_BASE + 0x0018)
47 #define WSA881X_DEM_BYPASS_DATA0                (WSA881X_DIGITAL_BASE + 0x0019)
48 #define WSA881X_DEM_BYPASS_DATA1                (WSA881X_DIGITAL_BASE + 0x001A)
49 #define WSA881X_DEM_BYPASS_DATA2                (WSA881X_DIGITAL_BASE + 0x001B)
50 #define WSA881X_DEM_BYPASS_DATA3                (WSA881X_DIGITAL_BASE + 0x001C)
51 #define WSA881X_OTP_CTRL0                       (WSA881X_DIGITAL_BASE + 0x001D)
52 #define WSA881X_OTP_CTRL1                       (WSA881X_DIGITAL_BASE + 0x001E)
53 #define WSA881X_HDRIVE_CTL_GROUP1               (WSA881X_DIGITAL_BASE + 0x001F)
54 #define WSA881X_INTR_MODE                       (WSA881X_DIGITAL_BASE + 0x0020)
55 #define WSA881X_INTR_MASK                       (WSA881X_DIGITAL_BASE + 0x0021)
56 #define WSA881X_INTR_STATUS                     (WSA881X_DIGITAL_BASE + 0x0022)
57 #define WSA881X_INTR_CLEAR                      (WSA881X_DIGITAL_BASE + 0x0023)
58 #define WSA881X_INTR_LEVEL                      (WSA881X_DIGITAL_BASE + 0x0024)
59 #define WSA881X_INTR_SET                        (WSA881X_DIGITAL_BASE + 0x0025)
60 #define WSA881X_INTR_TEST                       (WSA881X_DIGITAL_BASE + 0x0026)
61 #define WSA881X_PDM_TEST_MODE                   (WSA881X_DIGITAL_BASE + 0x0030)
62 #define WSA881X_ATE_TEST_MODE                   (WSA881X_DIGITAL_BASE + 0x0031)
63 #define WSA881X_PIN_CTL_MODE                    (WSA881X_DIGITAL_BASE + 0x0032)
64 #define WSA881X_PIN_CTL_OE                      (WSA881X_DIGITAL_BASE + 0x0033)
65 #define WSA881X_PIN_WDATA_IOPAD                 (WSA881X_DIGITAL_BASE + 0x0034)
66 #define WSA881X_PIN_STATUS                      (WSA881X_DIGITAL_BASE + 0x0035)
67 #define WSA881X_DIG_DEBUG_MODE                  (WSA881X_DIGITAL_BASE + 0x0037)
68 #define WSA881X_DIG_DEBUG_SEL                   (WSA881X_DIGITAL_BASE + 0x0038)
69 #define WSA881X_DIG_DEBUG_EN                    (WSA881X_DIGITAL_BASE + 0x0039)
70 #define WSA881X_SWR_HM_TEST1                    (WSA881X_DIGITAL_BASE + 0x003B)
71 #define WSA881X_SWR_HM_TEST2                    (WSA881X_DIGITAL_BASE + 0x003C)
72 #define WSA881X_TEMP_DETECT_DBG_CTL             (WSA881X_DIGITAL_BASE + 0x003D)
73 #define WSA881X_TEMP_DEBUG_MSB                  (WSA881X_DIGITAL_BASE + 0x003E)
74 #define WSA881X_TEMP_DEBUG_LSB                  (WSA881X_DIGITAL_BASE + 0x003F)
75 #define WSA881X_SAMPLE_EDGE_SEL                 (WSA881X_DIGITAL_BASE + 0x0044)
76 #define WSA881X_IOPAD_CTL                       (WSA881X_DIGITAL_BASE + 0x0045)
77 #define WSA881X_SPARE_0                         (WSA881X_DIGITAL_BASE + 0x0050)
78 #define WSA881X_SPARE_1                         (WSA881X_DIGITAL_BASE + 0x0051)
79 #define WSA881X_SPARE_2                         (WSA881X_DIGITAL_BASE + 0x0052)
80 #define WSA881X_OTP_REG_0                       (WSA881X_DIGITAL_BASE + 0x0080)
81 #define WSA881X_OTP_REG_1                       (WSA881X_DIGITAL_BASE + 0x0081)
82 #define WSA881X_OTP_REG_2                       (WSA881X_DIGITAL_BASE + 0x0082)
83 #define WSA881X_OTP_REG_3                       (WSA881X_DIGITAL_BASE + 0x0083)
84 #define WSA881X_OTP_REG_4                       (WSA881X_DIGITAL_BASE + 0x0084)
85 #define WSA881X_OTP_REG_5                       (WSA881X_DIGITAL_BASE + 0x0085)
86 #define WSA881X_OTP_REG_6                       (WSA881X_DIGITAL_BASE + 0x0086)
87 #define WSA881X_OTP_REG_7                       (WSA881X_DIGITAL_BASE + 0x0087)
88 #define WSA881X_OTP_REG_8                       (WSA881X_DIGITAL_BASE + 0x0088)
89 #define WSA881X_OTP_REG_9                       (WSA881X_DIGITAL_BASE + 0x0089)
90 #define WSA881X_OTP_REG_10                      (WSA881X_DIGITAL_BASE + 0x008A)
91 #define WSA881X_OTP_REG_11                      (WSA881X_DIGITAL_BASE + 0x008B)
92 #define WSA881X_OTP_REG_12                      (WSA881X_DIGITAL_BASE + 0x008C)
93 #define WSA881X_OTP_REG_13                      (WSA881X_DIGITAL_BASE + 0x008D)
94 #define WSA881X_OTP_REG_14                      (WSA881X_DIGITAL_BASE + 0x008E)
95 #define WSA881X_OTP_REG_15                      (WSA881X_DIGITAL_BASE + 0x008F)
96 #define WSA881X_OTP_REG_16                      (WSA881X_DIGITAL_BASE + 0x0090)
97 #define WSA881X_OTP_REG_17                      (WSA881X_DIGITAL_BASE + 0x0091)
98 #define WSA881X_OTP_REG_18                      (WSA881X_DIGITAL_BASE + 0x0092)
99 #define WSA881X_OTP_REG_19                      (WSA881X_DIGITAL_BASE + 0x0093)
100 #define WSA881X_OTP_REG_20                      (WSA881X_DIGITAL_BASE + 0x0094)
101 #define WSA881X_OTP_REG_21                      (WSA881X_DIGITAL_BASE + 0x0095)
102 #define WSA881X_OTP_REG_22                      (WSA881X_DIGITAL_BASE + 0x0096)
103 #define WSA881X_OTP_REG_23                      (WSA881X_DIGITAL_BASE + 0x0097)
104 #define WSA881X_OTP_REG_24                      (WSA881X_DIGITAL_BASE + 0x0098)
105 #define WSA881X_OTP_REG_25                      (WSA881X_DIGITAL_BASE + 0x0099)
106 #define WSA881X_OTP_REG_26                      (WSA881X_DIGITAL_BASE + 0x009A)
107 #define WSA881X_OTP_REG_27                      (WSA881X_DIGITAL_BASE + 0x009B)
108 #define WSA881X_OTP_REG_28                      (WSA881X_DIGITAL_BASE + 0x009C)
109 #define WSA881X_OTP_REG_29                      (WSA881X_DIGITAL_BASE + 0x009D)
110 #define WSA881X_OTP_REG_30                      (WSA881X_DIGITAL_BASE + 0x009E)
111 #define WSA881X_OTP_REG_31                      (WSA881X_DIGITAL_BASE + 0x009F)
112 #define WSA881X_OTP_REG_63                      (WSA881X_DIGITAL_BASE + 0x00BF)
113
114 /* Analog Register address space */
115 #define WSA881X_BIAS_REF_CTRL                   (WSA881X_ANALOG_BASE + 0x0000)
116 #define WSA881X_BIAS_TEST                       (WSA881X_ANALOG_BASE + 0x0001)
117 #define WSA881X_BIAS_BIAS                       (WSA881X_ANALOG_BASE + 0x0002)
118 #define WSA881X_TEMP_OP                         (WSA881X_ANALOG_BASE + 0x0003)
119 #define WSA881X_TEMP_IREF_CTRL                  (WSA881X_ANALOG_BASE + 0x0004)
120 #define WSA881X_TEMP_ISENS_CTRL                 (WSA881X_ANALOG_BASE + 0x0005)
121 #define WSA881X_TEMP_CLK_CTRL                   (WSA881X_ANALOG_BASE + 0x0006)
122 #define WSA881X_TEMP_TEST                       (WSA881X_ANALOG_BASE + 0x0007)
123 #define WSA881X_TEMP_BIAS                       (WSA881X_ANALOG_BASE + 0x0008)
124 #define WSA881X_TEMP_ADC_CTRL                   (WSA881X_ANALOG_BASE + 0x0009)
125 #define WSA881X_TEMP_DOUT_MSB                   (WSA881X_ANALOG_BASE + 0x000A)
126 #define WSA881X_TEMP_DOUT_LSB                   (WSA881X_ANALOG_BASE + 0x000B)
127 #define WSA881X_ADC_EN_MODU_V                   (WSA881X_ANALOG_BASE + 0x0010)
128 #define WSA881X_ADC_EN_MODU_I                   (WSA881X_ANALOG_BASE + 0x0011)
129 #define WSA881X_ADC_EN_DET_TEST_V               (WSA881X_ANALOG_BASE + 0x0012)
130 #define WSA881X_ADC_EN_DET_TEST_I               (WSA881X_ANALOG_BASE + 0x0013)
131 #define WSA881X_ADC_SEL_IBIAS                   (WSA881X_ANALOG_BASE + 0x0014)
132 #define WSA881X_ADC_EN_SEL_IBAIS                (WSA881X_ANALOG_BASE + 0x0015)
133 #define WSA881X_SPKR_DRV_EN                     (WSA881X_ANALOG_BASE + 0x001A)
134 #define WSA881X_SPKR_DRV_GAIN                   (WSA881X_ANALOG_BASE + 0x001B)
135 #define WSA881X_PA_GAIN_SEL_MASK                BIT(3)
136 #define WSA881X_PA_GAIN_SEL_REG                 BIT(3)
137 #define WSA881X_PA_GAIN_SEL_DRE                 0
138 #define WSA881X_SPKR_PAG_GAIN_MASK              GENMASK(7, 4)
139 #define WSA881X_SPKR_DAC_CTL                    (WSA881X_ANALOG_BASE + 0x001C)
140 #define WSA881X_SPKR_DRV_DBG                    (WSA881X_ANALOG_BASE + 0x001D)
141 #define WSA881X_SPKR_PWRSTG_DBG                 (WSA881X_ANALOG_BASE + 0x001E)
142 #define WSA881X_SPKR_OCP_CTL                    (WSA881X_ANALOG_BASE + 0x001F)
143 #define WSA881X_SPKR_OCP_MASK                   GENMASK(7, 6)
144 #define WSA881X_SPKR_OCP_EN                     BIT(7)
145 #define WSA881X_SPKR_OCP_HOLD                   BIT(6)
146 #define WSA881X_SPKR_CLIP_CTL                   (WSA881X_ANALOG_BASE + 0x0020)
147 #define WSA881X_SPKR_BBM_CTL                    (WSA881X_ANALOG_BASE + 0x0021)
148 #define WSA881X_SPKR_MISC_CTL1                  (WSA881X_ANALOG_BASE + 0x0022)
149 #define WSA881X_SPKR_MISC_CTL2                  (WSA881X_ANALOG_BASE + 0x0023)
150 #define WSA881X_SPKR_BIAS_INT                   (WSA881X_ANALOG_BASE + 0x0024)
151 #define WSA881X_SPKR_PA_INT                     (WSA881X_ANALOG_BASE + 0x0025)
152 #define WSA881X_SPKR_BIAS_CAL                   (WSA881X_ANALOG_BASE + 0x0026)
153 #define WSA881X_SPKR_BIAS_PSRR                  (WSA881X_ANALOG_BASE + 0x0027)
154 #define WSA881X_SPKR_STATUS1                    (WSA881X_ANALOG_BASE + 0x0028)
155 #define WSA881X_SPKR_STATUS2                    (WSA881X_ANALOG_BASE + 0x0029)
156 #define WSA881X_BOOST_EN_CTL                    (WSA881X_ANALOG_BASE + 0x002A)
157 #define WSA881X_BOOST_EN_MASK                   BIT(7)
158 #define WSA881X_BOOST_EN                        BIT(7)
159 #define WSA881X_BOOST_CURRENT_LIMIT             (WSA881X_ANALOG_BASE + 0x002B)
160 #define WSA881X_BOOST_PS_CTL                    (WSA881X_ANALOG_BASE + 0x002C)
161 #define WSA881X_BOOST_PRESET_OUT1               (WSA881X_ANALOG_BASE + 0x002D)
162 #define WSA881X_BOOST_PRESET_OUT2               (WSA881X_ANALOG_BASE + 0x002E)
163 #define WSA881X_BOOST_FORCE_OUT                 (WSA881X_ANALOG_BASE + 0x002F)
164 #define WSA881X_BOOST_LDO_PROG                  (WSA881X_ANALOG_BASE + 0x0030)
165 #define WSA881X_BOOST_SLOPE_COMP_ISENSE_FB      (WSA881X_ANALOG_BASE + 0x0031)
166 #define WSA881X_BOOST_RON_CTL                   (WSA881X_ANALOG_BASE + 0x0032)
167 #define WSA881X_BOOST_LOOP_STABILITY            (WSA881X_ANALOG_BASE + 0x0033)
168 #define WSA881X_BOOST_ZX_CTL                    (WSA881X_ANALOG_BASE + 0x0034)
169 #define WSA881X_BOOST_START_CTL                 (WSA881X_ANALOG_BASE + 0x0035)
170 #define WSA881X_BOOST_MISC1_CTL                 (WSA881X_ANALOG_BASE + 0x0036)
171 #define WSA881X_BOOST_MISC2_CTL                 (WSA881X_ANALOG_BASE + 0x0037)
172 #define WSA881X_BOOST_MISC3_CTL                 (WSA881X_ANALOG_BASE + 0x0038)
173 #define WSA881X_BOOST_ATEST_CTL                 (WSA881X_ANALOG_BASE + 0x0039)
174 #define WSA881X_SPKR_PROT_FE_GAIN               (WSA881X_ANALOG_BASE + 0x003A)
175 #define WSA881X_SPKR_PROT_FE_CM_LDO_SET         (WSA881X_ANALOG_BASE + 0x003B)
176 #define WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET1   (WSA881X_ANALOG_BASE + 0x003C)
177 #define WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET2   (WSA881X_ANALOG_BASE + 0x003D)
178 #define WSA881X_SPKR_PROT_ATEST1                (WSA881X_ANALOG_BASE + 0x003E)
179 #define WSA881X_SPKR_PROT_ATEST2                (WSA881X_ANALOG_BASE + 0x003F)
180 #define WSA881X_SPKR_PROT_FE_VSENSE_VCM         (WSA881X_ANALOG_BASE + 0x0040)
181 #define WSA881X_SPKR_PROT_FE_VSENSE_BIAS_SET1   (WSA881X_ANALOG_BASE + 0x0041)
182 #define WSA881X_BONGO_RESRV_REG1                (WSA881X_ANALOG_BASE + 0x0042)
183 #define WSA881X_BONGO_RESRV_REG2                (WSA881X_ANALOG_BASE + 0x0043)
184 #define WSA881X_SPKR_PROT_SAR                   (WSA881X_ANALOG_BASE + 0x0044)
185 #define WSA881X_SPKR_STATUS3                    (WSA881X_ANALOG_BASE + 0x0045)
186
187 #define SWRS_SCP_FRAME_CTRL_BANK(m)             (0x60 + 0x10 * (m))
188 #define SWRS_SCP_HOST_CLK_DIV2_CTL_BANK(m)      (0xE0 + 0x10 * (m))
189 #define SWR_SLV_MAX_REG_ADDR    0x390
190 #define SWR_SLV_START_REG_ADDR  0x40
191 #define SWR_SLV_MAX_BUF_LEN     20
192 #define BYTES_PER_LINE          12
193 #define SWR_SLV_RD_BUF_LEN      8
194 #define SWR_SLV_WR_BUF_LEN      32
195 #define SWR_SLV_MAX_DEVICES     2
196 #define WSA881X_MAX_SWR_PORTS   4
197 #define WSA881X_VERSION_ENTRY_SIZE 27
198 #define WSA881X_OCP_CTL_TIMER_SEC 2
199 #define WSA881X_OCP_CTL_TEMP_CELSIUS 25
200 #define WSA881X_OCP_CTL_POLL_TIMER_SEC 60
201
202 #define WSA881X_PA_GAIN_TLV(xname, reg, shift, max, invert, tlv_array) \
203 {       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, \
204         .access = SNDRV_CTL_ELEM_ACCESS_TLV_READ |\
205                  SNDRV_CTL_ELEM_ACCESS_READWRITE,\
206         .tlv.p = (tlv_array), \
207         .info = snd_soc_info_volsw, .get = snd_soc_get_volsw,\
208         .put = wsa881x_put_pa_gain, \
209         .private_value = SOC_SINGLE_VALUE(reg, shift, max, invert, 0) }
210
211 static struct reg_default wsa881x_defaults[] = {
212         { WSA881X_CHIP_ID0, 0x00 },
213         { WSA881X_CHIP_ID1, 0x00 },
214         { WSA881X_CHIP_ID2, 0x00 },
215         { WSA881X_CHIP_ID3, 0x02 },
216         { WSA881X_BUS_ID, 0x00 },
217         { WSA881X_CDC_RST_CTL, 0x00 },
218         { WSA881X_CDC_TOP_CLK_CTL, 0x03 },
219         { WSA881X_CDC_ANA_CLK_CTL, 0x00 },
220         { WSA881X_CDC_DIG_CLK_CTL, 0x00 },
221         { WSA881X_CLOCK_CONFIG, 0x00 },
222         { WSA881X_ANA_CTL, 0x08 },
223         { WSA881X_SWR_RESET_EN, 0x00 },
224         { WSA881X_TEMP_DETECT_CTL, 0x01 },
225         { WSA881X_TEMP_MSB, 0x00 },
226         { WSA881X_TEMP_LSB, 0x00 },
227         { WSA881X_TEMP_CONFIG0, 0x00 },
228         { WSA881X_TEMP_CONFIG1, 0x00 },
229         { WSA881X_CDC_CLIP_CTL, 0x03 },
230         { WSA881X_SDM_PDM9_LSB, 0x00 },
231         { WSA881X_SDM_PDM9_MSB, 0x00 },
232         { WSA881X_CDC_RX_CTL, 0x7E },
233         { WSA881X_DEM_BYPASS_DATA0, 0x00 },
234         { WSA881X_DEM_BYPASS_DATA1, 0x00 },
235         { WSA881X_DEM_BYPASS_DATA2, 0x00 },
236         { WSA881X_DEM_BYPASS_DATA3, 0x00 },
237         { WSA881X_OTP_CTRL0, 0x00 },
238         { WSA881X_OTP_CTRL1, 0x00 },
239         { WSA881X_HDRIVE_CTL_GROUP1, 0x00 },
240         { WSA881X_INTR_MODE, 0x00 },
241         { WSA881X_INTR_STATUS, 0x00 },
242         { WSA881X_INTR_CLEAR, 0x00 },
243         { WSA881X_INTR_LEVEL, 0x00 },
244         { WSA881X_INTR_SET, 0x00 },
245         { WSA881X_INTR_TEST, 0x00 },
246         { WSA881X_PDM_TEST_MODE, 0x00 },
247         { WSA881X_ATE_TEST_MODE, 0x00 },
248         { WSA881X_PIN_CTL_MODE, 0x00 },
249         { WSA881X_PIN_CTL_OE, 0x00 },
250         { WSA881X_PIN_WDATA_IOPAD, 0x00 },
251         { WSA881X_PIN_STATUS, 0x00 },
252         { WSA881X_DIG_DEBUG_MODE, 0x00 },
253         { WSA881X_DIG_DEBUG_SEL, 0x00 },
254         { WSA881X_DIG_DEBUG_EN, 0x00 },
255         { WSA881X_SWR_HM_TEST1, 0x08 },
256         { WSA881X_SWR_HM_TEST2, 0x00 },
257         { WSA881X_TEMP_DETECT_DBG_CTL, 0x00 },
258         { WSA881X_TEMP_DEBUG_MSB, 0x00 },
259         { WSA881X_TEMP_DEBUG_LSB, 0x00 },
260         { WSA881X_SAMPLE_EDGE_SEL, 0x0C },
261         { WSA881X_SPARE_0, 0x00 },
262         { WSA881X_SPARE_1, 0x00 },
263         { WSA881X_SPARE_2, 0x00 },
264         { WSA881X_OTP_REG_0, 0x01 },
265         { WSA881X_OTP_REG_1, 0xFF },
266         { WSA881X_OTP_REG_2, 0xC0 },
267         { WSA881X_OTP_REG_3, 0xFF },
268         { WSA881X_OTP_REG_4, 0xC0 },
269         { WSA881X_OTP_REG_5, 0xFF },
270         { WSA881X_OTP_REG_6, 0xFF },
271         { WSA881X_OTP_REG_7, 0xFF },
272         { WSA881X_OTP_REG_8, 0xFF },
273         { WSA881X_OTP_REG_9, 0xFF },
274         { WSA881X_OTP_REG_10, 0xFF },
275         { WSA881X_OTP_REG_11, 0xFF },
276         { WSA881X_OTP_REG_12, 0xFF },
277         { WSA881X_OTP_REG_13, 0xFF },
278         { WSA881X_OTP_REG_14, 0xFF },
279         { WSA881X_OTP_REG_15, 0xFF },
280         { WSA881X_OTP_REG_16, 0xFF },
281         { WSA881X_OTP_REG_17, 0xFF },
282         { WSA881X_OTP_REG_18, 0xFF },
283         { WSA881X_OTP_REG_19, 0xFF },
284         { WSA881X_OTP_REG_20, 0xFF },
285         { WSA881X_OTP_REG_21, 0xFF },
286         { WSA881X_OTP_REG_22, 0xFF },
287         { WSA881X_OTP_REG_23, 0xFF },
288         { WSA881X_OTP_REG_24, 0x03 },
289         { WSA881X_OTP_REG_25, 0x01 },
290         { WSA881X_OTP_REG_26, 0x03 },
291         { WSA881X_OTP_REG_27, 0x11 },
292         { WSA881X_OTP_REG_63, 0x40 },
293         /* WSA881x Analog registers */
294         { WSA881X_BIAS_REF_CTRL, 0x6C },
295         { WSA881X_BIAS_TEST, 0x16 },
296         { WSA881X_BIAS_BIAS, 0xF0 },
297         { WSA881X_TEMP_OP, 0x00 },
298         { WSA881X_TEMP_IREF_CTRL, 0x56 },
299         { WSA881X_TEMP_ISENS_CTRL, 0x47 },
300         { WSA881X_TEMP_CLK_CTRL, 0x87 },
301         { WSA881X_TEMP_TEST, 0x00 },
302         { WSA881X_TEMP_BIAS, 0x51 },
303         { WSA881X_TEMP_DOUT_MSB, 0x00 },
304         { WSA881X_TEMP_DOUT_LSB, 0x00 },
305         { WSA881X_ADC_EN_MODU_V, 0x00 },
306         { WSA881X_ADC_EN_MODU_I, 0x00 },
307         { WSA881X_ADC_EN_DET_TEST_V, 0x00 },
308         { WSA881X_ADC_EN_DET_TEST_I, 0x00 },
309         { WSA881X_ADC_EN_SEL_IBAIS, 0x10 },
310         { WSA881X_SPKR_DRV_EN, 0x74 },
311         { WSA881X_SPKR_DRV_DBG, 0x15 },
312         { WSA881X_SPKR_PWRSTG_DBG, 0x00 },
313         { WSA881X_SPKR_OCP_CTL, 0xD4 },
314         { WSA881X_SPKR_CLIP_CTL, 0x90 },
315         { WSA881X_SPKR_PA_INT, 0x54 },
316         { WSA881X_SPKR_BIAS_CAL, 0xAC },
317         { WSA881X_SPKR_STATUS1, 0x00 },
318         { WSA881X_SPKR_STATUS2, 0x00 },
319         { WSA881X_BOOST_EN_CTL, 0x18 },
320         { WSA881X_BOOST_CURRENT_LIMIT, 0x7A },
321         { WSA881X_BOOST_PRESET_OUT2, 0x70 },
322         { WSA881X_BOOST_FORCE_OUT, 0x0E },
323         { WSA881X_BOOST_LDO_PROG, 0x16 },
324         { WSA881X_BOOST_SLOPE_COMP_ISENSE_FB, 0x71 },
325         { WSA881X_BOOST_RON_CTL, 0x0F },
326         { WSA881X_BOOST_ZX_CTL, 0x34 },
327         { WSA881X_BOOST_START_CTL, 0x23 },
328         { WSA881X_BOOST_MISC1_CTL, 0x80 },
329         { WSA881X_BOOST_MISC2_CTL, 0x00 },
330         { WSA881X_BOOST_MISC3_CTL, 0x00 },
331         { WSA881X_BOOST_ATEST_CTL, 0x00 },
332         { WSA881X_SPKR_PROT_FE_GAIN, 0x46 },
333         { WSA881X_SPKR_PROT_FE_CM_LDO_SET, 0x3B },
334         { WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET1, 0x8D },
335         { WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET2, 0x8D },
336         { WSA881X_SPKR_PROT_ATEST1, 0x01 },
337         { WSA881X_SPKR_PROT_FE_VSENSE_VCM, 0x8D },
338         { WSA881X_SPKR_PROT_FE_VSENSE_BIAS_SET1, 0x4D },
339         { WSA881X_SPKR_PROT_SAR, 0x00 },
340         { WSA881X_SPKR_STATUS3, 0x00 },
341 };
342
343 static const struct reg_sequence wsa881x_pre_pmu_pa_2_0[] = {
344         { WSA881X_SPKR_DRV_GAIN, 0x41, 0 },
345         { WSA881X_SPKR_MISC_CTL1, 0x87, 0 },
346 };
347
348 static const struct reg_sequence wsa881x_vi_txfe_en_2_0[] = {
349         { WSA881X_SPKR_PROT_FE_VSENSE_VCM, 0x85, 0 },
350         { WSA881X_SPKR_PROT_ATEST2, 0x0A, 0 },
351         { WSA881X_SPKR_PROT_FE_GAIN, 0x47, 0 },
352 };
353
354 /* Default register reset values for WSA881x rev 2.0 */
355 static struct reg_sequence wsa881x_rev_2_0[] = {
356         { WSA881X_RESET_CTL, 0x00, 0x00 },
357         { WSA881X_TADC_VALUE_CTL, 0x01, 0x00 },
358         { WSA881X_INTR_MASK, 0x1B, 0x00 },
359         { WSA881X_IOPAD_CTL, 0x00, 0x00 },
360         { WSA881X_OTP_REG_28, 0x3F, 0x00 },
361         { WSA881X_OTP_REG_29, 0x3F, 0x00 },
362         { WSA881X_OTP_REG_30, 0x01, 0x00 },
363         { WSA881X_OTP_REG_31, 0x01, 0x00 },
364         { WSA881X_TEMP_ADC_CTRL, 0x03, 0x00 },
365         { WSA881X_ADC_SEL_IBIAS, 0x45, 0x00 },
366         { WSA881X_SPKR_DRV_GAIN, 0xC1, 0x00 },
367         { WSA881X_SPKR_DAC_CTL, 0x42, 0x00 },
368         { WSA881X_SPKR_BBM_CTL, 0x02, 0x00 },
369         { WSA881X_SPKR_MISC_CTL1, 0x40, 0x00 },
370         { WSA881X_SPKR_MISC_CTL2, 0x07, 0x00 },
371         { WSA881X_SPKR_BIAS_INT, 0x5F, 0x00 },
372         { WSA881X_SPKR_BIAS_PSRR, 0x44, 0x00 },
373         { WSA881X_BOOST_PS_CTL, 0xA0, 0x00 },
374         { WSA881X_BOOST_PRESET_OUT1, 0xB7, 0x00 },
375         { WSA881X_BOOST_LOOP_STABILITY, 0x8D, 0x00 },
376         { WSA881X_SPKR_PROT_ATEST2, 0x02, 0x00 },
377         { WSA881X_BONGO_RESRV_REG1, 0x5E, 0x00 },
378         { WSA881X_BONGO_RESRV_REG2, 0x07, 0x00 },
379 };
380
381 enum wsa_port_ids {
382         WSA881X_PORT_DAC,
383         WSA881X_PORT_COMP,
384         WSA881X_PORT_BOOST,
385         WSA881X_PORT_VISENSE,
386 };
387
388 /* 4 ports */
389 static struct sdw_dpn_prop wsa_sink_dpn_prop[WSA881X_MAX_SWR_PORTS] = {
390         {
391                 /* DAC */
392                 .num = 1,
393                 .type = SDW_DPN_SIMPLE,
394                 .min_ch = 1,
395                 .max_ch = 1,
396                 .simple_ch_prep_sm = true,
397                 .read_only_wordlength = true,
398         }, {
399                 /* COMP */
400                 .num = 2,
401                 .type = SDW_DPN_SIMPLE,
402                 .min_ch = 1,
403                 .max_ch = 1,
404                 .simple_ch_prep_sm = true,
405                 .read_only_wordlength = true,
406         }, {
407                 /* BOOST */
408                 .num = 3,
409                 .type = SDW_DPN_SIMPLE,
410                 .min_ch = 1,
411                 .max_ch = 1,
412                 .simple_ch_prep_sm = true,
413                 .read_only_wordlength = true,
414         }, {
415                 /* VISENSE */
416                 .num = 4,
417                 .type = SDW_DPN_SIMPLE,
418                 .min_ch = 1,
419                 .max_ch = 1,
420                 .simple_ch_prep_sm = true,
421                 .read_only_wordlength = true,
422         }
423 };
424
425 static struct sdw_port_config wsa881x_pconfig[WSA881X_MAX_SWR_PORTS] = {
426         {
427                 .num = 1,
428                 .ch_mask = 0x1,
429         }, {
430                 .num = 2,
431                 .ch_mask = 0xf,
432         }, {
433                 .num = 3,
434                 .ch_mask = 0x3,
435         }, {    /* IV feedback */
436                 .num = 4,
437                 .ch_mask = 0x3,
438         },
439 };
440
441 static bool wsa881x_readable_register(struct device *dev, unsigned int reg)
442 {
443         switch (reg) {
444         case WSA881X_CHIP_ID0:
445         case WSA881X_CHIP_ID1:
446         case WSA881X_CHIP_ID2:
447         case WSA881X_CHIP_ID3:
448         case WSA881X_BUS_ID:
449         case WSA881X_CDC_RST_CTL:
450         case WSA881X_CDC_TOP_CLK_CTL:
451         case WSA881X_CDC_ANA_CLK_CTL:
452         case WSA881X_CDC_DIG_CLK_CTL:
453         case WSA881X_CLOCK_CONFIG:
454         case WSA881X_ANA_CTL:
455         case WSA881X_SWR_RESET_EN:
456         case WSA881X_RESET_CTL:
457         case WSA881X_TADC_VALUE_CTL:
458         case WSA881X_TEMP_DETECT_CTL:
459         case WSA881X_TEMP_MSB:
460         case WSA881X_TEMP_LSB:
461         case WSA881X_TEMP_CONFIG0:
462         case WSA881X_TEMP_CONFIG1:
463         case WSA881X_CDC_CLIP_CTL:
464         case WSA881X_SDM_PDM9_LSB:
465         case WSA881X_SDM_PDM9_MSB:
466         case WSA881X_CDC_RX_CTL:
467         case WSA881X_DEM_BYPASS_DATA0:
468         case WSA881X_DEM_BYPASS_DATA1:
469         case WSA881X_DEM_BYPASS_DATA2:
470         case WSA881X_DEM_BYPASS_DATA3:
471         case WSA881X_OTP_CTRL0:
472         case WSA881X_OTP_CTRL1:
473         case WSA881X_HDRIVE_CTL_GROUP1:
474         case WSA881X_INTR_MODE:
475         case WSA881X_INTR_MASK:
476         case WSA881X_INTR_STATUS:
477         case WSA881X_INTR_CLEAR:
478         case WSA881X_INTR_LEVEL:
479         case WSA881X_INTR_SET:
480         case WSA881X_INTR_TEST:
481         case WSA881X_PDM_TEST_MODE:
482         case WSA881X_ATE_TEST_MODE:
483         case WSA881X_PIN_CTL_MODE:
484         case WSA881X_PIN_CTL_OE:
485         case WSA881X_PIN_WDATA_IOPAD:
486         case WSA881X_PIN_STATUS:
487         case WSA881X_DIG_DEBUG_MODE:
488         case WSA881X_DIG_DEBUG_SEL:
489         case WSA881X_DIG_DEBUG_EN:
490         case WSA881X_SWR_HM_TEST1:
491         case WSA881X_SWR_HM_TEST2:
492         case WSA881X_TEMP_DETECT_DBG_CTL:
493         case WSA881X_TEMP_DEBUG_MSB:
494         case WSA881X_TEMP_DEBUG_LSB:
495         case WSA881X_SAMPLE_EDGE_SEL:
496         case WSA881X_IOPAD_CTL:
497         case WSA881X_SPARE_0:
498         case WSA881X_SPARE_1:
499         case WSA881X_SPARE_2:
500         case WSA881X_OTP_REG_0:
501         case WSA881X_OTP_REG_1:
502         case WSA881X_OTP_REG_2:
503         case WSA881X_OTP_REG_3:
504         case WSA881X_OTP_REG_4:
505         case WSA881X_OTP_REG_5:
506         case WSA881X_OTP_REG_6:
507         case WSA881X_OTP_REG_7:
508         case WSA881X_OTP_REG_8:
509         case WSA881X_OTP_REG_9:
510         case WSA881X_OTP_REG_10:
511         case WSA881X_OTP_REG_11:
512         case WSA881X_OTP_REG_12:
513         case WSA881X_OTP_REG_13:
514         case WSA881X_OTP_REG_14:
515         case WSA881X_OTP_REG_15:
516         case WSA881X_OTP_REG_16:
517         case WSA881X_OTP_REG_17:
518         case WSA881X_OTP_REG_18:
519         case WSA881X_OTP_REG_19:
520         case WSA881X_OTP_REG_20:
521         case WSA881X_OTP_REG_21:
522         case WSA881X_OTP_REG_22:
523         case WSA881X_OTP_REG_23:
524         case WSA881X_OTP_REG_24:
525         case WSA881X_OTP_REG_25:
526         case WSA881X_OTP_REG_26:
527         case WSA881X_OTP_REG_27:
528         case WSA881X_OTP_REG_28:
529         case WSA881X_OTP_REG_29:
530         case WSA881X_OTP_REG_30:
531         case WSA881X_OTP_REG_31:
532         case WSA881X_OTP_REG_63:
533         case WSA881X_BIAS_REF_CTRL:
534         case WSA881X_BIAS_TEST:
535         case WSA881X_BIAS_BIAS:
536         case WSA881X_TEMP_OP:
537         case WSA881X_TEMP_IREF_CTRL:
538         case WSA881X_TEMP_ISENS_CTRL:
539         case WSA881X_TEMP_CLK_CTRL:
540         case WSA881X_TEMP_TEST:
541         case WSA881X_TEMP_BIAS:
542         case WSA881X_TEMP_ADC_CTRL:
543         case WSA881X_TEMP_DOUT_MSB:
544         case WSA881X_TEMP_DOUT_LSB:
545         case WSA881X_ADC_EN_MODU_V:
546         case WSA881X_ADC_EN_MODU_I:
547         case WSA881X_ADC_EN_DET_TEST_V:
548         case WSA881X_ADC_EN_DET_TEST_I:
549         case WSA881X_ADC_SEL_IBIAS:
550         case WSA881X_ADC_EN_SEL_IBAIS:
551         case WSA881X_SPKR_DRV_EN:
552         case WSA881X_SPKR_DRV_GAIN:
553         case WSA881X_SPKR_DAC_CTL:
554         case WSA881X_SPKR_DRV_DBG:
555         case WSA881X_SPKR_PWRSTG_DBG:
556         case WSA881X_SPKR_OCP_CTL:
557         case WSA881X_SPKR_CLIP_CTL:
558         case WSA881X_SPKR_BBM_CTL:
559         case WSA881X_SPKR_MISC_CTL1:
560         case WSA881X_SPKR_MISC_CTL2:
561         case WSA881X_SPKR_BIAS_INT:
562         case WSA881X_SPKR_PA_INT:
563         case WSA881X_SPKR_BIAS_CAL:
564         case WSA881X_SPKR_BIAS_PSRR:
565         case WSA881X_SPKR_STATUS1:
566         case WSA881X_SPKR_STATUS2:
567         case WSA881X_BOOST_EN_CTL:
568         case WSA881X_BOOST_CURRENT_LIMIT:
569         case WSA881X_BOOST_PS_CTL:
570         case WSA881X_BOOST_PRESET_OUT1:
571         case WSA881X_BOOST_PRESET_OUT2:
572         case WSA881X_BOOST_FORCE_OUT:
573         case WSA881X_BOOST_LDO_PROG:
574         case WSA881X_BOOST_SLOPE_COMP_ISENSE_FB:
575         case WSA881X_BOOST_RON_CTL:
576         case WSA881X_BOOST_LOOP_STABILITY:
577         case WSA881X_BOOST_ZX_CTL:
578         case WSA881X_BOOST_START_CTL:
579         case WSA881X_BOOST_MISC1_CTL:
580         case WSA881X_BOOST_MISC2_CTL:
581         case WSA881X_BOOST_MISC3_CTL:
582         case WSA881X_BOOST_ATEST_CTL:
583         case WSA881X_SPKR_PROT_FE_GAIN:
584         case WSA881X_SPKR_PROT_FE_CM_LDO_SET:
585         case WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET1:
586         case WSA881X_SPKR_PROT_FE_ISENSE_BIAS_SET2:
587         case WSA881X_SPKR_PROT_ATEST1:
588         case WSA881X_SPKR_PROT_ATEST2:
589         case WSA881X_SPKR_PROT_FE_VSENSE_VCM:
590         case WSA881X_SPKR_PROT_FE_VSENSE_BIAS_SET1:
591         case WSA881X_BONGO_RESRV_REG1:
592         case WSA881X_BONGO_RESRV_REG2:
593         case WSA881X_SPKR_PROT_SAR:
594         case WSA881X_SPKR_STATUS3:
595                 return true;
596         default:
597                 return false;
598         }
599 }
600
601 static bool wsa881x_volatile_register(struct device *dev, unsigned int reg)
602 {
603         switch (reg) {
604         case WSA881X_CHIP_ID0:
605         case WSA881X_CHIP_ID1:
606         case WSA881X_CHIP_ID2:
607         case WSA881X_CHIP_ID3:
608         case WSA881X_BUS_ID:
609         case WSA881X_TEMP_MSB:
610         case WSA881X_TEMP_LSB:
611         case WSA881X_SDM_PDM9_LSB:
612         case WSA881X_SDM_PDM9_MSB:
613         case WSA881X_OTP_CTRL1:
614         case WSA881X_INTR_STATUS:
615         case WSA881X_ATE_TEST_MODE:
616         case WSA881X_PIN_STATUS:
617         case WSA881X_SWR_HM_TEST2:
618         case WSA881X_SPKR_STATUS1:
619         case WSA881X_SPKR_STATUS2:
620         case WSA881X_SPKR_STATUS3:
621         case WSA881X_OTP_REG_0:
622         case WSA881X_OTP_REG_1:
623         case WSA881X_OTP_REG_2:
624         case WSA881X_OTP_REG_3:
625         case WSA881X_OTP_REG_4:
626         case WSA881X_OTP_REG_5:
627         case WSA881X_OTP_REG_31:
628         case WSA881X_TEMP_DOUT_MSB:
629         case WSA881X_TEMP_DOUT_LSB:
630         case WSA881X_TEMP_OP:
631         case WSA881X_SPKR_PROT_SAR:
632                 return true;
633         default:
634                 return false;
635         }
636 }
637
638 static struct regmap_config wsa881x_regmap_config = {
639         .reg_bits = 32,
640         .val_bits = 8,
641         .cache_type = REGCACHE_RBTREE,
642         .reg_defaults = wsa881x_defaults,
643         .num_reg_defaults = ARRAY_SIZE(wsa881x_defaults),
644         .volatile_reg = wsa881x_volatile_register,
645         .readable_reg = wsa881x_readable_register,
646         .reg_format_endian = REGMAP_ENDIAN_NATIVE,
647         .val_format_endian = REGMAP_ENDIAN_NATIVE,
648         .can_multi_write = true,
649 };
650
651 enum {
652         G_18DB = 0,
653         G_16P5DB,
654         G_15DB,
655         G_13P5DB,
656         G_12DB,
657         G_10P5DB,
658         G_9DB,
659         G_7P5DB,
660         G_6DB,
661         G_4P5DB,
662         G_3DB,
663         G_1P5DB,
664         G_0DB,
665 };
666
667 /*
668  * Private data Structure for wsa881x. All parameters related to
669  * WSA881X codec needs to be defined here.
670  */
671 struct wsa881x_priv {
672         struct regmap *regmap;
673         struct device *dev;
674         struct sdw_slave *slave;
675         struct sdw_stream_config sconfig;
676         struct sdw_stream_runtime *sruntime;
677         struct sdw_port_config port_config[WSA881X_MAX_SWR_PORTS];
678         struct gpio_desc *sd_n;
679         int version;
680         int active_ports;
681         bool port_prepared[WSA881X_MAX_SWR_PORTS];
682         bool port_enable[WSA881X_MAX_SWR_PORTS];
683 };
684
685 static void wsa881x_init(struct wsa881x_priv *wsa881x)
686 {
687         struct regmap *rm = wsa881x->regmap;
688         unsigned int val = 0;
689
690         regmap_read(rm, WSA881X_CHIP_ID1, &wsa881x->version);
691         regmap_register_patch(wsa881x->regmap, wsa881x_rev_2_0,
692                               ARRAY_SIZE(wsa881x_rev_2_0));
693
694         /* Enable software reset output from soundwire slave */
695         regmap_update_bits(rm, WSA881X_SWR_RESET_EN, 0x07, 0x07);
696
697         /* Bring out of analog reset */
698         regmap_update_bits(rm, WSA881X_CDC_RST_CTL, 0x02, 0x02);
699
700         /* Bring out of digital reset */
701         regmap_update_bits(rm, WSA881X_CDC_RST_CTL, 0x01, 0x01);
702         regmap_update_bits(rm, WSA881X_CLOCK_CONFIG, 0x10, 0x10);
703         regmap_update_bits(rm, WSA881X_SPKR_OCP_CTL, 0x02, 0x02);
704         regmap_update_bits(rm, WSA881X_SPKR_MISC_CTL1, 0xC0, 0x80);
705         regmap_update_bits(rm, WSA881X_SPKR_MISC_CTL1, 0x06, 0x06);
706         regmap_update_bits(rm, WSA881X_SPKR_BIAS_INT, 0xFF, 0x00);
707         regmap_update_bits(rm, WSA881X_SPKR_PA_INT, 0xF0, 0x40);
708         regmap_update_bits(rm, WSA881X_SPKR_PA_INT, 0x0E, 0x0E);
709         regmap_update_bits(rm, WSA881X_BOOST_LOOP_STABILITY, 0x03, 0x03);
710         regmap_update_bits(rm, WSA881X_BOOST_MISC2_CTL, 0xFF, 0x14);
711         regmap_update_bits(rm, WSA881X_BOOST_START_CTL, 0x80, 0x80);
712         regmap_update_bits(rm, WSA881X_BOOST_START_CTL, 0x03, 0x00);
713         regmap_update_bits(rm, WSA881X_BOOST_SLOPE_COMP_ISENSE_FB, 0x0C, 0x04);
714         regmap_update_bits(rm, WSA881X_BOOST_SLOPE_COMP_ISENSE_FB, 0x03, 0x00);
715
716         regmap_read(rm, WSA881X_OTP_REG_0, &val);
717         if (val)
718                 regmap_update_bits(rm, WSA881X_BOOST_PRESET_OUT1, 0xF0, 0x70);
719
720         regmap_update_bits(rm, WSA881X_BOOST_PRESET_OUT2, 0xF0, 0x30);
721         regmap_update_bits(rm, WSA881X_SPKR_DRV_EN, 0x08, 0x08);
722         regmap_update_bits(rm, WSA881X_BOOST_CURRENT_LIMIT, 0x0F, 0x08);
723         regmap_update_bits(rm, WSA881X_SPKR_OCP_CTL, 0x30, 0x30);
724         regmap_update_bits(rm, WSA881X_SPKR_OCP_CTL, 0x0C, 0x00);
725         regmap_update_bits(rm, WSA881X_OTP_REG_28, 0x3F, 0x3A);
726         regmap_update_bits(rm, WSA881X_BONGO_RESRV_REG1, 0xFF, 0xB2);
727         regmap_update_bits(rm, WSA881X_BONGO_RESRV_REG2, 0xFF, 0x05);
728 }
729
730 static int wsa881x_component_probe(struct snd_soc_component *comp)
731 {
732         struct wsa881x_priv *wsa881x = snd_soc_component_get_drvdata(comp);
733
734         snd_soc_component_init_regmap(comp, wsa881x->regmap);
735
736         return 0;
737 }
738
739 static int wsa881x_put_pa_gain(struct snd_kcontrol *kc,
740                                struct snd_ctl_elem_value *ucontrol)
741 {
742         struct snd_soc_component *comp = snd_soc_kcontrol_component(kc);
743         struct soc_mixer_control *mc =
744                         (struct soc_mixer_control *)kc->private_value;
745         int max = mc->max;
746         unsigned int mask = (1 << fls(max)) - 1;
747         int val, ret, min_gain, max_gain;
748
749         max_gain = (max - ucontrol->value.integer.value[0]) & mask;
750         /*
751          * Gain has to set incrementally in 4 steps
752          * as per HW sequence
753          */
754         if (max_gain > G_4P5DB)
755                 min_gain = G_0DB;
756         else
757                 min_gain = max_gain + 3;
758         /*
759          * 1ms delay is needed before change in gain
760          * as per HW requirement.
761          */
762         usleep_range(1000, 1010);
763
764         for (val = min_gain; max_gain <= val; val--) {
765                 ret = snd_soc_component_update_bits(comp,
766                               WSA881X_SPKR_DRV_GAIN,
767                               WSA881X_SPKR_PAG_GAIN_MASK,
768                               val << 4);
769                 if (ret < 0)
770                         dev_err(comp->dev, "Failed to change PA gain");
771
772                 usleep_range(1000, 1010);
773         }
774         return 0;
775 }
776
777 static int wsa881x_get_port(struct snd_kcontrol *kcontrol,
778                             struct snd_ctl_elem_value *ucontrol)
779 {
780         struct snd_soc_component *comp = snd_soc_kcontrol_component(kcontrol);
781         struct wsa881x_priv *data = snd_soc_component_get_drvdata(comp);
782         struct soc_mixer_control *mixer =
783                 (struct soc_mixer_control *)kcontrol->private_value;
784         int portidx = mixer->reg;
785
786         ucontrol->value.integer.value[0] = data->port_enable[portidx];
787
788
789         return 0;
790 }
791
792 static int wsa881x_boost_ctrl(struct snd_soc_component *comp, bool enable)
793 {
794         if (enable)
795                 snd_soc_component_update_bits(comp, WSA881X_BOOST_EN_CTL,
796                                               WSA881X_BOOST_EN_MASK,
797                                               WSA881X_BOOST_EN);
798         else
799                 snd_soc_component_update_bits(comp, WSA881X_BOOST_EN_CTL,
800                                               WSA881X_BOOST_EN_MASK, 0);
801         /*
802          * 1.5ms sleep is needed after boost enable/disable as per
803          * HW requirement
804          */
805         usleep_range(1500, 1510);
806         return 0;
807 }
808
809 static int wsa881x_set_port(struct snd_kcontrol *kcontrol,
810                             struct snd_ctl_elem_value *ucontrol)
811 {
812         struct snd_soc_component *comp = snd_soc_kcontrol_component(kcontrol);
813         struct wsa881x_priv *data = snd_soc_component_get_drvdata(comp);
814         struct soc_mixer_control *mixer =
815                 (struct soc_mixer_control *)kcontrol->private_value;
816         int portidx = mixer->reg;
817
818         if (ucontrol->value.integer.value[0])
819                 data->port_enable[portidx] = true;
820         else
821                 data->port_enable[portidx] = false;
822
823         if (portidx == WSA881X_PORT_BOOST) /* Boost Switch */
824                 wsa881x_boost_ctrl(comp, data->port_enable[portidx]);
825
826         return 0;
827 }
828
829 static const char * const smart_boost_lvl_text[] = {
830         "6.625 V", "6.750 V", "6.875 V", "7.000 V",
831         "7.125 V", "7.250 V", "7.375 V", "7.500 V",
832         "7.625 V", "7.750 V", "7.875 V", "8.000 V",
833         "8.125 V", "8.250 V", "8.375 V", "8.500 V"
834 };
835
836 static const struct soc_enum smart_boost_lvl_enum =
837         SOC_ENUM_SINGLE(WSA881X_BOOST_PRESET_OUT1, 0,
838                         ARRAY_SIZE(smart_boost_lvl_text),
839                         smart_boost_lvl_text);
840
841 static const DECLARE_TLV_DB_SCALE(pa_gain, 0, 150, 0);
842
843 static const struct snd_kcontrol_new wsa881x_snd_controls[] = {
844         SOC_ENUM("Smart Boost Level", smart_boost_lvl_enum),
845         WSA881X_PA_GAIN_TLV("PA Volume", WSA881X_SPKR_DRV_GAIN,
846                             4, 0xC, 1, pa_gain),
847         SOC_SINGLE_EXT("DAC Switch", WSA881X_PORT_DAC, 0, 1, 0,
848                        wsa881x_get_port, wsa881x_set_port),
849         SOC_SINGLE_EXT("COMP Switch", WSA881X_PORT_COMP, 0, 1, 0,
850                        wsa881x_get_port, wsa881x_set_port),
851         SOC_SINGLE_EXT("BOOST Switch", WSA881X_PORT_BOOST, 0, 1, 0,
852                        wsa881x_get_port, wsa881x_set_port),
853         SOC_SINGLE_EXT("VISENSE Switch", WSA881X_PORT_VISENSE, 0, 1, 0,
854                        wsa881x_get_port, wsa881x_set_port),
855 };
856
857 static const struct snd_soc_dapm_route wsa881x_audio_map[] = {
858         { "RDAC", NULL, "IN" },
859         { "RDAC", NULL, "DCLK" },
860         { "RDAC", NULL, "ACLK" },
861         { "RDAC", NULL, "Bandgap" },
862         { "SPKR PGA", NULL, "RDAC" },
863         { "SPKR", NULL, "SPKR PGA" },
864 };
865
866 static int wsa881x_visense_txfe_ctrl(struct snd_soc_component *comp,
867                                      bool enable)
868 {
869         struct wsa881x_priv *wsa881x = snd_soc_component_get_drvdata(comp);
870
871         if (enable) {
872                 regmap_multi_reg_write(wsa881x->regmap, wsa881x_vi_txfe_en_2_0,
873                                        ARRAY_SIZE(wsa881x_vi_txfe_en_2_0));
874         } else {
875                 snd_soc_component_update_bits(comp,
876                                               WSA881X_SPKR_PROT_FE_VSENSE_VCM,
877                                               0x08, 0x08);
878                 /*
879                  * 200us sleep is needed after visense txfe disable as per
880                  * HW requirement.
881                  */
882                 usleep_range(200, 210);
883                 snd_soc_component_update_bits(comp, WSA881X_SPKR_PROT_FE_GAIN,
884                                               0x01, 0x00);
885         }
886         return 0;
887 }
888
889 static int wsa881x_visense_adc_ctrl(struct snd_soc_component *comp,
890                                     bool enable)
891 {
892         snd_soc_component_update_bits(comp, WSA881X_ADC_EN_MODU_V, BIT(7),
893                                       (enable << 7));
894         snd_soc_component_update_bits(comp, WSA881X_ADC_EN_MODU_I, BIT(7),
895                                       (enable << 7));
896         return 0;
897 }
898
899 static int wsa881x_spkr_pa_event(struct snd_soc_dapm_widget *w,
900                                  struct snd_kcontrol *kcontrol, int event)
901 {
902         struct snd_soc_component *comp = snd_soc_dapm_to_component(w->dapm);
903         struct wsa881x_priv *wsa881x = snd_soc_component_get_drvdata(comp);
904
905         switch (event) {
906         case SND_SOC_DAPM_PRE_PMU:
907                 snd_soc_component_update_bits(comp, WSA881X_SPKR_OCP_CTL,
908                                               WSA881X_SPKR_OCP_MASK,
909                                               WSA881X_SPKR_OCP_EN);
910                 regmap_multi_reg_write(wsa881x->regmap, wsa881x_pre_pmu_pa_2_0,
911                                        ARRAY_SIZE(wsa881x_pre_pmu_pa_2_0));
912
913                 snd_soc_component_update_bits(comp, WSA881X_SPKR_DRV_GAIN,
914                                               WSA881X_PA_GAIN_SEL_MASK,
915                                               WSA881X_PA_GAIN_SEL_REG);
916                 break;
917         case SND_SOC_DAPM_POST_PMU:
918                 if (wsa881x->port_prepared[WSA881X_PORT_VISENSE]) {
919                         wsa881x_visense_txfe_ctrl(comp, true);
920                         snd_soc_component_update_bits(comp,
921                                                       WSA881X_ADC_EN_SEL_IBAIS,
922                                                       0x07, 0x01);
923                         wsa881x_visense_adc_ctrl(comp, true);
924                 }
925
926                 break;
927         case SND_SOC_DAPM_POST_PMD:
928                 if (wsa881x->port_prepared[WSA881X_PORT_VISENSE]) {
929                         wsa881x_visense_adc_ctrl(comp, false);
930                         wsa881x_visense_txfe_ctrl(comp, false);
931                 }
932
933                 snd_soc_component_update_bits(comp, WSA881X_SPKR_OCP_CTL,
934                                               WSA881X_SPKR_OCP_MASK,
935                                               WSA881X_SPKR_OCP_EN |
936                                               WSA881X_SPKR_OCP_HOLD);
937                 break;
938         }
939         return 0;
940 }
941
942 static const struct snd_soc_dapm_widget wsa881x_dapm_widgets[] = {
943         SND_SOC_DAPM_INPUT("IN"),
944         SND_SOC_DAPM_DAC_E("RDAC", NULL, WSA881X_SPKR_DAC_CTL, 7, 0,
945                            NULL,
946                            SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
947         SND_SOC_DAPM_PGA_E("SPKR PGA", SND_SOC_NOPM, 0, 0, NULL, 0,
948                            wsa881x_spkr_pa_event, SND_SOC_DAPM_PRE_PMU |
949                            SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
950         SND_SOC_DAPM_SUPPLY("DCLK", WSA881X_CDC_DIG_CLK_CTL, 0, 0, NULL,
951                             SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
952         SND_SOC_DAPM_SUPPLY("ACLK", WSA881X_CDC_ANA_CLK_CTL, 0, 0, NULL,
953                             SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
954         SND_SOC_DAPM_SUPPLY("Bandgap", WSA881X_TEMP_OP, 3, 0,
955                             NULL,
956                             SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
957         SND_SOC_DAPM_OUTPUT("SPKR"),
958 };
959
960 static int wsa881x_hw_params(struct snd_pcm_substream *substream,
961                              struct snd_pcm_hw_params *params,
962                              struct snd_soc_dai *dai)
963 {
964         struct wsa881x_priv *wsa881x = dev_get_drvdata(dai->dev);
965         int i;
966
967         wsa881x->active_ports = 0;
968         for (i = 0; i < WSA881X_MAX_SWR_PORTS; i++) {
969                 if (!wsa881x->port_enable[i])
970                         continue;
971
972                 wsa881x->port_config[wsa881x->active_ports] =
973                                                         wsa881x_pconfig[i];
974                 wsa881x->active_ports++;
975         }
976
977         return sdw_stream_add_slave(wsa881x->slave, &wsa881x->sconfig,
978                                     wsa881x->port_config, wsa881x->active_ports,
979                                     wsa881x->sruntime);
980 }
981
982 static int wsa881x_hw_free(struct snd_pcm_substream *substream,
983                            struct snd_soc_dai *dai)
984 {
985         struct wsa881x_priv *wsa881x = dev_get_drvdata(dai->dev);
986
987         sdw_stream_remove_slave(wsa881x->slave, wsa881x->sruntime);
988
989         return 0;
990 }
991
992 static int wsa881x_set_sdw_stream(struct snd_soc_dai *dai,
993                                   void *stream, int direction)
994 {
995         struct wsa881x_priv *wsa881x = dev_get_drvdata(dai->dev);
996
997         wsa881x->sruntime = stream;
998
999         return 0;
1000 }
1001
1002 static int wsa881x_digital_mute(struct snd_soc_dai *dai, int mute, int stream)
1003 {
1004         struct wsa881x_priv *wsa881x = dev_get_drvdata(dai->dev);
1005
1006         if (mute)
1007                 regmap_update_bits(wsa881x->regmap, WSA881X_SPKR_DRV_EN, 0x80,
1008                                    0x00);
1009         else
1010                 regmap_update_bits(wsa881x->regmap, WSA881X_SPKR_DRV_EN, 0x80,
1011                                    0x80);
1012
1013         return 0;
1014 }
1015
1016 static struct snd_soc_dai_ops wsa881x_dai_ops = {
1017         .hw_params = wsa881x_hw_params,
1018         .hw_free = wsa881x_hw_free,
1019         .mute_stream = wsa881x_digital_mute,
1020         .set_sdw_stream = wsa881x_set_sdw_stream,
1021 };
1022
1023 static struct snd_soc_dai_driver wsa881x_dais[] = {
1024         {
1025                 .name = "SPKR",
1026                 .id = 0,
1027                 .playback = {
1028                         .stream_name = "SPKR Playback",
1029                         .rate_max = 48000,
1030                         .rate_min = 48000,
1031                         .channels_min = 1,
1032                         .channels_max = 1,
1033                 },
1034                 .ops = &wsa881x_dai_ops,
1035         },
1036 };
1037
1038 static const struct snd_soc_component_driver wsa881x_component_drv = {
1039         .name = "WSA881x",
1040         .probe = wsa881x_component_probe,
1041         .controls = wsa881x_snd_controls,
1042         .num_controls = ARRAY_SIZE(wsa881x_snd_controls),
1043         .dapm_widgets = wsa881x_dapm_widgets,
1044         .num_dapm_widgets = ARRAY_SIZE(wsa881x_dapm_widgets),
1045         .dapm_routes = wsa881x_audio_map,
1046         .num_dapm_routes = ARRAY_SIZE(wsa881x_audio_map),
1047 };
1048
1049 static int wsa881x_update_status(struct sdw_slave *slave,
1050                                  enum sdw_slave_status status)
1051 {
1052         struct wsa881x_priv *wsa881x = dev_get_drvdata(&slave->dev);
1053
1054         if (status == SDW_SLAVE_ATTACHED && slave->dev_num > 0)
1055                 wsa881x_init(wsa881x);
1056
1057         return 0;
1058 }
1059
1060 static int wsa881x_port_prep(struct sdw_slave *slave,
1061                              struct sdw_prepare_ch *prepare_ch,
1062                              enum sdw_port_prep_ops state)
1063 {
1064         struct wsa881x_priv *wsa881x = dev_get_drvdata(&slave->dev);
1065
1066         if (state == SDW_OPS_PORT_POST_PREP)
1067                 wsa881x->port_prepared[prepare_ch->num - 1] = true;
1068         else
1069                 wsa881x->port_prepared[prepare_ch->num - 1] = false;
1070
1071         return 0;
1072 }
1073
1074 static int wsa881x_bus_config(struct sdw_slave *slave,
1075                               struct sdw_bus_params *params)
1076 {
1077         sdw_write(slave, SWRS_SCP_HOST_CLK_DIV2_CTL_BANK(params->next_bank),
1078                   0x01);
1079
1080         return 0;
1081 }
1082
1083 static struct sdw_slave_ops wsa881x_slave_ops = {
1084         .update_status = wsa881x_update_status,
1085         .bus_config = wsa881x_bus_config,
1086         .port_prep = wsa881x_port_prep,
1087 };
1088
1089 static int wsa881x_probe(struct sdw_slave *pdev,
1090                          const struct sdw_device_id *id)
1091 {
1092         struct wsa881x_priv *wsa881x;
1093
1094         wsa881x = devm_kzalloc(&pdev->dev, sizeof(*wsa881x), GFP_KERNEL);
1095         if (!wsa881x)
1096                 return -ENOMEM;
1097
1098         wsa881x->sd_n = devm_gpiod_get_optional(&pdev->dev, "powerdown",
1099                                                 GPIOD_FLAGS_BIT_NONEXCLUSIVE);
1100         if (IS_ERR(wsa881x->sd_n)) {
1101                 dev_err(&pdev->dev, "Shutdown Control GPIO not found\n");
1102                 return PTR_ERR(wsa881x->sd_n);
1103         }
1104
1105         dev_set_drvdata(&pdev->dev, wsa881x);
1106         wsa881x->slave = pdev;
1107         wsa881x->dev = &pdev->dev;
1108         wsa881x->sconfig.ch_count = 1;
1109         wsa881x->sconfig.bps = 1;
1110         wsa881x->sconfig.frame_rate = 48000;
1111         wsa881x->sconfig.direction = SDW_DATA_DIR_RX;
1112         wsa881x->sconfig.type = SDW_STREAM_PDM;
1113         pdev->prop.sink_ports = GENMASK(WSA881X_MAX_SWR_PORTS, 0);
1114         pdev->prop.sink_dpn_prop = wsa_sink_dpn_prop;
1115         gpiod_direction_output(wsa881x->sd_n, 1);
1116
1117         wsa881x->regmap = devm_regmap_init_sdw(pdev, &wsa881x_regmap_config);
1118         if (IS_ERR(wsa881x->regmap)) {
1119                 dev_err(&pdev->dev, "regmap_init failed\n");
1120                 return PTR_ERR(wsa881x->regmap);
1121         }
1122
1123         return devm_snd_soc_register_component(&pdev->dev,
1124                                                &wsa881x_component_drv,
1125                                                wsa881x_dais,
1126                                                ARRAY_SIZE(wsa881x_dais));
1127 }
1128
1129 static const struct sdw_device_id wsa881x_slave_id[] = {
1130         SDW_SLAVE_ENTRY(0x0217, 0x2010, 0),
1131         SDW_SLAVE_ENTRY(0x0217, 0x2110, 0),
1132         {},
1133 };
1134 MODULE_DEVICE_TABLE(sdw, wsa881x_slave_id);
1135
1136 static struct sdw_driver wsa881x_codec_driver = {
1137         .probe  = wsa881x_probe,
1138         .ops = &wsa881x_slave_ops,
1139         .id_table = wsa881x_slave_id,
1140         .driver = {
1141                 .name   = "wsa881x-codec",
1142         }
1143 };
1144 module_sdw_driver(wsa881x_codec_driver);
1145
1146 MODULE_DESCRIPTION("WSA881x codec driver");
1147 MODULE_LICENSE("GPL v2");