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 "";  // Gender
  26                 } else if (section == 2) {
  27                         return "Name";
  28                 } else {
  29                         return QVariant();
  30                 }
  31         } else {
  32                 return "";
  33         }
  34 }
  35
  36 QVariant PlayersModel::data(const QModelIndex &index, int role) const
  37 {
  38         if (role != Qt::DisplayRole) {
  39                 return QVariant();
  40         }
  41         if (index.column() == 0) {
  42                 return QString::fromUtf8(players[index.row()].number);
  43         } else if (index.column() == 1) {
  44                 return QString::fromUtf8(players[index.row()].gender);
  45         } else if (index.column() == 2) {
  46                 return QString::fromUtf8(players[index.row()].name);
  47         }
  48         return QVariant();
  49 }
  50
  51 string PlayersModel::get_player_name_by_id(unsigned player_id)
  52 {
  53         auto it = find_if(players.begin(), players.end(), [player_id](const Player &p) { return p.player_id == player_id; });
  54         return it->name;
  55 }
  56
  57 void PlayersModel::load_data()
  58 {
  59         players.clear();
  60
  61         // Read the players.
  62         sqlite3_stmt *stmt;
  63         int ret = sqlite3_prepare_v2(db, "SELECT player, number, name, gender FROM player ORDER BY (number+0), number", -1, &stmt, 0);
  64         if (ret != SQLITE_OK) {
  65                 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
  66                 abort();
  67         }
  68         for ( ;; ) {
  69                 ret = sqlite3_step(stmt);
  70                 if (ret == SQLITE_ROW) {
  71                         Player p;
  72                         p.player_id = sqlite3_column_int(stmt, 0);
  73                         p.number = (const char *)sqlite3_column_text(stmt, 1);
  74                         p.name = (const char *)sqlite3_column_text(stmt, 2);
  75                         p.gender = (const char *)sqlite3_column_text(stmt, 3);
  76                         players.push_back(p);
  77                 } else if (ret == SQLITE_DONE) {
  78                         break;
  79                 } else {
  80                         fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
  81                         abort();
  82                 }
  83         }
  84         ret = sqlite3_finalize(stmt);
  85         if (ret != SQLITE_OK) {
  86                 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
  87                 abort();
  88         }
  89 }