Make the player buttons actually fetch out the team.

[pkanalytics] / events.cpp

diff --git a/events.cpp b/events.cpp
index 5752d82ca4fb5c8d7cfed40ec705e1d17f698258..a2d5e07b82df5743f2bd13f5f11266de322f2b9c 100644 (file)
--- a/events.cpp
+++ b/events.cpp
@@ -10,6 +10,11 @@ using namespace std;
 
 string format_timestamp(uint64_t pos);
 
+EventsModel::EventsModel(sqlite3 *db) : db(db)
+{
+       load_data();
+}
+
 QVariant EventsModel::headerData(int section, Qt::Orientation orientation, int role) const
 {
        if (role != Qt::DisplayRole) {
@@ -33,13 +38,13 @@ QVariant EventsModel::data(const QModelIndex &index, int role) const
        if (role != Qt::DisplayRole) {
                return QVariant();
        }
-       refresh_if_needed();
        if (index.column() == 0) {
                return QString::fromUtf8(format_timestamp(events[index.row()].t));
        } else if (index.column() == 1) {
                optional<int> player_id = events[index.row()].player_id;
                if (player_id) {
-                       const Player &p = players[*player_id];
+                       auto p_it = players.find(*player_id);
+                       const Player &p = p_it->second;
                        return QString::fromUtf8(p.name + " (" + p.number + ")");
                } else {
                        return QVariant();
@@ -50,15 +55,10 @@ QVariant EventsModel::data(const QModelIndex &index, int role) const
        return QVariant();
 }
 
-void EventsModel::refresh_if_needed() const
+void EventsModel::load_data()
 {
-       if (!stale) {
-               return;
-       }
-
        players.clear();
        events.clear();
-       stale = false;
 
        // Read the players.
        sqlite3_stmt *stmt;
@@ -115,12 +115,9 @@ void EventsModel::refresh_if_needed() const
                fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
                abort();
        }
-
-       // TODO what if data changes externally?
-       //emit dataChanged(QModelIndex(
 }
 
-unsigned EventsModel::insert_event(uint64_t t, int player_id)
+unsigned EventsModel::insert_event(uint64_t t, int player_id, const string &type)
 {
        auto it = lower_bound(events.begin(), events.end(), t,
                [](const Event &e, uint64_t t) { return e.t < t; });
@@ -130,7 +127,7 @@ unsigned EventsModel::insert_event(uint64_t t, int player_id)
        Event e;
        e.t = t;
        e.player_id = player_id;
-       e.type = "unknown";
+       e.type = type;
        events.insert(events.begin() + pos, e);
 
        endInsertRows();
@@ -145,7 +142,7 @@ unsigned EventsModel::insert_event(uint64_t t, int player_id)
 
        sqlite3_bind_int64(stmt, 1, t);
        sqlite3_bind_int64(stmt, 2, player_id);
-       sqlite3_bind_text(stmt, 3, e.type.data(), e.type.size(), SQLITE_STATIC);
+       sqlite3_bind_text(stmt, 3, type.data(), type.size(), SQLITE_STATIC);
 
        ret = sqlite3_step(stmt);
        if (ret == SQLITE_ROW) {
@@ -260,3 +257,20 @@ EventsModel::Status EventsModel::get_status_at(uint64_t t)
        s.possession_sec = (s.offense && last_gained_possession != 0) ? (t - last_gained_possession) / 1000 : 0;
        return s;
 }
+
+set<int> EventsModel::get_team_at(uint64_t t)
+{
+       set<int> team;
+       for (const Event &e : events) {
+               if (e.t > t) {
+                       break;
+               }
+               if (e.type == "in") {
+                       team.insert(*e.player_id);
+               }
+               if (e.type == "out") {
+                       team.erase(*e.player_id);
+               }
+       }
+       return team;
+}