about summary refs log tree commit diff stats
path: root/apps/mandelbrot.mu
blob: 218fd4fac330d0d1bd2869bfe043483c0999efc5 (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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
# Mandelbrot set
#
# Install:
#   $ git clone https://github.com/akkartik/mu
#   $ cd mu
# Build on Linux:
#   $ ./translate apps/mandelbrot.mu
# Build on other platforms (slow):
#   $ ./translate_emulated apps/mandelbrot.mu
# Run:
#   $ qemu-system-i386 code.img

fn main screen: (addr screen), keyboard: (addr keyboard), data-disk: (addr disk) {
  # Initially the viewport is centered at 0, 0 in the scene.
  var zero: float
  var scene-cx/xmm1: float <- copy zero
  var scene-cy/xmm2: float <- copy zero
  # Initially the viewport shows a section of the scene 4 units wide.
  # scene-width-scale = 0.5
  var scene-width-scale: float
  var dest/eax: (addr float) <- address scene-width-scale
  fill-in-rational dest, 1, 2
  # scene-width = 4
  var four: float
  var dest/eax: (addr float) <- address four
  fill-in-rational dest, 4, 1
  var scene-width/xmm3: float <- copy four
  {
    mandelbrot screen scene-cx, scene-cy, scene-width
    # move the center some % of the current screen-width
    var adj/xmm0: float <- rational 2, 0x1c/28
    adj <- multiply scene-width
    scene-cx <- subtract adj
    scene-cy <- add adj
    # slowly shrink the scene width to zoom in
    scene-width <- multiply scene-width-scale
    loop
  }
}

fn mandelbrot screen: (addr screen), scene-cx: float, scene-cy: float, scene-width: float {
  var a/eax: int <- copy 0
  var b/ecx: int <- copy 0
  a, b <- screen-size screen
  var width/esi: int <- copy a
  width <- shift-left 3/log2-font-width
  var height/edi: int <- copy b
  height <- shift-left 4/log2-font-height
  var y/ecx: int <- copy 0
  {
    compare y, height
    break-if->=
    var imaginary/xmm1: float <- viewport-to-imaginary y, width, height, scene-cy, scene-width
    var x/ebx: int <- copy 0
    {
      compare x, width
      break-if->=
      var real/xmm0: float <- viewport-to-real x, width, scene-cx, scene-width
      var iterations/eax: int <- mandelbrot-iterations-for-point real, imaginary, 0x400/max
      iterations <- shift-right 3
      var color/edx: int <- copy 0
      iterations, color <- integer-divide iterations, 0x18/24/size-of-cycle-0
      color <- add 0x20/cycle-0
      pixel screen, x, y, color
      x <- increment
      loop
    }
    y <- increment
    loop
  }
}

fn mandelbrot-iterations-for-point real: float, imaginary: float, max: int -> _/eax: int {
  var zero: float
  var x/xmm0: float <- copy zero
  var y/xmm1: float <- copy zero
  var iterations/ecx: int <- copy 0
  {
    var done?/eax: boolean <- mandelbrot-done? x, y
    compare done?, 0/false
    break-if-!=
    compare iterations, max
    break-if->=
    var newx/xmm2: float <- mandelbrot-x x, y, real
    var newy/xmm3: float <- mandelbrot-y x, y, imaginary
    x <- copy newx
    y <- copy newy
    iterations <- increment
    loop
  }
  return iterations
}

fn mandelbrot-done? x: float, y: float -> _/eax: boolean {
  # x*x + y*y > 4
  var x2/xmm0: float <- copy x
  x2 <- multiply x
  var y2/xmm1: float <- copy y
  y2 <- multiply y
  var sum/xmm0: float <- copy x2
  sum <- add y2
  var four/eax: int <- copy 4
  var four-f/xmm1: float <- convert four
  compare sum, four-f
  {
    break-if-float>
    return 0/false
  }
  return 1/true
}

fn mandelbrot-x x: float, y: float, real: float -> _/xmm2: float {
  # x*x - y*y + real
  var x2/xmm0: float <- copy x
  x2 <- multiply x
  var y2/xmm1: float <- copy y
  y2 <- multiply y
  var result/xmm0: float <- copy x2
  result <- subtract y2
  result <- add real
  return result
}

fn mandelbrot-y x: float, y: float, imaginary: float -> _/xmm3: float {
  # 2*x*y + imaginary
  var two/eax: int <- copy 2
  var result/xmm0: float <- convert two
  result <- multiply x
  result <- multiply y
  result <- add imaginary
  return result
}

# Scale (x, y) pixel coordinates to a complex plane where the viewport width
# ranges from -2 to +2. Viewport height just follows the viewport's aspect
# ratio.

fn viewport-to-real x: int, width: int, scene-cx: float, scene-width: float -> _/xmm0: float {
  # 0 in the viewport       goes to scene-cx - scene-width/2 
  # width in the viewport   goes to scene-cx + scene-width/2
  # Therefore:
  # x in the viewport       goes to (scene-cx - scene-width/2) + x*scene-width/width
  # At most two numbers being multiplied before a divide, so no risk of overflow.
  var result/xmm0: float <- convert x
  result <- multiply scene-width
  var width-f/xmm1: float <- convert width
  result <- divide width-f
  result <- add scene-cx
  var two/eax: int <- copy 2
  var two-f/xmm2: float <- convert two
  var half-scene-width/xmm1: float <- copy scene-width
  half-scene-width <- divide two-f
  result <- subtract half-scene-width
  return result
}

fn viewport-to-imaginary y: int, width: int, height: int, scene-cy: float, scene-width: float -> _/xmm1: float {
  # 0 in the viewport       goes to scene-cy - scene-width/2*height/width
  # height in the viewport  goes to scene-cy + scene-width/2*height/width
  # Therefore:
  # y in the viewport       goes to (scene-cy - scene-width/2*height/width) + y*scene-width/width
  #  scene-cy - scene-width/width * (height/2 + y)
  # At most two numbers being multiplied before a divide, so no risk of overflow.
  var result/xmm0: float <- convert y
  result <- multiply scene-width
  var width-f/xmm1: float <- convert width
  result <- divide width-f
  result <- add scene-cy
  var two/eax: int <- copy 2
  var two-f/xmm2: float <- convert two
  var second-term/xmm1: float <- copy scene-width
  second-term <- divide two-f
  var height-f/xmm2: float <- convert height
  second-term <- multiply height-f
  var width-f/xmm2: float <- convert width
  second-term <- divide width-f
  result <- subtract second-term
  return result
}