From 303b8b6f818a1d2c9f99ff0fc6cbec4108ee75c3 Mon Sep 17 00:00:00 2001
From: Danny Lin <danny@kdrag0n.dev>
Date: Tue, 05 Oct 2021 21:00:57 -0700
Subject: [PATCH] SystemUI: Use colorkt-based dynamic theme engine for Monet

Dynamic wallpaper-based theming ("Monet") is a prominent aspect of
Android 12's Material You design system. Implement the system-wide
wallpaper-based theming using my colorkt-based [1] dynamic theme engine [2],
which many users prefer over Google's Pixel implementation in empirical
tests as it typically keeps colors closer to the wallpaper and doesn't
suffer from CAM16's blue-purple hue shift issues. It also provides
better perceived contrast than Google's implementation in most cases
because it uses ZCAM's modern lightness correlate as compared to CIELAB
L* [3].

NB: Unlike Google's implementation, we overlay the framework's default
modulated surface colors for performance (no need to modulate colors
when inflating views at runtime) and consistency (reduced dependency on
framework-side color code, which may not respect user preferences).

Demo screenshots: https://twitter.com/kdrag0n/status/1445584174790832134

[1] https://github.com/kdrag0n/colorkt
[2] https://github.com/kdrag0n/android12-extensions
[3] https://twitter.com/MyndexResearch/status/1435491919702888448

[kdrag0n: ported to 12L]

Change-Id: I48cbbacc32b2a8095eb07729aca1734f162d2391

@neobuddy89 - Updated for A13 and squashed with:

    From: SagarMakhar <sagarmakhar@gmail.com>
    Date: Tue, 2 Nov 2021 20:37:54 +0000
    Subject: [PATCH] Allow changing monet settings [1/2]

    From: shivatejapeddi <shivatejapeddi@gmail.com>
    Date: Fri, 5 Nov 2021 00:19:50 +0530
    Subject: SystemUI: Mover Tunable out of constructer to avoid conflict with pixel framework

Signed-off-by: Pranav Vashi <neobuddy89@gmail.com>
Change-Id: I1e7b239ae7afdf43d723eee0d73ba429a229c93b
---

diff --git a/packages/SystemUI/Android.bp b/packages/SystemUI/Android.bp
index f8b72f9..391bb13 100644
--- a/packages/SystemUI/Android.bp
+++ b/packages/SystemUI/Android.bp
@@ -124,6 +124,8 @@
         "lottie",
         "AicpGear-preference",
         "AicpGear-util",
+        "colorkt",
+        "themelib",
     ],
     manifest: "AndroidManifest.xml",
 
diff --git a/packages/SystemUI/src/com/android/systemui/dagger/SystemUICoreStartableModule.kt b/packages/SystemUI/src/com/android/systemui/dagger/SystemUICoreStartableModule.kt
index a9f34085..adf7de8 100644
--- a/packages/SystemUI/src/com/android/systemui/dagger/SystemUICoreStartableModule.kt
+++ b/packages/SystemUI/src/com/android/systemui/dagger/SystemUICoreStartableModule.kt
@@ -36,6 +36,7 @@
 import com.android.systemui.shortcut.ShortcutKeyDispatcher
 import com.android.systemui.statusbar.notification.InstantAppNotifier
 import com.android.systemui.statusbar.phone.KeyguardLiftController
+import com.android.systemui.theme.CustomThemeOverlayController
 import com.android.systemui.theme.ThemeOverlayController
 import com.android.systemui.toast.ToastUI
 import com.android.systemui.usb.StorageNotification
@@ -174,7 +175,7 @@
     @Binds
     @IntoMap
     @ClassKey(ThemeOverlayController::class)
-    abstract fun bindThemeOverlayController(sysui: ThemeOverlayController): CoreStartable
+    abstract fun bindThemeOverlayController(sysui: CustomThemeOverlayController): CoreStartable
 
     /** Inject into ToastUI.  */
     @Binds
@@ -205,4 +206,4 @@
     @IntoMap
     @ClassKey(KeyguardLiftController::class)
     abstract fun bindKeyguardLiftController(sysui: KeyguardLiftController): CoreStartable
