- def decode(self, data):
- """Nintendo Wii Nunchuk decoder"""
-
- out = []
- o = {}
-
- # We should accept a list of samples and iterate...
- # for sample in sampleiter(data['data'], self.unitsize):
- for p in example_packets:
-
- # TODO: Eliminate the need for ord().
- # s = ord(sample.data)
-
- if p['type'] == 'S': # TODO: Handle 'Sr' here, too?
- self.state = self.START
-
- elif p['type'] == 'Sr':
- pass # FIXME
-
- elif p['type'] == 'AR':
- # TODO: Error/Warning, not supported, I think.
- pass
-
- elif p['type'] == 'AW':
- # The Wii Nunchuk always has slave address 0x54.
- # TODO: Handle this stuff more correctly.
- if p['data'] == 0x54:
- pass # TODO
- else:
- pass # TODO: What to do here? Ignore? Error?
-
- elif p['type'] == 'DR' and self.state == self.INITIALIZED:
- if self.databytecount == 0:
- self.sx = p['data']
- elif self.databytecount == 1:
- self.sy = p['data']
- elif self.databytecount == 2:
- self.ax = p['data'] << 2
- elif self.databytecount == 3:
- self.ay = p['data'] << 2
- elif self.databytecount == 4:
- self.az = p['data'] << 2
- elif self.databytecount == 5:
- self.bz = (p['data'] & (1 << 0)) >> 0
- self.bc = (p['data'] & (1 << 1)) >> 1
- self.ax |= (p['data'] & (3 << 2)) >> 2
- self.ay |= (p['data'] & (3 << 4)) >> 4
- self.az |= (p['data'] & (3 << 6)) >> 6
- # del o
- o = {'type': 'D', 'range': (0, 0), 'data': []}
- o['data'] = [self.sx, self.sy, self.ax, self.ay, \
- self.az, self.bz, self.bc]
- # sx = sy = ax = ay = az = bz = bc = 0
- else:
- pass # TODO
-
- if 0 <= self.databytecount <= 5:
- self.databytecount += 1
-
- # TODO: If 6 bytes read -> save and reset
-
- # TODO
- elif p['type'] == 'DR' and self.state != self.INITIALIZED:
- pass
-
- elif p['type'] == 'DW':
- if p['data'] == 0x40 and self.state == self.START:
- self.state = self.INIT
- elif p['data'] == 0x00 and self.state == self.INIT:
- o = {'type': 'I', 'range': (0, 0), 'data': []}
- o['data'] = [0x40, 0x00]
- out.append(o)
- self.state = self.INITIALIZED
- else:
- pass # TODO
-
- elif p['type'] == 'P':
- out.append(o)
- self.state = self.INITIALIZED
- self.databytecount = 0
-
- sigrok.put(out)
-
-import sigrok
-
-sigrok.register(Decoder)
+ def decode(self, ss, es, data):
+
+ cmd, databyte, ack_bit = data
+
+ if cmd == 'START': # TODO: Handle 'Sr' here, too?
+ self.state = START
+
+ elif cmd == 'START_REPEAT':
+ pass # FIXME
+
+ elif cmd == 'ADDRESS_READ':
+ # TODO: Error/Warning, not supported, I think.
+ pass
+
+ elif cmd == 'ADDRESS_WRITE':
+ # The Wii Nunchuk always has slave address 0x54.
+ # TODO: Handle this stuff more correctly.
+ if databyte == 0x54:
+ pass # TODO
+ else:
+ pass # TODO: What to do here? Ignore? Error?
+
+ elif cmd == 'DATA_READ' and self.state == INITIALIZED:
+ if self.databytecount == 0:
+ self.sx = databyte
+ elif self.databytecount == 1:
+ self.sy = databyte
+ elif self.databytecount == 2:
+ self.ax = databyte << 2
+ elif self.databytecount == 3:
+ self.ay = databyte << 2
+ elif self.databytecount == 4:
+ self.az = databyte << 2
+ elif self.databytecount == 5:
+ self.bz = (databyte & (1 << 0)) >> 0
+ self.bc = (databyte & (1 << 1)) >> 1
+ self.ax |= (databyte & (3 << 2)) >> 2
+ self.ay |= (databyte & (3 << 4)) >> 4
+ self.az |= (databyte & (3 << 6)) >> 6
+
+ d = 'sx = 0x%02x, sy = 0x%02x, ax = 0x%02x, ay = 0x%02x, ' \
+ 'az = 0x%02x, bz = 0x%02x, bc = 0x%02x' % (self.sx, \
+ self.sy, self.ax, self.ay, self.az, self.bz, self.bc)
+ self.put(ss, es, self.out_ann, [0, [d]])
+
+ self.sx = self.sy = self.ax = self.ay = self.az = 0
+ self.bz = self.bc = 0
+ else:
+ pass # TODO
+
+ if 0 <= self.databytecount <= 5:
+ self.databytecount += 1
+
+ # TODO: If 6 bytes read -> save and reset
+
+ # TODO
+ elif cmd == 'DATA_READ' and self.state != INITIALIZED:
+ pass
+
+ elif cmd == 'DATA_WRITE':
+ if self.state == IDLE:
+ self.state = INITIALIZED
+ return
+
+ if databyte == 0x40 and self.state == START:
+ self.state = INIT
+ elif databyte == 0x00 and self.state == INIT:
+ self.put(ss, es, self.out_ann, [0, ['Initialize nunchuk']])
+ self.state = INITIALIZED
+ else:
+ pass # TODO
+
+ elif cmd == 'STOP':
+ self.state = INITIALIZED
+ self.databytecount = 0