from PyQt4 import QtGui from . import TrackWidget from projManagement import Validation import os from xml.etree import ElementTree as ET class SubcircuitTab(QtGui.QWidget): """ - This class creates Subcircuit Tab in KicadtoNgspice Window - It dynamically creates the widget for subcircuits, according to the .cir file - Creates `lineEdit` and `Add` button, which triggers `fileSelector` - Also, checks `Previous_value.xml` for previous subcircuit value to autofill, the `lineEdit` - Add button is bind to `trackSubcircuit` - Also `trackSubcircuit` without button is triggered if `lineEdit` filled """ def __init__(self, schematicInfo, clarg1): kicadFile = clarg1 (projpath, filename) = os.path.split(kicadFile) project_name = os.path.basename(projpath) try: f = open( os.path.join( projpath, project_name + "_Previous_Values.xml"), 'r') tree = ET.parse(f) parent_root = tree.getroot() for child in parent_root: if child.tag == "subcircuit": root = child except BaseException: print("Subcircuit Previous values XML is Empty") QtGui.QWidget.__init__(self) # Creating track widget object self.obj_trac = TrackWidget.TrackWidget() # Creating validation object self.obj_validation = Validation.Validation() # Row and column count self.row = 0 self.count = 1 # Entry count self.entry_var = {} self.subcircuit_dict_beg = {} self.subcircuit_dict_end = {} # List to hold information about subcircuit self.subDetail = {} # Stores the number of ports in each subcircuit self.numPorts = [] # Set Layout self.grid = QtGui.QGridLayout() self.setLayout(self.grid) for eachline in schematicInfo: words = eachline.split() if eachline[0] == 'x': # print("Subcircuit : Words", words[0]) self.obj_trac.subcircuitList[project_name + words[0]] = words self.subcircuit_dict_beg[words[0]] = self.count subbox = QtGui.QGroupBox() subgrid = QtGui.QGridLayout() subbox.setTitle("Add subcircuit for " + words[len(words) - 1]) self.entry_var[self.count] = QtGui.QLineEdit() self.entry_var[self.count].setText("") global path_name try: for child in root: if child.tag[0] == eachline[0] \ and child.tag[1] == eachline[1]: # print("Subcircuit MATCHING---", child.tag[0], \ # child.tag[1], eachline[0],eachline[1]) try: if child[0].text \ and os.path.exists(child[0].text): self.entry_var[self.count] \ .setText(child[0].text) path_name = child[0].text else: self.entry_var[self.count].setText("") except BaseException as e: print("Error when set text of " + "subcircuit :", str(e)) except BaseException as e: print("Error before subcircuit :", str(e)) subgrid.addWidget(self.entry_var[self.count], self.row, 1) self.addbtn = QtGui.QPushButton("Add") self.addbtn.setObjectName("%d" % self.count) # Send the number of ports specified with the given\ # subcircuit for verification. # eg. If the line is 'x1 4 0 3 ua741', there are 3 ports(4, 0 # and 3). self.numPorts.append(len(words) - 2) # print("Number of ports of sub circuit : ", self.numPorts) self.addbtn.clicked.connect(self.trackSubcircuit) subgrid.addWidget(self.addbtn, self.row, 2) subbox.setLayout(subgrid) # CSS subbox.setStyleSheet(" \ QGroupBox { border: 1px solid gray; border-radius:\ 9px; margin-top: 0.5em; } \ QGroupBox::title { subcontrol-origin: margin; left:\ 10px; padding: 0 3px 0 3px; } \ ") self.grid.addWidget(subbox) # Adding Subcircuit Details self.subDetail[self.count] = words[0] # Increment row and widget count if self.entry_var[self.count].text() == "": pass else: self.trackSubcircuitWithoutButton(self.count, path_name) self.subcircuit_dict_end[words[0]] = self.count self.row = self.row + 1 self.count = self.count + 1 self.show() def trackSubcircuit(self): """ - This function is use to keep track of all Subcircuit widget - Here the number of ports is tracked using the numPorts and `Add` button objectName property, which is refered using `sender` - Once a file is selected using the `QFileDialog` validate it - Pass the path of subciruit and the number of ports - According to validation state take further steps - If validated correctly, add to TrackWidget """ sending_btn = self.sender() # print "Object Called is ",sending_btn.objectName() self.widgetObjCount = int(sending_btn.objectName()) self.subfile = str( QtGui.QFileDialog.getExistingDirectory( self, "Open Subcircuit", "library/SubcircuitLibrary") ) self.reply = self.obj_validation.validateSub( self.subfile, self.numPorts[self.widgetObjCount - 1]) if self.reply == "True": # Setting Library to Text Edit Line self.entry_var[self.widgetObjCount].setText(self.subfile) self.subName = self.subDetail[self.widgetObjCount] # Storing to track it during conversion self.obj_trac.subcircuitTrack[self.subName] = self.subfile elif self.reply == "PORT": self.msg = QtGui.QErrorMessage(self) self.msg.setModal(True) self.msg.setWindowTitle("Error Message") self.msg.showMessage( "Please select a Subcircuit with correct number of ports.") self.msg.exec_() elif self.reply == "DIREC": self.msg = QtGui.QErrorMessage(self) self.msg.setModal(True) self.msg.setWindowTitle("Error Message") self.msg.showMessage( "Please select a valid Subcircuit directory " "(Containing '.sub' file).") self.msg.exec_() def trackSubcircuitWithoutButton(self, iter_value, path_value): """ - Same as trackSubcircuit, but here the count value is passed directly without using any button as in `Add` - This is triggered only once, initally """ self.widgetObjCount = iter_value self.subfile = path_value self.reply = self.obj_validation.validateSub( self.subfile, self.numPorts[self.widgetObjCount - 1]) if self.reply == "True": # Setting Library to Text Edit Line self.entry_var[self.widgetObjCount].setText(self.subfile) self.subName = self.subDetail[self.widgetObjCount] # Storing to track it during conversion self.obj_trac.subcircuitTrack[self.subName] = self.subfile elif self.reply == "PORT": self.msg = QtGui.QErrorMessage(self) self.msg.setModal(True) self.msg.setWindowTitle("Error Message") self.msg.showMessage( "Please select a Subcircuit with correct number of ports.") self.msg.exec_() elif self.reply == "DIREC": self.msg = QtGui.QErrorMessage(self) self.msg.setModal(True) self.msg.setWindowTitle("Error Message") self.msg.showMessage( "Please select a valid Subcircuit directory " "(Containing '.sub' file).") self.msg.exec_()