-}
\ No newline at end of file
+}
diff --git a/packages/SystemUI/src/com/android/systemui/theme/CustomThemeOverlayController.kt b/packages/SystemUI/src/com/android/systemui/theme/CustomThemeOverlayController.kt
new file mode 100644
index 0000000..903ad57
--- /dev/null
+++ b/packages/SystemUI/src/com/android/systemui/theme/CustomThemeOverlayController.kt
@@ -0,0 +1,233 @@
+/*
+ * Copyright (C) 2021 The Proton AOSP 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.systemui.theme
+
+import android.annotation.ColorInt
+import android.app.WallpaperColors
+import android.app.WallpaperManager
+import android.content.Context
+import android.content.om.FabricatedOverlay
+import android.content.res.Resources
+import android.os.Handler
+import android.os.UserManager
+import android.provider.Settings
+import android.util.Log
+import android.util.TypedValue
+import com.android.systemui.Dependency
+import com.android.systemui.broadcast.BroadcastDispatcher
+import com.android.systemui.dagger.SysUISingleton
+import com.android.systemui.dagger.qualifiers.Background
+import com.android.systemui.dagger.qualifiers.Main
+import com.android.systemui.dump.DumpManager
+import com.android.systemui.flags.FeatureFlags
+import com.android.systemui.keyguard.WakefulnessLifecycle
+import com.android.systemui.monet.Style
+import com.android.systemui.settings.UserTracker
+import com.android.systemui.statusbar.policy.ConfigurationController
+import com.android.systemui.statusbar.policy.DeviceProvisionedController
+import com.android.systemui.theme.ThemeOverlayApplier
+import com.android.systemui.theme.ThemeOverlayController
+import com.android.systemui.tuner.TunerService
+import com.android.systemui.tuner.TunerService.Tunable
+import com.android.systemui.util.settings.SecureSettings
+
+import dev.kdrag0n.colorkt.Color
+import dev.kdrag0n.colorkt.cam.Zcam
+import dev.kdrag0n.colorkt.conversion.ConversionGraph.convert
+import dev.kdrag0n.colorkt.data.Illuminants
+import dev.kdrag0n.colorkt.rgb.Srgb
+import dev.kdrag0n.colorkt.tristimulus.CieXyzAbs.Companion.toAbs
+import dev.kdrag0n.colorkt.ucs.lab.CieLab
+import dev.kdrag0n.monet.theme.DynamicColorScheme
+import dev.kdrag0n.monet.theme.MaterialYouTargets
+import java.util.concurrent.Executor
+import javax.inject.Inject
+import kotlin.math.log10
+import kotlin.math.pow
+
+@SysUISingleton
+class CustomThemeOverlayController @Inject constructor(
+    private val context: Context,
+    broadcastDispatcher: BroadcastDispatcher,
+    @Background bgHandler: Handler,
+    @Main mainExecutor: Executor,
+    @Background bgExecutor: Executor,
+    themeOverlayApplier: ThemeOverlayApplier,
+    secureSettings: SecureSettings,
+    wallpaperManager: WallpaperManager,
+    userManager: UserManager,
+    deviceProvisionedController: DeviceProvisionedController,
+    userTracker: UserTracker,
+    dumpManager: DumpManager,
+    featureFlags: FeatureFlags,
+    @Main resources: Resources,
+    wakefulnessLifecycle: WakefulnessLifecycle,
+    configurationController: ConfigurationController
+) : ThemeOverlayController(
+    context,
+    broadcastDispatcher,
+    bgHandler,
+    mainExecutor,
+    bgExecutor,
+    themeOverlayApplier,
+    secureSettings,
+    wallpaperManager,
+    userManager,
+    deviceProvisionedController,
+    userTracker,
+    dumpManager,
+    featureFlags,
+    resources,
+    wakefulnessLifecycle,
+    configurationController,
+), Tunable {
+    private lateinit var cond: Zcam.ViewingConditions
+    private lateinit var targets: MaterialYouTargets
+
+    private var colorOverride: Int = 0
+    private var chromaFactor: Double = Double.MIN_VALUE
+    private var accurateShades: Boolean = true
+    private var whiteLuminance: Double = Double.MIN_VALUE
+    private var linearLightness: Boolean = false
+    private var customColor: Boolean = false
+    private val mTunerService: TunerService = Dependency.get(TunerService::class.java)
+    override fun start() {
+        mTunerService.addTunable(this, PREF_COLOR_OVERRIDE, PREF_WHITE_LUMINANCE,
+                PREF_CHROMA_FACTOR, PREF_ACCURATE_SHADES, PREF_LINEAR_LIGHTNESS, PREF_CUSTOM_COLOR)
+        super.start()
+    }
+
+    override fun onTuningChanged(key: String?, newValue: String?) {
+        key?.let {
+            if (it.contains(PREF_PREFIX)) {
+                customColor = Settings.Secure.getInt(mContext.contentResolver, PREF_CUSTOM_COLOR, 0) == 1
+                colorOverride = Settings.Secure.getInt(mContext.contentResolver,
+                        PREF_COLOR_OVERRIDE, -1)
+                chromaFactor = (Settings.Secure.getFloat(mContext.contentResolver,
+                        PREF_CHROMA_FACTOR, 100.0f) / 100f).toDouble()
+                accurateShades = Settings.Secure.getInt(mContext.contentResolver, PREF_ACCURATE_SHADES, 1) != 0
+
+                whiteLuminance = parseWhiteLuminanceUser(
+                    Settings.Secure.getInt(mContext.contentResolver,
+                            PREF_WHITE_LUMINANCE, WHITE_LUMINANCE_USER_DEFAULT)
+                )
+                linearLightness = Settings.Secure.getInt(mContext.contentResolver,
+                        PREF_LINEAR_LIGHTNESS, 0) != 0
+                reevaluateSystemTheme(true /* forceReload */)
+            }
+        }
+    }
+
+    // Seed colors
+    override fun getNeutralColor(colors: WallpaperColors) = colors.primaryColor.toArgb()
+    override fun getAccentColor(colors: WallpaperColors) = getNeutralColor(colors)
+
+    override fun getOverlay(primaryColor: Int, type: Int, style: Style): FabricatedOverlay {
+        cond = Zcam.ViewingConditions(
+            surroundFactor = Zcam.ViewingConditions.SURROUND_AVERAGE,
+            // sRGB
+            adaptingLuminance = 0.4 * whiteLuminance,
+            // Gray world
+            backgroundLuminance = CieLab(
+                L = 50.0,
+                a = 0.0,
+                b = 0.0,
+            ).toXyz().y * whiteLuminance,
+            referenceWhite = Illuminants.D65.toAbs(whiteLuminance),
+        )
+
+        targets = MaterialYouTargets(
+            chromaFactor = chromaFactor,
+            useLinearLightness = linearLightness,
+            cond = cond,
+        )
+
+        // Generate color scheme
+        val colorScheme = DynamicColorScheme(
+            targets = targets,
+            seedColor = if (customColor) Srgb(colorOverride) else Srgb(primaryColor),
+            chromaFactor = chromaFactor,
+            cond = cond,
+            accurateShades = accurateShades,
+        )
+
+        val (groupKey, colorsList) = when (type) {
+            ACCENT -> "accent" to colorScheme.accentColors
+            NEUTRAL -> "neutral" to colorScheme.neutralColors
+            else -> error("Unknown type $type")
+        }
+
+        return FabricatedOverlay.Builder(context.packageName, groupKey, "android").run {
+            colorsList.withIndex().forEach { listEntry ->
+                val group = "$groupKey${listEntry.index + 1}"
+
+                listEntry.value.forEach { (shade, color) ->
+                    val colorSrgb = color.convert<Srgb>()
+                    Log.d(TAG, "Color $group $shade = ${colorSrgb.toHex()}")
+                    setColor("system_${group}_$shade", colorSrgb)
+                }
+            }
+
+            // Override special modulated surface colors for performance and consistency
+            if (type == NEUTRAL) {
+                // surface light = neutral1 20 (L* 98)
+                colorsList[0][20]?.let { setColor("surface_light", it) }
+
+                // surface highlight dark = neutral1 650 (L* 35)
+                colorsList[0][650]?.let { setColor("surface_highlight_dark", it) }
+
+                // surface_header_dark_sysui = neutral1 950 (L* 5)
+                colorsList[0][950]?.let { setColor("surface_header_dark_sysui", it) }
+            }
+
+            build()
+        }
+    }
+
+    companion object {
+        private const val TAG = "CustomThemeOverlayController"
+
+        private const val PREF_PREFIX = "monet_engine"
+        private const val PREF_CUSTOM_COLOR = "${PREF_PREFIX}_custom_color"
+        private const val PREF_COLOR_OVERRIDE = "${PREF_PREFIX}_color_override"
+        private const val PREF_CHROMA_FACTOR = "${PREF_PREFIX}_chroma_factor"
+        private const val PREF_ACCURATE_SHADES = "${PREF_PREFIX}_accurate_shades"
+        private const val PREF_LINEAR_LIGHTNESS = "${PREF_PREFIX}_linear_lightness"
+        private const val PREF_WHITE_LUMINANCE = "${PREF_PREFIX}_white_luminance_user"
+
+        private const val WHITE_LUMINANCE_MIN = 1.0
+        private const val WHITE_LUMINANCE_MAX = 10000.0
+        private const val WHITE_LUMINANCE_USER_MAX = 1000
+        private const val WHITE_LUMINANCE_USER_DEFAULT = 425 // ~200.0 divisible by step (decoded = 199.526)
+
+        private fun parseWhiteLuminanceUser(userValue: Int): Double {
+            val userSrc = userValue.toDouble() / WHITE_LUMINANCE_USER_MAX
+            val userInv = 1.0 - userSrc
+            return (10.0).pow(userInv * log10(WHITE_LUMINANCE_MAX))
+                    .coerceAtLeast(WHITE_LUMINANCE_MIN)
+        }
+
+        private fun FabricatedOverlay.Builder.setColor(name: String, @ColorInt color: Int) =
+            setResourceValue("android:color/$name", TypedValue.TYPE_INT_COLOR_ARGB8, color)
+
+        private fun FabricatedOverlay.Builder.setColor(name: String, color: Color): FabricatedOverlay.Builder {
+            val rgb = color.convert<Srgb>().toRgb8()
+            val argb = rgb or (0xff shl 24)
+            return setColor(name, argb)
+        }
+    }
+}