Fix pull button availability.

[pkanalytics] / events.cpp

diff --git a/events.cpp b/events.cpp
index a87d83c8d60c7c7efb3a66c05b0fa89f48ddd969..bf16202289f650913f49a1f1470149193966f33f 100644 (file)
--- a/events.cpp
+++ b/events.cpp
@@ -232,6 +232,7 @@ EventsModel::Status EventsModel::get_status_at(uint64_t t)
        s.their_score = 0;
        s.offense = true;
        s.stoppage = false;
+       s.should_pull = true;
        uint64_t last_gained_possession = 0;
        uint64_t last_stoppage = 0;
        uint64_t time_spent_in_stoppage = 0;
@@ -240,6 +241,15 @@ EventsModel::Status EventsModel::get_status_at(uint64_t t)
                if (e.t > t) {
                        break;
                }
+
+               if (e.type == "goal" || e.type == "their_goal") {
+                       s.should_pull = true;
+               } else if (e.type == "in" || e.type == "out" || e.type == "stoppage" || e.type == "restart" || e.type == "unknown") {
+                       // No effect on pull status.
+               } else {
+                       s.should_pull = false;
+               }
+
                if (e.type == "goal") {
                        ++s.our_score;
                        s.offense = false;