MainWindow.cpp

#include "MainWindow.h"

#include "DldReader.h"
#include "PlotWidget.h"

#include <QCoreApplication>
#include <QFileDialog>
#include <QHBoxLayout>
#include <QMessageBox>
#include <QProgressBar>
#include <QPushButton>
#include <QSplashScreen>
#include <QStatusBar>
#include <QVBoxLayout>
#include <QWidget>
#include <QTextStream>

MainWindow::MainWindow(QWidget *parent) : QMainWindow(parent) {
    setWindowTitle("DLD Viewer");

    QWidget *central = new QWidget(this);
    QVBoxLayout *vbox = new QVBoxLayout(central);
    QHBoxLayout *hbox = new QHBoxLayout();

    QPushButton *loadBtn = new QPushButton("Load DLD", central);
    QPushButton *homeBtn = new QPushButton("Reset view", central);
    QPushButton *quitBtn = new QPushButton("Quit", central);

    hbox->addWidget(loadBtn);
    hbox->addWidget(homeBtn);
    hbox->addWidget(quitBtn);
    hbox->addStretch();

    m_plot = new PlotWidget(central);

    vbox->addLayout(hbox);
    vbox->addWidget(m_plot, 1);

    setCentralWidget(central);
    m_progress = new QProgressBar(this);
    m_progress->setRange(0, 100);
    m_progress->setValue(0);
    m_progress->setTextVisible(true);
    m_progress->setVisible(false);
    statusBar()->addPermanentWidget(m_progress);
    statusBar()->showMessage("Ready");

    connect(loadBtn, &QPushButton::clicked, this, &MainWindow::loadFile);
    connect(homeBtn, &QPushButton::clicked, m_plot, &PlotWidget::resetView);
    connect(quitBtn, &QPushButton::clicked, this, &MainWindow::close);
    connect(m_plot, &PlotWidget::benchmarkDone, this, &MainWindow::benchmarkFinished);
}

void MainWindow::setSplashScreen(QSplashScreen *splash) {
    m_splash = splash;
}

void MainWindow::finishSplash() {
    if (m_splash) {
        m_splash->finish(this);
        m_splash = nullptr;
    }
}

bool MainWindow::loadFilePath(const QString &path) {
    QVector<float> data;
    DldLayout layout;
    QString error;
    m_progress->setValue(0);
    m_progress->setVisible(true);
    statusBar()->showMessage("Loading...");
    int lastPct = -1;
    auto progress = [&](int pct, const QString &stage) {
        if (pct == lastPct) {
            return;
        }
        lastPct = pct;
        m_progress->setValue(pct);
        QString label = stage.isEmpty() ? QString("Loading %1%").arg(pct)
                                        : QString("%1 (%2%)").arg(stage).arg(pct);
        statusBar()->showMessage(label);
        if (m_splash) {
            m_splash->showMessage(label, Qt::AlignHCenter | Qt::AlignBottom, Qt::white);
        }
        QCoreApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
    };

    if (!readDldFile(path, data, layout, &error, progress)) {
        m_progress->setVisible(false);
        finishSplash();
        if (QCoreApplication::arguments().contains("--bench")) {
            QTextStream(stderr) << "Failed to read DLD: " << error << "\n";
        } else {
            QMessageBox::critical(this, "Failed to read DLD", error);
        }
        return false;
    }
    m_progress->setVisible(false);
    finishSplash();

    m_plot->setData(data, layout.sampleRate);

    QString fmt = layout.bytesPerSample == 3 ? "24-bit" : (layout.bytesPerSample == 2 ? "16-bit" : "32-bit");
    QString endian = layout.littleEndian ? "LE" : "BE";
    statusBar()->showMessage(
        QString("Loaded %1 | blocks=%2 samples=%3 rate=%4 Hz | %5 %6")
            .arg(path)
            .arg(layout.blockCount)
            .arg(data.size())
            .arg(layout.sampleRate > 0.0 ? QString::number(layout.sampleRate, 'f', 2) : "unknown")
            .arg(fmt)
            .arg(endian)
    );
    return true;
}

void MainWindow::startBenchmark(int frames) {
    m_plot->startBenchmark(frames);
}

void MainWindow::loadFile() {
    QString path = QFileDialog::getOpenFileName(this, "Open DLD", QString(), "DLD files (*.DLD);;All files (*.*)");
    if (path.isEmpty()) {
        return;
    }
    loadFilePath(path);
}