class CommandStack(collections.abc.Collection):
def __init__(self):
self._parent = QObject(None)
self._index = 0
self._command_number = 1
def __len__(self):
return len(self._parent.children())
def __iter__(self):
self._index = 0
return self
def __next__(self):
if len(self) == 0:
raise StopIteration
if self._index < len(self):
self._index += 1
return self._parent.children()[self._index - 1]
raise StopIteration
def __contains__(self, item):
if isinstance(item, str):
# testing by string name and not reference
item_name = item
find_me = self._parent.findChild(item_name)
if find_me is not None:
return True
return False
# testing by references
for itm in self:
if item is itm:
return True
return False
def __eq__(self, other):
equals = [
len(self) == len(other),
]
equals = equals + list(
map(lambda x, y: x is y, iter(self), iter(other)))
return all(equals)
def __del__(self):
self._parent.deleteLater()
def push(self, command):
if not isinstance(command, QObject):
raise TypeError
if command.parent() is not None:
raise AttributeError('Command QObject already has a parent!')
command.setParent(self._parent)
command.setObjectName('Command_{}'.format(self._command_number))
self._command_number += 1
def pop(self):
last_index = len(self) - 1
last_guy = self._parent.children()[last_index]
self._command_number -= 1
last_guy.setParent(None)
last_guy.setObjectName(None)
return last_guy
class PolyFinderGUI(QMainWindow):
updated = pyqtSignal()
resetting = pyqtSignal()
function = -1
end = False
def enable_start_btn(self, index):
self.ui.pushBtn.setDisabled(False)
self.worker.set_data(index)
self.ui.f0Btn.setDisabled(True)
self.ui.f1Btn.setDisabled(True)
self.ui.f2Btn.setDisabled(True)
self.ui.f3Btn.setDisabled(True)
self.function = index
def enable_stop(self):
self.ui.pushBtn.setDisabled(True)
self.ui.stopBtn.setDisabled(False)
def reset(self):
self.ui.f0Btn.setDisabled(False)
self.ui.f1Btn.setDisabled(False)
self.ui.f2Btn.setDisabled(False)
self.ui.f3Btn.setDisabled(False)
self.ui.stopBtn.setDisabled(True)
self.resetting.emit()
def set_end(self, value):
self.end = value
def __init__(self):
super().__init__()
self.ui = Ui_Dialog()
self.ui.setupUi(self)
self.worker = genal.PolyFinder()
self.worker_thread = QThread()
self.worker.moveToThread(self.worker_thread)
self.worker_thread.start()
self.worker.generated.connect(self.update_graph)
self.worker.initialized.connect(self.worker.start_crunching)
self.ui.pushBtn.clicked.connect(self.worker.initialize)
self.ui.pushBtn.clicked.connect(self.enable_stop)
self.ui.stopBtn.clicked.connect(self.worker.finish)
self.ui.stopBtn.clicked.connect(
lambda: self.ui.stopBtn.setDisabled(True))
self.ui.stopBtn.clicked.connect(
lambda: self.ui.rstBtn.setDisabled(False))
self.ui.stopBtn.clicked.connect(
lambda: self.ui.quitBtn.setDisabled(False))
self.ui.stopBtn.clicked.connect(lambda: self.set_end(True))
self.ui.f0Btn.clicked.connect(lambda: self.enable_start_btn(0))
self.ui.f1Btn.clicked.connect(lambda: self.enable_start_btn(1))
self.ui.f2Btn.clicked.connect(lambda: self.enable_start_btn(2))
self.ui.f3Btn.clicked.connect(lambda: self.enable_start_btn(3))
self.ui.rstBtn.clicked.connect(self.reset)
self.ui.rstBtn.clicked.connect(
lambda: self.ui.rstBtn.setDisabled(True))
self.ui.rstBtn.clicked.connect(
lambda: self.ui.quitBtn.setDisabled(True))
self.ui.rstBtn.clicked.connect(lambda: self.set_end(False))
self.ui.quitBtn.clicked.connect(lambda: QApplication.quit())
self.resetting.connect(self.worker.reset)
self.updated.connect(self.worker.start_crunching)
self.sceneRef = QObject()
self.show()
def graph(self, x, coeffs):
y = 0
for i, c in enumerate(coeffs):
y += c * x**i
return y
def update_graph(self, Polinomials, f_data, generation):
"""
Polinomials is a tuple of 5 p polinomials, and each p the coefficients of the polinomial
"""
lowest = 1000
highest = -1000
f_x = (*(x[0] for x in f_data), )
f_y = (*(x[1] for x in f_data), )
for x in f_x:
if x < lowest:
lowest = x
if x > highest:
highest = x
x_start = lowest # change this, use the smallest X in the data
x_end = highest # change this, use the biggest X in the data
slices = len(f_x) # smooth the curve
x = linspace(x_start, x_end, slices)
plt.plot(f_x, f_y, color='black')
colors = ('#0000CC', '#0088FF', '#00CC00', '#CC8800', '#FF0000')
for i, p in enumerate(Polinomials):
ys = (*(genal.polimerize(x, p) for x in f_x), )
plt.plot(f_x, ys, color=colors[i], label=f'rank {i+1}')
image_path = f'generation.png'
plt.title(f'function: {self.function} - generation: {generation}')
plt.legend()
plt.savefig(image_path)
plt.clf()
scene = QGraphicsScene(self)
pixmap = QPixmap(image_path)
item = QGraphicsPixmapItem(pixmap)
scene.addItem(item)
self.ui.graphicsView.setScene(scene)
self.sceneRef.deleteLater()
self.sceneRef = scene
if self.end:
return
QTimer.singleShot(0, lambda: self.updated.emit())