[feat] add normal and hard data

shilokuma-inc · May 15, 2024 · f1a1791 · f1a1791
1 parent ded2c9e
commit f1a1791
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/PrimePickApp/QuizData/PrimeData.swift b/PrimePickApp/QuizData/PrimeData.swift
@@ -23,6 +23,32 @@ final class PrimeData {
         return twoDigitPrimes
     }
 
+    public func generateThreeDigitPrimes() -> [Int] {
+        var twoDigitPrimes = [Int]()
+        outerLoop: for num in 100..<1000 {
+            for i in 2..<num {
+                if num % i == 0 {
+                    continue outerLoop
+                }
+            }
+            twoDigitPrimes.append(num)
+        }
+        return twoDigitPrimes
+    }
+
+    public func generateFourDigitPrimes() -> [Int] {
+        var twoDigitPrimes = [Int]()
+        outerLoop: for num in 1000..<10000 {
+            for i in 2..<num {
+                if num % i == 0 {
+                    continue outerLoop
+                }
+            }
+            twoDigitPrimes.append(num)
+        }
+        return twoDigitPrimes
+    }
+
     public func generateThreeOrFourDigitPrimes() -> [Int] {
         var twoDigitPrimes = [Int]()
         outerLoop: for num in 100..<10000 {

diff --git a/PrimePickApp/QuizView.swift b/PrimePickApp/QuizView.swift
@@ -14,7 +14,15 @@ struct QuizView: View {
 
     init(difficulty: String) {
         self.difficulty = difficulty
-        primes = primeData.generateOneOrTwoDigitPrimes()
+        if self.difficulty == "Easy" {
+            primes = primeData.generateOneOrTwoDigitPrimes()
+        } else if self.difficulty == "Normal" {
+            primes = primeData.generateThreeDigitPrimes()
+        } else if self.difficulty == "Hard" {
+            primes = primeData.generateFourDigitPrimes()
+        } else {
+            primes = []
+        }
     }
 
     var body: some View {