Mercurial > code > home > repos > homeauto

annotate service/piNode/devices.py @ 183:634d6e477953

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
get piNode working, for motionsensor at least Ignore-this: 53f16373acb0613e01fdf3e98d8425f2
author drewp@bigasterisk.com
date Sun, 31 May 2015 22:53:21 -0700
parents 9fff29ebca71
children 40000fafad94
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
1 from __future__ import division
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
2
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
3 import time, logging
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
4 from rdflib import Namespace, RDF, URIRef, Literal
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
5
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
6 try:
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
7 import pigpio
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
8 except ImportError:
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
9 pigpio = None
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
10
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
11
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
12 log = logging.getLogger()
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
13 ROOM = Namespace('http://projects.bigasterisk.com/room/')
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
14
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
15 class DeviceType(object):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
16 deviceType = None
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
17 @classmethod
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
18 def findInstances(cls, graph, board, pi):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
19 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
20 return any number of instances of this class for all the separately
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
21 controlled devices on the board. Two LEDS makes two instances,
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
22 but two sensors on the same onewire bus makes only one device
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
23 (which yields more statements).
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
24 """
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
25 for row in graph.query("""SELECT ?dev ?gpioNumber WHERE {
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
26 ?board :hasPin ?pin .
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
27 ?pin :gpioNumber ?gpioNumber;
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
28 :connectedTo ?dev .
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
29 ?dev a ?thisType .
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
30 } ORDER BY ?dev""",
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
31 initBindings=dict(board=board,
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
32 thisType=cls.deviceType),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
33 initNs={'': ROOM}):
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
34 yield cls(graph, row.dev, pi, int(row.gpioNumber))
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
35
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
36 def __init__(self, graph, uri, pi, pinNumber):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
37 self.graph, self.uri, self.pi = graph, uri, pi
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
38 self.pinNumber = pinNumber
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
39
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
40 def description(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
41 return {
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
42 'uri': self.uri,
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
43 'className': self.__class__.__name__,
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
44 'pinNumber': getattr(self, 'pinNumber', None),
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
45 'outputPatterns': self.outputPatterns(),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
46 'watchPrefixes': self.watchPrefixes(),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
47 'outputWidgets': self.outputWidgets(),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
48 }
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
49
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
50 def watchPrefixes(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
51 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
52 subj,pred pairs of the statements that might be returned from
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
53 readFromPoll, so the dashboard knows what it should
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
54 watch. This should be eliminated, as the dashboard should just
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
55 always watch the whole tree of statements starting self.uri
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
56 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
57 return []
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
58
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
59 def poll(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
60 return [] # statements
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
61
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
62 def outputPatterns(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
63 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
64 Triple patterns, using None as a wildcard, that should be routed
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
65 to sendOutput
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
66 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
67 return []
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
68
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
69 def outputWidgets(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
70 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
71 structs to make output widgets on the dashboard. ~1 of these per
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
72 handler you have in sendOutput
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
73 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
74 return []
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
75
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
76 def sendOutput(self, statements):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
77 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
78 If we got statements that match this class's outputPatterns, this
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
79 will be called with the statements that matched.
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
80
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
81 Todo: it would be fine to read back confirmations or
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
82 whatever. Just need a way to collect them into graph statements.
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
83 """
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
84 raise NotImplementedError
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
85
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
86 _knownTypes = set()
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
87 def register(deviceType):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
88 _knownTypes.add(deviceType)
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
89 return deviceType
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
90
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
91 @register
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
92 class MotionSensorInput(DeviceType):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
93 deviceType = ROOM['MotionSensor']
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
94
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
95 def setup(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
96 self.pi.set_mode(17, pigpio.INPUT)
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
97 self.pi.set_pull_up_down(17, pigpio.PUD_DOWN)
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
98
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
99 def poll(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
100 motion = self.pi.read(17)
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
101
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
102 return [
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
103 (self.uri, ROOM['sees'],
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
104 ROOM['motion'] if motion else ROOM['noMotion']),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
105 self.recentMotionStatement(motion),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
106 ]
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
107
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
108 def recentMotionStatement(self, motion):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
109 if not hasattr(self, 'lastMotionTime'):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
110 self.lastMotionTime = 0
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
111 now = time.time()
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
112 if motion:
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
113 self.lastMotionTime = now
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
114 recentMotion = now - self.lastMotionTime < 60 * 10
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
115 return (self.uri, ROOM['seesRecently'],
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
116 ROOM['motion'] if recentMotion else ROOM['noMotion'])
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
117
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
118 def watchPrefixes(self):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
119 return [
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
120 (self.uri, ROOM['sees']),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
121 (self.uri, ROOM['seesRecently']),
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
122 ]
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
123
183
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
124
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
125 @register
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
126 class RgbStrip(DeviceType):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
127 deviceType = ROOM['RgbStrip']
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
128
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
129 @classmethod
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
130 def findInstances(cls, graph, board, pi):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
131 for row in graph.query("""SELECT DISTINCT ?dev ?r ?g ?b WHERE {
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
132 ?board
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
133 :hasPin ?rpin;
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
134 :hasPin ?gpin;
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
135 :hasPin ?bpin .
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
136 ?dev a :RgbStrip;
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
137 :redChannel ?rpin;
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
138 :greenChannel ?gpin;
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
139 :blueChannel ?bpin .
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
140 ?rpin :gpioNumber ?r .
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
141 ?gpin :gpioNumber ?g .
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
142 ?bpin :gpioNumber ?b .
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
143 } ORDER BY ?dev""",
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
144 initBindings=dict(board=board),
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
145 initNs={'': ROOM}):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
146 log.debug('found rgb %r', row)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
147 yield cls(graph, row.dev, pi, row.r, row.g, row.b)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
148
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
149 def __init__(self, graph, uri, pi, r, g, b):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
150 self.graph, self.uri, self.pi = graph, uri, pi
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
151 self.rgb = map(int, [r, g, b])
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
152
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
153 def setup(self):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
154 for i in self.rgb:
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
155 self.pi.set_mode(i, pigpio.OUTPUT)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
156 self.pi.set_PWM_frequency(i, 200)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
157 self.pi.set_PWM_dutycycle(i, 0)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
158
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
159 def outputPatterns(self):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
160 return [(self.uri, ROOM['color'], None)]
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
161
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
162 def sendOutput(self, statements):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
163 assert len(statements) == 1
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
164 assert statements[0][:2] == (self.uri, ROOM['color'])
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
165
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
166 rrggbb = statements[0][2].lstrip('#')
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
167 rgb = [int(x, 16) for x in [rrggbb[0:2], rrggbb[2:4], rrggbb[4:6]]]
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
168
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
169 for (i, v) in zip(self.rgb, rgb):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
170 self.pi.set_PWM_dutycycle(i, v)
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
171
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
172 def outputWidgets(self):
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
173 return [{
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
174 'element': 'output-rgb',
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
175 'subj': self.uri,
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
176 'pred': ROOM['color'],
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
177 }]
634d6e477953 get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents: 182
diff changeset
178
182
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
179 def makeDevices(graph, board, pi):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
180 out = []
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
181 for dt in sorted(_knownTypes, key=lambda cls: cls.__name__):
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
182 out.extend(dt.findInstances(graph, board, pi))
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
183 return out
9fff29ebca71 start pinode
drewp@bigasterisk.com
parents:
diff changeset
184