X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=futatabi%2Fmainwindow.cpp;h=5cbfb133927f18852b8c4367cd485e2e9a055c96;hb=2b37c51ea17b77b88e2f7fda443109acc260aa8f;hp=23840ff8441a0d4dfcadcdd434e3c5a3bf2e617f;hpb=44ee1966fb400dca1085e1e63d6305a5878414b7;p=nageru diff --git a/futatabi/mainwindow.cpp b/futatabi/mainwindow.cpp index 23840ff..5cbfb13 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;}"); @@ -138,30 +164,6 @@ MainWindow::MainWindow() connect(ui->stop_btn, &QPushButton::clicked, this, &MainWindow::stop_clicked); ui->stop_btn->setEnabled(false); - QShortcut *preview_1 = new QShortcut(QKeySequence(Qt::Key_1), this); - connect(preview_1, &QShortcut::activated, ui->preview_1_btn, &QPushButton::click); - connect(ui->input1_display, &JPEGFrameView::clicked, ui->preview_1_btn, &QPushButton::click); - connect(ui->preview_1_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(0); }); - ui->input1_display->set_overlay("1"); - - QShortcut *preview_2 = new QShortcut(QKeySequence(Qt::Key_2), this); - connect(preview_2, &QShortcut::activated, ui->preview_2_btn, &QPushButton::click); - connect(ui->input2_display, &JPEGFrameView::clicked, ui->preview_2_btn, &QPushButton::click); - connect(ui->preview_2_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(1); }); - ui->input2_display->set_overlay("2"); - - QShortcut *preview_3 = new QShortcut(QKeySequence(Qt::Key_3), this); - connect(preview_3, &QShortcut::activated, ui->preview_3_btn, &QPushButton::click); - connect(ui->input3_display, &JPEGFrameView::clicked, ui->preview_3_btn, &QPushButton::click); - connect(ui->preview_3_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(2); }); - ui->input3_display->set_overlay("3"); - - QShortcut *preview_4 = new QShortcut(QKeySequence(Qt::Key_4), this); - connect(preview_4, &QShortcut::activated, ui->preview_4_btn, &QPushButton::click); - connect(ui->input4_display, &JPEGFrameView::clicked, ui->preview_4_btn, &QPushButton::click); - connect(ui->preview_4_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(3); }); - ui->input4_display->set_overlay("4"); - connect(ui->playlist_duplicate_btn, &QPushButton::clicked, this, &MainWindow::playlist_duplicate); connect(ui->playlist_remove_btn, &QPushButton::clicked, this, &MainWindow::playlist_remove); @@ -173,21 +175,20 @@ 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->set_done_callback([this] { + post_to_main_thread([this] { live_player_clip_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); @@ -201,7 +202,66 @@ 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 new display rows. + unsigned display_rows = (num_cameras + 1) / 2; + ui->video_displays->setStretch(1, display_rows); + for (unsigned i = displays.size(); i < num_cameras; ++i) { + QFrame *frame = new QFrame(this); + frame->setAutoFillBackground(true); + + QLayout *layout = new QGridLayout(frame); + frame->setLayout(layout); + layout->setContentsMargins(3, 3, 3, 3); + + JPEGFrameView *display = new JPEGFrameView(frame); + display->setAutoFillBackground(true); + layout->addWidget(display); + + ui->input_displays->addWidget(frame, i / 2, i % 2); + display->set_overlay(to_string(i + 1)); + + QPushButton *preview_btn = new QPushButton(this); + preview_btn->setMaximumSize(20, 17); + preview_btn->setText(QString::fromStdString(to_string(i + 1))); + ui->preview_layout->addWidget(preview_btn); + + displays.emplace_back(FrameAndDisplay{ frame, display, preview_btn }); + + connect(display, &JPEGFrameView::clicked, preview_btn, &QPushButton::click); + 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); }); + } + + cliplist_clips->change_num_cameras(num_cameras); + playlist_clips->change_num_cameras(num_cameras); + + QMetaObject::invokeMethod(this, "relayout", Qt::QueuedConnection); } MainWindow::~MainWindow() @@ -216,7 +276,7 @@ void MainWindow::cue_in_clicked() return; } Clip clip; - clip.pts_in = current_pts; + clip.pts_in = max(current_pts - lrint(global_flags.cue_point_padding_seconds * TIMEBASE), 0); cliplist_clips->add_clip(clip); playlist_selection_changed(); ui->clip_list->scrollToBottom(); @@ -225,7 +285,7 @@ void MainWindow::cue_in_clicked() void MainWindow::cue_out_clicked() { if (!cliplist_clips->empty()) { - cliplist_clips->mutable_back()->pts_out = current_pts; + cliplist_clips->mutable_back()->pts_out = current_pts + lrint(global_flags.cue_point_padding_seconds * TIMEBASE); // TODO: select the row in the clip list? } } @@ -250,8 +310,7 @@ void MainWindow::queue_clicked() QModelIndex index = selected->currentIndex(); Clip clip = *cliplist_clips->clip(index.row()); - if (index.column() >= int(ClipList::Column::CAMERA_1) && - index.column() <= int(ClipList::Column::CAMERA_4)) { + 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(); @@ -277,7 +336,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; } } @@ -287,19 +346,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 (index.column() >= int(ClipList::Column::CAMERA_1) && - index.column() <= int(ClipList::Column::CAMERA_4)) { - 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) @@ -412,6 +470,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); } @@ -421,17 +480,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(Player::ClipWithRow{ *playlist_clips->clip(row), row }); + } + live_player->play(clips); + playlist_clips->set_progress({ { start_row, 0.0f } }); + playlist_clips->set_currently_playing(start_row, 0.0f); playlist_selection_changed(); ui->stop_btn->setEnabled(true); @@ -444,7 +506,7 @@ void MainWindow::stop_clicked() 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); + live_player->play(fake_clip); } void MainWindow::live_player_clip_done() @@ -455,7 +517,7 @@ void MainWindow::live_player_clip_done() playlist_clips->set_progress({}); playlist_clips->set_currently_playing(-1, 0.0f); } else { - playlist_clips->set_progress({{ row + 1, 0.0f }}); + playlist_clips->set_progress({ { row + 1, 0.0f } }); playlist_clips->set_currently_playing(row + 1, 0.0f); } ui->stop_btn->setEnabled(false); @@ -470,7 +532,7 @@ pair MainWindow::live_player_get_next_clip() // for Player to shut down, we could have a deadlock here. promise> clip_promise; future> clip = clip_promise.get_future(); - post_to_main_thread([this, &clip_promise] { + 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)); @@ -687,7 +749,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() || @@ -698,8 +761,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; } @@ -728,7 +790,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; @@ -769,7 +831,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() @@ -787,11 +849,11 @@ void MainWindow::playlist_selection_changed() if (!any_selected) { set_output_status("paused"); } else { - double remaining = 0.0; - for (int row = selected->selectedRows().front().row(); row < int(playlist_clips->size()); ++row) { - const Clip clip = *playlist_clips->clip(row); - remaining += double(clip.pts_out - clip.pts_in) / TIMEBASE / 0.5; // FIXME: stop hardcoding speed. + 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, { { selected->selectedRows().front().row(), 0.0 } }); set_output_status(format_duration(remaining) + " ready"); } } @@ -799,8 +861,7 @@ void MainWindow::playlist_selection_changed() void MainWindow::clip_list_selection_changed(const QModelIndex ¤t, const QModelIndex &) { int camera_selected = -1; - if (current.column() >= int(ClipList::Column::CAMERA_1) && - current.column() <= int(ClipList::Column::CAMERA_4)) { + if (cliplist_clips->is_camera_column(current.column())) { camera_selected = current.column() - int(ClipList::Column::CAMERA_1); } highlight_camera_input(camera_selected); @@ -969,34 +1030,31 @@ 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::highlight_camera_input(int stream_idx) +void MainWindow::padding_toggled(double seconds, bool checked) { - if (stream_idx == 0) { - ui->input1_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input1_frame->setStyleSheet(""); - } - if (stream_idx == 1) { - ui->input2_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input2_frame->setStyleSheet(""); - } - if (stream_idx == 2) { - ui->input3_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input3_frame->setStyleSheet(""); + if (!checked) { + return; } - if (stream_idx == 3) { - ui->input4_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input4_frame->setStyleSheet(""); + 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 (unsigned(stream_idx) == i) { + displays[i].frame->setStyleSheet("background: rgb(0,255,0)"); + } else { + displays[i].frame->setStyleSheet(""); + } } } @@ -1008,12 +1066,28 @@ void MainWindow::set_output_status(const string &status) 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" }; } -template +void MainWindow::display_frame(unsigned stream_idx, const FrameOnDisk &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 void MainWindow::replace_model(QTableView *view, Model **model, Model *new_model) { QItemSelectionModel *old_selection_model = view->selectionModel(); @@ -1023,3 +1097,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); +}