/
viz_update.py~
36 lines (30 loc) · 991 Bytes
/
viz_update.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
#!/usr/bin/env python
from tempfile import mkstemp
from shutil import move
from os import remove, close, environ
from xml.dom import minidom
sim_path = environ.get('PLUMESIM')
xmldoc = minidom.parse(sim_path+'/sim_properties.xml')
itemlist = xmldoc.getElementsByTagName('grid_x')
grid_x=float(itemlist[0].attributes['val'].value)
itemlist = xmldoc.getElementsByTagName('grid_y')
grid_y=float(itemlist[0].attributes['val'].value)
itemlist = xmldoc.getElementsByTagName('grid_z')
grid_z=float(itemlist[0].attributes['val'].value)
value = min([grid_x,grid_y,grid_z])
file_path = sim_path+"/rviz/marineplumesim.rviz"
pattern = ' Size (m):'
subst = ' Size (m): '+str(value)+'\n'
fh, abs_path = mkstemp()
new_file = open(abs_path,'w')
old_file = open(file_path)
for line in old_file:
new_file.write(subst if pattern in line else line)
#close temp file
new_file.close()
close(fh)
old_file.close()
#Remove original file
remove(file_path)
#Move new file
move(abs_path, file_path)