1 #include <QMediaPlayer>
3 #include <QApplication>
5 #include <QVideoWidget>
7 #include <QInputDialog>
14 #include "mainwindow.h"
15 #include "ui_mainwindow.h"
22 string format_timestamp(uint64_t pos)
32 snprintf(buf, sizeof(buf), "%d:%02d:%02d.%03d", hour, min, sec, ms);
36 MainWindow::MainWindow(EventsModel *events, PlayersModel *players) : events(events), players(players)
38 video = new QMediaPlayer;
39 //video->setSource(QUrl::fromLocalFile("/home/sesse/dev/stats/ultimate.mkv"));
40 video->setSource(QUrl::fromLocalFile("/home/sesse/dev/stats/ultimate-prores.mkv"));
43 ui = new Ui::MainWindow;
46 ui->event_view->setModel(events);
47 connect(ui->event_view->selectionModel(), &QItemSelectionModel::currentRowChanged,
48 [this, events](const QModelIndex ¤t, const QModelIndex &previous) {
49 uint64_t t = events->get_time(current.row());
50 if (t != video->position()) {
51 video->setPosition(events->get_time(current.row()));
53 // Selection could have changed, so we still need to update.
54 // (Just calling setPosition() would not give us the signal
56 update_ui_from_time(t);
60 ui->player_view->setModel(players);
61 ui->player_view->setColumnWidth(0, 30);
62 ui->player_view->setColumnWidth(1, 20);
63 ui->player_view->horizontalHeader()->setStretchLastSection(true);
65 connect(video, &QMediaPlayer::positionChanged, [this](uint64_t pos) {
66 position_changed(pos);
69 video->setVideoOutput(ui->video);
71 // It's not really clear whether PgUp should be forwards or backwards,
72 // but mpv does at least up = forwards, so that's probably standard.
73 QShortcut *pgdown = new QShortcut(QKeySequence(Qt::Key_PageDown), this);
74 connect(pgdown, &QShortcut::activated, [this]() { seek(-120000); });
75 QShortcut *pgup = new QShortcut(QKeySequence(Qt::Key_PageUp), this);
76 connect(pgup, &QShortcut::activated, [this]() { seek(120000); });
78 // Ugh. Used when Qt messes up and hangs the video.
79 QShortcut *f5 = new QShortcut(QKeySequence(Qt::Key_F5), this);
80 connect(f5, &QShortcut::activated, [this]() {
81 QVideoWidget *nvw = new QVideoWidget(ui->video->parentWidget());
82 nvw->setObjectName("video");
83 nvw->setMinimumSize(QSize(320, 240));
84 video->setVideoOutput(nvw);
85 ui->main_grid->replaceWidget(ui->video, nvw);
89 connect(ui->minus10s, &QPushButton::clicked, [this]() { seek(-10000); });
90 connect(ui->plus10s, &QPushButton::clicked, [this]() { seek(10000); });
92 connect(ui->minus2s, &QPushButton::clicked, [this]() { seek(-2000); });
93 connect(ui->plus2s, &QPushButton::clicked, [this]() { seek(2000); });
95 // TODO: Would be nice to actually have a frame...
96 connect(ui->minus1f, &QPushButton::clicked, [this]() { seek(-20); });
97 connect(ui->plus1f, &QPushButton::clicked, [this]() { seek(20); });
99 connect(ui->play_pause, &QPushButton::clicked, [this]() {
102 ui->play_pause->setText("Play (space)");
104 video->setPlaybackRate(1.0);
106 ui->play_pause->setText("Pause (space)");
110 // Needs to be set anew when we modify setText(), evidently.
111 ui->play_pause->setShortcut(QCoreApplication::translate("MainWindow", "Space", nullptr));
114 connect(ui->player_1, &QPushButton::clicked, [this]() { insert_player_event(0); });
115 connect(ui->player_2, &QPushButton::clicked, [this]() { insert_player_event(1); });
116 connect(ui->player_3, &QPushButton::clicked, [this]() { insert_player_event(2); });
117 connect(ui->player_4, &QPushButton::clicked, [this]() { insert_player_event(3); });
118 connect(ui->player_5, &QPushButton::clicked, [this]() { insert_player_event(4); });
119 connect(ui->player_6, &QPushButton::clicked, [this]() { insert_player_event(5); });
120 connect(ui->player_7, &QPushButton::clicked, [this]() { insert_player_event(6); });
123 connect(ui->offense_label, &ClickableLabel::clicked, [this]() { insert_noplayer_event("set_offense"); });
124 connect(ui->catch_, &QPushButton::clicked, [this]() { set_current_event_type("catch"); });
125 connect(ui->throwaway, &QPushButton::clicked, [this, events]() {
126 EventsModel::Status s = events->get_status_at(video->position());
127 if (s.attack_state == EventsModel::Status::DEFENSE && s.pull_state == EventsModel::Status::PULL_IN_AIR) {
128 insert_noplayer_event("pull_oob");
130 set_current_event_type("throwaway");
133 connect(ui->drop, &QPushButton::clicked, [this]() { set_current_event_type("drop"); });
134 connect(ui->goal, &QPushButton::clicked, [this]() { set_current_event_type("goal"); });
135 connect(ui->offensive_soft_plus, &QPushButton::clicked, [this]() { set_current_event_type("offensive_soft_plus"); });
136 connect(ui->offensive_soft_minus, &QPushButton::clicked, [this]() { set_current_event_type("offensive_soft_minus"); });
137 connect(ui->pull, &QPushButton::clicked, [this, events]() {
138 EventsModel::Status s = events->get_status_at(video->position());
139 if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
140 set_current_event_type("pull");
141 } else if (EventsModel::Status::PULL_IN_AIR) {
142 insert_noplayer_event("pull_landed");
146 // Defensive events (TODO add more)
147 connect(ui->interception, &QPushButton::clicked, [this]() { set_current_event_type("interception"); });
148 connect(ui->defense_label, &ClickableLabel::clicked, [this]() { insert_noplayer_event("set_defense"); });
149 connect(ui->their_throwaway, &QPushButton::clicked, [this]() { insert_noplayer_event("their_throwaway"); });
150 connect(ui->their_goal, &QPushButton::clicked, [this]() { insert_noplayer_event("their_goal"); });
151 connect(ui->their_pull, &QPushButton::clicked, [this, events]() {
152 EventsModel::Status s = events->get_status_at(video->position());
153 if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
154 insert_noplayer_event("their_pull");
157 connect(ui->our_defense, &QPushButton::clicked, [this]() { set_current_event_type("defense"); });
158 connect(ui->defensive_soft_plus, &QPushButton::clicked, [this]() { set_current_event_type("defensive_soft_plus"); });
159 connect(ui->defensive_soft_minus, &QPushButton::clicked, [this]() { set_current_event_type("defensive_soft_minus"); });
162 connect(ui->substitution, &QPushButton::clicked, [this]() { make_substitution(); });
163 connect(ui->stoppage, &QPushButton::clicked, [this, events]() {
164 EventsModel::Status s = events->get_status_at(video->position());
166 insert_noplayer_event("restart");
168 insert_noplayer_event("stoppage");
171 connect(ui->unknown, &QPushButton::clicked, [this]() { insert_noplayer_event("unknown"); });
173 QShortcut *key_delete = new QShortcut(QKeySequence(Qt::Key_Delete), this);
174 connect(key_delete, &QShortcut::activated, [this]() { ui->delete_->animateClick(); });
175 connect(ui->delete_, &QPushButton::clicked, [this]() { delete_current_event(); });
178 void MainWindow::position_changed(uint64_t pos)
180 ui->timestamp->setText(QString::fromUtf8(format_timestamp(pos)));
182 video->setPosition(*buffered_seek);
183 buffered_seek.reset();
186 video->pause(); // We only played to get a picture.
189 QModelIndex row = events->get_last_event_qt(video->position());
190 ui->event_view->scrollTo(row, QAbstractItemView::PositionAtCenter);
192 update_ui_from_time(pos);
195 void MainWindow::seek(int64_t delta_ms)
197 int64_t current_pos = buffered_seek ? *buffered_seek : video->position();
198 uint64_t pos = max<int64_t>(current_pos + delta_ms, 0);
201 video->setPlaybackRate(0.01);
202 video->play(); // Or Qt won't show the seek.
206 void MainWindow::insert_player_event(int button_id)
208 uint64_t t = video->position();
209 vector<int> team = events->sort_team(events->get_team_at(t));
210 if (button_id >= team.size()) {
213 int player_id = team[button_id];
215 EventsModel::Status s = events->get_status_at(t);
217 ui->event_view->selectionModel()->blockSignals(true);
218 if (s.attack_state == EventsModel::Status::OFFENSE) {
219 // TODO: Perhaps not if that player already did the last catch?
220 ui->event_view->selectRow(events->insert_event(t, player_id, "catch"));
222 ui->event_view->selectRow(events->insert_event(t, player_id));
224 ui->event_view->selectionModel()->blockSignals(false);
226 update_ui_from_time(t);
229 void MainWindow::insert_noplayer_event(const string &type)
231 uint64_t t = video->position();
233 ui->event_view->selectionModel()->blockSignals(true);
234 ui->event_view->selectRow(events->insert_event(t, nullopt, type));
235 ui->event_view->selectionModel()->blockSignals(false);
237 update_ui_from_time(t);
240 void MainWindow::set_current_event_type(const string &type)
242 QItemSelectionModel *select = ui->event_view->selectionModel();
243 if (!select->hasSelection()) {
246 int row = select->selectedRows().front().row(); // Should only be one, due to our selection behavior.
247 events->set_event_type(row, type);
248 update_ui_from_time(video->position());
251 void MainWindow::delete_current_event()
253 QItemSelectionModel *select = ui->event_view->selectionModel();
254 if (!select->hasSelection()) {
257 int row = select->selectedRows().front().row(); // Should only be one, due to our selection behavior.
258 ui->event_view->selectionModel()->blockSignals(true);
259 events->delete_event(row);
260 ui->event_view->selectionModel()->blockSignals(false);
261 update_ui_from_time(video->position());
264 void MainWindow::make_substitution()
266 QItemSelectionModel *select = ui->player_view->selectionModel();
268 for (QModelIndex row : select->selectedRows()) {
269 new_team.insert(players->get_player_id(row.row()));
271 events->set_team_at(video->position(), new_team);
274 void MainWindow::update_ui_from_time(uint64_t t)
277 update_player_buttons(t);
278 update_action_buttons(t);
281 void MainWindow::update_status(uint64_t t)
283 EventsModel::Status s = events->get_status_at(t);
285 const char *offense = "not started";
286 if (s.attack_state == EventsModel::Status::OFFENSE) {
288 } else if (s.attack_state == EventsModel::Status::DEFENSE) {
292 snprintf(buf, sizeof(buf), "%d–%d | %s | %d passes, %d sec possession",
293 s.our_score, s.their_score, offense, s.num_passes, s.possession_sec);
294 if (s.stoppage_sec > 0) {
296 snprintf(buf2, sizeof(buf2), "%s (plus %d sec stoppage)", buf, s.stoppage_sec);
297 ui->status->setText(buf2);
299 ui->status->setText(buf);
303 void MainWindow::update_player_buttons(uint64_t t)
305 QPushButton *buttons[] = {
314 const char shortcuts[] = "qweasdf";
316 for (int player_id : events->sort_team(events->get_team_at(t))) {
317 QPushButton *btn = buttons[num_players];
318 string label = players->get_player_name_by_id(player_id) + " (&" + shortcuts[num_players] + ")";
319 char shortcut[2] = "";
320 shortcut[0] = toupper(shortcuts[num_players]);
321 btn->setText(QString::fromUtf8(label));
322 btn->setShortcut(QCoreApplication::translate("MainWindow", shortcut, nullptr));
323 btn->setEnabled(true);
324 if (++num_players == 7) {
328 for (int i = num_players; i < 7; ++i) {
329 QPushButton *btn = buttons[i];
330 btn->setText("No player");
331 btn->setEnabled(false);
335 void MainWindow::update_action_buttons(uint64_t t)
337 EventsModel::Status s = events->get_status_at(t);
339 bool has_selection = false;
340 bool has_selection_with_player = false;
342 QItemSelectionModel *select = ui->event_view->selectionModel();
343 if (select->hasSelection()) {
344 has_selection = true;
345 int row = select->selectedRows().front().row(); // Should only be one, due to our selection behavior.
346 has_selection_with_player = events->get_player_id(row).has_value();
348 ui->delete_->setEnabled(has_selection);
351 ui->stoppage->setText("Restart (&v)");
352 ui->stoppage->setShortcut(QCoreApplication::translate("MainWindow", "V", nullptr));
353 ui->catch_->setEnabled(false);
354 ui->throwaway->setEnabled(false);
355 ui->drop->setEnabled(false);
356 ui->goal->setEnabled(false);
357 ui->offensive_soft_plus->setEnabled(false);
358 ui->offensive_soft_minus->setEnabled(false);
359 ui->pull->setEnabled(false);
360 ui->interception->setEnabled(false);
361 ui->their_throwaway->setEnabled(false);
362 ui->our_defense->setEnabled(false);
363 ui->their_goal->setEnabled(false);
364 ui->defensive_soft_plus->setEnabled(false);
365 ui->defensive_soft_minus->setEnabled(false);
366 ui->their_pull->setEnabled(false);
369 ui->stoppage->setText("Stoppage (&v)");
370 ui->stoppage->setShortcut(QCoreApplication::translate("MainWindow", "V", nullptr));
373 // Defaults for pull-related buttons.
374 ui->pull->setText("Pull (&p)");
375 ui->their_pull->setText("Their pull (&p)");
376 ui->pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
377 ui->their_pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
378 ui->throwaway->setText("Throwaway (&t)");
379 ui->throwaway->setShortcut(QCoreApplication::translate("MainWindow", "T", nullptr));
381 if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
382 ui->pull->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
383 ui->their_pull->setEnabled(s.attack_state == EventsModel::Status::OFFENSE);
385 ui->catch_->setEnabled(false);
386 ui->throwaway->setEnabled(false);
387 ui->drop->setEnabled(false);
388 ui->goal->setEnabled(false);
389 ui->offensive_soft_plus->setEnabled(false);
390 ui->offensive_soft_minus->setEnabled(false);
391 ui->interception->setEnabled(false);
392 ui->their_throwaway->setEnabled(false);
393 ui->our_defense->setEnabled(false);
394 ui->their_goal->setEnabled(false);
395 ui->defensive_soft_plus->setEnabled(false);
396 ui->defensive_soft_minus->setEnabled(false);
399 if (s.pull_state == EventsModel::Status::PULL_IN_AIR) {
400 if (s.attack_state == EventsModel::Status::DEFENSE) {
401 ui->pull->setText("Pull landed (&p)");
402 ui->pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
403 ui->pull->setEnabled(true);
405 ui->throwaway->setText("Pull OOB (&t)");
406 ui->throwaway->setShortcut(QCoreApplication::translate("MainWindow", "T", nullptr));
407 ui->throwaway->setEnabled(true);
409 ui->pull->setEnabled(false);
410 ui->throwaway->setEnabled(false);
412 ui->their_pull->setEnabled(false); // We don't track their pull landings; only by means of catch etc.
414 ui->catch_->setEnabled(false);
415 ui->drop->setEnabled(false);
416 ui->goal->setEnabled(false);
417 ui->offensive_soft_plus->setEnabled(false);
418 ui->offensive_soft_minus->setEnabled(false);
419 ui->interception->setEnabled(false);
420 ui->their_throwaway->setEnabled(false);
421 ui->our_defense->setEnabled(false);
422 ui->their_goal->setEnabled(false);
423 ui->defensive_soft_plus->setEnabled(false);
424 ui->defensive_soft_minus->setEnabled(false);
428 ui->catch_->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
429 ui->throwaway->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
430 ui->drop->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
431 ui->goal->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
432 ui->offensive_soft_plus->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
433 ui->offensive_soft_minus->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
434 ui->pull->setEnabled(false);
436 ui->interception->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
437 ui->their_throwaway->setEnabled(s.attack_state == EventsModel::Status::DEFENSE);
438 ui->our_defense->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
439 ui->their_goal->setEnabled(s.attack_state == EventsModel::Status::DEFENSE);
440 ui->defensive_soft_plus->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
441 ui->defensive_soft_minus->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
442 ui->their_pull->setEnabled(false);
445 sqlite3 *open_db(const char *filename)
448 int ret = sqlite3_open(filename, &db);
449 if (ret != SQLITE_OK) {
450 fprintf(stderr, "%s: %s\n", filename, sqlite3_errmsg(db));
455 CREATE TABLE IF NOT EXISTS player (player INTEGER PRIMARY KEY, number VARCHAR, name VARCHAR, gender VARCHAR(1));
456 )", nullptr, nullptr, nullptr); // Ignore errors.
459 CREATE TABLE IF NOT EXISTS match (match INTEGER PRIMARY KEY, description VARCHAR);
460 )", nullptr, nullptr, nullptr); // Ignore errors.
463 CREATE TABLE IF NOT EXISTS event (event INTEGER PRIMARY KEY, match INTEGER, t INTEGER, player INTEGER, type VARCHAR, FOREIGN KEY (player) REFERENCES player(player), FOREIGN KEY (match) REFERENCES match (match));
464 )", nullptr, nullptr, nullptr); // Ignore errors.
466 sqlite3_exec(db, "PRAGMA journal_mode=WAL", nullptr, nullptr, nullptr); // Ignore errors.
467 sqlite3_exec(db, "PRAGMA synchronous=NORMAL", nullptr, nullptr, nullptr); // Ignore errors.
468 sqlite3_exec(db, "PRAGMA foreign_keys=ON", nullptr, nullptr, nullptr); // Ignore errors.
472 int get_match_id(sqlite3 *db, QWidget *parent, int requested_match)
476 bool requested_match_ok = false;
478 // Read the list of matches already in the database.
480 int ret = sqlite3_prepare_v2(db, "SELECT match, description FROM match ORDER BY match", -1, &stmt, 0);
481 if (ret != SQLITE_OK) {
482 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
486 ret = sqlite3_step(stmt);
487 if (ret == SQLITE_ROW) {
489 snprintf(buf, sizeof(buf), "%s (%d)", sqlite3_column_text(stmt, 1), sqlite3_column_int(stmt, 0));
490 ids.push_back(sqlite3_column_int(stmt, 0));
491 if (ids.back() == requested_match) {
492 requested_match_ok = true;
494 items.push_back(buf);
495 } else if (ret == SQLITE_DONE) {
498 fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
502 ret = sqlite3_finalize(stmt);
503 if (ret != SQLITE_OK) {
504 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
507 items.push_back("Add new…");
509 if (requested_match_ok) {
510 return requested_match;
515 QInputDialog dialog(parent, Qt::WindowFlags());
516 dialog.setWindowTitle("Open game");
517 dialog.setLabelText("Choose game to analyze:");
518 dialog.setComboBoxItems(items);
519 dialog.setTextValue(items[items.size() - 2]);
520 dialog.setOption(QInputDialog::UseListViewForComboBoxItems, true);
521 if (!dialog.exec()) {
524 chosen_str = dialog.textValue();
527 for (unsigned i = 0; i < ids.size(); ++i) {
528 if (chosen_str == items[i]) {
533 // Must be a new game. Get its name and insert it into the database.
535 QString new_game_str = QInputDialog::getText(parent, "New game", "Choose name for new game:", QLineEdit::Normal, "", &ok);
536 if (!ok || new_game_str.isEmpty()) {
540 // Insert the new row into the database.
541 ret = sqlite3_prepare_v2(db, "INSERT INTO match (description) VALUES (?)", -1, &stmt, 0);
542 if (ret != SQLITE_OK) {
543 fprintf(stderr, "INSERT prepare: %s\n", sqlite3_errmsg(db));
547 QByteArray new_game_utf8 = new_game_str.toUtf8();
548 sqlite3_bind_text(stmt, 1, (const char *)new_game_utf8.data(), new_game_utf8.size(), SQLITE_STATIC);
550 ret = sqlite3_step(stmt);
551 if (ret == SQLITE_ROW) {
552 fprintf(stderr, "INSERT step: %s\n", sqlite3_errmsg(db));
556 ret = sqlite3_finalize(stmt);
557 if (ret != SQLITE_OK) {
558 fprintf(stderr, "INSERT finalize: %s\n", sqlite3_errmsg(db));
562 return sqlite3_last_insert_rowid(db);
565 int main(int argc, char *argv[])
567 QApplication app(argc, argv);
568 sqlite3 *db = open_db("ultimate.db");
570 // TODO: do this on-demand instead, from a menu
571 export_to_json(db, "ultimate.json");
573 int requested_match = -1;
575 requested_match = atoi(argv[1]);
578 int match_id = get_match_id(db, nullptr, requested_match);
579 if (match_id <= 0) { // Cancel.
583 MainWindow mainWindow(new EventsModel(db, match_id), new PlayersModel(db));
584 mainWindow.resize(QSize(1280, 720));