439a7bb0cd2bc0d783d1d01831f420e9b47a0a18
[saartuer.git] / actor.py
1 from libtuer import ThreadFunction, logger
2 import RPi.GPIO as GPIO
3 import time
4         
5 class Actor:
6         CMD_BUZZ = 0
7         CMD_UNLOCK = 1
8         CMD_LOCK = 2
9         
10         class CMD():
11                 def __init__(self, name, pin, tid, todo):
12                         self.name = name
13                         self.pin = pin
14                         self.tid = tid
15                         self.todo = todo
16                         # don't do the GPIO setup here, the main init did not yet run
17                 
18                 def execute(self):
19                         logger.info("Actor: Running command %s" % self.name)
20                         for (value, delay) in self.todo:
21                                 if value is not None:
22                                         logger.debug("Actor: Setting pin %d to %d" % (self.pin, value))
23                                         GPIO.output(self.pin, value)
24                                 time.sleep(delay)
25         
26         CMDs = {
27                 CMD_UNLOCK:  CMD("unlock", pin=12, tid=0, todo=[(True, 0.3), (False, 0.1)]),
28                 CMD_LOCK:  CMD("lock", pin=16, tid=0, todo=[(True, 0.3), (False, 0.1)]),
29                 CMD_BUZZ: CMD("buzz", pin=22, tid=1, todo=[(True, 2.0), (False, 0.1)]),
30         }
31         
32         def __init__(self):
33                 # launch threads, all running the "_execute" method
34                 self.threads = {}
35                 for cmd in Actor.CMDs.values():
36                         GPIO.setup(cmd.pin, GPIO.OUT)
37                         if not cmd.tid in self.threads:
38                                 self.threads[cmd.tid] = ThreadFunction(self._execute, "Actor TID %d" % cmd.tid)
39         
40         def _execute(self, cmd):
41                 Actor.CMDs[cmd].execute()
42         
43         def act(self, cmd):
44                 # dispatch command to correct thread
45                 self.threads[Actor.CMDs[cmd].tid](cmd)
46         
47         def stop(self):
48                 for thread in self.threads.values():
49                         thread.stop()