Как изменить форму узлов pyqtgraph - PullRequest
Новая
       5

Как изменить форму узлов pyqtgraph

0 голосов
/ 15 июня 2019

В настоящее время я работаю с pyqtgraph и хочу изменить форму пользовательских узлов, не создавая совершенно новый файл NodeCustom.Насколько я понимаю, функция рисования в классе NodeGraphicsItem класса pyqtgraph.Node управляет внешним видом узлов.

Я пытаюсь отобразить узел эллипса.Я хочу переопределить функцию рисования pyqtgraph и использовать собственный метод drawEllipse. 

class EllipseNodeItem(Node):
    nodeName = "EllipseNode"

    def __init__(self, name):
        self.view = None
        # Initialize node with only a single input terminal
        n = Node.__init__(self, name, terminals={'data': {'io': 'in'}})
        #NodeGraphicsItem.__init__(self, n)

    # drawing node in shape of ellipse, fitting into rectangle 200x100
    def paint(self, p, *args):  # pyqt function
        self.bounds = QtCore.QRectF(0, 0, 200, 100)

        p.setPen(self.pen)
        if self.isSelected(): #or self in sm.selectedNodes:
            p.setPen(self.selectPen)
            p.setBrush(self.selectBrush)
        else:
            p.setPen(self.pen)
            if self.hovered:
                p.setBrush(self.hoverBrush)
            else:
                p.setBrush(self.brush)

        p.drawEllipse(self.bounds)

    def setView(self, view):  # setView must be called by the program
        self.view = view

    def process(self, data, display=True):
        if display and self.view is not None:
            # the 'data' argument is the value given to the 'data' terminal
            if data is None:
                self.view.setImage(np.zeros((1, 1)))  # give a blank array to clear the view
            else:
                self.view.setImage(data)

Он обрабатывает EllipseNode как отдельный выбираемый узел.Очевидно, что краска не используется импортированными классами Node и NodeGraphicsItem, но это моя цель.

1 Ответ

0 голосов
/ 15 июня 2019

Я думаю, вам нужно добавить 

def boundingRect(self):       
    return  QtCore.QRectF(0, 0, 200, 100)

Если это не ответ для вас, я удалю свой ответ ...

...