X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=mainwindow.cpp;h=8f63bee0fbd05108c96e2f573cd45c4f4a00342d;hb=c4508efc504090116e146a1f38659fe7de8398df;hp=3b8561fac7a32fa3a14217a85ff8ff890cac23cb;hpb=498d1082773a5f394cb4fbd15b1084a4751f1409;p=nageru diff --git a/mainwindow.cpp b/mainwindow.cpp index 3b8561f..8f63bee 100644 --- a/mainwindow.cpp +++ b/mainwindow.cpp @@ -1,38 +1,61 @@ #include "mainwindow.h" #include "clip_list.h" +#include "disk_space_estimator.h" #include "player.h" #include "post_to_main_thread.h" +#include "timebase.h" #include "ui_mainwindow.h" +#include #include #include #include #include #include +#include + +#include using namespace std; +using namespace std::placeholders; MainWindow *global_mainwindow = nullptr; -ClipList *cliplist_clips; -PlayList *playlist_clips; +static ClipList *cliplist_clips; +static PlayList *playlist_clips; extern int64_t current_pts; extern mutex frame_mu; extern vector frames[MAX_STREAMS]; MainWindow::MainWindow() - : ui(new Ui::MainWindow) + : ui(new Ui::MainWindow), + db("futatabi.db") { global_mainwindow = this; ui->setupUi(this); - cliplist_clips = new ClipList(); + // The menus. + connect(ui->exit_action, &QAction::triggered, this, &MainWindow::exit_triggered); + + 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;}"); + ui->menuBar->setCornerWidget(disk_free_label); + + StateProto state = db.get_state(); + + cliplist_clips = new ClipList(state.clip_list()); ui->clip_list->setModel(cliplist_clips); + connect(cliplist_clips, &ClipList::any_content_changed, this, &MainWindow::content_changed); - playlist_clips = new PlayList(); + playlist_clips = new PlayList(state.play_list()); ui->playlist->setModel(playlist_clips); + connect(playlist_clips, &PlayList::any_content_changed, this, &MainWindow::content_changed); + + // For un-highlighting when we lose focus. + ui->clip_list->installEventFilter(this); // For scrubbing in the pts columns. ui->clip_list->viewport()->installEventFilter(this); @@ -60,24 +83,37 @@ MainWindow::MainWindow() 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); - // TODO: support the delete key iff the widget has focus? connect(ui->playlist_remove_btn, &QPushButton::clicked, this, &MainWindow::playlist_remove); + QShortcut *delete_key = new QShortcut(QKeySequence(Qt::Key_Delete), ui->playlist); + connect(delete_key, &QShortcut::activated, [this] { + if (ui->playlist->hasFocus()) { + playlist_remove(); + } + }); // TODO: support drag-and-drop. connect(ui->playlist_move_up_btn, &QPushButton::clicked, [this]{ playlist_move(-1); }); @@ -87,19 +123,32 @@ MainWindow::MainWindow() this, &MainWindow::playlist_selection_changed); playlist_selection_changed(); // First time set-up. - preview_player = new Player(ui->preview_display); - live_player = new Player(ui->live_display); + preview_player = new Player(ui->preview_display, /*also_output_to_stream=*/false); + live_player = new Player(ui->live_display, /*also_output_to_stream=*/true); 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](double played_this_clip, double total_length) { + post_to_main_thread([this, played_this_clip, total_length] { + live_player_clip_progress(played_this_clip, total_length); + }); + }); + + defer_timeout = new QTimer(this); + defer_timeout->setSingleShot(true); + connect(defer_timeout, &QTimer::timeout, this, &MainWindow::defer_timer_expired); + + connect(ui->clip_list->selectionModel(), &QItemSelectionModel::currentChanged, + this, &MainWindow::clip_list_selection_changed); } void MainWindow::cue_in_clicked() { if (!cliplist_clips->empty() && cliplist_clips->back()->pts_out < 0) { - cliplist_clips->back()->pts_in = current_pts; + cliplist_clips->mutable_back()->pts_in = current_pts; return; } Clip clip; @@ -111,7 +160,7 @@ void MainWindow::cue_in_clicked() void MainWindow::cue_out_clicked() { if (!cliplist_clips->empty()) { - cliplist_clips->back()->pts_out = current_pts; + cliplist_clips->mutable_back()->pts_out = current_pts; // TODO: select the row in the clip list? } } @@ -126,15 +175,23 @@ void MainWindow::queue_clicked() if (!selected->hasSelection()) { Clip clip = *cliplist_clips->back(); clip.stream_idx = 0; - playlist_clips->add_clip(clip); + if (clip.pts_out != -1) { + playlist_clips->add_clip(clip); + playlist_selection_changed(); + } return; } 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)) { - Clip clip = *cliplist_clips->clip(index.row()); clip.stream_idx = index.column() - int(ClipList::Column::CAMERA_1); + } else { + clip.stream_idx = ui->preview_display->get_stream_idx(); + } + + if (clip.pts_out != -1) { playlist_clips->add_clip(clip); playlist_selection_changed(); } @@ -151,11 +208,14 @@ 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)) { - unsigned stream_idx = index.column() - int(ClipList::Column::CAMERA_1); - preview_player->play_clip(*cliplist_clips->clip(index.row()), stream_idx); + stream_idx = index.column() - int(ClipList::Column::CAMERA_1); + } else { + stream_idx = ui->preview_display->get_stream_idx(); } + preview_player->play_clip(*cliplist_clips->clip(index.row()), stream_idx); } void MainWindow::preview_angle_clicked(unsigned stream_idx) @@ -222,6 +282,37 @@ void MainWindow::playlist_move(int delta) playlist_selection_changed(); } +void MainWindow::defer_timer_expired() +{ + state_changed(deferred_state); +} + +void MainWindow::content_changed() +{ + 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. + // So it's time to short-circuit that timer and do the work it wanted to do. + defer_timeout->stop(); + state_changed(deferred_state); + } + StateProto state; + *state.mutable_clip_list() = cliplist_clips->serialize(); + *state.mutable_play_list() = playlist_clips->serialize(); + if (currently_deferring_model_changes) { + deferred_change_id = current_change_id; + deferred_state = std::move(state); + defer_timeout->start(200); + return; + } + state_changed(state); +} + +void MainWindow::state_changed(const StateProto &state) +{ + db.store_state(state); +} + void MainWindow::play_clicked() { if (playlist_clips->empty()) return; @@ -236,23 +327,60 @@ void MainWindow::play_clicked() const Clip &clip = *playlist_clips->clip(row); live_player->play_clip(clip, clip.stream_idx); - playlist_clips->set_currently_playing(row); + playlist_clips->set_currently_playing(row, 0.0f); playlist_selection_changed(); } void MainWindow::live_player_clip_done() { int row = playlist_clips->get_currently_playing(); - if (row != -1 && row < int(playlist_clips->size()) - 1) { - ++row; - const Clip &clip = *playlist_clips->clip(row); - live_player->play_clip(clip, clip.stream_idx); - playlist_clips->set_currently_playing(row); + if (row == -1 || row == int(playlist_clips->size()) - 1) { + ui->live_label->setText("Current output (paused)"); + playlist_clips->set_currently_playing(-1, 0.0f); } else { - playlist_clips->set_currently_playing(-1); + playlist_clips->set_currently_playing(row + 1, 0.0f); } } +Clip 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. + promise clip_promise; + future clip = clip_promise.get_future(); + post_to_main_thread([this, &clip_promise] { + int row = playlist_clips->get_currently_playing(); + if (row != -1 && row < int(playlist_clips->size()) - 1) { + clip_promise.set_value(*playlist_clips->clip(row + 1)); + } else { + clip_promise.set_value(Clip()); + } + }); + return clip.get(); +} + +void MainWindow::live_player_clip_progress(double played_this_clip, double total_length) +{ + playlist_clips->set_currently_playing(playlist_clips->get_currently_playing(), played_this_clip / total_length); + + double remaining = total_length - played_this_clip; + for (int row = playlist_clips->get_currently_playing() + 1; 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. + } + int remaining_ms = lrint(remaining * 1e3); + + int ms = remaining_ms % 1000; + remaining_ms /= 1000; + int s = remaining_ms % 60; + remaining_ms /= 60; + int m = remaining_ms; + + char buf[256]; + snprintf(buf, sizeof(buf), "Current output (%d:%02d.%03d left)", m, s, ms); + ui->live_label->setText(buf); +} + void MainWindow::resizeEvent(QResizeEvent *event) { QMainWindow::resizeEvent(event); @@ -264,7 +392,19 @@ void MainWindow::resizeEvent(QResizeEvent *event) void MainWindow::relayout() { - ui->live_display->setMinimumHeight(ui->live_display->width() * 9 / 16); + ui->live_display->setMinimumWidth(ui->live_display->height() * 16 / 9); + ui->preview_display->setMinimumWidth(ui->preview_display->height() * 16 / 9); +} + +void set_pts_in(int64_t pts, int64_t current_pts, ClipProxy &clip) +{ + pts = std::max(pts, 0); + if (clip->pts_out == -1) { + pts = std::min(pts, current_pts); + } else { + pts = std::min(pts, clip->pts_out); + } + clip->pts_in = pts; } bool MainWindow::eventFilter(QObject *watched, QEvent *event) @@ -276,6 +416,13 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event) unsigned stream_idx = ui->preview_display->get_stream_idx(); + if (watched == ui->clip_list) { + if (event->type() == QEvent::FocusOut) { + highlight_camera_input(-1); + } + return false; + } + if (event->type() != QEvent::Wheel) { last_mousewheel_camera_row = -1; } @@ -340,34 +487,35 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event) } int64_t pts = scrub_pts_origin + adjusted_offset * scrub_sensitivity; - + currently_deferring_model_changes = true; if (scrub_type == SCRUBBING_CLIP_LIST) { - ClipProxy clip = cliplist_clips->clip(scrub_row); + ClipProxy clip = cliplist_clips->mutable_clip(scrub_row); if (scrub_column == int(ClipList::Column::IN)) { - pts = std::max(pts, 0); - pts = std::min(pts, clip->pts_out); - clip->pts_in = pts; + current_change_id = "cliplist:in:" + to_string(scrub_row); + set_pts_in(pts, current_pts, clip); preview_single_frame(pts, stream_idx, FIRST_AT_OR_AFTER); } else { + current_change_id = "cliplist:out" + to_string(scrub_row); pts = std::max(pts, clip->pts_in); pts = std::min(pts, current_pts); clip->pts_out = pts; preview_single_frame(pts, stream_idx, LAST_BEFORE); } } else { - ClipProxy clip = playlist_clips->clip(scrub_row); + ClipProxy clip = playlist_clips->mutable_clip(scrub_row); if (scrub_column == int(PlayList::Column::IN)) { - pts = std::max(pts, 0); - pts = std::min(pts, clip->pts_out); - clip->pts_in = pts; + current_change_id = "playlist:in:" + to_string(scrub_row); + set_pts_in(pts, current_pts, clip); preview_single_frame(pts, clip->stream_idx, FIRST_AT_OR_AFTER); } else { + current_change_id = "playlist:out:" + to_string(scrub_row); pts = std::max(pts, clip->pts_in); pts = std::min(pts, current_pts); clip->pts_out = pts; preview_single_frame(pts, clip->stream_idx, LAST_BEFORE); } } + currently_deferring_model_changes = false; return true; // Don't use this mouse movement for selecting things. } @@ -395,43 +543,49 @@ bool MainWindow::eventFilter(QObject *watched, QEvent *event) int row = destination->rowAt(wheel->y()); if (column == -1 || row == -1) return false; - ClipProxy clip = (watched == ui->clip_list->viewport()) ? - cliplist_clips->clip(row) : playlist_clips->clip(row); - if (watched == ui->playlist->viewport()) { - stream_idx = clip->stream_idx; - } + 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); + if (watched == ui->playlist->viewport()) { + stream_idx = clip->stream_idx; + } - if (column != camera_column) { - last_mousewheel_camera_row = -1; - } - if (column == in_column) { - int64_t pts = clip->pts_in + wheel->angleDelta().y() * wheel_sensitivity; - pts = std::max(pts, 0); - pts = std::min(pts, clip->pts_out); - clip->pts_in = pts; - preview_single_frame(pts, stream_idx, FIRST_AT_OR_AFTER); - } else if (column == out_column) { - int64_t pts = clip->pts_out + wheel->angleDelta().y() * wheel_sensitivity; - pts = std::max(pts, clip->pts_in); - pts = std::min(pts, current_pts); - clip->pts_out = pts; - preview_single_frame(pts, stream_idx, LAST_BEFORE); - } else if (column == camera_column) { - int angle_degrees = wheel->angleDelta().y(); - if (last_mousewheel_camera_row == row) { - angle_degrees += leftover_angle_degrees; + if (column != camera_column) { + last_mousewheel_camera_row = -1; } + if (column == in_column) { + current_change_id += "in:" + to_string(row); + int64_t pts = clip->pts_in + wheel->angleDelta().y() * wheel_sensitivity; + set_pts_in(pts, current_pts, clip); + preview_single_frame(pts, stream_idx, FIRST_AT_OR_AFTER); + } else if (column == out_column) { + current_change_id += "out:" + to_string(row); + int64_t pts = clip->pts_out + wheel->angleDelta().y() * wheel_sensitivity; + pts = std::max(pts, clip->pts_in); + pts = std::min(pts, current_pts); + clip->pts_out = pts; + preview_single_frame(pts, stream_idx, LAST_BEFORE); + } else if (column == camera_column) { + current_change_id += "camera:" + to_string(row); + int angle_degrees = wheel->angleDelta().y(); + if (last_mousewheel_camera_row == row) { + angle_degrees += leftover_angle_degrees; + } - 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); - clip->stream_idx = stream_idx; + 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); + clip->stream_idx = stream_idx; - last_mousewheel_camera_row = row; - leftover_angle_degrees = angle_degrees % camera_degrees_per_pixel; + last_mousewheel_camera_row = row; + leftover_angle_degrees = angle_degrees % camera_degrees_per_pixel; - // Don't update the live view, that's rarely what the operator wants. + // Don't update the live view, that's rarely what the operator wants. + } } + currently_deferring_model_changes = false; } else if (event->type() == QEvent::MouseButtonRelease) { scrubbing = false; } @@ -475,3 +629,75 @@ void MainWindow::playlist_selection_changed() any_selected && selected->selectedRows().back().row() < int(playlist_clips->size()) - 1); ui->play_btn->setEnabled(!playlist_clips->empty()); } + +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)) { + camera_selected = current.column() - int(ClipList::Column::CAMERA_1); + } + highlight_camera_input(camera_selected); +} + +void MainWindow::report_disk_space(off_t free_bytes, double estimated_seconds_left) +{ + char time_str[256]; + if (estimated_seconds_left < 60.0) { + strcpy(time_str, "Less than a minute"); + } else if (estimated_seconds_left < 1800.0) { // Less than half an hour: Xm Ys (red). + int s = lrintf(estimated_seconds_left); + int m = s / 60; + s %= 60; + snprintf(time_str, sizeof(time_str), "%dm %ds", m, s); + } else if (estimated_seconds_left < 3600.0) { // Less than an hour: Xm. + int m = lrintf(estimated_seconds_left / 60.0); + snprintf(time_str, sizeof(time_str), "%dm", m); + } else if (estimated_seconds_left < 36000.0) { // Less than ten hours: Xh Ym. + int m = lrintf(estimated_seconds_left / 60.0); + int h = m / 60; + m %= 60; + snprintf(time_str, sizeof(time_str), "%dh %dm", h, m); + } else { // More than ten hours: Xh. + int h = lrintf(estimated_seconds_left / 3600.0); + snprintf(time_str, sizeof(time_str), "%dh", h); + } + char buf[256]; + snprintf(buf, sizeof(buf), "Disk free: %'.0f MB (approx. %s)", free_bytes / 1048576.0, time_str); + + std::string label = buf; + + post_to_main_thread([this, label]{ + disk_free_label->setText(QString::fromStdString(label)); + ui->menuBar->setCornerWidget(disk_free_label); // Need to set this again for the sizing to get right. + }); +} + +void MainWindow::exit_triggered() +{ + close(); +} + +void MainWindow::highlight_camera_input(int stream_idx) +{ + 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 (stream_idx == 3) { + ui->input4_frame->setStyleSheet("background: rgb(0,255,0)"); + } else { + ui->input4_frame->setStyleSheet(""); + } +}