class CircuitItem(QGraphicsItem):
"""Graphical wrapper around the engine Circuit class."""
textH = 12
"""Height of text."""
ioH = 20
"""Height between to I/O pins."""
ioW = 15
"""Length of I/O pins."""
radius = 10
"""Radius of I/O pin heads."""
def __init__(self, circuit):
super(CircuitItem, self).__init__()
self.setFlag(QGraphicsItem.ItemIsMovable)
self.setFlag(QGraphicsItem.ItemIsSelectable)
self.setFlag(QGraphicsItem.ItemSendsGeometryChanges)
imgDir = filePath('icons/')
self.data = circuit
"""The real info. The class CircuitItem is just a graphical container
around it. data is saved / loaded to / from file.
"""
self.image = QImage(imgDir + circuit.__class__.__name__ + '.png')
"""The graphical representation of our item on screen."""
if not self.image:
self.image = QImage(imgDir + 'Default.png')
self.showCategory = True
self.showName = True
"""Is the item's name shown on screen?"""
self.showCategory = False
"""Is the item's category (circuit class) shown on screen?"""
self.name = QGraphicsSimpleTextItem(self)
# that won't rotate when the PlugItem is rotated by the user.
self.name.setFlag(QGraphicsItem.ItemIgnoresTransformations)
self.name.setText(self.data.name)
self.category = QGraphicsSimpleTextItem(self)
self.category.setFlag(QGraphicsItem.ItemIgnoresTransformations)
self.category.setText(self.data.category)
self.setupPaint()
def boundingRect(self):
"""Qt requires overloading this when overloading QGraphicsItem."""
W = 2 * self.radius + 2 * self.ioW + self.imgW
ni = self.data.nb_inputs()
no = self.data.nb_outputs()
t = (1 - int(max(ni, no) / 2)) * self.ioH - self.radius / 2
b = (1 + int(max(ni, no) / 2)) * self.ioH + self.radius / 2
return (
QRectF(-self.ioW - self.radius, t, W, b - t) if max(ni, no) > 1
else QRectF(-self.ioW - self.radius, 0, W, self.image.height()))
def handleAtPos(self, pos):
"""Is there an interactive handle where the mouse is? Return it."""
for i in range(self.nIn):
if self.inputPaths[i].contains(pos):
return self.data.inputList[i]
for i in range(self.nOut):
if self.outputPaths[i].contains(pos):
return self.data.outputList[i]
def itemChange(self, change, value):
"""Warning view it will soon have to correct pos."""
if change == QGraphicsItem.ItemPositionHasChanged:
# Restart till we stop moving.
self.scene().views()[0].timer.start()
return QGraphicsItem.itemChange(self, change, value)
def paint(self, painter, option, widget):
"""Draws the item."""
painter.setPen(QPen(QColor('black'), 2))
ni = self.data.nb_inputs()
no = self.data.nb_outputs()
for i in range(1 - int(ni / 2), 2 + int(ni / 2)):
if i != 1 or ni % 2:
painter.drawLine(-self.ioW, i * self.ioH, 0, i * self.ioH)
for i in range(1 - int(no / 2), 2 + int(no / 2)):
if i != 1 or no % 2:
painter.drawLine(
self.imgW, i * self.ioH, self.imgW + self.ioW,
i * self.ioH)
painter.drawImage(QRectF(0, 0, self.imgW, self.imgH), self.image)
for i in range(ni):
painter.drawPath(self.inputPaths[i])
painter.drawLine(
0, (1 - int(ni / 2)) * self.ioH, 0, (1 + int(ni / 2)) * self.ioH)
for i in range(no):
painter.drawPath(self.outputPaths[i])
painter.drawLine(
self.imgW,
(1 - int(no / 2)) * self.ioH,
self.imgW,
(1 + int(no / 2)) * self.ioH)
# Default selection box doesn't work; simple reimplementation.
if option.state & QStyle.State_Selected:
pen = QPen(Qt.black, 1, Qt.DashLine)
painter.setPen(pen)
painter.drawRect(self.boundingRect())
def setCategoryVisibility(self, isVisible):
"""Show/Hide circuit category (mostly useful for user circuits)."""
self.showCategory = isVisible
self.setupPaint()
def setNameVisibility(self, isVisible):
"""Shows/Hide the item name in the graphical view."""
self.showName = isVisible
self.setupPaint()
def setNbInputs(self, nb):
"""Add/Remove inputs (for logical gates)."""
if nb > self.data.nb_inputs():
for x in range(nb - self.data.nb_inputs()):
Plug(True, None, self.data)
elif nb < self.data.nb_inputs():
for x in range(self.data.nb_inputs() - nb):
self.data.remove_input(self.data.inputList[0])
self.setupPaint()
def setupPaint(self):
"""Offscreen rather than onscreen redraw (few changes)."""
self.nIn = self.data.nb_inputs()
self.nOut = self.data.nb_outputs()
# 3 sections with different heights must be aligned :
self.imgH = self.image.size().height() # central (png image)
self.imgW = self.image.size().width()
self.inH = (self.nIn - 1) * self.ioH + 2 * self.radius # inputs
self.outH = (self.nOut - 1) * self.ioH + 2 * self.radius # outputs
# therefore we calculate a vertical offset for each section :
self.maxH = max(self.imgH, self.inH, self.outH)
self.imgOff = (
0 if self.maxH == self.imgH else (self.maxH - self.imgH) / 2.)
self.inOff = (
0 if self.maxH == self.inH else (self.maxH - self.inH) / 2.)
self.outOff = (
0 if self.maxH == self.outH else (self.maxH - self.outH) / 2.)
# i/o mouseover detection. Create once, use on each mouseMoveEvent.
self.inputPaths = []
self.outputPaths = []
ni = self.data.nb_inputs()
no = self.data.nb_outputs()
for i in range(1 - int(ni / 2), 2 + int(ni / 2)):
if i != 1 or ni % 2:
path = QPainterPath()
path.addEllipse(
-self.ioW - self.radius, i * self.ioH - self.radius / 2,
self.radius, self.radius)
self.inputPaths.append(path)
for i in range(1 - int(no / 2), 2 + int(no / 2)):
if i != 1 or no % 2:
path = QPainterPath()
path.addEllipse(
self.imgW + self.ioW, i * self.ioH - self.radius / 2,
self.radius, self.radius)
self.outputPaths.append(path)
self.name.setVisible(self.showName)
self.category.setVisible(self.showCategory)
if self.showName or self.showCategory:
br = self.mapToScene(self.boundingRect())
w = self.boundingRect().width()
h = self.boundingRect().height()
realX = min([i.x() for i in br])
realY = min([i.y() for i in br])
firstY = realY + (w if self.rotation() % 180 else h) + 1
secondY = firstY + self.textH
if self.showName:
self.name.setBrush(QColor('red'))
self.name.setText(self.data.name)
self.name.setPos(self.mapFromScene(realX, firstY))
if self.showCategory:
self.category.setBrush(QColor('green'))
self.category.setText(
self.data.category if self.data.category
else self.data.__class__.__name__)
self.category.setPos(self.mapFromScene(
realX, secondY if self.showName else firstY))
self.prepareGeometryChange() # Must be called (cf Qt doc)
self.update() # Force onscreen redraw after changes.
class PlugItem(QGraphicsPathItem):
"""Graphical wrapper around the engine Plug class."""
bodyW = 30
"""The width of the body of plugs."""
pinW = 10
"""The width of the pin part of plugs."""
def __init__(self, plug):
super(PlugItem, self).__init__()
self.data = plug
"""The real info. The class PlugItem is just a graphical container
around it. data is saved / loaded to / from file.
"""
self.showName = False
"""Is the name of the item shown on screen?"""
self.setFlag(QGraphicsItem.ItemIsMovable)
self.setFlag(QGraphicsItem.ItemIsSelectable)
self.setFlag(QGraphicsItem.ItemSendsGeometryChanges)
self.setAcceptsHoverEvents(True)
self.setPen(QPen(QBrush(QColor(QColor('black'))), 2))
# This path is needed at each mouse over event, to check if
# the mouse is over a pin. We save it as an instance field,
# rather than recreate it at each event.
self.pinPath = QPainterPath()
if self.data.isInput:
self.pinPath.addEllipse(
self.bodyW + self.pinW / 2, self.bodyW / 2,
self.pinW, self.pinW)
else:
self.pinPath.addEllipse(
self.pinW / 2, self.bodyW / 2, self.pinW, self.pinW)
f = QFont('Times', 12, 75)
# Name and value text labels.
self.name = QGraphicsSimpleTextItem(self)
# that won't rotate when the PlugItem is rotated by the user.
self.name.setFlag(QGraphicsItem.ItemIgnoresTransformations)
self.name.setText(self.data.name)
self.name.setFont(f)
self.value = QGraphicsSimpleTextItem(self)
self.value.setFlag(QGraphicsItem.ItemIgnoresTransformations)
# Or else value would get the clicks, instead of the PlugItem.
self.value.setFlag(QGraphicsItem.ItemStacksBehindParent)
self.value.setFont(f)
self.setupPaint()
def handleAtPos(self, pos):
"""Is there an interactive handle where the mouse is?
Also return the Plug under this handle.
"""
return self.data if self.pinPath.contains(pos) else None
def itemChange(self, change, value):
"""Warning view it will soon have to correct pos."""
if change == QGraphicsItem.ItemPositionHasChanged:
# Restart till we stop moving.
self.scene().views()[0].timer.start()
return QGraphicsItem.itemChange(self, change, value)
def setAndUpdate(self):
"""Change the undelying plug's value, and force updates items."""
self.data.set(not self.data.value)
for i in self.scene().items():
if isinstance(i, PlugItem) or isinstance(i, WireItem):
i.setupPaint()
def setNameVisibility(self, isVisible):
"""Shows/Hide the item name in the graphical view."""
self.showName = isVisible
self.setupPaint()
def setupPaint(self):
"""Offscreen rather than onscreen redraw (few changes)."""
path = QPainterPath()
if self.data.isInput:
path.addEllipse(
self.pinW / 2, self.pinW / 2, self.bodyW, self.bodyW)
else:
path.addRect(
3 * self.pinW / 2 + 1, self.pinW / 2, self.bodyW, self.bodyW)
path.addPath(self.pinPath)
self.setPath(path)
self.name.setVisible(self.showName)
self.name.setText(self.data.name)
br = self.mapToScene(self.boundingRect())
w = self.boundingRect().width()
h = self.boundingRect().height()
realX = min([i.x() for i in br])
realY = min([i.y() for i in br])
self.name.setPos(self.mapFromScene(
realX, realY + (w if self.rotation() % 180 else h) + 1))
self.value.setText(
str(int(self.data.value)) if self.data.value is not None else 'E')
self.value.setPos(self.mapFromScene(realX + w / 3, realY + h / 3))
self.value.setBrush(QColor('green' if self.data.value else 'red'))
self.update() # Force onscreen redraw after changes.
