Support URI source for ImageScreenObject.

Author: shogo4405

haishinkit/src/main/java/com/haishinkit/screen/ImageScreenObject.kt

@@ -2,8 +2,33 @@ package com.haishinkit.screen
 
 import android.graphics.Bitmap
 import android.graphics.BitmapFactory
+import android.net.Uri
 import android.util.Base64
-import java.io.ByteArrayOutputStream
+import androidx.core.net.toUri
+import com.haishinkit.util.DataUriUtil
+
+private sealed interface ImageSource {
+    val uri: Uri
+
+    fun toBitmap(): Bitmap
+
+    class Data(override val uri: Uri) : ImageSource {
+        override fun toBitmap(): Bitmap {
+            val data = uri.schemeSpecificPart
+            val base64 = data.substringAfter("base64,")
+            val bytes = Base64.decode(base64, Base64.DEFAULT)
+            return BitmapFactory.decodeByteArray(bytes, 0, bytes.size)
+        }
+    }
+
+    companion object {
+        fun parseImageSource(uri: Uri?): ImageSource =
+            when (uri?.scheme) {
+                "data" -> Data(uri)
+                else -> error("unsupported")
+            }
+    }
+}
 
 /**
  * An object that manages offscreen rendering an image source.
@@ -13,7 +38,7 @@ open class ImageScreenObject(
 ) : ScreenObject(id) {
     private interface Keys {
         companion object {
-            const val BITMAP: String = "bitmap"
+            const val SOURCE: String = "source"
         }
     }
 
@@ -26,14 +51,28 @@ open class ImageScreenObject(
             invalidateLayout()
         }
 
+    private var source: String? = null
+        get() {
+            if (field == null && bitmap != null) {
+                return DataUriUtil.encode(bitmap!!)
+            }
+            return field
+        }
+        set(value) {
+            field = value
+            runCatching {
+                bitmap = ImageSource.parseImageSource(source?.toUri()).toBitmap()
+            }
+        }
+
     override var elements: Map<String, String>
         get() {
             return buildMap {
-                put(Keys.BITMAP, bitmap?.toBase64() ?: "")
+                put(Keys.SOURCE, source ?: "")
             }
         }
         set(value) {
-            bitmap = value[Keys.BITMAP]?.toBitmap()
+            source = value[Keys.SOURCE]
         }
 
     init {
@@ -42,22 +81,5 @@ open class ImageScreenObject(
 
     companion object {
         const val TYPE: String = "image"
-
-        private val TAG = VideoScreenObject::class.java.simpleName
     }
 }
-
-private fun String.toBitmap(): Bitmap {
-    val bytes = Base64.decode(this, Base64.NO_WRAP)
-    return BitmapFactory.decodeByteArray(bytes, 0, bytes.size)
-}
-
-private fun Bitmap.toBase64(
-    format: Bitmap.CompressFormat = Bitmap.CompressFormat.PNG,
-    quality: Int = 100,
-): String {
-    val outputStream = ByteArrayOutputStream()
-    compress(format, quality, outputStream)
-    val bytes = outputStream.toByteArray()
-    return Base64.encodeToString(bytes, Base64.NO_WRAP)
-}

haishinkit/src/main/java/com/haishinkit/screen/TextScreenObject.kt

@@ -97,6 +97,5 @@ class TextScreenObject(
 
     companion object {
         const val TYPE: String = "text"
-        private val TAG = VideoScreenObject::class.java.simpleName
     }
 }

haishinkit/src/main/java/com/haishinkit/util/DataUriUtil.kt

@@ -0,0 +1,36 @@
+package com.haishinkit.util
+
+import android.graphics.Bitmap
+import android.os.Build
+import android.util.Base64
+import java.io.ByteArrayOutputStream
+
+internal object DataUriUtil {
+    fun encode(
+        bitmap: Bitmap,
+        quality: Int = 90,
+    ): String {
+        val bytes = encodeToBytes(bitmap, quality)
+        val base64 = Base64.encodeToString(bytes, Base64.NO_WRAP)
+        return "data:image/webp;base64,$base64"
+    }
+
+    private fun encodeToBytes(
+        bitmap: Bitmap,
+        quality: Int,
+    ): ByteArray {
+        val out = ByteArrayOutputStream()
+        val format =
+            if (Build.VERSION.SDK_INT >= 30) {
+                if (bitmap.hasAlpha()) {
+                    Bitmap.CompressFormat.WEBP_LOSSLESS
+                } else {
+                    Bitmap.CompressFormat.WEBP_LOSSY
+                }
+            } else {
+                Bitmap.CompressFormat.WEBP
+            }
+        bitmap.compress(format, quality, out)
+        return out.toByteArray()
+    }
+}