kd: add ams support

Add support for AMS event notifications to KD module

Signed-off-by: Ievgen Ganakov <ievgen.ganakov@intel.com>

Author: iganakov

src/samples/audio/detect_test.c

@@ -17,7 +17,6 @@
 #include <rtos/alloc.h>
 #include <rtos/init.h>
 #include <sof/lib/memory.h>
-#include <sof/lib/notifier.h>
 #include <rtos/wait.h>
 #include <sof/lib/uuid.h>
 #include <sof/list.h>
@@ -41,6 +40,11 @@
 #include <stdint.h>
 #include <stdlib.h>
 #include <sof/samples/audio/kwd_nn_detect_test.h>
+#if CONFIG_AMS
+#include <sof/lib/ams.h>
+#else
+#include <sof/lib/notifier.h>
+#endif
 
 #define ACTIVATION_DEFAULT_SHIFT 3
 #define ACTIVATION_DEFAULT_COEF 0.05
@@ -85,19 +89,25 @@ struct comp_data {
 	uint32_t keyphrase_samples; /**< keyphrase length in samples */
 	uint32_t drain_req; /** defines draining size in bytes. */
 	uint16_t sample_valid_bytes;
-	struct kpb_event_data event_data;
 	struct kpb_client client_data;
 
 #if CONFIG_KWD_NN_SAMPLE_KEYPHRASE
 	int16_t *input;
 	size_t input_size;
 #endif
-
 	struct ipc_msg *msg;	/**< host notification */
 
 	void (*detect_func)(struct comp_dev *dev,
 			    const struct audio_stream __sparse_cache *source, uint32_t frames);
 	struct sof_ipc_comp_event event;
+
+#if CONFIG_AMS
+	/* AMS related data */
+	uint32_t ams_uuid_id;
+	struct ams_message_payload ams_payload;
+#else
+	struct kpb_event_data event_data;
+#endif /* CONFIG_AMS */
 };
 
 static inline bool detector_is_sample_width_supported(enum sof_ipc_frame sf)
@@ -140,6 +150,25 @@ static void notify_host(const struct comp_dev *dev)
 #endif /* CONFIG_IPC_MAJOR_4 */
 }
 
+#if CONFIG_AMS
+static int ams_notify_kpb(const struct comp_dev *dev)
+{
+	struct comp_data *cd = comp_get_drvdata(dev);
+
+	cd->client_data.r_ptr = NULL;
+	cd->client_data.sink = NULL;
+	cd->client_data.id = 0; /**< TODO: acquire proper id from kpb */
+	/* time in milliseconds */
+	cd->client_data.drain_req = (cd->drain_req != 0) ?
+				     cd->drain_req :
+				     cd->config.drain_req;
+
+	cd->ams_payload.message_length = sizeof(struct kpb_client);
+	cd->ams_payload.message = (uint8_t *)&cd->client_data;
+
+	return ams_send(&cd->ams_payload);
+}
+#else
 static void notify_kpb(const struct comp_dev *dev)
 {
 	struct comp_data *cd = comp_get_drvdata(dev);
@@ -160,11 +189,16 @@ static void notify_kpb(const struct comp_dev *dev)
 		       NOTIFIER_TARGET_CORE_ALL_MASK, &cd->event_data,
 		       sizeof(cd->event_data));
 }
+#endif /* CONFIG_AMS */
 
 void detect_test_notify(const struct comp_dev *dev)
 {
 	notify_host(dev);
+#if CONFIG_AMS
+	ams_notify_kpb(dev);
+#else
 	notify_kpb(dev);
+#endif
 }
 
 static void default_detect_test(struct comp_dev *dev,
@@ -719,6 +753,15 @@ static void test_keyword_free(struct comp_dev *dev)
 
 	comp_info(dev, "test_keyword_free()");
 
+#if CONFIG_AMS
+	int ret;
+
+	/* Unregister KD as AMS producer */
+	ret = kpb_unregister_ams_producer(dev, cd->ams_uuid_id);
+	if (ret)
+		comp_err(dev, "test_keyword_free(): unregister ams error %d", ret);
+#endif
+
 	ipc_msg_free(cd->msg);
 	comp_data_blob_handler_free(cd->model_handler);
 	rfree(cd);
@@ -798,6 +841,10 @@ static int test_keyword_params(struct comp_dev *dev,
 
 	cd->config.activation_threshold = err;
 
+#if CONFIG_AMS
+	cd->ams_uuid_id = AMS_INVALID_MSG_TYPE;
+#endif
+
 	return 0;
 }
 
@@ -874,6 +921,7 @@ static int test_keyword_prepare(struct comp_dev *dev)
 	struct comp_data *cd = comp_get_drvdata(dev);
 	uint16_t valid_bits = cd->sample_valid_bytes * 8;
 	uint16_t sample_width;
+	int ret;
 
 #if CONFIG_IPC_MAJOR_4
 	sample_width = cd->base_cfg.audio_fmt.depth;
@@ -887,7 +935,7 @@ static int test_keyword_prepare(struct comp_dev *dev)
 		/* Default threshold value has to be changed
 		 * according to host new format.
 		 */
-		int ret = test_keyword_get_threshold(dev, valid_bits);
+		ret = test_keyword_get_threshold(dev, valid_bits);
 
 		if (ret < 0) {
 			comp_err(dev, "test_keyword_prepare(): unsupported sample width %u",
@@ -902,6 +950,14 @@ static int test_keyword_prepare(struct comp_dev *dev)
 					   &cd->data_blob_size,
 					   &cd->data_blob_crc);
 
+#if CONFIG_AMS
+	/* Register KD as AMS producer */
+	ret = kpb_register_ams_producer(dev, &cd->ams_payload, ams_kpd_msg_uuid,
+					cd->ams_uuid_id);
+	if (ret)
+		return ret;
+#endif
+
 	return comp_set_state(dev, COMP_TRIGGER_PREPARE);
 }