aboutsummaryrefslogtreecommitdiff
path: root/Week11 Mandelbrot/src/com/camilstaps/mandelbrot/MandelbrotController.java
blob: f072805c4a896c7c6daf2114f62c771a8784ebbb (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
/*
 * Copyright (c) 2015 Camil Staps
 */
package com.camilstaps.mandelbrot;

import fractals.Grid;
import fractals.Mandelbrot;

/**
 *
 * @author camilstaps
 */
public class MandelbrotController {
    
    private final MandelbrotProvider mandelbrotProvider;
    private final Grid grid;
    
    public MandelbrotController(MandelbrotProvider mp, Grid grid) {
        mandelbrotProvider = mp;
        this.grid = grid;
    }
    
    public double getX(int x_on_screen) {
        double centerX = mandelbrotProvider.getCenterX();
        double scale = mandelbrotProvider.getScale();
        int grid_w = grid.getWidth();
        double  min_x = centerX - grid_w / scale / 2,
                max_x = centerX + grid_w / scale / 2;
        
        return min_x + (((float) x_on_screen) / grid_w) * (max_x - min_x);
    }
    
    public double getY(int y_on_screen) {
        double centerY = mandelbrotProvider.getCenterY();
        double scale = mandelbrotProvider.getScale();
        int grid_h = grid.getHeight();
        double  min_y = centerY - grid_h / scale / 2,
                max_y = centerY + grid_h / scale / 2;
        
        return min_y + (((float) y_on_screen) / grid_h) * (max_y - min_y);
    }
    
    public void redraw() {
        int repetitions = mandelbrotProvider.getRepetitions();
        
        int grid_w = grid.getWidth(), grid_h = grid.getHeight();
        
        for (int i = 0; i < grid_w; i++) {
            for (int j = 0; j < grid_h; j++) {
                double mandel = ((double) Mandelbrot.mandelNumber(getX(i), getY(j), repetitions) * Math.PI) / repetitions;
                int[] color = {
                    (int) (255 * Math.sin(mandel)), 
                    (int) (255 * Math.sin(mandel + Math.PI / 3)), 
                    (int) (255 * Math.sin(mandel + 2 * Math.PI / 3))};
                grid.setPixel(i, j, color);
            }               
        }
    }
    
    public interface MandelbrotProvider {
        public double getCenterX();
        public double getCenterY();
        public double getScale();
        public int getRepetitions();
    }
    
}