Пример #1
0
    def draw_stress(self, scale=None):
        self.clear_stress()

        scale = scale or self.settings['scale.stress']

        stress = [
            self.mesh.stress(key)
            for key in self.mesh.edges_where({'is_edge': True})
        ]
        cmap = Colormap(stress, 'rgb')

        lines = []
        for index, (u, v) in enumerate(self.mesh.edges_where({'is_edge':
                                                              True})):
            sp, ep = self.mesh.edge_coordinates(u, v)

            lines.append({
                'start': sp,
                'end': ep,
                'radius': scale,
                'color': cmap(stress[index]),
                'name': "{}.stress.{}-{}".format(self.mesh.name, u, v)
            })

        self._draw_cylinders(lines)
Пример #2
0
isolines = igl.groupsort_isolines(vertices, edges)

# ==============================================================================
# Visualisation
# ==============================================================================

viewer = ObjectViewer()

viewer.add(mesh,
           settings={
               'color': '#ffffff',
               'vertices.on': False,
               'edges.on': False
           })

cmap = Colormap(scalars, 'rgb')
for value, paths in isolines:
    for path in paths:
        points = [vertices[path[0][0]]]
        for i, j in path:
            points.append(vertices[j])
        viewer.add(Polyline(points),
                   settings={
                       'edges.color': rgb_to_hex(cmap(value)),
                       'edges.width': 5,
                       'vertices.on': False
                   })

viewer.update()
viewer.show()
Пример #3
0
cablenet = Cablenet.from_json(FILE_I)

# ==============================================================================
# Materialize
# ==============================================================================

mesh_materialize_cables(cablenet)

# ==============================================================================
# Visualize
# ==============================================================================

edges = list(cablenet.edges_where({'is_edge': True}))

stress = [cablenet.stress(key) for key in edges]
cmap = Colormap(stress, 'rgb')
edgecolor = {key: cmap(s) for key, s in zip(edges, stress)}

utilization = [cablenet.stress(key) / cablenet.edge_attribute(key, 'yield') for key in edges]
cmap = Colormap(utilization, 'red')
edgecolor = {key: cmap(u) for key, u in zip(edges, utilization)}

print(min(utilization))
print(max(utilization))

plotter = MeshPlotter(cablenet, figsize=(16, 9))
plotter.draw_vertices(radius=0.05, facecolor={key: (0.0, 0.0, 0.0) for key in cablenet.vertices_where({'is_anchor': True})})
plotter.draw_edges(width=2.0, color=edgecolor, keys=edges)

plotter.save(FILE_P, dpi=150)
# plotter.show()
Пример #4
0
# ==============================================================================

cablenet.vertices_attribute('is_anchor', True, keys=list(cablenet.vertices_on_boundary()))

# ==============================================================================
# Compute equilibrium
# ==============================================================================

update_xyz_numpy(cablenet)

# ==============================================================================
# Visualize
# ==============================================================================

heights = cablenet.vertices_attribute('z')
cmap = Colormap(heights, 'black')
vertexcolor = {key: cmap(z) for key, z in zip(cablenet.vertices(), heights)}

forces = cablenet.edges_attribute('f')
cmap = Colormap(forces, 'rgb')
edgecolor = {key: cmap(f) for key, f in zip(cablenet.edges(), forces)}

plotter = MeshPlotter(cablenet, figsize=(16, 9))
plotter.draw_vertices(facecolor=vertexcolor, radius=0.05)
plotter.draw_edges(width=2.0, color=edgecolor)

plotter.save(FILE_P, dpi=150)
# plotter.show()

# ==============================================================================
# Export
Пример #5
0
import compas_libigl as igl

HERE = os.path.dirname(__file__)
FILE = os.path.join(HERE, '..', 'data', 'tubemesh.json')

mesh = Mesh.from_json(FILE)
mesh_quads_to_triangles(mesh)

key_index = mesh.key_index()
index_key = mesh.index_key()

V = mesh.vertices_attributes('xyz')
F = [[key_index[key] for key in mesh.face_vertices(fkey)]
     for fkey in mesh.faces()]

