X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=main.cpp;h=3889023fa6154a51ade93615a9242c261cb282fd;hb=HEAD;hp=611e6c1a4579918eccb373a9840e9de7a479b206;hpb=6143142615bf1d908413109852e9834c8b857c66;p=pkanalytics

diff --git a/main.cpp b/main.cpp
index 611e6c1..3889023 100644
--- 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>
@@ -11,434 +11,136 @@
 #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);
-
-	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);
-	)", 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[])
@@ -446,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;
 }