11 string format_timestamp(uint64_t pos);
13 EventsModel::EventsModel(sqlite3 *db) : db(db)
18 QVariant EventsModel::headerData(int section, Qt::Orientation orientation, int role) const
20 if (role != Qt::DisplayRole) {
23 if (orientation == Qt::Horizontal) {
26 } else if (section == 1) {
36 QVariant EventsModel::data(const QModelIndex &index, int role) const
38 if (role != Qt::DisplayRole) {
41 if (index.column() == 0) {
42 return QString::fromUtf8(format_timestamp(events[index.row()].t));
43 } else if (index.column() == 1) {
44 optional<int> player_id = events[index.row()].player_id;
46 auto p_it = players.find(*player_id);
47 const Player &p = p_it->second;
48 return QString::fromUtf8(p.name + " (" + p.number + ")");
52 } else if (index.column() == 2) {
53 return QString::fromUtf8(events[index.row()].type);
58 void EventsModel::load_data()
65 int ret = sqlite3_prepare_v2(db, "SELECT player, number, name FROM player", -1, &stmt, 0);
66 if (ret != SQLITE_OK) {
67 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
71 ret = sqlite3_step(stmt);
72 if (ret == SQLITE_ROW) {
74 p.player_id = sqlite3_column_int(stmt, 0);
75 p.number = (const char *)sqlite3_column_text(stmt, 1);
76 p.name = (const char *) sqlite3_column_text(stmt, 2);
77 players[p.player_id] = std::move(p);
78 } else if (ret == SQLITE_DONE) {
81 fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
85 ret = sqlite3_finalize(stmt);
86 if (ret != SQLITE_OK) {
87 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
92 ret = sqlite3_prepare_v2(db, "SELECT event, t, player, type FROM event ORDER BY t", -1, &stmt, 0);
93 if (ret != SQLITE_OK) {
94 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
98 ret = sqlite3_step(stmt);
99 if (ret == SQLITE_ROW) {
101 e.event_id = sqlite3_column_int(stmt, 0);
102 e.t = sqlite3_column_int(stmt, 1);
103 e.player_id = sqlite3_column_int(stmt, 2);
104 e.type = (const char *)sqlite3_column_text(stmt, 3);
105 events.push_back(std::move(e));
106 } else if (ret == SQLITE_DONE) {
109 fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
113 ret = sqlite3_finalize(stmt);
114 if (ret != SQLITE_OK) {
115 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
120 unsigned EventsModel::insert_event(uint64_t t, optional<int> player_id, const string &type)
122 auto it = lower_bound(events.begin(), events.end(), t,
123 [](const Event &e, uint64_t t) { return e.t < t; });
124 unsigned pos = distance(events.begin(), it);
125 beginInsertRows(QModelIndex(), pos, pos);
129 e.player_id = player_id;
131 events.insert(events.begin() + pos, e);
135 // Insert the new row into the database.
137 int ret = sqlite3_prepare_v2(db, "INSERT INTO event (t, player, type) VALUES (?, ?, ?)", -1, &stmt, 0);
138 if (ret != SQLITE_OK) {
139 fprintf(stderr, "INSERT prepare: %s\n", sqlite3_errmsg(db));
143 sqlite3_bind_int64(stmt, 1, t);
145 sqlite3_bind_int64(stmt, 2, *player_id);
147 sqlite3_bind_null(stmt, 2);
149 sqlite3_bind_text(stmt, 3, type.data(), type.size(), SQLITE_STATIC);
151 ret = sqlite3_step(stmt);
152 if (ret == SQLITE_ROW) {
153 fprintf(stderr, "INSERT step: %s\n", sqlite3_errmsg(db));
157 ret = sqlite3_finalize(stmt);
158 if (ret != SQLITE_OK) {
159 fprintf(stderr, "INSERT finalize: %s\n", sqlite3_errmsg(db));
163 events[pos].event_id = sqlite3_last_insert_rowid(db);
167 void EventsModel::delete_event(unsigned pos)
169 int event_id = events[pos].event_id;
171 beginRemoveRows(QModelIndex(), pos, pos);
172 events.erase(events.begin() + pos);
175 // Delete the row from the database.
177 int ret = sqlite3_prepare_v2(db, "DELETE FROM event WHERE event=?", -1, &stmt, 0);
178 if (ret != SQLITE_OK) {
179 fprintf(stderr, "DELETE prepare: %s\n", sqlite3_errmsg(db));
183 sqlite3_bind_int64(stmt, 1, event_id);
185 ret = sqlite3_step(stmt);
186 if (ret == SQLITE_ROW) {
187 fprintf(stderr, "DELETE step: %s\n", sqlite3_errmsg(db));
191 ret = sqlite3_finalize(stmt);
192 if (ret != SQLITE_OK) {
193 fprintf(stderr, "DELETE finalize: %s\n", sqlite3_errmsg(db));
198 void EventsModel::set_event_type(unsigned pos, const string &type)
200 events[pos].type = type;
201 emit dataChanged(createIndex(pos, 0), createIndex(pos, 2));
204 int ret = sqlite3_prepare_v2(db, "UPDATE event SET type=? WHERE event=?", -1, &stmt, 0);
205 if (ret != SQLITE_OK) {
206 fprintf(stderr, "INSERT prepare: %s\n", sqlite3_errmsg(db));
210 sqlite3_bind_text(stmt, 1, type.data(), type.size(), SQLITE_STATIC);
211 sqlite3_bind_int64(stmt, 2, events[pos].event_id);
213 ret = sqlite3_step(stmt);
214 if (ret == SQLITE_ROW) {
215 fprintf(stderr, "UPDATE step: %s\n", sqlite3_errmsg(db));
219 ret = sqlite3_finalize(stmt);
220 if (ret != SQLITE_OK) {
221 fprintf(stderr, "UPDATE finalize: %s\n", sqlite3_errmsg(db));
226 EventsModel::Status EventsModel::get_status_at(uint64_t t)
232 uint64_t last_gained_possession = 0;
233 unsigned num_touches = 0;
234 for (const Event &e : events) {
238 if (e.type == "goal") {
243 if (e.type == "their_goal") {
248 if (e.type == "catch") {
249 if (num_touches == 0) { // Pick up.
250 last_gained_possession = e.t;
254 if (e.type == "interception") {
257 last_gained_possession = e.t;
260 s.num_passes = (num_touches == 0) ? 0 : num_touches - 1;
261 s.possession_sec = (s.offense && last_gained_possession != 0) ? (t - last_gained_possession) / 1000 : 0;
265 set<int> EventsModel::get_team_at(uint64_t t)
268 for (const Event &e : events) {
272 if (e.type == "in") {
273 team.insert(*e.player_id);
275 if (e.type == "out") {
276 team.erase(*e.player_id);