[Examples] Cleaned 01_quick_start.py

Author: arpastrana

examples/01_quick_start.py

@@ -14,12 +14,6 @@
 from compas_cem.plotters import Plotter
 
 
-# ------------------------------------------------------------------------------
-# Parameters
-# ------------------------------------------------------------------------------
-
-plot = False
-
 # ------------------------------------------------------------------------------
 # Instantiate a topology diagram
 # ------------------------------------------------------------------------------
@@ -67,144 +61,22 @@
 # Compute a state of static equilibrium
 # ------------------------------------------------------------------------------
 
-# form = static_equilibrium(topology, eta=1e-6, tmax=100, verbose=True)
-
-# ------------------------------------------------------------------------------
-# Compute a state of static equilibrium
-# ------------------------------------------------------------------------------
-
-import numpy as np
-import equinox as eqx
-
-class EquilibriumModel(eqx.Module):
-    pass
-
-
-# trails_sequences = topology.trails_sequences()
-# trails = {k: v[:] for k, v in topology.trails()}
-
-nodes = list(topology.nodes())
-node_index = topology.key_index()
-
-edges = list(topology.edges())
-edge_index = topology.uv_index()
-index_edge = topology.index_uv()
-
-deviation_mask = []
-for edge in edges:
-    mask_value = 0
-    if topology.is_deviation_edge(edge):
-        mask_value = 1
-    deviation_mask.append(mask_value)
-
-print(f"{edges=}")
-print(f"{deviation_mask=}")
-
-# topology.shift_trail(2, 1)
-sequences = np.ones((
-                     topology.number_of_sequences(),
-                     topology.number_of_trails()),
-                    dtype=np.int32)
-sequences *= -1
-
-for tidx, trail in enumerate(topology.trails()):
-    for node in trail:
-        seq = topology.node_sequence(node)
-        sequences[seq][tidx] = node_index[node]
-
-# trail_lengths = np.zeros(topology.number_of_edges())
-# for i, edge in enumerate(edges):
-#    if topology.is_trail_edge(edge):
-#        trail_lengths[i] = topology.edge_length_2(edge)
-
-trail_lengths = np.zeros(topology.number_of_nodes())
-
-for trail in topology.trails():
-    for u, v in pairwise(trail):
-        edge = (u, v)
-        if edge not in edges:
-            edge = (v, u)
-        trail_lengths[u] = topology.edge_length_2(edge)
-
-print(f"{trail_lengths=}")
-
-raise
-
-deviation_forces = np.zeros(topology.number_of_edges())
-for i, edge in enumerate(edges):
-    if topology.is_deviation_edge(edge):
-       deviation_forces[i] = topology.edge_force(edge)
-
-print(f"{deviation_forces=}")
-
-loads = list(topology.node_load(node) for node in nodes)
-loads = np.asarray(loads)
-
-positions = list(topology.node_coordinates(node) for node in nodes)
-positions = np.asarray(positions)
-
-residuals = []
-residual = np.zeros((topology.number_of_trails(), 3))
-
-
-def residual_vector(r, d, q):
-    return r - d - q
-
-def position_vector(p, r, l, t):
-    return p + l * (r / t)
-
-def trail_force(r):
-    return np.linalg.norm(residual, axis=1, keepdims=True)
-
-deviation = np.ones((topology.number_of_trails(), 3)) * -1.0
-
-# iterate over sequences
-for sequence in sequences:
-    load = loads[sequence, :]
-    position_last = positions[sequence, :]
-    # tlength = trail_lengths[sequence, :]
-
-    # deviation = deviation_vector(sequence, positions, adjacency, deviation_mask)
-    residual = residual_vector(residual, deviation, load)
-    residuals.append(residual)
-
-    tforce = trail_force(residual)
-    tlength = 1.
-
-    position_new = position_vector(position_last, residual, tlength, tforce)
-
-    print(sequence)
-    print(load)
-    print(position_last)
-    print(f"{position_last=}")
-    print("residual\n", residual)
-    print("trail force\n", tforce)
-    print(f"{position_new=}")
-
-    print()
-    position_last = position_new
-
-print("ok")
+form = static_equilibrium(topology, eta=1e-6, tmax=100, verbose=True)
 
 # ------------------------------------------------------------------------------
 # Plot results
 # ------------------------------------------------------------------------------
 
-if plot:
-    plotter = Plotter()
+plotter = Plotter()
 
-    # add topology diagram to scene
-    artist = plotter.add(topology,
-                         nodesize=0.2,
-                         nodetext="sequence",
-                         nodecolor="sequence",
-                         show_nodetext=True)
+# add topology diagram to scene
+artist = plotter.add(topology, nodesize=0.2, show_nodetext=True, show_edgetext=True)
 
 
-    # add shifted form diagram to the scene
-    # form = form.transformed(Translation.from_vector([0.0, -1.0, 0.0]))
-    # plotter.add(form, nodesize=0.2, show_edgetext=True, edgetext="sequence")
+# add shifted form diagram to the scene
+form = form.transformed(Translation.from_vector([0.0, -1.0, 0.0]))
+plotter.add(form, nodesize=0.2, show_nodetext=True, show_edgetext=True)
 
-    # show plotter contents
-    plotter.zoom_extents()
-    plotter.show()
+# show plotter contents
+plotter.zoom_extents()
+plotter.show()