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();
}
players.clear();
events.clear();
- // Read the players.
+ // Read the players. (The ordering is used to build the order map.)
sqlite3_stmt *stmt;
- int ret = sqlite3_prepare_v2(db, "SELECT player, number, name FROM player", -1, &stmt, 0);
+ int ret = sqlite3_prepare_v2(db, "SELECT player, number, name FROM player ORDER BY gender, (number+0), number", -1, &stmt, 0);
if (ret != SQLITE_OK) {
fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
abort();
}
+ int order = 0;
for ( ;; ) {
ret = sqlite3_step(stmt);
if (ret == SQLITE_ROW) {
p.number = (const char *)sqlite3_column_text(stmt, 1);
p.name = (const char *) sqlite3_column_text(stmt, 2);
players[p.player_id] = std::move(p);
+ player_ordering[p.player_id] = order++;
} else if (ret == SQLITE_DONE) {
break;
} else {
}
// 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) {
// 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) {
}
}
+unsigned EventsModel::get_last_event_pos(uint64_t t) const
+{
+ // upper_bound() gives first where e.t > t,
+ // and the one before that is the one we want.
+ auto it = upper_bound(events.begin(), events.end(), t,
+ [](uint64_t t, const Event &e) { return t < e.t; });
+ if (it == events.begin()) {
+ return 0;
+ } else {
+ return distance(events.begin(), it - 1);
+ }
+}
+
EventsModel::Status EventsModel::get_status_at(uint64_t t)
{
Status s;
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)) {
}
}
}
+
+vector<int> EventsModel::sort_team(const set<int> &team) const
+{
+ vector<int> ret(team.begin(), team.end());
+ std::sort(ret.begin(), ret.end(), [this](int a, int b) {
+ return player_ordering.find(a)->second < player_ordering.find(b)->second;
+ });
+ return ret;
+}