From daa563f8b359f923a3142ec527fe41a74c3ecf3b Mon Sep 17 00:00:00 2001
From: Yelin Jeong <yelini.jeong@samsung.com>
Date: Wed, 19 Jun 2024 11:41:16 +0900
Subject: [PATCH] App: Seperate pipeline pause and close

This patch seperates pipeline pause an close.
Also, now we can see pipeline status in main activity.

Signed-off-by: Yelin Jeong <yelini.jeong@samsung.com>
---
 .../ml/inference/offloading/MainActivity.kt   | 15 ++++--
 .../ml/inference/offloading/MainService.kt    | 52 +++++++++++++------
 .../src/main/res/layout/models.xml            | 10 +++-
 .../src/main/res/values/strings.xml           |  4 +-
 4 files changed, 59 insertions(+), 22 deletions(-)

diff --git a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainActivity.kt b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainActivity.kt
index abf09ca..afbc387 100644
--- a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainActivity.kt
+++ b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainActivity.kt
@@ -41,14 +41,23 @@ class MainActivity : ComponentActivity() {
         fun bind(info: ModelInfo) {
             val start = itemView.findViewById<Button>(R.id.start)
             val stop = itemView.findViewById<Button>(R.id.stop)
-            val port = itemView.findViewById<TextView>(R.id.port)
+            val close = itemView.findViewById<Button>(R.id.close)
+            val status = itemView.findViewById<TextView>(R.id.status)
 
             start.setOnClickListener(View.OnClickListener {
-                val serverPort = mService?.startServer(info.name, info.filter)
-                port.text = "Listening on port: " + serverPort.toString();
+                mService?.startServer(info.name, info.filter)
+                mService?.getPort(info.name)?.let { serverPort ->
+                    if (serverPort < 0) status.text = "Failed to start the server"
+                    else status.text = "Listening on port: " + serverPort.toString();
+                }
             })
             stop.setOnClickListener(View.OnClickListener {
                 mService?.stopServer(info.name)
+                status.text = resources.getString(R.string.server_paused)
+            })
+            close.setOnClickListener(View.OnClickListener {
+                mService?.closeServer(info.name)
+                status.text = resources.getString(R.string.server_unknown)
             })
         }
     }
diff --git a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
index e10cf8b..c28bfa7 100644
--- a/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
+++ b/ml_inference_offloading/src/main/java/ai/nnstreamer/ml/inference/offloading/MainService.kt
@@ -27,6 +27,12 @@ import java.net.Inet4Address
 import java.net.ServerSocket
 import kotlin.concurrent.thread
 
+// todo: Define DTO with generality and extensibility
+data class ServerInfo(
+    val pipeline: Pipeline,
+    val port: Int,
+    var status: Pipeline.State
+)
 
 class MainService : Service() {
     private inner class MainHandler(looper: Looper) : Handler(looper) {
@@ -73,8 +79,7 @@ class MainService : Service() {
     private lateinit var serviceLooper : Looper
     private lateinit var handlerThread: HandlerThread
     private var initialized = false
-    private var port = -1
-    private var serverInfo = mutableMapOf<String,Pipeline>()
+    private var serverInfoMap = mutableMapOf<String,ServerInfo>()
 
     private fun startForeground() {
         // Get NotificationManager
@@ -145,8 +150,8 @@ class MainService : Service() {
     }
 
     override fun onDestroy() {
-        serverInfo.values.forEach { pipeline ->
-            pipeline.close()
+        serverInfoMap.values.forEach { status ->
+            status.pipeline.close()
         }
         Toast.makeText(this, "The MainService has been gone", Toast.LENGTH_SHORT).show()
     }
@@ -228,22 +233,37 @@ class MainService : Service() {
         return port
     }
 
-    fun startServer(name:String, filter: String): Int {
-        val hostAddress = getIpAddress()
-        if (!isPortAvailable(port)) {
-            port = findPort()
-        }
+    fun getPort(name: String): Int {
+        return serverInfoMap[name]?.port ?: -1
+    }
 
-        val desc = "tensor_query_serversrc host=" + hostAddress + " port=" + port.toString() +
-                " ! " + filter + " ! tensor_query_serversink async=false"
-        val tensorQueryServer = Pipeline(desc, null)
-        serverInfo[name] = tensorQueryServer
-        tensorQueryServer.start()
+    fun startServer(name:String, filter: String) {
+        if (!serverInfoMap.containsKey(name)) {
+            val hostAddress = getIpAddress()
+            val port = findPort()
+            val desc = "tensor_query_serversrc host=" + hostAddress + " port=" + port.toString() +
+                    " ! " + filter + " ! tensor_query_serversink async=false"
+            val tensorQueryServer = Pipeline(desc, null)
+            serverInfoMap[name] = ServerInfo(tensorQueryServer, port, Pipeline.State.UNKNOWN)
+        }
 
-        return port
+        serverInfoMap[name]?.let { modelStatus ->
+            modelStatus.pipeline.start()
+            modelStatus.status = Pipeline.State.PLAYING
+        }
     }
 
     fun stopServer(name:String) {
-        serverInfo[name]?.close()
+        serverInfoMap[name]?.let { modelStatus ->
+            modelStatus.pipeline.stop()
+            modelStatus.status = Pipeline.State.PAUSED
+        }
+    }
+
+    fun closeServer(name:String) {
+        serverInfoMap[name]?.let { modelStatus ->
+            modelStatus.pipeline.close()
+            serverInfoMap.remove(name)
+        }
     }
 }
diff --git a/ml_inference_offloading/src/main/res/layout/models.xml b/ml_inference_offloading/src/main/res/layout/models.xml
index 662ce36..14ed50a 100644
--- a/ml_inference_offloading/src/main/res/layout/models.xml
+++ b/ml_inference_offloading/src/main/res/layout/models.xml
@@ -14,8 +14,14 @@
         android:layout_width="wrap_content"
         android:layout_height="wrap_content"
         android:text="Stop" />
+    <Button
+        android:id="@+id/close"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:text="Close" />
     <TextView
-        android:id="@+id/port"
+        android:id="@+id/status"
         android:layout_width="match_parent"
-        android:layout_height="match_parent" />
+        android:layout_height="wrap_content"
+        android:text="@string/server_unknown"/>
 </LinearLayout>
diff --git a/ml_inference_offloading/src/main/res/values/strings.xml b/ml_inference_offloading/src/main/res/values/strings.xml
index 1d4903e..de0e6f1 100644
--- a/ml_inference_offloading/src/main/res/values/strings.xml
+++ b/ml_inference_offloading/src/main/res/values/strings.xml
@@ -1,4 +1,6 @@
 <resources>
     <string name="app_name">ML Inference Offloading</string>
     <string name="title_activity_main">MainActivity</string>
-</resources>
\ No newline at end of file
+    <string name="server_paused">Server is paused</string>
+    <string name="server_unknown">There is no tensor query server</string>
+</resources>