Source code for direct.fsm.StatePush

# classes for event-driven programming
# http://en.wikipedia.org/wiki/Event-driven_programming

__all__ = ['StateVar', 'FunctionCall', 'EnterExit', 'Pulse', 'EventPulse',
           'EventArgument', ]

from direct.showbase.DirectObject import DirectObject


[docs]class PushesStateChanges: # base class for objects that broadcast state changes to a set of subscriber objects
[docs] def __init__(self, value): self._value = value # push state changes to these objects self._subscribers = set()
[docs] def destroy(self): if len(self._subscribers) != 0: raise Exception('%s object still has subscribers in destroy(): %s' % ( self.__class__.__name__, self._subscribers)) del self._subscribers del self._value
[docs] def getState(self): return self._value
[docs] def pushCurrentState(self): self._handleStateChange() return self
def _addSubscription(self, subscriber): self._subscribers.add(subscriber) subscriber._recvStatePush(self) def _removeSubscription(self, subscriber): self._subscribers.remove(subscriber) def _handlePotentialStateChange(self, value): oldValue = self._value self._value = value if oldValue != value: self._handleStateChange() def _handleStateChange(self): # push this object's state to the subscribing objects for subscriber in self._subscribers: subscriber._recvStatePush(self)
if __debug__: psc = PushesStateChanges(0) assert psc.getState() == 0 psc.destroy() del psc
[docs]class ReceivesStateChanges: # base class for objects that subscribe to state changes from PushesStateChanges objects
[docs] def __init__(self, source): self._source = None self._initSource = source
def _finishInit(self): # initialization is split across two functions to allow objects that derive from this # class to set everything up so that they can respond appropriately to the initial # state push from the state source self._subscribeTo(self._initSource) del self._initSource
[docs] def destroy(self): self._unsubscribe() del self._source
def _subscribeTo(self, source): self._unsubscribe() self._source = source if self._source: self._source._addSubscription(self) def _unsubscribe(self): if self._source: self._source._removeSubscription(self) self._source = None def _recvStatePush(self, source): pass
if __debug__: rsc = ReceivesStateChanges(None) rsc.destroy() del rsc
[docs]class StateVar(PushesStateChanges): # coder-friendly object that allows values to be set on it and pushes those values # as state changes
[docs] def set(self, value): PushesStateChanges._handlePotentialStateChange(self, value)
[docs] def get(self): return PushesStateChanges.getState(self)
if __debug__: sv = StateVar(0) assert sv.get() == 0 sv.set(1) assert sv.get() == 1 sv.destroy() del sv
[docs]class StateChangeNode(PushesStateChanges, ReceivesStateChanges): # base class that can be used to create a state-change notification chain
[docs] def __init__(self, source): ReceivesStateChanges.__init__(self, source) PushesStateChanges.__init__(self, source.getState()) ReceivesStateChanges._finishInit(self)
[docs] def destroy(self): PushesStateChanges.destroy(self) ReceivesStateChanges.destroy(self)
def _recvStatePush(self, source): # got a state push, apply new state to self self._handlePotentialStateChange(source._value)
if __debug__: sv = StateVar(0) assert sv.get() == 0 scn = StateChangeNode(sv) assert scn.getState() == 0 sv.set(1) assert sv.get() == 1 assert scn.getState() == 1 scn2 = StateChangeNode(scn) assert scn2.getState() == 1 sv.set(2) assert scn2.getState() == 2 scn3 = StateChangeNode(scn) assert scn3.getState() == 2 sv.set(3) assert scn2.getState() == 3 assert scn3.getState() == 3 scn3.destroy() scn2.destroy() scn.destroy() sv.destroy() del scn3 del scn2 del scn del sv
[docs]class ReceivesMultipleStateChanges: # base class for objects that subscribe to state changes from multiple PushesStateChanges # objects
[docs] def __init__(self): self._key2source = {} self._source2key = {}
[docs] def destroy(self): keys = list(self._key2source.keys()) for key in keys: self._unsubscribe(key) del self._key2source del self._source2key
def _subscribeTo(self, source, key): self._unsubscribe(key) self._key2source[key] = source self._source2key[source] = key source._addSubscription(self) def _unsubscribe(self, key): if key in self._key2source: source = self._key2source[key] source._removeSubscription(self) del self._key2source[key] del self._source2key[source] def _recvStatePush(self, source): self._recvMultiStatePush(self._source2key[source], source) def _recvMultiStatePush(self, key, source): pass
if __debug__: rsc = ReceivesMultipleStateChanges() sv = StateVar(0) sv2 = StateVar('b') rsc._subscribeTo(sv, 'a') rsc._subscribeTo(sv2, 2) rsc._unsubscribe('a') rsc.destroy() del rsc
[docs]class FunctionCall(ReceivesMultipleStateChanges, PushesStateChanges): # calls func with provided args whenever arguments' state changes
[docs] def __init__(self, func, *args, **kArgs): self._initialized = False ReceivesMultipleStateChanges.__init__(self) PushesStateChanges.__init__(self, None) self._func = func self._args = args self._kArgs = kArgs # keep a copy of the arguments ready to go, already filled in with # the value of arguments that push state self._bakedArgs = [] self._bakedKargs = {} for i, arg in enumerate(self._args): key = i if isinstance(arg, PushesStateChanges): self._bakedArgs.append(arg.getState()) self._subscribeTo(arg, key) else: self._bakedArgs.append(self._args[i]) for key, arg in self._kArgs.items(): if isinstance(arg, PushesStateChanges): self._bakedKargs[key] = arg.getState() self._subscribeTo(arg, key) else: self._bakedKargs[key] = arg self._initialized = True
# call pushCurrentState() instead ## push the current state to any listeners ##self._handleStateChange()
[docs] def destroy(self): ReceivesMultipleStateChanges.destroy(self) PushesStateChanges.destroy(self) del self._func del self._args del self._kArgs del self._bakedArgs del self._bakedKargs
[docs] def getState(self): # for any state recievers that are hooked up to us, they get a tuple # of (tuple(positional argument values), dict(keyword argument name->value)) return (tuple(self._bakedArgs), dict(self._bakedKargs))
def _recvMultiStatePush(self, key, source): # one of the arguments changed # pick up the new value if isinstance(key, str): self._bakedKargs[key] = source.getState() else: self._bakedArgs[key] = source.getState() # and send it out self._handlePotentialStateChange(self.getState()) def _handleStateChange(self): if self._initialized: self._func(*self._bakedArgs, **self._bakedKargs) PushesStateChanges._handleStateChange(self)
if __debug__: l = [] def handler(value, l=l): l.append(value) assert l == [] sv = StateVar(0) fc = FunctionCall(handler, sv) assert l == [] fc.pushCurrentState() assert l == [0,] sv.set(1) assert l == [0,1,] sv.set(2) assert l == [0,1,2,] fc.destroy() sv.destroy() del fc del sv del handler del l l = [] def handler(value, kDummy=None, kValue=None, l=l): l.append((value, kValue)) assert l == [] sv = StateVar(0) ksv = StateVar('a') fc = FunctionCall(handler, sv, kValue=ksv) assert l == [] fc.pushCurrentState() assert l == [(0,'a',),] sv.set(1) assert l == [(0,'a'),(1,'a'),] ksv.set('b') assert l == [(0,'a'),(1,'a'),(1,'b'),] fc.destroy() sv.destroy() del fc del sv del handler del l
[docs]class EnterExit(StateChangeNode): # call enterFunc when our state becomes true, exitFunc when it becomes false
[docs] def __init__(self, source, enterFunc, exitFunc): self._enterFunc = enterFunc self._exitFunc = exitFunc StateChangeNode.__init__(self, source)
[docs] def destroy(self): StateChangeNode.destroy(self) del self._exitFunc del self._enterFunc
def _handlePotentialStateChange(self, value): # convert the incoming state as a bool StateChangeNode._handlePotentialStateChange(self, bool(value)) def _handleStateChange(self): if self._value: self._enterFunc() else: self._exitFunc() StateChangeNode._handleStateChange(self)
if __debug__: l = [] def enter(l=l): l.append(1) def exit(l=l): l.append(0) sv = StateVar(0) ee = EnterExit(sv, enter, exit) sv.set(0) assert l == [] sv.set(1) assert l == [1,] sv.set(2) assert l == [1,] sv.set(0) assert l == [1,0,] sv.set(True) assert l == [1,0,1,] sv.set(False) assert l == [1,0,1,0,] ee.destroy() sv.destroy() del ee del sv del enter del exit del l
[docs]class Pulse(PushesStateChanges): # changes state to True then immediately to False whenever sendPulse is called
[docs] def __init__(self): PushesStateChanges.__init__(self, False)
[docs] def sendPulse(self): self._handlePotentialStateChange(True) self._handlePotentialStateChange(False)
if __debug__: l = [] def handler(value, l=l): l.append(value) p = Pulse() fc = FunctionCall(handler, p) assert l == [] fc.pushCurrentState() assert l == [False, ] p.sendPulse() assert l == [False, True, False, ] p.sendPulse() assert l == [False, True, False, True, False, ] fc.destroy() p.destroy() del fc del p del l del handler
[docs]class EventPulse(Pulse, DirectObject): # sends a True-False "pulse" whenever a specific messenger message is sent
[docs] def __init__(self, event): Pulse.__init__(self) self.accept(event, self.sendPulse)
[docs] def destroy(self): self.ignoreAll() Pulse.destroy(self)
[docs]class EventArgument(PushesStateChanges, DirectObject): # tracks a particular argument to a particular messenger event
[docs] def __init__(self, event, index=0): PushesStateChanges.__init__(self, None) self._index = index self.accept(event, self._handleEvent)
[docs] def destroy(self): self.ignoreAll() del self._index PushesStateChanges.destroy(self)
def _handleEvent(self, *args): self._handlePotentialStateChange(args[self._index])
[docs]class AttrSetter(StateChangeNode):
[docs] def __init__(self, source, object, attrName): self._object = object self._attrName = attrName StateChangeNode.__init__(self, source) self._handleStateChange()
def _handleStateChange(self): setattr(self._object, self._attrName, self._value) StateChangeNode._handleStateChange(self)
if __debug__: o = ScratchPad() svar = StateVar(0) aset = AttrSetter(svar, o, 'testAttr') assert hasattr(o, 'testAttr') assert o.testAttr == 0 svar.set('red') assert o.testAttr == 'red' aset.destroy() svar.destroy() o.destroy() del aset del svar del o