add docs to state machine

This commit is contained in:
petrucci4prez 2017-06-11 04:27:03 -04:00
parent df5eea8b71
commit 29f0383f0c
1 changed files with 54 additions and 0 deletions

View File

@ -1,3 +1,14 @@
'''
Controls and manages state of the alarm system. The design follows a reactor-
like model (similar to twisted) except the statemachine itself is passive (eg it
does not run/block on its own thread waiting for events). Instead the
StateMachine object holds the current status as well as the logic to move
between states, while child threads (listeners) deliver signals to the state
machine in response to external events. The state machine itself does not "wait"
but rather its selectState method could be called within any child thread (hence
the lock) which also means the entry/exit functions of each state are run within
the signal-originating child thread.
'''
import RPi.GPIO as GPIO import RPi.GPIO as GPIO
import time, logging, enum, os import time, logging, enum, os
from threading import Lock from threading import Lock
@ -17,6 +28,10 @@ from stream import Camera, FileDump
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class _SIGNALS(enum.Enum): class _SIGNALS(enum.Enum):
'''
Valid signals the statemachine can recieve. The consequence of each signal
depends on the current state
'''
ARM = enum.auto() ARM = enum.auto()
INSTANT_ARM = enum.auto() INSTANT_ARM = enum.auto()
DISARM = enum.auto() DISARM = enum.auto()
@ -63,6 +78,17 @@ def _resetUSBDevice(device):
logger.debug('Reset USB device: %s', devpath) logger.debug('Reset USB device: %s', devpath)
class _State: class _State:
'''
Represents one discrete status of the system. Each state has a set of entry
and exit functions and optionaly has sound that can play upon state entry.
States link to other states via the addTransition function, which links
another state with a signal. In this way, many states can be linked together
in a network.
There is currently nothing stopping multiple states from sharing the same
name...try not to be an idiot. This mostly matters in equality tests, which
only compares the name
'''
def __init__(self, name, entryCallbacks=[], exitCallbacks=[], sound=None): def __init__(self, name, entryCallbacks=[], exitCallbacks=[], sound=None):
self.name = name self.name = name
self.entryCallbacks = entryCallbacks self.entryCallbacks = entryCallbacks
@ -104,6 +130,32 @@ class _State:
return hash(self.name) return hash(self.name)
class StateMachine: class StateMachine:
'''
Manager for states. This is intended to be used as a context manager (eg
"with" statement) for brevity...and because there should only be one.
Init is responsible for setting up all objects (including child threads) as
well as contructing the state network. Each thread functions as some kind of
listener (or supports one) that wait for an event to happen.
Note we distinguish between "managed" and "non-managed" objects; the former
need to be started/stopped to ensure things get cleaned. Managed objects are
added to the managed list with _addManaged, which also returns a ref to the
object for other uses.
Upon opening the context, all threads are started. Note that not all threads
are managed; some are started and forgotten, as these require no cleanup.
Managed threads are started with _startManaged and stopped with _stopManaged
upon closing the context. This system has the added benefit of not trying to
stop an object that has not been initialized, as it cannot appear in the
managed list otherwise
During steady-state operation, the receiver for signals that make things
happen is selectState, intended to be called from any of the state machine's
child threads. This calls the current state's "next" method and sets the
result as the new current state. Note the lock because it could be called by
any number of listeners simultaneously
'''
def __init__(self): def __init__(self):
self._lock = Lock() self._lock = Lock()
self._managed = [] self._managed = []
@ -212,6 +264,7 @@ class StateMachine:
def __enter__(self): def __enter__(self):
_resetUSBDevice('1-1') _resetUSBDevice('1-1')
# start all managed threads (we retain ref to these to stop them later)
self._startManaged() self._startManaged()
def action(): def action():
@ -228,6 +281,7 @@ class StateMachine:
time.sleep(0.1) time.sleep(0.1)
self.fileDump.removeInitiator(pin) self.fileDump.removeInitiator(pin)
# start non-managed threads (we forget about these because they can exit with no cleanup)
startMotionSensor(5, 'Nate\'s room', action) startMotionSensor(5, 'Nate\'s room', action)
startMotionSensor(19, 'front door', action) startMotionSensor(19, 'front door', action)
startMotionSensor(26, 'Laura\'s room', action) startMotionSensor(26, 'Laura\'s room', action)