Module: Audio: Add s16/32 source_get_data and sink_get_buffer

This patch adds helper functions source_get_data_s16(),
source_get_data_s32(), sink_get_buffer_s16(), and
sink_get_buffer_s32().

The buffer_samples as number of samples simplifies the processing
function with no division or shift needed to convert buffer
size in bytes to samples. Also the int16_t and int32_t typed
arguments for data pointer and buffer start avoid type casts
in a typical simple processing function.

Signed-off-by: Seppo Ingalsuo <seppo.ingalsuo@linux.intel.com>

Author: singalsu

src/include/module/audio/sink_api.h

@@ -170,6 +170,44 @@ size_t sink_get_free_frames(struct sof_sink *sink);
 int sink_get_buffer(struct sof_sink *sink, size_t req_size, void **data_ptr, void **buffer_start,
 		    size_t *buffer_size);
 
+/**
+ * Get a circular buffer to operate on (to write).
+ *
+ * Same as sink_get_buffer() except that the size of circular buffer is returned as
+ * 16 bit samples count. The returned samples count simplifies pointer arithmetic in a
+ * samples process function. The data pointers are int16_t type.
+ *
+ * @param sink a handler to sink
+ * @param [in] req_size requested size of space
+ * @param [out] data_ptr a pointer to the space will be provided there
+ * @param [out] buffer_start pointer to circular buffer start
+ * @param [out] buffer_samples number of s16 samples total in circular buffer
+ *
+ * @retval -ENODATA if req_size is bigger than free space
+ *
+ */
+int sink_get_buffer_s16(struct sof_sink *sink, size_t req_size, int16_t **data_ptr,
+			int16_t **buffer_start, int *buffer_samples);
+
+/**
+ * Get a circular buffer to operate on (to write).
+ *
+ * Same as sink_get_buffer() except that the size of circular buffer is returned as
+ * 32 bit samples count. The returned samples count simplifies pointer arithmetic in a
+ * samples process function. The data pointers are int32_t type.
+ *
+ * @param sink a handler to sink
+ * @param [in] req_size requested size of space
+ * @param [out] data_ptr a pointer to the space will be provided there
+ * @param [out] buffer_start pointer to circular buffer start
+ * @param [out] buffer_samples number of s32 samples total in circular buffer
+ *
+ * @retval -ENODATA if req_size is bigger than free space
+ *
+ */
+int sink_get_buffer_s32(struct sof_sink *sink, size_t req_size, int32_t **data_ptr,
+			int32_t **buffer_start, int *buffer_samples);
+
 /**
  * Commits that the buffer previously obtained by get_buffer is filled with data
  * and ready to be used

src/include/module/audio/source_api.h

@@ -184,6 +184,42 @@ size_t source_get_data_frames_available(struct sof_source *source);
 int source_get_data(struct sof_source *source, size_t req_size, void const **data_ptr,
 		    void const **buffer_start, size_t *buffer_size);
 
+/**
+ * Retrieves a fragment of circular data (to read)
+ *
+ * Same as source_get_data() except that the size of circular buffer is returned as
+ * 16 bit samples count. The returned samples count simplifies pointer arithmetic in a
+ * samples process function. The data pointers are int16_t type.
+ *
+ * @param source a handler to source
+ * @param [in] req_size requested size of data.
+ * @param [out] data_ptr a pointer to data will be provided there
+ * @param [out] buffer_start pointer to circular buffer start
+ * @param [out] buffer_samples number of 16 bit samples total in circular buffer
+ *
+ * @retval -ENODATA if req_size is bigger than available data
+ */
+int source_get_data_s16(struct sof_source *source, size_t req_size, int16_t const **data_ptr,
+			int16_t const **buffer_start, int *buffer_samples);
+
+/**
+ * Retrieves a fragment of circular data (to read)
+ *
+ * Same as source_get_data() except that the size of circular buffer is returned as
+ * 32 bit samples count. The returned samples count simplifies pointer arithmetic in a
+ * samples process function. The data pointers are int32_t type.
+ *
+ * @param source a handler to source
+ * @param [in] req_size requested size of data.
+ * @param [out] data_ptr a pointer to data will be provided there
+ * @param [out] buffer_start pointer to circular buffer start
+ * @param [out] buffer_samples number of 32 bit samples total in circular buffer
+ *
+ * @retval -ENODATA if req_size is bigger than available data
+ */
+int source_get_data_s32(struct sof_source *source, size_t req_size, int32_t const **data_ptr,
+			int32_t const **buffer_start, int *buffer_samples);
+
 /**
  * Releases fragment previously obtained by source_get_data()
  * Once called, the data are no longer available for the caller

src/module/audio/sink_api.c

@@ -36,6 +36,38 @@ int sink_get_buffer(struct sof_sink *sink, size_t req_size,
 }
 EXPORT_SYMBOL(sink_get_buffer);
 
+int sink_get_buffer_s16(struct sof_sink *sink, size_t req_size,
+			int16_t **data_ptr, int16_t **buffer_start, int *buffer_samples)
+{
+	size_t buffer_size;
+	int ret;
+
+	ret = sink_get_buffer(sink, req_size, (void **)data_ptr, (void **)buffer_start,
+			      &buffer_size);
+	if (ret)
+		return ret;
+
+	*buffer_samples = buffer_size >> 1;
+	return 0;
+}
+EXPORT_SYMBOL(sink_get_buffer_s16);
+
+int sink_get_buffer_s32(struct sof_sink *sink, size_t req_size,
+			int32_t **data_ptr, int32_t **buffer_start, int *buffer_samples)
+{
+	size_t buffer_size;
+	int ret;
+
+	ret = sink_get_buffer(sink, req_size, (void **)data_ptr, (void **)buffer_start,
+			      &buffer_size);
+	if (ret)
+		return ret;
+
+	*buffer_samples = buffer_size >> 2;
+	return 0;
+}
+EXPORT_SYMBOL(sink_get_buffer_s32);
+
 int sink_commit_buffer(struct sof_sink *sink, size_t commit_size)
 {
 	int ret;

src/module/audio/source_api.c

@@ -35,6 +35,38 @@ int source_get_data(struct sof_source *source, size_t req_size,
 }
 EXPORT_SYMBOL(source_get_data);
 
+int source_get_data_s16(struct sof_source *source, size_t req_size, int16_t const **data_ptr,
+			int16_t const **buffer_start, int *buffer_samples)
+{
+	size_t buffer_size;
+	int ret;
+
+	ret = source_get_data(source, req_size, (void const **)data_ptr,
+			      (void const **)buffer_start, &buffer_size);
+	if (ret)
+		return ret;
+
+	*buffer_samples = buffer_size >> 1;
+	return 0;
+}
+EXPORT_SYMBOL(source_get_data_s16);
+
+int source_get_data_s32(struct sof_source *source, size_t req_size, int32_t const **data_ptr,
+			int32_t const **buffer_start, int *buffer_samples)
+{
+	size_t buffer_size;
+	int ret;
+
+	ret = source_get_data(source, req_size, (void const **)data_ptr,
+			      (void const **)buffer_start, &buffer_size);
+	if (ret)
+		return ret;
+
+	*buffer_samples = buffer_size >> 2;
+	return 0;
+}
+EXPORT_SYMBOL(source_get_data_s32);
+
 int source_release_data(struct sof_source *source, size_t free_size)
 {
 	int ret;