X-Git-Url: https://sigrok.org/gitweb/?p=libsigrokdecode.git;a=blobdiff_plain;f=decoders%2Fguess_bitrate%2Fpd.py;h=92d35208628dfcbd6fbb648db70e463246048ae2;hp=e86316fb86d7c384c2fb74ec0a1869c1d22aebdc;hb=92b7b49f6964f57a7d6fc4473645c993cfa4ba52;hpb=12851357e784b893e24880efc6cd22a0cbcc64ce diff --git a/decoders/guess_bitrate/pd.py b/decoders/guess_bitrate/pd.py index e86316f..92d3520 100644 --- a/decoders/guess_bitrate/pd.py +++ b/decoders/guess_bitrate/pd.py @@ -20,6 +20,9 @@ import sigrokdecode as srd +class SamplerateError(Exception): + pass + class Decoder(srd.Decoder): api_version = 2 id = 'guess_bitrate' @@ -39,7 +42,7 @@ class Decoder(srd.Decoder): def putx(self, data): self.put(self.ss_edge, self.samplenum, self.out_ann, data) - def __init__(self, **kwargs): + def __init__(self): self.olddata = None self.ss_edge = None self.first_transition = True @@ -50,11 +53,11 @@ class Decoder(srd.Decoder): def metadata(self, key, value): if key == srd.SRD_CONF_SAMPLERATE: - self.samplerate = value; + self.samplerate = value def decode(self, ss, es, data): - if self.samplerate is None: - raise Exception("Cannot decode without samplerate.") + if not self.samplerate: + raise SamplerateError('Cannot decode without samplerate.') for (self.samplenum, pins) in data: data = pins[0] @@ -64,22 +67,21 @@ class Decoder(srd.Decoder): continue # Initialize first self.olddata with the first sample value. - if self.olddata == None: + if self.olddata is None: self.olddata = data continue # Get the smallest distance between two transitions # and use that to calculate the bitrate/baudrate. - if self.first_transition == True: + if self.first_transition: self.ss_edge = self.samplenum self.first_transition = False else: b = self.samplenum - self.ss_edge - if self.bitwidth == None or b < self.bitwidth: + if self.bitwidth is None or b < self.bitwidth: self.bitwidth = b bitrate = int(float(self.samplerate) / float(b)) self.putx([0, ['%d' % bitrate]]) self.ss_edge = self.samplenum self.olddata = data -