class ToolBox(QTreeWidget):
"""A toolbox that contains logic gates and user circuits, for use
in the main designer window.
"""
def __init__(self):
super(ToolBox, self).__init__()
self.setDragEnabled(True)
self.setColumnCount(2)
self.header().setVisible(False)
gatesheader = QTreeWidgetItem(self, [self.str_basicGates])
gatesheader.setFlags(
~Qt.ItemFlag.ItemIsDragEnabled & ~Qt.ItemFlag.ItemIsSelectable)
gatesheader.setExpanded(True)
imgDir = filePath('icons/')
for name, class_ in inspect.getmembers(
gates,
lambda m: (
inspect.isclass(m) and m.__module__ == 'engine.gates')):
item = QTreeWidgetItem(gatesheader, [name[:-4]])
item.setIcon(0, QIcon(imgDir + name + '.png'))
item = QTreeWidgetItem(gatesheader, ['JKFlipFlop'])
item.setIcon(0, QIcon(imgDir + 'JKFlipFlop.png'))
item = QTreeWidgetItem(gatesheader, ['RSFlipFlop'])
item.setIcon(0, QIcon(imgDir + 'RSFlipFlop.png'))
ioheader = QTreeWidgetItem(self, [self.str_IO])
ioheader.setFlags(
~Qt.ItemFlag.ItemIsDragEnabled & ~Qt.ItemFlag.ItemIsSelectable)
ioheader.setExpanded(True)
[
QTreeWidgetItem(ioheader, [name])
for name in [self.str_I, self.str_O, self.str_Clock]]
self.userheader = QTreeWidgetItem(self, [self.str_userCircuits])
[QTreeWidgetItem(self.userheader, [name[:-4], 'user'])
for name in sorted(listdir(filePath('user/')))]
self.userheader.setFlags(
~Qt.ItemFlag.ItemIsDragEnabled & ~Qt.ItemFlag.ItemIsSelectable)
self.userheader.setExpanded(True)
self.setColumnWidth(0, 300)
def addUserCircuit(self, name):
"""When the user saves a circuit, add it at the correct
alphabetical spot, if it is not already present."""
for i in range(self.userheader.childCount()):
if self.userheader.child(i).text(0) == name:
return
QTreeWidgetItem(self.userheader, [name, 'user'])
self.userheader.sortChildren(0, Qt.AscendingOrder)
