]> git.sesse.net Git - pkanalytics/blobdiff - events.h
Drop QVideoWidget.
[pkanalytics] / events.h
index c1255578a50710de618e1b61ecdf6e938e54a4cd..df5fcb6a9b8c8b08871f2d76ef7804a5501652f7 100644 (file)
--- a/events.h
+++ b/events.h
@@ -12,7 +12,7 @@
 class EventsModel : public QAbstractTableModel
 {
 public:
-       EventsModel(sqlite3 *db);
+       EventsModel(sqlite3 *db, int match_id);
 
        int rowCount(const QModelIndex &parent) const override
        {
@@ -25,23 +25,38 @@ public:
        QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
        QVariant data(const QModelIndex &index, int role) const override;
 
-       unsigned insert_event(uint64_t t, std::optional<int> player_id, const std::string &type = "unknown");  // Returns the row.
+       unsigned insert_event(uint64_t t, std::optional<int> player_id, std::optional<int> formation_id, const std::string &type = "unknown");  // Returns the row.
        void delete_event(unsigned row);
+       std::string get_event_type(unsigned row) { return events[row].type; }
        void set_event_type(unsigned row, const std::string &type);
+       void set_event_formation(unsigned row, int formation_id);
        uint64_t get_time(unsigned row) { return events[row].t; }
+       unsigned get_last_event_pos(uint64_t t) const;  // Last event that happened at or before t.
+       QModelIndex get_last_event_qt(uint64_t t) const {
+               return createIndex(get_last_event_pos(t), 0);
+       }
        std::optional<int> get_player_id(unsigned row) { return events[row].player_id; }
 
        struct Status {
                unsigned our_score, their_score;
-               bool offense;
+               enum { NOT_STARTED, OFFENSE, DEFENSE } attack_state;
+               unsigned offensive_formation, defensive_formation;
                bool stoppage;
-               bool should_pull;
+               enum { NOT_PULLING, SHOULD_PULL, PULL_IN_AIR } pull_state;
                unsigned num_passes;
                unsigned possession_sec;
                unsigned stoppage_sec;
        };
        Status get_status_at(uint64_t t);
        std::set<int> get_team_at(uint64_t t);
+       void set_team_at(uint64_t, const std::set<int> &new_team);
+       std::vector<int> sort_team(const std::set<int> &team) const;  // Ordered first by gender, then by number.
+       void set_formation_at(uint64_t t, bool offense, unsigned formation);
+
+       // Used to notify when we've inserted a new one into the database.
+       void inserted_new_formation(int formation_id, const std::string &name) {
+               formations[formation_id] = Formation{ formation_id, name };
+       }
 
 private:
        struct Player {
@@ -50,16 +65,25 @@ private:
                std::string name;
        };
        std::map<int, Player> players;
+       std::map<int, int> player_ordering;  // From id to position.
+
+       struct Formation {
+               int formation_id;
+               std::string name;
+       };
+       std::map<int, Formation> formations;
 
        struct Event {
                int event_id;
                uint64_t t;
                std::optional<int> player_id;
+               std::optional<int> formation_id;
                std::string type;
        };
        std::vector<Event> events;
 
        sqlite3 *db;
+       int match_id;
 
        void load_data();
 };