def mouseDoubleClickEvent(self, event):
if self.__inspectorWidget is None:
parentWidget = self.getGraph().getGraphViewWidget()
self.__inspectorWidget = BackdropInspector(parent=parentWidget,
nodeItem=self)
result = self.__inspectorWidget.exec_()
else:
self.__inspectorWidget.setFocus()
super(KBackdrop, self).mouseDoubleClickEvent(event)
def mouseDoubleClickEvent(self, event):
if self.__inspectorWidget is None:
parentWidget = self.getGraph().getGraphViewWidget()
self.__inspectorWidget = BackdropInspector(parent=parentWidget, nodeItem=self)
result = self.__inspectorWidget.exec_()
else:
self.__inspectorWidget.setFocus()
super(KBackdrop, self).mouseDoubleClickEvent(event)
class KBackdrop(QtGui.QGraphicsWidget):
nameChanged = QtCore.Signal(str, str)
sizeChanged = QtCore.Signal(float)
__defaultColor = QtGui.QColor(65, 120, 122, 255)
__defaultUnselectedColor = QtGui.QColor(__defaultColor.darker(125))
__defaultSelectedColor = QtGui.QColor(__defaultColor.lighter(175))
__defaultHoverColor = QtGui.QColor(__defaultColor.lighter(110))
__defaultUnselectedPen = QtGui.QPen(__defaultUnselectedColor, 1.6)
__defaultSelectedPen = QtGui.QPen(__defaultSelectedColor, 1.6)
__defaultHoveredPen = QtGui.QPen(__defaultHoverColor, 1.6)
__defaultLinePen = QtGui.QPen(QtGui.QColor(25, 25, 25, 255), 1.25)
__resizeDistance = 16.0
__setCustomCursor = False
__hoveredOver = False
def __init__(self, graph, name):
super(KBackdrop, self).__init__()
self.setAcceptHoverEvents(True)
self.__name = name
self.__comment = None
self.__graph = graph
self.__color = self.__defaultColor
self.__color.setAlpha(25)
self.__unselectedColor = self.__defaultUnselectedColor
self.__selectedColor = self.__defaultSelectedColor
self.__hoverColor = self.__defaultHoverColor
self.__unselectedPen = QtGui.QPen(self.__defaultUnselectedPen)
self.__selectedPen = QtGui.QPen(self.__defaultSelectedPen)
self.__hoveredPen = QtGui.QPen(self.__defaultHoveredPen)
self.__linePen = QtGui.QPen(self.__defaultLinePen)
self.__inspectorWidget = None
self.setMinimumWidth(120)
self.setMinimumHeight(80)
self.setZValue(-100)
# Set defaults for interactions
self.__selected = False
self.__dragging = False
self.__resizing = False
self.__resizeCorner = -1
self.__resizeEdge = -1
self.createLayout()
self.createConnections()
# Initialize the comment with the name
self.setComment(name)
def createLayout(self):
layout = QtGui.QGraphicsLinearLayout()
layout.setContentsMargins(5, 0, 5, 7)
layout.setSpacing(7)
layout.setOrientation(QtCore.Qt.Vertical)
self.setLayout(layout)
self.__headerItem = KBackdropHeader(self.__name, self)
self.__titleWidget = self.__headerItem.getTitleWidget()
layout.addItem(self.__headerItem)
layout.setAlignment(self.__headerItem,
QtCore.Qt.AlignCenter | QtCore.Qt.AlignTop)
layout.addStretch(1)
def createConnections(self):
self.sizeChanged.connect(self.__titleWidget.nodeResized)
def getName(self):
return self.__name
def setName(self, name):
if name != self.__name:
origName = self.__name
self.__name = name
# self.__headerItem.setText(self.__name)
# Emit an event, so that the graph can update itsself.
self.nameChanged.emit(origName, name)
# Update the node so that the size is computed.
self.adjustSize()
def getComment(self):
return self.__comment
def setComment(self, comment):
self.__comment = comment
self.__headerItem.setText(comment)
# Resize the width of the backdrop based on title width
titleWidget = self.__headerItem.getTitleWidget()
titleWidth = titleWidget.textSize().width()
self.resize(titleWidth, self.size().height())
# Update the node so that the size is computed.
self.adjustSize()
def getColor(self):
color = QtGui.QColor(self.__color)
color.setAlpha(255)
return color
def setColor(self, color):
self.__color = color
self.__color.setAlpha(25)
self.update()
def getUnselectedColor(self):
return self.__unselectedColor
def setUnselectedColor(self, color):
self.__unselectedColor = color
self.__unselectedPen.setColor(self.__unselectedColor)
self.update()
def getSelectedColor(self):
return self.__selectedColor
def setSelectedColor(self, color):
self.__selectedColor = color
self.__selectedPen.setColor(self.__selectedColor)
self.update()
def getHoveredColor(self):
return self.__hoverColor
def setHoveredColor(self, color):
self.__hoverColor = color
self.__hoveredPen.setColor(self.__hoverColor)
self.update()
def getGraph(self):
return self.__graph
def getHeader(self):
return self.__headerItem
# ==========
# Selection
# ==========
def isSelected(self):
return self.__selected
def setSelected(self, selected=True):
self.__selected = selected
self.update()
# ==========
# Graph Pos
# ==========
def getGraphPos(self):
transform = self.transform()
size = self.size()
return QtCore.QPointF(transform.dx() + (size.width() * 0.5),
transform.dy() + (size.height() * 0.5))
def setGraphPos(self, graphPos):
size = self.size()
self.setTransform(
QtGui.QTransform.fromTranslate(
graphPos.x() - (size.width() * 0.5),
graphPos.y() - (size.height() * 0.5)), False)
def translate(self, x, y):
super(KBackdrop, self).translate(x, y)
def paint(self, painter, option, widget):
rect = self.windowFrameRect()
painter.setBrush(self.__color)
painter.setPen(QtGui.QPen(QtGui.QColor(0, 0, 0, 0), 0))
roundingY = 20.0 / (rect.height() / 80.0)
roundingX = rect.height() / rect.width() * roundingY
painter.drawRoundRect(rect,
roundingX,
roundingY,
mode=QtCore.Qt.AbsoluteSize)
# Title BG
titleHeight = self.__headerItem.size().height() - 3
darkerColor = self.__color.darker(125)
darkerColor.setAlpha(255)
painter.setBrush(darkerColor)
roundingYHeader = rect.width() * roundingX / titleHeight
painter.drawRoundRect(0, 0, rect.width(), titleHeight, roundingX,
roundingYHeader)
painter.drawRect(0, titleHeight * 0.5 + 2, rect.width(),
titleHeight * 0.5)
painter.setBrush(QtGui.QColor(0, 0, 0, 0))
if self.__selected:
painter.setPen(self.__selectedPen)
elif self.__hoveredOver:
painter.setPen(self.__hoveredPen)
else:
painter.setPen(self.__unselectedPen)
painter.drawRoundRect(rect,
roundingX,
roundingY,
mode=QtCore.Qt.AbsoluteSize)
super(KBackdrop, self).paint(painter, option, widget)
# =======
# Events
# =======
def mousePressEvent(self, event):
if event.button() is QtCore.Qt.MouseButton.LeftButton:
resizeCorner = self.getCorner(event.pos())
resizeEdge = self.getEdge(event.pos())
if resizeCorner != -1 and self.isSelected():
self.__resizing = True
self.__resizeCorner = resizeCorner
self._resizedBackdrop = False
elif resizeEdge != -1 and self.isSelected():
self.__resizing = True
self.__resizeEdge = resizeEdge
self._resizedBackdrop = False
else:
modifiers = event.modifiers()
if modifiers == QtCore.Qt.ControlModifier:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=False)
else:
self.__graph.deselectNode(self)
elif modifiers == QtCore.Qt.ShiftModifier:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=False)
else:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=True)
if self.__headerItem.contains(event.pos()):
self.setCursor(QtCore.Qt.ClosedHandCursor)
self.__dragging = True
self._nodesMoved = False
self._mouseDownPoint = self.mapToScene(event.pos())
self._mouseDelta = self._mouseDownPoint - self.getGraphPos()
self._lastDragPoint = self._mouseDownPoint
self._initPos = self.pos()
self._initScenePos = self.mapToScene(self.pos())
self._initBoundingRect = self.boundingRect()
self._initSceneBoundingRect = self.sceneBoundingRect()
else:
super(KBackdrop, self).mousePressEvent(event)
def mouseMoveEvent(self, event):
if self.__dragging:
newPos = self.mapToScene(event.pos())
graph = self.getGraph()
if graph.getSnapToGrid() is True:
gridSize = graph.getGridSize()
newNodePos = newPos - self._mouseDelta
snapPosX = math.floor(newNodePos.x() / gridSize) * gridSize
snapPosY = math.floor(newNodePos.y() / gridSize) * gridSize
snapPos = QtCore.QPointF(snapPosX, snapPosY)
newPosOffset = snapPos - newNodePos
newPos = newPos + newPosOffset
delta = newPos - self._lastDragPoint
self.__graph.moveSelectedNodes(delta)
self._lastDragPoint = newPos
self._nodesMoved = True
elif self.__resizing:
newPos = self.mapToScene(event.pos())
delta = newPos - self._mouseDownPoint
self._resizedBackdrop = True
newPosX = 0
newPosY = 0
newWidth = self._initBoundingRect.width()
newHeight = self._initBoundingRect.height()
if self.__resizeCorner == 0:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height() + (delta.y() *
-1.0)
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y() + delta.y()
elif self.__resizeCorner == 1:
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height() + (delta.y() *
-1.0)
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y() + delta.y()
elif self.__resizeCorner == 2:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height() + delta.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y()
elif self.__resizeCorner == 3:
newPosX = self._initPos.x()
newPosY = self._initPos.y()
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height() + delta.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
elif self.__resizeEdge == 0:
pass
elif self.__resizeEdge == 1:
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height()
newPosY = self._initPos.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x()
elif self.__resizeEdge == 2:
newWidth = self._initBoundingRect.width()
newHeight = self._initBoundingRect.height() + delta.y()
newPosX = self._initPos.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y()
elif self.__resizeEdge == 3:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height()
newPosY = self._initPos.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
self.setPos(newPosX, newPosY)
self.resize(newWidth, newHeight)
self.sizeChanged.emit(newWidth)
self.prepareGeometryChange()
else:
super(KBackdrop, self).mouseMoveEvent(event)
def mouseReleaseEvent(self, event):
if self.__dragging:
if self._nodesMoved:
newPos = self.mapToScene(event.pos())
delta = newPos - self._mouseDownPoint
self.__graph.endMoveSelectedNodes(delta)
self.setCursor(QtCore.Qt.ArrowCursor)
self.__dragging = False
elif self.__resizing:
self.setCursor(QtCore.Qt.ArrowCursor)
self.__resizing = False
self.__resizeCorner = -1
else:
super(KBackdrop, self).mouseReleaseEvent(event)
def mouseDoubleClickEvent(self, event):
if self.__inspectorWidget is None:
parentWidget = self.getGraph().getGraphViewWidget()
self.__inspectorWidget = BackdropInspector(parent=parentWidget,
nodeItem=self)
result = self.__inspectorWidget.exec_()
else:
self.__inspectorWidget.setFocus()
super(KBackdrop, self).mouseDoubleClickEvent(event)
def hoverEnterEvent(self, event):
self.__hoveredOver = True
self.update()
def hoverMoveEvent(self, event):
resizeCorner = self.getCorner(event.pos())
resizeEdge = self.getEdge(event.pos())
if resizeCorner == 0:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeFDiagCursor)
elif resizeCorner == 1:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeBDiagCursor)
elif resizeCorner == 2:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeBDiagCursor)
elif resizeCorner == 3:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeFDiagCursor)
elif resizeEdge == 0:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeVerCursor)
elif resizeEdge == 1:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeHorCursor)
elif resizeEdge == 2:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeVerCursor)
elif resizeEdge == 3:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeHorCursor)
elif self.__headerItem.contains(event.pos()) is True:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.OpenHandCursor)
else:
if self.__setCustomCursor is True:
self.setCursor(QtCore.Qt.ArrowCursor)
def hoverLeaveEvent(self, event):
self.setCursor(QtCore.Qt.ArrowCursor)
self.__hoveredOver = False
self.update()
# =============
# Misc Methods
# =============
def getCorner(self, pos):
topLeft = self.mapFromItem(self, self.boundingRect().topLeft())
bottomRight = self.mapFromItem(self, self.boundingRect().bottomRight())
rect = QtCore.QRectF(topLeft, bottomRight)
if (rect.topLeft() - pos).manhattanLength() < self.__resizeDistance:
return 0
elif (rect.topRight() - pos).manhattanLength() < self.__resizeDistance:
return 1
elif (rect.bottomLeft() -
pos).manhattanLength() < self.__resizeDistance:
return 2
elif (rect.bottomRight() -
pos).manhattanLength() < self.__resizeDistance:
return 3
return -1
def getEdge(self, pos):
topRectUpperLeft = self.mapFromItem(self,
self.boundingRect().topLeft())
topRectLowerRight = self.mapFromItem(self,
self.boundingRect().right(),
self.__resizeDistance * 0.25)
topRect = QtCore.QRectF(topRectUpperLeft, topRectLowerRight)
rightRectUpperLeft = self.mapFromItem(
self,
self.boundingRect().right() - self.__resizeDistance * 0.25, 0)
rightRectLowerRight = self.mapFromItem(
self,
self.boundingRect().bottomRight())
rightRect = QtCore.QRectF(rightRectUpperLeft, rightRectLowerRight)
bottomRectUpperLeft = self.mapFromItem(
self, 0,
self.boundingRect().bottom() - self.__resizeDistance * 0.25)
bottomRectLowerRight = self.mapFromItem(
self,
self.boundingRect().bottomRight())
bottomRect = QtCore.QRectF(bottomRectUpperLeft, bottomRectLowerRight)
leftRectUpperLeft = self.mapFromItem(self,
self.boundingRect().topLeft())
leftRectLowerRight = self.mapFromItem(
self,
self.boundingRect().left() + self.__resizeDistance * 0.25,
self.boundingRect().bottom())
leftRect = QtCore.QRectF(leftRectUpperLeft, leftRectLowerRight)
if topRect.contains(pos):
return -1 # Disabling the top resize by default.
elif rightRect.contains(pos):
return 1
elif bottomRect.contains(pos):
return 2
elif leftRect.contains(pos):
return 3
return -1
def inspectorClosed(self):
self.__inspectorWidget = None
# ==========
# Shut Down
# ==========
def disconnectAllPorts(self):
pass
# =============
# Data Methods
# =============
def getData(self):
"""Gets the essential data of the backdrop for saving.
Returns:
dict: Data for saving.
"""
data = {
'name': self.getName(),
'comment': self.getComment(),
'graphPos': self.getGraphPos().toTuple(),
'size': self.size().toTuple(),
'color': self.getColor().toTuple()
}
return data
def setData(self, data):
"""Sets the data on a backdrop after loading.
Args:
data (dict): Name, comment, graph pos, size, and color.
Returns:
bool: True if successful.
"""
self.setComment(data.get('comment', ''))
size = data.get('size', (self.minimumWidth(), self.minimumHeight()))
self.resize(size[0], size[1])
position = data.get('graphPos', (0, 0))
self.setGraphPos(QtCore.QPointF(position[0], position[1]))
color = data.get('color', self.__defaultColor.toTuple())
self.setColor(
color=QtGui.QColor(color[0], color[1], color[2], color[3]))
self.setUnselectedColor(self.getColor().darker(125))
self.setSelectedColor(self.getColor().lighter(175))
self.setHoveredColor(self.getColor().lighter(110))
return True
class KBackdrop(QtGui.QGraphicsWidget):
nameChanged = QtCore.Signal(str, str)
sizeChanged = QtCore.Signal(float)
__defaultColor = QtGui.QColor(65, 120, 122, 255)
__defaultUnselectedColor = QtGui.QColor(__defaultColor.darker(125))
__defaultSelectedColor = QtGui.QColor(__defaultColor.lighter(175))
__defaultHoverColor = QtGui.QColor(__defaultColor.lighter(110))
__defaultUnselectedPen = QtGui.QPen(__defaultUnselectedColor, 1.6)
__defaultSelectedPen = QtGui.QPen(__defaultSelectedColor, 1.6)
__defaultHoveredPen = QtGui.QPen(__defaultHoverColor, 1.6)
__defaultLinePen = QtGui.QPen(QtGui.QColor(25, 25, 25, 255), 1.25)
__resizeDistance = 16.0
__setCustomCursor = False
__hoveredOver = False
def __init__(self, graph, name):
super(KBackdrop, self).__init__()
self.setAcceptHoverEvents(True)
self.__name = name
self.__comment = None
self.__graph = graph
self.__color = self.__defaultColor
self.__color.setAlpha(25)
self.__unselectedColor = self.__defaultUnselectedColor
self.__selectedColor = self.__defaultSelectedColor
self.__hoverColor = self.__defaultHoverColor
self.__unselectedPen = QtGui.QPen(self.__defaultUnselectedPen)
self.__selectedPen = QtGui.QPen(self.__defaultSelectedPen)
self.__hoveredPen = QtGui.QPen(self.__defaultHoveredPen)
self.__linePen = QtGui.QPen(self.__defaultLinePen)
self.__inspectorWidget = None
self.setMinimumWidth(120)
self.setMinimumHeight(80)
self.setZValue(-100)
# Set defaults for interactions
self.__selected = False
self.__dragging = False
self.__resizing = False
self.__resizeCorner = -1
self.__resizeEdge = -1
self.createLayout()
self.createConnections()
# Initialize the comment with the name
self.setComment(name)
def createLayout(self):
layout = QtGui.QGraphicsLinearLayout()
layout.setContentsMargins(5, 0, 5, 7)
layout.setSpacing(7)
layout.setOrientation(QtCore.Qt.Vertical)
self.setLayout(layout)
self.__headerItem = KBackdropHeader(self.__name, self)
self.__titleWidget = self.__headerItem.getTitleWidget()
layout.addItem(self.__headerItem)
layout.setAlignment(self.__headerItem, QtCore.Qt.AlignCenter | QtCore.Qt.AlignTop)
layout.addStretch(1)
def createConnections(self):
self.sizeChanged.connect(self.__titleWidget.nodeResized)
def getName(self):
return self.__name
def setName(self, name):
if name != self.__name:
origName = self.__name
self.__name = name
# self.__headerItem.setText(self.__name)
# Emit an event, so that the graph can update itsself.
self.nameChanged.emit(origName, name)
# Update the node so that the size is computed.
self.adjustSize()
def getComment(self):
return self.__comment
def setComment(self, comment):
self.__comment = comment
self.__headerItem.setText(comment)
# Resize the width of the backdrop based on title width
titleWidget = self.__headerItem.getTitleWidget()
titleWidth = titleWidget.textSize().width()
self.resize(titleWidth, self.size().height())
# Update the node so that the size is computed.
self.adjustSize()
def getColor(self):
color = QtGui.QColor(self.__color)
color.setAlpha(255)
return color
def setColor(self, color):
self.__color = color
self.__color.setAlpha(25)
self.update()
def getUnselectedColor(self):
return self.__unselectedColor
def setUnselectedColor(self, color):
self.__unselectedColor = color
self.__unselectedPen.setColor(self.__unselectedColor)
self.update()
def getSelectedColor(self):
return self.__selectedColor
def setSelectedColor(self, color):
self.__selectedColor = color
self.__selectedPen.setColor(self.__selectedColor)
self.update()
def getHoveredColor(self):
return self.__hoverColor
def setHoveredColor(self, color):
self.__hoverColor = color
self.__hoveredPen.setColor(self.__hoverColor)
self.update()
def getGraph(self):
return self.__graph
def getHeader(self):
return self.__headerItem
# ==========
# Selection
# ==========
def isSelected(self):
return self.__selected
def setSelected(self, selected=True):
self.__selected = selected
self.update()
# ==========
# Graph Pos
# ==========
def getGraphPos(self):
transform = self.transform()
size = self.size()
return QtCore.QPointF(transform.dx()+(size.width()*0.5), transform.dy()+(size.height()*0.5))
def setGraphPos(self, graphPos):
size = self.size()
self.setTransform(QtGui.QTransform.fromTranslate(graphPos.x()-(size.width()*0.5), graphPos.y()-(size.height()*0.5)), False)
def translate(self, x, y):
super(KBackdrop, self).translate(x, y)
def paint(self, painter, option, widget):
rect = self.windowFrameRect()
painter.setBrush(self.__color)
painter.setPen(QtGui.QPen(QtGui.QColor(0, 0, 0, 0), 0))
roundingY = 20.0 / (rect.height() / 80.0)
roundingX = rect.height() / rect.width() * roundingY
painter.drawRoundRect(rect, roundingX, roundingY, mode=QtCore.Qt.AbsoluteSize)
# Title BG
titleHeight = self.__headerItem.size().height() - 3
darkerColor = self.__color.darker(125)
darkerColor.setAlpha(255)
painter.setBrush(darkerColor)
roundingYHeader = rect.width() * roundingX / titleHeight
painter.drawRoundRect(0, 0, rect.width(), titleHeight, roundingX, roundingYHeader)
painter.drawRect(0, titleHeight * 0.5 + 2, rect.width(), titleHeight * 0.5)
painter.setBrush(QtGui.QColor(0, 0, 0, 0))
if self.__selected:
painter.setPen(self.__selectedPen)
elif self.__hoveredOver:
painter.setPen(self.__hoveredPen)
else:
painter.setPen(self.__unselectedPen)
painter.drawRoundRect(rect, roundingX, roundingY, mode=QtCore.Qt.AbsoluteSize)
super(KBackdrop, self).paint(painter, option, widget)
# =======
# Events
# =======
def mousePressEvent(self, event):
if event.button() is QtCore.Qt.MouseButton.LeftButton:
resizeCorner = self.getCorner(event.pos())
resizeEdge = self.getEdge(event.pos())
if resizeCorner != -1 and self.isSelected():
self.__resizing = True
self.__resizeCorner = resizeCorner
self._resizedBackdrop = False
elif resizeEdge != -1 and self.isSelected():
self.__resizing = True
self.__resizeEdge = resizeEdge
self._resizedBackdrop = False
else:
modifiers = event.modifiers()
if modifiers == QtCore.Qt.ControlModifier:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=False)
else:
self.__graph.deselectNode(self)
elif modifiers == QtCore.Qt.ShiftModifier:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=False)
else:
if not self.isSelected():
self.__graph.selectNode(self, clearSelection=True)
if self.__headerItem.contains(event.pos()):
self.setCursor(QtCore.Qt.ClosedHandCursor)
self.__dragging = True
self._nodesMoved = False
self._mouseDownPoint = self.mapToScene(event.pos())
self._mouseDelta = self._mouseDownPoint - self.getGraphPos()
self._lastDragPoint = self._mouseDownPoint
self._initPos = self.pos()
self._initScenePos = self.mapToScene(self.pos())
self._initBoundingRect = self.boundingRect()
self._initSceneBoundingRect = self.sceneBoundingRect()
else:
super(KBackdrop, self).mousePressEvent(event)
def mouseMoveEvent(self, event):
if self.__dragging:
newPos = self.mapToScene(event.pos())
graph = self.getGraph()
if graph.getSnapToGrid() is True:
gridSize = graph.getGridSize()
newNodePos = newPos - self._mouseDelta
snapPosX = math.floor(newNodePos.x() / gridSize) * gridSize;
snapPosY = math.floor(newNodePos.y() / gridSize) * gridSize;
snapPos = QtCore.QPointF(snapPosX, snapPosY)
newPosOffset = snapPos - newNodePos
newPos = newPos + newPosOffset
delta = newPos - self._lastDragPoint
self.__graph.moveSelectedNodes(delta)
self._lastDragPoint = newPos
self._nodesMoved = True
elif self.__resizing:
newPos = self.mapToScene(event.pos())
delta = newPos - self._mouseDownPoint
self._resizedBackdrop = True
newPosX = 0
newPosY = 0
newWidth = self._initBoundingRect.width()
newHeight = self._initBoundingRect.height()
if self.__resizeCorner == 0:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height() + (delta.y() * -1.0)
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y() + delta.y()
elif self.__resizeCorner == 1:
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height() + (delta.y() * -1.0)
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y() + delta.y()
elif self.__resizeCorner == 2:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height() + delta.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y()
elif self.__resizeCorner == 3:
newPosX = self._initPos.x()
newPosY = self._initPos.y()
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height() + delta.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
elif self.__resizeEdge == 0:
pass
elif self.__resizeEdge == 1:
newWidth = self._initBoundingRect.width() + delta.x()
newHeight = self._initBoundingRect.height()
newPosY = self._initPos.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x()
elif self.__resizeEdge == 2:
newWidth = self._initBoundingRect.width()
newHeight = self._initBoundingRect.height() + delta.y()
newPosX = self._initPos.x()
if newHeight <= self.minimumHeight():
newHeight = self.minimumHeight()
else:
newPosY = self._initPos.y()
elif self.__resizeEdge == 3:
newWidth = self._initBoundingRect.width() + (delta.x() * -1.0)
newHeight = self._initBoundingRect.height()
newPosY = self._initPos.y()
if newWidth <= self.minimumWidth():
newWidth = self.minimumWidth()
else:
newPosX = self._initPos.x() + delta.x()
self.setPos(newPosX, newPosY)
self.resize(newWidth, newHeight)
self.sizeChanged.emit(newWidth)
self.prepareGeometryChange()
else:
super(KBackdrop, self).mouseMoveEvent(event)
def mouseReleaseEvent(self, event):
if self.__dragging:
if self._nodesMoved:
newPos = self.mapToScene(event.pos())
delta = newPos - self._mouseDownPoint
self.__graph.endMoveSelectedNodes(delta)
self.setCursor(QtCore.Qt.ArrowCursor)
self.__dragging = False
elif self.__resizing:
self.setCursor(QtCore.Qt.ArrowCursor)
self.__resizing = False
self.__resizeCorner = -1
else:
super(KBackdrop, self).mouseReleaseEvent(event)
def mouseDoubleClickEvent(self, event):
if self.__inspectorWidget is None:
parentWidget = self.getGraph().getGraphViewWidget()
self.__inspectorWidget = BackdropInspector(parent=parentWidget, nodeItem=self)
result = self.__inspectorWidget.exec_()
else:
self.__inspectorWidget.setFocus()
super(KBackdrop, self).mouseDoubleClickEvent(event)
def hoverEnterEvent(self, event):
self.__hoveredOver = True
self.update()
def hoverMoveEvent(self, event):
resizeCorner = self.getCorner(event.pos())
resizeEdge = self.getEdge(event.pos())
if resizeCorner == 0:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeFDiagCursor)
elif resizeCorner == 1:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeBDiagCursor)
elif resizeCorner == 2:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeBDiagCursor)
elif resizeCorner == 3:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeFDiagCursor)
elif resizeEdge == 0:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeVerCursor)
elif resizeEdge == 1:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeHorCursor)
elif resizeEdge == 2:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeVerCursor)
elif resizeEdge == 3:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.SizeHorCursor)
elif self.__headerItem.contains(event.pos()) is True:
self.__setCustomCursor = True
self.setCursor(QtCore.Qt.OpenHandCursor)
else:
if self.__setCustomCursor is True:
self.setCursor(QtCore.Qt.ArrowCursor)
def hoverLeaveEvent(self, event):
self.setCursor(QtCore.Qt.ArrowCursor)
self.__hoveredOver = False
self.update()
# =============
# Misc Methods
# =============
def getCorner(self, pos):
topLeft = self.mapFromItem(self, self.boundingRect().topLeft())
bottomRight = self.mapFromItem(self, self.boundingRect().bottomRight())
rect = QtCore.QRectF(topLeft, bottomRight)
if (rect.topLeft() - pos).manhattanLength() < self.__resizeDistance:
return 0
elif (rect.topRight() - pos).manhattanLength() < self.__resizeDistance:
return 1
elif (rect.bottomLeft() - pos).manhattanLength() < self.__resizeDistance:
return 2
elif (rect.bottomRight() - pos).manhattanLength() < self.__resizeDistance:
return 3
return -1
def getEdge(self, pos):
topRectUpperLeft = self.mapFromItem(self, self.boundingRect().topLeft())
topRectLowerRight = self.mapFromItem(self, self.boundingRect().right(), self.__resizeDistance * 0.25)
topRect = QtCore.QRectF(topRectUpperLeft, topRectLowerRight)
rightRectUpperLeft = self.mapFromItem(self, self.boundingRect().right() - self.__resizeDistance * 0.25, 0)
rightRectLowerRight = self.mapFromItem(self, self.boundingRect().bottomRight())
rightRect = QtCore.QRectF(rightRectUpperLeft, rightRectLowerRight)
bottomRectUpperLeft = self.mapFromItem(self, 0, self.boundingRect().bottom() - self.__resizeDistance * 0.25)
bottomRectLowerRight = self.mapFromItem(self, self.boundingRect().bottomRight())
bottomRect = QtCore.QRectF(bottomRectUpperLeft, bottomRectLowerRight)
leftRectUpperLeft = self.mapFromItem(self, self.boundingRect().topLeft())
leftRectLowerRight = self.mapFromItem(self, self.boundingRect().left() + self.__resizeDistance * 0.25, self.boundingRect().bottom())
leftRect = QtCore.QRectF(leftRectUpperLeft, leftRectLowerRight)
if topRect.contains(pos):
return -1 # Disabling the top resize by default.
elif rightRect.contains(pos):
return 1
elif bottomRect.contains(pos):
return 2
elif leftRect.contains(pos):
return 3
return -1
def inspectorClosed(self):
self.__inspectorWidget = None
# ==========
# Shut Down
# ==========
def disconnectAllPorts(self):
pass
# =============
# Data Methods
# =============
def getData(self):
"""Gets the essential data of the backdrop for saving.
Returns:
dict: Data for saving.
"""
data = {
'name': self.getName(),
'comment': self.getComment(),
'graphPos': self.getGraphPos().toTuple(),
'size': self.size().toTuple(),
'color': self.getColor().toTuple()
}
return data
def setData(self, data):
"""Sets the data on a backdrop after loading.
Args:
data (dict): Name, comment, graph pos, size, and color.
Returns:
bool: True if successful.
"""
self.setComment(data.get('comment', ''))
size = data.get('size', (self.minimumWidth(), self.minimumHeight()))
self.resize(size[0], size[1])
position = data.get('graphPos', (0, 0))
self.setGraphPos(QtCore.QPointF(position[0], position[1]))
color = data.get('color', self.__defaultColor.toTuple())
self.setColor(color=QtGui.QColor(color[0], color[1], color[2], color[3]))
self.setUnselectedColor(self.getColor().darker(125))
self.setSelectedColor(self.getColor().lighter(175))
self.setHoveredColor(self.getColor().lighter(110))
return True
