#include "mainwindow.h"
#include "clip_list.h"
+#include "disk_space_estimator.h"
+#include "flags.h"
+#include "frame_on_disk.h"
#include "player.h"
#include "post_to_main_thread.h"
+#include "timebase.h"
#include "ui_mainwindow.h"
-#include <string>
-#include <vector>
-
#include <QMouseEvent>
-#include <QWheelEvent>
#include <QShortcut>
+#include <QTimer>
+#include <QWheelEvent>
+#include <future>
+#include <sqlite3.h>
+#include <string>
+#include <vector>
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<int64_t> frames[MAX_STREAMS];
MainWindow::MainWindow()
- : ui(new Ui::MainWindow)
+ : ui(new Ui::MainWindow),
+ db(global_flags.working_directory + "/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);
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); });
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<size_t, double> &progress) {
+ post_to_main_thread([this, progress] {
+ live_player_clip_progress(progress);
+ });
+ });
+ set_output_status("paused");
+
+ 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;
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?
}
}
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();
}
void MainWindow::preview_clicked()
{
- if (cliplist_clips->empty()) return;
+ if (ui->playlist->hasFocus()) {
+ // Allow the playlist as preview iff it has focus and something is selected.
+ QItemSelectionModel *selected = ui->playlist->selectionModel();
+ 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);
+ return;
+ }
+ }
+
+ if (cliplist_clips->empty())
+ return;
QItemSelectionModel *selected = ui->clip_list->selectionModel();
if (!selected->hasSelection()) {
- preview_player->play_clip(*cliplist_clips->back(), 0);
+ preview_player->play_clip(*cliplist_clips->back(), cliplist_clips->size() - 1, 0);
return;
}
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()), index.row(), stream_idx);
}
void MainWindow::preview_angle_clicked(unsigned stream_idx)
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;
+ if (playlist_clips->empty())
+ return;
QItemSelectionModel *selected = ui->playlist->selectionModel();
int row;
}
const Clip &clip = *playlist_clips->clip(row);
- live_player->play_clip(clip, clip.stream_idx);
- playlist_clips->set_currently_playing(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);
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) {
+ set_output_status("paused");
+ playlist_clips->set_progress({});
+ playlist_clips->set_currently_playing(-1, 0.0f);
} else {
- playlist_clips->set_currently_playing(-1);
+ playlist_clips->set_progress({{ row + 1, 0.0f }});
+ playlist_clips->set_currently_playing(row + 1, 0.0f);
+ }
+}
+
+pair<Clip, size_t> 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<pair<Clip, size_t>> clip_promise;
+ future<pair<Clip, size_t>> 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(make_pair(*playlist_clips->clip(row + 1), row + 1));
+ } else {
+ clip_promise.set_value(make_pair(Clip(), 0));
+ }
+ });
+ return clip.get();
+}
+
+static string format_duration(double t)
+{
+ int t_ms = lrint(t * 1e3);
+
+ int ms = t_ms % 1000;
+ t_ms /= 1000;
+ int s = t_ms % 60;
+ t_ms /= 60;
+ int m = t_ms;
+
+ char buf[256];
+ snprintf(buf, sizeof(buf), "%d:%02d.%03d", m, s, ms);
+ return buf;
+}
+
+void MainWindow::live_player_clip_progress(const map<size_t, double> &progress)
+{
+ playlist_clips->set_progress(progress);
+
+ // Look at the last clip and then start counting from there.
+ assert(!progress.empty());
+ auto last_it = progress.end();
+ --last_it;
+ double remaining = 0.0;
+ double last_fade_time_seconds = 0.0;
+ for (size_t row = last_it->first; row < playlist_clips->size(); ++row) {
+ const Clip clip = *playlist_clips->clip(row);
+ double clip_length = double(clip.pts_out - clip.pts_in) / TIMEBASE / 0.5; // FIXME: stop hardcoding speed.
+ if (row == last_it->first) {
+ // A clip we're playing: Subtract the part we've already played.
+ remaining = clip_length * (1.0 - last_it->second);
+ } else {
+ // A clip we haven't played yet: Subtract the part that's overlapping
+ // with a previous clip (due to fade).
+ remaining += max(clip_length - last_fade_time_seconds, 0.0);
+ }
+ last_fade_time_seconds = min(clip_length, clip.fade_time_seconds);
}
+ set_output_status(format_duration(remaining) + " left");
}
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<int64_t>(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)
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;
}
}
int column = destination->columnAt(mouse->x());
int row = destination->rowAt(mouse->y());
- if (column == -1 || row == -1) return false;
+ if (column == -1 || row == -1)
+ return false;
if (type == SCRUBBING_CLIP_LIST) {
if (ClipList::Column(column) == ClipList::Column::IN) {
}
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<int64_t>(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<int64_t>(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.
}
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;
+ // Only adjust pts with the wheel if the given row is selected.
+ if (!destination->hasFocus() ||
+ row != destination->selectionModel()->currentIndex().row()) {
+ return false;
}
- 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<int64_t>(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;
+ 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;
}
- 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;
+ 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;
+ }
- last_mousewheel_camera_row = row;
- leftover_angle_degrees = angle_degrees % camera_degrees_per_pixel;
+ 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;
- // Don't update the live view, that's rarely what the operator wants.
+ 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.
+ }
}
+ currently_deferring_model_changes = false;
+ return true; // Don't scroll.
} else if (event->type() == QEvent::MouseButtonRelease) {
scrubbing = false;
}
{
if (rounding == LAST_BEFORE) {
lock_guard<mutex> lock(frame_mu);
- if (frames[stream_idx].empty()) return;
- auto it = lower_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts);
+ if (frames[stream_idx].empty())
+ return;
+ auto it = lower_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts,
+ [](const FrameOnDisk &frame, int64_t pts) { return frame.pts < pts; });
if (it != frames[stream_idx].end()) {
- pts = *it;
+ pts = it->pts;
}
} else {
assert(rounding == FIRST_AT_OR_AFTER);
lock_guard<mutex> lock(frame_mu);
- if (frames[stream_idx].empty()) return;
- auto it = upper_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts - 1);
+ if (frames[stream_idx].empty())
+ return;
+ auto it = upper_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts - 1,
+ [](int64_t pts, const FrameOnDisk &frame) { return pts < frame.pts; });
if (it != frames[stream_idx].end()) {
- pts = *it;
+ pts = it->pts;
}
}
Clip fake_clip;
fake_clip.pts_in = pts;
fake_clip.pts_out = pts + 1;
- preview_player->play_clip(fake_clip, stream_idx);
+ preview_player->play_clip(fake_clip, 0, stream_idx);
}
void MainWindow::playlist_selection_changed()
ui->playlist_move_down_btn->setEnabled(
any_selected && selected->selectedRows().back().row() < int(playlist_clips->size()) - 1);
ui->play_btn->setEnabled(!playlist_clips->empty());
+
+ 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.
+ }
+ set_output_status(format_duration(remaining) + " ready");
+ }
+}
+
+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, "<font color=\"red\">Less than a minute</font>");
+ } 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), "<font color=\"red\">%dm %ds</font>", 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("");
+ }
+}
+
+void MainWindow::set_output_status(const string &status)
+{
+ ui->live_label->setText(QString::fromStdString("Current output (" + status + ")"));
+
+ lock_guard<mutex> lock(queue_status_mu);
+ queue_status = status;
+}
+
+pair<string, string> MainWindow::get_queue_status() const {
+ lock_guard<mutex> lock(queue_status_mu);
+ return {queue_status, "text/plain"};
}
projects / nageru / blobdiff