def __init__(self):
QMainWindow.__init__(self)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.scene = NetworkViewer()
self.ui.view.setScene(self.scene)
self.ui.view.setRenderHint(QPainter.Antialiasing)
self.test_data = None
self.learning_data = None
self.creator = None
class MainWindow(QMainWindow):
def __init__(self):
QMainWindow.__init__(self)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.scene = NetworkViewer()
self.ui.view.setScene(self.scene)
self.ui.view.setRenderHint(QPainter.Antialiasing)
self.test_data = None
self.learning_data = None
self.creator = None
def _show_network(self):
self.scene.clear()
self.scene.show_network(self.network)
self.ui.view.fitInView(self.scene.sceneRect(), Qt.KeepAspectRatio)
def _activate_actions(self):
self.ui.launchAction.setEnabled(True)
self.ui.editNetworkAction.setEnabled(True)
self.ui.saveNetworkAction.setEnabled(True)
self.ui.menuLearning.setEnabled(True)
@pyqtSlot()
def on_newNetworkAction_triggered(self):
self.creator = NetworkCreatorDialog()
result = self.creator.exec_()
if result == QDialog.Accepted:
self.network = self.creator.create_network()
self._show_network()
self._activate_actions()
@pyqtSlot()
def on_editNetworkAction_triggered(self):
creator = NetworkCreatorDialog(NetworkModel.from_network(self.network))
result = creator.exec_()
if result == QDialog.Accepted:
self.network = creator.create_network()
self._show_network()
@pyqtSlot()
def on_launchAction_triggered(self):
creator = SimulationLauncherDialog(self.network, self.test_data)
creator.exec_()
self.test_data = creator.get_model()
@pyqtSlot()
def on_openNetworkAction_triggered(self):
filePath = QFileDialog().getOpenFileName(None, "Select input data file")
try:
with open(filePath, "r") as file:
self.network = yaml.load(file)
self._show_network()
self._activate_actions()
except Exception as e:
print e
@pyqtSlot()
def on_saveNetworkAction_triggered(self):
filePath = QFileDialog().getSaveFileName(None, "Select output file")
try:
with open(filePath, "w") as file:
yaml.dump(self.network, file)
except Exception as e:
print e
@pyqtSlot()
def on_learnAction_triggered(self):
launcher = LearningLauncher.dialog(self.network, self.learning_data)
result = launcher.exec_()
if result == QDialog.Accepted:
self._show_network()
self.learning_data = launcher.get_model()
def resizeEvent(self, QResizeEvent):
self.ui.view.fitInView(self.scene.sceneRect(), Qt.KeepAspectRatio)
def show(self):
QMainWindow.show(self)
self.resizeEvent(None)
