def init(self):
self.pause()
da = float(self.window.ui.daField.text())
db = float(self.window.ui.dbField.text())
ds = float(self.window.ui.dsField.text())
beta = float(self.window.ui.betaField.text())
width = self.window.ui.widthSlider.value()
height = self.window.ui.heightSlider.value()
self.texture = MorphogenesisImageData(width, height, ds, da, db, beta)
if self.window.ui.initComboBox.currentIndex() == 0:
self.texture.generate('random')
else:
self.texture.generate('stripe')
self.window.ui.widget.setTexture(self.texture)
self.window.ui.widget.makeCurrent()
class Controller:
def __init__(self, window):
self.window = window
self.worker = None
self.texture = None
self.running = False
self.dumpAtEndPath = None
self.timer = QtCore.QTimer()
def setThreadRunning(self, flag):
self.running = flag
self.window.ui.pauseButton.setEnabled(flag)
self.window.ui.runButton.setEnabled(not flag)
def setThreadFinished(self):
self.setThreadRunning(False)
if self.dumpAtEndPath is not None:
self.window.ui.widget.grabFrameBuffer().save(os.path.join(self.dumpAtEndPath, self.texture.imageName()) + '.png')
QtCore.QCoreApplication.instance().quit()
def saveScreenshot(self):
dialog = QtGui.QFileDialog()
dialog.setAcceptMode(QtGui.QFileDialog.AcceptSave)
dialog.selectFile(self.texture.imageName())
if dialog.exec_() == QtGui.QDialog.Accepted:
f = dialog.selectedFiles()[0]
self.window.ui.widget.grabFrameBuffer().save(f)
def updateUI(self):
self.window.ui.widget.updateGL()
if self.texture is not None:
self.window.ui.ipsLabel.setText("%1.f IPS" % (float(self.texture.iteration - self.lastIteration)/(time.time() - self.lastIterationTime)))
self.window.ui.iterationLabel.setText("Iteration %d" % self.texture.iteration)
self.lastIteration = self.texture.iteration
self.lastIterationTime = time.time()
def awake(self):
QtCore.QObject.connect(self.window.ui.runButton, QtCore.SIGNAL("clicked()"), self.run)
QtCore.QObject.connect(self.timer, QtCore.SIGNAL("timeout()"), self.updateUI )
QtCore.QObject.connect(self.window.ui.initButton, QtCore.SIGNAL("clicked()"), self.init)
QtCore.QObject.connect(self.window.ui.stepButton, QtCore.SIGNAL("clicked()"), self.step)
QtCore.QObject.connect(self.window.ui.pauseButton, QtCore.SIGNAL("clicked()"), self.pause)
QtCore.QObject.connect(self.window.ui.debugInfoMenuItem, QtCore.SIGNAL("triggered()"), self.logDebugInfo)
QtCore.QObject.connect(self.window.ui.saveScreenshotMenuItem, QtCore.SIGNAL("triggered()"), self.saveScreenshot)
QtCore.QObject.connect(QtCore.QCoreApplication.instance(), QtCore.SIGNAL("aboutToQuit()"), self.cleanup)
self.setThreadRunning(False)
def pause(self):
self.timer.stop()
if self.worker is not None:
self.worker.halt = True
def cleanup(self):
if self.worker is not None:
self.worker.halt = True
def init(self):
self.pause()
da = float(self.window.ui.daField.text())
db = float(self.window.ui.dbField.text())
ds = float(self.window.ui.dsField.text())
beta = float(self.window.ui.betaField.text())
width = self.window.ui.widthSlider.value()
height = self.window.ui.heightSlider.value()
self.texture = MorphogenesisImageData(width, height, ds, da, db, beta)
if self.window.ui.initComboBox.currentIndex() == 0:
self.texture.generate('random')
else:
self.texture.generate('stripe')
self.window.ui.widget.setTexture(self.texture)
self.window.ui.widget.makeCurrent()
def step(self):
if self.texture == None:
self.init()
self.texture.step()
self.window.ui.widget.updateGL()
def run(self, maxIterations = 0, dumpAtEndPath = None):
if self.texture == None:
self.init()
if not self.running:
self.dumpAtEndPath = dumpAtEndPath
self.lastIteration = self.texture.iteration
self.lastIterationTime = time.time()
self.worker = WorkerThread(self, self.texture, maxIterations)
QtCore.QObject.connect(self.worker, QtCore.SIGNAL("finished()"), self.setThreadFinished)
self.worker.start()
self.timer.start(40)
self.setThreadRunning(True)
def logDebugInfo(self):
if self.texture is not None:
self.texture.logDebugInfo()
def setOptions(self, options):
self.window.ui.daField.setText(str(options.D_a))
self.window.ui.dbField.setText(str(options.D_b))
self.window.ui.dsField.setText(str(options.D_s))
self.window.ui.betaField.setText(str(options.beta_i))
self.window.ui.widthSlider.setValue(options.width)
self.window.ui.heightSlider.setValue(options.height)