-
Notifications
You must be signed in to change notification settings - Fork 12
/
plants.py
47 lines (34 loc) · 875 Bytes
/
plants.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
# obstaclex, height, radius
csv = ''
with open('sublog.csv','r') as f:
csv = f.read()
csv = csv.split('\n')
csv = [l.strip() for l in csv]
csv = list(filter(lambda x:len(x)>0,csv))
print(len(csv),'lines')
csv = [[float(k) for k in l.split(',')] for l in csv]
lastabsdist = 0.
lastpsoas = -4.
tacos = []
psoas = []
for l in csv:
px = l[1]
lpsoas,rpsoas = l[36],l[37]
if abs(lpsoas-lastpsoas)>1e-8:
lastpsoas = lpsoas
psoas.append([lpsoas,rpsoas])
bdist = l[38]
bheight = l[39]
bradius = l[40]
if bdist == 100:
continue
absdist = bdist+px
if abs(absdist - lastabsdist)>1e-8:
lastabsdist = absdist
tacos.append([absdist,bheight,bradius])
print(tacos,psoas)
with open('rosetta.py','w') as f:
f.write('tacos=')
f.write(str(tacos))
f.write('\npsoas=')
f.write(str(psoas))