def plot(self, vertexcolor=None, edgecolor=None, vertexsize=None, edgewidth=None, vertextext=None, edgetext=None): """Plot a 2D representation of the network. Parameters ---------- vertexcolor : dict, optional A dictionary mapping vertex identifiers to colors. edgecolor : dict, optional A dictionary mapping edge identifiers to colors. vertexsize : dict, optional A dictionary mapping vertex identifiers to sizes. edgewidth : dict, optional A dictionary mapping edge identifiers to widths. vertextext : dict, optional A dictionary mappping vertex identifiers to labels. edgetext : dict, optional A dictionary mappping edge identifiers to labels. Examples -------- .. plot:: :include-source: import compas from compas.datastructures import Network network = Network.from_obj(compas.get('lines.obj')) network.plot() """ from compas.plotters import NetworkPlotter plotter = NetworkPlotter(self) plotter.draw_vertices(facecolor=vertexcolor, radius=vertexsize, text=vertextext) plotter.draw_edges(color=edgecolor, width=edgewidth, text=edgetext) plotter.show()
import compas from compas.datastructures import Network from compas.topology import network_embed_in_plane from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('fink.obj')) embedding = network.copy() fix = (1, 12) if network_embed_in_plane(embedding, fix=fix): plotter = NetworkPlotter(embedding) plotter.draw_lines([{ 'start': network.vertex_coordinates(u, 'xy'), 'end': network.vertex_coordinates(v, 'xy'), 'color': '#cccccc' } for u, v in network.edges()]) plotter.draw_vertices(radius=0.3, text={key: key for key in embedding.vertices()}, facecolor={key: '#ff0000' for key in fix}) plotter.draw_edges() plotter.show()
structure.set_vertices_attributes(['B', 'is_fixed'], [[0, 0, 0], True], structure.leaves()) structure.attributes['beams'] = {'beam': {'nodes': list(range(n))}} lines = [] for u, v in structure.edges(): lines.append({ 'start': structure.vertex_coordinates(u, 'xy'), 'end': structure.vertex_coordinates(v, 'xy'), 'color': '#cccccc' }) plotter = NetworkPlotter(structure, figsize=(10, 7)) plotter.draw_vertices( radius=0.005, facecolor={ i: '#ff0000' for i in structure.vertices_where({'is_fixed': True}) }) plotter.draw_lines(lines) plotter.draw_edges() def callback(X, k_i): for key in structure.vertices(): x, y, z = X[k_i[key], :] structure.set_vertex_attributes(key, 'xyz', [x, y, z]) plotter.update_edges() plotter.update(pause=0.01) drx_numpy(structure=structure, tol=0.01,
from compas.datastructures import Network from compas.topology import dijkstra_distances from compas.plotters import NetworkPlotter from compas.utilities import i_to_red network = Network.from_obj(compas.get('grid_irregular.obj')) adjacency = {key: network.vertex_neighbours(key) for key in network.vertices()} weight = {(u, v): network.edge_length(u, v) for u, v in network.edges()} weight.update({(v, u): weight[(u, v)] for u, v in network.edges()}) target = 22 distances = dijkstra_distances(adjacency, weight, target) plotter = NetworkPlotter(network) dmax = max(distances.values()) facecolor = { key: i_to_red(distances[key] / dmax) for key in network.vertices() } text = {key: '{:.1f}'.format(distances[key]) for key in network.vertices()} plotter.draw_vertices(text=text, facecolor=facecolor, radius=0.15) plotter.draw_edges() plotter.show()
# identify the fixed vertices network.set_vertices_attribute('is_fixed', True, keys=network.vertices_where( {'vertex_degree': 1})) # assign random prescribed force densities to the edges for uv in network.edges(): network.set_edge_attribute(uv, 'qpre', 1.0 * random.randint(1, 7)) # make a plotter for (dynamic) visualization plotter = NetworkPlotter(network, figsize=(10, 7)) # plot the starting configuration plotter.draw_vertices(facecolor={ key: '#000000' for key in network.vertices_where({'is_fixed': True}) }) plotter.draw_edges() plotter.update(pause=1.0) # run the DR network_dr(network, callback=callback) # plot the final configuration plotter.draw_vertices(facecolor={ key: '#000000' for key in network.vertices_where({'is_fixed': True}) }) plotter.draw_edges() plotter.update(pause=1.0)
if __name__ == '__main__': import compas from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.defaults['vertex.fontsize'] = 8 network.delete_vertex(17) plotter.draw_vertices(text='key', radius=0.2) plotter.draw_edges() plotter.show() vertices = {44: [0.0, 0.0, 0.0], 38: [1.0, 0.0, 0.0], 2: [2.0, 0.0, 0.0]} edges = [(44, 38), (38, 2)] network = Network.from_vertices_and_edges(vertices, edges) print(network) plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.defaults['vertex.fontsize'] = 8 plotter.draw_vertices(text='key', radius=0.2) plotter.draw_edges() plotter.show()
network = Network.from_obj(compas.get('fink.obj')) crossings = network_find_crossings(network) print(network_count_crossings(network)) print(crossings) print(len(crossings)) ecolor = {} for e1, e2 in crossings: ecolor[e1] = '#ff0000' ecolor[e2] = '#ff0000' plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.draw_vertices() plotter.draw_edges(color=ecolor) plotter.show() # embedding = network.copy() # fix = (1, 12) # if network_embed_in_plane(embedding, fix=fix): # plotter = NetworkPlotter(embedding, figsize=(10, 7)) # plotter.draw_lines([{'start': network.vertex_coordinates(u, 'xy'), # 'end': network.vertex_coordinates(v, 'xy'), # 'color': '#cccccc'} for u, v in network.edges()])
# compute the shortest path taking into account the edge weights path = shortest_path(network.adjacency, start, end) # convert the path to network edges edges = [(v, u) if not network.has_edge(u, v) else (u, v) for u, v in pairwise(path)] # make a plotter plotter = NetworkPlotter(network, figsize=(10, 7)) # set default font sizes plotter.defaults['vertex.fontsize'] = 6 plotter.defaults['edge.fontsize'] = 6 # draw the vertices plotter.draw_vertices( text='key', facecolor={key: '#ff0000' for key in (path[0], path[-1])}, radius=0.15) # draw the edges plotter.draw_edges(color={(u, v): '#ff0000' for u, v in edges}, width={(u, v): 5.0 for u, v in edges}) # show the plot plotter.show()
if __name__ == '__main__': import compas_ags from compas.plotters import NetworkPlotter form = FormDiagram.from_obj(compas_ags.get('paper/fink.obj')) lines = [] for u, v in form.edges(): lines.append({ 'start': form.vertex_coordinates(u), 'end': form.vertex_coordinates(v), 'color': '#cccccc', 'width': 0.5, }) form.identify_fixed() vcolor = {key: '#ff0000' for key in form.fixed()} vlabel = {key: key for key in form.vertices()} elabel = {(u, v): str(index) for index, (u, v) in enumerate(form.edges())} plotter = NetworkPlotter(form, figsize=(10.0, 7.0), fontsize=8) plotter.draw_lines(lines) plotter.draw_vertices(facecolor=vcolor, text=vlabel, radius=0.3) plotter.draw_edges(text=elabel) plotter.show()
from compas.datastructures import Network from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) u, v = network.get_any_edge() a = network.split_edge(u, v) lines = [] for u, v in network.edges(): lines.append({ 'start': network.vertex_coordinates(u, 'xy'), 'end' : network.vertex_coordinates(v, 'xy'), 'arrow': 'end', 'width': 4.0, 'color': '#00ff00' }) plotter = NetworkPlotter(network) plotter.draw_lines(lines) plotter.draw_vertices( radius=0.2, text={key: key for key in network.vertices()}, facecolor={key: '#ff0000' for key in (a,)} ) plotter.draw_edges() plotter.show()
if __name__ == '__main__': import compas from compas.datastructures import Network from compas.datastructures import network_is_planar from compas.datastructures import network_find_crossings from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) network.add_edge(6, 15) if not network_is_planar(network): crossings = network_find_crossings(network) else: crossings = [] print(crossings) plotter = NetworkPlotter(network) plotter.draw_vertices(radius=0.15, text={key: key for key in network.vertices()}) plotter.draw_edges( color={edge: '#ff0000' for edges in crossings for edge in edges}) plotter.show()
path = shortest_path(adjacency, start, end) edges = [] for i in range(len(path) - 1): u = path[i] v = path[i + 1] if v not in network.edge[u]: u, v = v, u edges.append([u, v]) plotter = NetworkPlotter(network, figsize=(10, 8), fontsize=6) plotter.draw_vertices( text={key: key for key in network.vertices()}, facecolor={key: '#ff0000' for key in (path[0], path[-1])}, radius=0.15) plotter.draw_edges(color={(u, v): '#ff0000' for u, v in edges}, width={(u, v): 5.0 for u, v in edges}) plotter.show() # ========================================================================== # testrun 3 # ========================================================================== if testrun == 3:
from compas.datastructures import network_transformed from compas.geometry import Rotation from compas_assembly.datastructures import Assembly from compas_assembly.datastructures import assembly_interfaces from compas_rbe.equilibrium import compute_interface_forces_cvx assembly = Assembly.from_json(compas_assembly.get('wall.json')) keys = list(assembly.vertices_where({'is_support': True})) print(keys) if keys: assembly_interfaces(assembly) R = Rotation.from_axis_and_angle([1.0, 0, 0], -pi / 2) network = network_transformed(assembly, R) plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.draw_vertices(facecolor={ key: '#ff0000' for key in network.vertices_where({'is_support': True}) }) plotter.draw_edges() plotter.show() compute_interface_forces_cvx(assembly, solver='CVXOPT', verbose=True) assembly.to_json(compas_assembly.get('assembly_result.json'))
width={uv: 5.0 for uv in edges}) plotter.update() # callback for the pick event def onpick(event): index = event.ind[0] shortest_path_to(index) # path to the sample file DATA = os.path.join(os.path.dirname(__file__), '..', 'data') FILE = os.path.join(DATA, 'grid_irregular.obj') # load a network from an OBJ file network = Network.from_obj(FILE) # define the starting point start = 21 # create plotter # draw the original configuration # register the pick callback # show the viewer plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.draw_vertices(facecolor={start: '#ff0000'}, radius=0.15, picker=10) plotter.draw_edges() plotter.register_listener(onpick) plotter.show()
cloads = Array2D(loads, 'double') cq = Array1D(q, 'double') cfixed = Array1D(fixed, 'int') cfree = Array1D(free, 'int') lib.fd.argtypes = [ ctypes.c_int, ctypes.c_int, ctypes.c_int, cvertices.ctype, cedges.ctype, cloads.ctype, cq.ctype, cfixed.ctype, cfree.ctype ] lib.fd(ctypes.c_int(len(vertices)), ctypes.c_int(len(edges)), ctypes.c_int(len(fixed)), cvertices.cdata, cedges.cdata, cloads.cdata, cq.cdata, cfixed.cdata, cfree.cdata) xyz = cvertices.pydata for key, attr in network.vertices(True): attr['x'] = float(xyz[key][0]) attr['y'] = float(xyz[key][1]) attr['z'] = float(xyz[key][2]) zmax = max(network.get_vertices_attribute('z')) plotter = NetworkPlotter(network, figsize=(10, 7)) plotter.draw_vertices(facecolor={ key: i_to_red(attr['z'] / zmax) for key, attr in network.vertices(True) }) plotter.draw_edges() plotter.show()
import compas from compas.datastructures import Network from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) plotter = NetworkPlotter(network) plotter.draw_vertices(text='key', facecolor={key: '#ff0000' for key in network.leaves()}, radius=0.15) plotter.draw_edges() plotter.show()
# -------------------------------------------------------------------------- # -------------------------------------------------------------------------- # selections # -------------------------------------------------------------------------- # inherited from VertexAttributesMixin # inherited from EdgeAttributesMixin # ============================================================================== # Main # ============================================================================== if __name__ == '__main__': import compas from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) plotter = NetworkPlotter(network) network.delete_vertex(17) plotter.draw_vertices(text='key') plotter.draw_edges() plotter.show()
weight = {(u, v): network.edge_length(u, v) for u, v in network.edges()} weight.update({(v, u): weight[(u, v)] for u, v in network.edges()}) path = dijkstra_path(adjacency, weight, start, end) # visualize the result plotter = NetworkPlotter(network, figsize=(10, 8), fontsize=6) edges = [] for u, v in pairwise(path): if v not in network.edge[u]: u, v = v, u edges.append([u, v]) plotter.draw_vertices( text={key: key for key in (start, end)}, facecolor={key: '#ff0000' for key in (path[0], path[-1])}, radius=0.15) plotter.draw_edges(color={(u, v): '#ff0000' for u, v in edges}, width={(u, v): 3.0 for u, v in edges}, text={(u, v): '{:.1f}'.format(weight[(u, v)]) for u, v in network.edges()}) plotter.show()
import compas from compas.datastructures import Network from compas.plotters import NetworkPlotter from compas.geometry import network_smooth_centroid network = Network.from_obj(compas.get('grid_irregular.obj')) fixed = [key for key in network.vertices() if network.vertex_degree(key) == 1] network_smooth_centroid(network, fixed=fixed) plotter = NetworkPlotter(network) plotter.draw_vertices(facecolor={key: '#ff0000' for key in fixed}) plotter.draw_edges() plotter.show()
plotter = NetworkPlotter(network, figsize=(10, 7)) lines = [] for u, v in network.edges(): lines.append({ 'start': network.vertex_coordinates(u, 'xy'), 'end': network.vertex_coordinates(v, 'xy'), 'color': '#cccccc', 'width': 1.0 }) plotter.draw_lines(lines) plotter.draw_vertices( radius=0.005, facecolor={ key: '#ff0000' for key in network.vertices_where({'is_fixed': True}) }) plotter.draw_edges() # Solver drx_numpy(network=network, tol=0.01, refresh=10, factor=30, update=True, callback=plot_iterations) plotter.show()
import compas from compas.datastructures import Network from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('lines.obj')) plotter = NetworkPlotter(network) plotter.draw_vertices(text={key: key for key in network.vertices()}, radius=0.2) plotter.draw_edges() plotter.show()
import compas from compas.datastructures import Network from compas.plotters import NetworkPlotter from compas.topology import vertex_coloring network = Network.from_obj(compas.get('grid_irregular.obj')) key_color = vertex_coloring(network.adjacency) colors = ['#ff0000', '#00ff00', '#0000ff'] plotter = NetworkPlotter(network) plotter.draw_vertices(facecolor={key: colors[key_color[key]] for key in network.vertices()}) plotter.draw_edges() plotter.show()
plotter = NetworkPlotter(network, figsize=(10, 7)) # Initial configuration lines = [] for u, v in network.edges(): lines.append({ 'start': network.vertex_coordinates(u, 'xy'), 'end': network.vertex_coordinates(v, 'xy'), 'color': '#cccccc', 'width': 1.0 }) plotter.draw_lines(lines) plotter.draw_vertices(radius=0.005, facecolor={key: '#ff0000' for key in pins}) plotter.draw_edges() plotter.update() # Callback for dynamic visualization def plot_iterations(X, radius=0.005): for i in network.vertices(): x, y, z = X[i, :] network.set_vertex_attributes(i, 'xyz', [x, y, z]) plotter.update_vertices(radius) plotter.update_edges() plotter.update(pause=0.01)
# ============================================================================== # Main # ============================================================================== if __name__ == "__main__": import compas from compas.datastructures import Network from compas.plotters import NetworkPlotter network = Network.from_obj(compas.get('grid_irregular.obj')) plotter = NetworkPlotter(network, figsize=(10, 8)) plotter.draw_vertices(radius=0.1, picker=10) plotter.draw_edges() default = [ plotter.defaults['vertex.facecolor'] for key in network.vertices() ] highlight = '#ff0000' def on_pick(event): index = event.ind[0] colors = default[:] colors[index] = highlight plotter.vertexcollection.set_facecolor(colors) plotter.update()
class AssemblyPlotter(Plotter): """An ``AssemblyPlotter`` combines the functionality of a ``NetworkPlotter`` and a ``MeshPlotter`` and uses the same set of axes for all drawing output. Parameters ---------- assembly : Assembly The assembly data structure. Notes ----- For all other relevant parameters, see ``Plotter``. Examples -------- .. code-block:: python plotter = AssemblyPlotter(assembly, tight=True, figsize=(12, 8)) plotter.draw_vertices() plotter.draw_blocks() plotter.show() """ def __init__(self, assembly, **kwargs): super(AssemblyPlotter, self).__init__(**kwargs) self.assembly = assembly self.assembly_plotter = NetworkPlotter(self.assembly, axes=self.axes) self.block_plotter = MeshPlotter(None, axes=self.axes) def draw_vertices(self, *args, **kwargs): """Draw the vertices of an assembly. """ return self.assembly_plotter.draw_vertices(*args, **kwargs) def draw_edges(self, *args, **kwargs): """Draw the edges of an assembly. """ self.assembly_plotter.draw_edges(*args, **kwargs) def draw_blocks(self, keys=None, facecolor=None, edgecolor=None, edgewidth=None, textcolor=None, fontsize=None): """Draw the blocks of an assembly. Notes ----- The blocks are drawn as the boundaing boxes of their vertices. """ keys = keys or list(self.assembly.vertices()) facecolordict = valuedict(keys, facecolor, self.block_plotter.defaults['face.facecolor']) edgecolordict = valuedict(keys, edgecolor, self.block_plotter.defaults['face.edgecolor']) edgewidthdict = valuedict(keys, edgewidth, self.block_plotter.defaults['face.edgewidth']) textcolordict = valuedict(keys, textcolor, self.block_plotter.defaults['face.textcolor']) fontsizedict = valuedict(keys, fontsize, self.block_plotter.defaults['face.fontsize']) polygons = [] for key, attr in self.assembly.vertices(True): block = self.assembly.blocks[key] xyz = block.get_vertices_attributes('xyz') box = bounding_box_xy(xyz) polygons.append({ 'points': box, 'edgecolor': edgecolordict[key], 'edgewidth': edgewidthdict[key], 'facecolor': facecolordict[key] }) self.draw_polygons(polygons)