Node-Mapper/InteractiveNodeSystem.py at main · mineable3/Node-Mapper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
frommathimportsqrt
fromConstantsimportConstants
fromNodeimportNode
importtkinterastk
importthreading

classInteractiveNodeSystem:
def__init__(self, root: tk.Tk):
self.root=root
self.root.title("Interactive Node System")
self.root.geometry("=800x600+300+20")

self.canvas=tk.Canvas(root, width=800, height=600, bg="white")
self.canvas.pack(expand=tk.YES, fill=tk.BOTH)
self.canvas.bind("<Button-1>", self.create_node)
self.canvas.bind("<Button-2>", self.grab_node)
self.canvas.bind("<Button2-Motion>", self.drag_node, '+')
self.canvas.bind("<ButtonRelease-2>", self.clear_selected_nodes)
self.canvas.bind("<Button-3>", self.connect_nodes)

self.nodes=list[Node]()
self.selected_nodes=list[Node]()

defclear_selected_nodes(self, doNotUseThis=None):
self.selected_nodes.clear()

defcreate_node(self, event) ->Node:
new_node=Node(self.canvas, event.x, event.y, (0.0,0.0))
self.nodes.append(new_node)
returnnew_node

defgrab_node(self, event):
self.clear_selected_nodes()
self.selected_nodes.append(self.get_clicked_node(event))

defdrag_node(self, event):
ifself.selected_nodes[0] !=None:
self.selected_nodes[0].move(event.x, event.y)

defconnect_nodes(self, event):

clicked_node=self.get_clicked_node(event)
ifclicked_node:
self.selected_nodes.append(clicked_node)

iflen(self.selected_nodes) ==2andself.selected_nodes[0] !=self.selected_nodes[1]:
self.selected_nodes[0].connect(self.selected_nodes[1])
self.selected_nodes= []

if(len(self.selected_nodes) >1):
self.selected_nodes.pop(0)

defget_clicked_node(self, event):
fornodeinself.nodes:
coords=node.get_position_np()
distance=sqrt((event.x-coords[0]) **2+ (event.y-coords[1]) **2)
ifdistance<Constants.NODE_SIZE:
returnnode
returnNone

defvelocityMethod(self):
while(True):
fornodeinself.nodes:
if(self.selected_nodes.count(node) ==0):
node.step_velocity()

defmainloop(self):
velocityThread=threading.Thread(target=self.velocityMethod)
velocityThread.start()

self.root.mainloop()