From 8e20d683ca3523bdcdc0135b7012e9addbd9c524 Mon Sep 17 00:00:00 2001
From: Yelin Jeong <yelini.jeong@samsung.com>
Date: Mon, 22 Apr 2024 20:56:21 +0900
Subject: [PATCH] App/Service: Bound service to MainActivity

This patch bounds service to MainActivity.

Signed-off-by: Yelin Jeong <yelini.jeong@samsung.com>
---
 .../src/main/AndroidManifest.xml              |  1 +
 .../ml/inference/offloading/MainActivity.kt   | 58 +++++++++++++++++--
 .../ml/inference/offloading/MainService.kt    | 17 +++++-
 .../src/main/res/layout/activity_main.xml     | 15 +++++
 4 files changed, 84 insertions(+), 7 deletions(-)
 create mode 100644 ml_inference_offloading/src/main/res/layout/activity_main.xml

diff --git a/ml_inference_offloading/src/main/AndroidManifest.xml b/ml_inference_offloading/src/main/AndroidManifest.xml
index c572016..fa91404 100644
--- a/ml_inference_offloading/src/main/AndroidManifest.xml
+++ b/ml_inference_offloading/src/main/AndroidManifest.xml
@@ -4,6 +4,7 @@
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE"/>
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE_SPECIAL_USE"/>
     <uses-permission android:name="android.permission.POST_NOTIFICATIONS" />
+    <uses-permission android:name="android.permission.INTERNET" />
 
     <application
         android:icon="@mipmap/ic_launcher"
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 395ceb0..7878702 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
@@ -1,19 +1,39 @@
 package ai.nnstreamer.ml.inference.offloading
 
+import ai.nnstreamer.ml.inference.offloading.ui.theme.NnstreamerandroidTheme
+import android.content.ComponentName
+import android.content.Context
+import android.content.Intent
+import android.content.ServiceConnection
 import android.os.Bundle
+import android.os.IBinder
+import android.view.View
+import android.widget.Button
 import androidx.activity.ComponentActivity
 import androidx.activity.compose.setContent
 import androidx.compose.foundation.layout.fillMaxSize
 import androidx.compose.material3.MaterialTheme
 import androidx.compose.material3.Surface
-import androidx.compose.material3.Text
-import androidx.compose.runtime.Composable
 import androidx.compose.ui.Modifier
-import androidx.compose.ui.tooling.preview.Preview
-import ai.nnstreamer.ml.inference.offloading.ui.theme.NnstreamerandroidTheme
-import android.content.Intent
 
 class MainActivity : ComponentActivity() {
+    private val TAG = "MainActivity"
+    private lateinit var mService: MainService
+    private var mBound: Boolean = false
+
+    private val connection = object : ServiceConnection {
+
+        override fun onServiceConnected(className: ComponentName, service: IBinder) {
+            val binder = service as MainService.LocalBinder
+            mService = binder.getService()
+            mBound = true
+        }
+
+        override fun onServiceDisconnected(arg0: ComponentName) {
+            mBound = false
+        }
+    }
+
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
         setContent {
@@ -25,7 +45,35 @@ class MainActivity : ComponentActivity() {
                 ) { }
             }
         }
+        setContentView(R.layout.activity_main)
 
         startForegroundService(Intent(this, MainService::class.java))
+
+        val start = findViewById<Button>(R.id.start)
+        start.setOnClickListener(View.OnClickListener {
+            if (mBound) {
+                mService.startServer()
+            }
+        })
+
+        val stop = findViewById<Button>(R.id.stop)
+        stop.setOnClickListener(View.OnClickListener {
+            if (mBound) {
+                mService.stopServer()
+            }
+        })
+    }
+
+    override fun onStart() {
+        super.onStart()
+        Intent(this, MainService::class.java).also { intent->
+            bindService(intent, connection, Context.BIND_AUTO_CREATE)
+        }
+    }
+
+    override fun onStop() {
+        super.onStop()
+        unbindService(connection)
+        mBound = false
     }
 }
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 c5732ed..ef6a0d8 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
@@ -7,6 +7,7 @@ import android.app.Service
 import android.content.Intent
 import android.content.pm.PackageManager
 import android.content.pm.ServiceInfo
+import android.os.Binder
 import android.os.Handler
 import android.os.HandlerThread
 import android.os.IBinder
@@ -35,12 +36,16 @@ class MainService : Service() {
         }
     }
 
+    inner class LocalBinder : Binder() {
+        fun getService(): MainService = this@MainService
+    }
+
     private val TAG = "MainService"
+    private val binder = LocalBinder()
     private lateinit var serviceHandler : MainHandler
     private lateinit var serviceLooper : Looper
     private lateinit var handlerThread: HandlerThread
     private var initialized = false
-
     private fun startForeground() {
         // Get NotificationManager
         val notificationManager = getSystemService(NOTIFICATION_SERVICE) as NotificationManager
@@ -101,7 +106,7 @@ class MainService : Service() {
     }
 
     override fun onBind(intent: Intent): IBinder {
-        TODO("Return the communication channel to the service.")
+        return binder
     }
 
     override fun onDestroy() {
@@ -125,4 +130,12 @@ class MainService : Service() {
             }
         }
     }
+
+    fun startServer() {
+        TODO("Not yet implemented")
+    }
+
+    fun stopServer() {
+        TODO("Not yet implemented")
+    }
 }
diff --git a/ml_inference_offloading/src/main/res/layout/activity_main.xml b/ml_inference_offloading/src/main/res/layout/activity_main.xml
new file mode 100644
index 0000000..f05a154
--- /dev/null
+++ b/ml_inference_offloading/src/main/res/layout/activity_main.xml
@@ -0,0 +1,15 @@
+<?xml version="1.0" encoding="utf-8"?>
+<LinearLayout xmlns:android="http://schemas.android.com/apk/res/android"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent">
+    <Button
+        android:id="@+id/start"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:text="Start" />
+    <Button
+        android:id="@+id/stop"
+        android:layout_width="wrap_content"
+        android:layout_height="wrap_content"
+        android:text="Stop" />
+</LinearLayout>