class MainWindow(QMainWindow, svLumpedGui3A.Ui_MainWindow):
def __init__(
self,
SettingDialog,
parent=None,
):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.BoundaryFace.clicked.connect(self.addBoundaryFaceButton)
self.Ground.clicked.connect(self.addGroundButton)
self.Export.clicked.connect(self.exportXML)
self.scene = self.CircuitView.scene()
def mousePressEvent(self, event):
global wireCount
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.selectedItem = self.scene.itemAt(self.start)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if self.spock.isFlat() == False:
self.end = QtCore.QPointF(self.itemX + 88, self.itemY)
elif (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
self.wire = QtGui.QGraphicsLineItem(QtCore.QLineF(
self.start, self.end))
self.scene.addItem(self.wire)
if self.start == self.end:
self.scene.removeItem(self.wire)
else:
wireCount = wireCount + 1
self.wire.setZValue(1)
def deleteWire(self):
self.scene.removeItem(self.wire)
def exportXML(self):
root = minidom.Document()
xml = root.createElement('root')
root.appendChild(xml)
wireChild = root.createElement('wire')
xml.appendChild(wireChild)
xmlStr = root.toprettyxml(indent='\t')
savePathFile = 'test.xml'
with open(savePathFile, 'w') as f:
f.write(xmlStr)
global wireCount, componentCount
allComponents = self.scene.items()
totalComponents = wireCount + componentCount
for QtGui.QGraphicsLineItem in allComponents[0:wireCount]:
WirePos = QtGui.QGraphicsLineItem.line()
print WirePos
for QtGui.QGraphicsProxyWidget in allComponents[
wireCount:totalComponents]:
ComponentWidget = QtGui.QGraphicsProxyWidget.widget()
if ComponentWidget.isFlat() == False:
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() + 88),
(QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ''
elif ComponentWidget.isChecked():
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49),
(QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49),
(QtGui.QGraphicsProxyWidget.pos().y() + 176))
else:
node1 = (QtGui.QGraphicsProxyWidget.pos().x(),
(QtGui.QGraphicsProxyWidget.pos().y() + 49))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() + 176),
(QtGui.QGraphicsProxyWidget.pos().y() + 49))
print ComponentWidget.componentType, ComponentWidget.name, ComponentWidget.value, ComponentWidget.unit, node1, node2
def addResistorButton(self):
global componentCount
self.Resistor2 = DragButton('')
## self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.setCheckable(True)
self.Resistor2.clicked.connect(self.handlebutton)
self.Resistor2.componentType = 'Resistor'
self.sceneResistor = self.scene.addWidget(self.Resistor2)
componentCount = componentCount + 1
def addCapacitorButton(self):
global componentCount
self.Capacitor2 = DragButton('')
## self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.setCheckable(True)
self.Capacitor2.setChecked(False)
self.Capacitor2.clicked.connect(self.handlebutton)
self.Capacitor2.componentType = 'Capacitor'
self.sceneCapacitor = self.scene.addWidget(self.Capacitor2)
componentCount = componentCount + 1
def addInductorButton(self):
global componentCount
self.Inductor2 = DragButton('')
## self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setCheckable(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.Inductor2.componentType = 'Inductor'
self.scene.addWidget(self.Inductor2)
componentCount = componentCount + 1
def addDiodeButton(self):
global componentCount
self.Diode2 = DragButton('')
## self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.Diode2.setCheckable(True)
self.Diode2.componentType = 'Diode'
self.scene.addWidget(self.Diode2)
componentCount = componentCount + 1
def addVoltageButton(self):
global componentCount
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(
QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DC Voltage Source"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.DCVoltageSource2.setCheckable(True)
self.DCVoltageSource2.componentType = 'DC Voltage Source'
self.scene.addWidget(self.DCVoltageSource2)
componentCount = componentCount + 1
def addBoundaryFaceButton(self):
global componentCount
self.BoundaryFace2 = DragButton('')
self.BoundaryFace2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.BoundaryFace2.setAcceptDrops(False)
self.BoundaryFace2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("BoundaryFace.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.BoundaryFace2.setIcon(icon4)
self.BoundaryFace2.setIconSize(QtCore.QSize(139, 70))
self.BoundaryFace2.setFlat(True)
self.BoundaryFace2.setObjectName(_fromUtf8("Boundary Face"))
self.BoundaryFace2.clicked.connect(self.handlebutton)
self.BoundaryFace2.componentType = 'Boundary Face'
self.scene.addWidget(self.BoundaryFace2)
componentCount = componentCount + 1
def addGroundButton(self):
global ground, componentCount
self.Ground2 = DragButton('')
self.Ground2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Ground2.setAcceptDrops(False)
self.Ground2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("Ground.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Ground2.setIcon(icon4)
self.Ground2.setIconSize(QtCore.QSize(139, 70))
self.Ground2.setFlat(False)
self.Ground2.setObjectName(_fromUtf8("Ground"))
self.Ground2.componentType = 'Ground'
self.scene.addWidget(self.Ground2)
componentCount = componentCount + 1
def handlebutton(self):
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
self.widgetSet = self.sender()
self.Settings = SettingDialog(self)
Type = self.widgetSet.componentType
self.Settings.Type.setText(Type)
self.Settings.exec_()
self.widgetSet.name = self.Settings.Name.text()
self.widgetSet.value = self.Settings.Value.value()
self.widgetSet.unit = self.Settings.Unit.currentText()
self.Settings.Name.clear()
self.Settings.Value.setValue(0)
self.Settings.Unit.setCurrentIndex(0)
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.selectedItem.setTransform(self.transform)
self.spock.setChecked(True)
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
self.scene.removeItem(self.selectedItem)
class MainWindow(QMainWindow, svLPMGui.Ui_MainWindow):
# creates the MainWindow Class and imports the window attributes from the UI generated python file.
def __init__(self, SettingDialog, parent=None,):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.BoundaryFace.clicked.connect(self.addBoundaryFaceButton)
self.Ground.clicked.connect(self.addGroundButton)
self.scene = self.CircuitView.scene()
self.actionOpen.triggered.connect(self.openFile)
self.actionSave_2.triggered.connect(self.XMLFileSave)
self.actionSaveAs.triggered.connect(self.fileSaveAs)
self.actionQuit_2.triggered.connect(self.Quit)
self.actionClear.triggered.connect(self.Clear)
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(0,0,0,0);color:black;font-weight:bold;}")
self.statusbar.messageChanged.connect(self.statusbarChanged)
self.windowCancel = False
# adds functionality to all the widgets/menubar/statusbar in the mainwindow and connects their trigger event to the relevent method
def mousePressEvent(self, event):
global wireCount
# method for enabling click events to draw wires: note that all the component widgets in the scene are QGraphicsProxyWidgets which have embedded custom QPushButtons
# called DragButtons with drag and drop functionality handelled by the QPushbutton widget itself.
try:
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.selectedItem = self.scene.itemAt(self.start)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
# This part of the method sets the position of the mouse click event.
# First the position of the click is mapped to global, then to the scene
# and accesses the embedded DragButton widget in the QGraphicsProxyWidget that was pressed.
# the middle button was needed due to the LeftButton already being used for the Drag and Drop functionality and the clash would break the drag and drop.
if (self.clickY > (self.itemY + 88)) &self.spock.rotated == True :
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY +88)) & self.spock.rotated == True:
self.start = QtCore.QPointF(self.itemX -49, self.itemY)
# checks whether the button has been rotated using the rotated class attribute from the DragButton class
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX +176, self.itemY + 49)
# This is the node assignment function: the dimensions of the QGraphicsProxy Widgets are all 176 by 98: therefore sets start position of button to node closest to the click.
# Note that the .geometry() function gives the x and y value of the "origin point" of the widget, which is the (0,0) of the widget but these co-ordinates have been mapped to the scene
# the node closest to the click is determined by whether the button is rotated or not: with the node assignment depending on whether the click is left or right of the central
# x axis for horizontal buttons or the y axis for rotated buttons
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
# reimplements position handling from the above, but with the RightButton on the mouse.
if self.spock.componentType == "Ground":
self.end = QtCore.QPointF(self.itemX +88, self.itemY)
# checks if the component is a ground button: if so sets its end to one node in the top centre of the button
elif (self.clickY > (self.itemY + 88)) & self.spock.rotated == True:
self.end = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY +88)) & self.spock.rotated == True:
self.end = QtCore.QPointF(self.itemX -49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX +176, self.itemY + 49)
wire = QtCore.QLineF(self.start, self.end)
self.wire = QtGui.QGraphicsLineItem(wire)
self.scene.addItem(self.wire)
# draws wire between two click positions
if self.start == self.end:
self.scene.removeItem(self.wire)
# intentional workaround: as both self.start and self.end are the same, it deletes the initial "wire" at the point of the first click. When setting another variable
#as the second click position (e.g. self.clickX2), it breaks the ability to draw another wire from a seperate click position as self.start will remain as the starting point. This workaround
# thus enables the self.start position to remain the start position of the wire (which is then removed) and then to be changed after the middle button is clicked on another component.
else:
wireCount = wireCount+1
self.wire.setZValue(1)
# increased the wireCount number to match the number of wires in the scene, and sets the ZValue for the QGraphicsLineItem to 1 to enable all the wires to stack first
#when the QGraphicsScene.items() function is run and returns an ordered list of all the items in the scene (Line Items and ProxyWidgets)
except Exception as err:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: Component Not Pressed!", msecs = 1500)
#QOL improvement: shows red error message when a component is not selected with a click event
def statusbarChanged(self, args):
if not args:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(0,0,0,0);color:black;font-weight:bold;}")
#when the statusbar is changed after the message timer has expired, resets the statusbar colour to transparent
def closeEvent(self, event):
self.Quit()
if self.windowCancel == True:
event.ignore()
self.confirmExit.close()
#method to overide the native closeEvent when exiting the program from the mainwindow X to bring up the confirmExit dialog
def Clear(self):
self.clearScene = QtGui.QMessageBox()
self.clearScene.setIcon(QtGui.QMessageBox.Warning)
self.clearScene.setText("Are you sure you want to clear the circuit?")
self.clearScene.setInformativeText("Any changes will be lost without saving")
self.clearScene.setStandardButtons(QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
self.clearScene.buttonClicked.connect(self.clearOptions)
self.clearScene.exec_()
#Creates a MessageBox Dialog to confirm clearing the scene of all items
def clearOptions(self, button):
buttonPressed = button.text()
if buttonPressed == "Cancel":
self.clearScene.close()
else:
self.scene.clear()
def deleteWire(self):
global wireCount
if wireCount > 0:
deleteWire = self.scene.items()[0]
# accesses the first item in the list (which will always be a wire due to the ZValue being 1 and the component's ZValue being default 0) and deletes it
self.scene.removeItem(deleteWire)
wireCount = wireCount - 1
else:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Wires Exist (yet!)", msecs = 1500)
# QOL message when no wires are in the scene
def fileSaveAs(self):
self.fileName = str(QtGui.QFileDialog.getSaveFileName(self, "Save File as XML"))
self.XMLFileSave()
# Opens the Save File Dialog native to pyqt, accesses the string of the file and implements the XMLFileSave method
def Quit(self):
self.confirmExit = QtGui.QMessageBox()
self.confirmExit.setIcon(QtGui.QMessageBox.Warning)
self.confirmExit.setText("Are you sure you want to Quit?")
self.confirmExit.setInformativeText("Any changes will be lost without saving!")
self.confirmExit.setStandardButtons(QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
self.confirmExit.buttonClicked.connect(self.quitOptions)
self.confirmExit.exec_()
def quitOptions(self, button):
buttonPressed = button.text()
if buttonPressed == "Cancel":
self.windowCancel = True
self.confirmExit.close()
else:
sys.exit()
# Quit Dialog when File>Quit is accessed
def XMLFileSave(self):
global wireCount, componentCount
# method for exporting the circuit drawn to XML File
allComponents = self.scene.items()
# returns a list of all the items in the scene, with the wires stacking first, then components
totalComponents = wireCount + componentCount
root = minidom.Document()
xml = root.createElement('circuit')
root.appendChild(xml)
# creates the framework for an XML File and sets the root of the XML File as "circuit" using the minidom API from xml.dom
if componentCount != 0 and wireCount != 0:
# Implements ability that user can only save file when there are both components and wires on the scene.
for QtGui.QGraphicsProxyWidget in allComponents[wireCount:totalComponents]:
ComponentWidget = QtGui.QGraphicsProxyWidget.widget()
# Slices the list to access the components which will be located between the wires and the end of the list and accesses the embedded widget
#loops over all the components in the allcomponents list
if ComponentWidget.componentType == "Ground":
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() +88), (QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ''
#the ground button only has 1 node
elif ComponentWidget.rotated == True :
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49), (QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() -49), (QtGui.QGraphicsProxyWidget.pos().y() + 176))
else:
node1 = (QtGui.QGraphicsProxyWidget.pos().x(), (QtGui.QGraphicsProxyWidget.pos().y() + 49))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() + 176), (QtGui.QGraphicsProxyWidget.pos().y() +49))
xmlcomponents = root.createElement('Component')
xmlcomponents.setAttribute('type', ComponentWidget.componentType)
xmlcomponents.setAttribute('value', str(ComponentWidget.value))
xmlcomponents.setAttribute('metricPrefix', str(ComponentWidget.unit))
xmlcomponents.setAttribute('name', str(ComponentWidget.name))
xmlcomponents.setAttribute('rotated', str(ComponentWidget.rotated))
xmlcomponents.setAttribute('DialogIndex', str(ComponentWidget.comboBoxIndex))
xml.appendChild(xmlcomponents)
# creates child elements in the circuit root element and adds attributes from the class members(attributes) from each DragButton Component (Note that all attributes must be strings)
componentNode1 = root.createElement("Node")
componentNode1.appendChild(root.createTextNode(str(node1)))
xmlcomponents.appendChild(componentNode1)
componentNode2 = root.createElement("Node")
componentNode2.appendChild(root.createTextNode(str(node2)))
xmlcomponents.appendChild(componentNode2)
# adds the node positions for the component as child elements of the component element
for wires in allComponents[0:wireCount]:
WirePos = wires.line()
WirePos = str(WirePos)
stringLength = len(WirePos)-1
wire = WirePos[20:stringLength]
xmlwire = root.createElement('wire')
xml.appendChild(xmlwire)
wirePos = root.createElement('wirePos')
wirePos.appendChild(root.createTextNode(wire))
xmlwire.appendChild(wirePos)
# slices the list to access only the wires which will stack first in the list and loops over each wire
xmlStr = root.toprettyxml(indent = '\t')
# adds the xml top string and indents for elements
try:
if self.sender() == self.actionSaveAs:
self.savePathFile = self.fileName + ".xml"
# checks if the sender of the method is File>SaveAs..., if so, it adds the .xml tag to the filename
elif self.sender() == self.actionSave_2:
self.savePathFile = self.fileName
# check if the sender of the method is File>Save, if so, it sets the self.filename (which was set when the user opened the XML file) as the filename and so overwrites its contents
# with the changes made
with open(self.savePathFile, 'w') as f:
f.write(xmlStr)
self.statusbar.showMessage("Circuit Saved to XML File", msecs = 3000)
# exports the circuit to the XML File or creates the file if it doesn't yet exist
except Exception as err:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: XML File does not exist! : Use Save As", msecs = 4000)
# an exception is generated if the user attempts to File>Save a previously unsaved circuit as the self.filename will not exist.
elif componentCount == 0:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Components Added!", msecs = 1600)
elif wireCount == 0:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Wires Added!", msecs = 1600)
# QOL improvement: error messages if the user's circuit has not wires or no components
def openFile(self):
global wireCount
# this method enables XML files to be opened, parsed, and rebuilds the circuit from the XML file parse
try:
self.fileName = str(QtGui.QFileDialog.getOpenFileName())
nameSplit = self.fileName.split("/") or self.fileName.split ("\\")
nameSplitNumber = (len(nameSplit)-1)
xmlFile = nameSplit[nameSplitNumber]
xmlLoad= ET.parse(xmlFile)
circuit = xmlLoad.getroot()
# updates the self.fileName to the string of the opened file (this time with the .xml tag)
# returns a string of the file path. therefore the string is split to access only the final index with the "name.xml"
# method then parses the XML file and accesses the root (which should be "circuit") using the element tree API which was chosen instead of using minidom again because
# it is easier to access sub or child elements using element tree.
for components in circuit.findall("Component"):
Type = components.get("type")
Value = float(components.get("value"))
Name = components.get("name")
MetricPrefix = components.get("metricPrefix")
Rotated = components.get("rotated")
DialogIndex = int(components.get("DialogIndex"))
#accesses all the component attributes from the XML File and assigns them to a local variable for each component
PosNode = components.find("Node").text
componentPos= PosNode.split(",")
componentX = float(componentPos[0].replace("(", ''))
componentY = float(componentPos[1].replace(")", ''))
#accesses the string of the FIRST node subelement for each component, splits it and returns the float of the X and Y co-ordinates
# the following code the rebuilds the scene from the XML File by checking the component Type string and checking if the componenet is rotated
# it then re-adds the component to the scene using the original add component methods below and adds back all the original settings
# the position of the component is achieved by accessing the first node position, then setting the co-ordinates back to the origin point of the widget in the scene co-ordinates
# for horizontal components, the origin point is the top left hand corner of the widget, while for rotated components it is the top right hand corner
# (which is rotated by 90 degrees about the origin point of a horizontal widget)
if Type == "Resistor" and Rotated == "False":
self.addResistorButton()
componentY = componentY-49
self.sceneResistor.setPos(componentX, componentY)
loadWidget = self.sceneResistor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Capacitor" and Rotated == "False":
self.addCapacitorButton()
componentY = componentY-49
self.sceneCapacitor.setPos(componentX, componentY)
loadWidget = self.sceneCapacitor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Inductor" and Rotated == "False":
self.addInductorButton()
componentY = componentY-49
self.sceneInductor.setPos(componentX, componentY)
loadWidget = self.sceneInductor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Diode" and Rotated == "False":
self.addDiodeButton()
componentY = componentY-49
self.sceneDiode.setPos(componentX, componentY)
loadWidget = self.sceneDiode.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "DC Voltage Source" and Rotated == "False":
self.addVoltageButton()
componentY = componentY-49
self.sceneDCVoltageSource.setPos(componentX, componentY)
loadWidget = self.sceneDCVoltageSource.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Boundary Face" and Rotated == "False":
self.addBoundaryFaceButton()
componentY = componentY-49
self.sceneBoundaryFace.setPos(componentX, componentY)
loadWidget = self.sceneBoundaryFace.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Ground":
self.addGroundButton()
componentX = componentX - 88
self.sceneGround.setPos(componentX, componentY)
loadWidget = self.sceneGround.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
# ground button only has single node
# code for all the rotated buttons:
if Type == "Resistor" and Rotated == "True":
self.addResistorButton()
componentX = componentX+49
self.sceneResistor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneResistor.setTransform(self.transform)
loadWidget = self.sceneResistor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Capacitor" and Rotated == "True":
self.addCapacitorButton()
componentX = componentX+49
self.sceneCapacitor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneCapacitor.setTransform(self.transform)
loadWidget = self.sceneCapacitor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Inductor" and Rotated == "True":
self.addInductorButton()
componentX = componentX+49
self.sceneInductor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneInductor.setTransform(self.transform)
loadWidget = self.sceneInductor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Diode" and Rotated == "True":
self.addDiodeButton()
componentX = componentX+49
self.sceneDiode.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneDiode.setTransform(self.transform)
loadWidget = self.sceneDiode.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "DC Voltage Source" and Rotated == "True":
self.addVoltageButton()
componentX = componentX+49
self.sceneDCVoltageSource.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneDCVoltageSource.setTransform(self.transform)
loadWidget = self.sceneDCVoltageSource.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Boundary Face" and Rotated == "True":
self.addBoundaryFaceButton()
componentX = componentX+49
self.sceneBoundaryFace.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneBoundaryFace.setTransform(self.transform)
loadWidget = self.sceneBoundaryFace.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
for wires in circuit.findall("wire"):
wireText = wires.find("wirePos").text
wirePos = wireText.split(",")
wireX1 = float(wirePos[0])
wireY1 = float(wirePos[1])
wireX2 = float(wirePos[2])
wireY2 = float(wirePos[3])
self.wire = QtGui.QGraphicsLineItem(QtCore.QLineF(wireX1, wireY1, wireX2, wireY2))
self.scene.addItem(self.wire)
self.wire.setZValue(1)
wireCount = wireCount + 1
# all the wires in the XML file are accessed, the string of the start and end position is split into 2 pairs of co-ordinate values
#the x and y co-ordinates are then accessed from the respective string index and the wire is re-added and wirecount updated
self.statusbar.showMessage("XML File \"" + xmlFile + "\" Loaded Successfully", msecs = 5000)
except Exception as err:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: Unable to Load File \"" + xmlFile + "\"", msecs = 4000)
# a try function was intiated at the start of the method as if the user opens a non XML file, it will print "unable to load file" error in the status bar
# The following are the original add component methods from the mainwindow once svLPM is launched. These add the specified component to the scene
# A custom QPushButton is created from the DragButtons class from dragbuttons.py in the svLPModeller file.
# The componentType class member/attribute of the Dragbutton is updated to the type of component created
# The DragButton is then added to the scene and embedded into a QGraphicsProxyWidget
# the component count is increased to match the number of total components in the scene
def addResistorButton(self):
global componentCount
self.Resistor2 = DragButton('')
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.clicked.connect(self.handlebutton)
self.Resistor2.componentType = 'Resistor'
self.sceneResistor = self.scene.addWidget(self.Resistor2)
componentCount = componentCount + 1
def addCapacitorButton(self):
global componentCount
self.Capacitor2 = DragButton('')
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.clicked.connect(self.handlebutton)
self.Capacitor2.componentType = 'Capacitor'
self.sceneCapacitor = self.scene.addWidget(self.Capacitor2)
componentCount = componentCount + 1
def addInductorButton(self):
global componentCount
self.Inductor2 = DragButton('')
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.Inductor2.componentType = 'Inductor'
self.sceneInductor = self.scene.addWidget(self.Inductor2)
componentCount = componentCount + 1
def addDiodeButton(self):
global componentCount
self.Diode2 = DragButton('')
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.Diode2.componentType = 'Diode'
self.sceneDiode = self.scene.addWidget(self.Diode2)
componentCount = componentCount + 1
def addVoltageButton(self):
global componentCount
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DC Voltage Source"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.DCVoltageSource2.componentType = 'DC Voltage Source'
self.sceneDCVoltageSource=self.scene.addWidget(self.DCVoltageSource2)
componentCount = componentCount + 1
def addBoundaryFaceButton(self):
global componentCount
self.BoundaryFace2 = DragButton('')
self.BoundaryFace2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.BoundaryFace2.setAcceptDrops(False)
self.BoundaryFace2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("BoundaryFace.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.BoundaryFace2.setIcon(icon4)
self.BoundaryFace2.setIconSize(QtCore.QSize(139, 70))
self.BoundaryFace2.setFlat(True)
self.BoundaryFace2.setObjectName(_fromUtf8("Boundary Face"))
self.BoundaryFace2.clicked.connect(self.handlebutton)
self.BoundaryFace2.componentType = 'Boundary Face'
self.sceneBoundaryFace= self.scene.addWidget(self.BoundaryFace2)
componentCount = componentCount + 1
def addGroundButton(self):
global ground, componentCount
self.Ground2 = DragButton('')
self.Ground2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Ground2.setAcceptDrops(False)
self.Ground2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("Ground.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Ground2.setIcon(icon4)
self.Ground2.setIconSize(QtCore.QSize(139, 70))
self.Ground2.setFlat(True)
self.Ground2.setObjectName(_fromUtf8("Ground"))
self.Ground2.componentType = 'Ground'
self.Ground2.ground = True
self.sceneGround= self.scene.addWidget(self.Ground2)
componentCount = componentCount + 1
#all the components added when clicked then call the handlebutton method below which can implement a variety of functions
def handlebutton(self):
global componentCount
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
# This function allows the user to assign settings to a component
self.widgetSet = self.sender()
# returns the DragButton which called the handlebutton method
self.Settings = SettingDialog(self)
# returns the Component Setting Dialog
Type = self.widgetSet.componentType
Name = self.widgetSet.name
Value = self.widgetSet.value
ComboBoxIndex = self.widgetSet.comboBoxIndex
# assigns the existing class member/attributes of the Dragbutton sender to local variables
self.Settings.Type.setText(Type)
self.Settings.Name.setText(Name)
self.Settings.Value.setValue(Value)
self.Settings.Unit.setCurrentIndex(ComboBoxIndex)
# sets the lineEdits, label and the combobox to the current class member/attribute of the DragButton
# by default if they have not been set previously they are empty strings, integers or floats respectively
self.Settings.exec_()
# executes and launches the Dialog (must be .exec_ () to retrieve inputs and NOT .show())
self.widgetSet.name = self.Settings.Name.text()
self.widgetSet.value = self.Settings.Value.value()
self.widgetSet.unit = self.Settings.Unit.currentText()
self.widgetSet.comboBoxIndex = self.Settings.Unit.currentIndex()
# when save is pressed, sets the current inputs of lineedits, doublespinbox and the combobox to the class member/attributes of the DragButton
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
# This function allows the user to rotate a componenent: NOTE need to press middle button and then left click and ALT/OPTION
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.selectedItem.setTransform(self.transform)
# THe middle button must be Pressed first to assign a self.selectedItem, which accesses QProxyWidget.
#The QProxyWidget is then rotated (couldn't use self.sender() because the DragButton cannot be rotated)
rotated = self.sender()
rotated.rotated = True
# The DragButton which called the function is then accessed (Note not the QGraphicsProxyWidget which it is embedded in)
# The boolean rotated class attribute is then set as True, and so when the XML File is generated or rebuilt, the rotated attribute is "True"
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
# This function allows the user to delete the component and all associated settings
self.scene.removeItem(self.selectedItem )
# removes the selected component QGraphicsProxyWidget from the scene (and so middle press first)
componentCount = componentCount -1
class MainWindow(QMainWindow, svLumpedGui3A.Ui_MainWindow):
def __init__(self, parent=None,):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.scene = self.CircuitView.scene()
def deleteWire(self):
self.scene.removeItem(self.wire)
def mousePressEvent(self, event):
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.selectedItem = self.scene.itemAt(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
## if (self.clickY < (self.itemY +98)) & rotated == True:
## self.start = QtCore.QPointF(self.itemX -49, self.itemY)
## global rotated
## rotated += False
if self.clickY > (self.itemY +98):
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
else:
self.start = QtCore.QPointF(self.itemX +176, self.itemY + 49)
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
if self.clickY > (self.itemY +98):
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
else:
self.end = QtCore.QPointF(self.itemX +176, self.itemY + 49)
self.wire = QtGui.QGraphicsLineItem(QtCore.QLineF(self.start, self.end))
self.scene.addItem(self.wire)
if self.start == self.end:
self.scene.removeItem(self.wire)
else:
print self.start, self.end
def addResistorButton(self,):
self.Resistor2 = DragButton('')
## self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.clicked.connect(self.handlebutton)
## self.Resistor2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
## self.Resistor2.customContextMenuRequested.connect(self.handlebutton)
self.scene_Resistor = self.scene.addWidget(self.Resistor2)
def addCapacitorButton(self):
self.Capacitor2 = DragButton('')
## self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.setCheckable(True)
self.Capacitor2.setChecked(False)
self.Capacitor2.clicked.connect(self.handlebutton)
self.scene_Capacitor = self.scene.addWidget(self.Capacitor2)
def addInductorButton(self):
self.Inductor2 = DragButton('')
## self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.scene.addWidget(self.Inductor2)
def addDiodeButton(self):
self.Diode2 = DragButton('')
## self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.scene.addWidget(self.Diode2)
def addVoltageButton(self):
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DCVoltageSource"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.scene.addWidget(self.DCVoltageSource2)
def handlebutton(self):
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
form = SettingDialog(self)
form.show()
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
transform = QtGui.QTransform()
transform.rotate(90)
self.selectedItem.setTransform(transform)
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
self.scene.removeItem(self.selectedItem)
class MainWindow(QMainWindow, svLumpedGui4.Ui_MainWindow):
def __init__(self, SettingDialog, parent=None,):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.BoundaryFace.clicked.connect(self.addBoundaryFaceButton)
self.Ground.clicked.connect(self.addGroundButton)
self.scene = self.CircuitView.scene()
self.actionOpen.triggered.connect(self.openFile)
self.actionSave_2.triggered.connect(self.XMLFileSave)
self.actionSaveAs.triggered.connect(self.fileSaveAs)
self.actionQuit_2.triggered.connect(self.Quit)
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(0,0,0,0);color:black;font-weight:bold;}")
self.statusbar.messageChanged.connect(self.statusbarChanged)
def mousePressEvent(self, event):
global wireCount
## try:
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.selectedItem = self.scene.itemAt(self.start)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if (self.clickY > (self.itemY + 88)) &self.spock.rotated == True :
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY +88)) & self.spock.rotated == True:
self.start = QtCore.QPointF(self.itemX -49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX +176, self.itemY + 49)
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if self.spock.componentType == "Ground":
self.end = QtCore.QPointF(self.itemX +88, self.itemY)
elif (self.clickY > (self.itemY + 88)) & self.spock.rotated == True:
self.end = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY +88)) & self.spock.rotated == True:
self.end = QtCore.QPointF(self.itemX -49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX +176, self.itemY + 49)
wire = QtCore.QLineF(self.start, self.end)
self.wire = QtGui.QGraphicsLineItem(wire)
self.scene.addItem(self.wire)
if self.start == self.end:
self.scene.removeItem(self.wire)
else:
wireCount = wireCount+1
self.wire.setZValue(1)
## except Exception as err:
## self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
## self.statusbar.showMessage("Error: Component Not Pressed!", msecs = 1500)
def statusbarChanged(self, args):
if not args:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(0,0,0,0);color:black;font-weight:bold;}")
def closeEvent(self, event):
self.Quit()
def deleteWire(self):
global wireCount
if wireCount > 0:
deleteWire = self.scene.items()[0]
self.scene.removeItem(deleteWire)
wireCount = wireCount - 1
else:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Wires Exist (yet!)", msecs = 1500)
def fileSaveAs(self):
self.fileName = str(QtGui.QFileDialog.getSaveFileName(self, "Save File as XML"))
self.XMLFileSave()
def Quit(self):
self.confirmExit = QtGui.QMessageBox()
self.confirmExit.setIcon(QtGui.QMessageBox.Warning)
self.confirmExit.setText("Are you sure you want to Quit?")
self.confirmExit.setInformativeText("Any changes will be lost without saving!")
self.confirmExit.setStandardButtons(QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel)
self.confirmExit.buttonClicked.connect(self.MessageBoxButton)
self.confirmExit.exec_()
def MessageBoxButton(self, button):
buttonPressed = button.text()
if buttonPressed == "Cancel":
self.confirmExit.close()
elif buttonPressed == "Yes":
sys.exit()
def XMLFileSave(self):
global wireCount, componentCount, fileOpened, fileSaved
allComponents = self.scene.items()
totalComponents = wireCount + componentCount
root = minidom.Document()
xml = root.createElement('circuit')
root.appendChild(xml)
if componentCount != 0 and wireCount != 0:
for QtGui.QGraphicsProxyWidget in allComponents[wireCount:totalComponents]:
ComponentWidget = QtGui.QGraphicsProxyWidget.widget()
if ComponentWidget.componentType == "Ground":
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() +88), (QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ''
elif ComponentWidget.rotated == True :
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49), (QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() -49), (QtGui.QGraphicsProxyWidget.pos().y() + 176))
else:
node1 = (QtGui.QGraphicsProxyWidget.pos().x(), (QtGui.QGraphicsProxyWidget.pos().y() + 49))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() + 176), (QtGui.QGraphicsProxyWidget.pos().y() +49))
xmlcomponents = root.createElement('Component')
xmlcomponents.setAttribute('type', ComponentWidget.componentType)
xmlcomponents.setAttribute('value', str(ComponentWidget.value))
xmlcomponents.setAttribute('metricPrefix', str(ComponentWidget.unit))
xmlcomponents.setAttribute('name', str(ComponentWidget.name))
xmlcomponents.setAttribute('rotated', str(ComponentWidget.rotated))
xmlcomponents.setAttribute('DialogIndex', str(ComponentWidget.comboBoxIndex))
xml.appendChild(xmlcomponents)
componentNode1 = root.createElement("Node")
componentNode1.appendChild(root.createTextNode(str(node1)))
xmlcomponents.appendChild(componentNode1)
componentNode2 = root.createElement("Node")
componentNode2.appendChild(root.createTextNode(str(node2)))
xmlcomponents.appendChild(componentNode2)
for wires in allComponents[0:wireCount]:
WirePos = wires.line()
WirePos = str(WirePos)
stringLength = len(WirePos)-1
wire = WirePos[20:stringLength]
xmlwire = root.createElement('wire')
xml.appendChild(xmlwire)
wirePos = root.createElement('wirePos')
wirePos.appendChild(root.createTextNode(wire))
xmlwire.appendChild(wirePos)
xmlStr = root.toprettyxml(indent = '\t')
try:
if self.sender() == self.actionSaveAs:
self.savePathFile = self.fileName + ".xml"
elif self.sender() == self.actionSave_2:
self.savePathFile = self.fileName
with open(self.savePathFile, 'w') as f:
f.write(xmlStr)
self.statusbar.showMessage("Circuit Saved to XML File", msecs = 3000)
fileSaved = True
except Exception as err:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: XML File does not exist! : Use Save As", msecs = 4000)
elif componentCount == 0:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Components Added!", msecs = 1600)
elif wireCount == 0:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: No Wires Added!", msecs = 1600)
def openFile(self):
global wireCount, fileOpened
try:
self.fileName = str(QtGui.QFileDialog.getOpenFileName())
nameSplit = self.fileName.split("/") or self.fileName.split ("\\")
nameSplitNumber = (len(nameSplit)-1)
xmlFile = nameSplit[nameSplitNumber]
xmlLoad= ET.parse(xmlFile)
circuit = xmlLoad.getroot()
for components in circuit.findall("Component"):
Type = components.get("type")
Value = float(components.get("value"))
Name = components.get("name")
MetricPrefix = components.get("metricPrefix")
Rotated = components.get("rotated")
DialogIndex = int(components.get("DialogIndex"))
PosNode = components.find("Node").text
componentPos= PosNode.split(",")
componentX = float(componentPos[0].replace("(", ''))
componentY = float(componentPos[1].replace(")", ''))
if Type == "Resistor" and Rotated == "False":
self.addResistorButton()
componentY = componentY-49
self.sceneResistor.setPos(componentX, componentY)
loadWidget = self.sceneResistor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Capacitor" and Rotated == "False":
self.addCapacitorButton()
componentY = componentY-49
self.sceneCapacitor.setPos(componentX, componentY)
loadWidget = self.sceneCapacitor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Inductor" and Rotated == "False":
self.addInductorButton()
componentY = componentY-49
self.sceneInductor.setPos(componentX, componentY)
loadWidget = self.sceneInductor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Diode" and Rotated == "False":
self.addDiodeButton()
componentY = componentY-49
self.sceneDiode.setPos(componentX, componentY)
loadWidget = self.sceneDiode.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "DC Voltage Source" and Rotated == "False":
self.addVoltageButton()
componentY = componentY-49
self.sceneDCVoltageSource.setPos(componentX, componentY)
loadWidget = self.sceneDCVoltageSource.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Boundary Face" and Rotated == "False":
self.addBoundaryFaceButton()
componentY = componentY-49
self.sceneBoundaryFace.setPos(componentX, componentY)
loadWidget = self.sceneBoundaryFace.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Ground":
self.addGroundButton()
componentY = componentY-49
componentY = componentY + 49
componentX = componentX - 88
self.sceneGround.setPos(componentX, componentY)
loadWidget = self.sceneGround.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
if Type == "Resistor" and Rotated == "True":
self.addResistorButton()
componentX = componentX+49
self.sceneResistor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneResistor.setTransform(self.transform)
loadWidget = self.sceneResistor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Capacitor" and Rotated == "True":
self.addCapacitorButton()
componentX = componentX+49
self.sceneCapacitor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneCapacitor.setTransform(self.transform)
loadWidget = self.sceneCapacitor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Inductor" and Rotated == "True":
self.addInductorButton()
componentX = componentX+49
self.sceneInductor.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneInductor.setTransform(self.transform)
loadWidget = self.sceneInductor.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Diode" and Rotated == "True":
self.addDiodeButton()
componentX = componentX+49
self.sceneDiode.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneDiode.setTransform(self.transform)
loadWidget = self.sceneDiode.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "DC Voltage Source" and Rotated == "True":
self.addVoltageButton()
componentX = componentX+49
self.sceneDCVoltageSource.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneDCVoltageSource.setTransform(self.transform)
loadWidget = self.sceneDCVoltageSource.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
if Type == "Boundary Face" and Rotated == "True":
self.addBoundaryFaceButton()
componentX = componentX+49
self.sceneBoundaryFace.setPos(componentX, componentY)
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.sceneBoundaryFace.setTransform(self.transform)
loadWidget = self.sceneBoundaryFace.widget()
loadWidget.value = Value
loadWidget.name = Name
loadWidget.unit = MetricPrefix
loadWidget.comboBoxIndex = DialogIndex
loadWidget.rotated = True
for wires in circuit.findall("wire"):
wireText = wires.find("wirePos").text
wirePos = wireText.split(",")
wireX1 = float(wirePos[0])
wireY1 = float(wirePos[1])
wireX2 = float(wirePos[2])
wireY2 = float(wirePos[3])
self.wire = QtGui.QGraphicsLineItem(QtCore.QLineF(wireX1, wireY1, wireX2, wireY2))
self.scene.addItem(self.wire)
self.wire.setZValue(1)
wireCount = wireCount + 1
self.statusbar.showMessage("XML File \"" + xmlFile + "\" Loaded Successfully", msecs = 5000)
fileOpened = True
except Exception as err:
self.statusbar.setStyleSheet("QStatusBar{padding-left:8px;background:rgba(255,0,0,255);color:black;font-weight:bold;}")
self.statusbar.showMessage("Error: Unable to Load File \"" + xmlFile + "\"", msecs = 4000)
def addResistorButton(self):
global componentCount
self.Resistor2 = DragButton('')
## self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.clicked.connect(self.handlebutton)
self.Resistor2.componentType = 'Resistor'
self.sceneResistor = self.scene.addWidget(self.Resistor2)
componentCount = componentCount + 1
def addCapacitorButton(self):
global componentCount
self.Capacitor2 = DragButton('')
## self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.clicked.connect(self.handlebutton)
self.Capacitor2.componentType = 'Capacitor'
self.sceneCapacitor = self.scene.addWidget(self.Capacitor2)
componentCount = componentCount + 1
def addInductorButton(self):
global componentCount
self.Inductor2 = DragButton('')
## self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.Inductor2.componentType = 'Inductor'
self.sceneInductor = self.scene.addWidget(self.Inductor2)
componentCount = componentCount + 1
def addDiodeButton(self):
global componentCount
self.Diode2 = DragButton('')
## self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.Diode2.componentType = 'Diode'
self.sceneDiode = self.scene.addWidget(self.Diode2)
componentCount = componentCount + 1
def addVoltageButton(self):
global componentCount
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DC Voltage Source"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.DCVoltageSource2.componentType = 'DC Voltage Source'
self.sceneDCVoltageSource=self.scene.addWidget(self.DCVoltageSource2)
componentCount = componentCount + 1
def addBoundaryFaceButton(self):
global componentCount
self.BoundaryFace2 = DragButton('')
self.BoundaryFace2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.BoundaryFace2.setAcceptDrops(False)
self.BoundaryFace2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("BoundaryFace.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.BoundaryFace2.setIcon(icon4)
self.BoundaryFace2.setIconSize(QtCore.QSize(139, 70))
self.BoundaryFace2.setFlat(True)
self.BoundaryFace2.setObjectName(_fromUtf8("Boundary Face"))
self.BoundaryFace2.clicked.connect(self.handlebutton)
self.BoundaryFace2.componentType = 'Boundary Face'
self.sceneBoundaryFace= self.scene.addWidget(self.BoundaryFace2)
componentCount = componentCount + 1
def addGroundButton(self):
global ground, componentCount
self.Ground2 = DragButton('')
self.Ground2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Ground2.setAcceptDrops(False)
self.Ground2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("Ground.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Ground2.setIcon(icon4)
self.Ground2.setIconSize(QtCore.QSize(139, 70))
self.Ground2.setFlat(True)
self.Ground2.setObjectName(_fromUtf8("Ground"))
self.Ground2.componentType = 'Ground'
self.Ground2.ground = True
self.sceneGround= self.scene.addWidget(self.Ground2)
componentCount = componentCount + 1
def handlebutton(self):
global componentCount
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
self.widgetSet = self.sender()
self.Settings = SettingDialog(self)
Type = self.widgetSet.componentType
Name = self.widgetSet.name
Value = self.widgetSet.value
ComboBoxIndex = self.widgetSet.comboBoxIndex
self.Settings.Type.setText(Type)
self.Settings.Name.setText(Name)
self.Settings.Value.setValue(Value)
self.Settings.Unit.setCurrentIndex(ComboBoxIndex)
self.Settings.exec_()
self.widgetSet.name = self.Settings.Name.text()
self.widgetSet.value = self.Settings.Value.value()
self.widgetSet.unit = self.Settings.Unit.currentText()
self.widgetSet.comboBoxIndex = self.Settings.Unit.currentIndex()
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.selectedItem.setTransform(self.transform)
rotated = self.sender()
rotated.rotated = True
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
self.scene.removeItem(self.selectedItem )
componentCount = componentCount -1
class MainWindow(QMainWindow, svLumpedGui2.Ui_MainWindow):
def __init__(
self,
parent=None,
):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
def mousePressEvent(self, event):
self.__mousePressPos = None
if event.button() == QtCore.Qt.RightButton:
self.x1 = event.globalPos().x()
self.y1 = event.globalPos().y()
self.__mousePressPos = None
if event.button() == QtCore.Qt.RightButton:
QtGui.QPushButton.mousePressEvent = self.getPos2
def getPos2(self, event):
self.x2 = event.globalPos().x()
self.y2 = event.globalPos().y()
print self.x1, self.y1, self.x2, self.y2
## def paintEvent(self, e):
##
## qp = QtGui.QPainter()
## qp.begin(self)
## self.drawLines(qp)
## qp.end()
##
## def drawLines(self, qp):
##
## pen = QtGui.QPen(QtCore.Qt.black, 2, QtCore.Qt.SolidLine)
##
## qp.setPen(pen)
## qp.drawLine(self.x1, self.y1, self.x2, self.y2)
def addResistorButton(self):
self.Resistor2 = DragButton('', self)
self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 67))
self.Resistor2.setFlat(True)
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setCheckable(True)
self.Resistor2.setChecked(False)
self.Resistor2.show()
self.Resistor2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.Resistor2.customContextMenuRequested.connect(self.handlebutton)
def addCapacitorButton(self):
self.Capacitor2 = DragButton('', self)
self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(150, 60))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.setCheckable(True)
self.Capacitor2.setChecked(False)
self.Capacitor2.show()
self.Capacitor2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.Capacitor2.customContextMenuRequested.connect(self.handlebutton)
def addInductorButton(self):
self.Inductor2 = DragButton('', self)
self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(135, 80))
self.Inductor2.setFlat(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.show()
self.Inductor2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.Inductor2.customContextMenuRequested.connect(self.handlebutton)
def addDiodeButton(self):
self.Diode2 = DragButton('', self)
self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap
(QtGui.QPixmap(_fromUtf8("Diode.png")), QtGui.QIcon.Normal,
QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(200, 67))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.show()
self.Diode2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.Diode2.customContextMenuRequested.connect(self.handlebutton)
def addVoltageButton(self):
self.DCVoltageSource2 = DragButton('', self)
self.DCVoltageSource2.setGeometry(QtCore.QRect(70, 350, 71, 71))
self.DCVoltageSource2.setCursor(
QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(100, 60))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DCVoltageSource"))
self.DCVoltageSource2.show()
self.DCVoltageSource2.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.DCVoltageSource2.customContextMenuRequested.connect(
self.handlebutton)
def handlebutton(self):
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
form = SettingDialog(self)
form.show()
class MainWindow(QMainWindow, svLumpedGui3A.Ui_MainWindow):
def __init__(
self,
SettingDialog,
parent=None,
):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.BoundaryFace.clicked.connect(self.addBoundaryFaceButton)
self.Ground.clicked.connect(self.addGroundButton)
self.Export.clicked.connect(self.exportXML)
self.scene = self.CircuitView.scene()
self.actionOpen.triggered.connect(self.openFile)
self.actionSave.triggered.connect(self.exportXML)
def mousePressEvent(self, event):
global wireCount
try:
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.selectedItem = self.scene.itemAt(self.start)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if self.spock.isFlat() == False:
self.end = QtCore.QPointF(self.itemX + 88, self.itemY)
elif (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
self.wire = QtGui.QGraphicsLineItem(
QtCore.QLineF(self.start, self.end))
self.scene.addItem(self.wire)
if self.start == self.end:
self.scene.removeItem(self.wire)
else:
wireCount = wireCount + 1
self.wire.setZValue(1)
except Exception as err:
self.statusbar.showMessage("Error: Component Not Pressed!")
def deleteWire(self):
global wireCount
if wireCount > 0:
deleteWire = self.scene.items()[0]
self.scene.removeItem(deleteWire)
wireCount = wireCount - 1
else:
self.statusbar.showMessage("Error: No Wires Exist (yet!)")
def exportXML(self):
global wireCount, componentCount
allComponents = self.scene.items()
totalComponents = wireCount + componentCount
root = minidom.Document()
xml = root.createElement('circuit')
root.appendChild(xml)
if componentCount != 0 and wireCount != 0:
for QtGui.QGraphicsProxyWidget in allComponents[
wireCount:totalComponents]:
ComponentWidget = QtGui.QGraphicsProxyWidget.widget()
if ComponentWidget.isFlat() == False:
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() + 88),
(QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ''
elif ComponentWidget.isChecked():
node1 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49),
(QtGui.QGraphicsProxyWidget.pos().y()))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() - 49),
(QtGui.QGraphicsProxyWidget.pos().y() + 176))
else:
node1 = (QtGui.QGraphicsProxyWidget.pos().x(),
(QtGui.QGraphicsProxyWidget.pos().y() + 49))
node2 = ((QtGui.QGraphicsProxyWidget.pos().x() + 176),
(QtGui.QGraphicsProxyWidget.pos().y() + 49))
xmlcomponents = root.createElement('Component')
xmlcomponents.setAttribute('type',
ComponentWidget.componentType)
xmlcomponents.setAttribute('value', str(ComponentWidget.value))
xmlcomponents.setAttribute('metricPrefix',
str(ComponentWidget.unit))
xmlcomponents.setAttribute('name', str(ComponentWidget.name))
xmlcomponents.setAttribute('rotated',
str(ComponentWidget.rotated))
xml.appendChild(xmlcomponents)
componentNode1 = root.createElement("Node")
componentNode1.appendChild(root.createTextNode(str(node1)))
xmlcomponents.appendChild(componentNode1)
componentNode2 = root.createElement("Node")
componentNode2.appendChild(root.createTextNode(str(node2)))
xmlcomponents.appendChild(componentNode2)
for QtGui.QGraphicsLineItem in allComponents[0:wireCount]:
WirePos = QtGui.QGraphicsLineItem.line()
WirePos = str(WirePos)
stringLength = len(WirePos) - 1
wire = WirePos[20:stringLength]
xmlwire = root.createElement('wire')
xml.appendChild(xmlwire)
wirePos = root.createElement('wirePos')
wirePos.appendChild(root.createTextNode(wire))
xmlwire.appendChild(wirePos)
xmlStr = root.toprettyxml(indent='\t')
savePathFile = 'test.xml'
with open(savePathFile, 'w') as f:
f.write(xmlStr)
self.statusbar.showMessage("Circuit Exported to XML File")
elif componentCount == 0:
self.statusbar.showMessage("Error: No Components Added!")
elif wireCount == 0:
self.statusbar.showMessage("Error: No Wires Added!")
def openFile(self):
name = str(QtGui.QFileDialog.getOpenFileName())
nameSplit = name.split("/") or name.split("\\")
nameSplitNumber = (len(nameSplit) - 1)
xmlFile = nameSplit[nameSplitNumber]
xmlLoad = ET.parse(xmlFile)
circuit = xmlLoad.getroot()
for components in circuit.findall("Component"):
Type = components.get("type")
Node = components.find("Node").text
print Type, Node
for wires in circuit.findall("wire"):
wirePos = wires.find("wirePos").text
print wirePos
def addResistorButton(self):
global componentCount
self.Resistor2 = DragButton('')
## self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.setCheckable(True)
self.Resistor2.clicked.connect(self.handlebutton)
self.Resistor2.componentType = 'Resistor'
self.sceneResistor = self.scene.addWidget(self.Resistor2)
componentCount = componentCount + 1
def addCapacitorButton(self):
global componentCount
self.Capacitor2 = DragButton('')
## self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.setCheckable(True)
self.Capacitor2.setChecked(False)
self.Capacitor2.clicked.connect(self.handlebutton)
self.Capacitor2.componentType = 'Capacitor'
self.sceneCapacitor = self.scene.addWidget(self.Capacitor2)
componentCount = componentCount + 1
def addInductorButton(self):
global componentCount
self.Inductor2 = DragButton('')
## self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setCheckable(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.Inductor2.componentType = 'Inductor'
self.scene.addWidget(self.Inductor2)
componentCount = componentCount + 1
def addDiodeButton(self):
global componentCount
self.Diode2 = DragButton('')
## self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.Diode2.setCheckable(True)
self.Diode2.componentType = 'Diode'
self.scene.addWidget(self.Diode2)
componentCount = componentCount + 1
def addVoltageButton(self):
global componentCount
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(
QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DC Voltage Source"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.DCVoltageSource2.setCheckable(True)
self.DCVoltageSource2.componentType = 'DC Voltage Source'
self.scene.addWidget(self.DCVoltageSource2)
componentCount = componentCount + 1
def addBoundaryFaceButton(self):
global componentCount
self.BoundaryFace2 = DragButton('')
self.BoundaryFace2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.BoundaryFace2.setAcceptDrops(False)
self.BoundaryFace2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("BoundaryFace.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.BoundaryFace2.setIcon(icon4)
self.BoundaryFace2.setIconSize(QtCore.QSize(139, 70))
self.BoundaryFace2.setFlat(True)
self.BoundaryFace2.setObjectName(_fromUtf8("Boundary Face"))
self.BoundaryFace2.clicked.connect(self.handlebutton)
self.BoundaryFace2.componentType = 'Boundary Face'
self.scene.addWidget(self.BoundaryFace2)
componentCount = componentCount + 1
def addGroundButton(self):
global ground, componentCount
self.Ground2 = DragButton('')
self.Ground2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Ground2.setAcceptDrops(False)
self.Ground2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("Ground.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Ground2.setIcon(icon4)
self.Ground2.setIconSize(QtCore.QSize(139, 70))
self.Ground2.setFlat(False)
self.Ground2.setObjectName(_fromUtf8("Ground"))
self.Ground2.componentType = 'Ground'
self.scene.addWidget(self.Ground2)
componentCount = componentCount + 1
def handlebutton(self):
global componentCount
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
self.widgetSet = self.sender()
self.Settings = SettingDialog(self)
Type = self.widgetSet.componentType
self.Settings.Type.setText(Type)
self.Settings.exec_()
self.widgetSet.name = self.Settings.Name.text()
self.widgetSet.value = self.Settings.Value.value()
self.widgetSet.unit = self.Settings.Unit.currentText()
self.Settings.Name.clear()
self.Settings.Value.setValue(0)
self.Settings.Unit.setCurrentIndex(0)
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.selectedItem.setTransform(self.transform)
self.spock.setChecked(True)
self.rotatedButton = self.sender()
self.rotatedButton.rotated = True
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
self.scene.removeItem(self.selectedItem)
componentCount = componentCount - 1
class MainWindow(QMainWindow, svLumpedGui3A.Ui_MainWindow):
def __init__(
self,
SettingDialog,
parent=None,
):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.Resistor.clicked.connect(self.addResistorButton)
self.Capacitor.clicked.connect(self.addCapacitorButton)
self.Inductor.clicked.connect(self.addInductorButton)
self.Diode.clicked.connect(self.addDiodeButton)
self.DCVoltageSource.clicked.connect(self.addVoltageButton)
self.DeleteWire.clicked.connect(self.deleteWire)
self.CircuitView.setScene(QtGui.QGraphicsScene(self))
self.BoundaryFace.clicked.connect(self.addBoundaryFaceButton)
self.Ground.clicked.connect(self.addGroundButton)
self.Export.clicked.connect(self.exportXML)
## self.Export.clicked.connect(self.nameTest)
## self.Export.clicked.connect(self.posTest)
self.scene = self.CircuitView.scene()
def mousePressEvent(self, event):
global ground
if event.buttons() == QtCore.Qt.MidButton:
self.start = event.pos()
self.start = self.mapToGlobal(event.pos())
self.start = self.CircuitView.mapFromGlobal(self.start)
self.start = self.CircuitView.mapToScene(self.start)
self.clickX = self.start.x()
self.clickY = self.start.y()
self.selectedItem = self.scene.itemAt(self.start)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.start = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.start = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
if event.buttons() == QtCore.Qt.RightButton:
self.end = event.pos()
self.end = self.mapToGlobal(event.pos())
self.end = self.CircuitView.mapFromGlobal(self.end)
self.end = self.CircuitView.mapToScene(self.end)
self.clickX = self.end.x()
self.clickY = self.end.y()
self.selectedItem = self.scene.itemAt(self.end)
self.itemX = self.selectedItem.geometry().x()
self.itemY = self.selectedItem.geometry().y()
self.spock = self.selectedItem.widget()
if ground == True:
if self.spock == self.Ground2:
self.scene.removeItem(self.wire)
self.end = QtCore.QPointF(self.itemX + 88, self.itemY)
ground = False
elif (self.clickY > (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY + 176)
elif (self.clickY < (self.itemY + 88)) & self.spock.isChecked():
self.end = QtCore.QPointF(self.itemX - 49, self.itemY)
elif self.clickX < (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX, self.itemY + 49)
elif self.clickX > (self.itemX + 88):
self.end = QtCore.QPointF(self.itemX + 176, self.itemY + 49)
self.wire = QtGui.QGraphicsLineItem(QtCore.QLineF(
self.start, self.end))
self.scene.addItem(self.wire)
if self.start == self.end:
self.scene.removeItem(self.wire)
else:
print self.start, self.end
def deleteWire(self):
self.scene.removeItem(self.wire)
def exportXML(self):
Dialog = SettingDialog()
self.spock.name = Dialog.Name.text()
self.spock.value = Dialog.Value.value()
self.spock.unit = Dialog.Unit.currentIndex()
self.componentPos = self.selectedItem.pos()
print self.spock.componentType, self.spock.name, self.spock.value, self.spock.unit, self.componentPos
## def posTest(self):
## self.allItems = self.scene.items()
## print self.allItems
##
def addResistorButton(self):
self.Resistor2 = DragButton('')
## self.Resistor2.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor2.setAcceptDrops(False)
self.Resistor2.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor2.setIcon(icon)
self.Resistor2.setIconSize(QtCore.QSize(130, 70))
self.Resistor2.setObjectName(_fromUtf8("Resistor"))
self.Resistor2.setFlat(True)
self.Resistor2.setCheckable(True)
self.Resistor2.clicked.connect(self.handlebutton)
self.Resistor2.componentType = 'Resistor'
self.scene_Resistor = self.scene.addWidget(self.Resistor2)
## self.Resistor2.name = SettingDialog.name
## self.Resistor2.value = SettingDialog.value
## self.Resistor2.unit = SettingDialog.unit
def addCapacitorButton(self):
self.Capacitor2 = DragButton('')
## self.Capacitor2.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor2.setAcceptDrops(False)
self.Capacitor2.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor2.setIcon(icon1)
self.Capacitor2.setIconSize(QtCore.QSize(130, 70))
self.Capacitor2.setFlat(True)
self.Capacitor2.setObjectName(_fromUtf8("Capacitor"))
self.Capacitor2.setCheckable(True)
self.Capacitor2.setChecked(False)
self.Capacitor2.clicked.connect(self.handlebutton)
self.Capacitor2.componentType = 'Capacitor'
self.scene_Capacitor = self.scene.addWidget(self.Capacitor2)
def addInductorButton(self):
self.Inductor2 = DragButton('')
## self.Inductor2.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor2.setAcceptDrops(False)
self.Inductor2.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor2.setIcon(icon3)
self.Inductor2.setIconSize(QtCore.QSize(130, 70))
self.Inductor2.setFlat(True)
self.Inductor2.setCheckable(True)
self.Inductor2.setObjectName(_fromUtf8("Inductor"))
self.Inductor2.clicked.connect(self.handlebutton)
self.Inductor2.componentType = 'Inductor'
self.scene.addWidget(self.Inductor2)
def addDiodeButton(self):
self.Diode2 = DragButton('')
## self.Diode2.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode2.setAcceptDrops(False)
self.Diode2.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode2.setIcon(icon2)
self.Diode2.setIconSize(QtCore.QSize(130, 70))
self.Diode2.setFlat(True)
self.Diode2.setObjectName(_fromUtf8("Diode"))
self.Diode2.clicked.connect(self.handlebutton)
self.Diode2.setCheckable(True)
self.Diode2.componentType = 'Diode'
self.scene.addWidget(self.Diode2)
print self.Diode2.componentType
def addVoltageButton(self):
self.DCVoltageSource2 = DragButton('')
self.DCVoltageSource2.setCursor(
QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource2.setAcceptDrops(False)
self.DCVoltageSource2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource2.setIcon(icon4)
self.DCVoltageSource2.setIconSize(QtCore.QSize(139, 70))
self.DCVoltageSource2.setFlat(True)
self.DCVoltageSource2.setObjectName(_fromUtf8("DCVoltageSource"))
self.DCVoltageSource2.clicked.connect(self.handlebutton)
self.DCVoltageSource2.setCheckable(True)
self.DCVoltageSource2.componentType = 'DCVoltageSource'
self.scene.addWidget(self.DCVoltageSource2)
print self.DCVoltageSource2.componentType
def addBoundaryFaceButton(self):
self.BoundaryFace2 = DragButton('')
self.BoundaryFace2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.BoundaryFace2.setAcceptDrops(False)
self.BoundaryFace2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("BoundaryFace.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.BoundaryFace2.setIcon(icon4)
self.BoundaryFace2.setIconSize(QtCore.QSize(139, 70))
self.BoundaryFace2.setFlat(True)
self.BoundaryFace2.setObjectName(_fromUtf8("BoundaryFace"))
self.BoundaryFace2.clicked.connect(self.handlebutton)
self.BoundaryFace2.componentType = 'BoundaryFace'
self.scene.addWidget(self.BoundaryFace2)
print self.BoundaryFace2.componentType
def addGroundButton(self):
global ground
self.Ground2 = DragButton('')
self.Ground2.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Ground2.setAcceptDrops(False)
self.Ground2.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("Ground.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Ground2.setIcon(icon4)
self.Ground2.setIconSize(QtCore.QSize(139, 70))
self.Ground2.setFlat(True)
self.Ground2.setObjectName(_fromUtf8("Ground"))
self.Ground2.clicked.connect(self.handlebutton)
self.Ground2.componentType = 'Ground'
self.scene.addWidget(self.Ground2)
print self.Ground2.componentType
ground = True
def handlebutton(self):
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ShiftModifier:
form = SettingDialog(self)
form.show()
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.AltModifier:
self.transform = QtGui.QTransform()
self.transform.rotate(90)
self.selectedItem.setTransform(self.transform)
self.spock.setChecked(True)
if QtGui.qApp.keyboardModifiers() & QtCore.Qt.ControlModifier:
self.scene.removeItem(self.selectedItem)
def nameTest(self):
dialog = SettingDialog()
## name = dialog.name
## print name
print dialog.Name.text()
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName(_fromUtf8("MainWindow"))
MainWindow.resize(991, 626)
MainWindow.setAcceptDrops(True)
self.centralwidget = QtGui.QWidget(MainWindow)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.ComponentList = QtGui.QToolBox(self.centralwidget)
self.ComponentList.setGeometry(QtCore.QRect(0, 0, 231, 531))
self.ComponentList.setMouseTracking(True)
self.ComponentList.setAcceptDrops(True)
self.ComponentList.setObjectName(_fromUtf8("ComponentList"))
self.Components = QtGui.QWidget()
self.Components.setGeometry(QtCore.QRect(0, 0, 231, 497))
self.Components.setObjectName(_fromUtf8("Components"))
self.Resistor = DragButton(self.Components)
self.Resistor.setGeometry(QtCore.QRect(40, 30, 131, 61))
self.Resistor.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Resistor.setAcceptDrops(False)
self.Resistor.setText(_fromUtf8(""))
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8("Resistor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Resistor.setIcon(icon)
self.Resistor.setIconSize(QtCore.QSize(130, 67))
self.Resistor.setFlat(True)
self.Resistor.setObjectName(_fromUtf8("Resistor"))
self.Capacitor = DragButton(self.Components)
self.Capacitor.setGeometry(QtCore.QRect(80, 120, 51, 51))
self.Capacitor.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Capacitor.setAcceptDrops(False)
self.Capacitor.setText(_fromUtf8(""))
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8("capacitor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Capacitor.setIcon(icon1)
self.Capacitor.setIconSize(QtCore.QSize(150, 60))
self.Capacitor.setCheckable(False)
self.Capacitor.setFlat(True)
self.Capacitor.setObjectName(_fromUtf8("Capacitor"))
self.Diode = DragButton(self.Components)
self.Diode.setGeometry(QtCore.QRect(50, 190, 111, 71))
self.Diode.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Diode.setAcceptDrops(False)
self.Diode.setText(_fromUtf8(""))
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8("Diode.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Diode.setIcon(icon2)
self.Diode.setIconSize(QtCore.QSize(200, 67))
self.Diode.setFlat(True)
self.Diode.setObjectName(_fromUtf8("Diode"))
self.Inductor = DragButton(self.Components)
self.Inductor.setGeometry(QtCore.QRect(40, 280, 141, 51))
self.Inductor.setCursor(QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.Inductor.setAcceptDrops(False)
self.Inductor.setText(_fromUtf8(""))
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8("Inductor.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.Inductor.setIcon(icon3)
self.Inductor.setIconSize(QtCore.QSize(135, 80))
self.Inductor.setFlat(True)
self.Inductor.setObjectName(_fromUtf8("Inductor"))
self.DCVoltageSource = DragButton(self.Components)
self.DCVoltageSource.setGeometry(QtCore.QRect(70, 350, 71, 71))
self.DCVoltageSource.setCursor(
QtGui.QCursor(QtCore.Qt.ClosedHandCursor))
self.DCVoltageSource.setAcceptDrops(False)
self.DCVoltageSource.setText(_fromUtf8(""))
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(_fromUtf8("DCVoltageSource.png")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.DCVoltageSource.setIcon(icon4)
self.DCVoltageSource.setIconSize(QtCore.QSize(100, 60))
self.DCVoltageSource.setFlat(True)
self.DCVoltageSource.setObjectName(_fromUtf8("DCVoltageSource"))
self.ComponentList.addItem(self.Components, _fromUtf8(""))
self.frame = QtGui.QFrame(self.centralwidget)
self.frame.setGeometry(QtCore.QRect(239, -1, 561, 651))
self.frame.setFrameShape(QtGui.QFrame.StyledPanel)
self.frame.setFrameShadow(QtGui.QFrame.Raised)
self.frame.setObjectName(_fromUtf8("frame"))
self.widget = QtGui.QWidget(self.frame)
self.widget.setGeometry(QtCore.QRect(-10, -10, 581, 581))
self.widget.setAcceptDrops(True)
self.widget.setObjectName(_fromUtf8("widget"))
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtGui.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 991, 22))
self.menubar.setObjectName(_fromUtf8("menubar"))
self.menuFile = QtGui.QMenu(self.menubar)
self.menuFile.setObjectName(_fromUtf8("menuFile"))
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtGui.QStatusBar(MainWindow)
self.statusbar.setObjectName(_fromUtf8("statusbar"))
MainWindow.setStatusBar(self.statusbar)
self.actionNew = QtGui.QAction(MainWindow)
self.actionNew.setObjectName(_fromUtf8("actionNew"))
self.actionOpem = QtGui.QAction(MainWindow)
self.actionOpem.setObjectName(_fromUtf8("actionOpem"))
self.actionOpen = QtGui.QAction(MainWindow)
self.actionOpen.setObjectName(_fromUtf8("actionOpen"))
self.actionSave = QtGui.QAction(MainWindow)
self.actionSave.setObjectName(_fromUtf8("actionSave"))
self.actionQuit = QtGui.QAction(MainWindow)
self.actionQuit.setObjectName(_fromUtf8("actionQuit"))
self.menuFile.addAction(self.actionNew)
self.menuFile.addAction(self.actionOpen)
self.menuFile.addSeparator()
self.menuFile.addAction(self.actionSave)
self.menuFile.addSeparator()
self.menuFile.addAction(self.actionQuit)
self.menubar.addAction(self.menuFile.menuAction())
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
MainWindow.setWindowTitle(
QtGui.QApplication.translate("MainWindow", "svLumpedPM", None,
QtGui.QApplication.UnicodeUTF8))
self.Resistor.setToolTip(
QtGui.QApplication.translate("MainWindow", "Resistor", None,
QtGui.QApplication.UnicodeUTF8))
self.Resistor.setStatusTip(
QtGui.QApplication.translate("MainWindow", "Resistor Selected",
None, QtGui.QApplication.UnicodeUTF8))
self.Capacitor.setToolTip(
QtGui.QApplication.translate("MainWindow", "Capacitor", None,
QtGui.QApplication.UnicodeUTF8))
self.Capacitor.setStatusTip(
QtGui.QApplication.translate("MainWindow", "Capacitor Selected",
None, QtGui.QApplication.UnicodeUTF8))
self.Diode.setToolTip(
QtGui.QApplication.translate("MainWindow", "Diode", None,
QtGui.QApplication.UnicodeUTF8))
self.Diode.setStatusTip(
QtGui.QApplication.translate("MainWindow", "Diode Selected", None,
QtGui.QApplication.UnicodeUTF8))
self.Inductor.setToolTip(
QtGui.QApplication.translate("MainWindow", "Inductor", None,
QtGui.QApplication.UnicodeUTF8))
self.Inductor.setStatusTip(
QtGui.QApplication.translate("MainWindow", "Inductor Selected",
None, QtGui.QApplication.UnicodeUTF8))
self.DCVoltageSource.setToolTip(
QtGui.QApplication.translate("MainWindow", "DC Voltage Source",
None, QtGui.QApplication.UnicodeUTF8))
self.DCVoltageSource.setStatusTip(
QtGui.QApplication.translate("MainWindow",
"DC Voltage Source Selected", None,
QtGui.QApplication.UnicodeUTF8))
self.ComponentList.setItemText(
self.ComponentList.indexOf(self.Components),
QtGui.QApplication.translate("MainWindow", "Circuit Components",
None, QtGui.QApplication.UnicodeUTF8))
self.menuFile.setTitle(
QtGui.QApplication.translate("MainWindow", "File", None,
QtGui.QApplication.UnicodeUTF8))
self.actionNew.setText(
QtGui.QApplication.translate("MainWindow", "New", None,
QtGui.QApplication.UnicodeUTF8))
self.actionOpem.setText(
QtGui.QApplication.translate("MainWindow", "Opem", None,
QtGui.QApplication.UnicodeUTF8))
self.actionOpen.setText(
QtGui.QApplication.translate("MainWindow", "Open", None,
QtGui.QApplication.UnicodeUTF8))
self.actionSave.setText(
QtGui.QApplication.translate("MainWindow", "Save", None,
QtGui.QApplication.UnicodeUTF8))
self.actionQuit.setText(
QtGui.QApplication.translate("MainWindow", "Quit", None,
QtGui.QApplication.UnicodeUTF8))
