X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=main.cpp;h=d23c63e8c51d3e15be86ac36e78c433d2e1689c1;hb=67da94217a271b0b76d42c75d1a78ee5ba942825;hp=4383f2e84958bc1dbfc06c6c72f58e9a959cba36;hpb=415442b56355392b207551b2cc84e663d8d7f914;p=pkanalytics

diff --git a/main.cpp b/main.cpp
index 4383f2e..d23c63e 100644
--- a/main.cpp
+++ b/main.cpp
@@ -15,6 +15,7 @@
 #include "ui_mainwindow.h"
 #include "events.h"
 #include "players.h"
+#include "formations.h"
 #include "json.h"
 
 using namespace std;
@@ -33,7 +34,9 @@ string format_timestamp(uint64_t pos)
 	return buf;
 }
 
-MainWindow::MainWindow(EventsModel *events, PlayersModel *players) : events(events), players(players)
+MainWindow::MainWindow(EventsModel *events, PlayersModel *players,
+                       FormationsModel *offensive_formations, FormationsModel *defensive_formations)
+	: events(events), players(players), offensive_formations(offensive_formations), defensive_formations(defensive_formations)
 {
 	video = new QMediaPlayer;
 	//video->setSource(QUrl::fromLocalFile("/home/sesse/dev/stats/ultimate.mkv"));
@@ -62,6 +65,20 @@ MainWindow::MainWindow(EventsModel *events, PlayersModel *players) : events(even
 	ui->player_view->setColumnWidth(1, 20);
 	ui->player_view->horizontalHeader()->setStretchLastSection(true);
 
+	auto formation_changed = [this](const QModelIndex &current, const QModelIndex &previous) {
+		update_action_buttons(video->position());
+	};
+	ui->offensive_formation_view->setModel(offensive_formations);
+	ui->defensive_formation_view->setModel(defensive_formations);
+	connect(ui->offensive_formation_view->selectionModel(), &QItemSelectionModel::currentRowChanged, formation_changed);
+	connect(ui->defensive_formation_view->selectionModel(), &QItemSelectionModel::currentRowChanged, formation_changed);
+	connect(ui->offensive_formation_view, &QListView::doubleClicked, [this](const QModelIndex &index) {
+		formation_double_clicked(true, index.row());
+	});
+	connect(ui->defensive_formation_view, &QListView::doubleClicked, [this](const QModelIndex &index) {
+		formation_double_clicked(false, index.row());
+	});
+
 	connect(video, &QMediaPlayer::positionChanged, [this](uint64_t pos) {
 		position_changed(pos);
 	});
@@ -217,9 +234,9 @@ void MainWindow::insert_player_event(int button_id)
 	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"));
+		ui->event_view->selectRow(events->insert_event(t, player_id, nullopt, "catch"));
 	} else {
-		ui->event_view->selectRow(events->insert_event(t, player_id));
+		ui->event_view->selectRow(events->insert_event(t, player_id, nullopt));
 	}
 	ui->event_view->selectionModel()->blockSignals(false);
 
@@ -231,7 +248,7 @@ void MainWindow::insert_noplayer_event(const string &type)
 	uint64_t t = video->position();
 
 	ui->event_view->selectionModel()->blockSignals(true);
-	ui->event_view->selectRow(events->insert_event(t, nullopt, type));
+	ui->event_view->selectRow(events->insert_event(t, nullopt, nullopt, type));
 	ui->event_view->selectionModel()->blockSignals(false);
 
 	update_ui_from_time(t);
@@ -334,6 +351,25 @@ void MainWindow::update_player_buttons(uint64_t t)
 
 void MainWindow::update_action_buttons(uint64_t t)
 {
+	{
+		QItemSelectionModel *select = ui->offensive_formation_view->selectionModel();
+		if (select->hasSelection()) {
+			int row = select->selectedRows().front().row();  // Should only be one, due to our selection behavior.
+			ui->offensive_formation->setEnabled(offensive_formations->get_formation_id(row) != -1);
+		} else {
+			ui->offensive_formation->setEnabled(false);
+		}
+	}
+	{
+		QItemSelectionModel *select = ui->defensive_formation_view->selectionModel();
+		if (select->hasSelection()) {
+			int row = select->selectedRows().front().row();  // Should only be one, due to our selection behavior.
+			ui->defensive_formation->setEnabled(defensive_formations->get_formation_id(row) != -1);
+		} else {
+			ui->defensive_formation->setEnabled(false);
+		}
+	}
+
 	EventsModel::Status s = events->get_status_at(t);
 
 	bool has_selection = false;
@@ -442,6 +478,23 @@ void MainWindow::update_action_buttons(uint64_t t)
 	ui->their_pull->setEnabled(false);
 }
 
+void MainWindow::formation_double_clicked(bool offense, unsigned row)
+{
+	FormationsModel *formations = offense ? offensive_formations : defensive_formations;
+	int id = formations->get_formation_id(row);
+	if (id == -1) {  // “Add new” clicked.
+		bool ok;
+		QString new_formation_str = QInputDialog::getText(this, "New formation", "Choose name for new formation:", QLineEdit::Normal, "", &ok);
+		if (!ok || new_formation_str.isEmpty()) {
+			return;
+		}
+
+		// FIXME insert
+	} else {
+		events->set_formation_at(video->position(), offense, id);
+	}
+}
+
 sqlite3 *open_db(const char *filename)
 {
 	sqlite3 *db;
@@ -460,7 +513,11 @@ sqlite3 *open_db(const char *filename)
 	)", nullptr, nullptr, nullptr);  // Ignore errors.
 
 	sqlite3_exec(db, R"(
-		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));
+		CREATE TABLE IF NOT EXISTS formation (formation INTEGER PRIMARY KEY, name VARCHAR, offense BOOLEAN NOT NULL);
+	)", nullptr, nullptr, nullptr);  // Ignore errors.
+
+	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.
 
 	sqlite3_exec(db, "PRAGMA journal_mode=WAL", nullptr, nullptr, nullptr);  // Ignore errors.
@@ -580,7 +637,8 @@ int main(int argc, char *argv[])
 		return 0;
 	}
 
-	MainWindow mainWindow(new EventsModel(db, match_id), new PlayersModel(db));
+	MainWindow mainWindow(new EventsModel(db, match_id), new PlayersModel(db),
+	                      new FormationsModel(db, true), new FormationsModel(db, false));
 	mainWindow.resize(QSize(1280, 720));
 	mainWindow.show();