]> git.sesse.net Git - nageru/blob - mainwindow.cpp
Convert Y'CbCr using Movit instead of in software with Qt. Saves a fair amount of...
[nageru] / mainwindow.cpp
1 #include "mainwindow.h"
2
3 #include "clip_list.h"
4 #include "player.h"
5 #include "post_to_main_thread.h"
6 #include "ui_mainwindow.h"
7
8 #include <string>
9 #include <vector>
10
11 #include <QShortcut>
12
13 using namespace std;
14
15 MainWindow *global_mainwindow = nullptr;
16 extern int64_t current_pts;
17 ClipList *cliplist_clips, *playlist_clips;
18
19 MainWindow::MainWindow()
20         : ui(new Ui::MainWindow)
21 {
22         global_mainwindow = this;
23         ui->setupUi(this);
24
25         cliplist_clips = new ClipList(ClipList::ListDisplay::CLIP_LIST);
26         ui->clip_list->setModel(cliplist_clips);
27
28         playlist_clips = new ClipList(ClipList::ListDisplay::PLAY_LIST);
29         ui->playlist->setModel(playlist_clips);
30
31         // TODO: These are too big for lambdas.
32         QShortcut *cue_in = new QShortcut(QKeySequence(Qt::Key_A), this);
33         connect(cue_in, &QShortcut::activated, ui->cue_in_btn, &QPushButton::click);
34         connect(ui->cue_in_btn, &QPushButton::clicked, []{
35                 if (!cliplist_clips->empty() && cliplist_clips->back()->pts_out < 0) {
36                         cliplist_clips->back()->pts_in = current_pts;
37                         return;
38                 }
39                 Clip clip;
40                 clip.pts_in = current_pts;
41                 cliplist_clips->add_clip(clip);
42         });
43
44         QShortcut *cue_out = new QShortcut(QKeySequence(Qt::Key_S), this);
45         connect(cue_out, &QShortcut::activated, ui->cue_out_btn, &QPushButton::click);
46         connect(ui->cue_out_btn, &QPushButton::clicked, []{
47                 if (!cliplist_clips->empty()) {
48                         cliplist_clips->back()->pts_out = current_pts;
49                         // TODO: select the row in the clip list?
50                 }
51         });
52
53         QShortcut *queue = new QShortcut(QKeySequence(Qt::Key_Q), this);
54         connect(queue, &QShortcut::activated, ui->queue_btn, &QPushButton::click);
55         connect(ui->queue_btn, &QPushButton::clicked, this, &MainWindow::queue_clicked);
56
57         QShortcut *preview = new QShortcut(QKeySequence(Qt::Key_W), this);
58         connect(preview, &QShortcut::activated, ui->preview_btn, &QPushButton::click);
59         connect(ui->preview_btn, &QPushButton::clicked, this, &MainWindow::preview_clicked);
60
61         QShortcut *play = new QShortcut(QKeySequence(Qt::Key_Space), this);
62         connect(play, &QShortcut::activated, ui->play_btn, &QPushButton::click);
63         connect(ui->play_btn, &QPushButton::clicked, this, &MainWindow::play_clicked);
64
65         preview_player = new Player(ui->preview_display);
66         live_player = new Player(ui->live_display);
67         live_player->set_done_callback([this]{
68                 post_to_main_thread([this]{
69                         live_player_clip_done();
70                 });
71         });
72 }
73
74 void MainWindow::queue_clicked()
75 {
76         QItemSelectionModel *selected = ui->clip_list->selectionModel();
77         if (!selected->hasSelection()) {
78                 Clip clip = *cliplist_clips->back();
79                 clip.stream_idx = 0;
80                 playlist_clips->add_clip(clip);
81                 return;
82         }
83
84         QModelIndex index = selected->currentIndex();
85         if (index.column() >= int(ClipList::ClipListColumn::CAMERA_1) &&
86             index.column() <= int(ClipList::ClipListColumn::CAMERA_4)) {
87                 Clip clip = *cliplist_clips->clip(index.row());
88                 clip.stream_idx = index.column() - int(ClipList::ClipListColumn::CAMERA_1);
89                 playlist_clips->add_clip(clip);
90         }
91 }
92
93 void MainWindow::preview_clicked()
94 {
95         QItemSelectionModel *selected = ui->clip_list->selectionModel();
96         if (!selected->hasSelection()) {
97                 preview_player->play_clip(*cliplist_clips->back(), 0);
98                 return;
99         }
100
101         QModelIndex index = selected->currentIndex();
102         if (index.column() >= int(ClipList::ClipListColumn::CAMERA_1) &&
103             index.column() <= int(ClipList::ClipListColumn::CAMERA_4)) {
104                 unsigned stream_idx = index.column() - int(ClipList::ClipListColumn::CAMERA_1);
105                 preview_player->play_clip(*cliplist_clips->clip(index.row()), stream_idx);
106         }
107 }
108
109 void MainWindow::play_clicked()
110 {
111         QItemSelectionModel *selected = ui->playlist->selectionModel();
112         int row;
113         if (!selected->hasSelection()) {
114                 row = 0;
115         } else {
116                 row = selected->selectedRows(0)[0].row();
117         }
118
119         const Clip &clip = *playlist_clips->clip(row);
120         live_player->play_clip(clip, clip.stream_idx);
121         playlist_clips->set_currently_playing(row);
122 }
123
124 void MainWindow::live_player_clip_done()
125 {
126         int row = playlist_clips->get_currently_playing();
127         if (row != -1 && row < int(playlist_clips->size()) - 1) {
128                 ++row;
129                 const Clip &clip = *playlist_clips->clip(row);
130                 live_player->play_clip(clip, clip.stream_idx);
131                 playlist_clips->set_currently_playing(row);
132         } else {
133                 playlist_clips->set_currently_playing(-1);
134         }
135 }
136
137 void MainWindow::resizeEvent(QResizeEvent *event)
138 {
139         QMainWindow::resizeEvent(event);
140
141         // Ask for a relayout, but only after the event loop is done doing relayout
142         // on everything else.
143         QMetaObject::invokeMethod(this, "relayout", Qt::QueuedConnection);
144 }
145
146 void MainWindow::relayout()
147 {
148         ui->live_display->setMinimumHeight(ui->live_display->width() * 9 / 16);
149 }