From: Hyoung Joo Ahn <hello.ahn@samsung.com>
Date: Thu, 30 Jul 2020 09:55:55 +0000 (+0900)
Subject: [C-API] add the EOS push at the destroy moment
X-Git-Tag: accepted/tizen/unified/20200802.223717^0
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=63d46c99e44d6c4263be075f08fbe28649a646af;p=platform%2Fupstream%2Fnnstreamer.git

[C-API] add the EOS push at the destroy moment

for the specific situation, tensor_mux(refresh), the EOS buffer has to be inserted to make the pipeline state as NULL

Signed-off-by: Hyoung Joo Ahn <hello.ahn@samsung.com>
---

diff --git a/api/capi/include/nnstreamer-capi-private.h b/api/capi/include/nnstreamer-capi-private.h
index 17038be..e4e04e6 100644
--- a/api/capi/include/nnstreamer-capi-private.h
+++ b/api/capi/include/nnstreamer-capi-private.h
@@ -88,6 +88,8 @@ typedef enum { MM_RESOURCE_MANAGER_RES_TYPE_MAX } mm_resource_manager_res_type_e
 #define TIZEN5PLUS 0
 #endif  /* __TIZEN__ */
 
+#define EOS_MESSAGE_TIME_LIMIT 100
+
 #ifdef __cplusplus
 extern "C" {
 #endif /* __cplusplus */
@@ -205,6 +207,7 @@ struct _ml_pipeline {
   GstBus *bus;            /**< The bus of the pipeline */
   gulong signal_msg;      /**< The message signal (connected to bus) */
   GMutex lock;            /**< Lock for pipeline operations */
+  gboolean isEOS;         /**< The pipeline is EOS state */
   GHashTable *namednodes; /**< hash table of "element"s. */
   GHashTable *resources;  /**< hash table of resources to construct the pipeline */
   pipeline_state_cb_s state_cb; /**< Callback to notify the change of pipeline state */
diff --git a/api/capi/src/nnstreamer-capi-pipeline.c b/api/capi/src/nnstreamer-capi-pipeline.c
index e923d20..dd44a80 100644
--- a/api/capi/src/nnstreamer-capi-pipeline.c
+++ b/api/capi/src/nnstreamer-capi-pipeline.c
@@ -294,6 +294,9 @@ cb_bus_sync_message (GstBus * bus, GstMessage * message, gpointer user_data)
     return;
 
   switch (GST_MESSAGE_TYPE (message)) {
+    case GST_MESSAGE_EOS:
+      pipe_h->isEOS = TRUE;
+      break;
     case GST_MESSAGE_STATE_CHANGED:
       if (GST_MESSAGE_SRC (message) == GST_OBJECT_CAST (pipe_h->element)) {
         GstState old_state, new_state;
@@ -335,6 +338,26 @@ cleanup_node (gpointer data)
   ml_pipeline_element *e = data;
 
   g_mutex_lock (&e->lock);
+  if (e->type == ML_PIPELINE_ELEMENT_APP_SRC && !e->pipe->isEOS) {
+    int eos_check_cnt = 0;
+
+    /** to push EOS event, the pipeline should be in PLAYING state */
+    gst_element_set_state (e->pipe->element, GST_STATE_PLAYING);
+    
+    if (gst_app_src_end_of_stream (GST_APP_SRC (e->element)) != GST_FLOW_OK) {
+      ml_logw ("Failed to set EOS in %s", e->name);
+    }
+    while (!e->pipe->isEOS) {
+      eos_check_cnt++;
+      /** check EOS every 1ms */
+      g_usleep (1000);
+      if (eos_check_cnt >= EOS_MESSAGE_TIME_LIMIT) {
+        ml_loge ("Failed to get EOS message");
+        break;
+      }
+    }
+  }
+
   g_free (e->name);
   if (e->src)
     gst_object_unref (e->src);
@@ -555,6 +578,7 @@ construct_pipeline_internal (const char *pipeline_description,
 
   g_mutex_init (&pipe_h->lock);
 
+  pipe_h->isEOS = FALSE;
   pipe_h->namednodes =
       g_hash_table_new_full (g_str_hash, g_str_equal, g_free, cleanup_node);
 
@@ -662,10 +686,6 @@ ml_pipeline_destroy (ml_pipeline_h pipe)
 
   /* Before changing the state, remove all callbacks. */
   p->state_cb.cb = NULL;
-  if (p->bus) {
-    g_signal_handler_disconnect (p->bus, p->signal_msg);
-    gst_object_unref (p->bus);
-  }
 
   g_hash_table_remove_all (p->namednodes);
   g_hash_table_remove_all (p->resources);
@@ -701,6 +721,11 @@ ml_pipeline_destroy (ml_pipeline_h pipe)
     p->element = NULL;
   }
 
+  if (p->bus) {
+    g_signal_handler_disconnect (p->bus, p->signal_msg);
+    gst_object_unref (p->bus);
+  }
+
   /* Destroy registered callback handles and resources */
   g_hash_table_destroy (p->namednodes);
   g_hash_table_destroy (p->resources);