]> git.sesse.net Git - nageru/blobdiff - futatabi/mainwindow.cpp
Make cue in/out highlight the right cell. More logical, and also helps with jogging.
[nageru] / futatabi / mainwindow.cpp
index 68b29e4be57127b97b3bd058f335ebbfecdf9d11..77a3ba35e35bebfdf31c5df258052c75dcffb1e7 100644 (file)
@@ -1,12 +1,12 @@
 #include "mainwindow.h"
 
-#include "shared/aboutdialog.h"
 #include "clip_list.h"
 #include "export.h"
-#include "shared/disk_space_estimator.h"
 #include "flags.h"
 #include "frame_on_disk.h"
 #include "player.h"
+#include "shared/aboutdialog.h"
+#include "shared/disk_space_estimator.h"
 #include "shared/post_to_main_thread.h"
 #include "shared/timebase.h"
 #include "ui_mainwindow.h"
@@ -15,6 +15,7 @@
 #include <QFileDialog>
 #include <QMessageBox>
 #include <QMouseEvent>
+#include <QNetworkReply>
 #include <QShortcut>
 #include <QTimer>
 #include <QWheelEvent>
@@ -39,13 +40,16 @@ MainWindow::MainWindow()
        global_mainwindow = this;
        ui->setupUi(this);
 
-       // Load settings from database if needed.
+       // Load settings from database.
+       SettingsProto settings = db.get_settings();
        if (!global_flags.interpolation_quality_set) {
-               SettingsProto settings = db.get_settings();
                if (settings.interpolation_quality() != 0) {
                        global_flags.interpolation_quality = settings.interpolation_quality() - 1;
                }
        }
+       if (!global_flags.cue_point_padding_set) {
+               global_flags.cue_point_padding_seconds = settings.cue_point_padding_seconds();  // Default 0 is fine.
+       }
        if (global_flags.interpolation_quality == 0) {
                // Allocate something just for simplicity; we won't be using it
                // unless the user changes runtime, in which case 1 is fine.
@@ -92,6 +96,28 @@ MainWindow::MainWindow()
        connect(ui->quality_3_action, &QAction::toggled, bind(&MainWindow::quality_toggled, this, 3, _1));
        connect(ui->quality_4_action, &QAction::toggled, bind(&MainWindow::quality_toggled, this, 4, _1));
 
+       // The cue point padding group.
+       QActionGroup *padding_group = new QActionGroup(ui->interpolation_menu);
+       padding_group->addAction(ui->padding_0_action);
+       padding_group->addAction(ui->padding_1_action);
+       padding_group->addAction(ui->padding_2_action);
+       padding_group->addAction(ui->padding_5_action);
+       if (global_flags.cue_point_padding_seconds <= 1e-3) {
+               ui->padding_0_action->setChecked(true);
+       } else if (fabs(global_flags.cue_point_padding_seconds - 1.0) < 1e-3) {
+               ui->padding_1_action->setChecked(true);
+       } else if (fabs(global_flags.cue_point_padding_seconds - 2.0) < 1e-3) {
+               ui->padding_2_action->setChecked(true);
+       } else if (fabs(global_flags.cue_point_padding_seconds - 5.0) < 1e-3) {
+               ui->padding_5_action->setChecked(true);
+       } else {
+               // Nothing to check, which is fine.
+       }
+       connect(ui->padding_0_action, &QAction::toggled, bind(&MainWindow::padding_toggled, this, 0.0, _1));
+       connect(ui->padding_1_action, &QAction::toggled, bind(&MainWindow::padding_toggled, this, 1.0, _1));
+       connect(ui->padding_2_action, &QAction::toggled, bind(&MainWindow::padding_toggled, this, 2.0, _1));
+       connect(ui->padding_5_action, &QAction::toggled, bind(&MainWindow::padding_toggled, this, 5.0, _1));
+
        global_disk_space_estimator = new DiskSpaceEstimator(bind(&MainWindow::report_disk_space, this, _1, _2));
        disk_free_label = new QLabel(this);
        disk_free_label->setStyleSheet("QLabel {padding-right: 5px;}");
@@ -149,24 +175,23 @@ MainWindow::MainWindow()
        });
 
        // TODO: support drag-and-drop.
