-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgenerator.py
55 lines (41 loc) · 1.3 KB
/
generator.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
import math
x_cells = 100
y_cells = 100
prefix = 'input/'
def gen_height():
f = open(prefix + 'height.txt', 'w')
f.write(str(x_cells) + ' ' + str(y_cells) + '\n')
for i in range(x_cells):
for j in range(y_cells):
#f.write('1.5 ')
f.write('3.0 ' if (i < 20 and j < 20) else '1.0 ')
#f.write('2.0 ' if (i*i*0.25 < j) else '1.0 ')
#f.write(str(math.exp(-((i-x_cells/2)**2 + (j - y_cells/2)**2)/100) + 1.0) + ' ')
#f.write('3.0 ' if ((i - j)**2 <= 16) else (str(abs(math.sqrt(i*i + 100)/100 - math.sqrt(j))) + ' '));
f.write('\n')
def gen_velocity_x():
f = open(prefix + 'velocity_x.txt', 'w')
f.write(str(x_cells + 1) + ' ' + str(y_cells) + '\n')
for i in range(x_cells + 1):
for j in range(y_cells):
f.write('0.0 ')
f.write('\n')
def gen_velocity_y():
f = open(prefix + 'velocity_y.txt', 'w')
f.write(str(x_cells) + ' ' + str(y_cells + 1) + '\n')
for i in range(x_cells):
for j in range(y_cells + 1):
f.write('0.0 ')
f.write('\n')
def gen_surface():
f = open(prefix + 'surface.txt', 'w')
f.write(str(x_cells) + ' ' + str(y_cells) + '\n')
for i in range(x_cells):
for j in range(y_cells):
#f.write('0.0 ')
f.write(str(math.exp(-((i-x_cells/2)**2 + (j-y_cells/2)**2)/100)) + ' ')
f.write('\n')
gen_height()
gen_velocity_x()
gen_velocity_y()
gen_surface()