1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
* Copyright (C) 2023 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.wallpaper.picker.preview.ui.viewmodel
import android.app.WallpaperColors
import android.content.Context
import android.graphics.Bitmap
import android.graphics.BitmapFactory
import android.graphics.ColorSpace
import android.graphics.Point
import android.graphics.Rect
import com.android.wallpaper.asset.Asset
import com.android.wallpaper.model.wallpaper.ScreenOrientation
import com.android.wallpaper.model.wallpaper.WallpaperModel.StaticWallpaperModel
import com.android.wallpaper.module.WallpaperPreferences
import com.android.wallpaper.picker.di.modules.BackgroundDispatcher
import com.android.wallpaper.picker.preview.domain.interactor.WallpaperPreviewInteractor
import com.android.wallpaper.picker.preview.ui.WallpaperPreviewActivity
import dagger.hilt.android.qualifiers.ApplicationContext
import dagger.hilt.android.scopes.ViewModelScoped
import java.io.ByteArrayOutputStream
import javax.inject.Inject
import kotlinx.coroutines.CancellableContinuation
import kotlinx.coroutines.CoroutineDispatcher
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.combine
import kotlinx.coroutines.flow.filterNotNull
import kotlinx.coroutines.flow.flowOn
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.suspendCancellableCoroutine
/** View model for static wallpaper preview used in [WallpaperPreviewActivity] and its fragments */
@ViewModelScoped
class StaticWallpaperPreviewViewModel
@Inject
constructor(
interactor: WallpaperPreviewInteractor,
@ApplicationContext private val context: Context,
private val wallpaperPreferences: WallpaperPreferences,
@BackgroundDispatcher private val bgDispatcher: CoroutineDispatcher,
) {
/** The state of static wallpaper crop in full preview, before user confirmation. */
var fullPreviewCrop: Rect? = null
private val _cropHints: MutableStateFlow<Map<ScreenOrientation, Rect>?> = MutableStateFlow(null)
private val staticWallpaperModel: Flow<StaticWallpaperModel> =
interactor.wallpaperModel.map { it as? StaticWallpaperModel }.filterNotNull()
val lowResBitmap: Flow<Bitmap> =
staticWallpaperModel
.map { it.staticWallpaperData.asset.getLowResBitmap(context) }
.filterNotNull()
.flowOn(bgDispatcher)
val subsamplingScaleImageViewModel: Flow<FullResWallpaperViewModel> =
staticWallpaperModel
.map { it.staticWallpaperData.asset }
.combine(_cropHints) { asset, cropHints ->
val dimensions = asset.decodeRawDimensions()
asset.decodeBitmap(dimensions)?.let { bitmap ->
FullResWallpaperViewModel(bitmap, dimensions, cropHints)
}
}
.filterNotNull()
.flowOn(bgDispatcher)
val wallpaperColors: Flow<WallpaperColors> =
staticWallpaperModel
.map { wallpaperPreferences.getWallpaperColors(it.commonWallpaperData.id.uniqueId) }
.filterNotNull()
fun updateCropHints(cropHints: Map<ScreenOrientation, Rect>) {
_cropHints.value = _cropHints.value?.plus(cropHints) ?: cropHints
}
// TODO b/296288298 Create a util class for Bitmap and Asset
private suspend fun Asset.decodeRawDimensions(): Point =
suspendCancellableCoroutine { k: CancellableContinuation<Point> ->
val callback =
Asset.DimensionsReceiver { it?.let { k.resumeWith(Result.success(Point(it))) } }
decodeRawDimensions(null, callback)
}
// TODO b/296288298 Create a util class functions for Bitmap and Asset
private suspend fun Asset.decodeBitmap(dimensions: Point): Bitmap? =
suspendCancellableCoroutine { k: CancellableContinuation<Bitmap?> ->
val callback = Asset.BitmapReceiver { k.resumeWith(Result.success(it)) }
decodeBitmap(dimensions.x, dimensions.y, callback)
}
// TODO b/296288298 Create a util class functions for Bitmap and Asset
private fun Bitmap.extractColors(): WallpaperColors? {
val tmpOut = ByteArrayOutputStream()
var shouldRecycle = false
var cropped = this
if (cropped.compress(Bitmap.CompressFormat.PNG, 100, tmpOut)) {
val outByteArray = tmpOut.toByteArray()
val options = BitmapFactory.Options()
options.inPreferredColorSpace = ColorSpace.get(ColorSpace.Named.SRGB)
cropped = BitmapFactory.decodeByteArray(outByteArray, 0, outByteArray.size)
}
if (cropped.config == Bitmap.Config.HARDWARE) {
cropped = cropped.copy(Bitmap.Config.ARGB_8888, false)
shouldRecycle = true
}
val colors = WallpaperColors.fromBitmap(cropped)
if (shouldRecycle) {
cropped.recycle()
}
return colors
}
}
|