mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 15:32:29 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
|
import os
|
||
|
|
import networkx as nx
|
||
|
|
from soil.simulation import SoilSimulation
|
||
|
|
|
||
|
|
|
||
|
|
class Simulator():
|
||
|
|
""" Simulator for running simulations. Using SOIL."""
|
||
|
|
|
||
|
|
def __init__(self, dump=False, dir_path='output'):
|
||
|
|
self.name = 'soil'
|
||
|
|
self.dump = dump
|
||
|
|
self.dir_path = dir_path
|
||
|
|
|
||
|
|
def run(self, config):
|
||
|
|
name = config['name']
|
||
|
|
print('Using config(s): {name}'.format(name=name))
|
||
|
|
|
||
|
|
sim = SoilSimulation(**config)
|
||
|
|
sim.dir_path = os.path.join(self.dir_path, name)
|
||
|
|
sim.dump = self.dump
|
||
|
|
|
||
|
|
print('Dumping results to {} : {}'.format(sim.dir_path, sim.dump))
|
||
|
|
|
||
|
|
simulation_results = sim.run_simulation()
|
||
|
|
|
||
|
|
# G = simulation_results[0].history_to_graph()
|
||
|
|
# for node in G.nodes():
|
||
|
|
# if 'pos' in G.node[node]:
|
||
|
|
# G.node[node]['viz'] = {"position": {"x": G.node[node]['pos'][0], "y": G.node[node]['pos'][1], "z": 0.0}}
|
||
|
|
# del (G.node[node]['pos'])
|
||
|
|
# nx.write_gexf(G, 'test.gexf', version='1.2draft')
|
||
|
|
|
||
|
|
return simulation_results
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
pass
|