-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
17 lines (14 loc) · 811 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from bricks_modeling.file_IO.model_reader import read_bricks_from_file
from bricks_modeling.file_IO.model_writer import write_bricks_to_file
from bricks_modeling.connectivity_graph import ConnectivityGraph
from util.debugger import MyDebugger
from visualization.model_visualizer import visualize_3D
if __name__ == "__main__":
debugger = MyDebugger("test")
bricks = read_bricks_from_file("./debug/2021-11-02_23-30-39_heart/heart b=24 ['3024', '3020', '3023', '3710'] .ldr")
structure_graph = ConnectivityGraph(bricks)
# print(structure_graph.to_json())
# structure_graph.show()
points = [b.get_translation() for b in structure_graph.bricks]
edges = [e["node_indices"] for e in structure_graph.connect_edges]
visualize_3D(points, lego_bricks=bricks, edges=edges, show_axis=True)