034_network_info_shortest_path.py

from compas.utilities import pairwise
import random

import compas
from compas.artists import Artist
from compas.datastructures import Network

network = Network.from_obj(compas.get('grid_irregular.obj'))

# Select random nodes + shortest path
start = random.choice(list(network.leaves()))
goal = random.choice(list(set(network.leaves()) - set([start])))
nodes = network.shortest_path(start, goal)

nodecolor = {}
edgecolor = {}

for u, v in pairwise(nodes):
    nodecolor[v] = (0, 255, 0)
    edgecolor[u, v] = edgecolor[v, u] = (0, 255, 0)

nodecolor[start] = (255, 0, 0)
nodecolor[goal] = (0, 0, 255)

print(network.summary())

artist = Artist(network, layer='network')
artist.clear_layer()
artist.draw_nodelabels(color=nodecolor)
artist.draw_edges(color=edgecolor)