]> git.sesse.net Git - nageru/blobdiff - mainwindow.cpp
Hook up white balance into the theme.
[nageru] / mainwindow.cpp
index 4804c088d98f7d3bc1f126d9a786ca181c18dea5..be46ded2bf82a96ea33a15f27f6cae24990c54f1 100644 (file)
 #include "mainwindow.h"
-#include "window.h"
-#include <thread>
 
-#include "context.h"
+#include <math.h>
+#include <stdio.h>
+#include <algorithm>
+#include <string>
+#include <vector>
+#include <QBoxLayout>
+#include <QKeySequence>
+#include <QLabel>
+#include <QMetaType>
+#include <QPushButton>
+#include <QResizeEvent>
+#include <QShortcut>
+#include <QSize>
+#include <QString>
+
+#include "glwidget.h"
+#include "lrameter.h"
 #include "mixer.h"
+#include "ui_display.h"
+#include "ui_mainwindow.h"
+#include "vumeter.h"
+
+class QResizeEvent;
+
+using namespace std;
+
+Q_DECLARE_METATYPE(std::vector<std::string>);
 
-using std::thread;
+MainWindow *global_mainwindow = nullptr;
 
 MainWindow::MainWindow()
+       : ui(new Ui::MainWindow)
+{
+       global_mainwindow = this;
+       ui->setupUi(this);
+
+       ui->me_live->set_output(Mixer::OUTPUT_LIVE);
+       ui->me_preview->set_output(Mixer::OUTPUT_PREVIEW);
+
+       // Hook up the transition buttons.
+       // TODO: Make them dynamic.
+       connect(ui->transition_btn1, &QPushButton::clicked, std::bind(&MainWindow::transition_clicked, this, 0));
+       connect(ui->transition_btn2, &QPushButton::clicked, std::bind(&MainWindow::transition_clicked, this, 1));
+       connect(ui->transition_btn3, &QPushButton::clicked, std::bind(&MainWindow::transition_clicked, this, 2));
+
+       // Aiee...
+       transition_btn1 = ui->transition_btn1;
+       transition_btn2 = ui->transition_btn2;
+       transition_btn3 = ui->transition_btn3;
+       qRegisterMetaType<std::vector<std::string>>("std::vector<std::string>");
+       connect(ui->me_preview, &GLWidget::transition_names_updated, this, &MainWindow::set_transition_names);
+}
+
+void MainWindow::resizeEvent(QResizeEvent* event)
+{
+       QMainWindow::resizeEvent(event);
+
+       // Ask for a relayout, but only after the event loop is done doing relayout
+       // on everything else.
+       QMetaObject::invokeMethod(this, "relayout", Qt::QueuedConnection);
+}
+
+void MainWindow::mixer_created(Mixer *mixer)
+{
+       // Make the previews.
+       unsigned num_previews = mixer->get_num_channels();
+
+       for (unsigned i = 0; i < num_previews; ++i) {
+               Mixer::Output output = Mixer::Output(Mixer::OUTPUT_INPUT0 + i);
+
+               QWidget *preview = new QWidget(this);
+               Ui::Display *ui_display = new Ui::Display;
+               ui_display->setupUi(preview);
+               ui_display->label->setText(mixer->get_channel_name(output).c_str());
+               ui_display->display->set_output(output);
+               ui->preview_displays->insertWidget(previews.size(), preview, 1);
+               previews.push_back(ui_display);
+
+               // Hook up the click.
+               connect(ui_display->display, &GLWidget::clicked, std::bind(&MainWindow::channel_clicked, this, i));
+
+               // Hook up the keyboard key.
+               QShortcut *shortcut = new QShortcut(QKeySequence(Qt::Key_1 + i), this);
+               connect(shortcut, &QShortcut::activated, std::bind(&MainWindow::channel_clicked, this, i));
+
+               // Hook up the white balance button (irrelevant if invisible).
+               ui_display->wb_button->setVisible(mixer->get_supports_set_wb(output));
+               connect(ui_display->wb_button, &QPushButton::clicked, std::bind(&MainWindow::wb_button_clicked, this, i));
+       }
+
+       mixer->set_audio_level_callback([this](float level_lufs, float peak_db, float global_level_lufs, float range_low_lufs, float range_high_lufs){
+               ui->vu_meter->set_level(level_lufs);
+               ui->lra_meter->set_levels(global_level_lufs, range_low_lufs, range_high_lufs);
+
+               char buf[256];
+               snprintf(buf, sizeof(buf), "%.1f", peak_db);
+               ui->peak_display->setText(buf);
+               if (peak_db > -0.1f) {  // -0.1 dBFS is EBU peak limit.
+                       ui->peak_display->setStyleSheet("QLabel { background-color: red; color: white; }");
+               } else {
+                       ui->peak_display->setStyleSheet("");
+               }
+       });
+}
+
+void MainWindow::relayout()
+{
+       int width = size().width();
+       int height = size().height();
+
+       // Allocate the height; the most important part is to keep the main displays
+       // at 16:9 if at all possible.
+       double me_width = ui->me_preview->width();
+       double me_height = me_width * 9.0 / 16.0 + ui->label_preview->height();
+
+       // TODO: Scale the widths when we need to do this.
+       if (me_height / double(height) > 0.8) {
+               me_height = height * 0.8;
+       }
+
+       // The previews will be constrained by the remaining height, and the width.
+       // FIXME: spacing?
+       double preview_label_height = previews[0]->label->height();
+       double preview_height = std::min(height - me_height - preview_label_height, (width / double(previews.size())) * 9.0 / 16.0);
+
+       ui->vertical_layout->setStretch(0, lrintf(me_height));
+       ui->vertical_layout->setStretch(1, std::max<int>(1, lrintf(height - me_height - preview_height)));
+       ui->vertical_layout->setStretch(2, lrintf(preview_height + preview_label_height));
+
+       // Set the widths for the previews.
+       double preview_width = preview_height * 16.0 / 9.0;  // FIXME: spacing?
+
+       for (unsigned i = 0; i < previews.size(); ++i) {
+               ui->preview_displays->setStretch(i, lrintf(preview_width));
+       }
+
+       // The spacer.
+       ui->preview_displays->setStretch(previews.size(), lrintf(width - previews.size() * preview_width));
+}
+
+void MainWindow::set_transition_names(vector<string> transition_names)
+{
+       if (transition_names.size() < 1) {
+               transition_btn1->setText(QString(""));
+       } else {
+               transition_btn1->setText(QString::fromStdString(transition_names[0]));
+       }
+       if (transition_names.size() < 2) {
+               transition_btn2->setText(QString(""));
+       } else {
+               transition_btn2->setText(QString::fromStdString(transition_names[1]));
+       }
+       if (transition_names.size() < 3) {
+               transition_btn3->setText(QString(""));
+       } else {
+               transition_btn3->setText(QString::fromStdString(transition_names[2]));
+       }
+}
+
+void MainWindow::transition_clicked(int transition_number)
+{
+       global_mixer->transition_clicked(transition_number);
+}
+
+void MainWindow::channel_clicked(int channel_number)
+{
+       if (current_wb_pick_display == channel_number) {
+               // The picking was already done from eventFilter(), since we don't get
+               // the mouse pointer here.
+       } else {
+               global_mixer->channel_clicked(channel_number);
+       }
+}
+
+void MainWindow::wb_button_clicked(int channel_number)
+{
+       current_wb_pick_display = channel_number;
+       QApplication::setOverrideCursor(Qt::CrossCursor);
+}
+
+bool MainWindow::eventFilter(QObject *watched, QEvent *event)
+{
+       if (current_wb_pick_display != -1 &&
+           event->type() == QEvent::MouseButtonRelease &&
+           watched->isWidgetType()) {
+               QApplication::restoreOverrideCursor();
+               if (watched == previews[current_wb_pick_display]->display) {
+                       const QMouseEvent *mouse_event = (QMouseEvent *)event;
+                       set_white_balance(current_wb_pick_display, mouse_event->x(), mouse_event->y());
+               } else {
+                       // The user clicked on something else, give up.
+                       // (The click goes through, which might not be ideal, but, yes.)
+                       current_wb_pick_display = -1;
+               }
+       }
+       return false;
+}
+
+namespace {
+
+double srgb_to_linear(double x)
+{
+       if (x < 0.04045) {
+               return x / 12.92;
+       } else {
+               return pow((x + 0.055) / 1.055, 2.4);
+       }
+}
+
+}  // namespace
+
+void MainWindow::set_white_balance(int channel_number, int x, int y)
 {
-        setCentralWidget(new Window(this));
+       // FIXME: This is wrong. We should really reset the white balance and then
+       // re-render, but renderToPixmap() crashes for me.
+       QRgb reference_color = previews[channel_number]->display->grabFrameBuffer().pixel(x, y);
+       double r = srgb_to_linear(qRed(reference_color) / 255.0);
+       double g = srgb_to_linear(qGreen(reference_color) / 255.0);
+       double b = srgb_to_linear(qBlue(reference_color) / 255.0);
+       global_mixer->set_wb(Mixer::OUTPUT_INPUT0 + channel_number, r, g, b);
 }