git.sesse.net Git - pkanalytics/blob - players.cpp

   1 #include <algorithm>
   2 #include <string>
   3 #include <vector>
   4 #include <sqlite3.h>
   5 #include "players.h"
   6
   7 using namespace std;
   8
   9 string format_timestamp(uint64_t pos);
  10
  11 PlayersModel::PlayersModel(sqlite3 *db) : db(db)
  12 {
  13         load_data();
  14 }
  15
  16 QVariant PlayersModel::headerData(int section, Qt::Orientation orientation, int role) const
  17 {
  18         if (role != Qt::DisplayRole) {
  19                 return QVariant();
  20         }
  21         if (orientation == Qt::Horizontal) {
  22                 if (section == 0) {
  23                         return "#";
  24                 } else if (section == 1) {
  25                         return "Name";
  26                 } else {
  27                         return QVariant();
  28                 }
  29         } else {
  30                 return "";
  31         }
  32 }
  33
  34 QVariant PlayersModel::data(const QModelIndex &index, int role) const
  35 {
  36         if (role != Qt::DisplayRole) {
  37                 return QVariant();
  38         }
  39         if (index.column() == 0) {
  40                 return QString::fromUtf8(players[index.row()].number);
  41         } else if (index.column() == 1) {
  42                 return QString::fromUtf8(players[index.row()].name);
  43         }
  44         return QVariant();
  45 }
  46
  47 string PlayersModel::get_player_name_by_id(unsigned player_id)
  48 {
  49         auto it = find_if(players.begin(), players.end(), [player_id](const Player &p) { return p.player_id == player_id; });
  50         return it->name;
  51 }
  52
  53 void PlayersModel::load_data()
  54 {
  55         players.clear();
  56
  57         // Read the players.
  58         sqlite3_stmt *stmt;
  59         int ret = sqlite3_prepare_v2(db, "SELECT player, number, name FROM player ORDER BY (number+0), number", -1, &stmt, 0);
  60         if (ret != SQLITE_OK) {
  61                 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
  62                 abort();
  63         }
  64         for ( ;; ) {
  65                 ret = sqlite3_step(stmt);
  66                 if (ret == SQLITE_ROW) {
  67                         Player p;
  68                         p.player_id = sqlite3_column_int(stmt, 0);
  69                         p.number = (const char *)sqlite3_column_text(stmt, 1);
  70                         p.name = (const char *) sqlite3_column_text(stmt, 2);
  71                         players.push_back(p);
  72                 } else if (ret == SQLITE_DONE) {
  73                         break;
  74                 } else {
  75                         fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
  76                         abort();
  77                 }
  78         }
  79         ret = sqlite3_finalize(stmt);
  80         if (ret != SQLITE_OK) {
  81                 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
  82                 abort();
  83         }
  84 }