def load_abaqus_geometry(self, abaqus_filename, name='main', plot=True):
"""loads abaqus input files into the gui"""
model_name = name
skip_reading = self.gui._remove_old_geometry(abaqus_filename)
if skip_reading:
return
self.gui.eid_maps[name] = {}
self.gui.nid_maps[name] = {}
model = Abaqus(log=self.gui.log, debug=False)
self.gui.model_type = 'abaqus'
#self.model_type = model.model_type
model.read_abaqus_inp(abaqus_filename)
self.gui.nid_map = {}
nnodes, all_nodes, nelements = get_nodes_nnodes_nelements(model)
self.gui.log.info('nnodes=%s nelements=%s' % (nnodes, nelements))
assert nelements > 0, nelements
#nodes = model.nodes
#elements = model.elements
self.gui.nnodes = nnodes
self.gui.nelements = nelements
grid = self.gui.grid
grid.Allocate(self.gui.nelements, 1000)
assert len(all_nodes) > 0 is not None, len(all_nodes)
if len(all_nodes) == 1:
nodes = all_nodes[0]
else:
nodes = np.vstack(all_nodes)
mmax = np.amax(nodes, axis=0)
mmin = np.amin(nodes, axis=0)
dim_max = (mmax - mmin).max()
self.gui.create_global_axes(dim_max)
points = vtk.vtkPoints()
points.SetNumberOfPoints(self.gui.nnodes)
data_type = vtk.VTK_FLOAT
points_array = numpy_to_vtk(
num_array=nodes,
deep=True,
array_type=data_type
)
points.SetData(points_array)
nid_offset = -1
nids = []
for unused_part_name, part in model.parts.items():
self.gui.log.info('part_name = %r' % unused_part_name)
nnodesi = part.nodes.shape[0]
nidsi = part.nids
nids.append(nidsi)
add_lines(grid, nidsi, part.r2d2, nid_offset)
add_tris(grid, nidsi, part.cps3, nid_offset)
add_tris(grid, nidsi, part.cpe3, nid_offset)
add_quads(grid, nidsi, part.cpe4, nid_offset)
add_quads(grid, nidsi, part.cpe4r, nid_offset)
add_quads(grid, nidsi, part.cps4, nid_offset)
add_quads(grid, nidsi, part.cps4r, nid_offset)
add_quads(grid, nidsi, part.coh2d4, nid_offset)
add_quads(grid, nidsi, part.cohax4, nid_offset)
add_tris(grid, nidsi, part.cax3, nid_offset)
#add_quads(grid, nidsi, part.cax4, nid_offset)
add_quads(grid, nidsi, part.cax4r, nid_offset)
# solids
add_tetras(grid, nidsi, part.c3d10h, nid_offset)
add_hexas(grid, nidsi, part.c3d8r, nid_offset)
nid_offset += nnodesi
nids = np.hstack(nids)
grid.SetPoints(points)
grid.Modified()
# loadCart3dResults - regions/loads
self.gui.scalar_bar_actor.VisibilityOn()
self.gui.scalar_bar_actor.Modified()
note = ''
self.gui.isubcase_name_map = {1: ['Abaqus%s' % note, '']}
#form = []
cases = OrderedDict()
ID = 1
form, cases, unused_icase, node_ids, element_ids = self._fill_abaqus_case(
cases, ID, nids, nodes, nelements, model)
#self._fill_cart3d_results(cases, form, icase, ID, model)
self.gui.node_ids = node_ids
self.gui.element_ids = element_ids
self.gui._finish_results_io2(model_name, form, cases)
def load_abaqus_geometry(self, abaqus_filename, name='main', plot=True):
"""loads abaqus input files into the gui"""
skip_reading = self._remove_old_geometry(abaqus_filename)
if skip_reading:
return
self.eid_maps[name] = {}
self.nid_maps[name] = {}
model = Abaqus(log=self.log, debug=False)
self.model_type = 'abaqus'
#self.model_type = model.model_type
model.read_abaqus_inp(abaqus_filename)
n_r2d2 = 0
n_cpe3 = 0
n_cpe4 = 0
n_cpe4r = 0
n_coh2d4 = 0
n_c3d10h = 0
n_cohax4 = 0
n_cax3 = 0
n_cax4r = 0
nnodes = 0
nelements = 0
all_nodes = []
for part_name, part in iteritems(model.parts):
nids = part.nids - 1
nodes = part.nodes
nnodes += nodes.shape[0]
if part.r2d2 is not None:
n_r2d2 += part.r2d2.shape[0]
if part.cpe3 is not None:
n_cpe3 += part.cpe3.shape[0]
if part.cpe4 is not None:
n_cpe4 += part.cpe4.shape[0]
if part.cpe4r is not None:
n_cpe4r += part.cpe4r.shape[0]
if part.coh2d4 is not None:
n_coh2d4 += part.coh2d4.shape[0]
if part.cohax4 is not None:
n_cohax4 += part.cohax4.shape[0]
if part.cax3 is not None:
n_cax3 += part.cax3.shape[0]
if part.cax4r is not None:
n_cax4r += part.cax4r.shape[0]
if part.c3d10h is not None:
n_c3d10h += part.c3d10h.shape[0]
all_nodes.append(nodes)
nelements += n_r2d2 + n_cpe3 + n_cpe4 + n_cpe4r + n_coh2d4 + n_c3d10h + n_cohax4 + n_cax3 + n_cax4r
assert nelements > 0, nelements
#nodes = model.nodes
#elements = model.elements
self.nNodes = nnodes
self.nElements = nelements
self.grid.Allocate(self.nElements, 1000)
points = vtk.vtkPoints()
points.SetNumberOfPoints(self.nNodes)
self.nid_map = {}
assert nodes is not None
nnodes = nodes.shape[0]
if len(all_nodes) == 1:
nodes = all_nodes[0]
else:
nodes = np.vstack(all_nodes)
mmax = np.amax(nodes, axis=0)
mmin = np.amin(nodes, axis=0)
dim_max = (mmax - mmin).max()
self.create_global_axes(dim_max)
data_type = vtk.VTK_FLOAT
points_array = numpy_to_vtk(num_array=nodes,
deep=True,
array_type=data_type)
points.SetData(points_array)
grid = self.grid
nid_offset = -1
for part_name, part in iteritems(model.parts):
nnodesi = part.nodes.shape[0]
n_r2d2 = 0
n_cpe3 = 0
n_cpe4 = 0
n_cpe4r = 0
n_coh2d4 = 0
n_c3d10h = 0
n_cohax4 = 0
n_cax3 = 0
n_cax4r = 0
if part.r2d2 is not None:
n_r2d2 += part.r2d2.shape[0]
if part.cpe3 is not None:
n_cpe3 += part.cpe3.shape[0]
if part.cpe4 is not None:
n_cpe4 += part.cpe4.shape[0]
if part.cpe4r is not None:
n_cpe4r += part.cpe4r.shape[0]
if part.coh2d4 is not None:
n_coh2d4 += part.coh2d4.shape[0]
if part.cohax4 is not None:
n_cohax4 += part.cohax4.shape[0]
if part.cax3 is not None:
n_cax3 += part.cax3.shape[0]
if part.cax4r is not None:
n_cax4r += part.cax4r.shape[0]
# solids
if part.c3d10h is not None:
n_c3d10h += part.c3d10h.shape[0]
if n_r2d2:
eids = part.r2d2[:, 0]
node_ids = part.r2d2[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkLine()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
if n_cpe3:
eids = part.cpe3[:, 0]
node_ids = part.cpe3[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkTriangle()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
grid.InsertNextCell(5, elem.GetPointIds())
if n_cpe4:
eids = part.cpe4[:, 0]
node_ids = part.cpe4[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkQuad()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
if n_cpe4r:
eids = part.cpe4r[:, 0]
node_ids = part.cpe4r[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkQuad()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
if n_coh2d4:
eids = part.coh2d4[:, 0]
node_ids = part.coh2d4[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkQuad()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
if n_cohax4:
eids = part.cohax4[:, 0]
node_ids = part.cohax4[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkQuad()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
if n_cax3:
eids = part.cax3[:, 0]
node_ids = part.cax3[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkTriangle()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
grid.InsertNextCell(5, elem.GetPointIds())
if n_cax4r:
eids = part.cax4r[:, 0]
node_ids = part.cax4r[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
elem = vtkQuad()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
# solids
if n_c3d10h:
eids = part.c3d10h[:, 0]
node_ids = part.c3d10h[:, 1:] + nid_offset
for eid, node_ids in zip(eids, node_ids):
#for eid, node_ids in part.c3d10h:
elem = vtkTetra()
elem.GetPointIds().SetId(0, node_ids[0])
elem.GetPointIds().SetId(1, node_ids[1])
elem.GetPointIds().SetId(2, node_ids[2])
elem.GetPointIds().SetId(3, node_ids[3])
grid.InsertNextCell(elem.GetCellType(), elem.GetPointIds())
nid_offset += nnodesi
grid.SetPoints(points)
grid.Modified()
if hasattr(grid, 'Update'):
grid.Update()
# loadCart3dResults - regions/loads
self.scalarBar.VisibilityOn()
self.scalarBar.Modified()
note = ''
self.iSubcaseNameMap = {1: ['Abaqus%s' % note, '']}
#form = []
cases = {}
ID = 1
form, cases, icase = self._fill_abaqus_case(cases, ID, nodes,
nelements, model)
#self._fill_cart3d_results(cases, form, icase, ID, model)
self._finish_results_io2(form, cases)