xref: /openbmc/linux/sound/soc/qcom/qdsp6/q6apm.c (revision c5a06fdc)
1 // SPDX-License-Identifier: GPL-2.0
2 // Copyright (c) 2020, Linaro Limited
3 
4 #include <dt-bindings/soc/qcom,gpr.h>
5 #include <linux/delay.h>
6 #include <linux/jiffies.h>
7 #include <linux/kernel.h>
8 #include <linux/module.h>
9 #include <linux/of.h>
10 #include <linux/of_platform.h>
11 #include <linux/sched.h>
12 #include <linux/slab.h>
13 #include <linux/soc/qcom/apr.h>
14 #include <linux/wait.h>
15 #include <sound/soc.h>
16 #include <sound/soc-dapm.h>
17 #include <sound/pcm.h>
18 #include "audioreach.h"
19 #include "q6apm.h"
20 
21 /* Graph Management */
22 struct apm_graph_mgmt_cmd {
23 	struct apm_module_param_data param_data;
24 	uint32_t num_sub_graphs;
25 	uint32_t sub_graph_id_list[];
26 } __packed;
27 
28 #define APM_GRAPH_MGMT_PSIZE(p, n) ALIGN(struct_size(p, sub_graph_id_list, n), 8)
29 
30 struct q6apm *g_apm;
31 
32 int q6apm_send_cmd_sync(struct q6apm *apm, struct gpr_pkt *pkt, uint32_t rsp_opcode)
33 {
34 	gpr_device_t *gdev = apm->gdev;
35 
36 	return audioreach_send_cmd_sync(&gdev->dev, gdev, &apm->result, &apm->lock,
37 					NULL, &apm->wait, pkt, rsp_opcode);
38 }
39 
40 static struct audioreach_graph *q6apm_get_audioreach_graph(struct q6apm *apm, uint32_t graph_id)
41 {
42 	struct audioreach_graph_info *info;
43 	struct audioreach_graph *graph;
44 	int id;
45 
46 	mutex_lock(&apm->lock);
47 	graph = idr_find(&apm->graph_idr, graph_id);
48 	mutex_unlock(&apm->lock);
49 
50 	if (graph) {
51 		kref_get(&graph->refcount);
52 		return graph;
53 	}
54 
55 	info = idr_find(&apm->graph_info_idr, graph_id);
56 
57 	if (!info)
58 		return ERR_PTR(-ENODEV);
59 
60 	graph = kzalloc(sizeof(*graph), GFP_KERNEL);
61 	if (!graph)
62 		return ERR_PTR(-ENOMEM);
63 
64 	graph->apm = apm;
65 	graph->info = info;
66 	graph->id = graph_id;
67 
68 	graph->graph = audioreach_alloc_graph_pkt(apm, info);
69 	if (IS_ERR(graph->graph)) {
70 		void *err = graph->graph;
71 
72 		kfree(graph);
73 		return ERR_CAST(err);
74 	}
75 
76 	mutex_lock(&apm->lock);
77 	id = idr_alloc(&apm->graph_idr, graph, graph_id, graph_id + 1, GFP_KERNEL);
78 	if (id < 0) {
79 		dev_err(apm->dev, "Unable to allocate graph id (%d)\n", graph_id);
80 		kfree(graph->graph);
81 		kfree(graph);
82 		mutex_unlock(&apm->lock);
83 		return ERR_PTR(id);
84 	}
85 	mutex_unlock(&apm->lock);
86 
87 	kref_init(&graph->refcount);
88 
89 	q6apm_send_cmd_sync(apm, graph->graph, 0);
90 
91 	return graph;
92 }
93 
94 static int audioreach_graph_mgmt_cmd(struct audioreach_graph *graph, uint32_t opcode)
95 {
96 	struct audioreach_graph_info *info = graph->info;
97 	int num_sub_graphs = info->num_sub_graphs;
98 	struct apm_module_param_data *param_data;
99 	struct apm_graph_mgmt_cmd *mgmt_cmd;
100 	struct audioreach_sub_graph *sg;
101 	struct q6apm *apm = graph->apm;
102 	int i = 0, rc, payload_size;
103 	struct gpr_pkt *pkt;
104 
105 	payload_size = APM_GRAPH_MGMT_PSIZE(mgmt_cmd, num_sub_graphs);
106 
107 	pkt = audioreach_alloc_apm_cmd_pkt(payload_size, opcode, 0);
108 	if (IS_ERR(pkt))
109 		return PTR_ERR(pkt);
110 
111 	mgmt_cmd = (void *)pkt + GPR_HDR_SIZE + APM_CMD_HDR_SIZE;
112 
113 	mgmt_cmd->num_sub_graphs = num_sub_graphs;
114 
115 	param_data = &mgmt_cmd->param_data;
116 	param_data->module_instance_id = APM_MODULE_INSTANCE_ID;
117 	param_data->param_id = APM_PARAM_ID_SUB_GRAPH_LIST;
118 	param_data->param_size = payload_size - APM_MODULE_PARAM_DATA_SIZE;
119 
120 	list_for_each_entry(sg, &info->sg_list, node)
121 		mgmt_cmd->sub_graph_id_list[i++] = sg->sub_graph_id;
122 
123 	rc = q6apm_send_cmd_sync(apm, pkt, 0);
124 
125 	kfree(pkt);
126 
127 	return rc;
128 }
129 
130 static void q6apm_put_audioreach_graph(struct kref *ref)
131 {
132 	struct audioreach_graph *graph;
133 	struct q6apm *apm;
134 
135 	graph = container_of(ref, struct audioreach_graph, refcount);
136 	apm = graph->apm;
137 
138 	audioreach_graph_mgmt_cmd(graph, APM_CMD_GRAPH_CLOSE);
139 
140 	mutex_lock(&apm->lock);
141 	graph = idr_remove(&apm->graph_idr, graph->id);
142 	mutex_unlock(&apm->lock);
143 
144 	kfree(graph->graph);
145 	kfree(graph);
146 }
147 
148 
149 static int q6apm_get_apm_state(struct q6apm *apm)
150 {
151 	struct gpr_pkt *pkt;
152 
153 	pkt = audioreach_alloc_apm_cmd_pkt(0, APM_CMD_GET_SPF_STATE, 0);
154 	if (IS_ERR(pkt))
155 		return PTR_ERR(pkt);
156 
157 	q6apm_send_cmd_sync(apm, pkt, APM_CMD_RSP_GET_SPF_STATE);
158 
159 	kfree(pkt);
160 
161 	return apm->state;
162 }
163 
164 bool q6apm_is_adsp_ready(void)
165 {
166 	if (g_apm)
167 		return q6apm_get_apm_state(g_apm);
168 
169 	return false;
170 }
171 EXPORT_SYMBOL_GPL(q6apm_is_adsp_ready);
172 
173 static struct audioreach_module *__q6apm_find_module_by_mid(struct q6apm *apm,
174 						    struct audioreach_graph_info *info,
175 						    uint32_t mid)
176 {
177 	struct audioreach_container *container;
178 	struct audioreach_sub_graph *sgs;
179 	struct audioreach_module *module;
180 
181 	list_for_each_entry(sgs, &info->sg_list, node) {
182 		list_for_each_entry(container, &sgs->container_list, node) {
183 			list_for_each_entry(module, &container->modules_list, node) {
184 				if (mid == module->module_id)
185 					return module;
186 			}
187 		}
188 	}
189 
190 	return NULL;
191 }
192 
193 int q6apm_graph_media_format_shmem(struct q6apm_graph *graph,
194 				   struct audioreach_module_config *cfg)
195 {
196 	struct audioreach_module *module;
197 
198 	if (cfg->direction == SNDRV_PCM_STREAM_CAPTURE)
199 		module = q6apm_find_module_by_mid(graph, MODULE_ID_RD_SHARED_MEM_EP);
200 	else
201 		module = q6apm_find_module_by_mid(graph, MODULE_ID_WR_SHARED_MEM_EP);
202 
203 	if (!module)
204 		return -ENODEV;
205 
206 	audioreach_set_media_format(graph, module, cfg);
207 
208 	return 0;
209 
210 }
211 EXPORT_SYMBOL_GPL(q6apm_graph_media_format_shmem);
212 
213 int q6apm_map_memory_regions(struct q6apm_graph *graph, unsigned int dir, phys_addr_t phys,
214 			     size_t period_sz, unsigned int periods)
215 {
216 	struct audioreach_graph_data *data;
217 	struct audio_buffer *buf;
218 	int cnt;
219 	int rc;
220 
221 	if (dir == SNDRV_PCM_STREAM_PLAYBACK)
222 		data = &graph->rx_data;
223 	else
224 		data = &graph->tx_data;
225 
226 	mutex_lock(&graph->lock);
227 
228 	if (data->buf) {
229 		mutex_unlock(&graph->lock);
230 		return 0;
231 	}
232 
233 	buf = kzalloc(((sizeof(struct audio_buffer)) * periods), GFP_KERNEL);
234 	if (!buf) {
235 		mutex_unlock(&graph->lock);
236 		return -ENOMEM;
237 	}
238 
239 	if (dir == SNDRV_PCM_STREAM_PLAYBACK)
240 		data = &graph->rx_data;
241 	else
242 		data = &graph->tx_data;
243 
244 	data->buf = buf;
245 
246 	buf[0].phys = phys;
247 	buf[0].size = period_sz;
248 
249 	for (cnt = 1; cnt < periods; cnt++) {
250 		if (period_sz > 0) {
251 			buf[cnt].phys = buf[0].phys + (cnt * period_sz);
252 			buf[cnt].size = period_sz;
253 		}
254 	}
255 	data->num_periods = periods;
256 
257 	mutex_unlock(&graph->lock);
258 
259 	rc = audioreach_map_memory_regions(graph, dir, period_sz, periods, 1);
260 	if (rc < 0) {
261 		dev_err(graph->dev, "Memory_map_regions failed\n");
262 		audioreach_graph_free_buf(graph);
263 	}
264 
265 	return rc;
266 }
267 EXPORT_SYMBOL_GPL(q6apm_map_memory_regions);
268 
269 int q6apm_unmap_memory_regions(struct q6apm_graph *graph, unsigned int dir)
270 {
271 	struct apm_cmd_shared_mem_unmap_regions *cmd;
272 	struct audioreach_graph_data *data;
273 	struct gpr_pkt *pkt;
274 	int rc;
275 
276 	if (dir == SNDRV_PCM_STREAM_PLAYBACK)
277 		data = &graph->rx_data;
278 	else
279 		data = &graph->tx_data;
280 
281 	if (!data->mem_map_handle)
282 		return 0;
283 
284 	pkt = audioreach_alloc_apm_pkt(sizeof(*cmd), APM_CMD_SHARED_MEM_UNMAP_REGIONS, dir,
285 				     graph->port->id);
286 	if (IS_ERR(pkt))
287 		return PTR_ERR(pkt);
288 
289 	cmd = (void *)pkt + GPR_HDR_SIZE;
290 	cmd->mem_map_handle = data->mem_map_handle;
291 
292 	rc = audioreach_graph_send_cmd_sync(graph, pkt, APM_CMD_SHARED_MEM_UNMAP_REGIONS);
293 	kfree(pkt);
294 
295 	audioreach_graph_free_buf(graph);
296 
297 	return rc;
298 }
299 EXPORT_SYMBOL_GPL(q6apm_unmap_memory_regions);
300 
301 int q6apm_remove_initial_silence(struct device *dev, struct q6apm_graph *graph, uint32_t samples)
302 {
303 	struct audioreach_module *module;
304 
305 	module = q6apm_find_module_by_mid(graph, MODULE_ID_PLACEHOLDER_DECODER);
306 	if (!module)
307 		return -ENODEV;
308 
309 	return audioreach_send_u32_param(graph, module, PARAM_ID_REMOVE_INITIAL_SILENCE, samples);
310 }
311 EXPORT_SYMBOL_GPL(q6apm_remove_initial_silence);
312 
313 int q6apm_remove_trailing_silence(struct device *dev, struct q6apm_graph *graph, uint32_t samples)
314 {
315 	struct audioreach_module *module;
316 
317 	module = q6apm_find_module_by_mid(graph, MODULE_ID_PLACEHOLDER_DECODER);
318 	if (!module)
319 		return -ENODEV;
320 
321 	return audioreach_send_u32_param(graph, module, PARAM_ID_REMOVE_TRAILING_SILENCE, samples);
322 }
323 EXPORT_SYMBOL_GPL(q6apm_remove_trailing_silence);
324 
325 int q6apm_enable_compress_module(struct device *dev, struct q6apm_graph *graph, bool en)
326 {
327 	struct audioreach_module *module;
328 
329 	module = q6apm_find_module_by_mid(graph, MODULE_ID_PLACEHOLDER_DECODER);
330 	if (!module)
331 		return -ENODEV;
332 
333 	return audioreach_send_u32_param(graph, module, PARAM_ID_MODULE_ENABLE, en);
334 }
335 EXPORT_SYMBOL_GPL(q6apm_enable_compress_module);
336 
337 int q6apm_set_real_module_id(struct device *dev, struct q6apm_graph *graph,
338 			     uint32_t codec_id)
339 {
340 	struct audioreach_module *module;
341 	uint32_t module_id;
342 
343 	module = q6apm_find_module_by_mid(graph, MODULE_ID_PLACEHOLDER_DECODER);
344 	if (!module)
345 		return -ENODEV;
346 
347 	switch (codec_id) {
348 	case SND_AUDIOCODEC_MP3:
349 		module_id = MODULE_ID_MP3_DECODE;
350 		break;
351 	case SND_AUDIOCODEC_AAC:
352 		module_id = MODULE_ID_AAC_DEC;
353 		break;
354 	case SND_AUDIOCODEC_FLAC:
355 		module_id = MODULE_ID_FLAC_DEC;
356 		break;
357 	default:
358 		return -EINVAL;
359 	}
360 
361 	return audioreach_send_u32_param(graph, module, PARAM_ID_REAL_MODULE_ID,
362 					 module_id);
363 }
364 EXPORT_SYMBOL_GPL(q6apm_set_real_module_id);
365 
366 int q6apm_graph_media_format_pcm(struct q6apm_graph *graph, struct audioreach_module_config *cfg)
367 {
368 	struct audioreach_graph_info *info = graph->info;
369 	struct audioreach_sub_graph *sgs;
370 	struct audioreach_container *container;
371 	struct audioreach_module *module;
372 
373 	list_for_each_entry(sgs, &info->sg_list, node) {
374 		list_for_each_entry(container, &sgs->container_list, node) {
375 			list_for_each_entry(module, &container->modules_list, node) {
376 				if ((module->module_id == MODULE_ID_WR_SHARED_MEM_EP) ||
377 					(module->module_id == MODULE_ID_RD_SHARED_MEM_EP))
378 					continue;
379 
380 				audioreach_set_media_format(graph, module, cfg);
381 			}
382 		}
383 	}
384 
385 	return 0;
386 
387 }
388 EXPORT_SYMBOL_GPL(q6apm_graph_media_format_pcm);
389 
390 static int q6apm_graph_get_tx_shmem_module_iid(struct q6apm_graph *graph)
391 {
392 	struct audioreach_module *module;
393 
394 	module = q6apm_find_module_by_mid(graph, MODULE_ID_RD_SHARED_MEM_EP);
395 	if (!module)
396 		return -ENODEV;
397 
398 	return module->instance_id;
399 
400 }
401 
402 int q6apm_graph_get_rx_shmem_module_iid(struct q6apm_graph *graph)
403 {
404 	struct audioreach_module *module;
405 
406 	module = q6apm_find_module_by_mid(graph, MODULE_ID_WR_SHARED_MEM_EP);
407 	if (!module)
408 		return -ENODEV;
409 
410 	return module->instance_id;
411 
412 }
413 EXPORT_SYMBOL_GPL(q6apm_graph_get_rx_shmem_module_iid);
414 
415 int q6apm_write_async(struct q6apm_graph *graph, uint32_t len, uint32_t msw_ts,
416 		      uint32_t lsw_ts, uint32_t wflags)
417 {
418 	struct apm_data_cmd_wr_sh_mem_ep_data_buffer_v2 *write_buffer;
419 	struct audio_buffer *ab;
420 	struct gpr_pkt *pkt;
421 	int rc, iid;
422 
423 	iid = q6apm_graph_get_rx_shmem_module_iid(graph);
424 	pkt = audioreach_alloc_pkt(sizeof(*write_buffer), DATA_CMD_WR_SH_MEM_EP_DATA_BUFFER_V2,
425 				   graph->rx_data.dsp_buf | (len << APM_WRITE_TOKEN_LEN_SHIFT),
426 				   graph->port->id, iid);
427 	if (IS_ERR(pkt))
428 		return PTR_ERR(pkt);
429 
430 	write_buffer = (void *)pkt + GPR_HDR_SIZE;
431 
432 	mutex_lock(&graph->lock);
433 	ab = &graph->rx_data.buf[graph->rx_data.dsp_buf];
434 
435 	write_buffer->buf_addr_lsw = lower_32_bits(ab->phys);
436 	write_buffer->buf_addr_msw = upper_32_bits(ab->phys);
437 	write_buffer->buf_size = len;
438 	write_buffer->timestamp_lsw = lsw_ts;
439 	write_buffer->timestamp_msw = msw_ts;
440 	write_buffer->mem_map_handle = graph->rx_data.mem_map_handle;
441 	write_buffer->flags = wflags;
442 
443 	graph->rx_data.dsp_buf++;
444 
445 	if (graph->rx_data.dsp_buf >= graph->rx_data.num_periods)
446 		graph->rx_data.dsp_buf = 0;
447 
448 	mutex_unlock(&graph->lock);
449 
450 	rc = gpr_send_port_pkt(graph->port, pkt);
451 
452 	kfree(pkt);
453 
454 	return rc;
455 }
456 EXPORT_SYMBOL_GPL(q6apm_write_async);
457 
458 int q6apm_read(struct q6apm_graph *graph)
459 {
460 	struct data_cmd_rd_sh_mem_ep_data_buffer_v2 *read_buffer;
461 	struct audioreach_graph_data *port;
462 	struct audio_buffer *ab;
463 	struct gpr_pkt *pkt;
464 	int rc, iid;
465 
466 	iid = q6apm_graph_get_tx_shmem_module_iid(graph);
467 	pkt = audioreach_alloc_pkt(sizeof(*read_buffer), DATA_CMD_RD_SH_MEM_EP_DATA_BUFFER_V2,
468 				   graph->tx_data.dsp_buf, graph->port->id, iid);
469 	if (IS_ERR(pkt))
470 		return PTR_ERR(pkt);
471 
472 	read_buffer = (void *)pkt + GPR_HDR_SIZE;
473 
474 	mutex_lock(&graph->lock);
475 	port = &graph->tx_data;
476 	ab = &port->buf[port->dsp_buf];
477 
478 	read_buffer->buf_addr_lsw = lower_32_bits(ab->phys);
479 	read_buffer->buf_addr_msw = upper_32_bits(ab->phys);
480 	read_buffer->mem_map_handle = port->mem_map_handle;
481 	read_buffer->buf_size = ab->size;
482 
483 	port->dsp_buf++;
484 
485 	if (port->dsp_buf >= port->num_periods)
486 		port->dsp_buf = 0;
487 
488 	mutex_unlock(&graph->lock);
489 
490 	rc = gpr_send_port_pkt(graph->port, pkt);
491 	kfree(pkt);
492 
493 	return rc;
494 }
495 EXPORT_SYMBOL_GPL(q6apm_read);
496 
497 static int graph_callback(struct gpr_resp_pkt *data, void *priv, int op)
498 {
499 	struct data_cmd_rsp_rd_sh_mem_ep_data_buffer_done_v2 *rd_done;
500 	struct data_cmd_rsp_wr_sh_mem_ep_data_buffer_done_v2 *done;
501 	struct apm_cmd_rsp_shared_mem_map_regions *rsp;
502 	struct gpr_ibasic_rsp_result_t *result;
503 	struct q6apm_graph *graph = priv;
504 	struct gpr_hdr *hdr = &data->hdr;
505 	struct device *dev = graph->dev;
506 	uint32_t client_event;
507 	phys_addr_t phys;
508 	int token;
509 
510 	result = data->payload;
511 
512 	switch (hdr->opcode) {
513 	case DATA_CMD_RSP_WR_SH_MEM_EP_DATA_BUFFER_DONE_V2:
514 		client_event = APM_CLIENT_EVENT_DATA_WRITE_DONE;
515 		mutex_lock(&graph->lock);
516 		token = hdr->token & APM_WRITE_TOKEN_MASK;
517 
518 		done = data->payload;
519 		phys = graph->rx_data.buf[token].phys;
520 		mutex_unlock(&graph->lock);
521 
522 		if (lower_32_bits(phys) == done->buf_addr_lsw &&
523 		    upper_32_bits(phys) == done->buf_addr_msw) {
524 			graph->result.opcode = hdr->opcode;
525 			graph->result.status = done->status;
526 			if (graph->cb)
527 				graph->cb(client_event, hdr->token, data->payload, graph->priv);
528 		} else {
529 			dev_err(dev, "WR BUFF Unexpected addr %08x-%08x\n", done->buf_addr_lsw,
530 				done->buf_addr_msw);
531 		}
532 
533 		break;
534 	case APM_CMD_RSP_SHARED_MEM_MAP_REGIONS:
535 		graph->result.opcode = hdr->opcode;
536 		graph->result.status = 0;
537 		rsp = data->payload;
538 
539 		if (hdr->token == SNDRV_PCM_STREAM_PLAYBACK)
540 			graph->rx_data.mem_map_handle = rsp->mem_map_handle;
541 		else
542 			graph->tx_data.mem_map_handle = rsp->mem_map_handle;
543 
544 		wake_up(&graph->cmd_wait);
545 		break;
546 	case DATA_CMD_RSP_RD_SH_MEM_EP_DATA_BUFFER_V2:
547 		client_event = APM_CLIENT_EVENT_DATA_READ_DONE;
548 		mutex_lock(&graph->lock);
549 		rd_done = data->payload;
550 		phys = graph->tx_data.buf[hdr->token].phys;
551 		mutex_unlock(&graph->lock);
552 
553 		if (upper_32_bits(phys) == rd_done->buf_addr_msw &&
554 		    lower_32_bits(phys) == rd_done->buf_addr_lsw) {
555 			graph->result.opcode = hdr->opcode;
556 			graph->result.status = rd_done->status;
557 			if (graph->cb)
558 				graph->cb(client_event, hdr->token, data->payload, graph->priv);
559 		} else {
560 			dev_err(dev, "RD BUFF Unexpected addr %08x-%08x\n", rd_done->buf_addr_lsw,
561 				rd_done->buf_addr_msw);
562 		}
563 		break;
564 	case DATA_CMD_WR_SH_MEM_EP_EOS_RENDERED:
565 		client_event = APM_CLIENT_EVENT_CMD_EOS_DONE;
566 		if (graph->cb)
567 			graph->cb(client_event, hdr->token, data->payload, graph->priv);
568 		break;
569 	case GPR_BASIC_RSP_RESULT:
570 		switch (result->opcode) {
571 		case APM_CMD_SHARED_MEM_UNMAP_REGIONS:
572 			graph->result.opcode = result->opcode;
573 			graph->result.status = 0;
574 			if (hdr->token == SNDRV_PCM_STREAM_PLAYBACK)
575 				graph->rx_data.mem_map_handle = 0;
576 			else
577 				graph->tx_data.mem_map_handle = 0;
578 
579 			wake_up(&graph->cmd_wait);
580 			break;
581 		case APM_CMD_SHARED_MEM_MAP_REGIONS:
582 		case DATA_CMD_WR_SH_MEM_EP_MEDIA_FORMAT:
583 		case APM_CMD_SET_CFG:
584 			graph->result.opcode = result->opcode;
585 			graph->result.status = result->status;
586 			if (result->status)
587 				dev_err(dev, "Error (%d) Processing 0x%08x cmd\n",
588 					result->status, result->opcode);
589 			wake_up(&graph->cmd_wait);
590 			break;
591 		default:
592 			break;
593 		}
594 		break;
595 	default:
596 		break;
597 	}
598 	return 0;
599 }
600 
601 struct q6apm_graph *q6apm_graph_open(struct device *dev, q6apm_cb cb,
602 				     void *priv, int graph_id)
603 {
604 	struct q6apm *apm = dev_get_drvdata(dev->parent);
605 	struct audioreach_graph *ar_graph;
606 	struct q6apm_graph *graph;
607 	int ret;
608 
609 	ar_graph = q6apm_get_audioreach_graph(apm, graph_id);
610 	if (IS_ERR(ar_graph)) {
611 		dev_err(dev, "No graph found with id %d\n", graph_id);
612 		return ERR_CAST(ar_graph);
613 	}
614 
615 	graph = kzalloc(sizeof(*graph), GFP_KERNEL);
616 	if (!graph) {
617 		ret = -ENOMEM;
618 		goto put_ar_graph;
619 	}
620 
621 	graph->apm = apm;
622 	graph->priv = priv;
623 	graph->cb = cb;
624 	graph->info = ar_graph->info;
625 	graph->ar_graph = ar_graph;
626 	graph->id = ar_graph->id;
627 	graph->dev = dev;
628 
629 	mutex_init(&graph->lock);
630 	init_waitqueue_head(&graph->cmd_wait);
631 
632 	graph->port = gpr_alloc_port(apm->gdev, dev, graph_callback, graph);
633 	if (IS_ERR(graph->port)) {
634 		ret = PTR_ERR(graph->port);
635 		goto free_graph;
636 	}
637 
638 	return graph;
639 
640 free_graph:
641 	kfree(graph);
642 put_ar_graph:
643 	kref_put(&ar_graph->refcount, q6apm_put_audioreach_graph);
644 	return ERR_PTR(ret);
645 }
646 EXPORT_SYMBOL_GPL(q6apm_graph_open);
647 
648 int q6apm_graph_close(struct q6apm_graph *graph)
649 {
650 	struct audioreach_graph *ar_graph = graph->ar_graph;
651 
652 	gpr_free_port(graph->port);
653 	kref_put(&ar_graph->refcount, q6apm_put_audioreach_graph);
654 	kfree(graph);
655 
656 	return 0;
657 }
658 EXPORT_SYMBOL_GPL(q6apm_graph_close);
659 
660 int q6apm_graph_prepare(struct q6apm_graph *graph)
661 {
662 	return audioreach_graph_mgmt_cmd(graph->ar_graph, APM_CMD_GRAPH_PREPARE);
663 }
664 EXPORT_SYMBOL_GPL(q6apm_graph_prepare);
665 
666 int q6apm_graph_start(struct q6apm_graph *graph)
667 {
668 	struct audioreach_graph *ar_graph = graph->ar_graph;
669 	int ret = 0;
670 
671 	if (ar_graph->start_count == 0)
672 		ret = audioreach_graph_mgmt_cmd(ar_graph, APM_CMD_GRAPH_START);
673 
674 	ar_graph->start_count++;
675 
676 	return ret;
677 }
678 EXPORT_SYMBOL_GPL(q6apm_graph_start);
679 
680 int q6apm_graph_stop(struct q6apm_graph *graph)
681 {
682 	struct audioreach_graph *ar_graph = graph->ar_graph;
683 
684 	if (--ar_graph->start_count > 0)
685 		return 0;
686 
687 	return audioreach_graph_mgmt_cmd(ar_graph, APM_CMD_GRAPH_STOP);
688 }
689 EXPORT_SYMBOL_GPL(q6apm_graph_stop);
690 
691 int q6apm_graph_flush(struct q6apm_graph *graph)
692 {
693 	return audioreach_graph_mgmt_cmd(graph->ar_graph, APM_CMD_GRAPH_FLUSH);
694 }
695 EXPORT_SYMBOL_GPL(q6apm_graph_flush);
696 
697 static int q6apm_audio_probe(struct snd_soc_component *component)
698 {
699 	return audioreach_tplg_init(component);
700 }
701 
702 static void q6apm_audio_remove(struct snd_soc_component *component)
703 {
704 	/* remove topology */
705 	snd_soc_tplg_component_remove(component);
706 }
707 
708 #define APM_AUDIO_DRV_NAME "q6apm-audio"
709 
710 static const struct snd_soc_component_driver q6apm_audio_component = {
711 	.name		= APM_AUDIO_DRV_NAME,
712 	.probe		= q6apm_audio_probe,
713 	.remove		= q6apm_audio_remove,
714 };
715 
716 static int apm_probe(gpr_device_t *gdev)
717 {
718 	struct device *dev = &gdev->dev;
719 	struct q6apm *apm;
720 	int ret;
721 
722 	apm = devm_kzalloc(dev, sizeof(*apm), GFP_KERNEL);
723 	if (!apm)
724 		return -ENOMEM;
725 
726 	dev_set_drvdata(dev, apm);
727 
728 	mutex_init(&apm->lock);
729 	apm->dev = dev;
730 	apm->gdev = gdev;
731 	init_waitqueue_head(&apm->wait);
732 
733 	INIT_LIST_HEAD(&apm->widget_list);
734 	idr_init(&apm->graph_idr);
735 	idr_init(&apm->graph_info_idr);
736 	idr_init(&apm->sub_graphs_idr);
737 	idr_init(&apm->containers_idr);
738 
739 	idr_init(&apm->modules_idr);
740 
741 	g_apm = apm;
742 
743 	q6apm_get_apm_state(apm);
744 
745 	ret = devm_snd_soc_register_component(dev, &q6apm_audio_component, NULL, 0);
746 	if (ret < 0) {
747 		dev_err(dev, "failed to register q6apm: %d\n", ret);
748 		return ret;
749 	}
750 
751 	return of_platform_populate(dev->of_node, NULL, NULL, dev);
752 }
753 
754 struct audioreach_module *q6apm_find_module_by_mid(struct q6apm_graph *graph, uint32_t mid)
755 {
756 	struct audioreach_graph_info *info = graph->info;
757 	struct q6apm *apm = graph->apm;
758 
759 	return __q6apm_find_module_by_mid(apm, info, mid);
760 
761 }
762 
763 static int apm_callback(struct gpr_resp_pkt *data, void *priv, int op)
764 {
765 	gpr_device_t *gdev = priv;
766 	struct q6apm *apm = dev_get_drvdata(&gdev->dev);
767 	struct device *dev = &gdev->dev;
768 	struct gpr_ibasic_rsp_result_t *result;
769 	struct gpr_hdr *hdr = &data->hdr;
770 
771 	result = data->payload;
772 
773 	switch (hdr->opcode) {
774 	case APM_CMD_RSP_GET_SPF_STATE:
775 		apm->result.opcode = hdr->opcode;
776 		apm->result.status = 0;
777 		/* First word of result it state */
778 		apm->state = result->opcode;
779 		wake_up(&apm->wait);
780 		break;
781 	case GPR_BASIC_RSP_RESULT:
782 		switch (result->opcode) {
783 		case APM_CMD_GRAPH_START:
784 		case APM_CMD_GRAPH_OPEN:
785 		case APM_CMD_GRAPH_PREPARE:
786 		case APM_CMD_GRAPH_CLOSE:
787 		case APM_CMD_GRAPH_FLUSH:
788 		case APM_CMD_GRAPH_STOP:
789 		case APM_CMD_SET_CFG:
790 			apm->result.opcode = result->opcode;
791 			apm->result.status = result->status;
792 			if (result->status)
793 				dev_err(dev, "Error (%d) Processing 0x%08x cmd\n", result->status,
794 					result->opcode);
795 			wake_up(&apm->wait);
796 			break;
797 		default:
798 			break;
799 		}
800 		break;
801 	default:
802 		break;
803 	}
804 
805 	return 0;
806 }
807 
808 #ifdef CONFIG_OF
809 static const struct of_device_id apm_device_id[]  = {
810 	{ .compatible = "qcom,q6apm" },
811 	{},
812 };
813 MODULE_DEVICE_TABLE(of, apm_device_id);
814 #endif
815 
816 static gpr_driver_t apm_driver = {
817 	.probe = apm_probe,
818 	.gpr_callback = apm_callback,
819 	.driver = {
820 		.name = "qcom-apm",
821 		.of_match_table = of_match_ptr(apm_device_id),
822 	},
823 };
824 
825 module_gpr_driver(apm_driver);
826 MODULE_DESCRIPTION("Audio Process Manager");
827 MODULE_LICENSE("GPL");
828