From 02693b33bfad0c9e761008db1185ab62a6cb44c8 Mon Sep 17 00:00:00 2001
From: Naveen <snaveen935@gmail.com>
Date: Wed, 12 Oct 2022 01:53:34 +0530
Subject: [PATCH] Use background color as state instead of using on/off
 drawables

---
 .../draw/pro/activities/MainActivity.kt       | 57 ++++++++++++-------
 1 file changed, 36 insertions(+), 21 deletions(-)

diff --git a/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt b/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
index ed66041..c38c6f9 100644
--- a/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
@@ -137,6 +137,7 @@ class MainActivity : SimpleActivity(), CanvasListener {
 
         requestedOrientation = if (config.forcePortraitMode) ActivityInfo.SCREEN_ORIENTATION_PORTRAIT else ActivityInfo.SCREEN_ORIENTATION_UNSPECIFIED
         refreshMenuItems()
+        updateButtonStates()
     }
 
     override fun onPause() {
@@ -348,6 +349,8 @@ class MainActivity : SimpleActivity(), CanvasListener {
     private fun eraserClicked() {
         if (isEyeDropperOn) {
             eyeDropperClicked()
+        } else if (isBucketFillOn) {
+            bucketFillClicked()
         }
 
         isEraserOn = !isEraserOn
@@ -355,13 +358,7 @@ class MainActivity : SimpleActivity(), CanvasListener {
     }
 
     private fun updateEraserState() {
-        val iconId = if (isEraserOn) {
-            R.drawable.ic_eraser_off_vector
-        } else {
-            R.drawable.ic_eraser_on_vector
-        }
-
-        eraser.setImageResource(iconId)
+        updateButtonStates()
         my_canvas.toggleEraser(isEraserOn)
     }
 
@@ -382,6 +379,8 @@ class MainActivity : SimpleActivity(), CanvasListener {
     private fun eyeDropperClicked() {
         if (isEraserOn) {
             eraserClicked()
+        } else if (isBucketFillOn) {
+            bucketFillClicked()
         }
 
         isEyeDropperOn = !isEyeDropperOn
@@ -391,28 +390,44 @@ class MainActivity : SimpleActivity(), CanvasListener {
             eyeDropper.stop()
         }
 
-        val iconId = if (isEyeDropperOn) {
-            R.drawable.ic_colorize_off_vector
-        } else {
-            R.drawable.ic_colorize_on_vector
-        }
-
-        eye_dropper.setImageResource(iconId)
+        updateButtonStates()
     }
 
     private fun bucketFillClicked() {
-        isBucketFillOn = !isBucketFillOn
-
-        val iconId = if (isBucketFillOn) {
-            R.drawable.ic_bucket_fill_off_vector
-        } else {
-            R.drawable.ic_bucket_fill_on_vector
+        if (isEraserOn) {
+            eraserClicked()
+        } else if (isEyeDropperOn) {
+            eyeDropperClicked()
         }
 
-        bucket_fill.setImageResource(iconId)
+        isBucketFillOn = !isBucketFillOn
+
+        updateButtonStates()
         my_canvas.toggleBucketFill(isBucketFillOn)
     }
 
+    private fun updateButtonStates() {
+        arrayOf(
+            eraser to isEraserOn,
+            eye_dropper to isEyeDropperOn,
+            bucket_fill to isBucketFillOn
+        ).forEach {
+            val view = it.first
+            val enabled = it.second
+
+            view.apply {
+                if (enabled) {
+                    val primaryColor = getProperPrimaryColor()
+                    background = ColorDrawable(primaryColor).apply { alpha = 220 }
+                    applyColorFilter(primaryColor.getContrastColor())
+                } else {
+                    background = null
+                    applyColorFilter(config.canvasBackgroundColor.getContrastColor())
+                }
+            }
+        }
+    }
+
     private fun confirmImage() {
         when {
             isEditIntent -> {