def draw_faces(self, keys=None, colors=None): self.clear_faces() self.clear_facelabels() keys = keys or list(self.datastructure.faces()) objects = [0] * len(keys) if colors is None: colors = {key: self.defaults['color.face'] for key in keys} for c, key in enumerate(keys): vertices = [ self.datastructure.vertex_coordinates(i) for i in self.datastructure.face[key] ] faces = [list(range(len(self.datastructure.face[key])))] name = 'F{0}'.format(key) objects[c] = xdraw_mesh(vertices=vertices, layer=self.layer, faces=faces, color=colors[key], name=name) self.face_objects = objects
def update(dofs, network, tol, plot, Xt, ds): x1, z1, r1, x2, z2, r2 = dofs dx1 = ds * cos(r1) dx2 = ds * cos(r2) dz1 = ds * sin(r1) dz2 = ds * sin(r2) sp, ep = network.leaves() network.set_vertex_attributes(sp, {'x': x1, 'z': z1}) network.set_vertex_attributes(ep, {'x': x2, 'z': z2}) network.set_vertex_attributes(sp + 1, {'x': dx1 + x1, 'z': dz1 + z1}) network.set_vertex_attributes(ep - 1, {'x': dx2 + x2, 'z': dz2 + z2}) X, f, l = drx_numba(network=network, factor=1, tol=tol) if plot: ind = argmin(cdist(X, Xt), axis=1) vertices = vstack([X, Xt[ind, :]]) n = X.shape[0] edges = [[i, i + n] for i in range(n)] + list(network.edges()) xdraw_mesh(name='norms', vertices=vertices, edges=edges) return X
def weld_meshes_from_layer(layer_input, layer_output): """ Grab meshes on an input layer and weld them onto an output layer. Parameters ---------- layer_input : str Layer containing the Blender meshes to weld. layer_output : str Layer to plot single welded mesh. Returns ------- None """ print('Welding meshes on layer:{0}'.format(layer_input)) S = Structure(path=' ') add_nodes_elements_from_layers(S, mesh_type='ShellElement', layers=layer_input) faces = [] for element in S.elements.values(): faces.append(element.nodes) try: clear_layer(layer_output) except: create_layer(layer_output) vertices = S.nodes_xyz() xdraw_mesh(name='welded_mesh', vertices=vertices, faces=faces, layer=layer_output)
def discretise_mesh(structure, mesh, layer, target, min_angle=15, factor=1): """ Discretise a mesh from an input triangulated coarse mesh into small denser meshes. Parameters ---------- structure : obj Structure object. mesh : obj The object of the Blender input mesh. layer : str Layer name to draw results. target : float Target length of each triangle. min_angle : float Minimum internal angle of triangles. factor : float Factor on the maximum area of each triangle. Returns ------- None """ blendermesh = BlenderMesh(mesh) vertices = list(blendermesh.get_vertices_coordinates().values()) faces = list(blendermesh.get_faces_vertex_indices().values()) try: points, tris = discretise_faces(vertices=vertices, faces=faces, target=target, min_angle=min_angle, factor=factor) for pts, tri in zip(points, tris): bmesh = xdraw_mesh(name='face', vertices=pts, faces=tri, layer=layer) add_nodes_elements_from_bmesh(structure=structure, bmesh=bmesh, mesh_type='ShellElement') except: print('***** Error using MeshPy (Triangle) or drawing faces *****')
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 add_tets_from_mesh(structure, name, mesh, draw_tets=False, volume=None, thermal=False): """ Adds tetrahedron elements from a mesh to the Structure object. Parameters ---------- structure : obj Structure object to update. name : str Name for the element set of tetrahedrons. mesh : obj The Blender mesh representing the outer surface. draw_tets : bool Layer to draw tetrahedrons on. volume : float Maximum volume for tets. thermal : bool Thermal properties on or off. Returns ------- None """ blendermesh = BlenderMesh(mesh) vertices = blendermesh.get_vertices_coordinates().values() faces = blendermesh.get_faces_vertex_indices().values() try: tets_points, tets_elements = tets_from_vertices_faces( vertices=vertices, faces=faces, volume=volume) for point in tets_points: structure.add_node(point) ekeys = [] for element in tets_elements: nodes = [ structure.check_node_exists(tets_points[i]) for i in element ] ekey = structure.add_element(nodes=nodes, type='TetrahedronElement', thermal=thermal) ekeys.append(ekey) structure.add_set(name=name, type='element', selection=ekeys) if draw_tets: tet_faces = [[0, 1, 2], [1, 3, 2], [1, 3, 0], [0, 2, 3]] for i, points in enumerate(tets_elements): xyz = [tets_points[j] for j in points] xdraw_mesh(name=str(i), vertices=xyz, faces=tet_faces, layer=draw_tets) print('***** MeshPy (TetGen) successfull *****') except: print('***** Error using MeshPy (TetGen) or drawing Tets *****')
# Step xv_all[n, :3] += dt * xv_all[n, 3:] xv_all[n, 3] += dt * Fx / rocket_mass xv_all[n, 4] += dt * Fy / rocket_mass xv_all[n, 5] += dt * Fz / rocket_mass return xv_all # Loop xv_all = vstack([xv_planets, xv_rocket]) for i in range(duration // day): # Update day xv_all = simulation(n, x_sun, xv_all, k, dt, planets_mass, rocket_mass, day, m, p) # Plot if i % refresh == 0: locations = [list(loc) for loc in list(xv_all[:, :3] / giga)] set_objects_location(objects=planets, locations=locations[:8]) set_objects_location(objects=[rocket], locations=[locations[8]]) mesh = xdraw_mesh(name='path', vertices=[locations[8]]) bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
__license__ = 'MIT License' __email__ = '*****@*****.**' clear_layer(layer=1) # Structure mdl = Structure(name='mesh_tris', path='/home/al/temp/') # Discretise blendermesh = BlenderMesh(get_objects(layer=0)[0]) pts = blendermesh.get_vertex_coordinates() fcs = blendermesh.get_face_vertex_indices() vertices, faces = functions.discretise_faces(vertices=pts, faces=fcs, target=0.1, min_angle=15, factor=1, iterations=50) for pts, fc in zip(vertices, faces): bmesh = xdraw_mesh(name='face', vertices=pts, faces=fc, layer=1, wire=True) blender.add_nodes_elements_from_bmesh(mdl, bmesh=bmesh, mesh_type='ShellElement') # Summary mdl.summary()
# Structure mdl = Structure(name='beam_simple', path='/home/al/temp/') # Clear clear_layers(layers=[0, 1, 2, 3]) # Lines L = 1.0 m = 100 x = [i * L / m for i in range(m + 1)] vertices = [[xi, 0, 0] for xi in x] edges = [[i, i + 1] for i in range(m)] bmesh = xdraw_mesh(name='beam', vertices=vertices, edges=edges, layer=0) # Points n = 5 xdraw_spheres([{'pos': [0, 0, 0], 'layer': 1, 'radius': 0.01}, {'pos': [L, 0, 0], 'layer': 2, 'radius': 0.01}]) xdraw_spheres([{'pos': [i, 0, 0], 'layer': 3, 'radius': 0.005} for i in x[n::n]]) # Elements network = network_from_bmesh(bmesh=bmesh) mdl.add_nodes_elements_from_network(network=network, element_type='BeamElement', elset='elset_lines', axes={'ex': [0, -1, 0]}) # Sets
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)
def add_tets_from_bmesh(structure, name, bmesh, draw_tets=False, volume=None, layer=19, acoustic=False, thermal=False): """ Adds tetrahedron elements from a Blender mesh to the Structure object. Parameters ---------- structure : obj Structure object to update. name : str Name for the element set of tetrahedrons. bmesh : obj The Blender mesh representing the outer surface. draw_tets : bool Draw the generated tetrahedrons. volume : float Maximum volume for tets. layer : int Layer to draw tetrahedrons if draw_tets=True. acoustic : bool Acoustic properties on or off. thermal : bool Thermal properties on or off. Returns ------- None """ blendermesh = BlenderMesh(bmesh) vertices = blendermesh.get_vertex_coordinates() faces = blendermesh.get_face_vertex_indices() tets_points, tets_elements = tets_from_vertices_faces(vertices=vertices, faces=faces, volume=volume) for point in tets_points: structure.add_node(point) ekeys = [] for element in tets_elements: nodes = [structure.check_node_exists(tets_points[i]) for i in element] ekey = structure.add_element(nodes=nodes, type='TetrahedronElement', acoustic=acoustic, thermal=thermal) ekeys.append(ekey) structure.add_set(name=name, type='element', selection=ekeys) if draw_tets: tet_faces = [[0, 1, 2], [1, 3, 2], [1, 3, 0], [0, 2, 3]] for i, points in enumerate(tets_elements): xyz = [tets_points[j] for j in points] xdraw_mesh(name=str(i), vertices=xyz, faces=tet_faces, layer=layer)