]> sigrok.org Git - libsigrokdecode.git/blame - decoders/guess_bitrate/pd.py
configure.ac: Also check for Python 3.7.
[libsigrokdecode.git] / decoders / guess_bitrate / pd.py
CommitLineData
56c93143
UH
1##
2## This file is part of the libsigrokdecode project.
3##
3aa79ba3 4## Copyright (C) 2013-2016 Uwe Hermann <uwe@hermann-uwe.de>
56c93143
UH
5##
6## This program is free software; you can redistribute it and/or modify
7## it under the terms of the GNU General Public License as published by
8## the Free Software Foundation; either version 2 of the License, or
9## (at your option) any later version.
10##
11## This program is distributed in the hope that it will be useful,
12## but WITHOUT ANY WARRANTY; without even the implied warranty of
13## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14## GNU General Public License for more details.
15##
16## You should have received a copy of the GNU General Public License
4539e9ca 17## along with this program; if not, see <http://www.gnu.org/licenses/>.
56c93143
UH
18##
19
56c93143
UH
20import sigrokdecode as srd
21
21cda951
UH
22class SamplerateError(Exception):
23 pass
24
56c93143 25class Decoder(srd.Decoder):
3aa79ba3 26 api_version = 3
56c93143
UH
27 id = 'guess_bitrate'
28 name = 'Guess bitrate'
29 longname = 'Guess bitrate/baudrate'
30 desc = 'Guess the bitrate/baudrate of a UART (or other) protocol.'
31 license = 'gplv2+'
32 inputs = ['logic']
33 outputs = ['guess_bitrate']
6a15597a 34 channels = (
56c93143 35 {'id': 'data', 'name': 'Data', 'desc': 'Data line'},
da9bcbd9
BV
36 )
37 annotations = (
38 ('bitrate', 'Bitrate / baudrate'),
39 )
56c93143
UH
40
41 def putx(self, data):
42 self.put(self.ss_edge, self.samplenum, self.out_ann, data)
43
92b7b49f 44 def __init__(self):
10aeb8ea
GS
45 self.reset()
46
47 def reset(self):
56c93143 48 self.ss_edge = None
56c93143 49
e2ff870e 50 def start(self):
be465111 51 self.out_ann = self.register(srd.OUTPUT_ANN)
56c93143 52
e2ff870e
UH
53 def metadata(self, key, value):
54 if key == srd.SRD_CONF_SAMPLERATE:
35b380b1 55 self.samplerate = value
e2ff870e 56
3aa79ba3 57 def decode(self):
21cda951
UH
58 if not self.samplerate:
59 raise SamplerateError('Cannot decode without samplerate.')
56c93143 60
ce7018ca
GS
61 # Get the first edge on the data line.
62 self.wait({0: 'e'})
63 self.ss_edge = self.samplenum
64
65 # Get any subsequent edge on the data line. Get the smallest
66 # distance between any two transitions, assuming it corresponds
67 # to one bit time of the respective bitrate of the input stream.
68 # This heuristics keeps getting better for longer captures.
69 bitwidth = None
3aa79ba3 70 while True:
3aa79ba3 71 self.wait({0: 'e'})
56c93143 72
ce7018ca
GS
73 b = self.samplenum - self.ss_edge
74 if bitwidth is None or b < bitwidth:
75 bitwidth = b
76 bitrate = int(float(self.samplerate) / float(b))
77 self.putx([0, ['%d' % bitrate]])
78 self.ss_edge = self.samplenum