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, *args, **kwargs):
super(Simulation,self).__init__(*args, **kwargs)
self._name = self.getMember('name')
self._data = self.getMember(DataObject.getTypeName())
self._robjs = self.getMember(RenderObject.getTypeName())
self.instanced = False
self.script = self.getMember(Scriptable.getTypeName())
def _createDataGUI(self):
pm.text( label='Data' )
pm.attrEnumOptionMenuGrp( l='Type',
at=self.node.name() +
'.'+CRSimulation_Node._dataTypeAttr,
ei=(0, DataObject.getTypeName()))
pm.button(label="Add", w=128,
c=pm.Callback(self.addChildEnumCB, CRDataObject,
self.datasrcs,
srcattr=CRSimulation_Node._dataTypeAttr))
def updateMembers(self):
self.setMember('name', self._name)
self.setMember(DataObject.getTypeName(), self._data)
self.setMember(RenderObject.getTypeName(), self._robjs)
self.setMember(Scriptable.getTypeName(), self.script)
def _initMembersDict(self):
super(Simulation, self)._initMembersDict()
self._members['name'] = [str, 'sim']
self._members[DataObject.getTypeName()] = [DataObject, None]
self._members[RenderObject.getTypeName()] = [RenderObject, []]
self._members[Scriptable.getTypeName()] = [Scriptable, None]
