forked from swiftlang/swift
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfractal.swift
166 lines (151 loc) · 9.41 KB
/
fractal.swift
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
// RUN: %empty-directory(%t)
// RUN: %target-build-swift -Xfrontend -disable-implicit-concurrency-module-import -emit-library -o %t/%target-library-name(complex) -emit-module %S/complex.swift -module-link-name complex
// RUN: %target-jit-run %s -I %t -L %t | %FileCheck %s
// RUN: grep -v import %s > %t/main.swift
// RUN: %target-jit-run %t/main.swift %S/complex.swift | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: swift_interpreter
import complex
func printDensity(_ d: Int) {
if (d > 40) {
print(" ", terminator: "")
} else if d > 6 {
print(".", terminator: "")
} else if d > 4 {
print("+", terminator: "")
} else if d > 2 {
print("*", terminator: "")
} else {
print("#", terminator: "")
}
}
extension Double {
func abs() -> Double {
if (self >= 0.0) { return self }
return self * -1.0
}
}
func getMandelbrotIterations(_ c: Complex, maxIterations: Int) -> Int {
var n = 0
var z = Complex()
while (n < maxIterations && z.magnitude() < 4.0) {
z = z*z + c
n += 1
}
return n
}
func fractal (_ densityFunc:(_ c: Complex, _ maxIterations: Int) -> Int,
xMin:Double, xMax:Double,
yMin:Double, yMax:Double,
rows:Int, cols:Int,
maxIterations:Int) {
// Set the spacing for the points in the Mandelbrot set.
var dX = (xMax - xMin) / Double(rows)
var dY = (yMax - yMin) / Double(cols)
// Iterate over the points an determine if they are in the
// Mandelbrot set.
for row in stride(from: xMin, to: xMax, by: dX) {
for col in stride(from: yMin, to: yMax, by: dY) {
var c = Complex(real: col, imag: row)
printDensity(densityFunc(c, maxIterations))
}
print("\n", terminator: "")
}
}
fractal(getMandelbrotIterations,
xMin: -1.35, xMax: 1.4, yMin: -2.0, yMax: 1.05, rows: 40, cols: 80,
maxIterations: 200)
// CHECK: ################################################################################
// CHECK: ##############################********************##############################
// CHECK: ########################********************************########################
// CHECK: ####################***************************+++**********####################
// CHECK: #################****************************++...+++**********#################
// CHECK: ##############*****************************++++......+************##############
// CHECK: ############******************************++++.......+++************############
// CHECK: ##########******************************+++++.... ...++++************##########
// CHECK: ########******************************+++++.... ..++++++**********#########
// CHECK: #######****************************+++++....... .....++++++**********#######
// CHECK: ######*************************+++++....... . .. ............++*********######
// CHECK: #####*********************+++++++++... .. . ... ..++*********#####
// CHECK: ####******************++++++++++++..... ..++**********####
// CHECK: ###***************++++++++++++++... . ...+++**********###
// CHECK: ##**************+++................. ....+***********##
// CHECK: ##***********+++++................. .++***********##
// CHECK: #**********++++++..... ..... ..++***********##
// CHECK: #*********++++++...... . ..++************#
// CHECK: #*******+++++....... ..+++************#
// CHECK: #++++............ ...+++************#
// CHECK: #++++............ ...+++************#
// CHECK: #******+++++........ ..+++************#
// CHECK: #********++++++..... . ..++************#
// CHECK: #**********++++++..... .... .++************#
// CHECK: #************+++++................. ..++***********##
// CHECK: ##*************++++................. . ..+***********##
// CHECK: ###***************+.+++++++++++..... ....++**********###
// CHECK: ###******************+++++++++++++..... ...+++*********####
// CHECK: ####*********************++++++++++.... .. ..++*********#####
// CHECK: #####*************************+++++........ . . . .......+*********######
// CHECK: #######***************************+++.......... .....+++++++*********#######
// CHECK: ########*****************************++++++.... ...++++++**********########
// CHECK: ##########*****************************+++++..... ....++++***********##########
// CHECK: ###########******************************+++++........+++***********############
// CHECK: #############******************************++++.. ...++***********##############
// CHECK: ################****************************+++...+++***********################
// CHECK: ###################***************************+.+++**********###################
// CHECK: #######################**********************************#######################
// CHECK: ############################************************############################
// CHECK: ################################################################################
func getBurningShipIterations(_ c: Complex, maxIterations: Int) -> Int {
var n = 0
var z = Complex()
while (n < maxIterations && z.magnitude() < 4.0) {
var zTmp = Complex(real: z.real.abs(), imag: z.imag.abs())
z = zTmp*zTmp + c
n += 1
}
return n
}
print("\n== BURNING SHIP ==\n\n", terminator: "")
fractal(getBurningShipIterations,
xMin: -2.0, xMax: 1.2, yMin: -2.1, yMax: 1.2, rows: 40, cols: 80,
maxIterations: 200)
// CHECK: ################################################################################
// CHECK: ################################################################################
// CHECK: ################################################################################
// CHECK: #####################################################################*****######
// CHECK: ################################################################*******+...+*###
// CHECK: #############################################################**********+...****#
// CHECK: ###########################################################************. .+****#
// CHECK: #########################################################***********++....+.****
// CHECK: ######################################################************+++......++***
// CHECK: ##############################*******************###************..... .....+++++
// CHECK: ########################*******+++*******************+ .+++++ . . ........+*
// CHECK: ####################**********+.. .+++*******+.+++**+. .....+.+**
// CHECK: #################**********++++...+...++ .. . . .+ ...+++++.***
// CHECK: ##############***********++..... . ... . ...++++++****#
// CHECK: ############*************....... . . ...+++********#
// CHECK: ##########***************. .. ...+++*********#
// CHECK: #########***************++. .. . . ...+++*********##
// CHECK: #######*****************. ... ...+++**********##
// CHECK: ######*****************+. ...+++**********###
// CHECK: #####****************+++ . .....++***********###
// CHECK: #####**********++..... . ....+++***********###
// CHECK: ####*********+++.. . ....+++***********####
// CHECK: ####********++++. ....+++***********####
// CHECK: ###*******++++. ...++++***********####
// CHECK: ###**++*+..+... ...+++************####
// CHECK: ### ...+++************####
// CHECK: ###*********+++++++++......... ...... ..++************####
// CHECK: ####****************++++++.................... .++***********#####
// CHECK: #####********************++++++++++++++++........ .+***********#####
// CHECK: ########****************************+++++++++....... ++***********#####
// CHECK: ###########*******************************++++++...... ..++**********######
// CHECK: ###############*******************************+++++.........++++*********#######
// CHECK: ####################****************************++++++++++++++**********########
// CHECK: ##########################*************************+++++++++***********#########
// CHECK: ################################**************************************##########
// CHECK: ####################################********************************############
// CHECK: ########################################***************************#############
// CHECK: ###########################################**********************###############
// CHECK: #############################################*****************##################
// CHECK: ################################################***********#####################