-       connect(ui->playlist_move_up_btn, &QPushButton::clicked, [this]{ playlist_move(-1); });
-       connect(ui->playlist_move_down_btn, &QPushButton::clicked, [this]{ playlist_move(1); });
+       connect(ui->playlist_move_up_btn, &QPushButton::clicked, [this] { playlist_move(-1); });
+       connect(ui->playlist_move_down_btn, &QPushButton::clicked, [this] { playlist_move(1); });
 
        connect(ui->playlist->selectionModel(), &QItemSelectionModel::selectionChanged,
-               this, &MainWindow::playlist_selection_changed);
+               this, &MainWindow::playlist_selection_changed);
        playlist_selection_changed();  // First time set-up.
 
        preview_player.reset(new Player(ui->preview_display, Player::NO_STREAM_OUTPUT));
        live_player.reset(new Player(ui->live_display, Player::HTTPD_STREAM_OUTPUT));
-       live_player->set_done_callback([this]{
-               post_to_main_thread([this]{
-                       live_player_clip_done();
+       live_player->set_done_callback([this] {
+               post_to_main_thread([this] {
+                       live_player_done();
                });
        });
-       live_player->set_next_clip_callback(bind(&MainWindow::live_player_get_next_clip, this));
-       live_player->set_progress_callback([this](const map<size_t, double> &progress) {
-               post_to_main_thread([this, progress] {
-                       live_player_clip_progress(progress);
+       live_player->set_progress_callback([this](const map<uint64_t, double> &progress, double time_remaining) {
+               post_to_main_thread([this, progress, time_remaining] {
+                       live_player_clip_progress(progress, time_remaining);
                });
        });
        set_output_status("paused");
@@ -177,12 +202,34 @@ MainWindow::MainWindow()
        ui->undo_action->setEnabled(true);
 
        connect(ui->clip_list->selectionModel(), &QItemSelectionModel::currentChanged,
-               this, &MainWindow::clip_list_selection_changed);
+               this, &MainWindow::clip_list_selection_changed);
+
+       // Find out how many cameras we have in the existing frames;
+       // if none, we start with two cameras.
+       num_cameras = 2;
+       {
+               lock_guard<mutex> lock(frame_mu);
+               for (size_t stream_idx = 2; stream_idx < MAX_STREAMS; ++stream_idx) {
+                       if (!frames[stream_idx].empty()) {
+                               num_cameras = stream_idx + 1;
+                       }
+               }
+       }
+       change_num_cameras();
+
+       if (!global_flags.tally_url.empty()) {
+               start_tally();
+       }
+}
+
+void MainWindow::change_num_cameras()
+{
+       assert(num_cameras >= displays.size());  // We only add, never remove.
 
-       // Make the display rows.
-       unsigned display_rows = (NUM_CAMERAS + 1) / 2;
+       // Make new display rows.
+       unsigned display_rows = (num_cameras + 1) / 2;
        ui->video_displays->setStretch(1, display_rows);
-       for (unsigned i = 0; i < NUM_CAMERAS; ++i) {
+       for (unsigned i = displays.size(); i < num_cameras; ++i) {
                QFrame *frame = new QFrame(this);
                frame->setAutoFillBackground(true);
 
@@ -208,8 +255,13 @@ MainWindow::MainWindow()
                QShortcut *shortcut = new QShortcut(QKeySequence(Qt::Key_1 + i), this);
                connect(shortcut, &QShortcut::activated, preview_btn, &QPushButton::click);
 
-               connect(preview_btn, &QPushButton::clicked, [this, i]{ preview_angle_clicked(i); });
+               connect(preview_btn, &QPushButton::clicked, [this, i] { preview_angle_clicked(i); });
        }
+
+       cliplist_clips->change_num_cameras(num_cameras);
+       playlist_clips->change_num_cameras(num_cameras);
+
+       QMetaObject::invokeMethod(this, "relayout", Qt::QueuedConnection);
 }
 
 MainWindow::~MainWindow()
@@ -221,21 +273,33 @@ void MainWindow::cue_in_clicked()
 {
        if (!cliplist_clips->empty() && cliplist_clips->back()->pts_out < 0) {
                cliplist_clips->mutable_back()->pts_in = current_pts;
-               return;
+       } else {
+               Clip clip;
+               clip.pts_in = max<int64_t>(current_pts - lrint(global_flags.cue_point_padding_seconds * TIMEBASE), 0);
+               cliplist_clips->add_clip(clip);
+               playlist_selection_changed();
        }
-       Clip clip;
-       clip.pts_in = current_pts;
-       cliplist_clips->add_clip(clip);
-       playlist_selection_changed();
+
+       // Select the item so that we can jog it.
+       ui->clip_list->setFocus();
+       QModelIndex index = cliplist_clips->index(cliplist_clips->size() - 1, int(ClipList::Column::IN));
+       ui->clip_list->selectionModel()->setCurrentIndex(index, QItemSelectionModel::ClearAndSelect);
        ui->clip_list->scrollToBottom();
 }
 
 void MainWindow::cue_out_clicked()
 {
-       if (!cliplist_clips->empty()) {
-               cliplist_clips->mutable_back()->pts_out = current_pts;
-               // TODO: select the row in the clip list?
+       if (cliplist_clips->empty()) {
+               return;
        }
+
+       cliplist_clips->mutable_back()->pts_out = current_pts + lrint(global_flags.cue_point_padding_seconds * TIMEBASE);
+
+       // Select the item so that we can jog it.
+       ui->clip_list->setFocus();
+       QModelIndex index = cliplist_clips->index(cliplist_clips->size() - 1, int(ClipList::Column::OUT));
+       ui->clip_list->selectionModel()->setCurrentIndex(index, QItemSelectionModel::ClearAndSelect);
+       ui->clip_list->scrollToBottom();
 }
 
 void MainWindow::queue_clicked()
@@ -258,7 +322,7 @@ void MainWindow::queue_clicked()
 
        QModelIndex index = selected->currentIndex();
        Clip clip = *cliplist_clips->clip(index.row());
-       if (ClipList::is_camera_column(index.column())) {
+       if (cliplist_clips->is_camera_column(index.column())) {
                clip.stream_idx = index.column() - int(ClipList::Column::CAMERA_1);
        } else {
                clip.stream_idx = ui->preview_display->get_stream_idx();
@@ -284,7 +348,7 @@ void MainWindow::preview_clicked()
                if (selected->hasSelection()) {
                        QModelIndex index = selected->currentIndex();
                        const Clip &clip = *playlist_clips->clip(index.row());
-                       preview_player->play_clip(clip, index.row(), clip.stream_idx);
+                       preview_player->play(clip);
                        return;
                }
        }
@@ -294,18 +358,18 @@ void MainWindow::preview_clicked()
 
        QItemSelectionModel *selected = ui->clip_list->selectionModel();
        if (!selected->hasSelection()) {
-               preview_player->play_clip(*cliplist_clips->back(), cliplist_clips->size() - 1, 0);
+               preview_player->play(*cliplist_clips->back());
                return;
        }
 
        QModelIndex index = selected->currentIndex();
-       unsigned stream_idx;
-       if (ClipList::is_camera_column(index.column())) {
-               stream_idx = index.column() - int(ClipList::Column::CAMERA_1);
+       Clip clip = *cliplist_clips->clip(index.row());
+       if (cliplist_clips->is_camera_column(index.column())) {
+               clip.stream_idx = index.column() - int(ClipList::Column::CAMERA_1);
        } else {
-               stream_idx = ui->preview_display->get_stream_idx();
+               clip.stream_idx = ui->preview_display->get_stream_idx();
        }
-       preview_player->play_clip(*cliplist_clips->clip(index.row()), index.row(), stream_idx);
+       preview_player->play(clip);
 }
 
 void MainWindow::preview_angle_clicked(unsigned stream_idx)
@@ -379,6 +443,14 @@ void MainWindow::defer_timer_expired()
 
 void MainWindow::content_changed()
 {
+       // If we are playing, update the part of the playlist that's not playing yet.
+       vector<ClipWithID> clips;
+       for (unsigned row = 0; row < playlist_clips->size(); ++row) {
+               clips.emplace_back(*playlist_clips->clip_with_id(row));
+       }
+       live_player->splice_play(clips);
+
+       // Serialize the state.
        if (defer_timeout->isActive() &&
            (!currently_deferring_model_changes || deferred_change_id != current_change_id)) {
                // There's some deferred event waiting, but this event is unrelated.
@@ -418,6 +490,7 @@ void MainWindow::save_settings()
 {
        SettingsProto settings;
        settings.set_interpolation_quality(global_flags.interpolation_quality + 1);
+       settings.set_cue_point_padding_seconds(global_flags.cue_point_padding_seconds);
        db.store_settings(settings);
 }
 
@@ -427,17 +500,20 @@ void MainWindow::play_clicked()
                return;
 
        QItemSelectionModel *selected = ui->playlist->selectionModel();
-       int row;
+       unsigned start_row;
        if (!selected->hasSelection()) {
-               row = 0;
+               start_row = 0;
        } else {
-               row = selected->selectedRows(0)[0].row();
+               start_row = selected->selectedRows(0)[0].row();
        }
 
-       const Clip &clip = *playlist_clips->clip(row);
-       live_player->play_clip(clip, row, clip.stream_idx);
-       playlist_clips->set_progress({{ row, 0.0f }});
-       playlist_clips->set_currently_playing(row, 0.0f);
+       vector<ClipWithID> clips;
+       for (unsigned row = start_row; row < playlist_clips->size(); ++row) {
+               clips.emplace_back(*playlist_clips->clip_with_id(row));
+       }
+       live_player->play(clips);
+       playlist_clips->set_progress({ { start_row, 0.0f } });
+       ui->playlist->selectionModel()->clear();
        playlist_selection_changed();
 
        ui->stop_btn->setEnabled(true);
@@ -448,70 +524,22 @@ void MainWindow::stop_clicked()
        Clip fake_clip;
        fake_clip.pts_in = 0;
        fake_clip.pts_out = 0;
-       size_t last_row = playlist_clips->size() - 1;
-       playlist_clips->set_currently_playing(last_row, 0.0f);
-       live_player->play_clip(fake_clip, last_row, 0);
-}
-
-void MainWindow::live_player_clip_done()
-{
-       int row = playlist_clips->get_currently_playing();
-       if (row == -1 || row == int(playlist_clips->size()) - 1) {
-               set_output_status("paused");
-               playlist_clips->set_progress({});
-               playlist_clips->set_currently_playing(-1, 0.0f);
-       } else {
-               playlist_clips->set_progress({{ row + 1, 0.0f }});
-               playlist_clips->set_currently_playing(row + 1, 0.0f);
-       }
+       playlist_clips->set_progress({});
+       live_player->play(fake_clip);
        ui->stop_btn->setEnabled(false);
 }
 
-pair<Clip, size_t> MainWindow::live_player_get_next_clip()
-{
-       // playlist_clips can only be accessed on the main thread.
-       // Hopefully, we won't have to wait too long for this to come back.
-       //
-       // TODO: If MainWindow is in the process of being destroyed and waiting
-       // for Player to shut down, we could have a deadlock here.
-       promise<pair<Clip, size_t>> clip_promise;
-       future<pair<Clip, size_t>> clip = clip_promise.get_future();
-       post_to_main_thread([&clip_promise] {
-               int row = playlist_clips->get_currently_playing();
-               if (row != -1 && row < int(playlist_clips->size()) - 1) {
-                       clip_promise.set_value(make_pair(*playlist_clips->clip(row + 1), row + 1));
-               } else {
-                       clip_promise.set_value(make_pair(Clip(), 0));
-               }
-       });
-       return clip.get();
-}
-
-static string format_duration(double t)
+void MainWindow::live_player_done()
 {
-       int t_ms = lrint(t * 1e3);
-
-       int ms = t_ms % 1000;
-       t_ms /= 1000;
-       int s = t_ms % 60;
-       t_ms /= 60;
-       int m = t_ms;
-
-       char buf[256];
-       snprintf(buf, sizeof(buf), "%d:%02d.%03d", m, s, ms);
-       return buf;
+       playlist_selection_changed();
+       playlist_clips->set_progress({});
+       ui->stop_btn->setEnabled(false);
 }
 
-void MainWindow::live_player_clip_progress(const map<size_t, double> &progress)
+void MainWindow::live_player_clip_progress(const map<uint64_t, double> &progress, double time_remaining)
 {
        playlist_clips->set_progress(progress);
-
-       vector<Clip> clips;
-       for (size_t row = 0; row < playlist_clips->size(); ++row) {
-               clips.push_back(*playlist_clips->clip(row));
-       }
-       double remaining = compute_time_left(clips, progress);
-       set_output_status(format_duration(remaining) + " left");
+       set_output_status(format_duration(time_remaining) + " left");
 }
 
 void MainWindow::resizeEvent(QResizeEvent *event)
@@ -693,7 +721,8 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event)
                }
                int column = destination->columnAt(wheel->x());
                int row = destination->rowAt(wheel->y());
-               if (column == -1 || row == -1) return false;
+               if (column == -1 || row == -1)
+                       return false;
 
                // Only adjust pts with the wheel if the given row is selected.
                if (!destination->hasFocus() ||
@@ -704,8 +733,7 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event)
                currently_deferring_model_changes = true;
                {
                        current_change_id = (watched == ui->clip_list->viewport()) ? "cliplist:" : "playlist:";
-                       ClipProxy clip = (watched == ui->clip_list->viewport()) ?
-                               cliplist_clips->mutable_clip(row) : playlist_clips->mutable_clip(row);
+                       ClipProxy clip = (watched == ui->clip_list->viewport()) ? cliplist_clips->mutable_clip(row) : playlist_clips->mutable_clip(row);
                        if (watched == ui->playlist->viewport()) {
                                stream_idx = clip->stream_idx;
                        }
@@ -734,7 +762,7 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event)
 
                                int stream_idx = clip->stream_idx + angle_degrees / camera_degrees_per_pixel;
                                stream_idx = std::max(stream_idx, 0);
-                               stream_idx = std::min(stream_idx, NUM_CAMERAS - 1);
+                               stream_idx = std::min<int>(stream_idx, num_cameras - 1);
                                clip->stream_idx = stream_idx;
 
                                last_mousewheel_camera_row = row;
@@ -775,7 +803,7 @@ void MainWindow::preview_single_frame(int64_t pts, unsigned stream_idx, MainWind
        Clip fake_clip;
        fake_clip.pts_in = pts;
        fake_clip.pts_out = pts + 1;
-       preview_player->play_clip(fake_clip, 0, stream_idx);
+       preview_player->play(fake_clip);
 }
 
 void MainWindow::playlist_selection_changed()
@@ -793,11 +821,11 @@ void MainWindow::playlist_selection_changed()
        if (!any_selected) {
                set_output_status("paused");
        } else {
-               vector<Clip> clips;
-               for (size_t row = 0; row < playlist_clips->size(); ++row) {
-                       clips.push_back(*playlist_clips->clip(row));
+               vector<ClipWithID> clips;
+               for (size_t row = selected->selectedRows().front().row(); row < playlist_clips->size(); ++row) {
+                       clips.emplace_back(*playlist_clips->clip_with_id(row));
                }
-               double remaining = compute_time_left(clips, {{selected->selectedRows().front().row(), 0.0}});
+               double remaining = compute_total_time(clips);
                set_output_status(format_duration(remaining) + " ready");
        }
 }
@@ -805,7 +833,7 @@ void MainWindow::playlist_selection_changed()
 void MainWindow::clip_list_selection_changed(const QModelIndex &current, const QModelIndex &)
 {
        int camera_selected = -1;
-       if (ClipList::is_camera_column(current.column())) {
+       if (cliplist_clips->is_camera_column(current.column())) {
                camera_selected = current.column() - int(ClipList::Column::CAMERA_1);
        }
        highlight_camera_input(camera_selected);
@@ -974,17 +1002,27 @@ void MainWindow::quality_toggled(int quality, bool checked)
                msgbox.setText(QString::fromStdString(
                        "The interpolation quality for the main output cannot be changed at runtime, "
                        "except being turned completely off; it will take effect for exported files "
-                       "only until next restart. The live output quality thus remains at " + to_string(flow_initialized_interpolation_quality) + "."));
+                       "only until next restart. The live output quality thus remains at " +
+                       to_string(flow_initialized_interpolation_quality) + "."));
                msgbox.exec();
        }
 
        save_settings();
 }
 
+void MainWindow::padding_toggled(double seconds, bool checked)
+{
+       if (!checked) {
+               return;
+       }
+       global_flags.cue_point_padding_seconds = seconds;
+       save_settings();
+}
+
 void MainWindow::highlight_camera_input(int stream_idx)
 {
-       for (unsigned i = 0; i < NUM_CAMERAS; ++i) {
-               if (stream_idx == i) {
+       for (unsigned i = 0; i < num_cameras; ++i) {
+               if (unsigned(stream_idx) == i) {
                        displays[i].frame->setStyleSheet("background: rgb(0,255,0)");
                } else {
                        displays[i].frame->setStyleSheet("");
@@ -995,24 +1033,36 @@ void MainWindow::highlight_camera_input(int stream_idx)
 void MainWindow::set_output_status(const string &status)
 {
        ui->live_label->setText(QString::fromStdString("Current output (" + status + ")"));
+       if (live_player != nullptr) {
+               live_player->set_pause_status(status);
+       }
 
        lock_guard<mutex> lock(queue_status_mu);
        queue_status = status;
 }
 
-pair<string, string> MainWindow::get_queue_status() const {
+pair<string, string> MainWindow::get_queue_status() const
+{
        lock_guard<mutex> lock(queue_status_mu);
-       return {queue_status, "text/plain"};
+       return { queue_status, "text/plain" };
 }
 
 void MainWindow::display_frame(unsigned stream_idx, const FrameOnDisk &frame)
 {
-       if (stream_idx < NUM_CAMERAS) {
-               displays[stream_idx].display->setFrame(stream_idx, frame);
+       if (stream_idx >= MAX_STREAMS) {
+               fprintf(stderr, "WARNING: Ignoring too-high stream index %u.\n", stream_idx);
+               return;
+       }
+       if (stream_idx >= num_cameras) {
+               post_to_main_thread_and_wait([this, stream_idx] {
+                       num_cameras = stream_idx + 1;
+                       change_num_cameras();
+               });
        }
+       displays[stream_idx].display->setFrame(stream_idx, frame);
 }
 
-template <class Model>
+template<class Model>
 void MainWindow::replace_model(QTableView *view, Model **model, Model *new_model)
 {
        QItemSelectionModel *old_selection_model = view->selectionModel();
@@ -1022,3 +1072,28 @@ void MainWindow::replace_model(QTableView *view, Model **model, Model *new_model
        *model = new_model;
        connect(new_model, &Model::any_content_changed, this, &MainWindow::content_changed);
 }
+
+void MainWindow::start_tally()
+{
+       http_reply = http.get(QNetworkRequest(QString::fromStdString(global_flags.tally_url)));
+       connect(http_reply, &QNetworkReply::finished, this, &MainWindow::tally_received);
+}
+
+void MainWindow::tally_received()
+{
+       unsigned time_to_next_tally_ms;
+       if (http_reply->error()) {
+               fprintf(stderr, "HTTP get of '%s' failed: %s\n", global_flags.tally_url.c_str(),
+                       http_reply->errorString().toStdString().c_str());
+               ui->live_frame->setStyleSheet("");
+               time_to_next_tally_ms = 1000;
+       } else {
+               string contents = http_reply->readAll().toStdString();
+               ui->live_frame->setStyleSheet(QString::fromStdString("background: " + contents));
+               time_to_next_tally_ms = 100;
+       }
+       http_reply->deleteLater();
+       http_reply = nullptr;
+
+       QTimer::singleShot(time_to_next_tally_ms, this, &MainWindow::start_tally);
+}