Mercurial > code > home > repos > homeauto
annotate service/piNode/devices.py @ 274:de541d0697b8
smooth output fading support
Ignore-this: 69d396a6c2ea8f693e270525e29d5b5d
author | drewp@bigasterisk.com |
---|---|
date | Thu, 14 Apr 2016 01:52:09 -0700 |
parents | 4685fbd208d1 |
children | e1b870f1274c |
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 |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
3 import time, logging, os |
243
141079644c45
piNode poll switches much faster. mirror the logic in arduinoNode though vari-rate poll is not supported yet
drewp@bigasterisk.com
parents:
233
diff
changeset
|
4 from rdflib import Namespace, URIRef, Literal |
274 | 5 from twisted.internet import reactor |
182 | 6 |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
7 try: |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
8 import pigpio |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
9 except ImportError: |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
10 pigpio = None |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
11 import w1thermsensor |
259 | 12 try: |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
13 import neopixel |
259 | 14 except ImportError: |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
15 neopixel = None |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
16 |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
17 def setupPwm(pi, pinNumber, hz=8000): |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
18 pi.set_mode(pinNumber, pigpio.OUTPUT) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
19 # see http://abyz.co.uk/rpi/pigpio/cif.html#gpioCfgClock |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
20 # and http://abyz.co.uk/rpi/pigpio/cif.html#gpioSetPWMfrequency |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
21 actual = pi.set_PWM_frequency(pinNumber, hz) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
22 if actual != hz: |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
23 raise ValueError('pwm actual=%s' % actual) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
24 pi.set_PWM_dutycycle(pinNumber, 0) |
259 | 25 |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
26 import sys |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
27 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
28 log = logging.getLogger() |
182 | 29 ROOM = Namespace('http://projects.bigasterisk.com/room/') |
209 | 30 XSD = Namespace('http://www.w3.org/2001/XMLSchema#') |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
31 RDFS = Namespace('http://www.w3.org/2000/01/rdf-schema#') |
182 | 32 |
33 class DeviceType(object): | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
34 deviceType = NotImplementedError |
182 | 35 @classmethod |
36 def findInstances(cls, graph, board, pi): | |
37 """ | |
38 return any number of instances of this class for all the separately | |
39 controlled devices on the board. Two LEDS makes two instances, | |
40 but two sensors on the same onewire bus makes only one device | |
41 (which yields more statements). | |
42 """ | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
43 log.debug("graph has any connected devices of type %s?", cls.deviceType) |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
44 for row in graph.query("""SELECT ?dev ?gpioNumber WHERE { |
182 | 45 ?board :hasPin ?pin . |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
46 ?pin :gpioNumber ?gpioNumber; |
182 | 47 :connectedTo ?dev . |
48 ?dev a ?thisType . | |
49 } ORDER BY ?dev""", | |
50 initBindings=dict(board=board, | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
51 thisType=cls.deviceType)): |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
52 yield cls(graph, row.dev, pi, int(row.gpioNumber)) |
182 | 53 |
54 def __init__(self, graph, uri, pi, pinNumber): | |
55 self.graph, self.uri, self.pi = graph, uri, pi | |
56 self.pinNumber = pinNumber | |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
57 self.hostStateInit() |
182 | 58 |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
59 def hostStateInit(self): |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
60 """ |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
61 If you don't want to use __init__, you can use this to set up |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
62 whatever storage you might need for hostStatements |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
63 """ |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
64 |
182 | 65 def description(self): |
66 return { | |
67 'uri': self.uri, | |
68 'className': self.__class__.__name__, | |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
69 'pinNumber': getattr(self, 'pinNumber', None), |
182 | 70 'outputPatterns': self.outputPatterns(), |
71 'watchPrefixes': self.watchPrefixes(), | |
72 'outputWidgets': self.outputWidgets(), | |
73 } | |
74 | |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
75 def hostStatements(self): |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
76 """ |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
77 Like readFromPoll but these statements come from the host-side |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
78 python code, not the connected device. Include output state |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
79 (e.g. light brightness) if its master version is in this |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
80 object. This method is called on /graph requests so it should |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
81 be fast. |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
82 """ |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
83 return [] |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
84 |
182 | 85 def watchPrefixes(self): |
86 """ | |
87 subj,pred pairs of the statements that might be returned from | |
88 readFromPoll, so the dashboard knows what it should | |
89 watch. This should be eliminated, as the dashboard should just | |
90 always watch the whole tree of statements starting self.uri | |
91 """ | |
92 return [] | |
93 | |
94 def poll(self): | |
95 return [] # statements | |
96 | |
97 def outputPatterns(self): | |
98 """ | |
99 Triple patterns, using None as a wildcard, that should be routed | |
100 to sendOutput | |
101 """ | |
102 return [] | |
103 | |
104 def outputWidgets(self): | |
105 """ | |
106 structs to make output widgets on the dashboard. ~1 of these per | |
107 handler you have in sendOutput | |
108 """ | |
109 return [] | |
110 | |
111 def sendOutput(self, statements): | |
112 """ | |
113 If we got statements that match this class's outputPatterns, this | |
114 will be called with the statements that matched. | |
115 | |
116 Todo: it would be fine to read back confirmations or | |
117 whatever. Just need a way to collect them into graph statements. | |
118 """ | |
119 raise NotImplementedError | |
120 | |
121 _knownTypes = set() | |
122 def register(deviceType): | |
123 _knownTypes.add(deviceType) | |
124 return deviceType | |
125 | |
126 @register | |
127 class MotionSensorInput(DeviceType): | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
128 # compare motion sensor lib at http://pythonhosted.org/gpiozero/inputs/ |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
129 # which is a bit fancier |
182 | 130 deviceType = ROOM['MotionSensor'] |
131 | |
132 def setup(self): | |
133 self.pi.set_mode(17, pigpio.INPUT) | |
134 self.pi.set_pull_up_down(17, pigpio.PUD_DOWN) | |
135 | |
254
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
136 def hostStateInit(self): |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
137 self.lastRead = None |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
138 |
182 | 139 def poll(self): |
140 motion = self.pi.read(17) | |
141 | |
254
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
142 oneshot = [] |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
143 if self.lastRead is not None and motion != self.lastRead: |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
144 oneshot = [(self.uri, ROOM['sees'], ROOM['motionStart'])] |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
145 self.lastRead = motion |
ba1149814877
pi device support for motionStart edge trigger
drewp@bigasterisk.com
parents:
251
diff
changeset
|
146 |
251
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
147 return {'latest': [ |
182 | 148 (self.uri, ROOM['sees'], |
149 ROOM['motion'] if motion else ROOM['noMotion']), | |
267 | 150 ] + self.recentMotionStatements(motion), |
151 'oneshot': oneshot} | |
182 | 152 |
267 | 153 def recentMotionStatements(self, motion): |
182 | 154 if not hasattr(self, 'lastMotionTime'): |
155 self.lastMotionTime = 0 | |
156 now = time.time() | |
157 if motion: | |
158 self.lastMotionTime = now | |
267 | 159 dt = now - self.lastMotionTime |
160 return [(self.uri, ROOM['seesRecently'], | |
161 ROOM['motion'] if (dt < 60 * 10) else ROOM['noMotion']), | |
162 (self.uri, ROOM['seesRecently30'], | |
163 ROOM['motion'] if (dt < 30) else ROOM['noMotion'])] | |
182 | 164 |
165 def watchPrefixes(self): | |
166 return [ | |
167 (self.uri, ROOM['sees']), | |
168 (self.uri, ROOM['seesRecently']), | |
267 | 169 (self.uri, ROOM['seesRecently30']), |
182 | 170 ] |
171 | |
183
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 @register |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
174 class RgbStrip(DeviceType): |
220 | 175 """3 PWMs for r/g/b on a strip""" |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
176 # pigpio daemon is working fine, but |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
177 # https://github.com/RPi-Distro/python-gpiozero/blob/59ba7154c5918745ac894ea03503667d6473c760/gpiozero/output_devices.py#L213 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
178 # can also apparently do PWM |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
179 deviceType = ROOM['RgbStrip'] |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
180 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
181 @classmethod |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
182 def findInstances(cls, graph, board, pi): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
183 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
|
184 ?board |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
185 :hasPin ?rpin; |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
186 :hasPin ?gpin; |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
187 :hasPin ?bpin . |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
188 ?dev a :RgbStrip; |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
189 :redChannel ?rpin; |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
190 :greenChannel ?gpin; |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
191 :blueChannel ?bpin . |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
192 ?rpin :gpioNumber ?r . |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
193 ?gpin :gpioNumber ?g . |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
194 ?bpin :gpioNumber ?b . |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
195 } ORDER BY ?dev""", |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
196 initBindings=dict(board=board), |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
197 initNs={'': ROOM}): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
198 log.debug('found rgb %r', row) |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
199 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
|
200 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
201 def __init__(self, graph, uri, pi, r, g, b): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
202 self.graph, self.uri, self.pi = graph, uri, pi |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
203 self.rgb = map(int, [r, g, b]) |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
204 self.value = '#000000' |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
205 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
206 def setup(self): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
207 for i in self.rgb: |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
208 setupPwm(self.pi, i) |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
209 |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
210 def hostStatements(self): |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
211 return [(self.uri, ROOM['color'], Literal(self.value))] |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
212 |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
213 def outputPatterns(self): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
214 return [(self.uri, ROOM['color'], None)] |
220 | 215 |
216 def _rgbFromHex(self, h): | |
217 rrggbb = h.lstrip('#') | |
218 return [int(x, 16) for x in [rrggbb[0:2], rrggbb[2:4], rrggbb[4:6]]] | |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
219 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
220 def sendOutput(self, statements): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
221 assert len(statements) == 1 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
222 assert statements[0][:2] == (self.uri, ROOM['color']) |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
223 |
220 | 224 rgb = self._rgbFromHex(statements[0][2]) |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
225 self.value = statements[0][2] |
183
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
226 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
227 for (i, v) in zip(self.rgb, rgb): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
228 self.pi.set_PWM_dutycycle(i, v) |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
229 |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
230 def outputWidgets(self): |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
231 return [{ |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
232 'element': 'output-rgb', |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
233 'subj': self.uri, |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
234 'pred': ROOM['color'], |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
235 }] |
634d6e477953
get piNode working, for motionsensor at least
drewp@bigasterisk.com
parents:
182
diff
changeset
|
236 |
210 | 237 |
238 @register | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
239 class TempHumidSensor(DeviceType): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
240 deviceType = ROOM['TempHumidSensor'] |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
241 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
242 def __init__(self, *a, **kw): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
243 DeviceType.__init__(self, *a, **kw) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
244 sys.path.append('/opt/pigpio/EXAMPLES/Python/DHT22_AM2302_SENSOR') |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
245 import DHT22 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
246 self.sensor = DHT22.sensor(self.pi, self.pinNumber) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
247 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
248 def poll(self): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
249 self.sensor.trigger() |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
250 humid, tempC = self.sensor.humidity(), self.sensor.temperature() |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
251 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
252 stmts = set() |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
253 if humid is not None: |
232
4e91f3ec460b
more round() workarounds for mismatching literals not patching themselves out
drewp@bigasterisk.com
parents:
221
diff
changeset
|
254 stmts.add((self.uri, ROOM['humidity'], Literal(round(humid, 2)))) |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
255 else: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
256 stmts.add((self.uri, RDFS['comment'], |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
257 Literal('DHT read returned None'))) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
258 if tempC is not None: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
259 stmts.add((self.uri, ROOM['temperatureF'], |
232
4e91f3ec460b
more round() workarounds for mismatching literals not patching themselves out
drewp@bigasterisk.com
parents:
221
diff
changeset
|
260 # see round() note in arduinoNode/devices.py |
4e91f3ec460b
more round() workarounds for mismatching literals not patching themselves out
drewp@bigasterisk.com
parents:
221
diff
changeset
|
261 Literal(round(tempC * 9 / 5 + 32, 2)))) |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
262 else: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
263 stmts.add((self.uri, RDFS['comment'], |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
264 Literal('DHT read returned None'))) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
265 return stmts |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
266 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
267 def watchPrefixes(self): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
268 return [ |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
269 (self.uri, ROOM['temperatureF']), |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
270 (self.uri, ROOM['humidity']), |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
271 ] |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
272 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
273 @register |
244 | 274 class PushbuttonInput(DeviceType): |
275 """add a switch to ground; we'll turn on pullup""" | |
276 deviceType = ROOM['Pushbutton'] | |
277 | |
278 def __init__(self, *a, **kw): | |
279 DeviceType.__init__(self, *a, **kw) | |
280 log.debug("setup switch on %r", self.pinNumber) | |
281 self.pi.set_mode(self.pinNumber, pigpio.INPUT) | |
282 self.pi.set_pull_up_down(self.pinNumber, pigpio.PUD_UP) | |
251
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
283 self.lastClosed = None |
262 | 284 self.invert = (self.uri, ROOM['style'], |
285 ROOM['inverted']) in self.graph | |
244 | 286 |
287 def poll(self): | |
288 closed = not self.pi.read(self.pinNumber) | |
262 | 289 if self.invert: |
290 closed = not closed | |
251
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
291 |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
292 if self.lastClosed is not None and closed != self.lastClosed: |
262 | 293 log.debug('%s changed to %s', self.uri, closed) |
251
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
294 oneshot = [ |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
295 (self.uri, ROOM['buttonState'], |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
296 ROOM['press'] if closed else ROOM['release']), |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
297 ] |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
298 else: |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
299 oneshot = [] |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
300 self.lastClosed = closed |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
301 |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
302 return {'latest': [ |
244 | 303 (self.uri, ROOM['buttonState'], |
304 ROOM['pressed'] if closed else ROOM['notPressed']), | |
251
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
305 ], |
254df9f881a6
start sending oneshot events from some devices
drewp@bigasterisk.com
parents:
244
diff
changeset
|
306 'oneshot':oneshot} |
244 | 307 |
308 def watchPrefixes(self): | |
309 return [ | |
310 (self.uri, ROOM['buttonState']), | |
311 ] | |
312 | |
313 @register | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
314 class OneWire(DeviceType): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
315 """ |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
316 Also see /my/proj/ansible/roles/raspi_io_node/tasks/main.yml for |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
317 some system config that contains the pin number that you want to |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
318 use for onewire. The pin number in this config is currently ignored. |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
319 """ |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
320 deviceType = ROOM['OneWire'] |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
321 # deliberately written like arduinoNode's one for an easier merge. |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
322 def __init__(self, *a, **kw): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
323 DeviceType.__init__(self, *a, **kw) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
324 log.info("scan for w1 devices") |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
325 self._sensors = w1thermsensor.W1ThermSensor.get_available_sensors() |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
326 for s in self._sensors: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
327 # Something looks different about these ids |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
328 # ('000003a5a94c') vs the ones I get from arduino |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
329 # ('2813bea50300003d'). Not sure if I'm parsing them |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
330 # differently or what. |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
331 s.uri = URIRef(os.path.join(self.uri, 'dev-%s' % s.id)) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
332 log.info(' found temperature sensor %s' % s.uri) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
333 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
334 def poll(self): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
335 try: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
336 stmts = [] |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
337 for sensor in self._sensors: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
338 stmts.append((self.uri, ROOM['connectedTo'], sensor.uri)) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
339 try: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
340 tempF = sensor.get_temperature(sensor.DEGREES_F) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
341 stmts.append((sensor.uri, ROOM['temperatureF'], |
232
4e91f3ec460b
more round() workarounds for mismatching literals not patching themselves out
drewp@bigasterisk.com
parents:
221
diff
changeset
|
342 # see round() note in arduinoNode/devices.py |
4e91f3ec460b
more round() workarounds for mismatching literals not patching themselves out
drewp@bigasterisk.com
parents:
221
diff
changeset
|
343 Literal(round(tempF, 2)))) |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
344 except w1thermsensor.core.SensorNotReadyError as e: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
345 log.warning(e) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
346 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
347 return stmts |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
348 except Exception as e: |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
349 log.error(e) |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
350 os.abort() |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
351 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
352 def watchPrefixes(self): |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
353 return [(s.uri, ROOM['temperatureF']) for s in self._sensors] |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
354 |
274 | 355 class FilteredValue(object): |
356 def __init__(self, setter, | |
357 slew=2.0, # step/sec max slew rate | |
358 accel=5, # step/sec^2 acceleration | |
359 ): | |
360 self.setter = setter | |
361 self.slew, self.accel = slew, accel | |
362 | |
363 self.x = None | |
364 self.dx = 0 | |
365 self.goal = self.x | |
366 self.lastStep = 0 | |
367 | |
368 def set(self, goal): | |
369 self.goal = goal | |
370 self.step() | |
371 | |
372 def step(self): | |
373 now = time.time() | |
374 dt = min(.1, now - self.lastStep) | |
375 self.lastStep = now | |
376 | |
377 if self.x is None: | |
378 self.x = self.goal | |
379 | |
380 if self.goal > self.x: | |
381 self.dx = min(self.slew, self.dx + self.accel * dt) | |
382 else: | |
383 self.dx = max(-self.slew, self.dx - self.accel * dt) | |
384 | |
385 nextX = self.x + self.dx * dt | |
386 if self.x == self.goal or (self.x < self.goal < nextX) or (self.x > self.goal > nextX): | |
387 self.x = self.goal | |
388 self.dx = 0 | |
389 else: | |
390 self.x = nextX | |
391 reactor.callLater(.05, self.step) | |
392 | |
393 #print "x= %(x)s dx= %(dx)s goal= %(goal)s" % self.__dict__ | |
394 self.setter(self.x) | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
395 |
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
396 @register |
210 | 397 class LedOutput(DeviceType): |
398 deviceType = ROOM['LedOutput'] | |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
399 |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
400 def hostStateInit(self): |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
401 self.value = 0 |
274 | 402 self.fv = FilteredValue(self._setPwm) |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
403 |
210 | 404 def setup(self): |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
405 setupPwm(self.pi, self.pinNumber) |
210 | 406 |
407 def outputPatterns(self): | |
408 return [(self.uri, ROOM['brightness'], None)] | |
409 | |
410 def sendOutput(self, statements): | |
411 assert len(statements) == 1 | |
412 assert statements[0][:2] == (self.uri, ROOM['brightness']) | |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
413 self.value = float(statements[0][2]) |
274 | 414 self.fv.set(self.value) |
415 | |
416 def _setPwm(self, x): | |
417 v = int(x * 255) | |
210 | 418 self.pi.set_PWM_dutycycle(self.pinNumber, v) |
233
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
419 |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
420 def hostStatements(self): |
4ebb5cc30002
server/browser graph sync. cut dependency on the WS version. merge some changes between arduino/pi code.
drewp@bigasterisk.com
parents:
232
diff
changeset
|
421 return [(self.uri, ROOM['brightness'], Literal(self.value))] |
210 | 422 |
423 def outputWidgets(self): | |
424 return [{ | |
425 'element': 'output-slider', | |
426 'min': 0, | |
427 'max': 1, | |
428 'step': 1 / 255, | |
429 'subj': self.uri, | |
430 'pred': ROOM['brightness'], | |
431 }] | |
432 | |
433 | |
205 | 434 @register |
435 class OnboardTemperature(DeviceType): | |
436 deviceType = ROOM['OnboardTemperature'] | |
243
141079644c45
piNode poll switches much faster. mirror the logic in arduinoNode though vari-rate poll is not supported yet
drewp@bigasterisk.com
parents:
233
diff
changeset
|
437 pollPeriod = 10 |
205 | 438 @classmethod |
439 def findInstances(cls, graph, board, pi): | |
259 | 440 log.debug('graph has any connected devices of type OnboardTemperature on %r?', board) |
207 | 441 for row in graph.query('''SELECT DISTINCT ?uri WHERE { |
205 | 442 ?board :onboardDevice ?uri . |
443 ?uri a :OnboardTemperature . | |
221
666e0e756ce6
piNode support for temp sensors. proper hostname lookup
drewp@bigasterisk.com
parents:
220
diff
changeset
|
444 }''', initBindings=dict(board=board)): |
259 | 445 log.debug(' found %s', row.uri) |
205 | 446 yield cls(graph, row.uri, pi, pinNumber=None) |
447 | |
208 | 448 def poll(self): |
205 | 449 milliC = open('/sys/class/thermal/thermal_zone0/temp').read().strip() |
450 c = float(milliC) / 1000. | |
451 f = c * 1.8 + 32 | |
452 return [ | |
453 (self.uri, ROOM['temperatureF'], Literal(f, datatype=XSD['decimal'])), | |
454 ] | |
455 | |
456 def watchPrefixes(self): | |
457 # these uris will become dynamic! see note on watchPrefixes | |
458 # about eliminating it. | |
459 return [(self.uri, ROOM['temperatureF']), | |
460 ] | |
259 | 461 |
462 @register | |
463 class RgbPixels(DeviceType): | |
464 """chain of ws2812 rgb pixels on pin GPIO18""" | |
465 deviceType = ROOM['RgbPixels'] | |
466 | |
467 def hostStateInit(self): | |
468 px = self.graph.value(self.uri, ROOM['pixels']) | |
469 self.pixelUris = list(self.graph.items(px)) | |
470 self.values = dict((uri, Literal('#000000')) for uri in self.pixelUris) | |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
471 colorOrder, stripType = self.getColorOrder(self.graph, self.uri) |
259 | 472 self.replace = {'ledArray': 'leds_%s' % self.pinNumber, |
473 'ledCount': len(self.pixelUris), | |
474 'pin': self.pinNumber, | |
475 'ledType': 'WS2812', | |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
476 'colorOrder': colorOrder |
259 | 477 } |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
478 self.neo = neopixel.Adafruit_NeoPixel(len(self.values), pin=18, strip_type=stripType) |
259 | 479 self.neo.begin() |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
480 |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
481 def getColorOrder(self, graph, uri): |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
482 colorOrder = graph.value(uri, ROOM['colorOrder'], |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
483 default=ROOM['ledColorOrder/RGB']) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
484 head, tail = str(colorOrder).rsplit('/', 1) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
485 if head != str(ROOM['ledColorOrder']): |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
486 raise NotImplementedError('%r colorOrder %r' % (uri, colorOrder)) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
487 stripType = getattr(neopixel.ws, 'WS2811_STRIP_%s' % tail) |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
488 return colorOrder, stripType |
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
489 |
259 | 490 def _rgbFromHex(self, h): |
491 rrggbb = h.lstrip('#') | |
492 return [int(x, 16) for x in [rrggbb[0:2], rrggbb[2:4], rrggbb[4:6]]] | |
493 | |
494 def sendOutput(self, statements): | |
495 px, pred, color = statements[0] | |
496 if pred != ROOM['color']: | |
497 raise ValueError(pred) | |
498 rgb = self._rgbFromHex(color) | |
499 if px not in self.values: | |
500 raise ValueError(px) | |
501 self.values[px] = Literal(color) | |
266
1ae78fe6f36e
neopixel lib change. factor out pwm setup and try another pwm rate
drewp@bigasterisk.com
parents:
262
diff
changeset
|
502 self.neo.setPixelColorRGB(self.pixelUris.index(px), rgb[0], rgb[1], rgb[2]) |
259 | 503 self.neo.show() |
504 | |
505 def hostStatements(self): | |
506 return [(uri, ROOM['color'], hexCol) | |
507 for uri, hexCol in self.values.items()] | |
508 | |
509 def outputPatterns(self): | |
510 return [(px, ROOM['color'], None) for px in self.pixelUris] | |
511 | |
512 def outputWidgets(self): | |
513 return [{ | |
514 'element': 'output-rgb', | |
515 'subj': px, | |
516 'pred': ROOM['color'], | |
517 } for px in self.pixelUris] | |
518 | |
205 | 519 |
182 | 520 def makeDevices(graph, board, pi): |
521 out = [] | |
522 for dt in sorted(_knownTypes, key=lambda cls: cls.__name__): | |
523 out.extend(dt.findInstances(graph, board, pi)) | |
524 return out | |
525 |