1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
##
## This file is part of the sigrok project.
##
## Copyright (C) 2010-2012 Uwe Hermann <uwe@hermann-uwe.de>
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
##
# Nintendo Wii Nunchuk protocol decoder
import sigrokdecode as srd
# States
IDLE = 0
START = 1
NUNCHUK_SLAVE = 2
INIT = 3
INITIALIZED = 4
class Decoder(srd.Decoder):
api_version = 1
id = 'nunchuk'
name = 'Nunchuk'
longname = 'Nintendo Wii Nunchuk'
desc = 'Decodes the Nintendo Wii Nunchuk I2C-based protocol.'
longdesc = '...'
license = 'gplv2+'
inputs = ['i2c']
outputs = ['nunchuck']
probes = []
optional_probes = [] # TODO
options = {}
annotations = [
['TODO', 'TODO'],
]
def __init__(self, **kwargs):
self.state = IDLE # TODO: Can we assume a certain initial state?
self.sx = self.sy = self.ax = self.ay = self.az = self.bz = self.bc = 0
self.databytecount = 0
def start(self, metadata):
# self.out_proto = self.add(srd.OUTPUT_PROTO, 'nunchuk')
self.out_ann = self.add(srd.OUTPUT_ANN, 'nunchuk')
def report(self):
pass
def decode(self, ss, es, data):
cmd, databyte = 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
|
