TheCloud/panda3d
1
0
Fork 0
mirror of https://github.com/panda3d/panda3d.git synced 2025-10-03 10:22:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

Applied patch from Nemesis13 partially ;-)

Browse Source
This commit is contained in:
Hubert Grzeskowiak 2009-11-03 18:08:29 +00:00 committed by rdb
parent fcd2a2ec0d
commit 617e9d5e8c
1 changed files with 33 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
direct/src/fsm/FSM.py
View File

@ -1,11 +1,9 @@
"""Undocumented Module""" """The new Finite State Machine module. This replaces the module
previously called FSM.py (now called ClassicFSM.py).
"""
__all__ = ['FSMException', 'FSM'] __all__ = ['FSMException', 'FSM']
"""
The new Finite State Machine module. This replaces the modules
previously called FSM.py (now called ClassicFSM.py).
"""
from direct.showbase.DirectObject import DirectObject from direct.showbase.DirectObject import DirectObject
from direct.directnotify import DirectNotifyGlobal from direct.directnotify import DirectNotifyGlobal
@ -145,6 +143,7 @@ class FSM(DirectObject):
def __init__(self, name): def __init__(self, name):
self.fsmLock = RLock() self.fsmLock = RLock()
self.name = name self.name = name
self.stateArray = []
self._serialNum = FSM.SerialNum self._serialNum = FSM.SerialNum
FSM.SerialNum += 1 FSM.SerialNum += 1
self._broadcastStateChanges = False self._broadcastStateChanges = False
@ -393,33 +392,42 @@ class FSM(DirectObject):
finally: finally:
self.fsmLock.release() self.fsmLock.release()
def requestNext(self, *args): def requestNext(self, *args):
"""request the 'next' state in the predefined state array""" """Request the 'next' state in the predefined state array."""
self.fsmLock.acquire() self.fsmLock.acquire()
try: try:
assert self.state in self.stateArray if self.stateArray:
if not self.state in self.stateArray:
self.request(self.stateArray[0])
else:
cur_index = self.stateArray.index(self.state)
new_index = (cur_index + 1) % len(self.stateArray)
self.request(self.stateArray[new_index], args)
else:
assert self.notifier.debug(
"stateArray empty. Can't switch to next.")
curIndex = self.stateArray.index(self.state)
newIndex = (curIndex + 1) % len(self.stateArray)
self.request(self.stateArray[newIndex], args)
finally: finally:
self.fsmLock.release() self.fsmLock.release()
def requestPrev(self, *args): def requestPrev(self, *args):
"""request the 'previous' state in the predefined state array""" """Request the 'previous' state in the predefined state array."""
self.fsmLock.acquire() self.fsmLock.acquire()
try: try:
assert self.state in self.stateArray if self.stateArray:
if not self.state in self.stateArray:
curIndex = self.stateArray.index(self.state) self.request(self.stateArray[0])
newIndex = (curIndex - 1) % len(self.stateArray) else:
cur_index = self.stateArray.index(self.state)
self.request(self.stateArray[newIndex], args) new_index = (cur_index - 1) % len(self.stateArray)
self.request(self.stateArray[new_index], args)
else:
assert self.notifier.debug(
"stateArray empty. Can't switch to next.")
finally: finally:
self.fsmLock.release() self.fsmLock.release()
def __setState(self, newState, *args): def __setState(self, newState, *args):
# Internal function to change unconditionally to the indicated # Internal function to change unconditionally to the indicated
# state. # state.