[Common] Updated copy tensor info

Added function to copy tensor_info with variable length

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/gst/nnstreamer/nnstreamer_plugin_api.h b/gst/nnstreamer/nnstreamer_plugin_api.h
index d8d8bce..e63a4e6 100644 (file)
--- a/gst/nnstreamer/nnstreamer_plugin_api.h
+++ b/gst/nnstreamer/nnstreamer_plugin_api.h
@@ -76,6 +76,14 @@ extern gboolean
 gst_tensor_info_is_equal (const GstTensorInfo * i1, const GstTensorInfo * i2);
 
 /**
+ * @brief Copy tensor info upto n elements
+ * @note Copied info should be freed with gst_tensor_info_free()
+ */
+extern void
+gst_tensor_info_copy_n (GstTensorInfo * dest, const GstTensorInfo * src,
+    const guint n);
+
+/**
  * @brief Copy tensor info
  * @note Copied info should be freed with gst_tensor_info_free()
  */

diff --git a/gst/nnstreamer/tensor_common.c b/gst/nnstreamer/tensor_common.c
index 13584b3..a756464 100644 (file)
--- a/gst/nnstreamer/tensor_common.c
+++ b/gst/nnstreamer/tensor_common.c
@@ -174,11 +174,12 @@ gst_tensor_info_is_equal (const GstTensorInfo * i1, const GstTensorInfo * i2)
 }
 
 /**
- * @brief Copy tensor info
+ * @brief Copy tensor info upto n elements
  * @note Copied info should be freed with gst_tensor_info_free()
  */
 void
-gst_tensor_info_copy (GstTensorInfo * dest, const GstTensorInfo * src)
+gst_tensor_info_copy_n (GstTensorInfo * dest, const GstTensorInfo * src,
+    const guint n)
 {
   guint i;
 
@@ -188,12 +189,22 @@ gst_tensor_info_copy (GstTensorInfo * dest, const GstTensorInfo * src)
   dest->name = (src->name) ? g_strdup (src->name) : NULL;
   dest->type = src->type;
 
-  for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++) {
+  for (i = 0; i < n; i++) {
     dest->dimension[i] = src->dimension[i];
   }
 }
 
 /**
+ * @brief Copy tensor info
+ * @note Copied info should be freed with gst_tensor_info_free()
+ */
+void
+gst_tensor_info_copy (GstTensorInfo * dest, const GstTensorInfo * src)
+{
+  gst_tensor_info_copy_n (dest, src, NNS_TENSOR_RANK_LIMIT);
+}
+
+/**
  * @brief Initialize the tensors info structure
  * @param info tensors info structure to be initialized
  */