def mousePressEvent(self, event): # CANVAS SCENE
printMouseEvent(self, event)
modifiers = getKeyboardModifiers()
# allow graphics view panning
if self.graph._panning:
event.ignore()
return
# if its not a port, then don't draw a line
modmidbutton_event = (event.button() == QtCore.Qt.LeftButton
and modifiers == QtCore.Qt.AltModifier)
if ((event.button() == QtCore.Qt.LeftButton) or (event.button() == QtCore.Qt.MidButton) or modmidbutton_event) \
and isinstance(self.itemAt(event.scenePos()), Port):
event.accept()
self.startLineDraw(event)
# rubber band select
# elif ((event.button() == QtCore.Qt.MidButton) or modmidbutton_event):
elif ((event.button() == QtCore.Qt.LeftButton) \
and not isinstance(self.itemAt(event.scenePos()), Node) \
and not isinstance(self.itemAt(event.scenePos()), PortEdge)):
event.accept()
self.unselectAllItems() # reset select before making another
self.origin = event.scenePos()
self.rubberBand = QtGui.QGraphicsRectItem(
QtCore.QRectF(self.origin, QtCore.QSizeF()))
self.rubberBand.setPen(
QtGui.QPen(QtCore.Qt.gray, 0, QtCore.Qt.SolidLine))
self.rubberBand.setBrush(QtGui.QBrush(QtCore.Qt.lightGray))
self.rubberBand.setZValue(0)
self.addItem(self.rubberBand)
else:
QtGui.QApplication.restoreOverrideCursor()
event.ignore()
super(CanvasScene, self).mousePressEvent(event)
def paint(self, painter, option, widget): # NODE
w = self.getNodeWidth()
# h = self.getTitleWidth()[1]
# draw shadow
painter.setPen(QtCore.Qt.NoPen)
painter.setBrush(QtCore.Qt.darkGray)
painter.drawRoundedRect(-8, -8, w, 20, 3, 3)
# choose module color
gradient = QtGui.QRadialGradient(-10, -10, 40)
# update the face node based on the state of internal nodes.
if self._role == 'Macro':
if self._macroParent.isProcessing():
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.gray).lighter(70))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkGray).lighter(70))
elif (option.state & QtGui.QStyle.State_Sunken) or (
self._macroParent.inComputeErrorState()):
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.red).lighter(150))
gradient.setColorAt(1,
QtGui.QColor(QtCore.Qt.red).lighter(170))
elif self._macroParent.inValidateErrorState():
gradient.setColorAt(
0,
QtGui.QColor(QtCore.Qt.yellow).lighter(190))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.yellow).lighter(170))
elif self._macroParent.inInitUIErrorState():
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.red).lighter(150))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.yellow).lighter(170))
else:
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.gray).lighter(150))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkGray).lighter(150))
# let the src and sink nodes update themselves normally
else:
conf = self.getCurState()
if self._computeState is conf:
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.gray).lighter(70))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkGray).lighter(70))
elif (option.state & QtGui.QStyle.State_Sunken) or (
self._computeErrorState is conf):
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.red).lighter(150))
gradient.setColorAt(1,
QtGui.QColor(QtCore.Qt.red).lighter(170))
elif self._validateError is conf:
gradient.setColorAt(
0,
QtGui.QColor(QtCore.Qt.yellow).lighter(190))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.yellow).lighter(170))
else:
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.gray).lighter(150))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkGray).lighter(150))
# draw module box (apply color)
painter.setBrush(QtGui.QBrush(gradient))
if self.beingHovered or self.isSelected():
#painter.setPen(QtGui.QPen(QtCore.Qt.red, 1))
fade = QtGui.QColor(QtCore.Qt.red)
fade.setAlpha(100)
painter.setPen(QtGui.QPen(fade, 2))
else:
#painter.setPen(QtGui.QPen(QtCore.Qt.black, 0))
fade = QtGui.QColor(QtCore.Qt.black)
fade.setAlpha(50)
painter.setPen(QtGui.QPen(fade, 0))
painter.drawRoundedRect(-10, -10, w, 20, 3, 3)
# title
painter.setPen(QtGui.QPen(QtCore.Qt.black, 0))
painter.setFont(self.title_font)
buf = self.getMacroNodeName()
# paint the node title
painter.drawText(-5, -9, w, 20,
(QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter),
str(buf))
def paint(self, painter, option, widget): # PORT
# choose module color
gradient = QtGui.QRadialGradient(-1, -1, 10)
if option.state & QtGui.QStyle.State_Sunken:
gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.red).lighter(150))
gradient.setColorAt(1,
QtGui.QColor(QtCore.Qt.darkRed).lighter(150))
# elif self._beingHovered:
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.darkRed).lighter(150))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.red).lighter(150))
elif isinstance(self, InPort):
# if self.menuWidget:
if self.isREQUIRED():
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.yellow).lighter(200))
# gradient.setColorAt(1,
# QtGui.QColor(QtCore.Qt.darkYellow).lighter(200))
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.blue).lighter(200))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkBlue).lighter(200))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.gray).lighter(300))
# gradient.setColorAt(0,
# QtGui.QColor(QtCore.Qt.darkGray).lighter(150))
else:
gradient.setColorAt(0,
QtGui.QColor(QtCore.Qt.green).lighter(200))
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.darkGreen).lighter(150))
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.gray).lighter(150))
# gradient.setColorAt(1,
# QtGui.QColor(QtCore.Qt.darkGray).lighter(100))
elif isinstance(self, OutPort):
# if self.menuWidget:
# orange=QtGui.QColor().fromRgbF(1.,0.5,0.)
# gradient.setColorAt(0, orange.lighter(200))
# gradient.setColorAt(1, orange.lighter(100))
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.blue).lighter(300))
# gradient.setColorAt(1,
# QtGui.QColor(QtCore.Qt.darkBlue).lighter(300))
if self.dataIsNone():
gradient.setColorAt(1,
QtGui.QColor(QtCore.Qt.red).lighter(300))
gradient.setColorAt(
0,
QtGui.QColor(QtCore.Qt.darkRed).lighter(150))
elif self.dataHasChanged():
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.gray).lighter(200))
# gradient.setColorAt(0,
# QtGui.QColor(QtCore.Qt.darkGray).lighter(100))
gradient.setColorAt(1,
QtGui.QColor(QtCore.Qt.blue).lighter(200))
gradient.setColorAt(
0,
QtGui.QColor(QtCore.Qt.darkBlue).lighter(200))
else:
gradient.setColorAt(
1,
QtGui.QColor(QtCore.Qt.yellow).lighter(200))
gradient.setColorAt(
0,
QtGui.QColor(QtCore.Qt.darkYellow).lighter(150))
# else:
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.blue).lighter(175))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.darkBlue).lighter(175))
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.gray).lighter(150))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.darkGray).lighter(100))
# if self.dataIsNone():
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.red).lighter(150))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.darkRed).lighter(100))
# elif self.dataHasChanged():
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.gray).lighter(150))
# gradient.setColorAt(1, QtGui.QColor(QtCore.Qt.darkGray).lighter(100))
# else:
# gradient.setColorAt(0, QtGui.QColor(QtCore.Qt.yellow).lighter(200))
# gradient.setColorAt(1,
# QtGui.QColor(QtCore.Qt.darkYellow).lighter(150))
# draw module box (apply color)
painter.setBrush(QtGui.QBrush(gradient))
#painter.setPen(QtGui.QPen(QtCore.Qt.black, 0))
fade = QtGui.QColor(QtCore.Qt.black)
fade.setAlpha(50)
painter.setPen(QtGui.QPen(fade, 0))
# if isinstance(self,InPort):
# if self.isREQUIRED():
# painter.setPen(QtGui.QPen(QtCore.Qt.red, 0))
# else:
# painter.setPen(QtGui.QPen(QtCore.Qt.black, 0))
# #painter.setPen(QtGui.QPen(QtCore.Qt.yellow, 0))
if self.isMemSaver():
painter.drawPolygon(self.portShape_memSave)
else:
if self.menuWidget:
painter.drawEllipse(1, 0, 5, 5)
else:
painter.drawPolygon(self.portShape)
