graph_managers.py

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor

from graphic_items import Node, Edge


class GraphManager:
    # Parameters as container nodes and edge_matrix
    # edge matrix stores the connections check dummy data for example
    # nodes is a dictionary of cordinates as values mapped to its indices as keys
    # converted nodes to self.nodes which is dictionary of nodes class object mapped to indices as keys
    # converted edge_matrix to self.edges which is 2d matrix of edge class objects
    
    def __init__(self, container, nodes, edge_matrix):
        self.container = container
        self.temp_edge = None
        self.selectedItem = None
        self.penColor = None

        if len(nodes) > 0:
            self.currentKey = max([key for key in nodes]) + 1
        else:
            self.currentKey = 0

        self.nodes = {}
        self.edges = []
        for _ in range(self.currentKey):
            self.edges.append([None] * self.currentKey)

        for key, val in nodes.items():
            self.addNode(key, val)

        for row in edge_matrix:
            for edge in row:
                if edge:
                    self.addEdge(edge)


    # adds edge in gui and in stored matrix
    # any doubt contact arib

    def addEdge(self, edge, temp=False):
        if self.penColor:
            edge.setPenColor(self.penColor)
        if edge:
            self.container.addItem(edge)
        if not temp:
            self.edges[edge.fromNode.key][edge.toNode.key] = edge

    # removes edge in gui and stored matrix
    # any doubt contact sid

    def removeEdge(self, edge, temp=False):
        self.container.removeItem(edge)  
        if not temp:
            self.edges[edge.fromNode.key][edge.toNode.key] = None


    # adds edge in gui and in stored matrix
    # any doubt contact arib

    def addNode(self, key, node):
        if self.penColor:
            node.setPenColor(self.penColor)
        self.container.addItem(node)
        self.nodes[key] = node
        if key == self.currentKey:
            self.edges.append([None]*key)
            for row in self.edges:
                row.append(None)


    # removes node in gui and stored matrix and 
    # any doubt contact sid

    def removeNode(self, node):
        self.container.removeItem(node)
        self.nodes.pop(node.key)
        for edge in self.edges[node.key]:
            if edge:
                self.removeEdge(edge)
        for i in range(len(self.edges)):
            if self.edges[i][node.key]:
                self.removeEdge(self.edges[i][node.key])

    # any doubt contact arib
    def toggleItem(self, item):

        if self.selectedItem and item is self.selectedItem:
            self.selectedItem.clicked = False
            self.selectedItem = None
        elif self.selectedItem:
            self.selectedItem.clicked = False
            self.selectedItem = item
            self.selectedItem.clicked = True
        else:
            self.selectedItem = item
            self.selectedItem.clicked = True

    def setPenColor(self, penColor):
        self.penColor = penColor

    def mouseMoveEvent(self, event, item):

        mousePos = event.scenePos()
        if item and item.type == 'node':

            if self.temp_edge:
                # This is the end node
                self.temp_edge.setEnd(item.center.x(), item.center.y())
            else:
                # This is the start node
                self.temp_edge = Edge(item, None)
                self.addEdge(self.temp_edge, temp=True)

            self.container.update()

        elif self.temp_edge:
            self.temp_edge.setEnd(mousePos.x(), mousePos.y())
            self.container.update()

    # any doubt contact arib

    def mouseReleaseEvent(self, event, item):
        if item and item.type == 'node':

            # If we are drawing edge and mouse is released at a node, then add that edge
            if self.temp_edge and item is not self.temp_edge.fromNode:
                self.removeEdge(self.temp_edge, temp=True)
                newEdge = Edge(self.temp_edge.fromNode, item)
                self.addEdge(newEdge)
                self.temp_edge = None

        elif self.temp_edge:
            self.removeEdge(self.temp_edge, temp=True)
            self.temp_edge = None

        self.container.update()

    # any doubt contact arib

    def mousePressEvent(self, event, item):
        mousePos = event.scenePos()
        if item and item.type == 'node':

            # Remove node if right-button clicked
            if item is self.selectedItem and event.button() == Qt.RightButton:
                self.removeNode(item)

            if event.button() == Qt.LeftButton:
                self.toggleItem(item)

        elif item and item.type == 'edge':

            if item is self.selectedItem and event.button() == Qt.RightButton:
                self.removeEdge(item)

            if event.button() == Qt.LeftButton:
                self.toggleItem(item)

        else:

            # Add Node
            # node = Node(mousePos.x(), mousePos.y(), self.currentKey)
            # self.addNode(self.currentKey, node)
            # self.currentKey += 1

            # self.toggleItem(node)
            if self.selectedItem:
                self.selectedItem.clicked = False
                self.selectedItem = None

    def mouseDoubleClickEvent(self, event, item):
        mousePos = event.scenePos()
        if not item:
            # Add Node
            node = Node(mousePos.x(), mousePos.y(), self.currentKey)
            self.addNode(self.currentKey, node)
            self.currentKey += 1


    def getData(self):
        return self.nodes, self.edges

    def printMatrix(self):
        for row in self.edges:
            for edge in row:
                if edge:
                    print("1", end=" ")
                else:
                    print("0", end=" ")
            print()
        print()