X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;ds=sidebyside;f=futatabi%2Fmainwindow.cpp;h=77a3ba35e35bebfdf31c5df258052c75dcffb1e7;hb=7ae1d56b56be5cc80333b2f7124ecfb6a86f16e9;hp=68b29e4be57127b97b3bd058f335ebbfecdf9d11;hpb=3295f7e178ffb254f7223e5c13d21c4f0689f2be;p=nageru diff --git a/futatabi/mainwindow.cpp b/futatabi/mainwindow.cpp index 68b29e4..77a3ba3 100644 --- a/futatabi/mainwindow.cpp +++ b/futatabi/mainwindow.cpp @@ -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 #include #include +#include #include #include #include @@ -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 &progress) { - post_to_main_thread([this, progress] { - live_player_clip_progress(progress); + live_player->set_progress_callback([this](const map &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 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(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 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 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 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> clip_promise; - future> 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 &progress) +void MainWindow::live_player_clip_progress(const map &progress, double time_remaining) { playlist_clips->set_progress(progress); - - vector 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(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 clips; - for (size_t row = 0; row < playlist_clips->size(); ++row) { - clips.push_back(*playlist_clips->clip(row)); + vector 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 ¤t, 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 lock(queue_status_mu); queue_status = status; } -pair MainWindow::get_queue_status() const { +pair MainWindow::get_queue_status() const +{ lock_guard 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 +template 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); +}