1 files changed, 36 insertions, 0 deletions
diff --git a/Week15 Mandelbrot/src/mandelbrot/ColorChooser.java b/Week15 Mandelbrot/src/mandelbrot/ColorChooser.java
new file mode 100644
index 0000000..c3aa441
--- /dev/null
+++ b/Week15 Mandelbrot/src/mandelbrot/ColorChooser.java
@@ -0,0 +1,36 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+package mandelbrot;
+
+/**
+ *
+ * @author Sjaak
+ */
+public class ColorChooser {
+    private int maxIndex;
+    
+    public ColorChooser () {
+        maxIndex = 0;
+    }
+    
+    public void setMaxIndex (int max_index) {
+        maxIndex = max_index;
+    }
+    
+    public int getColorIndex (double x0, double y0) {
+        double x = x0, y = y0;
+        int color_index = 0;
+        while (x * x + y * y < 4.0) {
+            double nx = x * x - y * y + x0;
+            y = 2 * x * y + y0;
+            x = nx;
+            color_index++;
+            if (color_index == maxIndex) {
+                return -1;
+            }
+        }
+        return color_index;
+    }
+}