uy = -y + r * np.sin(theta**2)
uz = 0. * z
return ux, uy, uz
def scalar_function(x, y, z, labels):
"""
Scalar function used to produced the plotted field.
"""
return x**2 + y**2
#MESH GENERATION
N1, N2 = 30, 30
l1, l2 = .75, 1.
m = RegularQuadMesh(N1=N1, N2=N2, l1=l1, l2=l2)
#FIELDS GENERATION
u = m.nodes.eval_vectorFunction(vector_function)
m.add_field(u, "u")
f = m.nodes.eval_function(scalar_function)
m.add_field(f, "f")
#PLOTS
fig = plt.figure(0)
plt.clf()
ax = fig.add_subplot(1, 1, 1)
m.draw(ax,
disp_func=lambda fields: fields["u"],
field_func=lambda fields: fields["f"],
cmap=cm.jet,
cbar_orientation="vertical",
contour=False,
from abapy.mesh import RegularQuadMesh, Mesh
from matplotlib import pyplot as plt
from array import array
from abapy.indentation import IndentationMesh
m = RegularQuadMesh(N1=2, N2=2)
m.connectivity[2] = array(m.dti, [5, 7, 4])
m.connectivity[3] = array(m.dti, [5, 6, 9])
m.add_set('el_set', [1, 2])
m.add_set('el_set2', [2, 4])
m.add_surface('my_surface', [
('el_set', 1),
])
m2 = m.sweep(sweep_angle=70., N=2, extrude=True)
x, y, z = m.get_edges()
x2, y2, z2 = m2.get_edges()
# Adding some 3D "home made" perspective:
zx, zy = .3, .3
for i in xrange(len(x2)):
if x2[i] != None:
x2[i] += zx * z2[i]
y2[i] += zy * z2[i]
# Plotting stuff
plt.figure()
plt.clf()
plt.gca().set_aspect('equal')
plt.axis('off')
plt.plot(x, y, 'b-', linewidth=4., label='Orginal Mesh')
plt.plot(x2, y2, 'r-', linewidth=1., label='Sweeped mesh')
