Update the timed transitions start time on every state change

Author: corna

Fsm.cpp

@@ -66,7 +66,7 @@ void Fsm::add_timed_transition(State* state_from, State* state_to,
 
   TimedTransition timed_transition;
   timed_transition.transition = transition;
-  timed_transition.start = 0;
+  timed_transition.start = millis();
   timed_transition.interval = interval;
 
   m_timed_transitions = (TimedTransition*) realloc(
@@ -97,6 +97,7 @@ void Fsm::trigger(int event)
         m_transitions[i].event == event)
     {
       m_current_state = m_transitions[i].make_transition();
+      update_timed_transitions_starts();
       return;
     }
   }
@@ -110,17 +111,10 @@ void Fsm::check_timer()
     TimedTransition* transition = &m_timed_transitions[i];
     if (transition->transition.state_from == m_current_state)
     {
-      if (transition->start == 0)
+      if (millis() - transition->start >= transition->interval)
       {
-        transition->start = millis();
-      }
-      else
-      {
-        if (millis() - transition->start >= transition->interval)
-        {
-          m_current_state = transition->transition.make_transition();
-          transition->start = 0;
-        }
+        m_current_state = transition->transition.make_transition();
+        update_timed_transitions_starts();
       }
     }
   }
@@ -141,3 +135,14 @@ State* Fsm::Transition::make_transition()
 
   return state_to;
 }
+
+void Fsm::update_timed_transitions_starts()
+{
+  for (int i = 0; i < m_num_timed_transitions; ++i)
+  {
+    TimedTransition* transition = &m_timed_transitions[i];
+
+    if (transition->transition.state_from == m_current_state)
+      transition->start = millis();
+  }
+}

Fsm.h

@@ -67,6 +67,7 @@ class Fsm
 
   static Transition create_transition(State* state_from, State* state_to,
                                       int event, void (*on_transition)());
+  void update_timed_transitions_starts();
 
 private:
   State* m_current_state;

LICENSE

@@ -1,5 +1,6 @@
 The MIT License (MIT)
 
+Copyright (c) 2016 Nicola Corna <nicola@corna.info>
 Copyright (c) 2015 Jon Black
 
 Permission is hereby granted, free of charge, to any person obtaining a copy