X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=futatabi%2Fmainwindow.cpp;h=152358b3e1da023160d8ced72e7a75990a6dfcc2;hb=f1cddd2fcaf4d166b31c7d28e4ccd2201c5508f2;hp=4e53590426cc38d21c348251e7ca3d7507a67f7d;hpb=56255d64099f0fbaa5271bcb246bc9510fd0e5d8;p=nageru diff --git a/futatabi/mainwindow.cpp b/futatabi/mainwindow.cpp index 4e53590..152358b 100644 --- a/futatabi/mainwindow.cpp +++ b/futatabi/mainwindow.cpp @@ -15,6 +15,7 @@ #include #include #include +#include #include #include #include @@ -32,17 +33,6 @@ static PlayList *playlist_clips; extern int64_t current_pts; -template -void replace_model(QTableView *view, Model **model, Model *new_model, MainWindow *main_window) -{ - QItemSelectionModel *old_selection_model = view->selectionModel(); - view->setModel(new_model); - delete *model; - delete old_selection_model; - *model = new_model; - main_window->connect(new_model, &Model::any_content_changed, main_window, &MainWindow::content_changed); -} - MainWindow::MainWindow() : ui(new Ui::MainWindow), db(global_flags.working_directory + "/futatabi.db") @@ -50,6 +40,25 @@ MainWindow::MainWindow() global_mainwindow = this; ui->setupUi(this); + // Load settings from database. + SettingsProto settings = db.get_settings(); + if (!global_flags.interpolation_quality_set) { + 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. + flow_initialized_interpolation_quality = 1; + } else { + flow_initialized_interpolation_quality = global_flags.interpolation_quality; + } + save_settings(); + // The menus. connect(ui->exit_action, &QAction::triggered, this, &MainWindow::exit_triggered); connect(ui->export_cliplist_clip_multitrack_action, &QAction::triggered, this, &MainWindow::export_cliplist_clip_multitrack_triggered); @@ -61,6 +70,54 @@ MainWindow::MainWindow() ui->undo_action->setEnabled(false); ui->redo_action->setEnabled(false); + // The quality group. + QActionGroup *quality_group = new QActionGroup(ui->interpolation_menu); + quality_group->addAction(ui->quality_0_action); + quality_group->addAction(ui->quality_1_action); + quality_group->addAction(ui->quality_2_action); + quality_group->addAction(ui->quality_3_action); + quality_group->addAction(ui->quality_4_action); + if (global_flags.interpolation_quality == 0) { + ui->quality_0_action->setChecked(true); + } else if (global_flags.interpolation_quality == 1) { + ui->quality_1_action->setChecked(true); + } else if (global_flags.interpolation_quality == 2) { + ui->quality_2_action->setChecked(true); + } else if (global_flags.interpolation_quality == 3) { + ui->quality_3_action->setChecked(true); + } else if (global_flags.interpolation_quality == 4) { + ui->quality_4_action->setChecked(true); + } else { + assert(false); + } + connect(ui->quality_0_action, &QAction::toggled, bind(&MainWindow::quality_toggled, this, 0, _1)); + connect(ui->quality_1_action, &QAction::toggled, bind(&MainWindow::quality_toggled, this, 1, _1)); + connect(ui->quality_2_action, &QAction::toggled, bind(&MainWindow::quality_toggled, this, 2, _1)); + 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;}"); @@ -104,29 +161,8 @@ MainWindow::MainWindow() connect(play, &QShortcut::activated, ui->play_btn, &QPushButton::click); connect(ui->play_btn, &QPushButton::clicked, this, &MainWindow::play_clicked); - 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->stop_btn, &QPushButton::clicked, this, &MainWindow::stop_clicked); + ui->stop_btn->setEnabled(false); connect(ui->playlist_duplicate_btn, &QPushButton::clicked, this, &MainWindow::playlist_duplicate); @@ -168,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() @@ -182,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(); @@ -191,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? } } @@ -216,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(); @@ -259,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(); @@ -374,6 +467,14 @@ void MainWindow::state_changed(const StateProto &state) } } +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); +} + void MainWindow::play_clicked() { if (playlist_clips->empty()) @@ -392,6 +493,18 @@ void MainWindow::play_clicked() playlist_clips->set_progress({{ row, 0.0f }}); playlist_clips->set_currently_playing(row, 0.0f); playlist_selection_changed(); + + ui->stop_btn->setEnabled(true); +} + +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() @@ -405,6 +518,7 @@ void MainWindow::live_player_clip_done() playlist_clips->set_progress({{ row + 1, 0.0f }}); playlist_clips->set_currently_playing(row + 1, 0.0f); } + ui->stop_btn->setEnabled(false); } pair MainWindow::live_player_get_next_clip() @@ -416,7 +530,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)); @@ -674,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; @@ -733,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"); } } @@ -745,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); @@ -878,8 +991,8 @@ void MainWindow::undo_triggered() StateProto state = undo_stack.back(); ui->undo_action->setEnabled(undo_stack.size() > 1); - replace_model(ui->clip_list, &cliplist_clips, new ClipList(state.clip_list()), this); - replace_model(ui->playlist, &playlist_clips, new PlayList(state.play_list()), this); + replace_model(ui->clip_list, &cliplist_clips, new ClipList(state.clip_list())); + replace_model(ui->playlist, &playlist_clips, new PlayList(state.play_list())); db.store_state(state); } @@ -897,33 +1010,48 @@ void MainWindow::redo_triggered() ui->redo_action->setEnabled(!redo_stack.empty()); const StateProto &state = undo_stack.back(); - replace_model(ui->clip_list, &cliplist_clips, new ClipList(state.clip_list()), this); - replace_model(ui->playlist, &playlist_clips, new PlayList(state.play_list()), this); + replace_model(ui->clip_list, &cliplist_clips, new ClipList(state.clip_list())); + replace_model(ui->playlist, &playlist_clips, new PlayList(state.play_list())); db.store_state(state); } -void MainWindow::highlight_camera_input(int stream_idx) +void MainWindow::quality_toggled(int quality, bool checked) { - if (stream_idx == 0) { - ui->input1_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input1_frame->setStyleSheet(""); + if (!checked) { + return; } - if (stream_idx == 1) { - ui->input2_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input2_frame->setStyleSheet(""); + global_flags.interpolation_quality = quality; + if (quality != 0 && // Turning interpolation off is always possible. + quality != flow_initialized_interpolation_quality) { + QMessageBox msgbox; + 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) + ".")); + msgbox.exec(); } - if (stream_idx == 2) { - ui->input3_frame->setStyleSheet("background: rgb(0,255,0)"); - } else { - ui->input3_frame->setStyleSheet(""); + + save_settings(); +} + +void MainWindow::padding_toggled(double seconds, bool checked) +{ + 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(""); + } } } @@ -939,3 +1067,54 @@ pair MainWindow::get_queue_status() const { lock_guard lock(queue_status_mu); 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) +{ + QItemSelectionModel *old_selection_model = view->selectionModel(); + view->setModel(new_model); + delete *model; + delete old_selection_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); +}