def element_plot(self):
try:
step = self.listboxes['steps'].currentText()
field = self.listboxes['fields_element'].currentText()
if field != 'axes':
iptype = self.listboxes['iptype'].currentText()
nodal = self.listboxes['nodal'].currentText()
cbar = [None, None]
data = self.structure.results[step]['element'][field]
result = postprocess(self.nodes, self.elements, self.ux,
self.uy, self.uz, data, 'element', 1,
cbar, 255, iptype, nodal)
toc, _, cnodes, *_ = result
self.update_vertices_colors(
{i: j
for i, j in enumerate(cnodes)})
self.update_statusbar('Plotting: {0:.3f} s'.format(toc))
except:
pass
def nodal_plot(self):
try:
step = self.listboxes['steps'].currentText()
field = self.listboxes['fields_nodal'].currentText()
cbar = [None, None]
data = [self.structure.results[step]['nodal']['{0}{1}'.format(field, self.mode)][i] for i in self.nkeys]
result = postprocess(self.nodes, self.elements, self.ux, self.uy, self.uz, data, 'nodal', 1, cbar,
255, None, None)
toc, _, cnodes, *_ = result
self.update_vertices_colors({i: j for i, j in enumerate(cnodes)})
self.update_statusbar('Plotting: {0:.3f} s'.format(toc))
except:
pass
def plot_voxels(structure,
step,
field='smises',
cbar=[None, None],
iptype='mean',
nodal='mean',
vdx=None,
mode=''):
"""
Voxel 4D visualisation.
Parameters
----------
structure : obj
Structure object.
step : str
Name of the Step.
field : str
Field to plot, e.g. 'smises'.
cbar : list
Minimum and maximum limits on the colorbar.
iptype : str
'mean', 'max' or 'min' of an element's integration point data.
nodal : str
'mean', 'max' or 'min' for nodal values.
vdx : float
Voxel spacing.
mode : int
mode or frequency number to plot, in case of modal, harmonic or buckling analysis.
Returns
-------
None
"""
# Node and element data
xyz = structure.nodes_xyz()
elements = [
structure.elements[i].nodes
for i in sorted(structure.elements, key=int)
]
nodal_data = structure.results[step]['nodal']
nkeys = sorted(structure.nodes, key=int)
ux = [nodal_data['ux{0}'.format(mode)][i] for i in nkeys]
uy = [nodal_data['uy{0}'.format(mode)][i] for i in nkeys]
uz = [nodal_data['uz{0}'.format(mode)][i] for i in nkeys]
try:
data = [nodal_data[field + str(mode)][key] for key in nkeys]
dtype = 'nodal'
except (Exception):
data = structure.results[step]['element'][field]
dtype = 'element'
# Postprocess
result = postprocess(xyz, elements, ux, uy, uz, data, dtype, 1, cbar, 1,
iptype, nodal)
try:
toc, U, cnodes, fabs, fscaled, celements, eabs = result
U = array(U)
print('\n***** Data processed : {0:.3f} s *****'.format(toc))
except:
print('\n***** Error post-processing *****')
try:
plotvoxels(values=fscaled, U=U, vdx=vdx)
except:
print('\n***** Error plotting voxels *****')
def plot_data(structure,
step,
field='um',
layer=None,
scale=1.0,
radius=0.05,
cbar=[None, None],
iptype='mean',
nodal='mean',
mode='',
cbar_size=1):
"""
Plots analysis results on the deformed shape of the Structure.
Parameters
----------
structure : obj
Structure object.
step : str
Name of the Step.
field : str
Field to plot, e.g. 'um', 'sxx', 'sm1'.
layer : str
Layer name for plotting.
scale : float
Scale on displacements for the deformed plot.
radius : float
Radius of the pipe visualisation meshes.
cbar : list
Minimum and maximum limits on the colorbar.
iptype : str
'mean', 'max' or 'min' of an element's integration point data.
nodal : str
'mean', 'max' or 'min' for nodal values.
mode : int
Mode or frequency number to plot, for modal, harmonic or buckling analysis.
cbar_size : float
Scale on the size of the colorbar.
Returns
-------
None
Notes
-----
- Pipe visualisation of line elements is not based on the element section.
"""
if field in ['smaxp', 'smises']:
nodal = 'max'
iptype = 'max'
elif field in ['sminp']:
nodal = 'min'
iptype = 'min'
# Create and clear Blender layer
if not layer:
layer = '{0}-{1}{2}'.format(step, field, mode)
try:
clear_layer(layer)
except:
create_layer(layer)
# Node and element data
nodes = structure.nodes_xyz()
elements = [
structure.elements[i].nodes
for i in sorted(structure.elements, key=int)
]
nodal_data = structure.results[step]['nodal']
nkeys = sorted(structure.nodes, key=int)
ux = [nodal_data['ux{0}'.format(mode)][i] for i in nkeys]
uy = [nodal_data['uy{0}'.format(mode)][i] for i in nkeys]
uz = [nodal_data['uz{0}'.format(mode)][i] for i in nkeys]
try:
data = [nodal_data['{0}{1}'.format(field, mode)][i] for i in nkeys]
dtype = 'nodal'
except (Exception):
data = structure.results[step]['element'][field]
dtype = 'element'
# Postprocess
result = postprocess(nodes, elements, ux, uy, uz, data, dtype, scale, cbar,
1, iptype, nodal)
try:
toc, U, cnodes, fabs, fscaled, celements, eabs = result
U = array(U)
print('\n***** Data processed : {0} s *****'.format(toc))
except:
print(
'\n***** Error encountered during data processing or plotting *****'
)
# Plot meshes
npts = 8
mesh_faces = []
block_faces = [[0, 1, 2, 3], [4, 5, 6, 7], [0, 1, 5, 4], [1, 2, 6, 5],
[2, 3, 7, 6], [3, 0, 4, 7]]
tet_faces = [[0, 2, 1], [1, 2, 3], [1, 3, 0], [0, 3, 2]]
pipes = []
mesh_add = []
for element, nodes in enumerate(elements):
n = len(nodes)
if n == 2:
u, v = nodes
pipe = draw_cylinder(start=U[u],
end=U[v],
radius=radius,
div=npts,
layer=layer)
pipes.append(pipe)
if dtype == 'element':
col1 = col2 = celements[element]
elif dtype == 'nodal':
col1 = cnodes[u]
col2 = cnodes[v]
try:
blendermesh = BlenderMesh(object=pipe)
blendermesh.set_vertices_colors(
{i: col1
for i in range(0, 2 * npts, 2)})
blendermesh.set_vertices_colors(
{i: col2
for i in range(1, 2 * npts, 2)})
except:
pass
elif n in [3, 4]:
if structure.elements[element].__name__ in [
'ShellElement', 'MembraneElement'
]:
mesh_faces.append(nodes)
else:
for face in tet_faces:
mesh_faces.append([nodes[i] for i in face])
elif n == 8:
for block in block_faces:
mesh_faces.append([nodes[i] for i in block])
if mesh_faces:
bmesh = xdraw_mesh(name='bmesh',
vertices=U,
faces=mesh_faces,
layer=layer)
blendermesh = BlenderMesh(bmesh)
blendermesh.set_vertices_colors(
{i: col
for i, col in enumerate(cnodes)})
mesh_add = [bmesh]
# Plot colourbar
xr, yr, _ = structure.node_bounds()
yran = yr[1] - yr[0] if yr[1] - yr[0] else 1
s = yran * 0.1 * cbar_size
xmin = xr[1] + 3 * s
ymin = yr[0]
cmesh = draw_plane(name='colorbar',
Lx=s,
dx=s,
Ly=10 * s,
dy=s,
layer=layer)
set_objects_coordinates(objects=[cmesh], coords=[[xmin, ymin, 0]])
blendermesh = BlenderMesh(object=cmesh)
vertices = blendermesh.get_vertices_coordinates().values()
y = array(list(vertices))[:, 1]
yn = yran * cbar_size
colors = colorbar(((y - ymin - 0.5 * yn) * 2 / yn)[:, newaxis],
input='array',
type=1)
blendermesh.set_vertices_colors(
{i: j
for i, j in zip(range(len(vertices)), colors)})
set_deselect()
set_select(objects=pipes + mesh_add + [cmesh])
bpy.context.view_layer.objects.active = cmesh
bpy.ops.object.join()
h = 0.6 * s
for i in range(5):
x0 = xmin + 1.2 * s
yu = ymin + (5.8 + i) * s
yl = ymin + (3.8 - i) * s
vu = +max([eabs, fabs]) * (i + 1) / 5.
vl = -max([eabs, fabs]) * (i + 1) / 5.
draw_text(radius=h,
pos=[x0, yu, 0],
text='{0:.3g}'.format(vu),
layer=layer)
draw_text(radius=h,
pos=[x0, yl, 0],
text='{0:.3g}'.format(vl),
layer=layer)
draw_text(radius=h, pos=[x0, ymin + 4.8 * s, 0], text='0', layer=layer)
draw_text(radius=h,
pos=[xmin, ymin + 12 * s, 0],
text='Step:{0} Field:{1}'.format(step, field),
layer=layer)
def plot_data(structure,
step,
field,
layer,
scale=1.0,
radius=0.05,
cbar=[None, None],
iptype='mean',
nodal='mean',
mode='',
colorbar_size=1):
""" Plots analysis results on the deformed shape of the Structure.
Parameters
----------
structure : obj
Structure object.
step : str
Name of the Step.
field : str
Field to plot, e.g. 'um', 'sxx', 'sm1'.
layer : int
Layer number for plotting.
scale : float
Scale on displacements for the deformed plot.
radius : float
Radius of the pipe visualisation meshes.
cbar : list
Minimum and maximum limits on the colorbar.
iptype : str
'mean', 'max' or 'min' of an element's integration point data.
nodal : str
'mean', 'max' or 'min' for nodal values.
mode : int
Mode or frequency number to plot, for modal, harmonic or buckling analysis.
colorbar_size : float
Scale on the size of the colorbar.
Returns
-------
None
Notes
-----
- Pipe visualisation of line elements is not based on the element section.
"""
clear_layer(layer=layer)
# Node and element data
nodes = structure.nodes_xyz()
elements = [
structure.elements[i].nodes
for i in sorted(structure.elements, key=int)
]
nodal_data = structure.results[step]['nodal']
nkeys = sorted(structure.nodes, key=int)
ux = [nodal_data['ux{0}'.format(mode)][i] for i in nkeys]
uy = [nodal_data['uy{0}'.format(mode)][i] for i in nkeys]
uz = [nodal_data['uz{0}'.format(mode)][i] for i in nkeys]
try:
data = [nodal_data['{0}{1}'.format(field, mode)][i] for i in nkeys]
dtype = 'nodal'
except (Exception):
data = structure.results[step]['element'][field]
dtype = 'element'
# Postprocess
result = postprocess(nodes, elements, ux, uy, uz, data, dtype, scale, cbar,
1, iptype, nodal)
try:
toc, U, cnodes, fabs, fscaled, celements, eabs = result
U = array(U)
print('\n***** Data processed : {0:.3f} s *****'.format(toc))
except:
print(
'\n***** Error encountered during data processing or plotting *****'
)
# Plot meshes
npts = 8
mesh_faces = []
for element, nodes in enumerate(elements):
n = len(nodes)
if n == 2:
u, v = nodes
pipe = draw_pipes(start=[U[u]],
end=[U[v]],
radius=radius,
layer=layer)[0]
if dtype == 'element':
col1 = col2 = [celements[element]] * npts
elif dtype == 'nodal':
col1 = [cnodes[u]] * npts
col2 = [cnodes[v]] * npts
blendermesh = BlenderMesh(pipe)
blendermesh.set_vertex_colors(vertices=range(0, 2 * npts, 2),
colors=col1)
blendermesh.set_vertex_colors(vertices=range(1, 2 * npts, 2),
colors=col2)
elif n in [3, 4]:
mesh_faces.append(nodes)
if mesh_faces:
bmesh = xdraw_mesh(name='bmesh',
vertices=U,
faces=mesh_faces,
layer=layer)
blendermesh = BlenderMesh(bmesh)
blendermesh.set_vertex_colors(vertices=range(U.shape[0]),
colors=cnodes)
# Plot colourbar
xr, yr, _ = structure.node_bounds()
yran = yr[1] - yr[0] if yr[1] - yr[0] else 1
s = yran * 0.1 * colorbar_size
xmin = xr[1] + 3 * s
ymin = yr[0]
cmesh = draw_plane(name='colorbar',
Lx=s,
dx=s,
Ly=10 * s,
dy=s,
layer=layer)
set_object_location(object=cmesh, location=[xmin, ymin, 0])
blendermesh = BlenderMesh(cmesh)
verts = blendermesh.get_vertex_coordinates()
y = array(verts)[:, 1]
yn = yran * colorbar_size
colors = colorbar(((y - ymin - 0.5 * yn) * 2 / yn)[:, newaxis],
input='array',
type=1)
blendermesh.set_vertex_colors(vertices=range(len(verts)), colors=colors)
h = 0.6 * s
texts = []
for i in range(5):
x0 = xmin + 1.2 * s
yu = ymin + (5.8 + i) * s
yl = ymin + (3.8 - i) * s
vu = float(+max(eabs, fabs) * (i + 1) / 5.)
vl = float(-max(eabs, fabs) * (i + 1) / 5.)
texts.extend([{
'radius': h,
'pos': [x0, yu, 0],
'text': '{0:.3g}'.format(vu),
'layer': layer
}, {
'radius': h,
'pos': [x0, yl, 0],
'text': '{0:.3g}'.format(vl),
'layer': layer
}])
texts.extend([{
'radius': h,
'pos': [x0, ymin + 4.8 * s, 0],
'text': '0',
'layer': layer
}, {
'radius': h,
'pos': [xmin, ymin + 12 * s, 0],
'text': 'Step:{0} Field:{1}'.format(step, field),
'layer': layer
}])
xdraw_texts(texts)