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

   1 #ifndef _PLAYERS_H
   2 #define _PLAYERS_H 1
   3
   4 #include <sqlite3.h>
   5 #include <QAbstractTableModel>
   6 #include <string>
   7 #include <vector>
   8
   9 class PlayersModel : public QAbstractTableModel
  10 {
  11 public:
  12         PlayersModel(sqlite3 *db);
  13
  14         int rowCount(const QModelIndex &parent) const override
  15         {
  16                 return players.size();
  17         }
  18         int columnCount(const QModelIndex &column) const override
  19         {
  20                 return 3;
  21         }
  22         QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
  23         QVariant data(const QModelIndex &index, int role) const override;
  24
  25         int get_player_id(unsigned row) const {
  26                 return players[row].player_id;
  27         }
  28         std::string get_player_gender(unsigned row) const {
  29                 return players[row].gender;
  30         }
  31         std::string get_player_name_by_id(unsigned player_id);
  32         std::string get_player_gender_by_id(unsigned player_id);
  33         QModelIndex get_row_start_qt(unsigned row) const {
  34                 return createIndex(row, 0);
  35         }
  36         QModelIndex get_row_end_qt(unsigned row) const {
  37                 return createIndex(row, 2);
  38         }
  39
  40 private:
  41         struct Player {
  42                 int player_id;
  43                 std::string number;
  44                 std::string name;
  45                 std::string gender;
  46         };
  47         std::vector<Player> players;
  48
  49         sqlite3 *db;
  50
  51         void load_data();
  52 };
  53
  54 #endif  // !defined(_PLAYERS_H)