|
|
# -*- encoding: utf-8 -*-
|
|
|
from head import *
|
|
|
|
|
|
class Node:
|
|
|
# 标记节点按钮
|
|
|
def node_mark_display(self):
|
|
|
for w in frame_right.winfo_children():
|
|
|
w.destroy()
|
|
|
# 清除右侧组件
|
|
|
curx = -1
|
|
|
y0 = 0.1
|
|
|
for i in d.Nodes:
|
|
|
if i[2] == 1:
|
|
|
curx = i[0]
|
|
|
lab = tk.Label(frame_right, text="节点操作", bg=RED, fg=BLACK, font=('微软雅黑', 20))
|
|
|
lab.place(relx=0.18, rely=0.02, width=200, height=30)
|
|
|
lab1 = tk.Label(frame_right, text=f"当前共有{len(d.Nodes)}个节点", bg=RED, fg=BLACK, font=('微软雅黑', 15))
|
|
|
lab1.place(relx=0.1, rely=0.0 + y0, width=200, height=30)
|
|
|
lab1 = tk.Label(frame_right, text="当前节点:", bg=RED, fg=BLACK, font=('微软雅黑', 12))
|
|
|
if curx == -1:
|
|
|
text = "未选择"
|
|
|
else:
|
|
|
text = f"{d.Nodes[curx][1]}"
|
|
|
|
|
|
lab2 = tk.Label(frame_right, text=text, bg=BLUE, fg=BLACK, font=('微软雅黑', 12))
|
|
|
lab1.place(relx=0.1, rely=0.05 + y0, width=80, height=30)
|
|
|
lab2.place(relx=0.4, rely=0.05 + y0, width=80, height=30)
|
|
|
B = tk.Button(frame_right, text="设置标签", command=lambda: self.node_rename())
|
|
|
B.place(relx=0.7, rely=0.05 + y0, width=80, height=30)
|
|
|
B = tk.Button(frame_right, text="删除当前节点", command=lambda: self.node_del_exact())
|
|
|
B.place(relx=0.1, rely=0.1 + y0, width=150, height=30)
|
|
|
if d.Nodes[curx][4] == 1:
|
|
|
text = '设为非必经节点'
|
|
|
else:
|
|
|
text = '设为必经节点'
|
|
|
B = tk.Button(frame_right, text=text, command=lambda: self.dominator(curx))
|
|
|
B.place(relx=0.1, rely=0.15 + y0, width=150, height=30)
|
|
|
B = tk.Button(frame_right, text="增加一个节点", command=lambda: self.node_add())
|
|
|
B.place(relx=0.1, rely=0.25 + y0, width=150, height=30)
|
|
|
B = tk.Button(frame_right, text="减少一个节点", command=lambda: self.node_del())
|
|
|
B.place(relx=0.1, rely=0.3 + y0, width=150, height=30)
|
|
|
|
|
|
def dominator(self, x):
|
|
|
d.Nodes[x][4] ^= 1
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_mark(self, name):
|
|
|
if name >= len(d.Nodes):
|
|
|
self.node_404(name)
|
|
|
return
|
|
|
|
|
|
for node in d.Nodes:
|
|
|
if node[0] == name:
|
|
|
node[2] = 1
|
|
|
else:
|
|
|
node[2] = 0
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_del_exact(self):
|
|
|
if d.nodes_num <= 2:# 若删除节点以后节点过少，则直接返回并警告
|
|
|
tk.messagebox.askokcancel(title='错误', message='节点过少')
|
|
|
return
|
|
|
flag = 0
|
|
|
cur = -1
|
|
|
for node in d.Nodes:# 寻找当前选定的节点
|
|
|
if node[2] == 1:
|
|
|
cur = node[0]
|
|
|
flag = 1
|
|
|
if flag == 0:
|
|
|
tk.messagebox.askokcancel(title='错误', message='未选择节点')
|
|
|
return
|
|
|
if d.source == 1:# 如果是均匀分布，则重新计算点的坐标
|
|
|
d.nodes_num -= 1#节点数目减一
|
|
|
d.ang = 360 / d.nodes_num#重新计算圆心角
|
|
|
nodes = d.Nodes.copy()#复制节点
|
|
|
d.Nodes.clear()#清除Data中的节点
|
|
|
num = 0
|
|
|
for n in nodes:#逐个复制节点
|
|
|
if n[0] == cur:#若是删除的节点，则不复制
|
|
|
continue
|
|
|
n[0] = num#重新编号
|
|
|
rad = math.radians(num * d.ang)#计算当前角度
|
|
|
x = int(math.cos(rad) * d.R)#计算当前点x坐标（相对）
|
|
|
y = int(math.sin(rad) * d.R)#计算当前点y坐标（相对）
|
|
|
n[3] = (d.center[0] + x, d.center[1] + y)#计算绝对坐标
|
|
|
d.Nodes.append(n)#将当前点加入点集合中
|
|
|
num += 1
|
|
|
edges = d.Edges.copy()#将原有边集复制出来
|
|
|
d.Edges.clear()#删除Data中的边集
|
|
|
num = 0
|
|
|
for e in edges:#复制边
|
|
|
if e[1] == cur or e[2] == cur:#删除的边不复制
|
|
|
continue
|
|
|
if e[1] > cur:#修改边对应的点的编号
|
|
|
e[1] -= 1
|
|
|
if e[2] > cur:#修改边对应的点的编号
|
|
|
e[2] -= 1
|
|
|
e[0] = num#重新编号
|
|
|
d.Edges.append(e)#将边加入边集中
|
|
|
num += 1
|
|
|
# 如果是按照经纬度分布，则重新计算比例尺
|
|
|
elif d.source == 2:
|
|
|
name = []
|
|
|
for i in range(len(d.name)):
|
|
|
if not i == cur:
|
|
|
name.append(d.name[i])
|
|
|
d.name = name.copy()
|
|
|
d.nodes_num -= 1
|
|
|
|
|
|
d.coordinate = d.coord_creat() # 纯节点坐标
|
|
|
d.Nodes = d.JI_nodes_creat() # 节点列表
|
|
|
d.Edges = d.JI_edges_creat()
|
|
|
d.JI_edge_add(2)
|
|
|
d.JI_edge_add(3)
|
|
|
|
|
|
elif d.source == 3:
|
|
|
name = []
|
|
|
for i in range(len(d.name)):
|
|
|
if not i == cur:
|
|
|
name.append(d.name[i])
|
|
|
d.name = name.copy()
|
|
|
d.nodes_num -= 1
|
|
|
|
|
|
d.coordinate = d.coord_creat() # 纯节点坐标
|
|
|
d.Nodes = d.LIAO_nodes_creat() # 节点列表
|
|
|
d.Edges = d.LIAO_edges_creat()
|
|
|
d.LIAO_edge_add(2)
|
|
|
d.LIAO_edge_add(3)
|
|
|
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_rename(self):
|
|
|
flag = 0
|
|
|
for node in d.Nodes:
|
|
|
if node[2] == 1:
|
|
|
cur = node
|
|
|
flag = 1
|
|
|
if flag == 1:
|
|
|
result = tkinter.simpledialog.askstring(title='修改节点标签', prompt='请输入修改后的标签：', initialvalue=cur[1])
|
|
|
else:
|
|
|
tk.messagebox.askokcancel(title='错误', message='未选择节点')
|
|
|
return
|
|
|
for node in d.Nodes: # 检查新的名称是否已经存在
|
|
|
if node[1] == result:
|
|
|
return "Error: name already exists"
|
|
|
if result:
|
|
|
cur[1] = result
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_clear(self):
|
|
|
for node in d.Nodes:
|
|
|
node[2] = False
|
|
|
G.draw()
|
|
|
|
|
|
|
|
|
|
|
|
def node_add(self):
|
|
|
# 增加一个节点
|
|
|
nodes = len(d.Nodes) + 1
|
|
|
# 如果节点数目大于五，则将连接详细信息改为不显示
|
|
|
if nodes > 5:
|
|
|
d.edgeinfo = 0
|
|
|
self.node_init(nodes)
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_set(self):
|
|
|
node = tkinter.simpledialog.askinteger(title='修改节点数目', prompt='请输入修改后的节点数目：',
|
|
|
initialvalue=str(len(d.Nodes)))
|
|
|
if node:
|
|
|
self.node_init(node)
|
|
|
G.draw()
|
|
|
|
|
|
def node_del(self):
|
|
|
if d.nodes_num <= 2:
|
|
|
tk.messagebox.askokcancel(title='错误', message='节点过少')
|
|
|
return
|
|
|
nodes = len(d.Nodes) - 1
|
|
|
if nodes < 6:
|
|
|
d.edgeinfo = 1
|
|
|
self.node_init(nodes)
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_init(self, node):
|
|
|
if not node:
|
|
|
return
|
|
|
if d.source == 1:
|
|
|
d.data(node)
|
|
|
elif d.source == 2:
|
|
|
if node > 13:
|
|
|
tk.messagebox.askokcancel(title='错误', message='节点过多')
|
|
|
return
|
|
|
d.HEI_data(node)
|
|
|
else:
|
|
|
if node > 13:
|
|
|
tk.messagebox.askokcancel(title='错误', message='节点过多')
|
|
|
return
|
|
|
d.LIAO_data(node)
|
|
|
if node > 5:
|
|
|
d.edgeinfo = 0
|
|
|
else:
|
|
|
d.edgeinfo = 1
|
|
|
|
|
|
def graph_del(self):
|
|
|
global d
|
|
|
del d
|
|
|
|
|
|
|
|
|
|
|
|
def node_refresh(self):
|
|
|
nodes = len(d.Nodes)
|
|
|
self.node_init(nodes)
|
|
|
G.draw()
|
|
|
self.node_mark_display()
|
|
|
|
|
|
def node_404(self, name):
|
|
|
# 弹出对话框
|
|
|
tk.messagebox.askokcancel(title='错误', message='该节点不存在')
|
|
|
# print(result)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
G.draw()
|
|
|
node = Node()
|
|
|
node.node_mark_display()
|
|
|
mainloop() |
