X-Git-Url: https://sigrok.org/gitweb/?p=libsigrokdecode.git;a=blobdiff_plain;f=decoders%2Fspi.py;h=383206b1914a6069e642f435516b594d575d9304;hp=152b2638da9a407cc2e370b9fd34ff03959725d6;hb=16ba97cf2252bc9543f70a0cf3e7d4b2ec6e7ee1;hpb=3b24e3781a0374ce2d66f1d3de59a5a5d10d657a diff --git a/decoders/spi.py b/decoders/spi.py index 152b263..383206b 100644 --- a/decoders/spi.py +++ b/decoders/spi.py @@ -1,109 +1,114 @@ +## +## This file is part of the sigrok project. +## +## Copyright (C) 2011 Gareth McMullin +## +## 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 __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])) + for i in range(0, len(data), unitsize): + yield(Sample(data[i:i+unitsize])) class Decoder(): - # Probe names with a set of defaults - probes = {'sdata':0, 'sck':1} - - def __init__(self, metadata, **kwargs): - # Metadata comes in here, we don't care for now - #print kwargs - self.unitsize = metadata['unitsize'] - - self.probes = Decoder.probes - self.oldsck = True - self.rxcount = 0 - self.rxdata = 0 - self.bytesreceived = 0 - - def summary(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", - } - print outdata - sigrok.put(outdata) - # Reset decoder state - self.rxdata = 0 - self.rxcount = 0 - # Keep stats for summary - self.bytesreceived += 1 - - -register = { - 'id': 'spi', - 'name': 'SPI Decoder', - 'longname': '...', - 'desc': 'Decodes SPI frames', - 'longdesc': '...', - 'author': 'Gareth McMullin', - 'email': 'gareth@blacksphere.co.nz', - 'license': 'gplv2+', - 'in': ['logic'], - 'out': ['spi'], - 'probes': [ - # All probes. - ], - 'options': { - # No options so far. - }, - # 'start': start, - # 'report': report, -} - -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() + 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 + import sigrok #Tested with: # sigrok-cli -d 0:samplerate=1000000:rle=on --time=1s -p 1,2 -a spidec -