Mercurial > code > home > repos > homeauto
annotate service/piNode/devices.py @ 210:75717a0be824
ledoutput support for piNode
Ignore-this: 3c803f823e2db77285885395db892835
| author | drewp@bigasterisk.com |
|---|---|
| date | Sat, 07 Nov 2015 12:24:59 -0800 |
| parents | 6a1462822c56 |
| children | fb23df31b642 |
| rev | line source |
|---|---|
| 182 | 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 | 4 from rdflib import Namespace, RDF, URIRef, Literal |
| 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 | 13 ROOM = Namespace('http://projects.bigasterisk.com/room/') |
| 209 | 14 XSD = Namespace('http://www.w3.org/2001/XMLSchema#') |
| 182 | 15 |
| 16 class DeviceType(object): | |
| 17 deviceType = None | |
| 18 @classmethod | |
| 19 def findInstances(cls, graph, board, pi): | |
| 20 """ | |
| 21 return any number of instances of this class for all the separately | |
| 22 controlled devices on the board. Two LEDS makes two instances, | |
| 23 but two sensors on the same onewire bus makes only one device | |
| 24 (which yields more statements). | |
| 25 """ | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
26 for row in graph.query("""SELECT ?dev ?gpioNumber WHERE { |
| 182 | 27 ?board :hasPin ?pin . |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
28 ?pin :gpioNumber ?gpioNumber; |
| 182 | 29 :connectedTo ?dev . |
| 30 ?dev a ?thisType . | |
| 31 } ORDER BY ?dev""", | |
| 32 initBindings=dict(board=board, | |
| 33 thisType=cls.deviceType), | |
| 34 initNs={'': ROOM}): | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
35 yield cls(graph, row.dev, pi, int(row.gpioNumber)) |
| 182 | 36 |
| 37 def __init__(self, graph, uri, pi, pinNumber): | |
| 38 self.graph, self.uri, self.pi = graph, uri, pi | |
| 39 self.pinNumber = pinNumber | |
| 40 | |
| 41 def description(self): | |
| 42 return { | |
| 43 'uri': self.uri, | |
| 44 'className': self.__class__.__name__, | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
45 'pinNumber': getattr(self, 'pinNumber', None), |
| 182 | 46 'outputPatterns': self.outputPatterns(), |
| 47 'watchPrefixes': self.watchPrefixes(), | |
| 48 'outputWidgets': self.outputWidgets(), | |
| 49 } | |
| 50 | |
| 51 def watchPrefixes(self): | |
| 52 """ | |
| 53 subj,pred pairs of the statements that might be returned from | |
| 54 readFromPoll, so the dashboard knows what it should | |
| 55 watch. This should be eliminated, as the dashboard should just | |
| 56 always watch the whole tree of statements starting self.uri | |
| 57 """ | |
| 58 return [] | |
| 59 | |
| 60 def poll(self): | |
| 61 return [] # statements | |
| 62 | |
| 63 def outputPatterns(self): | |
| 64 """ | |
| 65 Triple patterns, using None as a wildcard, that should be routed | |
| 66 to sendOutput | |
| 67 """ | |
| 68 return [] | |
| 69 | |
| 70 def outputWidgets(self): | |
| 71 """ | |
| 72 structs to make output widgets on the dashboard. ~1 of these per | |
| 73 handler you have in sendOutput | |
| 74 """ | |
| 75 return [] | |
| 76 | |
| 77 def sendOutput(self, statements): | |
| 78 """ | |
| 79 If we got statements that match this class's outputPatterns, this | |
| 80 will be called with the statements that matched. | |
| 81 | |
| 82 Todo: it would be fine to read back confirmations or | |
| 83 whatever. Just need a way to collect them into graph statements. | |
| 84 """ | |
| 85 raise NotImplementedError | |
| 86 | |
| 87 _knownTypes = set() | |
| 88 def register(deviceType): | |
| 89 _knownTypes.add(deviceType) | |
| 90 return deviceType | |
| 91 | |
| 92 @register | |
| 93 class MotionSensorInput(DeviceType): | |
| 94 deviceType = ROOM['MotionSensor'] | |
| 95 | |
| 96 def setup(self): | |
| 97 self.pi.set_mode(17, pigpio.INPUT) | |
| 98 self.pi.set_pull_up_down(17, pigpio.PUD_DOWN) | |
| 99 | |
| 100 def poll(self): | |
| 101 motion = self.pi.read(17) | |
| 102 | |
| 103 return [ | |
| 104 (self.uri, ROOM['sees'], | |
| 105 ROOM['motion'] if motion else ROOM['noMotion']), | |
| 106 self.recentMotionStatement(motion), | |
| 107 ] | |
| 108 | |
| 109 def recentMotionStatement(self, motion): | |
| 110 if not hasattr(self, 'lastMotionTime'): | |
| 111 self.lastMotionTime = 0 | |
| 112 now = time.time() | |
| 113 if motion: | |
| 114 self.lastMotionTime = now | |
| 115 recentMotion = now - self.lastMotionTime < 60 * 10 | |
| 116 return (self.uri, ROOM['seesRecently'], | |
| 117 ROOM['motion'] if recentMotion else ROOM['noMotion']) | |
| 118 | |
| 119 def watchPrefixes(self): | |
| 120 return [ | |
| 121 (self.uri, ROOM['sees']), | |
| 122 (self.uri, ROOM['seesRecently']), | |
| 123 ] | |
| 124 | |
|
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
125 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
126 @register |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
127 class RgbStrip(DeviceType): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
128 deviceType = ROOM['RgbStrip'] |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
129 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
130 @classmethod |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
131 def findInstances(cls, graph, board, pi): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
132 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
|
133 ?board |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
134 :hasPin ?rpin; |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
135 :hasPin ?gpin; |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
136 :hasPin ?bpin . |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
137 ?dev a :RgbStrip; |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
138 :redChannel ?rpin; |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
139 :greenChannel ?gpin; |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
140 :blueChannel ?bpin . |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
141 ?rpin :gpioNumber ?r . |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
142 ?gpin :gpioNumber ?g . |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
143 ?bpin :gpioNumber ?b . |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
144 } ORDER BY ?dev""", |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
145 initBindings=dict(board=board), |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
146 initNs={'': ROOM}): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
147 log.debug('found rgb %r', row) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
148 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
|
149 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
150 def __init__(self, graph, uri, pi, r, g, b): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
151 self.graph, self.uri, self.pi = graph, uri, pi |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
152 self.rgb = map(int, [r, g, b]) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
153 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
154 def setup(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
155 for i in self.rgb: |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
156 self.pi.set_mode(i, pigpio.OUTPUT) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
157 self.pi.set_PWM_frequency(i, 200) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
158 self.pi.set_PWM_dutycycle(i, 0) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
159 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
160 def outputPatterns(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
161 return [(self.uri, ROOM['color'], None)] |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
162 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
163 def sendOutput(self, statements): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
164 assert len(statements) == 1 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
165 assert statements[0][:2] == (self.uri, ROOM['color']) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
166 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
167 rrggbb = statements[0][2].lstrip('#') |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
168 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
|
169 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
170 for (i, v) in zip(self.rgb, rgb): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
171 self.pi.set_PWM_dutycycle(i, v) |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
172 |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
173 def outputWidgets(self): |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
174 return [{ |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
175 'element': 'output-rgb', |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
176 'subj': self.uri, |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
177 'pred': ROOM['color'], |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
178 }] |
|
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
179 |
| 210 | 180 |
| 181 @register | |
| 182 class LedOutput(DeviceType): | |
| 183 deviceType = ROOM['LedOutput'] | |
| 184 | |
| 185 def setup(self): | |
| 186 self.pi.set_mode(self.pinNumber, pigpio.OUTPUT) | |
| 187 self.pi.set_PWM_frequency(self.pinNumber, 200) | |
| 188 self.pi.set_PWM_dutycycle(self.pinNumber, 0) | |
| 189 | |
| 190 def outputPatterns(self): | |
| 191 return [(self.uri, ROOM['brightness'], None)] | |
| 192 | |
| 193 def sendOutput(self, statements): | |
| 194 assert len(statements) == 1 | |
| 195 assert statements[0][:2] == (self.uri, ROOM['brightness']) | |
| 196 v = int(float(statements[0][2]) * 255) | |
| 197 self.pi.set_PWM_dutycycle(self.pinNumber, v) | |
| 198 | |
| 199 def outputWidgets(self): | |
| 200 return [{ | |
| 201 'element': 'output-slider', | |
| 202 'min': 0, | |
| 203 'max': 1, | |
| 204 'step': 1 / 255, | |
| 205 'subj': self.uri, | |
| 206 'pred': ROOM['brightness'], | |
| 207 }] | |
| 208 | |
| 209 | |
| 205 | 210 @register |
| 211 class OnboardTemperature(DeviceType): | |
| 212 deviceType = ROOM['OnboardTemperature'] | |
| 213 @classmethod | |
| 214 def findInstances(cls, graph, board, pi): | |
| 207 | 215 for row in graph.query('''SELECT DISTINCT ?uri WHERE { |
| 205 | 216 ?board :onboardDevice ?uri . |
| 217 ?uri a :OnboardTemperature . | |
| 218 }'''): | |
| 219 yield cls(graph, row.uri, pi, pinNumber=None) | |
| 220 | |
| 208 | 221 def poll(self): |
| 205 | 222 milliC = open('/sys/class/thermal/thermal_zone0/temp').read().strip() |
| 223 c = float(milliC) / 1000. | |
| 224 f = c * 1.8 + 32 | |
| 225 return [ | |
| 226 (self.uri, ROOM['temperatureF'], Literal(f, datatype=XSD['decimal'])), | |
| 227 ] | |
| 228 | |
| 229 def watchPrefixes(self): | |
| 230 # these uris will become dynamic! see note on watchPrefixes | |
| 231 # about eliminating it. | |
| 232 return [(self.uri, ROOM['temperatureF']), | |
| 233 ] | |
| 234 | |
| 182 | 235 def makeDevices(graph, board, pi): |
| 236 out = [] | |
| 237 for dt in sorted(_knownTypes, key=lambda cls: cls.__name__): | |
| 238 out.extend(dt.findInstances(graph, board, pi)) | |
| 239 return out | |
| 240 |