adding getCurrentStateOrTransition()

Josh Wilson · Josh Wilson · commit f4e55629fa7d · 2010-02-19T02:01:15.000Z
diff --git a/direct/src/fsm/FSM.py b/direct/src/fsm/FSM.py
@@ -204,6 +204,18 @@ def getCurrentOrNextState(self):
         finally:
             self.fsmLock.release()
 
+    def getCurrentStateOrTransition(self):
+        # Returns the current state if we are in a state now, or the
+        # transition we are performing if we are currently within
+        # the enter or exit function for a state.
+        self.fsmLock.acquire()
+        try:
+            if self.state:
+                return self.state
+            return '%s -> %s' % (self.oldState, self.newState)
+        finally:
+            self.fsmLock.release()
+
     def isInTransition(self):
         self.fsmLock.acquire()
         try:
