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::TextAlignmentRole) {
  39                 if (index.column() == 0) {
  40                         return (Qt::AlignHCenter | Qt::AlignVCenter).toInt();
  41                 } else if (index.column() == 1) {
  42                         return (Qt::AlignHCenter | Qt::AlignVCenter).toInt();
  43                 } else {
  44                         return (Qt::AlignLeft | Qt::AlignVCenter).toInt();
  45                 }
  46         }
  47         if (role != Qt::DisplayRole) {
  48                 return QVariant();
  49         }
  50         if (index.column() == 0) {
  51                 return QString::fromUtf8(players[index.row()].number);
  52         } else if (index.column() == 1) {
  53                 return QString::fromUtf8(players[index.row()].gender);
  54         } else if (index.column() == 2) {
  55                 return QString::fromUtf8(players[index.row()].name);
  56         }
  57         return QVariant();
  58 }
  59
  60 string PlayersModel::get_player_name_by_id(unsigned player_id)
  61 {
  62         auto it = find_if(players.begin(), players.end(), [player_id](const Player &p) { return p.player_id == int(player_id); });
  63         return it->name;
  64 }
  65
  66 string PlayersModel::get_player_gender_by_id(unsigned player_id)
  67 {
  68         auto it = find_if(players.begin(), players.end(), [player_id](const Player &p) { return p.player_id == int(player_id); });
  69         return it->gender;
  70 }
  71
  72 void PlayersModel::load_data()
  73 {
  74         players.clear();
  75
  76         // Read the players.
  77         sqlite3_stmt *stmt;
  78         int ret = sqlite3_prepare_v2(db, "SELECT player, number, name, gender FROM player ORDER BY (number+0), number", -1, &stmt, 0);
  79         if (ret != SQLITE_OK) {
  80                 fprintf(stderr, "SELECT prepare: %s\n", sqlite3_errmsg(db));
  81                 abort();
  82         }
  83         for ( ;; ) {
  84                 ret = sqlite3_step(stmt);
  85                 if (ret == SQLITE_ROW) {
  86                         Player p;
  87                         p.player_id = sqlite3_column_int(stmt, 0);
  88                         p.number = (const char *)sqlite3_column_text(stmt, 1);
  89                         p.name = (const char *)sqlite3_column_text(stmt, 2);
  90                         p.gender = (const char *)sqlite3_column_text(stmt, 3);
  91                         players.push_back(p);
  92                 } else if (ret == SQLITE_DONE) {
  93                         break;
  94                 } else {
  95                         fprintf(stderr, "SELECT step: %s\n", sqlite3_errmsg(db));
  96                         abort();
  97                 }
  98         }
  99         ret = sqlite3_finalize(stmt);
 100         if (ret != SQLITE_OK) {
 101                 fprintf(stderr, "SELECT finalize: %s\n", sqlite3_errmsg(db));
 102                 abort();
 103         }
 104 }