]> git.sesse.net Git - pkanalytics/blobdiff - main.cpp
Support filtering passes by thrower and receiver.
[pkanalytics] / main.cpp
index d9e605894ddee0f5d3a91934ff6df5a308b2cfce..3889023fa6154a51ade93615a9242c261cb282fd 100644 (file)
--- a/main.cpp
+++ b/main.cpp
@@ -1,9 +1,9 @@
-#include <QMediaPlayer>
 #include <QMainWindow>
 #include <QApplication>
 #include <QGridLayout>
-#include <QVideoWidget>
 #include <QShortcut>
+#include <QInputDialog>
+#include <QTimer>
 #include <algorithm>
 #include <string>
 #include <map>
 #include <optional>
 #include <sqlite3.h>
 #include "mainwindow.h"
-#include "ui_mainwindow.h"
 #include "events.h"
 #include "players.h"
+#include "formations.h"
+#include "json.h"
 
 using namespace std;
 
-string format_timestamp(uint64_t pos)
-{
-       int ms = pos % 1000;
-       pos /= 1000;
-       int sec = pos % 60;
-       pos /= 60;
-       int min = pos % 60;
-       int hour = pos / 60;
-
-       char buf[256];
-       snprintf(buf, sizeof(buf), "%d:%02d:%02d.%03d", hour, min, sec, ms);
-       return buf;
-}
-
-MainWindow::MainWindow(EventsModel *events, PlayersModel *players) : events(events), players(players)
-{
-       video = new QMediaPlayer;
-       //video->setSource(QUrl::fromLocalFile("/home/sesse/dev/stats/ultimate.mkv"));
-       video->setSource(QUrl::fromLocalFile("/home/sesse/dev/stats/ultimate-prores.mkv"));
-       video->play();
-
-       ui = new Ui::MainWindow;
-       ui->setupUi(this);
-
-       ui->event_view->setModel(events);
-       connect(ui->event_view->selectionModel(), &QItemSelectionModel::currentRowChanged,
-               [this, events](const QModelIndex &current, const QModelIndex &previous) {
-                       uint64_t t = events->get_time(current.row());
-                       if (t != video->position()) {
-                               video->setPosition(events->get_time(current.row()));
-                       } else {
-                               // Selection could have changed, so we still need to update.
-                               // (Just calling setPosition() would not give us the signal
-                               // in this case.)
-                               update_ui_from_time(t);
-                       }
-               });
-
-       ui->player_view->setModel(players);
-       ui->player_view->setColumnWidth(0, 30);
-       ui->player_view->setColumnWidth(1, 20);
-       ui->player_view->horizontalHeader()->setStretchLastSection(true);
-
-       connect(video, &QMediaPlayer::positionChanged, [this](uint64_t pos) {
-               position_changed(pos);
-       });
-
-       video->setVideoOutput(ui->video);
-
-       connect(ui->minus10s, &QPushButton::clicked, [this]() { seek(-10000); });
-       connect(ui->plus10s, &QPushButton::clicked, [this]() { seek(10000); });
-
-       connect(ui->minus2s, &QPushButton::clicked, [this]() { seek(-2000); });
-       connect(ui->plus2s, &QPushButton::clicked, [this]() { seek(2000); });
-
-       // TODO: Would be nice to actually have a frame...
-       connect(ui->minus1f, &QPushButton::clicked, [this]() { seek(-20); });
-       connect(ui->plus1f, &QPushButton::clicked, [this]() { seek(20); });
-
-       connect(ui->play_pause, &QPushButton::clicked, [this]() {
-               if (playing) {
-                       video->pause();
-                       ui->play_pause->setText("Play (space)");
-               } else {
-                       video->setPlaybackRate(1.0);
-                       video->play();
-                       ui->play_pause->setText("Pause (space)");
-               }
-               playing = !playing;
-
-               // Needs to be set anew when we modify setText(), evidently.
-               ui->play_pause->setShortcut(QCoreApplication::translate("MainWindow", "Space", nullptr));
-       });
-
-       connect(ui->player_1, &QPushButton::clicked, [this]() { insert_event(1); });
-       connect(ui->player_2, &QPushButton::clicked, [this]() { insert_event(2); });
-       connect(ui->player_3, &QPushButton::clicked, [this]() { insert_event(3); });
-       connect(ui->player_4, &QPushButton::clicked, [this]() { insert_event(4); });
-       connect(ui->player_5, &QPushButton::clicked, [this]() { insert_event(5); });
-       connect(ui->player_6, &QPushButton::clicked, [this]() { insert_event(6); });
-       connect(ui->player_7, &QPushButton::clicked, [this]() { insert_event(7); });
-
-       // Offensive events
-       connect(ui->offense_label, &ClickableLabel::clicked, [this]() { insert_noplayer_event("set_offense"); });
-       connect(ui->catch_, &QPushButton::clicked, [this]() { set_current_event_type("catch"); });
-       connect(ui->throwaway, &QPushButton::clicked, [this, events]() {
-               EventsModel::Status s = events->get_status_at(video->position());
-               if (s.attack_state == EventsModel::Status::DEFENSE && s.pull_state == EventsModel::Status::PULL_IN_AIR) {
-                       insert_noplayer_event("pull_oob");
-               } else {
-                       set_current_event_type("throwaway");
-               }
-       });
-       connect(ui->drop, &QPushButton::clicked, [this]() { set_current_event_type("drop"); });
-       connect(ui->goal, &QPushButton::clicked, [this]() { set_current_event_type("goal"); });
-       connect(ui->offensive_soft_plus, &QPushButton::clicked, [this]() { set_current_event_type("offensive_soft_plus"); });
-       connect(ui->offensive_soft_minus, &QPushButton::clicked, [this]() { set_current_event_type("offensive_soft_minus"); });
-       connect(ui->pull, &QPushButton::clicked, [this, events]() {
-               EventsModel::Status s = events->get_status_at(video->position());
-               if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
-                       set_current_event_type("pull");
-               } else if (EventsModel::Status::PULL_IN_AIR) {
-                       insert_noplayer_event("pull_landed");
-               }
-       });
-
-       // Defensive events (TODO add more)
-       connect(ui->defense_label, &ClickableLabel::clicked, [this]() { insert_noplayer_event("set_defense"); });
-       connect(ui->their_throwaway, &QPushButton::clicked, [this]() { insert_noplayer_event("their_throwaway"); });
-       connect(ui->their_goal, &QPushButton::clicked, [this]() { insert_noplayer_event("their_goal"); });
-       connect(ui->their_pull, &QPushButton::clicked, [this, events]() {
-               EventsModel::Status s = events->get_status_at(video->position());
-               if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
-                       insert_noplayer_event("their_pull");
-               }
-       });
-       connect(ui->our_defense, &QPushButton::clicked, [this]() { set_current_event_type("defense"); });
-       connect(ui->defensive_soft_plus, &QPushButton::clicked, [this]() { set_current_event_type("defensive_soft_plus"); });
-       connect(ui->defensive_soft_minus, &QPushButton::clicked, [this]() { set_current_event_type("defensive_soft_minus"); });
-
-       // Misc. events
-       connect(ui->substitution, &QPushButton::clicked, [this]() { make_substitution(); });
-       connect(ui->stoppage, &QPushButton::clicked, [this, events]() {
-               EventsModel::Status s = events->get_status_at(video->position());
-               if (s.stoppage) {
-                       insert_noplayer_event("restart");
-               } else {
-                       insert_noplayer_event("stoppage");
-               }
-       });
-       connect(ui->unknown, &QPushButton::clicked, [this]() { insert_noplayer_event("unknown"); });
-
-       QShortcut *key_delete = new QShortcut(QKeySequence(Qt::Key_Delete), this);
-       connect(key_delete, &QShortcut::activated, [this]() { ui->delete_->animateClick(); });
-       connect(ui->delete_, &QPushButton::clicked, [this]() { delete_current_event(); });
-}
-
-void MainWindow::position_changed(uint64_t pos)
-{
-       ui->timestamp->setText(QString::fromUtf8(format_timestamp(pos)));
-       if (buffered_seek) {
-               video->setPosition(*buffered_seek);
-               buffered_seek.reset();
-       }
-       if (!playing) {
-               video->pause();  // We only played to get a picture.
-       }
-       if (playing) {
-               QModelIndex row = events->get_last_event_qt(video->position());
-               ui->event_view->scrollTo(row, QAbstractItemView::PositionAtCenter);
-       }
-       update_ui_from_time(pos);
-}
-
-void MainWindow::seek(int64_t delta_ms)
-{
-       int64_t current_pos = buffered_seek ? *buffered_seek : video->position();
-       uint64_t pos = max<int64_t>(current_pos + delta_ms, 0);
-       buffered_seek = pos;
-       if (!playing) {
-               video->setPlaybackRate(0.01);
-               video->play();  // Or Qt won't show the seek.
-       }
-}
-
-void MainWindow::insert_event(int button_id)
+sqlite3 *open_db(const char *filename)
 {
-       uint64_t t = video->position();
-       set<int> team = events->get_team_at(t);
-       if (button_id > team.size()) {
-               return;
-       }
-       int player_id = *next(team.begin(), button_id - 1);
-
-       EventsModel::Status s = events->get_status_at(t);
-
-       ui->event_view->selectionModel()->blockSignals(true);
-       if (s.attack_state == EventsModel::Status::OFFENSE) {
-               // TODO: Perhaps not if that player already did the last catch?
-               ui->event_view->selectRow(events->insert_event(t, player_id, "catch"));
-       } else {
-               ui->event_view->selectRow(events->insert_event(t, player_id));
+       sqlite3 *db;
+       int ret = sqlite3_open(filename, &db);
+       if (ret != SQLITE_OK) {
+               fprintf(stderr, "%s: %s\n", filename, sqlite3_errmsg(db));
+               exit(1);
        }
-       ui->event_view->selectionModel()->blockSignals(false);
 
-       update_ui_from_time(t);
-}
+       sqlite3_exec(db, R"(
+               CREATE TABLE IF NOT EXISTS player (player INTEGER PRIMARY KEY, number VARCHAR, name VARCHAR, gender VARCHAR(1));
+       )", nullptr, nullptr, nullptr);  // Ignore errors.
 
-void MainWindow::insert_noplayer_event(const string &type)
-{
-       uint64_t t = video->position();
+       sqlite3_exec(db, R"(
+               CREATE TABLE IF NOT EXISTS match (match INTEGER PRIMARY KEY, description VARCHAR, video_filename VARCHAR, gender_rule_a BOOLEAN DEFAULT false, gender_pull_rule BOOLEAN DEFAULT false);
+       )", nullptr, nullptr, nullptr);  // Ignore errors.
 
-       ui->event_view->selectionModel()->blockSignals(true);
-       ui->event_view->selectRow(events->insert_event(t, nullopt, type));
-       ui->event_view->selectionModel()->blockSignals(false);
+       sqlite3_exec(db, R"(
+               CREATE TABLE IF NOT EXISTS formation (formation INTEGER PRIMARY KEY, name VARCHAR, offense BOOLEAN NOT NULL);
+       )", nullptr, nullptr, nullptr);  // Ignore errors.
 
-       update_ui_from_time(t);
-}
+       sqlite3_exec(db, R"(
+               CREATE TABLE IF NOT EXISTS event (event INTEGER PRIMARY KEY, match INTEGER, t INTEGER, player INTEGER, type VARCHAR, formation INTEGER, FOREIGN KEY (player) REFERENCES player(player), FOREIGN KEY (match) REFERENCES match (match), FOREIGN KEY (formation) REFERENCES formation (formation));
+       )", nullptr, nullptr, nullptr);  // Ignore errors.
 
-void MainWindow::set_current_event_type(const string &type)
-{
-       QItemSelectionModel *select = ui->event_view->selectionModel();
-       if (!select->hasSelection()) {
-               return;
-       }
-       int row = select->selectedRows().front().row();  // Should only be one, due to our selection behavior.
-       events->set_event_type(row, type);
-       update_ui_from_time(video->position());
+       sqlite3_exec(db, "PRAGMA journal_mode=WAL", nullptr, nullptr, nullptr);  // Ignore errors.
+       sqlite3_exec(db, "PRAGMA synchronous=NORMAL", nullptr, nullptr, nullptr);  // Ignore errors.
+       sqlite3_exec(db, "PRAGMA foreign_keys=ON", nullptr, nullptr, nullptr);  // Ignore errors.
+       return db;
 }
-
-void MainWindow::delete_current_event()
+int get_match_id(sqlite3 *db, QWidget *parent, int requested_match)
 {
-       QItemSelectionModel *select = ui->event_view->selectionModel();
-       if (!select->hasSelection()) {
-               return;
-       }
-       int row = select->selectedRows().front().row();  // Should only be one, due to our selection behavior.
-       ui->event_view->selectionModel()->blockSignals(true);
-       events->delete_event(row);
-       ui->event_view->selectionModel()->blockSignals(false);
-       update_ui_from_time(video->position());
-}
+       QStringList items;
+       vector<int> ids;
+       bool requested_match_ok = false;
 
-void MainWindow::make_substitution()
-{
-       QItemSelectionModel *select = ui->player_view->selectionModel();
-       set<int> new_team;
-       for (QModelIndex row : select->selectedRows()) {
-               new_team.insert(players->get_player_id(row.row()));
+       // Read the list of matches already in the database.
+       sqlite3_stmt *stmt;
+       int ret = sqlite3_prepare_v2(db, "SELECT match, description FROM match ORDER BY match", -1, &stmt, 0);
+       if (ret != SQLITE_OK) {
+               fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
+               abort();
+       }
+       for ( ;; ) {
+               ret = sqlite3_step(stmt);
+               if (ret == SQLITE_ROW) {
+                       char buf[256];
+                       snprintf(buf, sizeof(buf), "%s (%d)", sqlite3_column_text(stmt, 1), sqlite3_column_int(stmt, 0));
+                       ids.push_back(sqlite3_column_int(stmt, 0));
+                       if (ids.back() == requested_match) {
+                               requested_match_ok = true;
+                       }
+                       items.push_back(buf);
+               } else if (ret == SQLITE_DONE) {
+                       break;
+               } else {
+                       fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
+                       abort();
+               }
        }
-       events->set_team_at(video->position(), new_team);
-}
-
-void MainWindow::update_ui_from_time(uint64_t t)
-{
-       update_status(t);
-       update_player_buttons(t);
-       update_action_buttons(t);
-}
-
-void MainWindow::update_status(uint64_t t)
-{
-       EventsModel::Status s = events->get_status_at(t);
-       char buf[256];
-       const char *offense = "not started";
-       if (s.attack_state == EventsModel::Status::OFFENSE) {
-               offense = "offense";
-       } else if (s.attack_state == EventsModel::Status::DEFENSE) {
-               offense = "defense";
+       ret = sqlite3_finalize(stmt);
+       if (ret != SQLITE_OK) {
+               fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
+               abort();
        }
+       items.push_back("Add new…");
 
-       snprintf(buf, sizeof(buf), "%d–%d | %s | %d passes, %d sec possession",
-               s.our_score, s.their_score, offense, s.num_passes, s.possession_sec);
-       if (s.stoppage_sec > 0) {
-               char buf2[256];
-               snprintf(buf2, sizeof(buf2), "%s (plus %d sec stoppage)", buf, s.stoppage_sec);
-               ui->status->setText(buf2);
-       } else {
-               ui->status->setText(buf);
+       if (requested_match_ok) {
+               return requested_match;
        }
-}
 
-void MainWindow::update_player_buttons(uint64_t t)
-{
-       // FIXME: sort by number, instead of by internal ID
-       QPushButton *buttons[] = {
-               ui->player_1,
-               ui->player_2,
-               ui->player_3,
-               ui->player_4,
-               ui->player_5,
-               ui->player_6,
-               ui->player_7
-       };
-       const char shortcuts[] = "qweasdf";
-       int num_players = 0;
-       for (int player_id : events->get_team_at(t)) {
-               QPushButton *btn = buttons[num_players];
-               string label = players->get_player_name_by_id(player_id) + " (&" + shortcuts[num_players] + ")";
-               char shortcut[2] = "";
-               shortcut[0] = toupper(shortcuts[num_players]);
-               btn->setText(QString::fromUtf8(label));
-               btn->setShortcut(QCoreApplication::translate("MainWindow", shortcut, nullptr));
-               btn->setEnabled(true);
-               if (++num_players == 7) {
-                       break;
+       QString chosen_str;
+       {
+               QInputDialog dialog(parent, Qt::WindowFlags());
+               dialog.setWindowTitle("Open game");
+               dialog.setLabelText("Choose game to analyze:");
+               dialog.setComboBoxItems(items);
+               if (items.size() >= 2) {
+                       dialog.setTextValue(items[items.size() - 2]);
                }
+               dialog.setOption(QInputDialog::UseListViewForComboBoxItems, true);
+               if (!dialog.exec()) {
+                       return -1;
+               }
+               chosen_str = dialog.textValue();
        }
-       for (int i = num_players; i < 7; ++i) {
-               QPushButton *btn = buttons[i];
-               btn->setText("No player");
-               btn->setEnabled(false);
-       }
-}
-
-void MainWindow::update_action_buttons(uint64_t t)
-{
-       EventsModel::Status s = events->get_status_at(t);
 
-       bool has_selection = false;
-       bool has_selection_with_player = false;
-
-       QItemSelectionModel *select = ui->event_view->selectionModel();
-       if (select->hasSelection()) {
-               has_selection = true;
-               int row = select->selectedRows().front().row();  // Should only be one, due to our selection behavior.
-               has_selection_with_player = events->get_player_id(row).has_value();
+       for (unsigned i = 0; i < ids.size(); ++i) {
+               if (chosen_str == items[i]) {
+                       return ids[i];
+               }
        }
-       ui->delete_->setEnabled(has_selection);
 
-       if (s.stoppage) {
-               ui->stoppage->setText("Restart (&v)");
-               ui->stoppage->setShortcut(QCoreApplication::translate("MainWindow", "V", nullptr));
-               ui->catch_->setEnabled(false);
-               ui->throwaway->setEnabled(false);
-               ui->drop->setEnabled(false);
-               ui->goal->setEnabled(false);
-               ui->offensive_soft_plus->setEnabled(false);
-               ui->offensive_soft_minus->setEnabled(false);
-               ui->pull->setEnabled(false);
-               ui->interception->setEnabled(false);
-               ui->their_throwaway->setEnabled(false);
-               ui->our_defense->setEnabled(false);
-               ui->their_goal->setEnabled(false);
-               ui->defensive_soft_plus->setEnabled(false);
-               ui->defensive_soft_minus->setEnabled(false);
-               ui->their_pull->setEnabled(false);
-               return;
-       } else {
-               ui->stoppage->setText("Stoppage (&v)");
-               ui->stoppage->setShortcut(QCoreApplication::translate("MainWindow", "V", nullptr));
+       // Must be a new game. Get its name and insert it into the database.
+       bool ok;
+       QString new_game_str = QInputDialog::getText(parent, "New game", "Choose name for new game:", QLineEdit::Normal, "", &ok);
+       if (!ok || new_game_str.isEmpty()) {
+               return -1;
        }
 
-       // Defaults for pull-related buttons.
-       ui->pull->setText("Pull (&p)");
-       ui->their_pull->setText("Their pull (&p)");
-       ui->pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
-       ui->their_pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
-       ui->throwaway->setText("Throwaway (&t)");
-       ui->throwaway->setShortcut(QCoreApplication::translate("MainWindow", "T", nullptr));
-
-       if (s.pull_state == EventsModel::Status::SHOULD_PULL) {
-               ui->pull->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
-               ui->their_pull->setEnabled(s.attack_state == EventsModel::Status::OFFENSE);
-
-               ui->catch_->setEnabled(false);
-               ui->throwaway->setEnabled(false);
-               ui->drop->setEnabled(false);
-               ui->goal->setEnabled(false);
-               ui->offensive_soft_plus->setEnabled(false);
-               ui->offensive_soft_minus->setEnabled(false);
-               ui->interception->setEnabled(false);
-               ui->their_throwaway->setEnabled(false);
-               ui->our_defense->setEnabled(false);
-               ui->their_goal->setEnabled(false);
-               ui->defensive_soft_plus->setEnabled(false);
-               ui->defensive_soft_minus->setEnabled(false);
-               return;
+       // Insert the new row into the database.
+       ret = sqlite3_prepare_v2(db, "INSERT INTO match (description, gender_rule_a, gender_pull_rule) VALUES (?, COALESCE((SELECT gender_rule_a FROM match ORDER BY match DESC LIMIT 1),false), COALESCE((SELECT gender_pull_rule FROM match ORDER BY match DESC LIMIT 1),false))", -1, &stmt, 0);
+       if (ret != SQLITE_OK) {
+               fprintf(stderr, "INSERT prepare: %s\n", sqlite3_errmsg(db));
+               abort();
        }
-       if (s.pull_state == EventsModel::Status::PULL_IN_AIR) {
-               if (s.attack_state == EventsModel::Status::DEFENSE) {
-                       ui->pull->setText("Pull landed (&p)");
-                       ui->pull->setShortcut(QCoreApplication::translate("MainWindow", "P", nullptr));
-                       ui->pull->setEnabled(true);
 
-                       ui->throwaway->setText("Pull OOB (&t)");
-                       ui->throwaway->setShortcut(QCoreApplication::translate("MainWindow", "T", nullptr));
-                       ui->throwaway->setEnabled(true);
-               } else {
-                       ui->pull->setEnabled(false);
-                       ui->throwaway->setEnabled(false);
-               }
-               ui->their_pull->setEnabled(false);  // We don't track their pull landings; only by means of catch etc.
+       QByteArray new_game_utf8 = new_game_str.toUtf8();
+       sqlite3_bind_text(stmt, 1, (const char *)new_game_utf8.data(), new_game_utf8.size(), SQLITE_STATIC);
 
-               ui->catch_->setEnabled(false);
-               ui->drop->setEnabled(false);
-               ui->goal->setEnabled(false);
-               ui->offensive_soft_plus->setEnabled(false);
-               ui->offensive_soft_minus->setEnabled(false);
-               ui->interception->setEnabled(false);
-               ui->their_throwaway->setEnabled(false);
-               ui->our_defense->setEnabled(false);
-               ui->their_goal->setEnabled(false);
-               ui->defensive_soft_plus->setEnabled(false);
-               ui->defensive_soft_minus->setEnabled(false);
-               return;
+       ret = sqlite3_step(stmt);
+       if (ret == SQLITE_ROW) {
+               fprintf(stderr, "INSERT step: %s\n", sqlite3_errmsg(db));
+               abort();
        }
 
-       ui->catch_->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->throwaway->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->drop->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->goal->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->offensive_soft_plus->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->offensive_soft_minus->setEnabled(s.attack_state == EventsModel::Status::OFFENSE && has_selection_with_player);
-       ui->pull->setEnabled(false);
-
-       ui->interception->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
-       ui->their_throwaway->setEnabled(s.attack_state == EventsModel::Status::DEFENSE);
-       ui->our_defense->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
-       ui->their_goal->setEnabled(s.attack_state == EventsModel::Status::DEFENSE);
-       ui->defensive_soft_plus->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
-       ui->defensive_soft_minus->setEnabled(s.attack_state == EventsModel::Status::DEFENSE && has_selection_with_player);
-       ui->their_pull->setEnabled(false);
-}
-
-sqlite3 *open_db(const char *filename)
-{
-       sqlite3 *db;
-       int ret = sqlite3_open(filename, &db);
+       ret = sqlite3_finalize(stmt);
        if (ret != SQLITE_OK) {
-               fprintf(stderr, "%s: %s\n", filename, sqlite3_errmsg(db));
-               exit(1);
+               fprintf(stderr, "INSERT finalize: %s\n", sqlite3_errmsg(db));
+               abort();
        }
 
-       sqlite3_exec(db, R"(
-               CREATE TABLE IF NOT EXISTS player (player INTEGER PRIMARY KEY, number VARCHAR, name VARCHAR, gender VARCHAR(1));
-       )", nullptr, nullptr, nullptr);  // Ignore errors.
-
-       sqlite3_exec(db, R"(
-               CREATE TABLE IF NOT EXISTS event (event INTEGER PRIMARY KEY, t INTEGER, player INTEGER, type VARCHAR, FOREIGN KEY (player) REFERENCES player(player));
-       )", nullptr, nullptr, nullptr);  // Ignore errors.
-
-       sqlite3_exec(db, "PRAGMA journal_mode=WAL", nullptr, nullptr, nullptr);  // Ignore errors.
-       sqlite3_exec(db, "PRAGMA synchronous=NORMAL", nullptr, nullptr, nullptr);  // Ignore errors.
-       return db;
+       return sqlite3_last_insert_rowid(db);
 }
 
 int main(int argc, char *argv[])
@@ -449,10 +148,23 @@ int main(int argc, char *argv[])
        QApplication app(argc, argv);
        sqlite3 *db = open_db("ultimate.db");
 
-       MainWindow mainWindow(new EventsModel(db), new PlayersModel(db));
+       int requested_match = -1;
+       if (argc >= 2) {
+               requested_match = atoi(argv[1]);
+       }
+
+       int match_id = get_match_id(db, nullptr, requested_match);
+       if (match_id <= 0) {  // Cancel.
+               return 0;
+       }
+
+       MainWindow mainWindow(new EventsModel(db, match_id), new PlayersModel(db),
+                             new FormationsModel(db, true), new FormationsModel(db, false),
+                             db, match_id);
        mainWindow.resize(QSize(1280, 720));
        mainWindow.show();
 
-       return app.exec();
+       int ret = app.exec();
 
+       return ret;
 }