soil/visualization.py

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=True, 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))
		
		sim.run_simulation()


	def get_trial(self, name, trial):
		graph = nx.read_gexf(os.path.join(self.dir_path, name, '{}_trial_{}.gexf'.format(name, trial)))

		attributes = self.get_attributes(os.path.join(self.dir_path, name, '{}_trial_{}.gexf'.format(name, trial)))
		json = {}
		json['graph'] = nx.node_link_data(graph)
		json['models'] = attributes
		return json

	def reset(self):
		pass

	def get_attributes(self, path_gexf):
		attributes = {}
		tree = ElementTree.parse(path_gexf)
		root = tree.getroot()

		ns = { 'gexf': 'http://www.gexf.net/1.2draft' }

		for mode in root[0].findall('gexf:attributes', ns):
			attributes[mode.attrib['mode']] = []
			for attribute in mode:
				values = {
					'id'	: attribute.attrib['id'],
					'title'	: attribute.attrib['title'],
					'type'	: attribute.attrib['type']
				}
				attributes[mode.attrib['mode']].append(values)

		return attributes


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