Handling the process death scenario for Bitmap

Author: srikrishnasakunia

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt

@@ -16,6 +16,7 @@
 package com.android.developers.androidify.customize
 
 import android.app.Application
+import android.graphics.Bitmap
 import android.net.Uri
 import android.util.Log
 import androidx.compose.foundation.layout.fillMaxSize
@@ -29,13 +30,11 @@ import dagger.assisted.Assisted
 import dagger.assisted.AssistedFactory
 import dagger.assisted.AssistedInject
 import dagger.hilt.android.lifecycle.HiltViewModel
-import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
-import javax.inject.Inject
 
 @HiltViewModel(assistedFactory = CustomizeExportViewModel.Factory::class)
 class CustomizeExportViewModel @AssistedInject constructor(
@@ -65,11 +64,12 @@ class CustomizeExportViewModel @AssistedInject constructor(
 
     init {
         _state.update {
-            CustomizeExportState(
+            it.copy(
                 originalImageUrl = originalImageUrl,
                 exportImageCanvas = it.exportImageCanvas.copy(imageUri = resultImageUrl),
             )
         }
+        loadInitialBitmap(resultImageUrl)
     }
 
 
@@ -221,8 +221,7 @@ class CustomizeExportViewModel @AssistedInject constructor(
                 }
                 return@launch
             }
-
-            val image = state.value.exportImageCanvas.imageUri?.let { uri -> localFileProvider.loadBitmapFromUri(uri) }
+            val image = state.value.exportImageCanvas.imageBitmap
             if (image == null) {
                 return@launch
             }
@@ -278,4 +277,19 @@ class CustomizeExportViewModel @AssistedInject constructor(
             it.copy(selectedTool = tool)
         }
     }
+
+    private fun loadInitialBitmap(uri: Uri) {
+        viewModelScope.launch {
+            try {
+                val bitmap = localFileProvider.loadBitmapFromUri(uri)
+                _state.update {
+                    it.copy(
+                        exportImageCanvas = it.exportImageCanvas.copy(imageBitmap = bitmap)
+                    )
+                }
+            } catch (e: Exception) {
+                _snackbarHostState.value.showSnackbar("Could not load image.")
+            }
+        }
+    }
 }

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt

@@ -76,6 +76,7 @@ data class BackgroundToolState(
 
 data class ExportImageCanvas(
     val imageUri: Uri? = null,
+    val imageBitmap: Bitmap? = null,
     val imageBitmapRemovedBackground: Bitmap? = null,
     val aspectRatioOption: SizeOption = SizeOption.Square,
     val canvasSize: Size = Size(1000f, 1000f),