diff --git a/mainwindow.cpp b/mainwindow.cpp index 379cd2b..89f86a3 100644 --- a/mainwindow.cpp +++ b/mainwindow.cpp @@ -1,6 +1,10 @@ #include #include #include +#include +#include +#include +#include #include "mainwindow.h" #include "ui_mainwindow.h" #include "triggerwidget.h" @@ -8,7 +12,9 @@ MainWindow::MainWindow(QWidget *parent): QMainWindow(parent), ui(new Ui::MainWindow), - codeEditor(this) + codeEditor(this), + currentFilePath(""), + isFileModified(false) { ui->setupUi(this); enumerateDevices(); @@ -23,7 +29,22 @@ MainWindow::MainWindow(QWidget *parent): ui->codeLayout->addWidget(&codeEditor); + // Set up keyboard shortcuts + ui->actionOpen->setShortcut(QKeySequence::Open); + ui->actionSave->setShortcut(QKeySequence::Save); + ui->actionSave_As->setShortcut(QKeySequence::SaveAs); + ui->actionQuit->setShortcut(QKeySequence::Quit); + connect(ui->actionQuit, &QAction::triggered, this, [this]() {close();}); + connect(ui->actionOpen, &QAction::triggered, this, &MainWindow::onActionOpenTriggered); + connect(ui->actionSave, &QAction::triggered, this, &MainWindow::onActionSaveTriggered); + connect(ui->actionSave_As, &QAction::triggered, this, &MainWindow::onActionSaveAsTriggered); + + // Connect text changed signal to track modifications + connect(&codeEditor, &QTextEdit::textChanged, this, [this]() { + isFileModified = true; + updateStatus(); + }); } MainWindow::~MainWindow() @@ -31,6 +52,90 @@ MainWindow::~MainWindow() delete ui; } +void MainWindow::updateStatus() +{ + if (!currentFilePath.isEmpty()) { + QString status = "EisMultiplexer-Qt"; + status = QString("%1").arg(currentFilePath); + if (isFileModified) { + status += " - [Unsaved Changes]"; + } + ui->statusbar->showMessage(status); + } +} + +void MainWindow::onActionOpenTriggered() +{ + QString filePath = QFileDialog::getOpenFileName(this, tr("Open Python Script"), + "", + tr("Python Files (*.py);;All Files (*)")); + + if (filePath.isEmpty()) { + return; + } + + QFile file(filePath); + if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) { + QMessageBox::warning(this, tr("Error"), tr("Could not open file: %1").arg(filePath)); + return; + } + + QTextStream in(&file); + QString content = in.readAll(); + file.close(); + + codeEditor.setPlainText(content); + currentFilePath = filePath; + isFileModified = false; + updateStatus(); +} + +void MainWindow::onActionSaveTriggered() +{ + if (currentFilePath.isEmpty()) { + onActionSaveAsTriggered(); + return; + } + + QFile file(currentFilePath); + if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) { + QMessageBox::warning(this, tr("Error"), tr("Could not save file: %1").arg(currentFilePath)); + return; + } + + QTextStream out(&file); + out << codeEditor.toPlainText(); + file.close(); + + isFileModified = false; + updateStatus(); +} + +void MainWindow::onActionSaveAsTriggered() +{ + QString filePath = QFileDialog::getSaveFileName(this, tr("Save Python Script"), + "", + tr("Python Files (*.py);;All Files (*)")); + + if (filePath.isEmpty()) { + return; + } + + QFile file(filePath); + if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) { + QMessageBox::warning(this, tr("Error"), tr("Could not save file: %1").arg(filePath)); + return; + } + + QTextStream out(&file); + out << codeEditor.toPlainText(); + file.close(); + + currentFilePath = filePath; + isFileModified = false; + updateStatus(); +} + void MainWindow::enumerateDevices() { size_t count = 0; @@ -114,17 +219,47 @@ void MainWindow::enumerateDevices() ui->statusbar->showMessage("Ready"); free(serials); + generateExample(); } void MainWindow::generateExample() { QString example = - "import eismultiplexer\n\n" - "from time import sleep"; + "# This is an example script to show you how\n# to drive eismultiplexer using the python api\n" + "import eismultiplexer as multi\n" + "from time import sleep\n\n" + "# First initalize the device(s)\n"; - for (const auto& channel : channels) + std::set serials; + for (size_t i = 0; i < channels.size(); ++i) + serials.insert(channels[i]->getDeviceSerial()); + + size_t i = 0; + for (uint16_t serial : serials) { - example.append(QString("eismultiplexer")); + example.append(QString("multiplexer_") + QString::number(i) + " multi.Multiplexer(serial=" + QString::number(serial) + ")\n"); + ++i; } + + example.append("\nprint('\\nListing the nummber of channels per unit')\n"); + for (size_t i = 0; i < serials.size(); ++i) + { + QString printLine = "print(f'Found unit with serial number {" + QString::number(channels[i]->getDeviceSerial()) + "} and {multiplexer_" + QString::number(i) + ".getChannelCount()} channels')\n"; + example.append(printLine); + } + + example.append("\nprint('Connecting the first and second channel on the first unit')\n"); + example.append("multiplexer_0.connectChannel(multi.Channel.A)\n"); + example.append("multiplexer_0.connectChannel(multi.Channel.B)\n\n"); + example.append("print('Waiting for half a second for something to happen')\n"); + example.append("sleep(0.5)\n\n"); + example.append("print('Disconnect first channel')\n"); + example.append("multiplexer_0.disconnectChannel(multi.Channel.A)\n\n"); + example.append("print('Waiting up to 5000 milliseconds for a trigger')\n"); + example.append("multiplexer_0.setTriggerState(0, multi.TriggerState.INPUT)\n"); + example.append("multiplexer_0.waitTrigger(0, multi.TriggerState.HIGHLEVEL, 5000)\n\n"); + example.append("print('Disconnecting all channels')\n"); + example.append("multiplexer_0.clear()\n"); + codeEditor.setText(example); } diff --git a/mainwindow.h b/mainwindow.h index 7373d2b..70bbc2f 100644 --- a/mainwindow.h +++ b/mainwindow.h @@ -34,9 +34,17 @@ public: explicit MainWindow(QWidget *parent = nullptr); ~MainWindow(); +private slots: + void onActionOpenTriggered(); + void onActionSaveTriggered(); + void onActionSaveAsTriggered(); + private: void enumerateDevices(); - void generateExample(); + void generateExample(); + void updateStatus(); + QString currentFilePath; + bool isFileModified; }; #endif // MAINWINDOW_H