# -*- coding: utf-8 -*- # # AWL simulator - FUP draw widget # # Copyright 2016 Michael Buesch # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License along # with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. # from __future__ import division, absolute_import, print_function, unicode_literals from awlsim.common.compat import * from awlsim.common.xmlfactory import * from awlsim.gui.icons import * from awlsim.gui.util import * from awlsim.gui.fup.fup_grid import * class FupContextMenu(QMenu): """FUP/FBD draw widget context menu.""" add = Signal(FupElem) edit = Signal() remove = Signal() def __init__(self, parent=None): QMenu.__init__(self, parent) self.gridX = 0 self.gridY = 0 self.__actInsAND = self.addAction(getIcon("doc_new"), "Insert &AND", self.__addAND) self.__actInsOR = self.addAction(getIcon("doc_new"), "Insert &OR", self.__addOR) self.__actInsXOR = self.addAction(getIcon("doc_new"), "Insert &XOR", self.__addXOR) self.addSeparator() self.__actInsLOAD = self.addAction(getIcon("doc_new"), "Insert &LOAD operand", self.__addLOAD) self.__actInsASSIGN = self.addAction(getIcon("doc_new"), "Insert A&SSIGN operand", self.__addASSIGN) self.addSeparator() self.__actEdit = self.addAction(getIcon("doc_edit"), "&Edit element...", self.__edit) self.__actDel = self.addAction(getIcon("doc_close"), "&Remove element", self.__del) self.addSeparator() self.__actAddInp = self.addAction(getIcon("new"), "Add input &connection", self.__addInput) def __addAND(self): self.add.emit(FupElem_AND(self.gridX, self.gridY)) def __addOR(self): self.add.emit(FupElem_OR(self.gridX, self.gridY)) def __addXOR(self): self.add.emit(FupElem_XOR(self.gridX, self.gridY)) def __addLOAD(self): self.add.emit(FupElem_LOAD(self.gridX, self.gridY)) def __addASSIGN(self): self.add.emit(FupElem_ASSIGN(self.gridX, self.gridY)) def __edit(self): self.edit.emit() def __del(self): self.remove.emit() def __addInput(self): pass def enableInsert(self, en=True): self.__actInsAND.setEnabled(en) self.__actInsOR.setEnabled(en) self.__actInsXOR.setEnabled(en) self.__actInsLOAD.setEnabled(en) self.__actInsASSIGN.setEnabled(en) def enableEdit(self, en=True): self.__actEdit.setEnabled(en) def enableDelete(self, en=True): self.__actDel.setEnabled(en) def enableAddInput(self, en=True): self.__actAddInp.setEnabled(en) class FupDrawWidget(QWidget): """FUP/FBD draw widget.""" # Signal: Something in the FUP diagram changed diagramChanged = Signal() def __init__(self, parent=None): QWidget.__init__(self, parent) self.__suppressMousePress = 0 self.__repaintBlocked = Blocker() self.__contextMenu = FupContextMenu(self) self.__contextMenu.add.connect(self.addElem) self.__contextMenu.remove.connect(self.removeSelElems) self.__contextMenu.edit.connect(self.editSelElems) self.__bgBrush = QBrush(QColor("#F5F5F5")) self.__gridPen = QPen(QColor("#E0E0E0")) self.__gridPen.setWidth(1) self.__textPen = QColor("#808080") self.__dragConnPenOpen = QPen(QColor("#FF0000")) self.__dragConnPenOpen.setWidth(4) self.__dragConnPenClosed = QPen(QColor("#000000")) self.__dragConnPenClosed.setWidth(4) self.__selRectPen = QPen(QColor("#0000FF")) self.__selRectPen.setWidth(3) # Start and end pixel coordinates of a selection rectangle. self.__selectStartPix = None self.__selectEndPix = None # Start grid coordinates of an element drag. self.__dragStart = None # The dragged connection. self.__draggedConn = None self.__cellHeight = 20 self.__cellWidth = 60 self.__grid = FupGrid(self, 12, 32) self.resize(self.__grid.width * self.__cellWidth, self.__grid.height * self.__cellHeight) self.setFocusPolicy(Qt.ClickFocus | Qt.WheelFocus | Qt.StrongFocus) self.setMouseTracking(True) def repaint(self): if not self.__repaintBlocked: QWidget.repaint(self) def __contentChanged(self): self.repaint() self.diagramChanged.emit() @property def grid(self): return self.__grid @property def cellPixHeight(self): return self.__cellHeight @property def cellPixWidth(self): return self.__cellWidth def addElem(self, elem): if self.__grid.placeElem(elem): self.__grid.deselectAll() self.__grid.selectElem(elem) self.__contentChanged() def removeElem(self, elem): self.__grid.removeElem(elem) self.__contentChanged() def removeSelElems(self): with self.__repaintBlocked: for elem in self.__grid.selectedElems.copy(): self.removeElem(elem) self.__contentChanged() def moveElem(self, elem, toGridX, toGridY, relativeCoords=False, checkOnly=False, excludeCheckElems=set()): if self.__grid.moveElemTo(elem, toGridX, toGridY, relativeCoords=relativeCoords, checkOnly=checkOnly, excludeCheckElems=excludeCheckElems): if not checkOnly: self.__contentChanged() return True return False def establishWire(self, fromConn, toConn): if fromConn and toConn: if fromConn.canConnectTo(toConn): with contextlib.suppress(ValueError): fromConn.connectTo(toConn) self.__contentChanged() def disconnectConn(self, conn): if conn: conn.disconnect() self.__contentChanged() def editSelElems(self): chg = 0 if self.__grid: for elem in self.__grid.selectedElems: chg += int(elem.edit(self)) if chg: self.__contentChanged() def paintEvent(self, event=None): grid = self.__grid if not grid: return size = self.size() width, height = size.width(), size.height() p = QPainter(self) # Draw background p.fillRect(self.rect(), self.__bgBrush) # Set font font = p.font() font.setFamily("Mono") font.setPointSize(7) p.setFont(font) # Draw the grid p.setPen(self.__gridPen) # vertical lines for x in range(0, width, self.__cellWidth): p.drawLine(x, 0, x, height) p.drawText(x, -5, self.__cellWidth, self.__cellHeight, Qt.AlignCenter | Qt.AlignTop, str(x // self.__cellWidth)) # horizontal lines for y in range(0, height, self.__cellHeight): p.drawLine(0, y, width, y) p.drawText(5, y, self.__cellWidth, self.__cellHeight, Qt.AlignLeft | Qt.AlignVCenter, str(y // self.__cellHeight)) # Draw the help text, if the grid is empty. if not grid.elems: p.setPen(self.__textPen) font = p.font() font.setPointSize(9) p.setFont(font) x, y = self.__cellWidth + 5, self.__cellHeight * 2 - 5 p.drawText(x, y, width - x, height - y, Qt.AlignLeft | Qt.AlignTop, "Hints:\n" "* Right click here to insert FUP/FBD element\n" "* Left-drag to connect inputs and outputs\n" "* Middle-click to delete connections and wires") # Draw the elements. First background elements (selected/expanded). prevX, prevY = 0, 0 for wantForeground in (False, True): for elem in grid.elems: isForeground = elem.selected or elem.expanded if wantForeground == isForeground: xAbs, yAbs = elem.pixCoords p.translate(xAbs - prevX, yAbs - prevY) prevX, prevY = xAbs, yAbs elem.draw(p) p.translate(-prevX, -prevY) # Draw the connection wires for wire in grid.wires: wire.draw(p) # Draw the dragged connection draggedConn = self.__draggedConn if draggedConn and draggedConn.elem: xAbs, yAbs = draggedConn.pixCoords gridX, gridY = self.posToGridCoords(xAbs, yAbs) mousePos = self.mapFromGlobal(QCursor.pos()) elem, _, _, _, _, elemRelX, elemRelY = self.posToElem( mousePos.x(), mousePos.y()) # Check if we hit a possible target connection p.setPen(self.__dragConnPenOpen) if elem: targetConn = elem.getConnViaPixCoord(elemRelX, elemRelY) if targetConn and draggedConn.canConnectTo(targetConn): p.setPen(self.__dragConnPenClosed) p.drawLine(xAbs, yAbs, mousePos.x(), mousePos.y()) # Draw selection rectangle if self.__selectStartPix and self.__selectEndPix: xAbs0 = min(self.__selectStartPix[0], self.__selectEndPix[0]) yAbs0 = min(self.__selectStartPix[1], self.__selectEndPix[1]) xAbs1 = max(self.__selectStartPix[0], self.__selectEndPix[0]) yAbs1 = max(self.__selectStartPix[1], self.__selectEndPix[1]) selWidth, selHeight = xAbs1 - xAbs0, yAbs1 - yAbs0 r = 2 p.setBrush(Qt.NoBrush) p.setPen(self.__selRectPen) p.drawRoundedRect(xAbs0, yAbs0, selWidth, selHeight, r, r) def posToGridCoords(self, pixX, pixY): """Convert pixel coordinates to grid coordinates. """ return pixX // self.__cellWidth,\ pixY // self.__cellHeight def posToElem(self, pixX, pixY): """Convert pixel coordinates to element and element coordinates. Returns: (FupElem, FupConn, FupElem.AREA_xxx, gridX, gridY, elemRelativePixX, elemRelativePixY) """ # Get the grid coordinates. gridX, gridY = self.posToGridCoords(pixX, pixY) # Get the element (if any). elem = self.__grid.getElemAt(gridX, gridY) # Get the coordinates relative to the element. elemRelX = pixX % self.__cellWidth elemRelY = pixY % self.__cellHeight if elem: elemRelX += (gridX - elem.x) * self.__cellWidth elemRelY += (gridY - elem.y) * self.__cellHeight # Get the connection and area (if any). conn, area = None, None if elem: area, areaIdx = elem.getAreaViaPixCoord(elemRelX, elemRelY) if area == FupElem.AREA_INPUT: conn = elem.getInput(areaIdx) elif area == FupElem.AREA_OUTPUT: conn = elem.getOutput(areaIdx) return elem, conn, area, gridX, gridY, elemRelX, elemRelY def mousePressEvent(self, event): if self.__suppressMousePress: self.__suppressMousePress -= 1 return x, y = event.x(), event.y() modifiers = QGuiApplication.keyboardModifiers() # Get the element (if any) elem, conn, area, gridX, gridY, elemRelX, elemRelY = self.posToElem(x, y) self.__contextMenu.gridX = gridX self.__contextMenu.gridY = gridY # Handle left button press if event.button() == Qt.LeftButton: if elem: if area == FupElem.AREA_BODY: # Start dragging of the selected element(s). self.__dragStart = (gridX, gridY) if not elem.selected: # Select this element. if not (modifiers & Qt.ControlModifier): self.__grid.deselectAll() self.__grid.selectElem(elem) self.repaint() if conn and (not conn.wire or conn.OUT): # Start dragging of the selected connection. self.__draggedConn = conn self.repaint() else: # Start a multi-selection if not (modifiers & Qt.ControlModifier): self.__grid.deselectAll() self.__selectStartPix = (x, y) self.__selectEndPix = None self.repaint() # Handle middle button press if event.button() == Qt.MidButton: self.disconnectConn(conn) # Handle right button press if event.button() == Qt.RightButton: if elem and not elem.selected: if not (modifiers & Qt.ControlModifier): self.__grid.deselectAll() self.__grid.selectElem(elem) self.repaint() # Open the context menu self.__contextMenu.enableInsert(elem is None) self.__contextMenu.enableDelete(elem is not None) self.__contextMenu.enableEdit(False) self.__contextMenu.enableAddInput(False) if elem: elem.prepareContextMenu(self.__contextMenu) self.__contextMenu.exec_(self.mapToGlobal(event.pos())) QWidget.mousePressEvent(self, event) def mouseReleaseEvent(self, event): x, y = event.x(), event.y() elem, conn, area, gridX, gridY, elemRelX, elemRelY = self.posToElem(x, y) # Handle end of multi-selection if self.__selectStartPix: self.__selectStartPix = None self.__selectEndPix = None self.repaint() # Handle end of element dragging self.__dragStart = None # Handle end of connection dragging draggedConn = self.__draggedConn if draggedConn: # Try to establish the dragged connection. if elem: targetConn = elem.getConnViaPixCoord(elemRelX, elemRelY) self.establishWire(draggedConn, targetConn) self.__draggedConn = None self.repaint() QWidget.mouseReleaseEvent(self, event) def mouseMoveEvent(self, event): x, y = event.x(), event.y() modifiers = QGuiApplication.keyboardModifiers() elem, conn, area, gridX, gridY, elemRelX, elemRelY = self.posToElem(x, y) # Temporarily expand elements on mouse-over if event.buttons() == Qt.NoButton: chg = 0 if elem and area == FupElem.AREA_BODY: if elem not in self.__grid.expandedElems: chg += int(self.__grid.unexpandAllElems()) chg += int(self.__grid.expandElem(elem, True)) else: chg += int(self.__grid.unexpandAllElems()) if chg: self.repaint() # Handle multi-selection if self.__selectStartPix: self.__selectEndPix = (x, y) # Mark all elements within the rectangle as selected. startGridX, startGridY = self.posToGridCoords(*self.__selectStartPix) clear = not (modifiers & Qt.ControlModifier) self.__grid.selectElemsInRect(startGridX, startGridY, gridX, gridY, clear=clear) self.repaint() # Handle element dragging if self.__dragStart: deltaX, deltaY = gridX - self.__dragStart[0],\ gridY - self.__dragStart[1] if deltaX or deltaY: with self.__repaintBlocked: selectedElems = self.__grid.selectedElems # First check if we can move all elements allOk = True for elem in selectedElems: if not self.moveElem(elem, deltaX, deltaY, relativeCoords=True, checkOnly=True, excludeCheckElems=selectedElems): allOk = False break # If everything is Ok, move all elements. if allOk: for elem in selectedElems: self.moveElem(elem, deltaX, deltaY, relativeCoords=True, checkOnly=False, excludeCheckElems=selectedElems) self.__dragStart = (gridX, gridY) self.repaint() # Handle connection dragging if self.__draggedConn: self.repaint() QWidget.mouseMoveEvent(self, event) def mouseDoubleClickEvent(self, event): x, y = event.x(), event.y() elem, conn, area, gridX, gridY, elemRelX, elemRelY = self.posToElem(x, y) # Force end of dragging. self.__dragStart = None self.__draggedConn = None # Handle left button double click if event.button() == Qt.LeftButton: if elem: if conn and not conn.isConnected: # Double click on an unconnected IN or OUT connection # adds a LOAD or ASSIGN operator element. connGridX, connGridY = self.posToGridCoords(*conn.pixCoords) newElem, newConn = None, None if conn.IN and not conn.OUT: newElem = FupElem_LOAD(connGridX - 1, connGridY) if newElem.outputs: newConn = newElem.outputs[0] elif conn.OUT and not conn.IN: newElem = FupElem_ASSIGN(connGridX + 1, connGridY) if newElem.inputs: newConn = newElem.inputs[0] if newElem and newConn: self.addElem(newElem) with contextlib.suppress(ValueError): newConn.connectTo(conn) self.repaint() else: # Edit the element's contents if elem.edit(self): self.__contentChanged() # Suppress the next press event self.__suppressMousePress += 1 QWidget.mouseDoubleClickEvent(self, event) def keyPressEvent(self, event): if event.matches(QKeySequence.Delete): self.removeSelElems() elif event.matches(QKeySequence.Cancel) or\ event.matches(QKeySequence.Deselect): self.__grid.deselectAll() self.repaint() elif event.matches(QKeySequence.SelectAll): for elem in self.__grid.elems: self.__grid.selectElem(elem) self.repaint() QWidget.keyPressEvent(self, event) def keyReleaseEvent(self, event): QWidget.keyReleaseEvent(self, event)