-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.py
56 lines (40 loc) · 1.33 KB
/
main.py
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
import sys,pprint
from random import randint
from utils import *
from constants import *
#x,y - position in main grid
def DSFractal(main, x, y, grid):
square(grid)
diamond(main, x, y, grid)
minis = subgrid(grid)
half = len(grid) / 2
# Order of recursion
# 1 2
# 3 4
#
if len(minis[0]) > 2 :
insert(grid, main, x, y)
insert(DSFractal(main, x, y, minis[0]), grid, 0, 0)
if len(minis[1]) > 2 :
insert(grid, main, x, y)
insert(DSFractal(main, (half) x, y, minis[1]), grid, half, 0)
if len(minis[2]) > 2 :
insert(grid, main, x, y)
insert(DSFractal(main, x, (half) y, minis[2]), grid, 0, half)
if len(minis[3]) > 2 :
insert(grid, main, x, y)
insert(DSFractal(main, (half) x, (half) y, minis[3]), grid, half, half)
return grid
# Validate size of grid
if (grid_size - 1 & (grid_size - 2)) != 0:
sys.exit("Invalid Number. Must be (2^n) 1 ")
pp = pprint.PrettyPrinter(indent=4)
#init 2d array with 0s
grid = [[0 for x in range(grid_size)] for x in range(grid_size)]
#set corners to half height
grid[0][0] = initial_height
grid[0][len(grid) - 1] = initial_height
grid[len(grid) - 1][0] = initial_height
grid[len(grid) - 1][len(grid) - 1] = initial_height
DSFractal(grid, 0, 0, grid)
display(output_type, grid)