mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 15:32:29 +00:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import os
|
|
import networkx as nx
|
|
from server import VisualizationElement
|
|
from soil.simulation import SoilSimulation
|
|
from xml.etree import ElementTree
|
|
|
|
|
|
class Model():
|
|
|
|
def __init__(self, dump=False, dir_path='output'):
|
|
self.name = 'soil'
|
|
self.dump = dump
|
|
self.dir_path = dir_path
|
|
self.simulation = list()
|
|
|
|
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
|
|
|
|
|
|
class GraphVisualization(VisualizationElement):
|
|
package_includes = []
|
|
|
|
# TODO: esta por definir todos los ajustes de simulacion
|
|
def __init__(self, params=None):
|
|
new_element = ("new funcion()")
|
|
self.js_code = "elements.push(" + new_element + ");"
|
|
|
|
def render(self, model):
|
|
pass
|