fix: generate cages for killer

kaajjo · Jul 15, 2024 · 9b6c6b0 · 9b6c6b0
1 parent b49cbef
commit 9b6c6b0
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/app/src/main/java/com/kaajjo/libresudoku/ui/home/HomeViewModel.kt b/app/src/main/java/com/kaajjo/libresudoku/ui/home/HomeViewModel.kt
@@ -7,6 +7,7 @@ import androidx.lifecycle.ViewModel
 import androidx.lifecycle.viewModelScope
 import com.kaajjo.libresudoku.core.Cell
 import com.kaajjo.libresudoku.core.PreferencesConstants
+import com.kaajjo.libresudoku.core.qqwing.Cage
 import com.kaajjo.libresudoku.core.qqwing.CageGenerator
 import com.kaajjo.libresudoku.core.qqwing.GameDifficulty
 import com.kaajjo.libresudoku.core.qqwing.GameType
@@ -124,8 +125,16 @@ class HomeViewModel
                     }
                 }
 
-                val generator = CageGenerator(solvedPuzzle, gameTypeToGenerate)
-                val cages = generator.generate(2, 5)
+                var cages: List<Cage>? = null
+                if (gameTypeToGenerate in setOf(
+                        GameType.Killer9x9,
+                        GameType.Killer12x12,
+                        GameType.Killer6x6
+                    )
+                ) {
+                    val generator = CageGenerator(solvedPuzzle, gameTypeToGenerate)
+                    cages = generator.generate(2, 5)
+                }
                 withContext(Dispatchers.IO) {
                     val sudokuParser = SudokuParser()
                     insertedBoardUid = boardRepository.insert(
@@ -135,7 +144,9 @@ class HomeViewModel
                             solvedBoard = sudokuParser.boardToString(solvedPuzzle),
                             difficulty = selectedDifficulty,
                             type = selectedType,
-                            killerCages = SudokuParser().killerSudokuCagesToString(cages)
+                            killerCages = if (cages != null) sudokuParser.killerSudokuCagesToString(
+                                cages
+                            ) else null
                         )
                     )
                 }