def test_simulation(self):
datasrc = CSVDataSource(resource='input/data/stationary/*.dat', delim=',', fields= [["id", "integer"], ["pos_z", "float"], ["pos_y", "float"], ["pos_x", "float"], ["euler_x", "float"], ["euler_y", "float"], ["euler_z", "float"]])
dataobj = DataObject()
dataobj.addDataSource(datasrc)
sph = Sphere()
robj = RenderObject(condition="id > 0", geometry=[sph.getSerialized()])
sim = Simulation()
sim.setData(dataobj)
sim.addRenderObject(robj)
md = MDReaderFactory.build('sim.yml')
md.addElement(Simulation.getTypeName(), sim.getSerialized())
md.writeToDisk()
def __init__(self, factories, typename='', **kwargs):
super(CRSimulation, self).__init__(factories, typename, **kwargs)
self.sim_factories = self.factories.getFactory(Simulation.getTypeName())
self.datasrcs = weakref.WeakValueDictionary()
self.robjs = weakref.WeakValueDictionary()
def export(self, md):
sim = self._constructSim()
md.addElement(Simulation.getTypeName(), sim.getSerialized())
del sim
def _constructSim(self):
attrdict = self.attrs2Dict()
simdict = attrdict[Simulation.getTypeName()]
sim = self.sim_factories.build(Simulation.getTypeName(), **simdict)
return sim
