def load_stl_geometry(self, stl_filename, dirname, name='main', plot=True):
print("load_stl_geometry...")
skip_reading = self.removeOldGeometry(stl_filename)
if skip_reading:
return
model = STL(log=self.log, debug=False)
#self.model_type = model.model_type
model.read_stl(stl_filename)
nodes = model.nodes
elements = model.elements
normals = model.get_normals(elements)
areas = model.get_area(elements)
#nnormals = model.get_normals_at_nodes(elements)
self.nNodes = nodes.shape[0]
self.nElements = elements.shape[0]
print("nnodes = %s" % self.nNodes)
print("nelements = %s" % self.nElements)
self.grid.Allocate(self.nElements, 1000)
#self.gridResult.SetNumberOfComponents(self.nElements)
points = vtk.vtkPoints()
points.SetNumberOfPoints(self.nNodes)
#self.gridResult.Allocate(self.nNodes, 1000)
#vectorReselt.SetNumberOfComponents(3)
self.nid_map = {}
#elem.SetNumberOfPoints(nNodes)
if 0:
fraction = 1. / self.nNodes # so you can color the nodes by ID
for nid, node in sorted(iteritems(nodes)):
points.InsertPoint(nid - 1, *node)
self.gridResult.InsertNextValue(nid * fraction)
#print str(element)
#elem = vtk.vtkVertex()
#elem.GetPointIds().SetId(0, i)
#self.aQuadGrid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
#vectorResult.InsertTuple3(0, 0.0, 0.0, 1.0)
assert nodes is not None
nnodes = nodes.shape[0]
xmax, ymax, zmax = nodes.max(axis=0)
xmin, ymin, zmin = nodes.min(axis=0)
self.log.info('xmax=%s xmin=%s' % (xmax, xmin))
self.log.info('ymax=%s ymin=%s' % (ymax, ymin))
self.log.info('zmax=%s zmin=%s' % (zmax, zmin))
dim_max = max(xmax-xmin, ymax-ymin, zmax-zmin)
self.create_global_axes(dim_max)
nid = 0
print("nnodes=%s" % nnodes)
for i in range(nnodes):
points.InsertPoint(nid, nodes[i, :])
nid += 1
nelements = elements.shape[0]
#elements -= 1
for eid in range(nelements):
elem = vtkTriangle()
node_ids = elements[eid, :]
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
self.grid.InsertNextCell(5, elem.GetPointIds()) #elem.GetCellType() = 5 # vtkTriangle
self.grid.SetPoints(points)
self.grid.Modified()
if hasattr(self.grid, 'Update'):
self.grid.Update()
print("updated grid")
# loadSTLResults - regions/loads
self. turn_text_on()
self.scalarBar.VisibilityOff()
self.scalarBar.Modified()
cases = {}
self.iSubcaseNameMap = {}
ID = 1
form, cases = self._fill_stl_case(cases, ID, elements, nodes, normals, areas)
self._finish_results_io2(form, cases)
def load_stl_geometry(self, stl_filename, dirname, name='main', plot=True):
#print("load_stl_geometry...")
skip_reading = self._remove_old_geometry(stl_filename)
if skip_reading:
return
model = STL(log=self.log, debug=False)
#self.model_type = model.model_type
model.read_stl(stl_filename)
nodes = model.nodes
elements = model.elements
normals = model.get_normals(elements, stop_on_failure=False)
areas = model.get_area(elements, stop_on_failure=False)
#nnormals = model.get_normals_at_nodes(elements)
self.nNodes = nodes.shape[0]
self.nElements = elements.shape[0]
self.log.info('nnodes=%s nelements=%s' % (self.nNodes, self.nElements))
self.grid.Allocate(self.nElements, 1000)
#self.gridResult.SetNumberOfComponents(self.nElements)
points = vtk.vtkPoints()
points.SetNumberOfPoints(self.nNodes)
#self.gridResult.Allocate(self.nNodes, 1000)
#vectorReselt.SetNumberOfComponents(3)
self.nid_map = {}
#elem.SetNumberOfPoints(nNodes)
if 0:
fraction = 1. / self.nNodes # so you can color the nodes by ID
for nid, node in sorted(iteritems(nodes)):
points.InsertPoint(nid - 1, *node)
self.gridResult.InsertNextValue(nid * fraction)
#print str(element)
#elem = vtk.vtkVertex()
#elem.GetPointIds().SetId(0, i)
#self.aQuadGrid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
#vectorResult.InsertTuple3(0, 0.0, 0.0, 1.0)
assert nodes is not None
nnodes = nodes.shape[0]
xmax, ymax, zmax = nodes.max(axis=0)
xmin, ymin, zmin = nodes.min(axis=0)
self.log.info('xmax=%s xmin=%s' % (xmax, xmin))
self.log.info('ymax=%s ymin=%s' % (ymax, ymin))
self.log.info('zmax=%s zmin=%s' % (zmax, zmin))
dim_max = max(xmax - xmin, ymax - ymin, zmax - zmin)
self.create_global_axes(dim_max)
nid = 0
#print("nnodes=%s" % nnodes)
for i in range(nnodes):
points.InsertPoint(nid, nodes[i, :])
nid += 1
nelements = elements.shape[0]
#elements -= 1
for eid in range(nelements):
elem = vtkTriangle()
node_ids = elements[eid, :]
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
self.grid.InsertNextCell(
5, elem.GetPointIds()) #elem.GetCellType() = 5 # vtkTriangle
self.grid.SetPoints(points)
self.grid.Modified()
if hasattr(self.grid, 'Update'):
self.grid.Update()
self.log_info("updated grid")
# loadSTLResults - regions/loads
self.turn_text_on()
self.scalarBar.VisibilityOff()
self.scalarBar.Modified()
cases = {}
self.iSubcaseNameMap = {}
ID = 1
form, cases = self._fill_stl_case(cases, ID, elements, nodes, normals,
areas)
self._finish_results_io2(form, cases)
