cylinder = Cylinder(circle, 0.7 * height) circle = [[0, 0, 0.7 * height], [0, 0, 1]], 0.1 cone = Cone(circle, 0.3 * height) for node in network.nodes(): a = network.node_attributes(node, 'xyz') for nbr in network.neighbors(node): edge = node, nbr if not network.has_edge(*edge): edge = nbr, node b = network.node_attributes(nbr, 'xyz') force = network.edge_attribute(edge, 'f') direction = normalize_vector(subtract_vectors(b, a)) frame = Frame.from_plane([a, direction]) X = Transformation.from_frame_to_frame(world, frame) S = Scale.from_factors([force, 1, 1]) X = X * S shaft = cylinder.transformed(X) tip = cone.transformed(X) artist = CylinderArtist(shaft, layer=layer, color=(255, 0, 0)) artist.draw(u=16) artist = ConeArtist(tip, layer=layer, color=(255, 0, 0)) artist.draw(u=16)
import random from compas_rhino.artists import NetworkArtist from compas.datastructures import Network network = Network() network.add_edge(1, 2) network.add_edge(2, 3) network.add_edge(1, 4) network.add_edge(4, 5) network.add_edge(4, 6) for node in network.nodes(): x = random.choice(range(5)) y = random.choice(range(5)) z = random.choice(range(5)) network.node_attributes(node, 'xyz', [x, y, z]) print(network.summary()) text = {node: str(node) for node in network.nodes()} artist = NetworkArtist(network, layer='network') artist.clear_layer() artist.draw_nodelabels(text) artist.draw() artist.redraw()
for node in network.nodes(): point = network.node_coordinates(node) for nbr in tree.nearest_neighbors(point, 4, distance_sort=True): if nbr[2] < 1e-6: continue if not network.has_edge(node, nbr[1], directed=False): network.add_edge(node, nbr[1]) start = network.get_any_node() goal = network.get_any_node() path = network.shortest_path(start, goal) edges = [(u, v) if network.has_edge(u, v) else (v, u) for u, v in pairwise(path)] artist = NetworkArtist(network, layer="ITA20::Network") artist.clear_layer() artist.draw_nodes(color={start: (255, 0, 0), goal: (0, 0, 255)}) artist.draw_edges(color={edge: (0, 255, 0) for edge in edges}) for u, v in edges: o = network.edge_midpoint(u, v) n = network.edge_direction(u, v) h = network.edge_length(u, v) cylinder = Cylinder([(o, n), 0.02], h) artist = CylinderArtist(cylinder, color=(0, 255, 0), layer="ITA20::Network") artist.draw(show_vertices=False)