X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=futatabi%2Fmainwindow.cpp;h=0537f0099688bda5b31d39bbca94750d7b32ee24;hb=add8bb3a499cfeaad58bf5bdbda68391bab64059;hp=23840ff8441a0d4dfcadcdd434e3c5a3bf2e617f;hpb=44ee1966fb400dca1085e1e63d6305a5878414b7;p=nageru diff --git a/futatabi/mainwindow.cpp b/futatabi/mainwindow.cpp index 23840ff..0537f00 100644 --- a/futatabi/mainwindow.cpp +++ b/futatabi/mainwindow.cpp @@ -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); @@ -187,7 +189,6 @@ MainWindow::MainWindow() 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); @@ -202,6 +203,65 @@ MainWindow::MainWindow() connect(ui->clip_list->selectionModel(), &QItemSelectionModel::currentChanged, 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,23 @@ 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); + live_player_index_to_row.clear(); + + vector clips; + for (unsigned row = start_row; row < playlist_clips->size(); ++row) { + live_player_index_to_row.emplace(clips.size(), row); + clips.push_back(*playlist_clips->clip(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 +509,8 @@ 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_index_to_row.clear(); + live_player->play({fake_clip}); } void MainWindow::live_player_clip_done() @@ -470,7 +536,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)); @@ -498,7 +564,13 @@ static string format_duration(double t) void MainWindow::live_player_clip_progress(const map &progress) { - playlist_clips->set_progress(progress); + map converted_progress; + for (const auto &it : progress) { + if (live_player_index_to_row.count(it.first)) { + converted_progress.emplace(live_player_index_to_row[it.first], it.second); + } + } + playlist_clips->set_progress(converted_progress); vector clips; for (size_t row = 0; row < playlist_clips->size(); ++row) { @@ -728,7 +800,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 +841,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 +859,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 +871,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); @@ -976,27 +1047,23 @@ void MainWindow::quality_toggled(int quality, bool checked) 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(""); + } } } @@ -1013,6 +1080,21 @@ pair MainWindow::get_queue_status() const { return {queue_status, "text/plain"}; } +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) { @@ -1023,3 +1105,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); +}