summaryrefslogtreecommitdiff
path: root/decoders/spi.py
blob: 731c77cc3c8ae0214697cd5e0b99c4b13b5b2da6 (plain)
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
##
## This file is part of the sigrok project.
##
## Copyright (C) 2011 Gareth McMullin <gareth@blacksphere.co.nz>
##
## 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
##

class Sample():
    def __init__(self, data):
        self.data = data
    def probe(self, probe):
        s = ord(self.data[probe / 8]) & (1 << (probe % 8))
        return True if s else False

def sampleiter(data, unitsize):
    for i in range(0, len(data), unitsize):
        yield(Sample(data[i:i+unitsize]))

class Decoder():
    name = 'SPI Decoder'
    desc = '...desc...'
    longname = '...longname...'
    longdesc = '...longdesc...'
    author = 'Gareth McMullin'
    email = 'gareth@blacksphere.co.nz'
    license = 'gplv2+'
    inputs = ['logic']
    outputs = ['spi']
    # Probe names with a set of defaults
    probes = {'sdata':0, 'sck':1}
    options = {}

    def __init__(self):
        self.probes = Decoder.probes.copy()
        self.oldsck = True
        self.rxcount = 0
        self.rxdata = 0
        self.bytesreceived = 0

    def start(self, metadata):
        self.unitsize = metadata["unitsize"]

    def report(self):
        return "SPI: %d bytes received" % self.bytesreceived

    def decode(self, data):
        # We should accept a list of samples and iterate...
        for sample in sampleiter(data["data"], self.unitsize):

            sck = sample.probe(self.probes["sck"])
            # Sample SDATA on rising SCK
            if sck == self.oldsck:
                continue
            self.oldsck = sck
            if not sck:
                continue

            # If this is first bit, save timestamp
            if self.rxcount == 0:
                self.time = data["time"]
            # Receive bit into our shift register
            sdata = sample.probe(self.probes["sdata"])
            if sdata:
                self.rxdata |= 1 << (7 - self.rxcount)
            self.rxcount += 1
            # Continue to receive if not a byte yet
            if self.rxcount != 8:
                continue
            # Received a byte, pass up to sigrok
            outdata = {"time":self.time,
                "duration":data["time"] + data["duration"] - self.time,
                "data":self.rxdata,
                "display":("%02X" % self.rxdata),
                "type":"spi",
            }
            sigrok.put(outdata)
            # Reset decoder state
            self.rxdata = 0
            self.rxcount = 0
            # Keep stats for summary
            self.bytesreceived += 1

if __name__ == "__main__":
    data = open("spi_dump.bin").read()

    # dummy class to keep Decoder happy for test
    class Sigrok():
        def put(self, data):
            print "\t", data
    sigrok = Sigrok()

    dec = Decoder(driver='ols', unitsize=1, starttime=0)
    dec.decode({"time":0, "duration":len(data), "data":data, "type":"logic"})

    print dec.summary()
else:
    import sigrok

#Tested with:
#  sigrok-cli -d 0:samplerate=1000000:rle=on --time=1s -p 1,2 -a spidec