X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=futatabi%2Fmainwindow.cpp;h=152358b3e1da023160d8ced72e7a75990a6dfcc2;hb=931727fc4cccafc84023a053a6eff174b5ea8190;hp=0c12b5c17637c6ce648b7a24841cd09f92e29f9a;hpb=db71f5501fe92501c09a32c778fbacccadd26206;p=nageru diff --git a/futatabi/mainwindow.cpp b/futatabi/mainwindow.cpp index 0c12b5c..152358b 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); @@ -202,6 +204,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 +277,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 +286,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 +311,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(); @@ -293,8 +353,7 @@ void MainWindow::preview_clicked() QModelIndex index = selected->currentIndex(); unsigned stream_idx; - if (index.column() >= int(ClipList::Column::CAMERA_1) && - index.column() <= int(ClipList::Column::CAMERA_4)) { + if (cliplist_clips->is_camera_column(index.column())) { stream_idx = index.column() - int(ClipList::Column::CAMERA_1); } else { stream_idx = ui->preview_display->get_stream_idx(); @@ -412,6 +471,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); } @@ -728,7 +788,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; @@ -787,11 +847,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 +859,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 +1035,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 +1068,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 +1093,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); +}