Fix category saving

Signed-off-by: enricocid <[email protected]>
enricocid · Feb 24, 2022 · 98f56ec · 98f56ec
1 parent 18d9e76
commit 98f56ec
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/project/app/src/main/java/com/iven/vectorify/ui/MainActivity.kt b/project/app/src/main/java/com/iven/vectorify/ui/MainActivity.kt
@@ -38,6 +38,7 @@ import com.iven.vectorify.ui.PreviewActivity.Companion.TEMP_VECTOR
 import com.iven.vectorify.ui.PreviewActivity.Companion.TEMP_VECTOR_COLOR
 import com.iven.vectorify.ui.PreviewActivity.Companion.TEMP_V_OFFSET
 import com.iven.vectorify.utils.Utils
+import com.iven.vectorify.utils.VectorsCategories
 import com.maxkeppeler.sheets.color.ColorSheet
 import dev.chrisbanes.insetter.Insetter
 import dev.chrisbanes.insetter.windowInsetTypesOf
@@ -264,7 +265,7 @@ class MainActivity : AppCompatActivity(), SharedPreferences.OnSharedPreferenceCh
                         TEMP_BACKGROUND_COLOR to mTempBackgroundColor,
                         TEMP_VECTOR_COLOR to mTempVectorColor,
                         TEMP_VECTOR to mTempVector,
-                        TEMP_CATEGORY to mTempCategory,
+                        TEMP_CATEGORY to VectorsCategories.safeGetVectorCategory(mTempVector),
                         TEMP_SCALE to mTempScale,
                         TEMP_H_OFFSET to mTempHorizontalOffset,
                         TEMP_V_OFFSET to mTempVerticalOffset

diff --git a/project/app/src/main/java/com/iven/vectorify/utils/VectorsCategories.kt b/project/app/src/main/java/com/iven/vectorify/utils/VectorsCategories.kt
@@ -12,6 +12,25 @@ object VectorsCategories {
     //https://iconscout.com/icon/naruto
     //https://github.com/PotatoProject/website/blob/master/src/assets/
 
+    @JvmStatic
+    fun safeGetVectorCategory(vector: Int): Int {
+        val categories = listOf(
+            TECH, SYMBOLS, ANIMALS, EMOTICONS, FUN, FOOD, NATURE,
+            WEATHER, SPORT, MATH, SCIENCE, CHERNOFF, MUSIC, NERDY,
+            BUILDINGS, ALERT, ALPHABET, ROMAN, ZODIAC, OTHERS
+        )
+
+        categories.iterator().withIndex().run {
+            while (hasNext()) {
+                val item = next()
+                if (item.value.contains(vector)) {
+                    return item.index
+                }
+            }
+        }
+        return 0
+    }
+
     @JvmStatic
     val TECH = listOf(
         R.drawable.android,