GraphManipulation.py/main.py at master · LBeghini/GraphManipulation.py · 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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import sys
from PyQt5.QtWidgets import *
import numpy as np
import networkx as nx
from graph import Graph
import pyqtgraph as pg
from components.tool_bar import ToolBar


class Window(QMainWindow):
    def __init__(self):
        super(Window, self).__init__()
        self.pos = []
        self.adj = []
        self.texts = []

        self.toolbar = ToolBar()

        self.graph = Graph()

        self.nx_graph = None

        self.w = pg.GraphicsLayoutWidget(show=True)
        self.v = self.w.addViewBox()

        self.connect_events()
        self.set_up()

    def connect_events(self):
        self.graph.remove_signal.connect(self.remove)
        self.graph.change_position_signal.connect(self.update_pos)
        self.graph.canvas_clicked_signal.connect(self.onMouseClick)

    def update_pos(self, pos):
        for i in range(len(pos)):
            self.pos[i][0] = pos[i][0]
            self.pos[i][1] = pos[i][1]

    def set_up(self):
        self.addToolBar(self.toolbar)

        self.v.setAspectLocked()
        self.v.addItem(self.graph)

        self.setCentralWidget(self.w)

        self.transform_g6_in_graph("L??????????^~@")
        self.get_pos()
        self.get_adj()
        self.define_graph()

    def remove(self, id_pos=None, id_edge=None):
        if self.toolbar.remove_node.isChecked():
            if len(self.pos) == 1:
                return
            self.pos.pop(id_pos)
            for i in reversed(id_edge):
                self.adj.pop(i)
            self.nx_graph.remove_node(self.texts[id_pos])
            self.texts.pop(id_pos)
            self.update_graph_index(id_pos)
            print(self.texts)
            print(id_pos)
            self.define_graph()
            self.graph.createPen(len(self.pos))

    def update_graph_index(self, id_pos):
        for i in range(0, len(self.adj)):
            if self.adj[i][0] > id_pos:
                self.adj[i][0] -= 1

            if self.adj[i][1] > id_pos:
                self.adj[i][1] -= 1

    def get_pos(self):
        points = nx.drawing.layout.spring_layout(self.nx_graph)

        for i, point in enumerate(points.values()):
            aux = [point[0] * 10 // 1, point[1] * 10 // 1]
            self.pos.append(aux)
            self.texts.append(i)

    def get_adj(self):
        nx_edges = self.nx_graph.edges(data=True)
        for e in nx_edges:
            aux = [e[0], e[1]]
            if not reversed(aux) in self.adj:
                self.adj.append(aux)

    def transform_g6_in_graph(self, g6):
        self.nx_graph = nx.from_graph6_bytes(g6.encode('utf-8'))
        pass

    def define_graph(self):
        pos = np.array(self.pos, dtype=float)
        adj = np.array(self.adj)

        texts = ["%d" % i for i in self.texts]

        self.graph.setData(pos=pos, adj=adj, size=1, pxMode=False, text=texts)

    def onMouseClick(self, x, y):
        if self.toolbar.add_node.isChecked():
            self.pos.append([x, y])
            self.nx_graph.add_node(max(self.nx_graph.nodes) + 1)
            self.texts.append(max(self.nx_graph.nodes))
            self.define_graph()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = Window()
    window.show()
    app.exec_()