Implement filter-by-player-on-field.

[pkanalytics] / events.cpp

diff --git a/events.cpp b/events.cpp
index cb552c571bcd7cba8bf42b8540ff404c1568802b..e6ed5796340ff3521ab051512f25920d1d04c022 100644 (file)
--- a/events.cpp
+++ b/events.cpp
@@ -10,7 +10,7 @@ using namespace std;
 
 string format_timestamp(uint64_t pos);
 
-EventsModel::EventsModel(sqlite3 *db) : db(db)
+EventsModel::EventsModel(sqlite3 *db, int match_id) : db(db), match_id(match_id)
 {
        load_data();
 }
@@ -91,11 +91,12 @@ void EventsModel::load_data()
        }
 
        // Read the events.
-       ret = sqlite3_prepare_v2(db, "SELECT event, t, player, type FROM event ORDER BY t", -1, &stmt, 0);
+       ret = sqlite3_prepare_v2(db, "SELECT event, t, player, type FROM event WHERE match=? ORDER BY t", -1, &stmt, 0);
        if (ret != SQLITE_OK) {
                fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
                abort();
        }
+       sqlite3_bind_int64(stmt, 1, match_id);
        for ( ;; ) {
                ret = sqlite3_step(stmt);
                if (ret == SQLITE_ROW) {
@@ -138,19 +139,20 @@ unsigned EventsModel::insert_event(uint64_t t, optional<int> player_id, const st
 
        // Insert the new row into the database.
        sqlite3_stmt *stmt;
-       int ret = sqlite3_prepare_v2(db, "INSERT INTO event (t, player, type) VALUES (?, ?, ?)", -1, &stmt, 0);
+       int ret = sqlite3_prepare_v2(db, "INSERT INTO event (match, t, player, type) VALUES (?, ?, ?, ?)", -1, &stmt, 0);
        if (ret != SQLITE_OK) {
                fprintf(stderr, "INSERT prepare: %s\n", sqlite3_errmsg(db));
                abort();
        }
 
-       sqlite3_bind_int64(stmt, 1, t);
+       sqlite3_bind_int64(stmt, 1, match_id);
+       sqlite3_bind_int64(stmt, 2, t);
        if (player_id) {
-               sqlite3_bind_int64(stmt, 2, *player_id);
+               sqlite3_bind_int64(stmt, 3, *player_id);
        } else {
-               sqlite3_bind_null(stmt, 2);
+               sqlite3_bind_null(stmt, 3);
        }
-       sqlite3_bind_text(stmt, 3, type.data(), type.size(), SQLITE_STATIC);
+       sqlite3_bind_text(stmt, 4, type.data(), type.size(), SQLITE_STATIC);
 
        ret = sqlite3_step(stmt);
        if (ret == SQLITE_ROW) {
@@ -359,7 +361,7 @@ void EventsModel::set_team_at(uint64_t t, const set<int> &new_team)
                if (e.t > t) {
                        break;
                }
-               if (e.type == "goal" || e.type == "their_goal" || e.type == "stoppage" || e.type == "reset") {
+               if (e.type == "goal" || e.type == "their_goal" || e.type == "stoppage" || e.type == "reset" || e.type == "set_offense" || e.type == "set_defense") {
                        backdate_point = e.t + 1;
                }
                if (e.player_id.has_value() && !new_team.count(*e.player_id)) {