git.sesse.net Git - nageru/blob - mainwindow.cpp

   1 #include "mainwindow.h"
   2
   3 #include "clip_list.h"
   4 #include "disk_space_estimator.h"
   5 #include "player.h"
   6 #include "post_to_main_thread.h"
   7 #include "timebase.h"
   8 #include "ui_mainwindow.h"
   9
  10 #include <QMouseEvent>
  11 #include <QShortcut>
  12 #include <QTimer>
  13 #include <QWheelEvent>
  14 #include <future>
  15 #include <sqlite3.h>
  16 #include <string>
  17 #include <vector>
  18
  19 using namespace std;
  20 using namespace std::placeholders;
  21
  22 MainWindow *global_mainwindow = nullptr;
  23 static ClipList *cliplist_clips;
  24 static PlayList *playlist_clips;
  25
  26 extern int64_t current_pts;
  27 extern mutex frame_mu;
  28 extern vector<int64_t> frames[MAX_STREAMS];
  29
  30 MainWindow::MainWindow()
  31         : ui(new Ui::MainWindow),
  32           db("futatabi.db")
  33 {
  34         global_mainwindow = this;
  35         ui->setupUi(this);
  36
  37         // The menus.
  38         connect(ui->exit_action, &QAction::triggered, this, &MainWindow::exit_triggered);
  39
  40         global_disk_space_estimator = new DiskSpaceEstimator(bind(&MainWindow::report_disk_space, this, _1, _2));
  41         disk_free_label = new QLabel(this);
  42         disk_free_label->setStyleSheet("QLabel {padding-right: 5px;}");
  43         ui->menuBar->setCornerWidget(disk_free_label);
  44
  45         StateProto state = db.get_state();
  46
  47         cliplist_clips = new ClipList(state.clip_list());
  48         ui->clip_list->setModel(cliplist_clips);
  49         connect(cliplist_clips, &ClipList::any_content_changed, this, &MainWindow::content_changed);
  50
  51         playlist_clips = new PlayList(state.play_list());
  52         ui->playlist->setModel(playlist_clips);
  53         connect(playlist_clips, &PlayList::any_content_changed, this, &MainWindow::content_changed);
  54
  55         // For un-highlighting when we lose focus.
  56         ui->clip_list->installEventFilter(this);
  57
  58         // For scrubbing in the pts columns.
  59         ui->clip_list->viewport()->installEventFilter(this);
  60         ui->playlist->viewport()->installEventFilter(this);
  61
  62         QShortcut *cue_in = new QShortcut(QKeySequence(Qt::Key_A), this);
  63         connect(cue_in, &QShortcut::activated, ui->cue_in_btn, &QPushButton::click);
  64         connect(ui->cue_in_btn, &QPushButton::clicked, this, &MainWindow::cue_in_clicked);
  65
  66         QShortcut *cue_out = new QShortcut(QKeySequence(Qt::Key_S), this);
  67         connect(cue_out, &QShortcut::activated, ui->cue_out_btn, &QPushButton::click);
  68         connect(ui->cue_out_btn, &QPushButton::clicked, this, &MainWindow::cue_out_clicked);
  69
  70         QShortcut *queue = new QShortcut(QKeySequence(Qt::Key_Q), this);
  71         connect(queue, &QShortcut::activated, ui->queue_btn, &QPushButton::click);
  72         connect(ui->queue_btn, &QPushButton::clicked, this, &MainWindow::queue_clicked);
  73
  74         QShortcut *preview = new QShortcut(QKeySequence(Qt::Key_W), this);
  75         connect(preview, &QShortcut::activated, ui->preview_btn, &QPushButton::click);
  76         connect(ui->preview_btn, &QPushButton::clicked, this, &MainWindow::preview_clicked);
  77
  78         QShortcut *play = new QShortcut(QKeySequence(Qt::Key_Space), this);
  79         connect(play, &QShortcut::activated, ui->play_btn, &QPushButton::click);
  80         connect(ui->play_btn, &QPushButton::clicked, this, &MainWindow::play_clicked);
  81
  82         QShortcut *preview_1 = new QShortcut(QKeySequence(Qt::Key_1), this);
  83         connect(preview_1, &QShortcut::activated, ui->preview_1_btn, &QPushButton::click);
  84         connect(ui->input1_display, &JPEGFrameView::clicked, ui->preview_1_btn, &QPushButton::click);
  85         connect(ui->preview_1_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(0); });
  86         ui->input1_display->set_overlay("1");
  87
  88         QShortcut *preview_2 = new QShortcut(QKeySequence(Qt::Key_2), this);
  89         connect(preview_2, &QShortcut::activated, ui->preview_2_btn, &QPushButton::click);
  90         connect(ui->input2_display, &JPEGFrameView::clicked, ui->preview_2_btn, &QPushButton::click);
  91         connect(ui->preview_2_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(1); });
  92         ui->input2_display->set_overlay("2");
  93
  94         QShortcut *preview_3 = new QShortcut(QKeySequence(Qt::Key_3), this);
  95         connect(preview_3, &QShortcut::activated, ui->preview_3_btn, &QPushButton::click);
  96         connect(ui->input3_display, &JPEGFrameView::clicked, ui->preview_3_btn, &QPushButton::click);
  97         connect(ui->preview_3_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(2); });
  98         ui->input3_display->set_overlay("3");
  99
 100         QShortcut *preview_4 = new QShortcut(QKeySequence(Qt::Key_4), this);
 101         connect(preview_4, &QShortcut::activated, ui->preview_4_btn, &QPushButton::click);
 102         connect(ui->input4_display, &JPEGFrameView::clicked, ui->preview_4_btn, &QPushButton::click);
 103         connect(ui->preview_4_btn, &QPushButton::clicked, [this]{ preview_angle_clicked(3); });
 104         ui->input4_display->set_overlay("4");
 105
 106         connect(ui->playlist_duplicate_btn, &QPushButton::clicked, this, &MainWindow::playlist_duplicate);
 107
 108         connect(ui->playlist_remove_btn, &QPushButton::clicked, this, &MainWindow::playlist_remove);
 109         QShortcut *delete_key = new QShortcut(QKeySequence(Qt::Key_Delete), ui->playlist);
 110         connect(delete_key, &QShortcut::activated, [this] {
 111                 if (ui->playlist->hasFocus()) {
 112                         playlist_remove();
 113                 }
 114         });
 115
 116         // TODO: support drag-and-drop.
 117         connect(ui->playlist_move_up_btn, &QPushButton::clicked, [this]{ playlist_move(-1); });
 118         connect(ui->playlist_move_down_btn, &QPushButton::clicked, [this]{ playlist_move(1); });
 119
 120         connect(ui->playlist->selectionModel(), &QItemSelectionModel::selectionChanged,
 121                 this, &MainWindow::playlist_selection_changed);
 122         playlist_selection_changed();  // First time set-up.
 123
 124         preview_player = new Player(ui->preview_display, /*also_output_to_stream=*/false);
 125         live_player = new Player(ui->live_display, /*also_output_to_stream=*/true);
 126         live_player->set_done_callback([this]{
 127                 post_to_main_thread([this]{
 128                         live_player_clip_done();
 129                 });
 130         });
 131         live_player->set_next_clip_callback(bind(&MainWindow::live_player_get_next_clip, this));
 132         live_player->set_progress_callback([this](double played_this_clip, double total_length) {
 133                 post_to_main_thread([this, played_this_clip, total_length] {
 134                         live_player_clip_progress(played_this_clip, total_length);
 135                 });
 136         });
 137         set_output_status("paused");
 138
 139         defer_timeout = new QTimer(this);
 140         defer_timeout->setSingleShot(true);
 141         connect(defer_timeout, &QTimer::timeout, this, &MainWindow::defer_timer_expired);
 142
 143         connect(ui->clip_list->selectionModel(), &QItemSelectionModel::currentChanged,
 144                 this, &MainWindow::clip_list_selection_changed);
 145 }
 146
 147 void MainWindow::cue_in_clicked()
 148 {
 149         if (!cliplist_clips->empty() && cliplist_clips->back()->pts_out < 0) {
 150                 cliplist_clips->mutable_back()->pts_in = current_pts;
 151                 return;
 152         }
 153         Clip clip;
 154         clip.pts_in = current_pts;
 155         cliplist_clips->add_clip(clip);
 156         playlist_selection_changed();
 157 }
 158
 159 void MainWindow::cue_out_clicked()
 160 {
 161         if (!cliplist_clips->empty()) {
 162                 cliplist_clips->mutable_back()->pts_out = current_pts;
 163                 // TODO: select the row in the clip list?
 164         }
 165 }
 166
 167 void MainWindow::queue_clicked()
 168 {
 169         if (cliplist_clips->empty()) {
 170                 return;
 171         }
 172
 173         QItemSelectionModel *selected = ui->clip_list->selectionModel();
 174         if (!selected->hasSelection()) {
 175                 Clip clip = *cliplist_clips->back();
 176                 clip.stream_idx = 0;
 177                 if (clip.pts_out != -1) {
 178                         playlist_clips->add_clip(clip);
 179                         playlist_selection_changed();
 180                 }
 181                 return;
 182         }
 183
 184         QModelIndex index = selected->currentIndex();
 185         Clip clip = *cliplist_clips->clip(index.row());
 186         if (index.column() >= int(ClipList::Column::CAMERA_1) &&
 187             index.column() <= int(ClipList::Column::CAMERA_4)) {
 188                 clip.stream_idx = index.column() - int(ClipList::Column::CAMERA_1);
 189         } else {
 190                 clip.stream_idx = ui->preview_display->get_stream_idx();
 191         }
 192
 193         if (clip.pts_out != -1) {
 194                 playlist_clips->add_clip(clip);
 195                 playlist_selection_changed();
 196         }
 197 }
 198
 199 void MainWindow::preview_clicked()
 200 {
 201         if (cliplist_clips->empty())
 202                 return;
 203
 204         QItemSelectionModel *selected = ui->clip_list->selectionModel();
 205         if (!selected->hasSelection()) {
 206                 preview_player->play_clip(*cliplist_clips->back(), 0);
 207                 return;
 208         }
 209
 210         QModelIndex index = selected->currentIndex();
 211         unsigned stream_idx;
 212         if (index.column() >= int(ClipList::Column::CAMERA_1) &&
 213             index.column() <= int(ClipList::Column::CAMERA_4)) {
 214                 stream_idx = index.column() - int(ClipList::Column::CAMERA_1);
 215         } else {
 216                 stream_idx = ui->preview_display->get_stream_idx();
 217         }
 218         preview_player->play_clip(*cliplist_clips->clip(index.row()), stream_idx);
 219 }
 220
 221 void MainWindow::preview_angle_clicked(unsigned stream_idx)
 222 {
 223         preview_player->override_angle(stream_idx);
 224
 225         // Change the selection if we were previewing a clip from the clip list.
 226         // (The only other thing we could be showing is a pts scrub, and if so,
 227         // that would be selected.)
 228         QItemSelectionModel *selected = ui->clip_list->selectionModel();
 229         if (selected->hasSelection()) {
 230                 QModelIndex cell = selected->selectedIndexes()[0];
 231                 int column = int(ClipList::Column::CAMERA_1) + stream_idx;
 232                 selected->setCurrentIndex(cell.sibling(cell.row(), column), QItemSelectionModel::ClearAndSelect);
 233         }
 234 }
 235
 236 void MainWindow::playlist_duplicate()
 237 {
 238         QItemSelectionModel *selected = ui->playlist->selectionModel();
 239         if (!selected->hasSelection()) {
 240                 // Should have been grayed out, but OK.
 241                 return;
 242         }
 243         QModelIndexList rows = selected->selectedRows();
 244         int first = rows.front().row(), last = rows.back().row();
 245         playlist_clips->duplicate_clips(first, last);
 246         playlist_selection_changed();
 247 }
 248
 249 void MainWindow::playlist_remove()
 250 {
 251         QItemSelectionModel *selected = ui->playlist->selectionModel();
 252         if (!selected->hasSelection()) {
 253                 // Should have been grayed out, but OK.
 254                 return;
 255         }
 256         QModelIndexList rows = selected->selectedRows();
 257         int first = rows.front().row(), last = rows.back().row();
 258         playlist_clips->erase_clips(first, last);
 259
 260         // TODO: select the next one in the list?
 261
 262         playlist_selection_changed();
 263 }
 264
 265 void MainWindow::playlist_move(int delta)
 266 {
 267         QItemSelectionModel *selected = ui->playlist->selectionModel();
 268         if (!selected->hasSelection()) {
 269                 // Should have been grayed out, but OK.
 270                 return;
 271         }
 272
 273         QModelIndexList rows = selected->selectedRows();
 274         int first = rows.front().row(), last = rows.back().row();
 275         if ((delta == -1 && first == 0) ||
 276             (delta == 1 && size_t(last) == playlist_clips->size() - 1)) {
 277                 // Should have been grayed out, but OK.
 278                 return;
 279         }
 280
 281         playlist_clips->move_clips(first, last, delta);
 282         playlist_selection_changed();
 283 }
 284
 285 void MainWindow::defer_timer_expired()
 286 {
 287         state_changed(deferred_state);
 288 }
 289
 290 void MainWindow::content_changed()
 291 {
 292         if (defer_timeout->isActive() &&
 293             (!currently_deferring_model_changes || deferred_change_id != current_change_id)) {
 294                 // There's some deferred event waiting, but this event is unrelated.
 295                 // So it's time to short-circuit that timer and do the work it wanted to do.
 296                 defer_timeout->stop();
 297                 state_changed(deferred_state);
 298         }
 299         StateProto state;
 300         *state.mutable_clip_list() = cliplist_clips->serialize();
 301         *state.mutable_play_list() = playlist_clips->serialize();
 302         if (currently_deferring_model_changes) {
 303                 deferred_change_id = current_change_id;
 304                 deferred_state = std::move(state);
 305                 defer_timeout->start(200);
 306                 return;
 307         }
 308         state_changed(state);
 309 }
 310
 311 void MainWindow::state_changed(const StateProto &state)
 312 {
 313         db.store_state(state);
 314 }
 315
 316 void MainWindow::play_clicked()
 317 {
 318         if (playlist_clips->empty())
 319                 return;
 320
 321         QItemSelectionModel *selected = ui->playlist->selectionModel();
 322         int row;
 323         if (!selected->hasSelection()) {
 324                 row = 0;
 325         } else {
 326                 row = selected->selectedRows(0)[0].row();
 327         }
 328
 329         const Clip &clip = *playlist_clips->clip(row);
 330         live_player->play_clip(clip, clip.stream_idx);
 331         playlist_clips->set_currently_playing(row, 0.0f);
 332         playlist_selection_changed();
 333 }
 334
 335 void MainWindow::live_player_clip_done()
 336 {
 337         int row = playlist_clips->get_currently_playing();
 338         if (row == -1 || row == int(playlist_clips->size()) - 1) {
 339                 set_output_status("paused");
 340                 playlist_clips->set_currently_playing(-1, 0.0f);
 341         } else {
 342                 playlist_clips->set_currently_playing(row + 1, 0.0f);
 343         }
 344 }
 345
 346 Clip MainWindow::live_player_get_next_clip()
 347 {
 348         // playlist_clips can only be accessed on the main thread.
 349         // Hopefully, we won't have to wait too long for this to come back.
 350         promise<Clip> clip_promise;
 351         future<Clip> clip = clip_promise.get_future();
 352         post_to_main_thread([this, &clip_promise] {
 353                 int row = playlist_clips->get_currently_playing();
 354                 if (row != -1 && row < int(playlist_clips->size()) - 1) {
 355                         clip_promise.set_value(*playlist_clips->clip(row + 1));
 356                 } else {
 357                         clip_promise.set_value(Clip());
 358                 }
 359         });
 360         return clip.get();
 361 }
 362
 363 void MainWindow::live_player_clip_progress(double played_this_clip, double total_length)
 364 {
 365         playlist_clips->set_currently_playing(playlist_clips->get_currently_playing(), played_this_clip / total_length);
 366
 367         double remaining = total_length - played_this_clip;
 368         for (int row = playlist_clips->get_currently_playing() + 1; row < int(playlist_clips->size()); ++row) {
 369                 const Clip clip = *playlist_clips->clip(row);
 370                 remaining += double(clip.pts_out - clip.pts_in) / TIMEBASE / 0.5;  // FIXME: stop hardcoding speed.
 371         }
 372         int remaining_ms = lrint(remaining * 1e3);
 373
 374         int ms = remaining_ms % 1000;
 375         remaining_ms /= 1000;
 376         int s = remaining_ms % 60;
 377         remaining_ms /= 60;
 378         int m = remaining_ms;
 379
 380         char buf[256];
 381         snprintf(buf, sizeof(buf), "%d:%02d.%03d left", m, s, ms);
 382         set_output_status(buf);
 383 }
 384
 385 void MainWindow::resizeEvent(QResizeEvent *event)
 386 {
 387         QMainWindow::resizeEvent(event);
 388
 389         // Ask for a relayout, but only after the event loop is done doing relayout
 390         // on everything else.
 391         QMetaObject::invokeMethod(this, "relayout", Qt::QueuedConnection);
 392 }
 393
 394 void MainWindow::relayout()
 395 {
 396         ui->live_display->setMinimumWidth(ui->live_display->height() * 16 / 9);
 397         ui->preview_display->setMinimumWidth(ui->preview_display->height() * 16 / 9);
 398 }
 399
 400 void set_pts_in(int64_t pts, int64_t current_pts, ClipProxy &clip)
 401 {
 402         pts = std::max<int64_t>(pts, 0);
 403         if (clip->pts_out == -1) {
 404                 pts = std::min(pts, current_pts);
 405         } else {
 406                 pts = std::min(pts, clip->pts_out);
 407         }
 408         clip->pts_in = pts;
 409 }
 410
 411 bool MainWindow::eventFilter(QObject *watched, QEvent *event)
 412 {
 413         constexpr int dead_zone_pixels = 3;  // To avoid that simple clicks get misinterpreted.
 414         constexpr int scrub_sensitivity = 100;  // pts units per pixel.
 415         constexpr int wheel_sensitivity = 100;  // pts units per degree.
 416         constexpr int camera_degrees_per_pixel = 15;  // One click of most mice.
 417
 418         unsigned stream_idx = ui->preview_display->get_stream_idx();
 419
 420         if (watched == ui->clip_list) {
 421                 if (event->type() == QEvent::FocusOut) {
 422                         highlight_camera_input(-1);
 423                 }
 424                 return false;
 425         }
 426
 427         if (event->type() != QEvent::Wheel) {
 428                 last_mousewheel_camera_row = -1;
 429         }
 430
 431         if (event->type() == QEvent::MouseButtonPress) {
 432                 QMouseEvent *mouse = (QMouseEvent *)event;
 433
 434                 QTableView *destination;
 435                 ScrubType type;
 436
 437                 if (watched == ui->clip_list->viewport()) {
 438                         destination = ui->clip_list;
 439                         type = SCRUBBING_CLIP_LIST;
 440                 } else if (watched == ui->playlist->viewport()) {
 441                         destination = ui->playlist;
 442                         type = SCRUBBING_PLAYLIST;
 443                 } else {
 444                         return false;
 445                 }
 446                 int column = destination->columnAt(mouse->x());
 447                 int row = destination->rowAt(mouse->y());
 448                 if (column == -1 || row == -1)
 449                         return false;
 450
 451                 if (type == SCRUBBING_CLIP_LIST) {
 452                         if (ClipList::Column(column) == ClipList::Column::IN) {
 453                                 scrub_pts_origin = cliplist_clips->clip(row)->pts_in;
 454                                 preview_single_frame(scrub_pts_origin, stream_idx, FIRST_AT_OR_AFTER);
 455                         } else if (ClipList::Column(column) == ClipList::Column::OUT) {
 456                                 scrub_pts_origin = cliplist_clips->clip(row)->pts_out;
 457                                 preview_single_frame(scrub_pts_origin, stream_idx, LAST_BEFORE);
 458                         } else {
 459                                 return false;
 460                         }
 461                 } else {
 462                         if (PlayList::Column(column) == PlayList::Column::IN) {
 463                                 scrub_pts_origin = playlist_clips->clip(row)->pts_in;
 464                                 preview_single_frame(scrub_pts_origin, stream_idx, FIRST_AT_OR_AFTER);
 465                         } else if (PlayList::Column(column) == PlayList::Column::OUT) {
 466                                 scrub_pts_origin = playlist_clips->clip(row)->pts_out;
 467                                 preview_single_frame(scrub_pts_origin, stream_idx, LAST_BEFORE);
 468                         } else {
 469                                 return false;
 470                         }
 471                 }
 472
 473                 scrubbing = true;
 474                 scrub_row = row;
 475                 scrub_column = column;
 476                 scrub_x_origin = mouse->x();
 477                 scrub_type = type;
 478         } else if (event->type() == QEvent::MouseMove) {
 479                 if (scrubbing) {
 480                         QMouseEvent *mouse = (QMouseEvent *)event;
 481                         int offset = mouse->x() - scrub_x_origin;
 482                         int adjusted_offset;
 483                         if (offset >= dead_zone_pixels) {
 484                                 adjusted_offset = offset - dead_zone_pixels;
 485                         } else if (offset < -dead_zone_pixels) {
 486                                 adjusted_offset = offset + dead_zone_pixels;
 487                         } else {
 488                                 adjusted_offset = 0;
 489                         }
 490
 491                         int64_t pts = scrub_pts_origin + adjusted_offset * scrub_sensitivity;
 492                         currently_deferring_model_changes = true;
 493                         if (scrub_type == SCRUBBING_CLIP_LIST) {
 494                                 ClipProxy clip = cliplist_clips->mutable_clip(scrub_row);
 495                                 if (scrub_column == int(ClipList::Column::IN)) {
 496                                         current_change_id = "cliplist:in:" + to_string(scrub_row);
 497                                         set_pts_in(pts, current_pts, clip);
 498                                         preview_single_frame(pts, stream_idx, FIRST_AT_OR_AFTER);
 499                                 } else {
 500                                         current_change_id = "cliplist:out" + to_string(scrub_row);
 501                                         pts = std::max(pts, clip->pts_in);
 502                                         pts = std::min(pts, current_pts);
 503                                         clip->pts_out = pts;
 504                                         preview_single_frame(pts, stream_idx, LAST_BEFORE);
 505                                 }
 506                         } else {
 507                                 ClipProxy clip = playlist_clips->mutable_clip(scrub_row);
 508                                 if (scrub_column == int(PlayList::Column::IN)) {
 509                                         current_change_id = "playlist:in:" + to_string(scrub_row);
 510                                         set_pts_in(pts, current_pts, clip);
 511                                         preview_single_frame(pts, clip->stream_idx, FIRST_AT_OR_AFTER);
 512                                 } else {
 513                                         current_change_id = "playlist:out:" + to_string(scrub_row);
 514                                         pts = std::max(pts, clip->pts_in);
 515                                         pts = std::min(pts, current_pts);
 516                                         clip->pts_out = pts;
 517                                         preview_single_frame(pts, clip->stream_idx, LAST_BEFORE);
 518                                 }
 519                         }
 520                         currently_deferring_model_changes = false;
 521
 522                         return true;  // Don't use this mouse movement for selecting things.
 523                 }
 524         } else if (event->type() == QEvent::Wheel) {
 525                 QWheelEvent *wheel = (QWheelEvent *)event;
 526
 527                 QTableView *destination;
 528                 int in_column, out_column, camera_column;
 529                 if (watched == ui->clip_list->viewport()) {
 530                         destination = ui->clip_list;
 531                         in_column = int(ClipList::Column::IN);
 532                         out_column = int(ClipList::Column::OUT);
 533                         camera_column = -1;
 534                         last_mousewheel_camera_row = -1;
 535                 } else if (watched == ui->playlist->viewport()) {
 536                         destination = ui->playlist;
 537                         in_column = int(PlayList::Column::IN);
 538                         out_column = int(PlayList::Column::OUT);
 539                         camera_column = int(PlayList::Column::CAMERA);
 540                 } else {
 541                         last_mousewheel_camera_row = -1;
 542                         return false;
 543                 }
 544                 int column = destination->columnAt(wheel->x());
 545                 int row = destination->rowAt(wheel->y());
 546                 if (column == -1 || row == -1) return false;
 547
 548                 currently_deferring_model_changes = true;
 549                 {
 550                         current_change_id = (watched == ui->clip_list->viewport()) ? "cliplist:" : "playlist:";
 551                         ClipProxy clip = (watched == ui->clip_list->viewport()) ?
 552                                 cliplist_clips->mutable_clip(row) : playlist_clips->mutable_clip(row);
 553                         if (watched == ui->playlist->viewport()) {
 554                                 stream_idx = clip->stream_idx;
 555                         }
 556
 557                         if (column != camera_column) {
 558                                 last_mousewheel_camera_row = -1;
 559                         }
 560                         if (column == in_column) {
 561                                 current_change_id += "in:" + to_string(row);
 562                                 int64_t pts = clip->pts_in + wheel->angleDelta().y() * wheel_sensitivity;
 563                                 set_pts_in(pts, current_pts, clip);
 564                                 preview_single_frame(pts, stream_idx, FIRST_AT_OR_AFTER);
 565                         } else if (column == out_column) {
 566                                 current_change_id += "out:" + to_string(row);
 567                                 int64_t pts = clip->pts_out + wheel->angleDelta().y() * wheel_sensitivity;
 568                                 pts = std::max(pts, clip->pts_in);
 569                                 pts = std::min(pts, current_pts);
 570                                 clip->pts_out = pts;
 571                                 preview_single_frame(pts, stream_idx, LAST_BEFORE);
 572                         } else if (column == camera_column) {
 573                                 current_change_id += "camera:" + to_string(row);
 574                                 int angle_degrees = wheel->angleDelta().y();
 575                                 if (last_mousewheel_camera_row == row) {
 576                                         angle_degrees += leftover_angle_degrees;
 577                                 }
 578
 579                                 int stream_idx = clip->stream_idx + angle_degrees / camera_degrees_per_pixel;
 580                                 stream_idx = std::max(stream_idx, 0);
 581                                 stream_idx = std::min(stream_idx, NUM_CAMERAS - 1);
 582                                 clip->stream_idx = stream_idx;
 583
 584                                 last_mousewheel_camera_row = row;
 585                                 leftover_angle_degrees = angle_degrees % camera_degrees_per_pixel;
 586
 587                                 // Don't update the live view, that's rarely what the operator wants.
 588                         }
 589                 }
 590                 currently_deferring_model_changes = false;
 591         } else if (event->type() == QEvent::MouseButtonRelease) {
 592                 scrubbing = false;
 593         }
 594         return false;
 595 }
 596
 597 void MainWindow::preview_single_frame(int64_t pts, unsigned stream_idx, MainWindow::Rounding rounding)
 598 {
 599         if (rounding == LAST_BEFORE) {
 600                 lock_guard<mutex> lock(frame_mu);
 601                 if (frames[stream_idx].empty())
 602                         return;
 603                 auto it = lower_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts);
 604                 if (it != frames[stream_idx].end()) {
 605                         pts = *it;
 606                 }
 607         } else {
 608                 assert(rounding == FIRST_AT_OR_AFTER);
 609                 lock_guard<mutex> lock(frame_mu);
 610                 if (frames[stream_idx].empty())
 611                         return;
 612                 auto it = upper_bound(frames[stream_idx].begin(), frames[stream_idx].end(), pts - 1);
 613                 if (it != frames[stream_idx].end()) {
 614                         pts = *it;
 615                 }
 616         }
 617
 618         Clip fake_clip;
 619         fake_clip.pts_in = pts;
 620         fake_clip.pts_out = pts + 1;
 621         preview_player->play_clip(fake_clip, stream_idx);
 622 }
 623
 624 void MainWindow::playlist_selection_changed()
 625 {
 626         QItemSelectionModel *selected = ui->playlist->selectionModel();
 627         bool any_selected = selected->hasSelection();
 628         ui->playlist_duplicate_btn->setEnabled(any_selected);
 629         ui->playlist_remove_btn->setEnabled(any_selected);
 630         ui->playlist_move_up_btn->setEnabled(
 631                 any_selected && selected->selectedRows().front().row() > 0);
 632         ui->playlist_move_down_btn->setEnabled(
 633                 any_selected && selected->selectedRows().back().row() < int(playlist_clips->size()) - 1);
 634         ui->play_btn->setEnabled(!playlist_clips->empty());
 635 }
 636
 637 void MainWindow::clip_list_selection_changed(const QModelIndex &current, const QModelIndex &)
 638 {
 639         int camera_selected = -1;
 640         if (current.column() >= int(ClipList::Column::CAMERA_1) &&
 641             current.column() <= int(ClipList::Column::CAMERA_4)) {
 642                 camera_selected = current.column() - int(ClipList::Column::CAMERA_1);
 643         }
 644         highlight_camera_input(camera_selected);
 645 }
 646
 647 void MainWindow::report_disk_space(off_t free_bytes, double estimated_seconds_left)
 648 {
 649         char time_str[256];
 650         if (estimated_seconds_left < 60.0) {
 651                 strcpy(time_str, "<font color=\"red\">Less than a minute</font>");
 652         } else if (estimated_seconds_left < 1800.0) {  // Less than half an hour: Xm Ys (red).
 653                 int s = lrintf(estimated_seconds_left);
 654                 int m = s / 60;
 655                 s %= 60;
 656                 snprintf(time_str, sizeof(time_str), "<font color=\"red\">%dm %ds</font>", m, s);
 657         } else if (estimated_seconds_left < 3600.0) {  // Less than an hour: Xm.
 658                 int m = lrintf(estimated_seconds_left / 60.0);
 659                 snprintf(time_str, sizeof(time_str), "%dm", m);
 660         } else if (estimated_seconds_left < 36000.0) {  // Less than ten hours: Xh Ym.
 661                 int m = lrintf(estimated_seconds_left / 60.0);
 662                 int h = m / 60;
 663                 m %= 60;
 664                 snprintf(time_str, sizeof(time_str), "%dh %dm", h, m);
 665         } else {  // More than ten hours: Xh.
 666                 int h = lrintf(estimated_seconds_left / 3600.0);
 667                 snprintf(time_str, sizeof(time_str), "%dh", h);
 668         }
 669         char buf[256];
 670         snprintf(buf, sizeof(buf), "Disk free: %'.0f MB (approx. %s)", free_bytes / 1048576.0, time_str);
 671
 672         std::string label = buf;
 673
 674         post_to_main_thread([this, label] {
 675                 disk_free_label->setText(QString::fromStdString(label));
 676                 ui->menuBar->setCornerWidget(disk_free_label);  // Need to set this again for the sizing to get right.
 677         });
 678 }
 679
 680 void MainWindow::exit_triggered()
 681 {
 682         close();
 683 }
 684
 685 void MainWindow::highlight_camera_input(int stream_idx)
 686 {
 687         if (stream_idx == 0) {
 688                 ui->input1_frame->setStyleSheet("background: rgb(0,255,0)");
 689         } else {
 690                 ui->input1_frame->setStyleSheet("");
 691         }
 692         if (stream_idx == 1) {
 693                 ui->input2_frame->setStyleSheet("background: rgb(0,255,0)");
 694         } else {
 695                 ui->input2_frame->setStyleSheet("");
 696         }
 697         if (stream_idx == 2) {
 698                 ui->input3_frame->setStyleSheet("background: rgb(0,255,0)");
 699         } else {
 700                 ui->input3_frame->setStyleSheet("");
 701         }
 702         if (stream_idx == 3) {
 703                 ui->input4_frame->setStyleSheet("background: rgb(0,255,0)");
 704         } else {
 705                 ui->input4_frame->setStyleSheet("");
 706         }
 707 }
 708
 709 void MainWindow::set_output_status(const string &status)
 710 {
 711         ui->live_label->setText(QString::fromStdString("Current output (" + status + ")"));
 712
 713         lock_guard<mutex> lock(queue_status_mu);
 714         queue_status = status;
 715 }
 716
 717 pair<string, string> MainWindow::get_queue_status() const {
 718         lock_guard<mutex> lock(queue_status_mu);
 719         return {queue_status, "text/plain"};
 720 }