root = mesh.get_any_vertex()

# D = igl.trimesh_geodistance_exact(V, F, 0)
D = igl.trimesh_geodistance_heat(V, F, key_index[root])

cmap = Colormap(D, 'red')

plotter = MeshPlotter(mesh, figsize=(8, 5))
plotter.draw_faces()
plotter.draw_vertices(
    text={root: 'root'},
    radius=0.2,
    facecolor={key: cmap(d)
               for key, d in zip(mesh.vertices(), D)})
plotter.show()
# and its dual
mesh = Mesh.from_polyhedron(20)
dual = mesh.dual()

# mesh transformation for drawing various representations
# of the mesh in non-overlapping way
# the mesh will be transformed in-place
# therefore the same transformation can be applied consecutively
bbox = mesh.bounding_box_xy()
dx = bbox[1][0] - bbox[0][0]
X = Translation([1.5 * dx, 0, 0])

# vertex coloring
key_color = vertex_coloring(mesh.adjacency)
c = len(set(key_color.values()))
colors = Colormap(list(range(c)), 'rgb')
vertexcolor = {key: colors(key_color[key]) for key in mesh.vertices()}

# face coloring
# which is the same as a vertex coloring of the dual
key_color = vertex_coloring(dual.adjacency)
c = len(set(key_color.values()))
colors = Colormap(list(range(c)), 'rgb')
facecolor = {key: colors(key_color[key]) for key in dual.vertices()}

# the artist for drawing various versions of the mesh
artist = MeshArtist(mesh)

# mesh
mesh.name = "Mesh"
artist.clear()
Пример #7
0
cablenet.vertices_attribute('t', 0.05)

# ==============================================================================
# Unload the cablenet
# ==============================================================================

apply_loads_numpy(cablenet)

# ==============================================================================
# Visualize
# ==============================================================================

edges = list(cablenet.edges_where({'is_edge': True}))

stress = [cablenet.stress(key) for key in edges]
cmap = Colormap(stress, 'rgb')
edgecolor = {key: cmap(s) for key, s in zip(edges, stress)}

print(stress)

utilization = [
    cablenet.stress(key) / cablenet.edge_attribute(key, 'yield')
    for key in edges
]
# cmap = Colormap(utilization, 'red')
# edgecolor = {key: cmap(u) for key, u in zip(edges, utilization)}

print(min(utilization))
print(max(utilization))

plotter = MeshPlotter(cablenet, figsize=(16, 9))
Пример #8
0
from compas.utilities import Colormap
from compas_plotters import MeshPlotter

# ==============================================================================
# Input geometry
# ==============================================================================

mesh = Mesh.from_off(igl.get('tubemesh.off'))
mesh.quads_to_triangles()

# ==============================================================================
# Mass matrix
# ==============================================================================

mass = igl.trimesh_massmatrix(mesh)

# ==============================================================================
# Visualisation
# ==============================================================================

cmap = Colormap(mass, 'red')

plotter = MeshPlotter(mesh, figsize=(8, 5))
plotter.draw_vertices(radius=0.2,
                      facecolor={
                          key: cmap(mass[index])
                          for index, key in enumerate(mesh.vertices())
                      })
plotter.draw_faces()
plotter.show()
Пример #9
0
# ==============================================================================
# Input
# ==============================================================================

mesh = Mesh.from_off(igl.get('tubemesh.off'))
mesh.quads_to_triangles()

# ==============================================================================
# Geodesic distance
# ==============================================================================

source = mesh.get_any_vertex()
distance = igl.trimesh_geodistance(mesh.to_vertices_and_faces(),
                                   source,
                                   method='heat')

# ==============================================================================
# Visualize
# ==============================================================================

cmap = Colormap(distance, 'red')

plotter = MeshPlotter(mesh, figsize=(16, 9), tight=True)
plotter.draw_vertices(radius=0.2,
                      facecolor={
                          key: cmap(distance[index])
                          for index, key in enumerate(mesh.vertices())
                      })
plotter.draw_faces()
plotter.save('examples/geodistance.png')